Beispiel #1
0
    def __init__(self, parent=None, **kwargs):
        Flowchart.__init__(self, **kwargs)
        self.parent = parent
        self.inputNode.graphicsItem().hide()
        self.outputNode.graphicsItem().hide()
        self._nodeCopyPasteBuffer = None

        self.widget().chartWidget.hoverOver = self.hoverOver  # this thing is not working....
Beispiel #2
0
    def __init__(self, parent=None, **kwargs):
        Flowchart.__init__(self, **kwargs)
        self.parent = parent
        self.inputNode.graphicsItem().hide()
        self.outputNode.graphicsItem().hide()
        self._nodeCopyPasteBuffer = None

        self.widget(
        ).chartWidget.hoverOver = self.hoverOver  # this thing is not working....
Beispiel #3
0
    def __init__(self, prototxt=None, weights=None):
        """
        A flowchart that enables visualizing, editing, and running a caffe neural network model

        :param prototxt: string
            Optional path to a caffe prototxt file, which will be used as the initial model for the network
        :param weights: string
            Optional path to a caffe caffemodel file, which will be used as the initial weights for the network
        """

        self.nodeList = []
        self.displayNodes = {}
        self.netNeedsUpdate = True
        self.netNeedsEval = True
        self.holdUpdateConnects = True
        self.nextData = None
        Flowchart.__init__(self, terminals={
            'dataIn': {'io': 'in'},
            'dataOut': {'io': 'out'}
        })

        if prototxt is None:
            self.proto = NetProto()
            self.layerList = self.proto.layer
            self.plotList = {}
        else:
            self.proto = parsePrototxt(prototxt, 'net')
            self.layerList = self.proto.layer
            # for old proto format
            if len(self.layerList) == 0:
                self.layerList = self.proto.layers

        self.weights = weights

        for proto in self.layerList:
            print proto

        self.setupNodes()

        fcw = self.widget().chartWidget
        fcw.viewDock.sizePolicy().setHorizontalPolicy(QtGui.QSizePolicy.Maximum)

        fcw.moveDock(fcw.hoverDock, 'bottom', fcw.viewDock)
        fcw.moveDock(fcw.selDock, 'right', fcw.hoverDock)
Beispiel #4
0
    def create_widgets(self, parent):
        vbox = QtGui.QVBoxLayout()
        self.label = QtGui.QLabel()
        self.label.setText("Constellation")
        self.label.setFixedHeight(20)

        self.fc = Flowchart(terminals={'Connection': {'io': 'in'}})
        self.fc._nodes['Input'].close()
        self.fc._nodes['Output'].close()

        self.view = self.fc.widget()
        wig = self.view.chartWidget
        self.view.chartWidget.hoverDock.hide()
        self.view.chartWidget.selectedTree.hide()
        self.view.chartWidget.selDock.label.setText("Selected Comp.")

        # add selection label
        self.selection_label = self.view.chartWidget.selDescLabel
        self.selection_label.setText("")
        self.selection_label.setFixedHeight(15)

        # add information table
        self.info_table = GBuilder().table(self.view.chartWidget.selDock, 1, 2,
                                           ["Settings", "Value"])
        for r in range(self.info_table.rowCount()):
            self.info_table.setRowHeight(r, 20)
        self.info_table.horizontalHeader().hide()
        self.info_table.verticalHeader().hide()
        self.info_table.setSelectionBehavior(QAbstractItemView.SelectRows)
        self.info_table.setSortingEnabled(True)
        #         self.view.chartWidget.selInfoLayout.addWidget(self.selection_label)
        self.view.chartWidget.selInfoLayout.addWidget(self.info_table)

        # override on selection
        self.view.chartWidget._scene.selectionChanged.connect(
            self._selection_changed)

        vbox.addWidget(self.label)
        vbox.addWidget(wig)
        self.setLayout(vbox)
Beispiel #5
0
 def createNode(self, nodeType, name=None, pos=None):
     """
     overrides Flowchart.createNode to pick sensible names that caffe will accept
     """
     if name is None:
         name = nodeType.lower()
     if name in self._nodes:
         name2 = name
         n = 1
         while name2 in self._nodes:
             name2 = "%s.%d" % (name, n)
             n += 1
         name = name2
     return Flowchart.createNode(self, nodeType, name=name, pos=pos)
    def create_widgets(self, parent):
        vbox = QtGui.QVBoxLayout()        
        self.label = QtGui.QLabel()
        self.label.setText("Constellation")
        self.label.setFixedHeight(20)
        
        self.fc = Flowchart(terminals={'Connection': {'io': 'in'}})        
        self.fc._nodes['Input'].close()
        self.fc._nodes['Output'].close()

        self.view = self.fc.widget()
        wig = self.view.chartWidget
        self.view.chartWidget.hoverDock.hide()
        self.view.chartWidget.selectedTree.hide()        
        self.view.chartWidget.selDock.label.setText("Selected Comp.")
        
        # add selection label
        self.selection_label = self.view.chartWidget.selDescLabel
        self.selection_label.setText("")
        self.selection_label.setFixedHeight(15)
        
        # add information table
        self.info_table = GBuilder().table(self.view.chartWidget.selDock, 1, 2, ["Settings", "Value"])
        for r in range(self.info_table.rowCount()):
            self.info_table.setRowHeight(r, 20)
        self.info_table.horizontalHeader().hide()
        self.info_table.verticalHeader().hide()
        self.info_table.setSelectionBehavior(QAbstractItemView.SelectRows); 
        self.info_table.setSortingEnabled(True)
#         self.view.chartWidget.selInfoLayout.addWidget(self.selection_label)
        self.view.chartWidget.selInfoLayout.addWidget(self.info_table)
        
        # override on selection
        self.view.chartWidget._scene.selectionChanged.connect(self._selection_changed)
        
        vbox.addWidget(self.label)
        vbox.addWidget(wig)
        self.setLayout(vbox)
