如何在Python中将INET套接字连接到PTY设备

Question

我试图在Python中复制此socat命令的行为：

socat PTY,link=/tmp/mypty tcp-listen:1234

与服务器端口1234的TCP连接将连接到/ tmp / mypty（到/ dev / pts / N的链接），运行screen /tmp/mypty将允许用户与该端口另一端的内容进行交互连接。 换句话说，如果这用于连接到该侦听端口：

socat exec:'bash -li',pty,stderr,setsid,sigint,sane tcp-connect:172.16.30.3:1234

然后，运行screen /tmp/mypty将用户连接到在TCP连接的另一端运行的bash实例。

获取TCP套接字非常简单。 而且我可以使用os.openpty（）获得PTY，这会给我带来一对文件描述符，它们只是整数。

我无法做的是以一种可以无缝地来回传递数据的方式钩住该套接字和那些文件描述符。 如果它们都是插座，那将是微不足道的。 使用带有select（）的循环将新的读取数据传递到另一个套接字的write 。

我尝试过的

1. 使用socket.fromfd使用一个PTY文件描述符构建一个套接字。 这导致OSError: [Errno 88] Socket operation on non-socket 。 此代码示例，其中clisock是有效的套接字：

def sockToPty(clisock,addr):
        (master, slave) = os.openpty()
        print('PTY: Opening {}'.format(os.ttyname(slave)))
        ptymsock = socket.fromfd(master, socket.AF_UNIX, socket.SOCK_STREAM)
        ptyssock = socket.fromfd(slave,  socket.AF_UNIX, socket.SOCK_STREAM)
        print('CLISOCK: {}'.format(clisock))
        print('PTYMSOCK: {}'.format(ptymsock))
        print('PTYSSOCK: {}'.format(ptyssock))
        peer[clisock] = ptymsock
        peer[ptyssock] = clisock
        while True:
            iready, oready, eready = select.select([clisock, ptyssock], [], [])

            for sock in iready:
                data = sock.recv(4096)
                if not len(data):
                    return
                peer[sock].send(data)

导致此错误：

PTY: Opening /dev/pts/2
    CLISOCK: 
    PTYMSOCK: 
    PTYSSOCK: 
    Exception in thread Thread-1:
    Traceback (most recent call last):
      File "/usr/lib/python3.6/threading.py", line 916, in _bootstrap_inner
        self.run()
      File "/usr/lib/python3.6/threading.py", line 864, in run
        self._target(*self._args, **self._kwargs)
      File "./tcpmux.py", line 62, in sockToPty
        peer[sock].send(data)
    OSError: [Errno 88] Socket operation on non-socket

如果我使用AF_INET而不是AF_UNIX，这没有什么区别。

2. 使用os.dup2（）将SSLsocket的fileno（）映射到stdin，stdout和stderr，然后运行os.openpty（）（la tcp_pty_bind.py ）。 这不会产生任何错误，但是当我使用screen打开PTY时，没有明显的数据流。

几乎所有的Python PTY示例都集中在pty.spawn（）上，这使得很难找到这种（公认的）PTY使用的示例代码。

旁注：我之所以使用os.openpty而不是pty.openpty是因为似乎没有意义。 pty模块的广告似乎是为了提高可移植性，但并没有更大的可移植性，而且我正在Linux上工作，因此可移植性不是问题。 而且由于我正在获得正确的os.ttyname（）的有效文件描述符，所以我没有理由质疑openpty（）不能正常工作。

Answer 1

可以使用select.poll（）而不是select.select（）来完成此操作，因为select.poll（）可以与文件描述符和套接字一起使用。

def sockToPty(clisock,addr):
    (master, slave) = os.openpty()
    tty.setraw(master, termios.TCSANOW)
    print('PTY: Opened {} for {}'.format(os.ttyname(slave), addr))
    mypoll = select.poll()
    mypoll.register(clisock, select.POLLIN)
    mypoll.register(master, select.POLLIN)
    try:
        while True:
            fdlist = mypoll.poll(1000)
            for fd,event in fdlist:
                # We treat sockets and FDs differently
                if fd == master:
                    data = os.read(fd, 4096)
                    if len(data) == 0:
                        print('PTY: {}, {} exiting due to Zero read.'.format(addr, os.ttyname(slave)))
                        raise GetOut
                    clisock.send(data)
                else:
                    data = clisock.recv(4096)
                    if len(data) == 0:
                        print('PTY: {}, {} exiting due to Zero read.'.format(addr, os.ttyname(slave)))
                        raise GetOut
                    os.write(master, data)
    except GetOut:
        os.close(master)
        os.close(slave)
        clisock.shutdown(socket.SHUT_RDWR)
        clisock.close()

这段代码比简单的套接字要复杂得多-使用TCP套接字，您可以简单地将该套接字的fileno（）用作文件描述符，然后以相同的方式对待双方（os.read（）， os.write（））。 但是，如果要将它与SSLSocket一起使用，则需要上面更复杂的版本，因为如果从SSLSocket中获取fileno（），则它希望您将处理TLS层以及数据。