Python重新使用相同的套接字并遇到问题

Question

我正在使用 Python 编写一个简单的脚本，该脚本将使用套接字连接到主机/端口。我知道套接字只能使用一次，这就是为什么我根本不关闭套接字，但是当我在端口 80 上连接到 localhost 并尝试像GET / 这样的简单命令时，它第一次工作但第二次执行@ 987654322@ 或任何其他 HTTP 命令，它不会打印输出。这是我所拥有的

import socket
size = 1024
host = 'localhost'
port = 80
def connectsocket(userHost, userPort):
    s = socket.socket(socket.AF_INET, socket.SOCK_STREAM) #TCP socket
    s.connect((userHost, userPort))
    while(1):
        input = raw_input("Command: ")
        s.send(input + '\r\n\r\n') #Send command
        r = s.recv(size) #Recieve output
        print(r)
connectsocket(host, port)

我认为这会起作用，但这是一个示例输出：

amartin@homebox:~$ python socketconn.py
Command: GET /
[BUNCH OF HTML CODE]
Command: GET /

Command:

如您所见，它适用于第一个GET /，但不适用于第二个。我怎样才能解决这个问题？

Answer 1

鉴于来自 cmets 的关于 @samplebias 答案的信息，我认为这与您想要完成的任务类似：

import errno
import socket

size = 1024
host = 'localhost'
port = 80


def connectsocket(userHost, userPort):

    while(1):
        s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)  # TCP socket
        s.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
        s.connect((userHost, userPort))

        command = raw_input("Command: ")

        s.send(command + '\r\n\r\n')  # Send command
        response = s.recv(size)  # Recieve output
        print response

        try:
            s.shutdown(socket.SHUT_RDWR)
        except socket.error, exc:
            # Depending on the platform, shutting down one half of the
            # connection can also close the opposite half
            if exc.errno != errno.ENOTCONN:
                raise

        s.close()


connectsocket(host, port)

您也可以查看Twisted 库。你也可以看看这本书：Foundations of Python Network Programming。

python 文档网站上还有一个很好的 socket tutorial 可以提供帮助。最后但同样重要的是，我在 Google 上找到了一个 slightly more comprehensive tutorial，它看起来非常适合初学者。

HTH。

Answer 2

您需要告诉服务器保持 TCP 连接打开并期待更多请求：

headers = 'Connection: keep-alive\r\n'
s.send(input + '\r\n' + headers + '\r\n')

另外，请尝试指定 HTTP 版本：

python socketconn.py 
Command: GET / HTTP/1.1