Beispiel #7
0
class ConstellationViewPluginGUI(QWidget):
    def __init__(self, parent):
        QWidget.__init__(self, parent)

        self._radius = 350
        self._gw_distance = 2 * self._radius
        self._bus_distance = 4 * self._radius
        self._last_bus_pos = [0, 0]
        self._last_bus_steps = [0, 0]
        self._bus_node = {}
        self._bus_position = {}

        self._ecu_terminals = {}
        self._ecu_position = {}
        self._ecu_node = {}

        self._show_dict = {}

        self.lastClicked = []
        self._all_points = []
        self.create_widgets(parent)

        self.map_points = {}

        self.COLOR_ECU_AUTH = (255, 0, 0)
        self.COLOR_STR_AUTH = (0, 255, 0)
        self.COLOR_SIMPLE = (0, 0, 255)
        self.COLOR_PROCESS = (123, 123, 0)
        self.COLOR_PROCESS_2 = (0, 123, 123)

        self._init_categories()
        self._mode = 'LW_AUTH'

        self._pts_ecu = {}

    def _clicked(self, plot, points):

        for p in self.lastClicked:
            p.resetPen()

        try:
            info = points[0].data()
        except:
            info = False

        if info:
            try:
                info[5]
            except:
                info += [0, 0, 0, 0, 0]

            if len(str(info[2])) > 100:
                showos = info[2][:99]
            else:
                showos = info[2]

            self.label.setText(
                "ECU:  %s\t\t Time:%s \t\nMessageID:  %s \tMessage:  %s \t\nSize:  %s \t\t\tCorresponding ID:  %s \tStream ID:  %s"
                % (info[0], info[-1], self._id_to_str(
                    info[1]), showos, info[3], info[6], info[5]))

        for p in points:
            p.setPen('b', width=2)
        self.lastClicked = points

    def _init_categories(self):

        # TESLA
        self.tesla_time_sync_send = [
            MonitorTags.CP_SEND_SYNC_MESSAGE,
            MonitorTags.CP_SEND_SYNC_RESPONSE_MESSAGE
        ]
        self.tesla_time_sync_rec = [
            MonitorTags.CP_RECEIVE_SYNC_RESPONSE_MESSAGE
        ]

        self.tesla_setup_send = [
            MonitorTags.CP_ENCRYPTED_EXCHANGE_FIRST_KEY_KN
        ]
        self.tesla_setup_rec = [MonitorTags.CP_RECEIVED_EXCHANGE_FIRST_KEY_KN]

        self.tesla_simple_message_send = [
            MonitorTags.CP_MACED_TRANSMIT_MESSAGE
        ]
        self.tesla_simple_message_rec = [
            MonitorTags.CP_BUFFERED_SIMPLE_MESSAGE
        ]

        self.tesla_message_authenticated = [
            MonitorTags.CP_RETURNED_AUTHENTICATED_SIMPLE_MESSAGE
        ]

        self.tesla = self.tesla_time_sync_send + self.tesla_time_sync_rec + self.tesla_setup_send + self.tesla_setup_rec + self.tesla_simple_message_send + self.tesla_simple_message_rec + self.tesla_message_authenticated

        # TLS
        self.hand_shake_tag_server_send = [MonitorTags.CP_SEND_SERVER_HELLO, MonitorTags.CP_SEND_SERVER_CERTIFICATE, MonitorTags.CP_SEND_SERVER_KEYEXCHANGE, MonitorTags.CP_SEND_CERTIFICATE_REQUEST, MonitorTags.CP_SEND_SERVER_HELLO_DONE, \
                                           MonitorTags.CP_CLIENT_FINISHED_GENERATED_HASH_PRF]
        self.hand_shake_tag_server_rec = [MonitorTags.CP_RECEIVE_CLIENT_CERTIFICATE, MonitorTags.CP_RECEIVE_CLIENT_KEYEXCHANGE, MonitorTags.CP_RECEIVE_CERTIFICATE_VERIFY, MonitorTags.CP_RECEIVED_CHANGE_CIPHER_SPEC, \
                                          MonitorTags.CP_RECEIVE_CLIENT_FINISHED]
        self.hand_shake_tag_server_process = [MonitorTags.CP_CLIENT_CERTIFICATE_VALIDATED, MonitorTags.CP_DECRYPTED_CLIENT_KEYEXCHANGE, MonitorTags.CP_DECRYPTED_CERTIFICATE_VERIFY , MonitorTags.CP_GENERATED_MASTER_SECRET_CERT_VERIFY, \
                                              MonitorTags.CP_CLIENT_FINISHED_HASHED_COMPARISON_HASH , MonitorTags.CP_CLIENT_AUTHENTICATED]

        self.hand_shake_tag_client_send = [MonitorTags.CP_SEND_CLIENT_HELLO, MonitorTags.CP_SEND_CLIENT_CERTIFICATE , MonitorTags.CP_INIT_SEND_CLIENT_KEYEXCHANGE , \
            MonitorTags.CP_SEND_CIPHER_SPEC , MonitorTags.CP_GENERATED_HASH_FROM_PRF_CLIENT_FINISHED, MonitorTags.CP_GENERATED_HASH_FROM_PRF_SERVER_FINISHED]


        self.hand_shake_tag_client_rec = [MonitorTags.CP_RECEIVE_CLIENT_HELLO, MonitorTags.CP_RECEIVE_SERVER_HELLO, MonitorTags.CP_RECEIVE_SERVER_CERTIFICATE , MonitorTags.CP_RECEIVE_SERVER_KEYEXCHANGE, \
                                          MonitorTags.CP_RECEIVE_CERTIFICATE_REQUEST, MonitorTags.CP_RECEIVE_SERVER_HELLO_DONE, MonitorTags.CP_RECEIVE_SERVER_FINISHED  ]

        self.hand_shake_tag_client_process = [MonitorTags.CP_SERVER_HELLO_DONE_VALIDATED_CERT, MonitorTags.CP_ENCRYPTED_CLIENT_KEYEXCHANGE , MonitorTags.CP_GENERATED_MASTERSEC_CLIENT_KEYEXCHANGE , MonitorTags.CP_INIT_SEND_CERTIFICATE_VERIFY, \
                                              MonitorTags.CP_ENCRYPTED_CERTIFICATE_VERIFY, MonitorTags.CP_INIT_CLIENT_FINISHED , MonitorTags.CP_HASHED_CLIENT_FINISHED, MonitorTags.CP_SERVER_FINISHED_HASHED_COMPARISON_HASH , \
                                              MonitorTags.CP_SERVER_FINISHED_GENERATED_HASH_PRF, MonitorTags.CP_INIT_SERVER_FINISHED , MonitorTags.CP_HASHED_SERVER_FINISHED, MonitorTags.CP_SERVER_AUTHENTICATED ]

        self.simple_tags_send = [MonitorTags.CP_SESSION_AVAILABLE_SEND_MESSAGE]
        self.simple_tags_rec = [MonitorTags.CP_RECEIVE_SIMPLE_MESSAGE]

        self.tls = self.hand_shake_tag_server_send + self.hand_shake_tag_server_rec + self.hand_shake_tag_server_process + self.hand_shake_tag_client_send + self.hand_shake_tag_client_rec + self.hand_shake_tag_client_process \
                + self.simple_tags_send + self.simple_tags_rec

        # authentication
        self.sec_mod_tags = [MonitorTags.CP_SEC_INIT_AUTHENTICATION, MonitorTags.CP_SEC_ECNRYPTED_CONFIRMATION_MESSAGE, MonitorTags.CP_SEC_COMPARED_HASH_REG_MSG, \
                             MonitorTags.CP_SEC_ENCRYPTED_DENY_MESSAGE, MonitorTags.CP_SEC_ENCRYPTED_GRANT_MESSAGE, MonitorTags.CP_SEC_DECRYPTED_REQ_MESSAGE, MonitorTags.CP_SEC_RECEIVE_REG_MESSAGE]

        self.authent_tags_send = [
            MonitorTags.CP_SEC_INIT_AUTHENTICATION,
            MonitorTags.CP_SEC_ECNRYPTED_CONFIRMATION_MESSAGE,
            MonitorTags.CP_ECU_SEND_REG_MESSAGE
        ]
        self.authent_tags_receive = [
            MonitorTags.CP_SEC_COMPARED_HASH_REG_MSG,
            MonitorTags.CP_ECU_VALIDATED_SEC_MOD_CERTIFICATE,
            MonitorTags.CP_ECU_DECRYPTED_CONF_MESSAGE
        ]

        self.author_tags_send = [
            MonitorTags.CP_SEC_ENCRYPTED_DENY_MESSAGE,
            MonitorTags.CP_SEC_ENCRYPTED_GRANT_MESSAGE,
            MonitorTags.CP_ECU_ENCRYPTED_REQ_MESSAGE
        ]
        self.author_tags_receive = [
            MonitorTags.CP_ECU_DECRYPTED_DENY_MESSAGE,
            MonitorTags.CP_ECU_DECRYPTED_GRANT_MESSAGE,
            MonitorTags.CP_SEC_DECRYPTED_REQ_MESSAGE
        ]

        self.simp_tags_send = [
            MonitorTags.CP_ECU_ENCRYPTED_SEND_SIMPLE_MESSAGE
        ]
        self.simp_tags_receive = [MonitorTags.CP_ECU_DECRYPTED_SIMPLE_MESSAGE]

        self.lw_auth = self.sec_mod_tags + self.authent_tags_send + self.authent_tags_receive + self.author_tags_send + self.author_tags_receive + self.simp_tags_send + self.simp_tags_receive

    def create_widgets(self, parent):
        vbox = QtGui.QVBoxLayout()
        self.label = QtGui.QLabel()
        self.label.setText("Constellation")
        self.label.setFixedHeight(20)

        self.fc = Flowchart(terminals={'Connection': {'io': 'in'}})
        self.fc._nodes['Input'].close()
        self.fc._nodes['Output'].close()

        self.view = self.fc.widget()
        wig = self.view.chartWidget
        self.view.chartWidget.hoverDock.hide()
        self.view.chartWidget.selectedTree.hide()
        self.view.chartWidget.selDock.label.setText("Selected Comp.")

        # add selection label
        self.selection_label = self.view.chartWidget.selDescLabel
        self.selection_label.setText("")
        self.selection_label.setFixedHeight(15)

        # add information table
        self.info_table = GBuilder().table(self.view.chartWidget.selDock, 1, 2,
                                           ["Settings", "Value"])
        for r in range(self.info_table.rowCount()):
            self.info_table.setRowHeight(r, 20)
        self.info_table.horizontalHeader().hide()
        self.info_table.verticalHeader().hide()
        self.info_table.setSelectionBehavior(QAbstractItemView.SelectRows)
        self.info_table.setSortingEnabled(True)
        #         self.view.chartWidget.selInfoLayout.addWidget(self.selection_label)
        self.view.chartWidget.selInfoLayout.addWidget(self.info_table)

        # override on selection
        self.view.chartWidget._scene.selectionChanged.connect(
            self._selection_changed)

        vbox.addWidget(self.label)
        vbox.addWidget(wig)
        self.setLayout(vbox)

    def save(self):

        data = {}

        bus_ecu_connections = self.bus_ecu_connections  # pur so uebernehmen
        bus_data = self._get_save_bus()
        ecu_data = self._get_save_ecu()

        data['bus_data'] = bus_data
        data['ecu_data'] = ecu_data
        data['ecu_positions'] = self._ecu_position
        data['bus_positions'] = self._bus_position
        data['bus_ecu_connections'] = bus_ecu_connections

        print("SSVR I")
        return data

    def load(self, data):

        self._ecu_position = data['ecu_positions']
        self._bus_position = data['bus_positions']
        bus_data = data['bus_data']
        ecu_data = data['ecu_data']
        bus_ecu_connections = data['bus_ecu_connections']

        # add busses
        self._add_buses_load(bus_data)

        # add ecus
        self._add_ecus_load(ecu_data, bus_ecu_connections)

    def update_gui(self, monitor_input_list):
        ''' this method is called exactly once and should show all 
            information
        '''

        # extract information
        try:
            constellation = monitor_input_list[0].data
        except:
            return
        ecu_groups = constellation[0]
        bus_groups = constellation[1]
        bus_ecu_connections = constellation[2]
        self.bus_ecu_connections = bus_ecu_connections

        # set cans
        self._add_buses(bus_groups)

        # add ecus
        self._add_ecus(ecu_groups, bus_ecu_connections)

    def _selection_changed(self):

        items = self.view.chartWidget._scene.selectedItems()

        if len(items) == 0:
            data = None
        else:
            item = items[0]

            if hasattr(item, 'node') and isinstance(item.node, Node):
                print(item.node)
                print("SELECTED %s" % item.node.settings_dict)
                self._show_node_information(item.node)

            else:
                data = None


