何时/为什么要使用s.shutdown(socket.SHUT_WR)?

时间:2016-01-31 12:13:27

标签: python sockets network-programming python-sockets

我刚开始学习python网络编程。我正在阅读Python网络编程基础,无法理解s.shutdown(socket.SHUT_WR)的使用,其中s是套接字对象。 下面是使用它的代码(其中sys.argv [2]是用户想要发送的字节数,舍入为16的倍数):

import socket, sys
s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)

HOST = '127.0.0.1'
PORT = 1060

if sys.argv[1:] == ['server']:
    s.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
    s.bind((HOST, PORT))
    s.listen(1)
    while True:
        print 'Listening at', s.getsockname()
        sc, sockname = s.accept()
        print 'Processing up to 1024 bytes at a time from', sockname
        n = 0
        while True:
            message = sc.recv(1024)
            if not message:
                break
            sc.sendall(message.upper())  # send it back uppercase
            n += len(message)
            print '\r%d bytes processed so far' % (n,),
            sys.stdout.flush()
        print
        sc.close()
        print 'Completed processing'

elif len(sys.argv) == 3 and sys.argv[1] == 'client' and sys.argv[2].isdigit():

    bytes = (int(sys.argv[2]) + 15) // 16 * 16  # round up to // 16
    message = 'capitalize this!'  # 16-byte message to repeat over and over

    print 'Sending', bytes, 'bytes of data, in chunks of 16 bytes'
    s.connect((HOST, PORT))

    sent = 0
    while sent < bytes:
        s.sendall(message)
        sent += len(message)
        print '\r%d bytes sent' % (sent,),
        sys.stdout.flush()

    print
    s.shutdown(socket.SHUT_WR)

    print 'Receiving all the data the server sends back'

    received = 0
    while True:
        data = s.recv(42)
        if not received:
            print 'The first data received says', repr(data)
        received += len(data)
        if not data:
            break
        print '\r%d bytes received' % (received,),

    s.close()

else:
    print >>sys.stderr, 'usage: tcp_deadlock.py server | client <bytes>'

这是作者提供的解释,我发现很难理解:

  

其次,您将看到客户端在完成发送传输后对套接字进行shutdown()调用。这解决了一个重要的问题:如果服务器要永远读取直到它看到文件结束,那么客户端将如何避免在套接字上执行完全关闭(),从而禁止自己执行许多recv( )调用它仍然需要接收服务器的响应?解决方案是“半关闭”套接字 - 也就是说,永久关闭一个方向的通信但不破坏套接字本身 - 这样服务器就不能再读取任何数据,但仍然可以发送任何剩余的回复另一个方向,仍然是开放的。

我对它的作用的理解是它会阻止客户端应用程序进一步发送数据,因此也会阻止服务器端进一步尝试读取任何数据。

我无法理解的是,为什么在这个程序中使用它以及我应该在什么情况下考虑在我的程序中使用它?

2 个答案:

答案 0 :(得分:2)

  

我对它会做什么的理解是它会阻止客户   应用程序从进一步发送数据,因此也将阻止   服务器端进一步尝试读取任何数据。

您的理解是正确的。

  

我无法理解的是,为什么在这个程序中使用它...

正如您自己的声明所暗示的那样,如果没有客户端s.shutdown(socket.SHUT_WR),服务器将不会退出等待数据,而是永远坚持其sc.recv(1024),因为没有连接终止请求发送到服务器。
由于服务器永远不会到达sc.close(),客户端也不会退出等待数据,而是永远坚持s.recv(42),因为不会发送连接终止请求服务器。
阅读this answer to "close vs shutdown socket?"也可能具有启发性。

答案 1 :(得分:0)

解释是半生不熟的,它仅适用 到这个特定的代码,总的来说,我会全力投票,这是不好的做法。

现在要明白为什么会这样,你需要查看服务器代码。此服务器通过阻止执行直到收到1024个字节为止。接收后,它处理数据(使其成为大写)并将其发回。现在的问题是硬编码值为1024.如果你的字符串短于1024字节怎么办?

要解决此问题,您需要告诉服务器 - 嘿,没有更多的数据可供您使用,所以从message = sc.recv(1024)返回,然后通过向一个方向关闭套接字来执行此操作。

您不想完全关闭套接字,因为服务器无法向您发送回复。