| # Copyright (C) 2011 Google Inc. All rights reserved. |
| # Copyright (C) 2016 Apple Inc. All rights reserved. |
| # |
| # Redistribution and use in source and binary forms, with or without |
| # modification, are permitted provided that the following conditions are |
| # met: |
| # |
| # * Redistributions of source code must retain the above copyright |
| # notice, this list of conditions and the following disclaimer. |
| # * Redistributions in binary form must reproduce the above |
| # copyright notice, this list of conditions and the following disclaimer |
| # in the documentation and/or other materials provided with the |
| # distribution. |
| # * Neither the name of Google Inc. nor the names of its |
| # contributors may be used to endorse or promote products derived from |
| # this software without specific prior written permission. |
| # |
| # THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS |
| # "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT |
| # LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR |
| # A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT |
| # OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, |
| # SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT |
| # LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, |
| # DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY |
| # THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT |
| # (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE |
| # OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
| |
| """A class to help start/stop the PyWebSocket server used by layout tests.""" |
| |
| import logging |
| import os |
| import sys |
| import time |
| |
| from webkitpy.layout_tests.servers import http_server, http_server_base |
| |
| _log = logging.getLogger(__name__) |
| |
| |
| _WS_LOG_NAME = 'pywebsocket.ws.log' |
| _WSS_LOG_NAME = 'pywebsocket.wss.log' |
| |
| |
| class PyWebSocket(http_server.Lighttpd): |
| DEFAULT_WS_PORT = 8880 |
| DEFAULT_WSS_PORT = 9323 |
| |
| def __init__(self, port_obj, output_dir, port=DEFAULT_WS_PORT, |
| root=None, use_tls=False, |
| private_key=None, certificate=None, ca_certificate=None, |
| pidfile=None): |
| """Args: |
| output_dir: the absolute path to the layout test result directory |
| """ |
| http_server.Lighttpd.__init__(self, port_obj, output_dir, |
| port=port, |
| root=root) |
| self._output_dir = output_dir |
| self._pid_file = pidfile |
| self._process = None |
| |
| self._port = port |
| self._root = root |
| self._use_tls = use_tls |
| |
| self._name = 'pywebsocket' |
| if self._use_tls: |
| self._name = 'pywebsocket_secure' |
| |
| if private_key: |
| self._private_key = private_key |
| else: |
| self._private_key = self._pem_file |
| if certificate: |
| self._certificate = certificate |
| else: |
| self._certificate = self._pem_file |
| self._ca_certificate = ca_certificate |
| if self._port: |
| self._port = int(self._port) |
| self._wsin = None |
| self._wsout = None |
| self._mappings = [{'port': self._port}] |
| |
| if not self._pid_file: |
| self._pid_file = self._filesystem.join(self._runtime_path, '%s.pid' % self._name) |
| |
| # Webkit tests |
| # FIXME: This is the wrong way to detect if we're in Chrome vs. WebKit! |
| # The port objects are supposed to abstract this. |
| if self._root: |
| self._layout_tests = self._filesystem.abspath(self._root) |
| self._web_socket_tests = self._filesystem.abspath(self._filesystem.join(self._root, 'http', 'tests', 'websocket', 'tests')) |
| else: |
| try: |
| self._layout_tests = self._port_obj.layout_tests_dir() |
| self._web_socket_tests = self._filesystem.join(self._layout_tests, 'http', 'tests', 'websocket', 'tests') |
| except: |
| self._web_socket_tests = None |
| |
| if self._use_tls: |
| self._log_prefix = _WSS_LOG_NAME |
| else: |
| self._log_prefix = _WS_LOG_NAME |
| |
| def ports_to_forward(self): |
| return [self._port] |
| |
| def _prepare_config(self): |
| self._filesystem.maybe_make_directory(self._output_dir) |
| log_file_name = self._log_prefix |
| # FIXME: Doesn't Executive have a devnull, so that we don't have to use os.devnull directly? |
| self._wsin = open(os.devnull, 'r') |
| |
| error_log = self._filesystem.join(self._output_dir, log_file_name + "-err.txt") |
| output_log = self._filesystem.join(self._output_dir, log_file_name + "-out.txt") |
| self._wsout = self._filesystem.open_text_file_for_writing(output_log) |
| |
| # FIXME https://bugs.webkit.org/show_bug.cgi?id=206546: Should be using the same version of Python run-webkit-tests is |
| python_interp = sys.executable |
| if sys.version_info > (3, 0): |
| python_interp = 'python2' |
| |
| wpt_tools_base = self._filesystem.join(self._layout_tests, "imported", "w3c", "web-platform-tests", "tools") |
| pywebsocket_base = self._filesystem.join(wpt_tools_base, "pywebsocket") |
| pywebsocket_deps = [self._filesystem.join(wpt_tools_base, "third_party", "six")] |
| pywebsocket_script = self._filesystem.join(pywebsocket_base, 'mod_pywebsocket', 'standalone.py') |
| start_cmd = [ |
| python_interp, '-u', pywebsocket_script, |
| '--server-host', 'localhost', |
| '--port', str(self._port), |
| # FIXME: Don't we have a self._port_obj.layout_test_path? |
| '--document-root', self._filesystem.join(self._layout_tests, 'http', 'tests'), |
| '--scan-dir', self._web_socket_tests, |
| '--cgi-paths', '/websocket/tests', |
| '--log-file', error_log, |
| ] |
| |
| handler_map_file = self._filesystem.join(self._web_socket_tests, 'handler_map.txt') |
| if self._filesystem.exists(handler_map_file): |
| _log.debug('Using handler_map_file: %s' % handler_map_file) |
| start_cmd.append('--websock-handlers-map-file') |
| start_cmd.append(handler_map_file) |
| else: |
| _log.warning('No handler_map_file found') |
| |
| if self._use_tls: |
| start_cmd.extend(['-t', '-k', self._private_key, |
| '-c', self._certificate]) |
| if self._ca_certificate: |
| start_cmd.append('--tls-client-ca') |
| start_cmd.append(self._ca_certificate) |
| |
| self._start_cmd = start_cmd |
| server_name = self._filesystem.basename(pywebsocket_script) |
| self._env = self._port_obj.setup_environ_for_server(server_name) |
| self._env['PYTHONPATH'] = os.path.pathsep.join([pywebsocket_base] + pywebsocket_deps + [self._env.get('PYTHONPATH', '')]) |
| |
| def _remove_stale_logs(self): |
| try: |
| self._remove_log_files(self._output_dir, self._log_prefix) |
| except OSError as e: |
| _log.warning('Failed to remove stale %s log files: %s' % (self._name, str(e))) |
| |
| def _spawn_process(self): |
| _log.debug('Starting %s server, cmd="%s"' % (self._name, self._start_cmd)) |
| self._process = self._executive.popen(self._start_cmd, env=self._env, shell=False, stdin=self._wsin, stdout=self._wsout, stderr=self._executive.STDOUT) |
| self._filesystem.write_text_file(self._pid_file, str(self._process.pid)) |
| return self._process.pid |
| |
| def _stop_running_server(self): |
| super(PyWebSocket, self)._stop_running_server() |
| |
| if self._wsin: |
| self._wsin.close() |
| self._wsin = None |
| if self._wsout: |
| self._wsout.close() |
| self._wsout = None |
| |
| |
| def is_web_socket_server_running(): |
| return http_server_base.HttpServerBase._is_running_on_port(PyWebSocket.DEFAULT_WS_PORT) |