#         self.selectedTree.setData(data, hideRoot=True)

    def _show_node_information(self, node):

        # hide all rows
        for r in range(self.info_table.rowCount()):
            self.info_table.setRowHidden(r, True)

        # add new information save dict[ecu_id][settings_id] = row_nr
        comp_id = node.settings_dict['comp_id']
        self.selection_label.setText("Selected ECU:      %s" % comp_id)

        for set_key in node.settings_dict.keys():

            # if already exists -> show its row and skip
            try:
                row_nr = self._show_dict[comp_id][set_key]
                self.info_table.setRowHidden(row_nr, False)
                continue
            except:
                pass

            # else add it to table and show row
            self.info_table.setRowCount(self.info_table.rowCount() + 1)
            row_nr = self.info_table.rowCount()
            item_1 = QTableWidgetItem()
            item_1.setData(QtCore.Qt.EditRole, set_key)
            item_2 = QTableWidgetItem()
            item_2.setText(str(node.settings_dict[set_key]))
            General().add_to_three_dict(self._show_dict, comp_id, set_key,
                                        row_nr)
            self.info_table.setItem(row_nr - 1, 0, item_1)
            self.info_table.setItem(row_nr - 1, 1, item_2)
            self.info_table.setRowHeight(row_nr - 1, 20)

    def _get_save_bus(self):

        buses = []
        for bus_id in self._bus_node:
            node = self._bus_node[bus_id]
            settings = node.get_string_settings()

            buses.append([bus_id, settings])
        return buses

    def _add_buses_load(self, bus_data):

        nr_buses = len(bus_data)

        for b_group in bus_data:
            bus_id = b_group[0]
            bus_settings = b_group[1]

            # place the busses
            [x, y] = self._bus_position[bus_id]

            node = InformationNode(bus_id)
            node.settings_dict = bus_settings
            self._bus_node[bus_id] = node
            self.fc.addNode(node, bus_id, [x, y])
            self._bus_position[bus_id] = [x, y]

    def _add_buses(self, bus_groups):
        first = True
        nr_buses = len(bus_groups)

        for b_group in bus_groups:
            bus = b_group[0]

            # place the busses
            if not first: [x, y] = self._next_bus_pos(nr_buses)
            else: [x, y] = [0, 0]
            first = False
            node = InformationNode(bus.comp_id)
            node.set_settings(bus)
            self._bus_node[bus.comp_id] = node
            self.fc.addNode(node, bus.comp_id, [x, y])
            self._bus_position[bus.comp_id] = [x, y]

    def _get_save_ecu(self):
        ecus = []
        for ecu_id in self._ecu_node:
            node = self._ecu_node[ecu_id]
            settings = node.get_string_settings()
            ecus.append([ecu_id, settings])
        return ecus

    def _add_ecus_load(self, ecu_data, bus_ecu_connections):
        ''' add all ecus to the respective buses '''
        ecu_dict = self._can_ecu_list_dictionary(bus_ecu_connections)

        for bus_id in ecu_dict:
            idx = 0
            ecu_list = ecu_dict[bus_id]
            for ecu_id in ecu_list:
                if isinstance(ecu_id, UUID):
                    ecu_id = "GW\n%s" % str(ecu_id.hex)

                ecu_dat = self._ecu_dat_from_id(ecu_id, ecu_data)

                ecu_id = ecu_dat[0]
                ecu_settings = ecu_dat[1]

                [x, y] = self._ecu_position[ecu_id]

                ecu_id_show = ecu_id

                # shorten name if to long
                if len(ecu_id) >= 13:
                    ecu_id_show = ecu_id[:13]

                # calculate positions
                idx += 1
                try:
                    # if this already exists gets a second connection
                    ecu_terminal = False
                    if ecu_id in self._ecu_node.keys():
                        ecu_terminal = self._ecu_node[ecu_id].addInput('C%s' %
                                                                       idx)

                    # create new node
                    if not ecu_terminal:
                        node = InformationNode(ecu_id_show)
                        # set information
                        node.settings_dict = ecu_settings
                        self.fc.addNode(node, ecu_id_show, [x, y])
                        self._ecu_position[ecu_id] = [x, y]
                        self._ecu_node[ecu_id] = node
                        ecu_terminal = node.addInput('C%s' % idx)

                    bus_terminal = self._bus_node[bus_id].addInput('C%s' % idx)
                except:
                    traceback.print_exc()

                # connector to bus
                try:
                    self.fc.connectTerminals(bus_terminal, ecu_terminal)
                except:
                    pass

    def _add_ecus(self, ecu_groups, bus_ecu_connections):
        ''' add all ecus to the respective buses '''
        ecu_dict = self._can_ecu_list_dictionary(bus_ecu_connections)

        for bus_id in ecu_dict:
            ecu_list = ecu_dict[bus_id]
            nr_ecus = len(ecu_list)
            pos_bus = self._bus_position[bus_id]
            idx = 0

            if bus_id == "CAN0":
                a = 0
            for ecu_id in ecu_list:
                ecu_id_show = ecu_id
                ecu = self._ecu_from_group(ecu_groups, ecu_id)

                [x, y] = self._next_ecu_pos(pos_bus, nr_ecus, idx)

                if isinstance(ecu_id, UUID):
                    ecu_id = "GW\n%s" % ecu_id.hex
                    [x, y] = self._get_gw_position(ecu)

                # shorten name if to long
                if len(ecu_id) >= 13:
                    ecu_id_show = ecu_id[:13]

                # calculate positions
                idx += 1
                try:
                    # if this already exists gets a second connection
                    ecu_terminal = False
                    if ecu_id in self._ecu_node.keys():
                        ecu_terminal = self._ecu_node[ecu_id].addInput('C%s' %
                                                                       idx)

                    # create new node
                    if not ecu_terminal:
                        node = InformationNode(ecu_id_show)
                        # set information
                        node.set_settings(ecu)
                        self.fc.addNode(node, ecu_id_show, [x, y])
                        self._ecu_position[ecu_id] = [x, y]
                        self._ecu_node[ecu_id] = node
                        ecu_terminal = node.addInput('C%s' % idx)

                    bus_terminal = self._bus_node[bus_id].addInput('C%s' % idx)
                except:
                    traceback.print_exc()

                # connector to bus
                try:
                    self.fc.connectTerminals(bus_terminal, ecu_terminal)
                except:
                    pass

    def _ecu_dat_from_id(self, ecu_id, ecu_data):

        for ecu_dat in ecu_data:
            if ecu_dat[0] == ecu_id:
                return ecu_dat
        return None

    def _ecu_from_group(self, ecu_groups, ecu_id):
        for ecu_lists in ecu_groups:
            ecu_list = ecu_lists[0]
            ecu_spec = ecu_lists[1]
            for ecu in ecu_list:
                if ecu.ecu_id == ecu_id:
                    return ecu
        return None

    def _get_gw_position(self, gateway):
        x_points = []
        y_points = []

        for bus in gateway.connected_bus:
            x_points.append(self._bus_position[bus.comp_id][0])
            y_points.append(self._bus_position[bus.comp_id][1])

        x = float(sum(x_points)) / float(len(x_points))
        y = float(sum(y_points)) / float(len(y_points))

        for bus in gateway.connected_bus:
            if self._bus_position[bus.comp_id][0] == x and self._bus_position[
                    bus.comp_id][1] == y:
                y += self._gw_distance

        return [x, y]

    def _can_ecu_list_dictionary(self, bus_ecu_connections):
        ''' creates a dictinary of form key: bus id and
            value list of ecu ids connected to the bus
        
            Input:     bus_ecu_connections     list
            Output:    ecu_dict                dictionary
        '''
        out_dict = {}
        for connection in bus_ecu_connections:

            General().force_add_dict_list(out_dict, connection[0],
                                          connection[1])

        return out_dict

    def _next_ecu_pos(self, pos_bus, nr_ecus, idx):
        ''' determines the next position of the ecu'''

        alpha = (360 / nr_ecus) * 1.00000001
        angle = idx * alpha
        # radius varies with the number of ECUs connected (less means closer)
        radius = (math.sqrt(2) * 190) / (2 * math.sin(2 * math.pi *
                                                      (alpha / 360)))

        angle /= 360
        angle *= 2 * math.pi

        x = radius * math.cos(angle) + pos_bus[0]
        y = radius * math.sin(angle) + pos_bus[1]

        return [x, y]

    def _next_bus_pos(self, nr_buses):
        ''' determines the next position of the bus'''

        if nr_buses < 5:
            [x, y] = self._last_bus_pos
            [lr, ud] = self._last_bus_steps
            x += self._bus_distance
            self._last_bus_pos[0] = x
            return [x, y]

        [x, y] = self._last_bus_pos
        [lr, ud] = self._last_bus_steps
        step = round(sqrt(nr_buses / 3))

        y += self._bus_distance

        if ud == step:
            y = 0
            ud = 0
            x += self._bus_distance
            self._last_bus_steps = [lr + 1, 0]
            lr += 1
        ud += 1
        self._last_bus_steps = [lr, ud]
        self._last_bus_pos = [x, y]
        return [x, y]

    def _is_sec_mod(self, ecu):
        try:
            ecu._SECMODULE
            return True
        except:
            pass
        return False
