def __init__(self):
Gtk.Window.__init__(self,
title="Network Traffic Based Software Generation")
# screenWidth = Gdk.Screen.get_width(Gdk.Screen.get_default())
# screenHeigth = Gdk.Screen.get_height(Gdk.Screen.get_default())
# Gtk.Window.resize(self,screenWidth/10,screenHeigth/3)
self.PCAPWidget = ""
self.PacketList = {}
self.ProtocolList = {}
self.filterExpression = "le Filter"
self.PacketAreaBox = PacketArea(self)
mainGrid = Gtk.Grid()
# mainGrid.set_column_spacing(4)
# mainGrid.set_row_spacing(4)
# mainGrid.set_row_homogeneous(False)
# mainGrid.set_hexpand(False)
self.add(mainGrid)
#
# for x in range(4):
# mainGrid.set_row_baseline_position(x,Gtk.BaselinePosition.CENTER)
#Start of Header
HeaderBox = Header(self, "Network Traffic Based Software Generation")
HeaderBox.addButton("State Machine")
HeaderBox.addButton("Create Session")
HeaderBox.addButton("Open Session")
HeaderBox.addButton("Close Session")
HeaderBox.addButton("Switch Workspace")
HeaderBox.addButton("Open PCAP")
HeaderBox.addButton("Terminal")
HeaderBox.showButtons()
HeaderButtons = HeaderBox._buttons
HeaderButtons[6].connect("clicked", HeaderBox.Terminal_clicked)
HeaderButtons[5].connect("clicked", HeaderBox.PCAP_clicked)
HeaderButtons[4].connect("clicked", HeaderBox.Workspace_clicked)
HeaderButtons[3].connect("clicked", HeaderBox.OpenSession_clicked)
HeaderButtons[2].connect("clicked", HeaderBox.OpenSession_clicked)
HeaderButtons[1].connect("clicked", HeaderBox.NewSession_clicked)
HeaderButtons[0].connect("clicked", HeaderBox.StateMachine_clicked)
mainGrid.attach(HeaderBox, 0, 0, 4, 1)
#End of Header
#Image Containing the status.
StatusIndicator = Gtk.Image.new_from_file("statusIndicator.png")
StatusFrame = Gtk.Frame()
StatusBox = Gtk.Box(spacing=0)
StatusBox.add(StatusFrame)
mainGrid.attach(StatusBox, 0, 1, 4, 1)
StatusBox.pack_start(StatusIndicator, False, True, 0)
#End of Image containing the Status.
#Start of Sessions View
SessionsBox = Session()
SessionsBox.addSession("Session A")
SessionsBox.addSession("Session B")
SessionsBox.addSession("Session C")
SessionsBox.showSessions()
SessionsBox.set_vexpand(True)
mainGrid.attach(SessionsBox, 0, 2, 1, 1)
#End of Sessions View
#Start of PDML View
# PDMLFrame = Gtk.Frame()
PDMLBox = Gtk.Box(orientation=Gtk.Orientation.VERTICAL, spacing=0)
PDMLLabel = Gtk.Label("PDML View")
PDMLLabel.modify_bg(Gtk.StateType.NORMAL,
Gdk.color_parse("light blue"))
PDMLBox.pack_start(PDMLLabel, False, False, 0)
PDMLViewBox = PDMLView(self)
PDMLBox.pack_start(PDMLViewBox, False, False, 0)
PDMLBox.add(self.PacketAreaBox)
PDMLBox.set_hexpand(True)
# PDMLBox.pack_start(self.PacketAreaBox,True,True,0)
mainGrid.attach(PDMLBox, 1, 2, 4, 1)
#End of PDML View
#Start of Tagging View
TagBox = TagArea()
TagBox.addCombobox("Saved Tag")
TagBox.addField("Tag Name")
TagBox.addField("Tagged Field")
TagBox.addField("Tag Description")
TagBox.populateFields()
TagBox.showFields()
TagBox.addButton("Update")
TagBox.addButton("Cancel")
TagBox.showButtons()
mainGrid.attach(TagBox, 0, 3, 1, 1)
#End of Tagging View
#Start of field area View
FieldAreaBox = FieldArea()
mainGrid.attach(FieldAreaBox, 1, 3, 1, 1)
FieldAreaBox.addField("icmp.type", "Type 8 [Echo ping request]", 1, 34,
"8", "8", 2)
#Start of Message Type View
MessageTypeBox = Gtk.Box(orientation=Gtk.Orientation.VERTICAL,
spacing=0)
mainGrid.attach(MessageTypeBox, 2, 3, 2, 1)
MessageTypeFrame = Gtk.Frame()
MessageTypeBox.add(MessageTypeFrame)
#
#
#
#IMPORTANT: Notebook requires all views to be inside a box. This section is all the views until I find a way to have it set up more efficiently.
#
#Start of New/Modify
MessageTypeTabs = Gtk.Notebook()
NewModifyView = Gtk.Box(orientation=Gtk.Orientation.VERTICAL,
spacing=0)
InstructionsLabel = Gtk.Label(
"To create a new message type, please enter a message type name and select message type field value pair(s). \n To Update/delete an existing message type, please select from the existing message type first \n and the previously selected name and field value pair(s) will be pre-populated."
)
InstructionsLabel.set_line_wrap(True)
ExistingMessageTypeBox = Gtk.Box(spacing=1)
ExistingMessageLabel = Gtk.Label("Existing Message Type")
ExistingMessageTypeBox.pack_start(ExistingMessageLabel, False, False,
0)
ExistingMessageType = Gtk.Entry()
ExistingMessageTypeBox.pack_start(ExistingMessageType, False, False, 0)
MessageTypeNameBox = Gtk.Box(spacing=2)
MessageTypeLabel = Gtk.Label("Message Type Name")
MessageTypeNameBox.pack_start(MessageTypeLabel, False, False, 0)
MessageTypeName = Gtk.Entry()
MessageTypeNameBox.pack_start(MessageTypeName, False, False, 0)
MessageTypePairsBox = Gtk.Box(spacing=2)
MessageTypePairsLabel = Gtk.Label("Message Type Field\n Value Pair(s)")
MessageTypePairsBox.pack_start(MessageTypePairsLabel, False, False, 0)
ValuePairs = Gtk.Entry()
MessageTypePairsBox.pack_start(ValuePairs, False, False, 0)
NewModifyView.pack_start(InstructionsLabel, False, False, 0)
NewModifyView.pack_start(ExistingMessageTypeBox, False, False, 0)
NewModifyView.pack_start(MessageTypeNameBox, False, False, 0)
NewModifyView.pack_start(MessageTypePairsBox, False, False, 0)
SelectBothField = Gtk.CheckButton(
label="Select both field name and value")
SelectFieldNameOnly = Gtk.CheckButton(label="Select field name only")
NewModifyView.pack_start(SelectBothField, False, False, 0)
NewModifyView.pack_start(SelectFieldNameOnly, False, False, 0)
OperationButtons = Gtk.Box(spacing=2)
SaveButtonMessageType = Gtk.Button(label="Save")
DeleteButtonMessageType = Gtk.Button(label="Delete")
ClearButtonMessageType = Gtk.Button(label="Clear")
OperationButtons.pack_start(SaveButtonMessageType, False, False, 0)
OperationButtons.pack_start(DeleteButtonMessageType, False, False, 0)
OperationButtons.pack_start(ClearButtonMessageType, False, False, 0)
NewModifyView.pack_start(OperationButtons, False, False, 0)
MessageTypeTabs.append_page(NewModifyView)
MessageTypeTabs.set_tab_label_text(NewModifyView, "New/Modify")
#End of New/Modify
#Start of Template
TemplateBox = Gtk.Box(orientation=Gtk.Orientation.VERTICAL, spacing=0)
ExistingMessageTypebox = Gtk.Box(spacing=0)
ExistingMessageTypeLabel = Gtk.Label("Existing Message Type")
ExistingMessageType = Gtk.Entry()
ExistingMessageTypebox.pack_start(ExistingMessageTypeLabel, False,
False, 0)
ExistingMessageTypebox.pack_start(ExistingMessageType, False, False, 0)
TemplateBox.pack_start(ExistingMessageTypebox, False, False, 0)
CycleThroughPackets = Gtk.Button(label="Cycle Through Packets")
TemplateBox.pack_start(CycleThroughPackets, False, False, 0)
MessageTypeTemplateBox = Gtk.Box(spacing=0)
MessageTypeTemplateLabel = Gtk.Label(
"Message Type \n Template Field Value \n Pair(s)")
PacketAreaField = Gtk.ListStore(str)
PacketAreaField.append(["No packets"])
PacketView = Gtk.TreeView(PacketAreaField)
Render_Packet = Gtk.CellRendererText()
PacketViewCol = Gtk.TreeViewColumn("Packets", Render_Packet, text=0)
PacketView.append_column(PacketViewCol)
MessageTypeTemplateBox.pack_start(MessageTypeTemplateLabel, False,
False, 0)
MessageTypeTemplateBox.pack_start(PacketView, False, False, 0)
TemplateBox.pack_start(MessageTypeTemplateBox, False, False, 0)
OperationButtonsTypeTemplate = Gtk.Box(spacing=2)
SaveButtonTypeTemplate = Gtk.Button(label="Save")
ClearButtonTypeTemplate = Gtk.Button(label="Clear")
OperationButtonsTypeTemplate.pack_start(SaveButtonTypeTemplate, False,
False, 0)
OperationButtonsTypeTemplate.pack_start(ClearButtonTypeTemplate, False,
False, 0)
TemplateBox.pack_start(OperationButtonsTypeTemplate, False, False, 0)
MessageTypeTabs.append_page(TemplateBox)
MessageTypeTabs.set_tab_label_text(TemplateBox, "Template")
#End of Template
#Start of Equivalency
EquivalencyBox = Gtk.Box(orientation=Gtk.Orientation.VERTICAL,
spacing=0)
FieldEquivalencyEqualBox = Gtk.Box(spacing=0)
FieldEquivalencyLabel = Gtk.Label("Field Equivalency")
FieldName = Gtk.Entry()
FieldOR = Gtk.Label("OR")
MessageType = Gtk.Entry()
FieldEqual = Gtk.Label("=")
FieldEquivalencyEqualBox.pack_start(FieldEquivalencyLabel, False,
False, 0)
FieldEquivalencyEqualBox.pack_start(FieldName, False, False, 0)
FieldEquivalencyEqualBox.pack_start(FieldOR, False, False, 0)
FieldEquivalencyEqualBox.pack_start(MessageType, False, False, 0)
FieldEquivalencyEqualBox.pack_start(FieldEqual, False, False, 0)
EquivalencyBox.pack_start(FieldEquivalencyEqualBox, False, False, 0)
SecondFieldEquivalencyEqualBox = Gtk.Box(spacing=0)
FieldName = Gtk.Entry()
FieldOR = Gtk.Label("OR")
IndentLabel = Gtk.Label(" ")
MessageType = Gtk.Entry()
SecondFieldEquivalencyEqualBox.pack_start(IndentLabel, False, False, 0)
SecondFieldEquivalencyEqualBox.pack_start(FieldName, False, False, 0)
SecondFieldEquivalencyEqualBox.pack_start(FieldOR, False, False, 0)
SecondFieldEquivalencyEqualBox.pack_start(MessageType, False, False, 0)
FieldPLUS = Gtk.Label("+")
SecondFieldEquivalencyEqualBox.pack_start(FieldPLUS, False, False, 0)
EquivalencyBox.pack_start(SecondFieldEquivalencyEqualBox, False, False,
0)
OperationButtonsEquivalency = Gtk.Box(spacing=2)
SaveButtonEquivalency = Gtk.Button(label="Save")
ClearButtonEquivalency = Gtk.Button(label="Clear")
OperationButtonsEquivalency.pack_start(SaveButtonEquivalency, False,
False, 0)
OperationButtonsEquivalency.pack_start(ClearButtonEquivalency, False,
False, 0)
EquivalencyBox.pack_start(OperationButtonsEquivalency, False, False, 0)
MessageTypeTabs.append_page(EquivalencyBox)
MessageTypeTabs.set_tab_label_text(EquivalencyBox, "Equivalency")
#End of Equivalency
#Start of Generation
GenerationBox = Gtk.Box(orientation=Gtk.Orientation.VERTICAL,
spacing=0)
ExistingMessageTypeInGeneration = Gtk.Box(spacing=0)
ExistingMessageTypeLabel = Gtk.Label("Existing Message Type")
ExistingMessageTypeInGeneration.pack_start(ExistingMessageTypeLabel,
False, False, 0)
MessageTypeName = Gtk.Entry()
ExistingMessageTypeInGeneration.pack_start(MessageTypeName, False,
False, 0)
GenerationBox.pack_start(ExistingMessageTypeInGeneration, False, False,
0)
MessageTemplateFormatBox = Gtk.Box(spacing=0)
MessageTemplateFormatLabel = Gtk.Label(
"Message Template \n Output Format")
MessageTemplateFormat = Gtk.Entry()
MessageTemplateFormatBox.pack_start(MessageTemplateFormatLabel, False,
False, 0)
MessageTemplateFormatBox.pack_start(MessageTemplateFormat, False,
False, 0)
GenerationBox.pack_start(MessageTemplateFormatBox, False, False, 0)
MessageTemplateNameBox = Gtk.Box(spacing=0)
MessageTemplateNameLabel = Gtk.Label("Message Template \n Name")
MessageTemplateName = Gtk.Entry()
MessageTemplateNameBox.pack_start(MessageTemplateNameLabel, False,
False, 0)
MessageTemplateNameBox.pack_start(MessageTemplateName, False, False, 0)
GenerationBox.pack_start(MessageTemplateNameBox, False, False, 0)
ButtonsMessageType = Gtk.Box(spacing=0)
GenerateButton = Gtk.Button(label="Generate")
ClearMessageType = Gtk.Button(label="Clear")
ButtonsMessageType.pack_start(GenerateButton, False, False, 0)
ButtonsMessageType.pack_start(ClearMessageType, False, False, 0)
GenerationBox.pack_start(ButtonsMessageType, False, False, 0)
MessageTypeTabs.append_page(GenerationBox)
MessageTypeTabs.set_tab_label_text(GenerationBox, "Generation")
#End of Generation
MessageTypeArea = Gtk.Label("Message Type Area")
MessageTypeArea.set_text("Message Type")
MessageTypeBox.pack_start(MessageTypeArea, False, False, 0)
MessageTypeBox.pack_start(MessageTypeTabs, False, False, 0)
