执行的主程序

manage.py

@@ -0,0 +1,52 @@
+from msvcrt import getch
+import os
+import rainbow_table
+
+while True:
+    print("""\n
+欢迎使用随便做的破烂版彩虹表/生成破解程序。
+使用说明如下：
+
+1. 输入1生成彩虹表
+2. 输入2破解哈希值
+3. 输入3退出程序
+
+""")
+    op = input("请输入操作选项：")
+    if op == '1':
+        print("""
+生成命令格式如下：algorithm charset min_length max_length cahin_length chain_number index\n
+参数详解：
+    -algorithm:LM,NTLM,MD5,SHA1,SHA256
+    -charset:'numeric', 'alpha', 'alpha_numeric', 'loweralpha', 'loweralpha_numeric', 
+             'mixalpha', 'mixalpha_numeric', 'ascii_32_95', 'ascii_32_65_123_4', 'alpha_numeric_symbol32_space'
+             等，具体可以看同目录下的charset.txt进行确认。
+    -min_length and max_length:指定想生成的表的密码范围。
+    -chain_length:指定单条链的长度（进行几次约简函数和哈希函数操作）。
+    -chain_number:指定表内总共的链数量。
+    -index:索引值，用于确保使用的归约函数不同。""")
+        try:
+            algorithm, charset_type, min_length, max_length, chain_length, chain_number, table_index = input("").split()
+            rainbow_table.rtgen(algorithm, charset_type, int(min_length), int(max_length), int(chain_length), int(chain_number), int(table_index)).table_generate()
+        except Exception as e:
+            print(f'发生错误，请检查输入的指令格式是否有误.',e)
+    elif op == '2':
+        print("""
+破解指令格式如下：to_crack_hash_string rainbow_table_name\n
+参数详解：
+    -to_ceack_hash_string:输入要查表的哈希值。
+    -rainbow_table_name:默认读取工作目录下'.\\output'内的文件，请确保其中存在生成的表。\n""")
+        try:
+            value = input("").split()
+            hash_string,f_name = value
+            file_path = os.path.join(".\\output", f_name)
+            search = rainbow_table.rtsearch(hash_string,file_path)
+            search.crack()
+        except Exception as e:
+            print(f'\n发生错误，请检查输入的指令格式或哈希格式是否有误.',e)
+    elif op == '3':
+        print("\n感谢您的使用^_^")
+        break
+    else:
+        print("还没有定义这个操作哦。")
+    input("按任意键继续...")