Beispiel #8
0
 def clear(self):
     self.nodeList = []
     Flowchart.clear(self)
class ConstellationViewPluginGUI(QWidget):
            
    def __init__(self, parent):
        QWidget.__init__(self, parent)
        
        self._radius = 350
        self._gw_distance = 2 * self._radius
        self._bus_distance = 4 * self._radius
        self._last_bus_pos = [0, 0]
        self._last_bus_steps = [0, 0]
        self._bus_node = {}
        self._bus_position = {}
        
        self._ecu_terminals = {}
        self._ecu_position = {}
        self._ecu_node = {}
        
        self._show_dict = {}
        
        
        self.lastClicked = []     
        self._all_points = []           
        self.create_widgets(parent)
        
        
        self.map_points = {}
        
        self.COLOR_ECU_AUTH = (255, 0, 0)
        self.COLOR_STR_AUTH = (0, 255, 0)
        self.COLOR_SIMPLE = (0, 0, 255)
        self.COLOR_PROCESS = (123, 123, 0)
        self.COLOR_PROCESS_2 = (0, 123, 123)
        
        self._init_categories()
        self._mode = 'LW_AUTH'
            
        self._pts_ecu = {} 
            
    def _clicked(self, plot, points):  
              
        for p in self.lastClicked:
            p.resetPen()
                    
        try: info = points[0].data()
        except: info = False
        
        if info:
            try: info[5]
            except: info += [0, 0, 0, 0, 0]
            
            if len(str(info[2])) > 100:
                showos = info[2][:99]
            else:
                showos = info[2]
            
            self.label.setText("ECU:  %s\t\t Time:%s \t\nMessageID:  %s \tMessage:  %s \t\nSize:  %s \t\t\tCorresponding ID:  %s \tStream ID:  %s" % (info[0], info[-1], self._id_to_str(info[1]), showos, info[3], info[6], info[5]))

        for p in points:
            p.setPen('b', width=2)
        self.lastClicked = points   
                
    def _init_categories(self):
        
        # TESLA
        self.tesla_time_sync_send = [MonitorTags.CP_SEND_SYNC_MESSAGE, MonitorTags.CP_SEND_SYNC_RESPONSE_MESSAGE]
        self.tesla_time_sync_rec = [MonitorTags.CP_RECEIVE_SYNC_RESPONSE_MESSAGE]
        
        self.tesla_setup_send = [MonitorTags.CP_ENCRYPTED_EXCHANGE_FIRST_KEY_KN]
        self.tesla_setup_rec = [MonitorTags.CP_RECEIVED_EXCHANGE_FIRST_KEY_KN]
        
        self.tesla_simple_message_send = [MonitorTags.CP_MACED_TRANSMIT_MESSAGE]
        self.tesla_simple_message_rec = [MonitorTags.CP_BUFFERED_SIMPLE_MESSAGE]
        
        self.tesla_message_authenticated = [MonitorTags.CP_RETURNED_AUTHENTICATED_SIMPLE_MESSAGE]
    
        self.tesla = self.tesla_time_sync_send + self.tesla_time_sync_rec + self.tesla_setup_send + self.tesla_setup_rec + self.tesla_simple_message_send + self.tesla_simple_message_rec + self.tesla_message_authenticated
        
        # TLS
        self.hand_shake_tag_server_send = [MonitorTags.CP_SEND_SERVER_HELLO, MonitorTags.CP_SEND_SERVER_CERTIFICATE, MonitorTags.CP_SEND_SERVER_KEYEXCHANGE, MonitorTags.CP_SEND_CERTIFICATE_REQUEST, MonitorTags.CP_SEND_SERVER_HELLO_DONE, \
                                           MonitorTags.CP_CLIENT_FINISHED_GENERATED_HASH_PRF]
        self.hand_shake_tag_server_rec = [MonitorTags.CP_RECEIVE_CLIENT_CERTIFICATE, MonitorTags.CP_RECEIVE_CLIENT_KEYEXCHANGE, MonitorTags.CP_RECEIVE_CERTIFICATE_VERIFY, MonitorTags.CP_RECEIVED_CHANGE_CIPHER_SPEC, \
                                          MonitorTags.CP_RECEIVE_CLIENT_FINISHED]
        self.hand_shake_tag_server_process = [MonitorTags.CP_CLIENT_CERTIFICATE_VALIDATED, MonitorTags.CP_DECRYPTED_CLIENT_KEYEXCHANGE, MonitorTags.CP_DECRYPTED_CERTIFICATE_VERIFY , MonitorTags.CP_GENERATED_MASTER_SECRET_CERT_VERIFY, \
                                              MonitorTags.CP_CLIENT_FINISHED_HASHED_COMPARISON_HASH , MonitorTags.CP_CLIENT_AUTHENTICATED]
                
        self.hand_shake_tag_client_send = [MonitorTags.CP_SEND_CLIENT_HELLO, MonitorTags.CP_SEND_CLIENT_CERTIFICATE , MonitorTags.CP_INIT_SEND_CLIENT_KEYEXCHANGE , \
            MonitorTags.CP_SEND_CIPHER_SPEC , MonitorTags.CP_GENERATED_HASH_FROM_PRF_CLIENT_FINISHED, MonitorTags.CP_GENERATED_HASH_FROM_PRF_SERVER_FINISHED]
        
        
        self.hand_shake_tag_client_rec = [MonitorTags.CP_RECEIVE_CLIENT_HELLO, MonitorTags.CP_RECEIVE_SERVER_HELLO, MonitorTags.CP_RECEIVE_SERVER_CERTIFICATE , MonitorTags.CP_RECEIVE_SERVER_KEYEXCHANGE, \
                                          MonitorTags.CP_RECEIVE_CERTIFICATE_REQUEST, MonitorTags.CP_RECEIVE_SERVER_HELLO_DONE, MonitorTags.CP_RECEIVE_SERVER_FINISHED  ]
        
        self.hand_shake_tag_client_process = [MonitorTags.CP_SERVER_HELLO_DONE_VALIDATED_CERT, MonitorTags.CP_ENCRYPTED_CLIENT_KEYEXCHANGE , MonitorTags.CP_GENERATED_MASTERSEC_CLIENT_KEYEXCHANGE , MonitorTags.CP_INIT_SEND_CERTIFICATE_VERIFY, \
                                              MonitorTags.CP_ENCRYPTED_CERTIFICATE_VERIFY, MonitorTags.CP_INIT_CLIENT_FINISHED , MonitorTags.CP_HASHED_CLIENT_FINISHED, MonitorTags.CP_SERVER_FINISHED_HASHED_COMPARISON_HASH , \
                                              MonitorTags.CP_SERVER_FINISHED_GENERATED_HASH_PRF, MonitorTags.CP_INIT_SERVER_FINISHED , MonitorTags.CP_HASHED_SERVER_FINISHED, MonitorTags.CP_SERVER_AUTHENTICATED ]
        
        self.simple_tags_send = [MonitorTags.CP_SESSION_AVAILABLE_SEND_MESSAGE]
        self.simple_tags_rec = [ MonitorTags.CP_RECEIVE_SIMPLE_MESSAGE  ]
                
        self.tls = self.hand_shake_tag_server_send + self.hand_shake_tag_server_rec + self.hand_shake_tag_server_process + self.hand_shake_tag_client_send + self.hand_shake_tag_client_rec + self.hand_shake_tag_client_process \
                + self.simple_tags_send + self.simple_tags_rec 
                
        # authentication
        self.sec_mod_tags = [MonitorTags.CP_SEC_INIT_AUTHENTICATION, MonitorTags.CP_SEC_ECNRYPTED_CONFIRMATION_MESSAGE, MonitorTags.CP_SEC_COMPARED_HASH_REG_MSG, \
                             MonitorTags.CP_SEC_ENCRYPTED_DENY_MESSAGE, MonitorTags.CP_SEC_ENCRYPTED_GRANT_MESSAGE, MonitorTags.CP_SEC_DECRYPTED_REQ_MESSAGE, MonitorTags.CP_SEC_RECEIVE_REG_MESSAGE]
            
        self.authent_tags_send = [MonitorTags.CP_SEC_INIT_AUTHENTICATION, MonitorTags.CP_SEC_ECNRYPTED_CONFIRMATION_MESSAGE, MonitorTags.CP_ECU_SEND_REG_MESSAGE]
        self.authent_tags_receive = [MonitorTags.CP_SEC_COMPARED_HASH_REG_MSG, MonitorTags.CP_ECU_VALIDATED_SEC_MOD_CERTIFICATE, MonitorTags.CP_ECU_DECRYPTED_CONF_MESSAGE]
                            
        self.author_tags_send = [MonitorTags.CP_SEC_ENCRYPTED_DENY_MESSAGE, MonitorTags.CP_SEC_ENCRYPTED_GRANT_MESSAGE, MonitorTags.CP_ECU_ENCRYPTED_REQ_MESSAGE]
        self.author_tags_receive = [MonitorTags.CP_ECU_DECRYPTED_DENY_MESSAGE, MonitorTags.CP_ECU_DECRYPTED_GRANT_MESSAGE, MonitorTags.CP_SEC_DECRYPTED_REQ_MESSAGE]
        
        self.simp_tags_send = [MonitorTags.CP_ECU_ENCRYPTED_SEND_SIMPLE_MESSAGE]
        self.simp_tags_receive = [MonitorTags.CP_ECU_DECRYPTED_SIMPLE_MESSAGE]
             
        self.lw_auth = self.sec_mod_tags + self.authent_tags_send + self.authent_tags_receive + self.author_tags_send + self.author_tags_receive + self.simp_tags_send + self.simp_tags_receive 
            
    def create_widgets(self, parent):
        vbox = QtGui.QVBoxLayout()        
        self.label = QtGui.QLabel()
        self.label.setText("Constellation")
        self.label.setFixedHeight(20)
        
        self.fc = Flowchart(terminals={'Connection': {'io': 'in'}})        
        self.fc._nodes['Input'].close()
        self.fc._nodes['Output'].close()

        self.view = self.fc.widget()
        wig = self.view.chartWidget
        self.view.chartWidget.hoverDock.hide()
        self.view.chartWidget.selectedTree.hide()        
        self.view.chartWidget.selDock.label.setText("Selected Comp.")
        
        # add selection label
        self.selection_label = self.view.chartWidget.selDescLabel
        self.selection_label.setText("")
        self.selection_label.setFixedHeight(15)
        
        # add information table
        self.info_table = GBuilder().table(self.view.chartWidget.selDock, 1, 2, ["Settings", "Value"])
        for r in range(self.info_table.rowCount()):
            self.info_table.setRowHeight(r, 20)
        self.info_table.horizontalHeader().hide()
        self.info_table.verticalHeader().hide()
        self.info_table.setSelectionBehavior(QAbstractItemView.SelectRows); 
        self.info_table.setSortingEnabled(True)
