You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
87 lines
2.4 KiB
Python
87 lines
2.4 KiB
Python
from util import find_data_file
|
|
from util import setup_child
|
|
from util import fprint
|
|
from util import run_cmd
|
|
from util import win32
|
|
from util import linux
|
|
from util import kill
|
|
import util
|
|
import time
|
|
import csv
|
|
import ssh
|
|
|
|
def get_blocklist(config, appendbad):
|
|
setup_child()
|
|
fprint("Downloading deny list from server")
|
|
data = ssh.check_for_file(config, "BadIPs.csv", "receive")
|
|
#fprint(data.stdout)
|
|
csvreader = csv.reader(data.stdout.split("\n"), delimiter=',', quotechar='|')
|
|
data2 = list()
|
|
for row in csvreader:
|
|
data2.append(row)
|
|
data2 = [i for i in data2 if i]
|
|
#fprint(data2)
|
|
for line in appendbad:
|
|
data2.append(line)
|
|
|
|
|
|
|
|
fprint(data2)
|
|
with open(find_data_file("blocklist.csv"), "w", newline="") as f:
|
|
writer = csv.writer(f)
|
|
writer.writerows(data2)
|
|
fprint("done creating csv")
|
|
return data2
|
|
|
|
def block_conn(config, datafile, res):
|
|
setup_child()
|
|
fprint("Searching block data")
|
|
mydata = list()
|
|
badapps = list()
|
|
badips = list()
|
|
badlines = list()
|
|
with open(find_data_file(datafile), newline='') as csvfile:
|
|
csvreader = csv.reader(csvfile, delimiter=',', quotechar='|')
|
|
|
|
for row in csvreader:
|
|
mydata.append(row)
|
|
|
|
baddata = res
|
|
fprint("Local loaded successfully")
|
|
goodct = 0
|
|
#fprint(mydata)
|
|
for line in mydata:
|
|
#fprint(line)
|
|
#fprint(line)
|
|
if line[0].find("0x") >= 0:
|
|
continue
|
|
|
|
srcip = line[2].split(":")[0]
|
|
destip = line[3].split(":")[0]
|
|
pid = line[5]
|
|
try:
|
|
pid = int(pid)
|
|
except ValueError:
|
|
continue
|
|
found = False
|
|
for line in baddata:
|
|
#fprint(destip + " " + line[4])
|
|
badsrcip = line[2]
|
|
baddestip = line[4]
|
|
badpid = line[11]
|
|
|
|
if srcip == badsrcip or destip == baddestip and not pid in badapps:
|
|
found = True
|
|
fprint("FLAG " + srcip + " " + destip + " " + str(pid))
|
|
badapps.append(pid)
|
|
badips.append(baddestip)
|
|
badlines.append(line)
|
|
#fprint(badapps)
|
|
#fprint("FLAG " + srcip + " " + destip + " " + str(pid))
|
|
#kill(pid)
|
|
|
|
if not found:
|
|
goodct = goodct + 1
|
|
|
|
return badapps, badips, badlines, goodct
|