fix UI data for blocked connections

block.py

@@ -34,6 +34,7 @@ def block_conn(config, datafile, res):
     fprint("Searching block data")
     mydata = list()
     badapps = list()
+    badips = list()
     with open(find_data_file(datafile), newline='') as csvfile:
         csvreader = csv.reader(csvfile, delimiter=',', quotechar='|')
         
@@ -66,7 +67,8 @@ def block_conn(config, datafile, res):
             if srcip == badsrcip or destip == baddestip and not pid in badapps:
                 fprint("FLAG " + srcip + " " + destip + " " + str(pid))
                 badapps.append(pid)
+                badips.append(baddestip)
                 #fprint(badapps)
                 #fprint("FLAG " + srcip + " " + destip + " " + str(pid))
                 #kill(pid)
-    return badapps
+    return badapps, badips

ippigeon.py

@@ -82,17 +82,29 @@ def login_done(res):
 def blockdata_done(res):
     global settings
     fprint("FINISHED downloading block data")
+    tmpkill = settings["kill"]
+    settings["kill"] = False
     #block_res = pool.apply_async(block.block_conn, (config, datafile, res, settings))
-    block_res = block.block_conn(config, datafile, res)
+    block_pids, block_ips = block.block_conn(config, datafile, res)
     tmplist = settings["badapps"]
-    for x in block_res:
+    
+    for x in block_pids:
         fprint(x)
         if not x in tmplist:
             tmplist.append(x)
-        
     settings["badapps"] = tmplist
     fprint(settings["badapps"])
     
+    tmplist = settings["badips"]
+    for x in block_ips:
+        fprint(x)
+        if not x in tmplist:
+            tmplist.append(x)
+    settings["badips"] = tmplist
+    fprint(settings["badips"])
+
+    settings["kill"] = tmpkill
+        
 def killall():
     kids = active_children()
     for kid in kids:
@@ -152,10 +164,12 @@ def mainloop(pool):
         settings["block"] = False
 
     if settings["kill"] == True:
-        for x in settings["badapps"]:
-            kill(x)
+        tmplist = settings["badapps"]
         settings["badapps"] = list()
 
+        for x in tmplist:
+            kill(x)
+
     sleep(interval / (interval * config["core"]["clockspeed"]))
     counter += 1
     if counter == interval * config["core"]["clockspeed"]:
@@ -202,6 +216,7 @@ if __name__ == '__main__':
             settings["continueui"] = False
             settings["killbox"] = list()
             settings["badapps"] = list()
+            settings["badips"] = list()
             settings["block"] = False
             settings["kill"] = False
             killme = manager.Value('d', 0)

panel.py

@@ -187,11 +187,12 @@ class ServerPanel(wx.Panel):
         else:
             fprint("Unable to delete")
         for i in range(len(TEST_FILE)):
+            
             idx = 0
-            for pid in settings["badapps"]:
+            for ip in settings["badips"]:
                 #fprint(pid)
                 idx = i
-                if str(TEST_FILE.iloc[i, 5]) == str(pid): # "bad" pid, highlight in red at the top
+                if str(TEST_FILE.iloc[i, 3]).find(ip) >= 0 and str(TEST_FILE.iloc[i, 4]).find("TIME_WAIT") < 0: # "bad" pid, highlight in red at the top
                     idx = self.list_ctrl.InsertItem(0, TEST_FILE.iloc[i, 0])
                     self.list_ctrl.SetItemBackgroundColour(idx, wx.Colour(200, 51, 51))
                     break
@@ -202,8 +203,8 @@ class ServerPanel(wx.Panel):
             
                     #fprint("Got " + TEST_FILE.iloc[i, 5])
             for j in range(1, 6):
-                fprint(str(idx) + " " + str(TEST_FILE.iloc[i, 0]))
-                self.list_ctrl.SetItem(idx, j, str(TEST_FILE.iloc[idx, j]))
+                #fprint(str(idx) + " " + str(TEST_FILE.iloc[i, 0]))
+                self.list_ctrl.SetItem(idx, j, str(TEST_FILE.iloc[i, j]))
 
                 #fprint(i, j, TEST_FILE.iloc[i, j])
         #self.SetSizer(self.main_sizer)

util.py

@@ -42,6 +42,7 @@ def time():
 
 def kill(pid):
     setup_child()
+    if pid > 0:
         fprint("Killing PID " + str(pid))
         os.kill(int(pid), 9)
         fprint("Signal 9 sent to PID " + str(pid))