Upload app.py

app.py

@@ -16,7 +16,11 @@ def try_decrypt_wallet(wallet_data_bytes, password_attempt):
         if w.is_encrypted():
             return w.decrypt(password_attempt)
         else:
-            st.session_state.info_message = "钱包文件未加密。"
+            # 使用 session_state 存储需要在主线程显示的消息
+            if 'info_message' not in st.session_state:
+                st.session_state.info_message = "钱包文件未加密。"
+            elif st.session_state.info_message is None: # 避免覆盖已有的消息直到它被显示
+                 st.session_state.info_message = "钱包文件未加密。"
             return True
     except Exception:
         return False
@@ -28,10 +32,6 @@ def generate_passwords_bruteforce(charset, max_len):
 
 # --- 后台工作函数 ---
 def background_cracker_task(wallet_bytes_data, passwords_iterable, task_id, method_name):
-    """
-    在后台线程中运行破解过程。
-    更新 st.session_state 来反映进度和结果。
-    """
     st.session_state[f'task_{task_id}_status'] = 'running'
     st.session_state[f'task_{task_id}_found_password'] = None
     st.session_state[f'task_{task_id}_error_message'] = None
@@ -42,14 +42,7 @@ def background_cracker_task(wallet_bytes_data, passwords_iterable, task_id, meth
     found = False
     processed_passwords = 0
 
-    # 特殊处理暴力破解，因为它是一个生成器，无法预知总数
-    is_bruteforce = (method_name == "bruteforce")
-    total_passwords_for_progress = 0
-    if not is_bruteforce and isinstance(passwords_iterable, list):
-        total_passwords_for_progress = len(passwords_iterable)
-
     for pwd in passwords_iterable:
-        # 检查是否应该停止 (如果用户启动了新任务)
         if st.session_state.get('current_task_id') != task_id:
             st.session_state[f'task_{task_id}_error_message'] = "任务被另一个操作中止。"
             st.session_state[f'task_{task_id}_status'] = 'failed'
@@ -58,30 +51,20 @@ def background_cracker_task(wallet_bytes_data, passwords_iterable, task_id, meth
         processed_passwords += 1
         st.session_state[f'task_{task_id}_attempts'] = processed_passwords
 
-        # 为了避免过于频繁地更新session_state (可能导致性能问题)
-        # 可以在这里添加一些逻辑，例如每隔N次尝试或每隔X秒更新一次用于显示的尝试计数
-        # 不过对于最终结果的设置，这个不是大问题
-
         if try_decrypt_wallet(wallet_bytes_data, pwd):
             st.session_state[f'task_{task_id}_found_password'] = pwd
             st.session_state[f'task_{task_id}_status'] = 'success'
             found = True
             break
-        # time.sleep(0.0001) # 极小的延时，以允许GIL切换，但对CPU密集型任务效果不大
+        # time.sleep(0.00001) # 极小的延时，CPU密集型任务中效果有限
 
     end_time = time.time()
     st.session_state[f'task_{task_id}_time_taken'] = end_time - start_time
 
-    if not found and st.session_state[f'task_{task_id}_status'] == 'running': # 确保不是因为中止而退出
+    if not found and st.session_state[f'task_{task_id}_status'] == 'running':
         st.session_state[f'task_{task_id}_error_message'] = "未找到密码。"
         st.session_state[f'task_{task_id}_status'] = 'failed'
 
-    # 主动触发一次 Streamlit 刷新，让UI能立刻感知到状态变化
-    # 这个在线程中调用需要小心，但对于更新session_state然后让Streamlit自己刷新是安全的
-    # st.experimental_rerun() # 旧版，如果需要可以尝试
-    # Streamlit 通常会自动检测 session_state 的变化并重新运行脚本
-
-
 # --- Streamlit UI ---
 st.set_page_config(page_title="Wallet.dat Cracker (Background)", layout="wide")
 st.title("Bitcoin Wallet.dat 密码破解尝试 (后台运行)")
@@ -96,23 +79,20 @@ st.warning(
     """
 )
 
-# 初始化 session_state 变量
 if 'current_task_id' not in st.session_state:
     st.session_state.current_task_id = None
-if 'task_counter' not in st.session_state: # 用于生成唯一的 task_id
+if 'task_counter' not in st.session_state:
     st.session_state.task_counter = 0
 if 'info_message' not in st.session_state:
     st.session_state.info_message = None
 
 
-# --- 文件上传 ---
 uploaded_wallet_file = st.file_uploader("1. 上传你的 wallet.dat 文件", type=["dat"])
 
 if uploaded_wallet_file:
     wallet_bytes = uploaded_wallet_file.getvalue()
     st.success(f"已上传 {uploaded_wallet_file.name} ({len(wallet_bytes)} bytes)")
 
-    # --- 破解方法选择 ---
     crack_method = st.radio(
         "2. 选择破解方法:",
         ('使用密码列表', '尝试单个密码', '基本暴力破解 (极慢，后台运行)')
@@ -123,22 +103,16 @@ if uploaded_wallet_file:
     if current_task_id:
         task_status = st.session_state.get(f'task_{current_task_id}_status')
 
-    # --- 显示任务状态和结果 ---
     if task_status == 'running':
         attempts = st.session_state.get(f'task_{current_task_id}_attempts', 0)
         with st.spinner(f"正在后台破解... 已尝试 {attempts} 个密码。请勿关闭此页面。"):
-            # 由于 Streamlit 的特性，spinner 会在脚本重新运行时自动更新。
-            # 我们需要一种方式让脚本在线程运行时定期重新运行以更新spinner的文本。
-            # 一个简单的方法是使用 st.empty() 和 time.sleep() 在主线程中轮询，
-            # 但这又会阻塞主线程。
-            # 更好的方式是依赖 Streamlit 对 session_state 变化的自动刷新。
-            # 为了让 spinner 中的数字更新，后台线程可能需要更频繁地更新一个 session_state 变量
-            # 但为了简单，我们这里只显示一个通用的 "正在后台破解..."
-            # 并且依赖于任务完成后 session_state 的变化来刷新整个页面。
-            # 或者，我们可以在这里放一个占位符，后台线程完成后，其设置的session_state会导致页面刷新
-            st.empty() # 只是为了保持 spinner 可见
-            time.sleep(1) # 让streamlit有机会重跑脚本并显示spinner
-            st.rerun() # 强制重跑以更新spinner中的计数（如果做了计数更新）
+            st.empty()
+            time.sleep(1)
+            try:
+                st.experimental_rerun()
+            except AttributeError:
+                st.warning("您的 Streamlit 版本较旧，spinner 状态可能不会实时更新。请考虑升级 Streamlit 或手动刷新页面。")
+                pass
 
 
     elif task_status == 'success':
@@ -149,8 +123,11 @@ if uploaded_wallet_file:
         st.balloons()
         st.info(f"总耗时: {time_taken:.2f} 秒。尝试密码数: {attempts}。")
         if st.button("清除结果并重试", key="clear_success"):
-            st.session_state.current_task_id = None # 清除任务标记
-            st.rerun()
+            st.session_state.current_task_id = None
+            try:
+                st.experimental_rerun()
+            except AttributeError:
+                st.warning("您的 Streamlit 版本较旧。请考虑升级 Streamlit 或手动刷新页面。")
 
 
     elif task_status == 'failed':
@@ -160,15 +137,19 @@ if uploaded_wallet_file:
         st.error(f"破解失败: {error_msg}")
         st.info(f"总耗时: {time_taken:.2f} 秒。尝试密码数: {attempts}。")
         if st.button("清除结果并重试", key="clear_failed"):
-            st.session_state.current_task_id = None # 清除任务标记
-            st.rerun()
+            st.session_state.current_task_id = None
+            try:
+                st.experimental_rerun()
+            except AttributeError:
+                st.warning("您的 Streamlit 版本较旧。请考虑升级 Streamlit 或手动刷新页面。")
 
+    # 显示来自 try_decrypt_wallet 的一次性消息
     if st.session_state.get('info_message'):
         st.info(st.session_state.info_message)
         st.session_state.info_message = None # 显示后清除
 
-    # --- 启动按钮 (仅当没有任务运行时显示) ---
-    if task_status not in ['running', 'success', 'failed'] or not task_status :
+
+    if not task_status or task_status not in ['running', 'success', 'failed']:
         passwords_to_try_list = []
         brute_force_params = {}
         can_start = False
@@ -193,7 +174,6 @@ if uploaded_wallet_file:
             else:
                 st.info("请上传密码列表文件。")
 
-
         elif crack_method == '尝试单个密码':
             single_password = st.text_input("输入要尝试的密码:", type="password", key="single_pwd_input")
             if single_password:
@@ -204,26 +184,29 @@ if uploaded_wallet_file:
             else:
                 st.info("请输入要尝试的密码。")
 
-
         elif crack_method == '基本暴力破解 (极慢，后台运行)':
             charset_input = st.text_input("字符集:", DEFAULT_CHARSET, key="charset_input")
             max_len_input = st.number_input(
                 "最大密码长度 (警告: 长度大于4会非常慢):",
                 min_value=1, max_value=8, value=MAX_BRUTEFORCE_LEN, key="maxlen_input"
             )
-            estimated_total = sum(len(charset_input)**i for i in range(1, max_len_input + 1))
-            st.warning(f"使用字符集 '{charset_input}' 和最大长度 {max_len_input} 进行暴力破解，"
-                       f"可能需要尝试 {estimated_total} 个密码。这会非常耗时。")
-            brute_force_params = {'charset': charset_input, 'max_len': max_len_input}
-            can_start = True
-            start_button_label = "开始暴力破解 (后台)"
-            method_name_for_task = "bruteforce"
+            if charset_input: # 确保字符集不为空
+                estimated_total = sum(len(charset_input)**i for i in range(1, max_len_input + 1))
+                st.warning(f"使用字符集 '{charset_input}' 和最大长度 {max_len_input} 进行暴力破解，"
+                           f"可能需要尝试 {estimated_total} 个密码。这会��常耗时。")
+                brute_force_params = {'charset': charset_input, 'max_len': max_len_input}
+                can_start = True
+                start_button_label = "开始暴力破解 (后台)"
+                method_name_for_task = "bruteforce"
+            else:
+                st.error("字符集不能为空。")
 
 
         if can_start and st.button(start_button_label, key=f"start_button_{method_name_for_task}"):
             st.session_state.task_counter += 1
             new_task_id = st.session_state.task_counter
             st.session_state.current_task_id = new_task_id
+            st.session_state.info_message = None # 清除任何旧的钱包未加密消息
 
             if method_name_for_task == "bruteforce":
                 passwords_iterable = generate_passwords_bruteforce(
@@ -232,14 +215,16 @@ if uploaded_wallet_file:
             else:
                 passwords_iterable = passwords_to_try_list
 
-            # 启动后台线程
             thread = threading.Thread(
                 target=background_cracker_task,
                 args=(wallet_bytes, passwords_iterable, new_task_id, method_name_for_task)
             )
-            thread.daemon = True # 允许主程序退出时线程也退出
+            thread.daemon = True
             thread.start()
-            st.rerun() # 立即重新运行脚本以显示 spinner
+            try:
+                st.experimental_rerun()
+            except AttributeError:
+                st.warning("您的 Streamlit 版本较旧。请考虑升级 Streamlit 或手动刷新页面。")
 
 else:
     st.info("请先上传 wallet.dat 文件。")