#         self.view.chartWidget.selInfoLayout.addWidget(self.selection_label)
        self.view.chartWidget.selInfoLayout.addWidget(self.info_table)
        
        # override on selection
        self.view.chartWidget._scene.selectionChanged.connect(self._selection_changed)
        
        vbox.addWidget(self.label)
        vbox.addWidget(wig)
        self.setLayout(vbox)
    
    def save(self):
        
        data = {}
        
        bus_ecu_connections = self.bus_ecu_connections  # pur so uebernehmen
        bus_data = self._get_save_bus()
        ecu_data = self._get_save_ecu()
                
        data['bus_data'] = bus_data
        data['ecu_data'] = ecu_data
        data['ecu_positions'] = self._ecu_position
        data['bus_positions'] = self._bus_position
        data['bus_ecu_connections'] = bus_ecu_connections

        print("SSVR I")
        return data
    
    def load(self, data):
        
        self._ecu_position = data['ecu_positions']
        self._bus_position = data['bus_positions']
        bus_data = data['bus_data']
        ecu_data = data['ecu_data']
        bus_ecu_connections = data['bus_ecu_connections']
        
        # add busses
        self._add_buses_load(bus_data)
    
        # add ecus
        self._add_ecus_load(ecu_data, bus_ecu_connections)
         
    def update_gui(self, monitor_input_list):
        ''' this method is called exactly once and should show all 
            information
        '''
        
        # extract information
        try: constellation = monitor_input_list[0].data
        except: return
        ecu_groups = constellation[0]
        bus_groups = constellation[1]        
        bus_ecu_connections = constellation[2]
        self.bus_ecu_connections = bus_ecu_connections
        
        # set cans
        self._add_buses(bus_groups)
        
        # add ecus
        self._add_ecus(ecu_groups, bus_ecu_connections)
             
    
    def _selection_changed(self):
        
        items = self.view.chartWidget._scene.selectedItems()

        if len(items) == 0:
            data = None
        else:
            item = items[0]

            if hasattr(item, 'node') and isinstance(item.node, Node):
                print(item.node)
                print("SELECTED %s" % item.node.settings_dict)
                self._show_node_information(item.node)

            else:
                data = None
        
