基于Python Cmd模块的Socket远程Shell：Readline/补全功能失效问题

我明白你遇到的痛点了——用cmd模块做的Shell在本地用标准IO时，Tab补全、光标键都正常，但通过Socket用Telnet连接时，这些功能就失效了，甚至显示乱码。这其实是因为终端控制字符的处理和上下文缺失导致的，下面给你具体的解决方案：

问题根源 #

cmd模块依赖Python的readline库，而readline需要和**交互式终端（TTY）**交互才能正常工作。当你用Socket传输数据时，数据是原始字节流，没有终端的属性（比如行编辑、回显、控制字符解析），所以readline的光标键映射、Tab补全功能无法生效，甚至会把控制字符直接打印成乱码。

解决方案：用伪终端（PTY）模拟真实终端 #

在Unix-like系统上，我们可以用pty模块创建伪终端，让cmd Shell以为自己在和真实终端交互，同时把伪终端的IO和Socket做转发。这样Telnet客户端的控制字符会被PTY正确解析，readline功能就能正常工作了。

修改后的完整代码 #

import socket
import pty
import os
from cmd import Cmd

class my_shell(Cmd):
    """Simple custom shell"""
    prompt = 'my-shell> '  # 设置自定义提示符

    def do_quit(self, arg):
        """Exit the shell session"""
        print("Goodbye!")
        return True

    def do_echo(self, arg):
        """Echo back your input"""
        print(f"Echo: {arg}")

    def do_help(self, arg):
        """Show help for commands"""
        if arg:
            super().do_help(arg)
        else:
            print("Available commands: quit, echo, help")

def handle_telnet_client(conn):
    # 创建伪终端（master用于父进程转发数据，slave给子进程做终端）
    master_fd, slave_fd = pty.openpty()
    
    # Fork子进程来运行cmd Shell
    pid = os.fork()
    if pid == 0:
        # 子进程：将标准IO绑定到伪终端的slave端
        os.dup2(slave_fd, 0)  # stdin
        os.dup2(slave_fd, 1)  # stdout
        os.dup2(slave_fd, 2)  # stderr
        
        # 关闭不需要的文件描述符
        os.close(master_fd)
        os.close(slave_fd)
        
        # 启动自定义Shell
        shell = my_shell()
        shell.cmdloop()
        os._exit(0)  # 子进程退出
    else:
        # 父进程：转发Socket和伪终端master端的数据
        os.close(slave_fd)
        try:
            while True:
                # 从Telnet客户端读数据，写入伪终端
                client_data = conn.recv(1024)
                if not client_data:
                    break  # 客户端断开连接
                os.write(master_fd, client_data)
                
                # 从伪终端读输出，发送给Telnet客户端
                terminal_output = os.read(master_fd, 1024)
                conn.send(terminal_output)
        except Exception as e:
            print(f"Connection error: {str(e)}")
        finally:
            # 清理资源
            os.close(master_fd)
            conn.close()
            os.waitpid(pid, 0)  # 等待子进程结束

def start_server(host='0.0.0.0', port=9999):
    server_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    server_socket.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
    server_socket.bind((host, port))
    server_socket.listen(5)
    print(f"Shell server running on {host}:{port}")
    
    while True:
        client_conn, client_addr = server_socket.accept()
        print(f"New connection from {client_addr[0]}:{client_addr[1]}")
        handle_telnet_client(client_conn)

if __name__ == '__main__':
    start_server()

为什么这样有效？ #

1. 伪终端模拟真实终端：PTY会处理Telnet发送的ANSI控制字符（比如光标上下键的\x1b[A序列），让readline库能正确解析这些指令，实现光标移动和Tab补全。
2. IO转发：父进程负责在Socket和PTY之间传递数据，相当于把Telnet客户端和伪终端做了桥接。

额外注意事项 #

• 这个方案仅适用于Linux/macOS等Unix-like系统，Windows系统没有pty模块，需要用winpty等第三方工具模拟。
• 如果遇到Telnet输入重复显示的问题，可以在数据转发时处理Telnet的选项协商（比如回应WILL ECHO指令），避免终端回显重复。

内容的提问来源于stack exchange，提问作者Jorge Chavez