Docker/pandora-box
Docker/pandora-box
1
0
Fork 0
mirror of https://github.com/dbarzin/pandora-box.git synced 2025-07-21 14:29:41 +02:00
Code Issues Activity

work in progress

Browse source
This commit is contained in:
Didier 2022-06-13 20:10:33 +00:00
parent 1052b54dc7
commit 145ebb41a0
1 changed files with 56 additions and 47 deletions
Download patch file Download diff file Expand all files Collapse all files

103
pandorabox.py
View file

@ -18,9 +18,9 @@ import time
# -----------------------------------------------------------
NO_SCAN = True
USB_AUTO_MOUNT = True
USB_AUTO_MOUNT = False
PANDORA_ROOT_URL = "http://127.0.0.1:6100"
FAKE_SCAN = True
FAKE_SCAN = False
# ----------------------------------------------------------
@ -223,12 +223,19 @@ def mount_device(device):
else:
return ""
else:
res = os.system("pmount " + device.device_node + " box")
if res == 1:
return "/media/box"
else:
return ""
res = os.system("pmount " + device.device_node + " /media/box")
found = False
loop = 0
while (not found) and (loop < 10):
time.sleep(1)
try:
statvfs=os.statvfs(mount_point)
except Exception as e :
loop +=1
continue
break;
log("Device mounted at /media/box")
return "/media/box"
"""Unmount USB device"""
@ -261,7 +268,8 @@ def device_loop():
try:
statvfs=os.statvfs(mount_point)
except Exception as e :
logging.error("Unexpected error: ", e)
log("Unexpected error1: %s" % e)
logging.exception("An exception was thrown!")
continue
print_size(human_readable_size(statvfs.f_frsize * statvfs.f_blocks))
print_used(human_readable_size(statvfs.f_frsize * (statvfs.f_blocks - statvfs.f_bfree)))
@ -294,7 +302,8 @@ def device_loop():
umount_device()
update_bar(0)
except Exception as e:
log("Unexpected error: %s" % e )
log("Unexpected error2: %s" % e )
logging.exception("An exception was thrown!")
finally:
log("Done.")
@ -325,47 +334,46 @@ def log_device_info(dev):
"""Scan a mount point with Pandora"""
def scan(mount_point, used):
global infected_filed
infected_files = array()
infected_files = []
scanned = 0
file_count = 0
scan_start_time = time.time()
if FAKE_SCAN:
for root, dirs, files in os.walk(mount_point):
for file in files:
try :
full_path = os.path.join(root,file)
file_size = os.path.getsize(full_path)
log("Check %-s [%s]" % (file, human_readable_size(file_size)))
file_scan_start_time = time.time()
if not FAKE_SCAN:
pandora = pypandora.PyPandora(root_url=PANDORA_ROOT_URL)
for root, dirs, files in os.walk(mount_point):
for file in files:
try :
full_path = os.path.join(root,file)
file_size = os.path.getsize(full_path)
log("Check %s [%s]" % (file, human_readable_size(file_size)))
file_scan_start_time = time.time()
if FAKE_SCAN :
time.sleep(0.1)
file_scan_end_time = time.time()
log("Check %s (%ds) -> %-s" % (file,(file_scan_end_time - file_scan_start_time),"SKIPPED"))
scanned += os.path.getsize(full_path)
file_count += 1
update_bar(scanned * 100 // used)
except Exception as e :
log("Unexpected error: %s" % e)
return False
update_bar(100)
log("Scan done in %ds" % (time.time() - scan_start_time))
log("%d files scanned" % file_count)
else:
pp = pypandora.PyPandora(root_url=PANDORA_ROOT_URL)
for arg in sys.argv[1:]:
log("Scan %-80s" % arg)
res = pp.submit_from_disk(arg)
while True:
time.sleep(1)
res = pp.task_status(res["taskId"])
if res["status"] != "WAITING":
break
break;
log("Scan %s -> %s" % (arg,res["status"]))
status = "SKIPPED"
else:
if file_size > (1024*1024*1024):
status = "TOO BIG"
else:
res = pandora.submit_from_disk(full_path)
time.sleep(0.1)
while True:
res = pandora.task_status(res["taskId"])
status = res["status"]
if status != "WAITING":
break
time.sleep(0.5)
file_scan_end_time = time.time()
log("Check %s (%ds) -> %-s" % (file,(file_scan_end_time - file_scan_start_time),status))
scanned += os.path.getsize(full_path)
file_count += 1
update_bar(scanned * 100 // used)
except Exception as e :
log("Unexpected error3: %s" % e)
logging.exception("An exception was thrown!")
return False
update_bar(100)
log("Scan done in %ds" % (time.time() - scan_start_time))
log("%d files scanned" % file_count)
return True
@ -381,7 +389,8 @@ def main(stdscr):
while True:
device_loop()
except Exception as e :
logging.error("Unexpected error: ", e)
log("Unexpected error4: %s" % e)
logging.exception("An exception was thrown!")
# logging.error(traceback.format_exc())
finally:
end_curses()