#         self.selectedTree.setData(data, hideRoot=True)
    
    def _show_node_information(self, node):
        
        # hide all rows
        for r in range(self.info_table.rowCount()):
            self.info_table.setRowHidden(r, True)
        
        # add new information save dict[ecu_id][settings_id] = row_nr
        comp_id = node.settings_dict['comp_id']      
        self.selection_label.setText("Selected ECU:      %s" % comp_id)

        for set_key in node.settings_dict.keys():
            
            # if already exists -> show its row and skip
            try: 
                row_nr = self._show_dict[comp_id][set_key]
                self.info_table.setRowHidden(row_nr, False)
                continue
            except: pass
                        
            # else add it to table and show row
            self.info_table.setRowCount(self.info_table.rowCount() + 1)
            row_nr = self.info_table.rowCount()          
            item_1 = QTableWidgetItem()
            item_1.setData(QtCore.Qt.EditRole, set_key)  
            item_2 = QTableWidgetItem()
            item_2.setText(str(node.settings_dict[set_key]));  
            General().add_to_three_dict(self._show_dict, comp_id, set_key, row_nr)            
            self.info_table.setItem(row_nr - 1, 0, item_1)
            self.info_table.setItem(row_nr - 1, 1, item_2)
            self.info_table.setRowHeight(row_nr - 1, 20)
    
    def _get_save_bus(self):
        
        buses = []
        for bus_id in self._bus_node:
            node = self._bus_node[bus_id]
            settings = node.get_string_settings()
            
            buses.append([bus_id, settings])
        return buses
       
    def _add_buses_load(self, bus_data):
        
        nr_buses = len(bus_data)        
      
        
        for b_group in bus_data:
            bus_id = b_group[0]
            bus_settings = b_group[1]
             
            # place the busses
            [x, y] = self._bus_position[bus_id]
            
            node = InformationNode(bus_id)
            node.settings_dict = bus_settings
            self._bus_node[bus_id] = node
            self.fc.addNode(node, bus_id, [x, y])         
            self._bus_position[bus_id] = [x, y]
            
    def _add_buses(self, bus_groups):
        first = True
        nr_buses = len(bus_groups)
        
        for b_group in bus_groups:
            bus = b_group[0]
                         
            # place the busses
            if not first: [x, y] = self._next_bus_pos(nr_buses)
            else: [x, y] = [0, 0]
            first = False
            node = InformationNode(bus.comp_id)
            node.set_settings(bus)
            self._bus_node[bus.comp_id] = node
            self.fc.addNode(node, bus.comp_id, [x, y])         
            self._bus_position[bus.comp_id] = [x, y]

    def _get_save_ecu(self):
        ecus = []
        for ecu_id in self._ecu_node:
            node = self._ecu_node[ecu_id]
            settings = node.get_string_settings()
            ecus.append([ecu_id, settings])
        return ecus
        
    def _add_ecus_load(self, ecu_data, bus_ecu_connections):
        ''' add all ecus to the respective buses '''
        ecu_dict = self._can_ecu_list_dictionary(bus_ecu_connections)        
        
        for bus_id in ecu_dict:
            idx = 0    
            ecu_list = ecu_dict[bus_id]
            for ecu_id in ecu_list:
                if isinstance(ecu_id, UUID): ecu_id = "GW\n%s" % str(ecu_id.hex)
                
                ecu_dat = self._ecu_dat_from_id(ecu_id, ecu_data)
                
                ecu_id = ecu_dat[0]
                ecu_settings = ecu_dat[1]
                
                [x, y] = self._ecu_position[ecu_id]
                
                ecu_id_show = ecu_id
                                    
                # shorten name if to long
                if len(ecu_id) >= 13:
                    ecu_id_show = ecu_id[:13]
                    
                # calculate positions       
                idx += 1         
                try:
                    # if this already exists gets a second connection
                    ecu_terminal = False
                    if ecu_id in self._ecu_node.keys():
                        ecu_terminal = self._ecu_node[ecu_id].addInput('C%s' % idx)                                                
                        
                    # create new node
                    if not ecu_terminal:
                        node = InformationNode(ecu_id_show)
                        # set information
                        node.settings_dict = ecu_settings
                        self.fc.addNode(node, ecu_id_show, [x, y])         
                        self._ecu_position[ecu_id] = [x, y]
                        self._ecu_node[ecu_id] = node
                        ecu_terminal = node.addInput('C%s' % idx)
                    
                    bus_terminal = self._bus_node[bus_id].addInput('C%s' % idx)
                except:
                    traceback.print_exc()
                
                # connector to bus
                try:
                    self.fc.connectTerminals(bus_terminal, ecu_terminal)
                except:
                    pass

    def _add_ecus(self, ecu_groups, bus_ecu_connections):
        ''' add all ecus to the respective buses '''
        ecu_dict = self._can_ecu_list_dictionary(bus_ecu_connections)        
        
        for bus_id in ecu_dict:
            ecu_list = ecu_dict[bus_id]
            nr_ecus = len(ecu_list)
            pos_bus = self._bus_position[bus_id]
            idx = 0
            
            if bus_id == "CAN0":
                a = 0
            for ecu_id in ecu_list:
                ecu_id_show = ecu_id
                ecu = self._ecu_from_group(ecu_groups, ecu_id)
                
                [x, y] = self._next_ecu_pos(pos_bus, nr_ecus, idx)
                
                if isinstance(ecu_id, UUID):
                    ecu_id = "GW\n%s" % ecu_id.hex                    
                    [x, y] = self._get_gw_position(ecu)
                    
                # shorten name if to long
                if len(ecu_id) >= 13:
                    ecu_id_show = ecu_id[:13]
                    
                # calculate positions
                idx += 1
                try:
                    # if this already exists gets a second connection
                    ecu_terminal = False
                    if ecu_id in self._ecu_node.keys():
                        ecu_terminal = self._ecu_node[ecu_id].addInput('C%s' % idx)                                                
                        
                    # create new node
                    if not ecu_terminal:
                        node = InformationNode(ecu_id_show)
                        # set information
                        node.set_settings(ecu)
                        self.fc.addNode(node, ecu_id_show, [x, y])         
                        self._ecu_position[ecu_id] = [x, y]
                        self._ecu_node[ecu_id] = node
                        ecu_terminal = node.addInput('C%s' % idx)
                    
                    bus_terminal = self._bus_node[bus_id].addInput('C%s' % idx)
                except:
                    traceback.print_exc()
                
                # connector to bus
                try:
                    self.fc.connectTerminals(bus_terminal, ecu_terminal)
                except:
                    pass
    
    def _ecu_dat_from_id(self, ecu_id, ecu_data):    
        
        for ecu_dat in ecu_data:
            if ecu_dat[0] == ecu_id:
                return ecu_dat
        return None
            
    def _ecu_from_group(self, ecu_groups, ecu_id):
        for ecu_lists in ecu_groups:
            ecu_list = ecu_lists[0]
            ecu_spec = ecu_lists[1]
            for ecu in ecu_list:
                if ecu.ecu_id == ecu_id:
                    return ecu
        return None
            
    def _get_gw_position(self, gateway):
        x_points = []
        y_points = []
        
        for bus in gateway.connected_bus:            
            x_points.append(self._bus_position[bus.comp_id][0])
            y_points.append(self._bus_position[bus.comp_id][1])        
            
        x = float(sum(x_points)) / float(len(x_points))
        y = float(sum(y_points)) / float(len(y_points))
        
        for bus in gateway.connected_bus:    
            if self._bus_position[bus.comp_id][0] == x and self._bus_position[bus.comp_id][1] == y:
                y += self._gw_distance
        
        return [x, y]
        
                
    def _can_ecu_list_dictionary(self, bus_ecu_connections):
        ''' creates a dictinary of form key: bus id and
            value list of ecu ids connected to the bus
        
            Input:     bus_ecu_connections     list
            Output:    ecu_dict                dictionary
        '''
        out_dict = {}
        for connection in bus_ecu_connections:
            
            General().force_add_dict_list(out_dict, connection[0], connection[1])
            
        return out_dict
    
    def _next_ecu_pos(self, pos_bus, nr_ecus, idx):
        ''' determines the next position of the ecu'''
        
        
        alpha = (360 / nr_ecus) * 1.00000001
        angle = idx * alpha
        # radius varies with the number of ECUs connected (less means closer)
        radius = (math.sqrt(2) * 190) / (2 * math.sin(2 * math.pi * (alpha / 360)))  
        
        angle /= 360
        angle *= 2 * math.pi
        
        x = radius * math.cos(angle) + pos_bus[0]
        y = radius * math.sin(angle) + pos_bus[1]
                   
        return [x, y]
    
    def _next_bus_pos(self, nr_buses):
        ''' determines the next position of the bus'''
        
        if nr_buses < 5:
            [x, y] = self._last_bus_pos      
            [lr, ud] = self._last_bus_steps
            x += self._bus_distance
            self._last_bus_pos[0] = x
            return [x, y]
        
        [x, y] = self._last_bus_pos      
        [lr, ud] = self._last_bus_steps
        step = round(sqrt(nr_buses / 3))
        
        y += self._bus_distance
        
        if ud == step:
            y = 0
            ud = 0
            x += self._bus_distance
            self._last_bus_steps = [lr + 1, 0]       
            lr += 1
        ud += 1
        self._last_bus_steps = [lr, ud]
        self._last_bus_pos = [x, y]
        return [x, y]
                    
    def _is_sec_mod(self, ecu):
        try:   
            ecu._SECMODULE
            return True              
        except:
            pass
        return False