Index: net/tools/testserver/testserver_base.py |
diff --git a/net/tools/testserver/testserver_base.py b/net/tools/testserver/testserver_base.py |
deleted file mode 100644 |
index 0d3f65fdca95a2482530367083ba8d7f6df4f0a4..0000000000000000000000000000000000000000 |
--- a/net/tools/testserver/testserver_base.py |
+++ /dev/null |
@@ -1,258 +0,0 @@ |
-# Copyright 2013 The Chromium Authors. All rights reserved. |
-# Use of this source code is governed by a BSD-style license that can be |
-# found in the LICENSE file. |
- |
-import BaseHTTPServer |
-import errno |
-import json |
-import optparse |
-import os |
-import re |
-import socket |
-import SocketServer |
-import struct |
-import sys |
-import warnings |
- |
-import tlslite.errors |
- |
-# Ignore deprecation warnings, they make our output more cluttered. |
-warnings.filterwarnings("ignore", category=DeprecationWarning) |
- |
-if sys.platform == 'win32': |
- import msvcrt |
- |
-# Using debug() seems to cause hangs on XP: see http://crbug.com/64515. |
-debug_output = sys.stderr |
-def debug(string): |
- debug_output.write(string + "\n") |
- debug_output.flush() |
- |
- |
-class Error(Exception): |
- """Error class for this module.""" |
- |
- |
-class OptionError(Error): |
- """Error for bad command line options.""" |
- |
- |
-class FileMultiplexer(object): |
- def __init__(self, fd1, fd2) : |
- self.__fd1 = fd1 |
- self.__fd2 = fd2 |
- |
- def __del__(self) : |
- if self.__fd1 != sys.stdout and self.__fd1 != sys.stderr: |
- self.__fd1.close() |
- if self.__fd2 != sys.stdout and self.__fd2 != sys.stderr: |
- self.__fd2.close() |
- |
- def write(self, text) : |
- self.__fd1.write(text) |
- self.__fd2.write(text) |
- |
- def flush(self) : |
- self.__fd1.flush() |
- self.__fd2.flush() |
- |
- |
-class ClientRestrictingServerMixIn: |
- """Implements verify_request to limit connections to our configured IP |
- address.""" |
- |
- def verify_request(self, _request, client_address): |
- return client_address[0] == self.server_address[0] |
- |
- |
-class BrokenPipeHandlerMixIn: |
- """Allows the server to deal with "broken pipe" errors (which happen if the |
- browser quits with outstanding requests, like for the favicon). This mix-in |
- requires the class to derive from SocketServer.BaseServer and not override its |
- handle_error() method. """ |
- |
- def handle_error(self, request, client_address): |
- value = sys.exc_info()[1] |
- if isinstance(value, tlslite.errors.TLSClosedConnectionError): |
- print "testserver.py: Closed connection" |
- return |
- if isinstance(value, socket.error): |
- err = value.args[0] |
- if sys.platform in ('win32', 'cygwin'): |
- # "An established connection was aborted by the software in your host." |
- pipe_err = 10053 |
- else: |
- pipe_err = errno.EPIPE |
- if err == pipe_err: |
- print "testserver.py: Broken pipe" |
- return |
- if err == errno.ECONNRESET: |
- print "testserver.py: Connection reset by peer" |
- return |
- SocketServer.BaseServer.handle_error(self, request, client_address) |
- |
- |
-class StoppableHTTPServer(BaseHTTPServer.HTTPServer): |
- """This is a specialization of BaseHTTPServer to allow it |
- to be exited cleanly (by setting its "stop" member to True).""" |
- |
- def serve_forever(self): |
- self.stop = False |
- self.nonce_time = None |
- while not self.stop: |
- self.handle_request() |
- self.socket.close() |
- |
- |
-def MultiplexerHack(std_fd, log_fd): |
- """Creates a FileMultiplexer that will write to both specified files. |
- |
- When running on Windows XP bots, stdout and stderr will be invalid file |
- handles, so log_fd will be returned directly. (This does not occur if you |
- run the test suite directly from a console, but only if the output of the |
- test executable is redirected.) |
- """ |
- if std_fd.fileno() <= 0: |
- return log_fd |
- return FileMultiplexer(std_fd, log_fd) |
- |
- |
-class BasePageHandler(BaseHTTPServer.BaseHTTPRequestHandler): |
- |
- def __init__(self, request, client_address, socket_server, |
- connect_handlers, get_handlers, head_handlers, post_handlers, |
- put_handlers): |
- self._connect_handlers = connect_handlers |
- self._get_handlers = get_handlers |
- self._head_handlers = head_handlers |
- self._post_handlers = post_handlers |
- self._put_handlers = put_handlers |
- BaseHTTPServer.BaseHTTPRequestHandler.__init__( |
- self, request, client_address, socket_server) |
- |
- def log_request(self, *args, **kwargs): |
- # Disable request logging to declutter test log output. |
- pass |
- |
- def _ShouldHandleRequest(self, handler_name): |
- """Determines if the path can be handled by the handler. |
- |
- We consider a handler valid if the path begins with the |
- handler name. It can optionally be followed by "?*", "/*". |
- """ |
- |
- pattern = re.compile('%s($|\?|/).*' % handler_name) |
- return pattern.match(self.path) |
- |
- def do_CONNECT(self): |
- for handler in self._connect_handlers: |
- if handler(): |
- return |
- |
- def do_GET(self): |
- for handler in self._get_handlers: |
- if handler(): |
- return |
- |
- def do_HEAD(self): |
- for handler in self._head_handlers: |
- if handler(): |
- return |
- |
- def do_POST(self): |
- for handler in self._post_handlers: |
- if handler(): |
- return |
- |
- def do_PUT(self): |
- for handler in self._put_handlers: |
- if handler(): |
- return |
- |
- |
-class TestServerRunner(object): |
- """Runs a test server and communicates with the controlling C++ test code. |
- |
- Subclasses should override the create_server method to create their server |
- object, and the add_options method to add their own options. |
- """ |
- |
- def __init__(self): |
- self.option_parser = optparse.OptionParser() |
- self.add_options() |
- |
- def main(self): |
- self.options, self.args = self.option_parser.parse_args() |
- |
- logfile = open(self.options.log_file, 'w') |
- sys.stderr = MultiplexerHack(sys.stderr, logfile) |
- if self.options.log_to_console: |
- sys.stdout = MultiplexerHack(sys.stdout, logfile) |
- else: |
- sys.stdout = logfile |
- |
- server_data = { |
- 'host': self.options.host, |
- } |
- self.server = self.create_server(server_data) |
- self._notify_startup_complete(server_data) |
- self.run_server() |
- |
- def create_server(self, server_data): |
- """Creates a server object and returns it. |
- |
- Must populate server_data['port'], and can set additional server_data |
- elements if desired.""" |
- raise NotImplementedError() |
- |
- def run_server(self): |
- try: |
- self.server.serve_forever() |
- except KeyboardInterrupt: |
- print 'shutting down server' |
- self.server.stop = True |
- |
- def add_options(self): |
- self.option_parser.add_option('--startup-pipe', type='int', |
- dest='startup_pipe', |
- help='File handle of pipe to parent process') |
- self.option_parser.add_option('--log-to-console', action='store_const', |
- const=True, default=False, |
- dest='log_to_console', |
- help='Enables or disables sys.stdout logging ' |
- 'to the console.') |
- self.option_parser.add_option('--log-file', default='testserver.log', |
- dest='log_file', |
- help='The name of the server log file.') |
- self.option_parser.add_option('--port', default=0, type='int', |
- help='Port used by the server. If ' |
- 'unspecified, the server will listen on an ' |
- 'ephemeral port.') |
- self.option_parser.add_option('--host', default='127.0.0.1', |
- dest='host', |
- help='Hostname or IP upon which the server ' |
- 'will listen. Client connections will also ' |
- 'only be allowed from this address.') |
- self.option_parser.add_option('--data-dir', dest='data_dir', |
- help='Directory from which to read the ' |
- 'files.') |
- |
- def _notify_startup_complete(self, server_data): |
- # Notify the parent that we've started. (BaseServer subclasses |
- # bind their sockets on construction.) |
- if self.options.startup_pipe is not None: |
- server_data_json = json.dumps(server_data) |
- server_data_len = len(server_data_json) |
- print 'sending server_data: %s (%d bytes)' % ( |
- server_data_json, server_data_len) |
- if sys.platform == 'win32': |
- fd = msvcrt.open_osfhandle(self.options.startup_pipe, 0) |
- else: |
- fd = self.options.startup_pipe |
- startup_pipe = os.fdopen(fd, "w") |
- # First write the data length as an unsigned 4-byte value. This |
- # is _not_ using network byte ordering since the other end of the |
- # pipe is on the same machine. |
- startup_pipe.write(struct.pack('=L', server_data_len)) |
- startup_pipe.write(server_data_json) |
- startup_pipe.close() |