python - HTTP 服务器在接受数据包时挂起

Question

我写了一个简单的 http 服务器来处理 POST 请求:

class MyHandler( BaseHTTPServer.BaseHTTPRequestHandler):
    def do_POST( self ):
        ctype, pdict = cgi.parse_header(self.headers.getheader('content-type'))
        postvars = {}
        try:
          if ctype == 'application/x-www-form-urlencoded':
              length = int(self.headers.getheader('content-length'))
              postvars = cgi.parse_qs(self.rfile.read(length), keep_blank_values=1)

          self.send_response( 200 )
          self.send_header( "Content-type", "text")
          self.send_header( "Content-length", str(len(body)) )
          self.end_headers()
          self.wfile.write(body)
        except:
          print "Error"


def httpd(handler_class=MyHandler, server_address = ('2.3.4.5', 80)):
    try:
        print "Server started"
        srvr = BaseHTTPServer.HTTPServer(server_address, handler_class)
        srvr.serve_forever() # serve_forever
    except KeyboardInterrupt:
        server.socket.close()


if __name__ == "__main__":
    httpd( )

服务器运行良好，但有时会挂起。当我按 CTRL+C 时，出现以下错误，然后继续接收数据:

Exception happened during processing of request from ('1.1.1.2', 50928)
Traceback (most recent call last):
  File "/usr/lib/python2.6/SocketServer.py", line 281, in _handle_request_noblock
    self.process_request(request, client_address)
  File "/usr/lib/python2.6/SocketServer.py", line 307, in process_request
    self.finish_request(request, client_address)
  File "/usr/lib/python2.6/SocketServer.py", line 320, in finish_request
    self.RequestHandlerClass(request, client_address, self)
  File "/usr/lib/python2.6/SocketServer.py", line 615, in __init__
    self.handle()
  File "/usr/lib/python2.6/BaseHTTPServer.py", line 329, in handle
    self.handle_one_request()
  File "/usr/lib/python2.6/BaseHTTPServer.py", line 312, in handle_one_request
    self.raw_requestline = self.rfile.readline()
  File "/usr/lib/python2.6/socket.py", line 406, in readline
    data = self._sock.recv(self._rbufsize)
KeyboardInterrupt

谁能告诉我如何纠正这个问题？我无法理解这些错误。

Answer 1

我完全重写了我的解决方案以解决两个缺点:

• 即使客户端只打开了一个连接但没有开始通信，它也可以处理超时。
• 如果客户端打开一个连接，一个新的服务器进程被派生出来。慢速客户端无法阻止其他客户端。

它是尽可能基于你的代码。它现在是一个完整的独立演示。如果您在 http://localhost:8000/上打开浏览器并在表单中写入“模拟错误”并按提交，它会模拟运行时错误。

from http.server import HTTPServer, BaseHTTPRequestHandler
from socketserver import ForkingMixIn
from urllib.parse import parse_qs


class MyHandler(BaseHTTPRequestHandler):
    def do_POST(self):
        ctype, pdict = self.headers.get_params(header='content-type')[0]
        postvars = {}
        try:
            if ctype == 'application/x-www-form-urlencoded':
                length = int(self.headers['content-length'])
                postvars = parse_qs(self.rfile.read(length), keep_blank_values=1
                                    encoding='utf-8')
                assert postvars.get('foo', '') != ['bar']  # can simulate error
            body = 'Something\n'.encode('ascii')
            self.send_response(200)
            self.send_header("Content-type", "text")
            self.send_header("Content-length", str(len(body)))
            self.end_headers()
            self.wfile.write(body)
        except Exception:
            self.send_error(500)
            raise

    def do_GET(self):
        # demo for testing POST by web browser - without valid html
        body = ('Something\n<form method="post" action="http://%s:%d/">\n'
                '<input name="foo" type="text"><input type="submit"></form>\n'
                % self.server.server_address).encode('ascii')
        self.send_response(200)
        self.send_header("Content-type", "text/html")
        self.send_header("Content-length", str(len(body)))
        self.end_headers()
        self.wfile.write(body)


class ForkingHTTPServer(ForkingMixIn, HTTPServer):
    def finish_request(self, request, client_address):
        request.settimeout(30)
        # "super" can not be used because BaseServer is not created from object
        HTTPServer.finish_request(self, request, client_address)


def httpd(handler_class=MyHandler, server_address=('localhost', 8000)):
    try:
        print("Server started")
        srvr = ForkingHTTPServer(server_address, handler_class)
        srvr.serve_forever()  # serve_forever
    except KeyboardInterrupt:
        srvr.socket.close()


if __name__ == "__main__":
    httpd()

fork 可以被禁用，例如通过从代码中删除类 SocketServer.ForkingMixIn 进行调试。

针对 Python 3 进行了编辑更新，但应注意来自 Python 3 文档的警告:

Warning: http.server is not recommended for production. It only implements basic security checks.