Method update

src/ETW/real-time/ProvGraph.py:156–260

(self, anomaly_cutoff, topK = 20)

    #             sim_list.append()

    def update(self, anomaly_cutoff, topK = 20):
        self.lock.acquire()
#        self.G = nx.compose(self.G,self.TmpG)
        self.G = self.TmpG.copy()
        self.TmpG.clear()
        self.lock.release()

        nodes = self.G.nodes()
        pnode = []
        for node in nodes:
            if self.GetNodeType(node) == NODE_TYPE.PROCESS and self.GetNodeCmd(node) != '':
                pnode.append(node)

        self.node_set |= set(pnode)
        update_node_list = self.caculate_anomaly_score(pnode,anomaly_cutoff)


        if update_node_list == -1:
            self.thread_lock.release()
            return
            
        relabel_dic = {}
        for node in self.G.nodes():
            origion = node
            while node in self.taylor_map:
                node = self.taylor_map[node]
            if origion != node:
                self.taylor_map[origion] = node
                relabel_dic[origion] = node
        print(len(relabel_dic))
        self.G = nx.relabel_nodes(self.G, relabel_dic, copy=False)
        update_node_list = list(update_node_list)
        for i,node in enumerate(update_node_list):
            if node in relabel_dic:
                update_node_list[i] = relabel_dic[node]
        update_node_list = set(update_node_list)

        # remove_edge_list = []

        # for e in self.G.edges:
        #     e_type = self.G.edges[e]['type']
        #     if ETYPE[e_type] in IGNTYPE:
        #         remove_edge_list.append(e)

        # self.G.remove_edges_from(remove_edge_list)

        # remove_node_list = []
        # cron_list = []
        # for node in self.G.nodes:
        #     attr = self.GetNodeName(node)
        #     if attr == "":
        #         remove_node_list.append(node)
        #     if attr == "cron" or attr == "bash" or attr == "sh" or attr == "sshd":
        #     if attr == "cron":
        #         cron_list.append(node)
        
        self.G = nx.DiGraph(self.G)
        # cron_process = set()