| OLD | NEW |
|---|
| (Empty) |
| 1 #!/usr/bin/env python | |
| 2 # | |
| 3 # Copyright (c) 2012 The Chromium Authors. All rights reserved. | |
| 4 # Use of this source code is governed by a BSD-style license that can be | |
| 5 # found in the LICENSE file. | |
| 6 | |
| 7 """Provides a convenient wrapper for spawning a test lighttpd instance. | |
| 8 | |
| 9 Usage: | |
| 10 lighttpd_server PATH_TO_DOC_ROOT | |
| 11 """ | |
| 12 | |
| 13 import codecs | |
| 14 import contextlib | |
| 15 import httplib | |
| 16 import os | |
| 17 import random | |
| 18 import shutil | |
| 19 import socket | |
| 20 import subprocess | |
| 21 import sys | |
| 22 import tempfile | |
| 23 import time | |
| 24 | |
| 25 from pylib import constants | |
| 26 from pylib import pexpect | |
| 27 | |
| 28 class LighttpdServer(object): | |
| 29 """Wraps lighttpd server, providing robust startup. | |
| 30 | |
| 31 Args: | |
| 32 document_root: Path to root of this server's hosted files. | |
| 33 port: TCP port on the _host_ machine that the server will listen on. If | |
| 34 ommitted it will attempt to use 9000, or if unavailable it will find | |
| 35 a free port from 8001 - 8999. | |
| 36 lighttpd_path, lighttpd_module_path: Optional paths to lighttpd binaries. | |
| 37 base_config_path: If supplied this file will replace the built-in default | |
| 38 lighttpd config file. | |
| 39 extra_config_contents: If specified, this string will be appended to the | |
| 40 base config (default built-in, or from base_config_path). | |
| 41 config_path, error_log, access_log: Optional paths where the class should | |
| 42 place temprary files for this session. | |
| 43 """ | |
| 44 | |
| 45 def __init__(self, document_root, port=None, | |
| 46 lighttpd_path=None, lighttpd_module_path=None, | |
| 47 base_config_path=None, extra_config_contents=None, | |
| 48 config_path=None, error_log=None, access_log=None): | |
| 49 self.temp_dir = tempfile.mkdtemp(prefix='lighttpd_for_chrome_android') | |
| 50 self.document_root = os.path.abspath(document_root) | |
| 51 self.fixed_port = port | |
| 52 self.port = port or constants.LIGHTTPD_DEFAULT_PORT | |
| 53 self.server_tag = 'LightTPD ' + str(random.randint(111111, 999999)) | |
| 54 self.lighttpd_path = lighttpd_path or '/usr/sbin/lighttpd' | |
| 55 self.lighttpd_module_path = lighttpd_module_path or '/usr/lib/lighttpd' | |
| 56 self.base_config_path = base_config_path | |
| 57 self.extra_config_contents = extra_config_contents | |
| 58 self.config_path = config_path or self._Mktmp('config') | |
| 59 self.error_log = error_log or self._Mktmp('error_log') | |
| 60 self.access_log = access_log or self._Mktmp('access_log') | |
| 61 self.pid_file = self._Mktmp('pid_file') | |
| 62 self.process = None | |
| 63 | |
| 64 def _Mktmp(self, name): | |
| 65 return os.path.join(self.temp_dir, name) | |
| 66 | |
| 67 @staticmethod | |
| 68 def _GetRandomPort(): | |
| 69 # The ports of test server is arranged in constants.py. | |
| 70 return random.randint(constants.LIGHTTPD_RANDOM_PORT_FIRST, | |
| 71 constants.LIGHTTPD_RANDOM_PORT_LAST) | |
| 72 | |
| 73 def StartupHttpServer(self): | |
| 74 """Starts up a http server with specified document root and port.""" | |
| 75 # If we want a specific port, make sure no one else is listening on it. | |
| 76 if self.fixed_port: | |
| 77 self._KillProcessListeningOnPort(self.fixed_port) | |
| 78 while True: | |
| 79 if self.base_config_path: | |
| 80 # Read the config | |
| 81 with codecs.open(self.base_config_path, 'r', 'utf-8') as f: | |
| 82 config_contents = f.read() | |
| 83 else: | |
| 84 config_contents = self._GetDefaultBaseConfig() | |
| 85 if self.extra_config_contents: | |
| 86 config_contents += self.extra_config_contents | |
| 87 # Write out the config, filling in placeholders from the members of |self| | |
| 88 with codecs.open(self.config_path, 'w', 'utf-8') as f: | |
| 89 f.write(config_contents % self.__dict__) | |
| 90 if (not os.path.exists(self.lighttpd_path) or | |
| 91 not os.access(self.lighttpd_path, os.X_OK)): | |
| 92 raise EnvironmentError( | |
| 93 'Could not find lighttpd at %s.\n' | |
| 94 'It may need to be installed (e.g. sudo apt-get install lighttpd)' | |
| 95 % self.lighttpd_path) | |
| 96 self.process = pexpect.spawn(self.lighttpd_path, | |
| 97 ['-D', '-f', self.config_path, | |
| 98 '-m', self.lighttpd_module_path], | |
| 99 cwd=self.temp_dir) | |
| 100 client_error, server_error = self._TestServerConnection() | |
| 101 if not client_error: | |
| 102 assert int(open(self.pid_file, 'r').read()) == self.process.pid | |
| 103 break | |
| 104 self.process.close() | |
| 105 | |
| 106 if self.fixed_port or not 'in use' in server_error: | |
| 107 print 'Client error:', client_error | |
| 108 print 'Server error:', server_error | |
| 109 return False | |
| 110 self.port = self._GetRandomPort() | |
| 111 return True | |
| 112 | |
| 113 def ShutdownHttpServer(self): | |
| 114 """Shuts down our lighttpd processes.""" | |
| 115 if self.process: | |
| 116 self.process.terminate() | |
| 117 shutil.rmtree(self.temp_dir, ignore_errors=True) | |
| 118 | |
| 119 def _TestServerConnection(self): | |
| 120 # Wait for server to start | |
| 121 server_msg = '' | |
| 122 for timeout in xrange(1, 5): | |
| 123 client_error = None | |
| 124 try: | |
| 125 with contextlib.closing(httplib.HTTPConnection( | |
| 126 '127.0.0.1', self.port, timeout=timeout)) as http: | |
| 127 http.set_debuglevel(timeout > 3) | |
| 128 http.request('HEAD', '/') | |
| 129 r = http.getresponse() | |
| 130 r.read() | |
| 131 if (r.status == 200 and r.reason == 'OK' and | |
| 132 r.getheader('Server') == self.server_tag): | |
| 133 return (None, server_msg) | |
| 134 client_error = ('Bad response: %s %s version %s\n ' % | |
| 135 (r.status, r.reason, r.version) + | |
| 136 '\n '.join([': '.join(h) for h in r.getheaders()])) | |
| 137 except (httplib.HTTPException, socket.error) as client_error: | |
| 138 pass # Probably too quick connecting: try again | |
| 139 # Check for server startup error messages | |
| 140 ix = self.process.expect([pexpect.TIMEOUT, pexpect.EOF, '.+'], | |
| 141 timeout=timeout) | |
| 142 if ix == 2: # stdout spew from the server | |
| 143 server_msg += self.process.match.group(0) | |
| 144 elif ix == 1: # EOF -- server has quit so giveup. | |
| 145 client_error = client_error or 'Server exited' | |
| 146 break | |
| 147 return (client_error or 'Timeout', server_msg) | |
| 148 | |
| 149 @staticmethod | |
| 150 def _KillProcessListeningOnPort(port): | |
| 151 """Checks if there is a process listening on port number |port| and | |
| 152 terminates it if found. | |
| 153 | |
| 154 Args: | |
| 155 port: Port number to check. | |
| 156 """ | |
| 157 if subprocess.call(['fuser', '-kv', '%d/tcp' % port]) == 0: | |
| 158 # Give the process some time to terminate and check that it is gone. | |
| 159 time.sleep(2) | |
| 160 assert subprocess.call(['fuser', '-v', '%d/tcp' % port]) != 0, \ | |
| 161 'Unable to kill process listening on port %d.' % port | |
| 162 | |
| 163 @staticmethod | |
| 164 def _GetDefaultBaseConfig(): | |
| 165 return """server.tag = "%(server_tag)s" | |
| 166 server.modules = ( "mod_access", | |
| 167 "mod_accesslog", | |
| 168 "mod_alias", | |
| 169 "mod_cgi", | |
| 170 "mod_rewrite" ) | |
| 171 | |
| 172 # default document root required | |
| 173 #server.document-root = "." | |
| 174 | |
| 175 # files to check for if .../ is requested | |
| 176 index-file.names = ( "index.php", "index.pl", "index.cgi", | |
| 177 "index.html", "index.htm", "default.htm" ) | |
| 178 # mimetype mapping | |
| 179 mimetype.assign = ( | |
| 180 ".gif" => "image/gif", | |
| 181 ".jpg" => "image/jpeg", | |
| 182 ".jpeg" => "image/jpeg", | |
| 183 ".png" => "image/png", | |
| 184 ".svg" => "image/svg+xml", | |
| 185 ".css" => "text/css", | |
| 186 ".html" => "text/html", | |
| 187 ".htm" => "text/html", | |
| 188 ".xhtml" => "application/xhtml+xml", | |
| 189 ".xhtmlmp" => "application/vnd.wap.xhtml+xml", | |
| 190 ".js" => "application/x-javascript", | |
| 191 ".log" => "text/plain", | |
| 192 ".conf" => "text/plain", | |
| 193 ".text" => "text/plain", | |
| 194 ".txt" => "text/plain", | |
| 195 ".dtd" => "text/xml", | |
| 196 ".xml" => "text/xml", | |
| 197 ".manifest" => "text/cache-manifest", | |
| 198 ) | |
| 199 | |
| 200 # Use the "Content-Type" extended attribute to obtain mime type if possible | |
| 201 mimetype.use-xattr = "enable" | |
| 202 | |
| 203 ## | |
| 204 # which extensions should not be handle via static-file transfer | |
| 205 # | |
| 206 # .php, .pl, .fcgi are most often handled by mod_fastcgi or mod_cgi | |
| 207 static-file.exclude-extensions = ( ".php", ".pl", ".cgi" ) | |
| 208 | |
| 209 server.bind = "127.0.0.1" | |
| 210 server.port = %(port)s | |
| 211 | |
| 212 ## virtual directory listings | |
| 213 dir-listing.activate = "enable" | |
| 214 #dir-listing.encoding = "iso-8859-2" | |
| 215 #dir-listing.external-css = "style/oldstyle.css" | |
| 216 | |
| 217 ## enable debugging | |
| 218 #debug.log-request-header = "enable" | |
| 219 #debug.log-response-header = "enable" | |
| 220 #debug.log-request-handling = "enable" | |
| 221 #debug.log-file-not-found = "enable" | |
| 222 | |
| 223 #### SSL engine | |
| 224 #ssl.engine = "enable" | |
| 225 #ssl.pemfile = "server.pem" | |
| 226 | |
| 227 # Autogenerated test-specific config follows. | |
| 228 | |
| 229 cgi.assign = ( ".cgi" => "/usr/bin/env", | |
| 230 ".pl" => "/usr/bin/env", | |
| 231 ".asis" => "/bin/cat", | |
| 232 ".php" => "/usr/bin/php-cgi" ) | |
| 233 | |
| 234 server.errorlog = "%(error_log)s" | |
| 235 accesslog.filename = "%(access_log)s" | |
| 236 server.upload-dirs = ( "/tmp" ) | |
| 237 server.pid-file = "%(pid_file)s" | |
| 238 server.document-root = "%(document_root)s" | |
| 239 | |
| 240 """ | |
| 241 | |
| 242 | |
| 243 def main(argv): | |
| 244 server = LighttpdServer(*argv[1:]) | |
| 245 try: | |
| 246 if server.StartupHttpServer(): | |
| 247 raw_input('Server running at http://127.0.0.1:%s -' | |
| 248 ' press Enter to exit it.' % server.port) | |
| 249 else: | |
| 250 print 'Server exit code:', server.process.exitstatus | |
| 251 finally: | |
| 252 server.ShutdownHttpServer() | |
| 253 | |
| 254 | |
| 255 if __name__ == '__main__': | |
| 256 sys.exit(main(sys.argv)) | |
| OLD | NEW |
|---|
Chromium Code Reviews
Issue 2392643003:
Removes files from //build that we don't need (Closed)
