python套接字流重定向实例汇总

将套接字流重定向到标准输入或输出流

    #!/usr/bin/env python3
    """
    测试socket-stream 重定向模式
    """
    import sys,os,time
    from multiprocessing import Process
    from socket import *

    def initListenerSocket(port=50008,host=''):
        """ 
        初始化在服务器模式下调用者用于监听连接的套接字
        """
        sock=socket()
        try:
            sock.bind((host,port))
        except OSError as e:
            print('Address already in use')
            os._exit(1)
        sock.listen(5)
        conn,addr=sock.accept()
        return conn

    def redirecOut(port=50008,host='localhost'):
        """ 
        在接受之前其他连接都失败，连接调用者标准输出流
        到一个套接字，这个套接字用于gui监听,在收听者启动后，启动调用者
        """
        sock=socket()
        try:
            sock.connect((host,port))
        except ConnectionRefusedError as e:
            print('connection refuse')
            os._exit(1)
        file=sock.makefile('w')
        sys.stdout=file
        return sock

    def redirecIn(port=50008,host='localhost'):
        """ 
        连接调用者标准输入流到用于gui来提供的套接字
        """
        sock=socket()
        try:
            sock.connect((host,port))
        except ConnectionRefusedError as e:
            print('conenction refuse')
            os._exit(1)
        file=sock.makefile('r')
        sys.stdin=file
        return sock

    def redirecBothAsClient(port=50008,host='localhost'):
        """
        在这种模式下，连接调用者标准输入和输出流到相同的套接字
        调用者对于服务器来说就是客户端:发送消息，接受响应答复
        """
        sock=socket()
        try:
            sock.connect((host,port))
        except ConnectionRefusedError as e:
            print('connection refuse')
            os._exit(1)
        ofile=sock.makefile('w')
        ifile=sock.makefile('r')
        sys.stdout=ofile
        sys.stdin=ifile
        return sock

    def redirecBothAsServer(port=50008,host='localhost'):
        """
        在这种模式下，连接调用者标准输入和输出流到相同的套接字，调用者对于
        服务器来说就是服务端：接受消息，发送响应答复
        """
        sock=socket()
        try:
            sock.bind((host,port))
        except OSError as e:
            print('Address already in use')
            os._exit(1)
        sock.listen(5)
        conn,addr=sock.accept()
        ofile=conn.makefile('w')
        ifile=conn.makefile('r')
        sys.stdout=ofile
        sys.stdin=ifile
        return conn

    def server1():
        mypid=os.getpid()
        conn=initListenerSocket()
        file=conn.makefile('r')
        for i in range(3):
            data=file.readline().rstrip()
            print('server %s got [%s]' %(mypid,data))

    def client1():
        time.sleep(1)
        mypid=os.getpid()
        redirecOut()
        for i in range(3):
            print('client: %s:%s' % (mypid,i))
            sys.stdout.flush()

    def server2():
        mypid=os.getpid()
        conn=initListenerSocket()
        for i in range(3):
            conn.send(('server %s got [%s]\n' %(mypid,i)).encode())

    def client2():
        time.sleep(1)
        mypid=os.getpid()
        redirecIn()
        for i in range(3):
            data=input()
            print('client %s got [%s]]'%(mypid,data))

    def server3():
        mypid=os.getpid()
        conn=initListenerSocket()
        file=conn.makefile('r')
        for i in range(3):
            data=file.readline().rstrip()
            conn.send(('server %s got [%s]\n' % (mypid,data)).encode())

    def client3():
        time.sleep(1)
        mypid=os.getpid()
        redirecBothAsClient()
        for i in range(3):
            print('Client %s: %s' %(mypid,data))
            data=input()
            sys.stderr.write('client %s got [%s]\n' %(mypid,data))

    def server4(port=50008,host='localhost'):
        mypid=os.getpid()
        sock=socket()
        try:
            sock.connect((host,port))
        ConnectionRefusedError as e:
            print('connection refuse')
            os._exit(1)
        file=sock.makefile('r')
        for i in range(3):
            sock.send(('server %s: %S\n' %(mypid,i)).encode())
            data=file.readline().rstrip()
            print('server %s got [%s]' %(mypid,data))

    def client4():
        time.sleep(1)
        mypid=os.getpid()
        redirecBothAsServer()
        for i in range(3):
            data=input()
            print('client %s got [%s]'%(mypid,data))
            sys.stdout.flush()

    def server5():
        mypid=os.getpid()
        conn=initListenerSocket()
        file=conn.makefile('r')
        for i in range(3):
            conn.send(('server %s:%s\n' %(mypid,i)).encode())
            data=file.readline().rstrip()
            print('server %s got [%s]' % (mypid,data))

    def client5():
        mypid=os.getpid()
        s=redirecBothAsClient()
        for i in range(3):
            data=input()
            print('client %s got [%s]'%(mypid,data))
            sys.stdout.flush()

    def main():
        server=eval('server'+sys.argv[1])
        client=eval('client'+sys.argv[1])
        Process(target=server).start()
        client()

    if __name__=='__main__':
        main()