你的 Python 脚本整体逻辑是：通过 `paramiko` 连接目标主机，使用普通用户登录后切换到 `root`，然后执行后续操作。但你只贴了一部分代码（在 `while not buff.endswith('# '):` 之后中断了），并且存在几个潜在问题和优化建议。 --- ### ✅ **当前脚本存在的主要问题** 1. **`buff.endswith('# ')` 的判断可能失败** - `su -` 切换后，有些系统提示符是 `#`，有些是 `# `（带空格），还有可能是 `#` 在行首。 - 更严重的是：`recv(9999)` 只接收一次数据，而 `buff` 是拼接的字符串，**如果命令输出被分块发送，容易漏掉关键字符**。 2. **未处理 `su` 失败的情况** - 如果 `root_pass` 错误，`su` 会提示 `Password:` 重复出现，但你没有重试或超时退出机制。 3. **缺少异常处理和超时控制** - 当前 `try` 块中没有对 `ssh.send()` 或 `recv()` 的异常处理，一旦网络波动就崩溃。 4. **`print(resp)` 打印原始字节流，可读性差** - 建议用 `.decode('utf-8', errors='ignore')` 避免编码错误。 5. **`passwd6` 文件格式假设太强** - 假设每行是 `ip username gzrst_pass root_pass new_pass`，中间用空格分隔，若文件有空行、注释、多余空格会导致解析错误。 6. **脚本未完成后续操作** - 你在 `while not buff.endswith('# ')` 后中断了，无法继续执行 `new_pass` 设置等操作。 7. **安全性隐患** - 硬编码密码在文件中，且明文传输（即使加密连接，也建议避免）。 - 没有记录日志或审计。 --- ### ✅ **修复 & 完善版本（推荐）** ```python #!/usr/bin/env python3 # -*- coding: utf-8 -*- import paramiko import time import os # 清理旧结果文件 test_fail = 'result.fail' test_ok = 'result.ok' if os.path.exists(test_fail): os.remove(test_fail) if os.path.exists(test_ok): os.remove(test_ok) # 密码文件路径 pass_file_path = 'passwd6' # 检查文件是否存在 if not os.path.exists(pass_file_path): print(f"❌ 密码文件 {pass_file_path} 不存在！") exit(1) # 读取并处理每一行 with open(pass_file_path, 'r') as f: lines = [line.strip() for line in f if line.strip() and not line.startswith('#')] for idx, line in enumerate(lines): parts = line.split() if len(parts) < 5: print(f"⚠️ 第 {idx+1} 行格式错误（跳过）: {line}") continue ipaddr, username, gzrst_pass, root_pass, new_pass = parts[:5] port = 22 print(f"\n🔄 正在尝试连接 {ipaddr} (用户: {username}) ...") try: # 创建 SSH 客户端 client = paramiko.SSHClient() client.load_system_host_keys() client.set_missing_host_key_policy(paramiko.AutoAddPolicy()) # 连接 client.connect( hostname=ipaddr, port=port, username=username, password=gzrst_pass, timeout=10 ) print(f"✅ {ipaddr} 连接成功") # 获取交互式 shell shell = client.invoke_shell() time.sleep(1) # 缓冲区接收函数 def recv_buffer(max_len=9999, timeout=5): buffer = "" start_time = time.time() while time.time() - start_time < timeout: if shell.recv_ready(): data = shell.recv(max_len).decode('utf-8', errors='ignore') buffer += data # 检查是否包含关键提示符 if any(pattern in buffer for pattern in ['$', '#', 'Password:']): return buffer time.sleep(0.1) return buffer # 等待用户登录提示符 prompt = recv_buffer(timeout=10) if '$ ' in prompt or '#' in prompt: print("👉 已进入 shell 环境") else: print("⚠️ 未识别到预期提示符，跳过后续操作") client.close() continue # 执行 su - 切换到 root shell.send("su -\n") time.sleep(1) # 接收密码提示 resp = recv_buffer(timeout=10) if "Password:" not in resp: print(f"❌ 未能检测到 root 密码提示（响应: {resp[:100]})") client.close() continue print(f"🔑 正在输入 root 密码...") shell.send(root_pass + "\n") time.sleep(1) # 等待 root 提示符 resp = recv_buffer(timeout=15) if "# " in resp or "#" in resp: print(f"✅ 成功切换到 root 权限") else: print(f"❌ root 切换失败（响应: {resp[:150]})") with open(test_fail, 'a') as f: f.write(f"{ipaddr}\n") client.close() continue # === 可选：修改 root 密码 === # 示例：设置新密码 shell.send(f"echo '{new_pass}' | passwd --stdin root\n") time.sleep(1) # 再次检查是否成功 resp = recv_buffer(timeout=5) if "password updated successfully" in resp or "passwd: all authentication tokens updated successfully" in resp: print(f"🔐 root 密码已更新为: {new_pass}") else: print(f"⚠️ 密码更新可能失败（响应: {resp[:150]})") # 可选：保存成功结果 with open(test_ok, 'a') as f: f.write(f"{ipaddr} {username} {new_pass}\n") # 关闭连接 client.close() print(f"✅ {ipaddr} 处理完成") except Exception as e: print(f"❌ {ipaddr} 连接/操作失败: {str(e)}") with open(test_fail, 'a') as f: f.write(f"{ipaddr}\n") if 'client' in locals(): client.close() ``` --- ### ✅ **使用说明** 1. **准备 `passwd6` 文件**（示例内容）： ```text 192.168.1.100 admin password123 rootpass newpass123 192.168.1.101 user1 pass123 root123 newroot123 ``` 2. **安装依赖**： ```bash pip install paramiko ``` 3. **运行脚本**： ```bash python3 ssh_root_password_changer.py ``` 4. **输出结果**： - `result.ok`：成功连接并更改密码的主机列表。 - `result.fail`：失败的主机列表。 --- ### ✅ **安全建议** - ❌ 不要将密码明文写入文件。 - ✅ 使用密钥认证代替密码。 - ✅ 若用于批量管理，推荐使用 Ansible / SaltStack 等工具。 - ✅ 添加日志记录（如 `logging` 模块）便于排查。 --- ### 📌 总结 你原来的脚本思路正确，但需要： - 改进缓冲区处理方式（避免卡死） - 加强异常捕获 - 优化提示符匹配逻辑 - 增加容错与日志输出 我提供的完整版本已解决上述问题，并具备可扩展性。 如需添加“自动改密码”、“批量执行命令”、“并发多线程”等功能，也可以继续告诉我！