def digger(data_type, count):
rto = raintableoperator()
dbo = dboperator()
fo = fileoperator()
unknow_list = []
if data_type == "md5":
while 1:
unknow_list = dbo.get_unknown_md5_list(count)
if len(unknow_list) > 0:
print(" ")
print("md5 digger:get " + str(len(unknow_list)) + " hash to dig")
print("md5 digger:hash list is: " + str(unknow_list))
weak_en_list, weak_pl_list = rto.ask_raintable(unknow_list, "md5")
strong_en_list = set(unknow_list) - set(weak_en_list)
dbo.refresh_md5table(weak_en_list, weak_pl_list, strong_en_list)
else:
time.sleep(10)
fo.write_log("md5 digger is waitting for work")
print(" ")
print("md5 digger:get " + str(len(unknow_list)) + " hash to dig")
print("md5 digger:hash list is: " + str(unknow_list))
print("md5 digger:duang")
elif data_type == "sha1":
while 1:
unknow_list = dbo.get_unknown_sha1_list(count)
if len(unknow_list) > 0:
print(" ")
print("sha1 digger:get " + str(len(unknow_list)) + " hash to dig")
print("sha1 digger:hash list is: " + str(unknow_list))
weak_en_list, weak_pl_list = rto.ask_raintable(unknow_list, "sha1")
strong_en_list = set(unknow_list) - set(weak_en_list)
dbo.refresh_sha1table(weak_en_list, weak_pl_list, strong_en_list)
else:
time.sleep(10)
fo.write_log("sha1 digger is waitting for work")
print(" ")
print("sha1 digger:get " + str(len(unknow_list)) + " hash to dig")
print("sha1 digger:hash list is: " + str(unknow_list))
print("sha1 digger:duang")
else:
fo.write_log("you input an unknow data_type in digger(data_type,count)")
def print_help():
print("help :")
print(" hashpump.py -h")
print(" hashpump.py topn filename")
print(" hashpump.py custom filename")
def append_knowledge(data_list, note):
dbo = dboperator()
dbo.append_knowledge(data_list, note)
if __name__ == "__main__":
fo = fileoperator()
if len(sys.argv) == 1 or len(sys.argv) == 2:
print_help()
elif len(sys.argv) == 3:
if sys.argv[1] == "topn":
temp_data_list = fo.read_list_from_file(sys.argv[2])
append_knowledge(temp_data_list, TOP_N)
elif sys.argv[1] == "custom":
temp_data_list = fo.read_list_from_file(sys.argv[2])
append_knowledge(temp_data_list, USER_CUSTOM)
elif sys.argv[1] == "l8":
temp_data_list = fo.read_list_from_file(sys.argv[2])
append_knowledge(temp_data_list, LESS_THAN_8)
else:
print_help()
else:
