01tcp_deadlock.py

import argparse,socket

def recvall(sock,length):
    """传入一个套接字，和一个长度"""
    # 定义一个二进制的变量
    data = b''
    # 循环
    while len(data) < length:
        print(len(data),length)
        # more用来接收规定长度减去data的长度
        # 注意这里，recv函数是阻塞的，如果他没有接收到数据，那么就一直卡在那里。
        # 有两种情况，如果每次发送的数据都比我们设定的长度要少，那么第二次循环，就会卡主
        # 还有一种就是我们发送的字节数正好是我们规定长度的整数倍，那么最后一次循环就为空了。
        # 针对这个，我们可以在发送端和接收端都来一个结束标识。
        more = sock.recv(length - len(data))
        if more == b'efo':break
        if not more:
            raise EOFError('was expecting %d bytes but only received '
                           'd% bytes before the socket closed ' %(length ,len(data)))
        data += more
    return data

def server(interface,port):
    # 定义一个套接字，用于TCP连接
    sock = socket.socket(socket.AF_INET,socket.SOCK_STREAM)
    # 设置套接字连接
    sock.setsockopt(socket.SOL_SOCKET,socket.SO_REUSEADDR,1)
    # sock.setsockopt(socket.SOL_SOCKET,socket.SO_REUSEADDR,1)
    # 套接字绑定端口和IP地址
    sock.bind((interface,port))
    # 设置套接字的监听客户端的监听数量。
    sock.listen(1)
    # 打印出服务端连接绑定的IP地址和端口
    print("listening at",sock.getsockname())
    # 死循环遍历，用来重复的进行和监听服务端，然后进行通信。
    while True:
        # 服务端套接字用来接收客服端的连接请求。
        sc,socketname = sock.accept()
        print("We have accepted a connection from",socketname)
        print("  Socket name:",sc.getsockname())
        print("  Socket peer:",sc.getpeername())
        # 规定每次接收客户端传过来的消息长度
        message = recvall(sc,16)
        print('222222')
        # 打印出来接受到的数据。
        print(' Incoming sixteen-octet message:',repr(message))
        # 服务端发送二进制消息。
        sc.sendall(b'Farewell,client')
        sc.sendall(b'efo')
        sc.close()
        print("  Reply sent ，scoket closed")

# 定义客户端函数，入参为IP地址和端口
def client(host,port):
    # 定义一个TCP 连接的套接字。
    sock = socket.socket(socket.AF_INET,socket.SOCK_STREAM)
    # 建立和服务端的TCP连接。
    # 注意这里传入的是一个元祖
    sock.connect((host,port))
    # 打印客户端连接的IP和端口
    print('Client has been assigned socket name',sock.getsockname())
    # 客户端发送二进制消息
    sock.sendall(b'Hi there,server')
    # 发送一个结束标识
    sock.sendall(b'efo')
    # 客户端接收服务端发送过来的消息
    reply = recvall(sock,16)
    # 打印出来服务端发送的消息
    print('222222')
    print('The server said' ,repr(reply))
    # 关闭套接字
    sock.close()

if __name__ == "__main__":
    # 定义一个字典
    choices = {'client':client,'server':server}
    parser = argparse.ArgumentParser(description = 'Send and receive over TCP')
    parser.add_argument('role',choices = choices,help = 'which role to play')
    parser.add_argument('host',help = 'interface the server listens at;'
                                      ' host the client sends to')
    parser.add_argument('-p',metavar = "PORT",type = int,default = 1060,help = "TCP port (default 1060")
    args = parser.parse_args()
    function = choices[args.role]
    function(args.host,args.p)