Chris Sosa | 7c93136 | 2010-10-11 19:49:01 -0700 | [diff] [blame] | 1 | #!/usr/bin/python |
| 2 | |
Chris Sosa | 781ba6d | 2012-04-11 12:44:43 -0700 | [diff] [blame] | 3 | # Copyright (c) 2009-2012 The Chromium OS Authors. All rights reserved. |
rtc@google.com | ded2240 | 2009-10-26 22:36:21 +0000 | [diff] [blame] | 4 | # Use of this source code is governed by a BSD-style license that can be |
| 5 | # found in the LICENSE file. |
| 6 | |
Chris Sosa | 7c93136 | 2010-10-11 19:49:01 -0700 | [diff] [blame] | 7 | """A CherryPy-based webserver to host images and build packages.""" |
| 8 | |
Chris Sosa | dbc2008 | 2012-12-10 13:39:11 -0800 | [diff] [blame] | 9 | import cherrypy |
Gilad Arnold | 55a2a37 | 2012-10-02 09:46:32 -0700 | [diff] [blame] | 10 | import json |
Sean O'Connor | 14b6a0a | 2010-03-20 23:23:48 -0700 | [diff] [blame] | 11 | import optparse |
rtc@google.com | ded2240 | 2009-10-26 22:36:21 +0000 | [diff] [blame] | 12 | import os |
Scott Zawalski | 4647ce6 | 2012-01-03 17:17:28 -0500 | [diff] [blame] | 13 | import re |
Simran Basi | 4baad08 | 2013-02-14 13:39:18 -0800 | [diff] [blame] | 14 | import shutil |
Mandeep Singh Baines | 38dcdda | 2012-12-07 17:55:33 -0800 | [diff] [blame] | 15 | import socket |
chocobo@google.com | 4dc2581 | 2009-10-27 23:46:26 +0000 | [diff] [blame] | 16 | import sys |
Chris Masone | 816e38c | 2012-05-02 12:22:36 -0700 | [diff] [blame] | 17 | import subprocess |
| 18 | import tempfile |
Gilad Arnold | d5ebaaa | 2012-10-02 11:52:38 -0700 | [diff] [blame] | 19 | import types |
rtc@google.com | ded2240 | 2009-10-26 22:36:21 +0000 | [diff] [blame] | 20 | |
Chris Sosa | 0356d3b | 2010-09-16 15:46:22 -0700 | [diff] [blame] | 21 | import autoupdate |
Gilad Arnold | c65330c | 2012-09-20 15:17:48 -0700 | [diff] [blame] | 22 | import common_util |
Chris Sosa | 47a7d4e | 2012-03-28 11:26:55 -0700 | [diff] [blame] | 23 | import downloader |
Gilad Arnold | c65330c | 2012-09-20 15:17:48 -0700 | [diff] [blame] | 24 | import log_util |
| 25 | |
| 26 | |
| 27 | # Module-local log function. |
Chris Sosa | 6a3697f | 2013-01-29 16:44:43 -0800 | [diff] [blame] | 28 | def _Log(message, *args): |
| 29 | return log_util.LogWithTag('DEVSERVER', message, *args) |
Chris Sosa | 0356d3b | 2010-09-16 15:46:22 -0700 | [diff] [blame] | 30 | |
Frank Farzan | 4016087 | 2011-12-12 18:39:18 -0800 | [diff] [blame] | 31 | |
Chris Sosa | 417e55d | 2011-01-25 16:40:48 -0800 | [diff] [blame] | 32 | CACHED_ENTRIES = 12 |
Don Garrett | f90edf0 | 2010-11-16 17:36:14 -0800 | [diff] [blame] | 33 | |
Simran Basi | 4baad08 | 2013-02-14 13:39:18 -0800 | [diff] [blame] | 34 | TELEMETRY_FOLDER = 'telemetry_src' |
| 35 | TELEMETRY_DEPS = ['dep-telemetry_dep.tar.bz2', |
| 36 | 'dep-page_cycler_dep.tar.bz2', |
| 37 | 'dep-chrome_test.tar.bz2'] |
| 38 | |
Chris Sosa | 0356d3b | 2010-09-16 15:46:22 -0700 | [diff] [blame] | 39 | # Sets up global to share between classes. |
rtc@google.com | 21a5ca3 | 2009-11-04 18:23:23 +0000 | [diff] [blame] | 40 | updater = None |
rtc@google.com | ded2240 | 2009-10-26 22:36:21 +0000 | [diff] [blame] | 41 | |
Frank Farzan | 4016087 | 2011-12-12 18:39:18 -0800 | [diff] [blame] | 42 | |
Chris Sosa | 9164ca3 | 2012-03-28 11:04:50 -0700 | [diff] [blame] | 43 | class DevServerError(Exception): |
Chris Sosa | 47a7d4e | 2012-03-28 11:26:55 -0700 | [diff] [blame] | 44 | """Exception class used by this module.""" |
| 45 | pass |
| 46 | |
| 47 | |
Scott Zawalski | 4647ce6 | 2012-01-03 17:17:28 -0500 | [diff] [blame] | 48 | def _LeadingWhiteSpaceCount(string): |
| 49 | """Count the amount of leading whitespace in a string. |
| 50 | |
| 51 | Args: |
| 52 | string: The string to count leading whitespace in. |
| 53 | Returns: |
| 54 | number of white space chars before characters start. |
| 55 | """ |
| 56 | matched = re.match('^\s+', string) |
| 57 | if matched: |
| 58 | return len(matched.group()) |
| 59 | |
| 60 | return 0 |
| 61 | |
| 62 | |
| 63 | def _PrintDocStringAsHTML(func): |
| 64 | """Make a functions docstring somewhat HTML style. |
| 65 | |
| 66 | Args: |
| 67 | func: The function to return the docstring from. |
| 68 | Returns: |
| 69 | A string that is somewhat formated for a web browser. |
| 70 | """ |
| 71 | # TODO(scottz): Make this parse Args/Returns in a prettier way. |
| 72 | # Arguments could be bolded and indented etc. |
| 73 | html_doc = [] |
| 74 | for line in func.__doc__.splitlines(): |
| 75 | leading_space = _LeadingWhiteSpaceCount(line) |
| 76 | if leading_space > 0: |
Chris Sosa | 47a7d4e | 2012-03-28 11:26:55 -0700 | [diff] [blame] | 77 | line = ' ' * leading_space + line |
Scott Zawalski | 4647ce6 | 2012-01-03 17:17:28 -0500 | [diff] [blame] | 78 | |
| 79 | html_doc.append('<BR>%s' % line) |
| 80 | |
| 81 | return '\n'.join(html_doc) |
| 82 | |
| 83 | |
Chris Sosa | 7c93136 | 2010-10-11 19:49:01 -0700 | [diff] [blame] | 84 | def _GetConfig(options): |
| 85 | """Returns the configuration for the devserver.""" |
Mandeep Singh Baines | 38dcdda | 2012-12-07 17:55:33 -0800 | [diff] [blame] | 86 | |
| 87 | # On a system with IPv6 not compiled into the kernel, |
| 88 | # AF_INET6 sockets will return a socket.error exception. |
| 89 | # On such systems, fall-back to IPv4. |
| 90 | socket_host = '::' |
| 91 | try: |
| 92 | socket.socket(socket.AF_INET6, socket.SOCK_STREAM) |
| 93 | except socket.error: |
| 94 | socket_host = '0.0.0.0' |
| 95 | |
Chris Sosa | 7c93136 | 2010-10-11 19:49:01 -0700 | [diff] [blame] | 96 | base_config = { 'global': |
| 97 | { 'server.log_request_headers': True, |
| 98 | 'server.protocol_version': 'HTTP/1.1', |
Mandeep Singh Baines | 38dcdda | 2012-12-07 17:55:33 -0800 | [diff] [blame] | 99 | 'server.socket_host': socket_host, |
Chris Sosa | 7c93136 | 2010-10-11 19:49:01 -0700 | [diff] [blame] | 100 | 'server.socket_port': int(options.port), |
Chris Sosa | 374c62d | 2010-10-14 09:13:54 -0700 | [diff] [blame] | 101 | 'response.timeout': 6000, |
Chris Sosa | 6fe2394 | 2012-07-02 15:44:46 -0700 | [diff] [blame] | 102 | 'request.show_tracebacks': True, |
Chris Sosa | 72333d1 | 2012-06-13 11:28:05 -0700 | [diff] [blame] | 103 | 'server.socket_timeout': 60, |
Zdenek Behan | 1347a31 | 2011-02-10 03:59:17 +0100 | [diff] [blame] | 104 | 'tools.staticdir.root': |
| 105 | os.path.dirname(os.path.abspath(sys.argv[0])), |
Chris Sosa | 7c93136 | 2010-10-11 19:49:01 -0700 | [diff] [blame] | 106 | }, |
Dale Curtis | c9aaf3a | 2011-08-09 15:47:40 -0700 | [diff] [blame] | 107 | '/api': |
| 108 | { |
| 109 | # Gets rid of cherrypy parsing post file for args. |
| 110 | 'request.process_request_body': False, |
| 111 | }, |
Chris Sosa | a1ef010 | 2010-10-21 16:22:35 -0700 | [diff] [blame] | 112 | '/build': |
| 113 | { |
| 114 | 'response.timeout': 100000, |
| 115 | }, |
Chris Sosa | 7c93136 | 2010-10-11 19:49:01 -0700 | [diff] [blame] | 116 | '/update': |
| 117 | { |
| 118 | # Gets rid of cherrypy parsing post file for args. |
| 119 | 'request.process_request_body': False, |
Chris Sosa | f65f4b9 | 2010-10-21 15:57:51 -0700 | [diff] [blame] | 120 | 'response.timeout': 10000, |
Chris Sosa | 7c93136 | 2010-10-11 19:49:01 -0700 | [diff] [blame] | 121 | }, |
| 122 | # Sets up the static dir for file hosting. |
| 123 | '/static': |
| 124 | { 'tools.staticdir.dir': 'static', |
| 125 | 'tools.staticdir.on': True, |
Chris Sosa | f65f4b9 | 2010-10-21 15:57:51 -0700 | [diff] [blame] | 126 | 'response.timeout': 10000, |
Chris Sosa | 7c93136 | 2010-10-11 19:49:01 -0700 | [diff] [blame] | 127 | }, |
| 128 | } |
Chris Sosa | 5f118ef | 2012-07-12 11:37:50 -0700 | [diff] [blame] | 129 | if options.production: |
Chris Sosa | d1ea86b | 2012-07-12 13:35:37 -0700 | [diff] [blame] | 130 | base_config['global'].update({'server.thread_pool': 75}) |
Scott Zawalski | 1c5e7cd | 2012-02-27 13:12:52 -0500 | [diff] [blame] | 131 | |
Chris Sosa | 7c93136 | 2010-10-11 19:49:01 -0700 | [diff] [blame] | 132 | return base_config |
rtc@google.com | 6424466 | 2009-11-12 00:52:08 +0000 | [diff] [blame] | 133 | |
Darin Petkov | e17164a | 2010-08-11 13:24:41 -0700 | [diff] [blame] | 134 | |
Zdenek Behan | 608f46c | 2011-02-19 00:47:16 +0100 | [diff] [blame] | 135 | def _PrepareToServeUpdatesOnly(image_dir, static_dir): |
Chris Sosa | 0356d3b | 2010-09-16 15:46:22 -0700 | [diff] [blame] | 136 | """Sets up symlink to image_dir for serving purposes.""" |
| 137 | assert os.path.exists(image_dir), '%s must exist.' % image_dir |
| 138 | # If we're serving out of an archived build dir (e.g. a |
| 139 | # buildbot), prepare this webserver's magic 'static/' dir with a |
| 140 | # link to the build archive. |
Gilad Arnold | c65330c | 2012-09-20 15:17:48 -0700 | [diff] [blame] | 141 | _Log('Preparing autoupdate for "serve updates only" mode.') |
Zdenek Behan | 608f46c | 2011-02-19 00:47:16 +0100 | [diff] [blame] | 142 | if os.path.lexists('%s/archive' % static_dir): |
| 143 | if image_dir != os.readlink('%s/archive' % static_dir): |
Gilad Arnold | c65330c | 2012-09-20 15:17:48 -0700 | [diff] [blame] | 144 | _Log('removing stale symlink to %s' % image_dir) |
Zdenek Behan | 608f46c | 2011-02-19 00:47:16 +0100 | [diff] [blame] | 145 | os.unlink('%s/archive' % static_dir) |
| 146 | os.symlink(image_dir, '%s/archive' % static_dir) |
Chris Sosa | cde6bf4 | 2012-05-31 18:36:39 -0700 | [diff] [blame] | 147 | |
Chris Sosa | 0356d3b | 2010-09-16 15:46:22 -0700 | [diff] [blame] | 148 | else: |
Zdenek Behan | 608f46c | 2011-02-19 00:47:16 +0100 | [diff] [blame] | 149 | os.symlink(image_dir, '%s/archive' % static_dir) |
Chris Sosa | cde6bf4 | 2012-05-31 18:36:39 -0700 | [diff] [blame] | 150 | |
Gilad Arnold | c65330c | 2012-09-20 15:17:48 -0700 | [diff] [blame] | 151 | _Log('archive dir: %s ready to be used to serve images.' % image_dir) |
Chris Sosa | 7c93136 | 2010-10-11 19:49:01 -0700 | [diff] [blame] | 152 | |
| 153 | |
Gilad Arnold | d5ebaaa | 2012-10-02 11:52:38 -0700 | [diff] [blame] | 154 | def _GetRecursiveMemberObject(root, member_list): |
| 155 | """Returns an object corresponding to a nested member list. |
| 156 | |
| 157 | Args: |
| 158 | root: the root object to search |
| 159 | member_list: list of nested members to search |
| 160 | Returns: |
| 161 | An object corresponding to the member name list; None otherwise. |
| 162 | """ |
| 163 | for member in member_list: |
| 164 | next_root = root.__class__.__dict__.get(member) |
| 165 | if not next_root: |
| 166 | return None |
| 167 | root = next_root |
| 168 | return root |
| 169 | |
| 170 | |
| 171 | def _IsExposed(name): |
| 172 | """Returns True iff |name| has an `exposed' attribute and it is set.""" |
| 173 | return hasattr(name, 'exposed') and name.exposed |
| 174 | |
| 175 | |
Gilad Arnold | 748c832 | 2012-10-12 09:51:35 -0700 | [diff] [blame] | 176 | def _GetExposedMethod(root, nested_member, ignored=None): |
Gilad Arnold | d5ebaaa | 2012-10-02 11:52:38 -0700 | [diff] [blame] | 177 | """Returns a CherryPy-exposed method, if such exists. |
| 178 | |
| 179 | Args: |
| 180 | root: the root object for searching |
| 181 | nested_member: a slash-joined path to the nested member |
| 182 | ignored: method paths to be ignored |
| 183 | Returns: |
| 184 | A function object corresponding to the path defined by |member_list| from |
| 185 | the |root| object, if the function is exposed and not ignored; None |
| 186 | otherwise. |
| 187 | """ |
Gilad Arnold | 748c832 | 2012-10-12 09:51:35 -0700 | [diff] [blame] | 188 | method = (not (ignored and nested_member in ignored) and |
Gilad Arnold | d5ebaaa | 2012-10-02 11:52:38 -0700 | [diff] [blame] | 189 | _GetRecursiveMemberObject(root, nested_member.split('/'))) |
| 190 | if (method and type(method) == types.FunctionType and _IsExposed(method)): |
| 191 | return method |
| 192 | |
| 193 | |
Gilad Arnold | 748c832 | 2012-10-12 09:51:35 -0700 | [diff] [blame] | 194 | def _FindExposedMethods(root, prefix, unlisted=None): |
Gilad Arnold | d5ebaaa | 2012-10-02 11:52:38 -0700 | [diff] [blame] | 195 | """Finds exposed CherryPy methods. |
| 196 | |
| 197 | Args: |
| 198 | root: the root object for searching |
| 199 | prefix: slash-joined chain of members leading to current object |
| 200 | unlisted: URLs to be excluded regardless of their exposed status |
| 201 | Returns: |
| 202 | List of exposed URLs that are not unlisted. |
| 203 | """ |
| 204 | method_list = [] |
| 205 | for member in sorted(root.__class__.__dict__.keys()): |
| 206 | prefixed_member = prefix + '/' + member if prefix else member |
Gilad Arnold | 748c832 | 2012-10-12 09:51:35 -0700 | [diff] [blame] | 207 | if unlisted and prefixed_member in unlisted: |
Gilad Arnold | d5ebaaa | 2012-10-02 11:52:38 -0700 | [diff] [blame] | 208 | continue |
| 209 | member_obj = root.__class__.__dict__[member] |
| 210 | if _IsExposed(member_obj): |
| 211 | if type(member_obj) == types.FunctionType: |
| 212 | method_list.append(prefixed_member) |
| 213 | else: |
| 214 | method_list += _FindExposedMethods( |
| 215 | member_obj, prefixed_member, unlisted) |
| 216 | return method_list |
| 217 | |
| 218 | |
Dale Curtis | c9aaf3a | 2011-08-09 15:47:40 -0700 | [diff] [blame] | 219 | class ApiRoot(object): |
| 220 | """RESTful API for Dev Server information.""" |
| 221 | exposed = True |
| 222 | |
| 223 | @cherrypy.expose |
| 224 | def hostinfo(self, ip): |
| 225 | """Returns a JSON dictionary containing information about the given ip. |
| 226 | |
Gilad Arnold | 1b90839 | 2012-10-05 11:36:27 -0700 | [diff] [blame] | 227 | Args: |
| 228 | ip: address of host whose info is requested |
| 229 | Returns: |
| 230 | A JSON dictionary containing all or some of the following fields: |
| 231 | last_event_type (int): last update event type received |
| 232 | last_event_status (int): last update event status received |
| 233 | last_known_version (string): last known version reported in update ping |
| 234 | forced_update_label (string): update label to force next update ping to |
| 235 | use, set by setnextupdate |
| 236 | See the OmahaEvent class in update_engine/omaha_request_action.h for |
| 237 | event type and status code definitions. If the ip does not exist an empty |
| 238 | string is returned. |
Dale Curtis | c9aaf3a | 2011-08-09 15:47:40 -0700 | [diff] [blame] | 239 | |
Gilad Arnold | 1b90839 | 2012-10-05 11:36:27 -0700 | [diff] [blame] | 240 | Example URL: |
| 241 | http://myhost/api/hostinfo?ip=192.168.1.5 |
| 242 | """ |
Dale Curtis | c9aaf3a | 2011-08-09 15:47:40 -0700 | [diff] [blame] | 243 | return updater.HandleHostInfoPing(ip) |
| 244 | |
| 245 | @cherrypy.expose |
Gilad Arnold | 286a006 | 2012-01-12 13:47:02 -0800 | [diff] [blame] | 246 | def hostlog(self, ip): |
Gilad Arnold | 1b90839 | 2012-10-05 11:36:27 -0700 | [diff] [blame] | 247 | """Returns a JSON object containing a log of host event. |
| 248 | |
| 249 | Args: |
| 250 | ip: address of host whose event log is requested, or `all' |
| 251 | Returns: |
| 252 | A JSON encoded list (log) of dictionaries (events), each of which |
| 253 | containing a `timestamp' and other event fields, as described under |
| 254 | /api/hostinfo. |
| 255 | |
| 256 | Example URL: |
| 257 | http://myhost/api/hostlog?ip=192.168.1.5 |
| 258 | """ |
Gilad Arnold | 286a006 | 2012-01-12 13:47:02 -0800 | [diff] [blame] | 259 | return updater.HandleHostLogPing(ip) |
| 260 | |
| 261 | @cherrypy.expose |
Dale Curtis | c9aaf3a | 2011-08-09 15:47:40 -0700 | [diff] [blame] | 262 | def setnextupdate(self, ip): |
| 263 | """Allows the response to the next update ping from a host to be set. |
| 264 | |
| 265 | Takes the IP of the host and an update label as normally provided to the |
Gilad Arnold | 1b90839 | 2012-10-05 11:36:27 -0700 | [diff] [blame] | 266 | /update command. |
| 267 | """ |
Dale Curtis | c9aaf3a | 2011-08-09 15:47:40 -0700 | [diff] [blame] | 268 | body_length = int(cherrypy.request.headers['Content-Length']) |
| 269 | label = cherrypy.request.rfile.read(body_length) |
| 270 | |
| 271 | if label: |
| 272 | label = label.strip() |
| 273 | if label: |
| 274 | return updater.HandleSetUpdatePing(ip, label) |
| 275 | raise cherrypy.HTTPError(400, 'No label provided.') |
| 276 | |
| 277 | |
Gilad Arnold | 55a2a37 | 2012-10-02 09:46:32 -0700 | [diff] [blame] | 278 | @cherrypy.expose |
| 279 | def fileinfo(self, *path_args): |
| 280 | """Returns information about a given staged file. |
| 281 | |
| 282 | Args: |
| 283 | path_args: path to the file inside the server's static staging directory |
| 284 | Returns: |
| 285 | A JSON encoded dictionary with information about the said file, which may |
| 286 | contain the following keys/values: |
Gilad Arnold | 1b90839 | 2012-10-05 11:36:27 -0700 | [diff] [blame] | 287 | size (int): the file size in bytes |
| 288 | sha1 (string): a base64 encoded SHA1 hash |
| 289 | sha256 (string): a base64 encoded SHA256 hash |
| 290 | |
| 291 | Example URL: |
| 292 | http://myhost/api/fileinfo/some/path/to/file |
Gilad Arnold | 55a2a37 | 2012-10-02 09:46:32 -0700 | [diff] [blame] | 293 | """ |
| 294 | file_path = os.path.join(updater.static_dir, *path_args) |
| 295 | if not os.path.exists(file_path): |
| 296 | raise DevServerError('file not found: %s' % file_path) |
| 297 | try: |
| 298 | file_size = os.path.getsize(file_path) |
| 299 | file_sha1 = common_util.GetFileSha1(file_path) |
| 300 | file_sha256 = common_util.GetFileSha256(file_path) |
| 301 | except os.error, e: |
| 302 | raise DevServerError('failed to get info for file %s: %s' % |
| 303 | (file_path, str(e))) |
| 304 | return json.dumps( |
| 305 | {'size': file_size, 'sha1': file_sha1, 'sha256': file_sha256}) |
| 306 | |
Chris Sosa | 76e44b9 | 2013-01-31 12:11:38 -0800 | [diff] [blame] | 307 | |
David Rochberg | 7c79a81 | 2011-01-19 14:24:45 -0500 | [diff] [blame] | 308 | class DevServerRoot(object): |
Chris Sosa | 7c93136 | 2010-10-11 19:49:01 -0700 | [diff] [blame] | 309 | """The Root Class for the Dev Server. |
| 310 | |
| 311 | CherryPy works as follows: |
| 312 | For each method in this class, cherrpy interprets root/path |
| 313 | as a call to an instance of DevServerRoot->method_name. For example, |
| 314 | a call to http://myhost/build will call build. CherryPy automatically |
| 315 | parses http args and places them as keyword arguments in each method. |
| 316 | For paths http://myhost/update/dir1/dir2, you can use *args so that |
| 317 | cherrypy uses the update method and puts the extra paths in args. |
| 318 | """ |
Gilad Arnold | f8f769f | 2012-09-24 08:43:01 -0700 | [diff] [blame] | 319 | # Method names that should not be listed on the index page. |
| 320 | _UNLISTED_METHODS = ['index', 'doc'] |
| 321 | |
Dale Curtis | c9aaf3a | 2011-08-09 15:47:40 -0700 | [diff] [blame] | 322 | api = ApiRoot() |
Chris Sosa | 7c93136 | 2010-10-11 19:49:01 -0700 | [diff] [blame] | 323 | |
David Rochberg | 7c79a81 | 2011-01-19 14:24:45 -0500 | [diff] [blame] | 324 | def __init__(self): |
Nick Sanders | 7dcaa2e | 2011-08-04 15:20:41 -0700 | [diff] [blame] | 325 | self._builder = None |
Simran Basi | 4baad08 | 2013-02-14 13:39:18 -0800 | [diff] [blame] | 326 | self._telemetry_lock_dict = common_util.LockDict() |
David Rochberg | 7c79a81 | 2011-01-19 14:24:45 -0500 | [diff] [blame] | 327 | |
Dale Curtis | c9aaf3a | 2011-08-09 15:47:40 -0700 | [diff] [blame] | 328 | @cherrypy.expose |
David Rochberg | 7c79a81 | 2011-01-19 14:24:45 -0500 | [diff] [blame] | 329 | def build(self, board, pkg, **kwargs): |
Chris Sosa | 7c93136 | 2010-10-11 19:49:01 -0700 | [diff] [blame] | 330 | """Builds the package specified.""" |
Nick Sanders | 7dcaa2e | 2011-08-04 15:20:41 -0700 | [diff] [blame] | 331 | import builder |
| 332 | if self._builder is None: |
| 333 | self._builder = builder.Builder() |
David Rochberg | 7c79a81 | 2011-01-19 14:24:45 -0500 | [diff] [blame] | 334 | return self._builder.Build(board, pkg, kwargs) |
Chris Sosa | 7c93136 | 2010-10-11 19:49:01 -0700 | [diff] [blame] | 335 | |
Chris Sosa | cde6bf4 | 2012-05-31 18:36:39 -0700 | [diff] [blame] | 336 | @staticmethod |
| 337 | def _canonicalize_archive_url(archive_url): |
| 338 | """Canonicalizes archive_url strings. |
| 339 | |
| 340 | Raises: |
| 341 | DevserverError: if archive_url is not set. |
| 342 | """ |
| 343 | if archive_url: |
Chris Sosa | 76e44b9 | 2013-01-31 12:11:38 -0800 | [diff] [blame] | 344 | if not archive_url.startswith('gs://'): |
| 345 | raise DevServerError("Archive URL isn't from Google Storage.") |
| 346 | |
Chris Sosa | cde6bf4 | 2012-05-31 18:36:39 -0700 | [diff] [blame] | 347 | return archive_url.rstrip('/') |
| 348 | else: |
| 349 | raise DevServerError("Must specify an archive_url in the request") |
| 350 | |
Dale Curtis | c9aaf3a | 2011-08-09 15:47:40 -0700 | [diff] [blame] | 351 | @cherrypy.expose |
Frank Farzan | bcb571e | 2012-01-03 11:48:17 -0800 | [diff] [blame] | 352 | def download(self, **kwargs): |
| 353 | """Downloads and archives full/delta payloads from Google Storage. |
| 354 | |
Chris Sosa | 76e44b9 | 2013-01-31 12:11:38 -0800 | [diff] [blame] | 355 | THIS METHOD IS DEPRECATED: use stage(..., artifacts=...) instead. |
Chris Sosa | 47a7d4e | 2012-03-28 11:26:55 -0700 | [diff] [blame] | 356 | This methods downloads artifacts. It may download artifacts in the |
| 357 | background in which case a caller should call wait_for_status to get |
| 358 | the status of the background artifact downloads. They should use the same |
| 359 | args passed to download. |
| 360 | |
Frank Farzan | bcb571e | 2012-01-03 11:48:17 -0800 | [diff] [blame] | 361 | Args: |
| 362 | archive_url: Google Storage URL for the build. |
| 363 | |
| 364 | Example URL: |
Gilad Arnold | f8f769f | 2012-09-24 08:43:01 -0700 | [diff] [blame] | 365 | http://myhost/download?archive_url=gs://chromeos-image-archive/ |
| 366 | x86-generic/R17-1208.0.0-a1-b338 |
Frank Farzan | bcb571e | 2012-01-03 11:48:17 -0800 | [diff] [blame] | 367 | """ |
Chris Sosa | 76e44b9 | 2013-01-31 12:11:38 -0800 | [diff] [blame] | 368 | return self.stage(archive_url=kwargs.get('archive_url'), |
| 369 | artifacts='full_payload,test_suites,stateful') |
| 370 | |
| 371 | @cherrypy.expose |
| 372 | def stage(self, **kwargs): |
| 373 | """Downloads and caches the artifacts from Google Storage URL. |
| 374 | |
| 375 | Downloads and caches the artifacts Google Storage URL. Returns once these |
| 376 | have been downloaded on the devserver. A call to this will attempt to cache |
| 377 | non-specified artifacts in the background for the given from the given URL |
| 378 | following the principle of spatial locality. Spatial locality of different |
| 379 | artifacts is explicitly defined in the build_artifact module. |
| 380 | |
| 381 | These artifacts will then be available from the static/ sub-directory of |
| 382 | the devserver. |
| 383 | |
| 384 | Args: |
| 385 | archive_url: Google Storage URL for the build. |
| 386 | artifacts: Comma separated list of artifacts to download. |
| 387 | |
| 388 | Example: |
| 389 | To download the autotest and test suites tarballs: |
| 390 | http://devserver_url:<port>/stage?archive_url=gs://your_url/path& |
| 391 | artifacts=autotest,test_suites |
| 392 | To download the full update payload: |
| 393 | http://devserver_url:<port>/stage?archive_url=gs://your_url/path& |
| 394 | artifacts=full_payload |
| 395 | |
| 396 | For both these examples, one could find these artifacts at: |
| 397 | http://devserver_url:<port>/static/archive/<relative_path>* |
| 398 | |
| 399 | Note for this example, relative path is the archive_url stripped of its |
| 400 | basename i.e. path/ in the examples above. Specific example: |
| 401 | |
| 402 | gs://chromeos-image-archive/x86-mario-release/R26-3920.0.0 |
| 403 | |
| 404 | Will get staged to: |
| 405 | |
| 406 | http://devserver_url:<port>/static/archive/x86-mario-release/R26-3920.0.0 |
| 407 | """ |
Chris Sosa | cde6bf4 | 2012-05-31 18:36:39 -0700 | [diff] [blame] | 408 | archive_url = self._canonicalize_archive_url(kwargs.get('archive_url')) |
Chris Sosa | 76e44b9 | 2013-01-31 12:11:38 -0800 | [diff] [blame] | 409 | artifacts = kwargs.get('artifacts', '') |
| 410 | if not artifacts: |
| 411 | raise DevServerError('No artifacts specified.') |
Chris Sosa | 47a7d4e | 2012-03-28 11:26:55 -0700 | [diff] [blame] | 412 | |
Chris Sosa | 76e44b9 | 2013-01-31 12:11:38 -0800 | [diff] [blame] | 413 | downloader.Downloader(updater.static_dir, archive_url).Download( |
| 414 | artifacts.split(',')) |
| 415 | return 'Success' |
Chris Sosa | cde6bf4 | 2012-05-31 18:36:39 -0700 | [diff] [blame] | 416 | |
| 417 | @cherrypy.expose |
Simran Basi | 4baad08 | 2013-02-14 13:39:18 -0800 | [diff] [blame] | 418 | def setup_telemetry(self, **kwargs): |
| 419 | """Extracts and sets up telemetry |
| 420 | |
| 421 | This method goes through the telemetry deps packages, and stages them on |
| 422 | the devserver to be used by the drones and the telemetry tests. |
| 423 | |
| 424 | Args: |
| 425 | archive_url: Google Storage URL for the build. |
| 426 | |
| 427 | Returns: |
| 428 | Path to the source folder for the telemetry codebase once it is staged. |
| 429 | """ |
| 430 | archive_url = kwargs.get('archive_url') |
| 431 | self.stage(archive_url=archive_url, artifacts='autotest') |
| 432 | |
| 433 | build = '/'.join(downloader.Downloader.ParseUrl(archive_url)) |
| 434 | build_path = os.path.join(updater.static_dir, build) |
| 435 | deps_path = os.path.join(build_path, 'autotest/packages') |
| 436 | telemetry_path = os.path.join(build_path, TELEMETRY_FOLDER) |
| 437 | src_folder = os.path.join(telemetry_path, 'src') |
| 438 | |
| 439 | with self._telemetry_lock_dict.lock(telemetry_path): |
| 440 | if os.path.exists(src_folder): |
| 441 | # Telemetry is already fully stage return |
| 442 | return src_folder |
| 443 | |
| 444 | common_util.MkDirP(telemetry_path) |
| 445 | |
| 446 | # Copy over the required deps tar balls to the telemetry directory. |
| 447 | for dep in TELEMETRY_DEPS: |
| 448 | dep_path = os.path.join(deps_path, dep) |
| 449 | try: |
| 450 | common_util.ExtractTarball(dep_path, telemetry_path) |
| 451 | except common_util.CommonUtilError as e: |
| 452 | shutil.rmtree(telemetry_path) |
| 453 | raise DevServerError(str(e)) |
| 454 | |
| 455 | # By default all the tarballs extract to test_src but some parts of |
| 456 | # the telemetry code specifically hardcoded to exist inside of 'src'. |
| 457 | test_src = os.path.join(telemetry_path, 'test_src') |
| 458 | try: |
| 459 | shutil.move(test_src, src_folder) |
| 460 | except shutil.Error: |
| 461 | # This can occur if src_folder already exists. Remove and retry move. |
| 462 | shutil.rmtree(src_folder) |
| 463 | raise DevServerError('Failure in telemetry setup for build %s. Appears' |
| 464 | ' that the test_src to src move failed.' % build) |
| 465 | |
| 466 | return src_folder |
| 467 | |
| 468 | @cherrypy.expose |
Chris Sosa | cde6bf4 | 2012-05-31 18:36:39 -0700 | [diff] [blame] | 469 | def wait_for_status(self, **kwargs): |
| 470 | """Waits for background artifacts to be downloaded from Google Storage. |
| 471 | |
Chris Sosa | 76e44b9 | 2013-01-31 12:11:38 -0800 | [diff] [blame] | 472 | THIS METHOD IS DEPRECATED: use stage(..., artifacts=...) instead. |
Chris Sosa | cde6bf4 | 2012-05-31 18:36:39 -0700 | [diff] [blame] | 473 | Args: |
| 474 | archive_url: Google Storage URL for the build. |
| 475 | |
| 476 | Example URL: |
Gilad Arnold | f8f769f | 2012-09-24 08:43:01 -0700 | [diff] [blame] | 477 | http://myhost/wait_for_status?archive_url=gs://chromeos-image-archive/ |
| 478 | x86-generic/R17-1208.0.0-a1-b338 |
Chris Sosa | cde6bf4 | 2012-05-31 18:36:39 -0700 | [diff] [blame] | 479 | """ |
Chris Sosa | 76e44b9 | 2013-01-31 12:11:38 -0800 | [diff] [blame] | 480 | return self.stage(archive_url=kwargs.get('archive_url'), |
| 481 | artifacts='full_payload,test_suites,autotest,stateful') |
Chris Sosa | 47a7d4e | 2012-03-28 11:26:55 -0700 | [diff] [blame] | 482 | |
| 483 | @cherrypy.expose |
Chris Masone | 816e38c | 2012-05-02 12:22:36 -0700 | [diff] [blame] | 484 | def stage_debug(self, **kwargs): |
| 485 | """Downloads and stages debug symbol payloads from Google Storage. |
| 486 | |
Chris Sosa | 76e44b9 | 2013-01-31 12:11:38 -0800 | [diff] [blame] | 487 | THIS METHOD IS DEPRECATED: use stage(..., artifacts=...) instead. |
| 488 | This methods downloads the debug symbol build artifact |
| 489 | synchronously, and then stages it for use by symbolicate_dump. |
Chris Masone | 816e38c | 2012-05-02 12:22:36 -0700 | [diff] [blame] | 490 | |
| 491 | Args: |
| 492 | archive_url: Google Storage URL for the build. |
| 493 | |
| 494 | Example URL: |
Gilad Arnold | f8f769f | 2012-09-24 08:43:01 -0700 | [diff] [blame] | 495 | http://myhost/stage_debug?archive_url=gs://chromeos-image-archive/ |
| 496 | x86-generic/R17-1208.0.0-a1-b338 |
Chris Masone | 816e38c | 2012-05-02 12:22:36 -0700 | [diff] [blame] | 497 | """ |
Chris Sosa | 76e44b9 | 2013-01-31 12:11:38 -0800 | [diff] [blame] | 498 | return self.stage(archive_url=kwargs.get('archive_url'), |
| 499 | artifacts='symbols') |
Chris Masone | 816e38c | 2012-05-02 12:22:36 -0700 | [diff] [blame] | 500 | |
| 501 | @cherrypy.expose |
Chris Sosa | 76e44b9 | 2013-01-31 12:11:38 -0800 | [diff] [blame] | 502 | def symbolicate_dump(self, minidump, **kwargs): |
Chris Masone | 816e38c | 2012-05-02 12:22:36 -0700 | [diff] [blame] | 503 | """Symbolicates a minidump using pre-downloaded symbols, returns it. |
| 504 | |
| 505 | Callers will need to POST to this URL with a body of MIME-type |
| 506 | "multipart/form-data". |
| 507 | The body should include a single argument, 'minidump', containing the |
| 508 | binary-formatted minidump to symbolicate. |
| 509 | |
Chris Masone | 816e38c | 2012-05-02 12:22:36 -0700 | [diff] [blame] | 510 | Args: |
Chris Sosa | 76e44b9 | 2013-01-31 12:11:38 -0800 | [diff] [blame] | 511 | archive_url: Google Storage URL for the build. |
Chris Masone | 816e38c | 2012-05-02 12:22:36 -0700 | [diff] [blame] | 512 | minidump: The binary minidump file to symbolicate. |
| 513 | """ |
Chris Sosa | 76e44b9 | 2013-01-31 12:11:38 -0800 | [diff] [blame] | 514 | # Ensure the symbols have been staged. |
| 515 | archive_url = self._canonicalize_archive_url(kwargs.get('archive_url')) |
| 516 | if self.stage(archive_url=archive_url, artifacts='symbols') != 'Success': |
| 517 | raise DevServerError('Failed to stage symbols for %s' % archive_url) |
| 518 | |
Chris Masone | 816e38c | 2012-05-02 12:22:36 -0700 | [diff] [blame] | 519 | to_return = '' |
| 520 | with tempfile.NamedTemporaryFile() as local: |
| 521 | while True: |
| 522 | data = minidump.file.read(8192) |
| 523 | if not data: |
| 524 | break |
| 525 | local.write(data) |
Chris Sosa | 76e44b9 | 2013-01-31 12:11:38 -0800 | [diff] [blame] | 526 | |
Chris Masone | 816e38c | 2012-05-02 12:22:36 -0700 | [diff] [blame] | 527 | local.flush() |
Chris Sosa | 76e44b9 | 2013-01-31 12:11:38 -0800 | [diff] [blame] | 528 | |
| 529 | symbols_directory = os.path.join(downloader.Downloader.GetBuildDir( |
| 530 | updater.static_dir, archive_url), 'debug', 'breakpad') |
| 531 | |
| 532 | stackwalk = subprocess.Popen( |
| 533 | ['minidump_stackwalk', local.name, symbols_directory], |
| 534 | stdout=subprocess.PIPE, stderr=subprocess.PIPE) |
| 535 | |
Chris Masone | 816e38c | 2012-05-02 12:22:36 -0700 | [diff] [blame] | 536 | to_return, error_text = stackwalk.communicate() |
| 537 | if stackwalk.returncode != 0: |
| 538 | raise DevServerError("Can't generate stack trace: %s (rc=%d)" % ( |
| 539 | error_text, stackwalk.returncode)) |
| 540 | |
| 541 | return to_return |
| 542 | |
| 543 | @cherrypy.expose |
Scott Zawalski | 1695453 | 2012-03-20 15:31:36 -0400 | [diff] [blame] | 544 | def latestbuild(self, **params): |
| 545 | """Return a string representing the latest build for a given target. |
| 546 | |
| 547 | Args: |
| 548 | target: The build target, typically a combination of the board and the |
| 549 | type of build e.g. x86-mario-release. |
| 550 | milestone: The milestone to filter builds on. E.g. R16. Optional, if not |
| 551 | provided the latest RXX build will be returned. |
| 552 | Returns: |
| 553 | A string representation of the latest build if one exists, i.e. |
| 554 | R19-1993.0.0-a1-b1480. |
| 555 | An empty string if no latest could be found. |
| 556 | """ |
| 557 | if not params: |
| 558 | return _PrintDocStringAsHTML(self.latestbuild) |
| 559 | |
| 560 | if 'target' not in params: |
| 561 | raise cherrypy.HTTPError('500 Internal Server Error', |
| 562 | 'Error: target= is required!') |
| 563 | try: |
Gilad Arnold | c65330c | 2012-09-20 15:17:48 -0700 | [diff] [blame] | 564 | return common_util.GetLatestBuildVersion( |
Scott Zawalski | 1695453 | 2012-03-20 15:31:36 -0400 | [diff] [blame] | 565 | updater.static_dir, params['target'], |
| 566 | milestone=params.get('milestone')) |
Gilad Arnold | 17fe03d | 2012-10-02 10:05:01 -0700 | [diff] [blame] | 567 | except common_util.CommonUtilError as errmsg: |
Scott Zawalski | 1695453 | 2012-03-20 15:31:36 -0400 | [diff] [blame] | 568 | raise cherrypy.HTTPError('500 Internal Server Error', str(errmsg)) |
| 569 | |
| 570 | @cherrypy.expose |
Scott Zawalski | 84a39c9 | 2012-01-13 15:12:42 -0500 | [diff] [blame] | 571 | def controlfiles(self, **params): |
Scott Zawalski | 4647ce6 | 2012-01-03 17:17:28 -0500 | [diff] [blame] | 572 | """Return a control file or a list of all known control files. |
| 573 | |
| 574 | Example URL: |
| 575 | To List all control files: |
Scott Zawalski | 84a39c9 | 2012-01-13 15:12:42 -0500 | [diff] [blame] | 576 | http://dev-server/controlfiles?board=x86-alex-release&build=R18-1514.0.0 |
Scott Zawalski | 4647ce6 | 2012-01-03 17:17:28 -0500 | [diff] [blame] | 577 | To return the contents of a path: |
Scott Zawalski | 84a39c9 | 2012-01-13 15:12:42 -0500 | [diff] [blame] | 578 | http://dev-server/controlfiles?board=x86-alex-release&build=R18-1514.0.0&control_path=client/sleeptest/control |
Scott Zawalski | 4647ce6 | 2012-01-03 17:17:28 -0500 | [diff] [blame] | 579 | |
| 580 | Args: |
Scott Zawalski | 84a39c9 | 2012-01-13 15:12:42 -0500 | [diff] [blame] | 581 | build: The build i.e. x86-alex-release/R18-1514.0.0-a1-b1450. |
Scott Zawalski | 4647ce6 | 2012-01-03 17:17:28 -0500 | [diff] [blame] | 582 | control_path: If you want the contents of a control file set this |
| 583 | to the path. E.g. client/site_tests/sleeptest/control |
| 584 | Optional, if not provided return a list of control files is returned. |
| 585 | Returns: |
| 586 | Contents of a control file if control_path is provided. |
| 587 | A list of control files if no control_path is provided. |
| 588 | """ |
Scott Zawalski | 4647ce6 | 2012-01-03 17:17:28 -0500 | [diff] [blame] | 589 | if not params: |
| 590 | return _PrintDocStringAsHTML(self.controlfiles) |
| 591 | |
Scott Zawalski | 84a39c9 | 2012-01-13 15:12:42 -0500 | [diff] [blame] | 592 | if 'build' not in params: |
Scott Zawalski | 4647ce6 | 2012-01-03 17:17:28 -0500 | [diff] [blame] | 593 | raise cherrypy.HTTPError('500 Internal Server Error', |
Scott Zawalski | 84a39c9 | 2012-01-13 15:12:42 -0500 | [diff] [blame] | 594 | 'Error: build= is required!') |
Scott Zawalski | 4647ce6 | 2012-01-03 17:17:28 -0500 | [diff] [blame] | 595 | |
| 596 | if 'control_path' not in params: |
Gilad Arnold | c65330c | 2012-09-20 15:17:48 -0700 | [diff] [blame] | 597 | return common_util.GetControlFileList( |
| 598 | updater.static_dir, params['build']) |
Scott Zawalski | 4647ce6 | 2012-01-03 17:17:28 -0500 | [diff] [blame] | 599 | else: |
Gilad Arnold | c65330c | 2012-09-20 15:17:48 -0700 | [diff] [blame] | 600 | return common_util.GetControlFile( |
| 601 | updater.static_dir, params['build'], params['control_path']) |
Frank Farzan | 4016087 | 2011-12-12 18:39:18 -0800 | [diff] [blame] | 602 | |
| 603 | @cherrypy.expose |
Gilad Arnold | 6f99b98 | 2012-09-12 10:49:40 -0700 | [diff] [blame] | 604 | def stage_images(self, **kwargs): |
| 605 | """Downloads and stages a Chrome OS image from Google Storage. |
| 606 | |
Chris Sosa | 76e44b9 | 2013-01-31 12:11:38 -0800 | [diff] [blame] | 607 | THIS METHOD IS DEPRECATED: use stage(..., artifacts=...) instead. |
Gilad Arnold | 6f99b98 | 2012-09-12 10:49:40 -0700 | [diff] [blame] | 608 | This method downloads a zipped archive from a specified GS location, then |
| 609 | extracts and stages the specified list of images and stages them under |
Chris Sosa | 76e44b9 | 2013-01-31 12:11:38 -0800 | [diff] [blame] | 610 | static/BOARD/BUILD/. Download is synchronous. |
Gilad Arnold | 6f99b98 | 2012-09-12 10:49:40 -0700 | [diff] [blame] | 611 | |
| 612 | Args: |
| 613 | archive_url: Google Storage URL for the build. |
| 614 | image_types: comma-separated list of images to download, may include |
| 615 | 'test', 'recovery', and 'base' |
| 616 | |
| 617 | Example URL: |
| 618 | http://myhost/stage_images?archive_url=gs://chromeos-image-archive/ |
| 619 | x86-generic/R17-1208.0.0-a1-b338&image_types=test,base |
| 620 | """ |
Gilad Arnold | 6f99b98 | 2012-09-12 10:49:40 -0700 | [diff] [blame] | 621 | image_types = kwargs.get('image_types').split(',') |
Chris Sosa | 76e44b9 | 2013-01-31 12:11:38 -0800 | [diff] [blame] | 622 | image_types_list = [image + '_image' for image in image_types] |
| 623 | self.stage(archive_url=kwargs.get('archive_url'), artifacts=','.join( |
| 624 | image_types_list)) |
Gilad Arnold | 6f99b98 | 2012-09-12 10:49:40 -0700 | [diff] [blame] | 625 | |
| 626 | @cherrypy.expose |
Chris Sosa | 7c93136 | 2010-10-11 19:49:01 -0700 | [diff] [blame] | 627 | def index(self): |
Gilad Arnold | f8f769f | 2012-09-24 08:43:01 -0700 | [diff] [blame] | 628 | """Presents a welcome message and documentation links.""" |
Gilad Arnold | f8f769f | 2012-09-24 08:43:01 -0700 | [diff] [blame] | 629 | return ('Welcome to the Dev Server!<br>\n' |
| 630 | '<br>\n' |
| 631 | 'Here are the available methods, click for documentation:<br>\n' |
| 632 | '<br>\n' |
| 633 | '%s' % |
| 634 | '<br>\n'.join( |
| 635 | [('<a href=doc/%s>%s</a>' % (name, name)) |
Gilad Arnold | d5ebaaa | 2012-10-02 11:52:38 -0700 | [diff] [blame] | 636 | for name in _FindExposedMethods( |
| 637 | self, '', unlisted=self._UNLISTED_METHODS)])) |
Gilad Arnold | f8f769f | 2012-09-24 08:43:01 -0700 | [diff] [blame] | 638 | |
| 639 | @cherrypy.expose |
| 640 | def doc(self, *args): |
| 641 | """Shows the documentation for available methods / URLs. |
| 642 | |
| 643 | Example: |
| 644 | http://myhost/doc/update |
| 645 | """ |
Gilad Arnold | d5ebaaa | 2012-10-02 11:52:38 -0700 | [diff] [blame] | 646 | name = '/'.join(args) |
| 647 | method = _GetExposedMethod(self, name) |
Gilad Arnold | f8f769f | 2012-09-24 08:43:01 -0700 | [diff] [blame] | 648 | if not method: |
| 649 | raise DevServerError("No exposed method named `%s'" % name) |
| 650 | if not method.__doc__: |
| 651 | raise DevServerError("No documentation for exposed method `%s'" % name) |
| 652 | return '<pre>\n%s</pre>' % method.__doc__ |
Chris Sosa | 7c93136 | 2010-10-11 19:49:01 -0700 | [diff] [blame] | 653 | |
Dale Curtis | c9aaf3a | 2011-08-09 15:47:40 -0700 | [diff] [blame] | 654 | @cherrypy.expose |
Chris Sosa | 7c93136 | 2010-10-11 19:49:01 -0700 | [diff] [blame] | 655 | def update(self, *args): |
Gilad Arnold | f8f769f | 2012-09-24 08:43:01 -0700 | [diff] [blame] | 656 | """Handles an update check from a Chrome OS client. |
| 657 | |
| 658 | The HTTP request should contain the standard Omaha-style XML blob. The URL |
| 659 | line may contain an additional intermediate path to the update payload. |
| 660 | |
| 661 | Example: |
| 662 | http://myhost/update/optional/path/to/payload |
| 663 | """ |
Chris Sosa | 7c93136 | 2010-10-11 19:49:01 -0700 | [diff] [blame] | 664 | label = '/'.join(args) |
Gilad Arnold | 286a006 | 2012-01-12 13:47:02 -0800 | [diff] [blame] | 665 | body_length = int(cherrypy.request.headers.get('Content-Length', 0)) |
Chris Sosa | 7c93136 | 2010-10-11 19:49:01 -0700 | [diff] [blame] | 666 | data = cherrypy.request.rfile.read(body_length) |
| 667 | return updater.HandleUpdatePing(data, label) |
| 668 | |
Chris Sosa | 0356d3b | 2010-09-16 15:46:22 -0700 | [diff] [blame] | 669 | |
Chris Sosa | dbc2008 | 2012-12-10 13:39:11 -0800 | [diff] [blame] | 670 | def _CleanCache(cache_dir, wipe): |
| 671 | """Wipes any excess cached items in the cache_dir. |
| 672 | |
| 673 | Args: |
| 674 | cache_dir: the directory we are wiping from. |
| 675 | wipe: If True, wipe all the contents -- not just the excess. |
| 676 | """ |
| 677 | if wipe: |
| 678 | # Clear the cache and exit on error. |
| 679 | cmd = 'rm -rf %s/*' % cache_dir |
| 680 | if os.system(cmd) != 0: |
| 681 | _Log('Failed to clear the cache with %s' % cmd) |
| 682 | sys.exit(1) |
| 683 | else: |
| 684 | # Clear all but the last N cached updates |
| 685 | cmd = ('cd %s; ls -tr | head --lines=-%d | xargs rm -rf' % |
| 686 | (cache_dir, CACHED_ENTRIES)) |
| 687 | if os.system(cmd) != 0: |
| 688 | _Log('Failed to clean up old delta cache files with %s' % cmd) |
| 689 | sys.exit(1) |
| 690 | |
| 691 | |
Chris Sosa | cde6bf4 | 2012-05-31 18:36:39 -0700 | [diff] [blame] | 692 | def main(): |
Sean O'Connor | 14b6a0a | 2010-03-20 23:23:48 -0700 | [diff] [blame] | 693 | usage = 'usage: %prog [options]' |
Gilad Arnold | 286a006 | 2012-01-12 13:47:02 -0800 | [diff] [blame] | 694 | parser = optparse.OptionParser(usage=usage) |
Gilad Arnold | 9714d9b | 2012-10-04 10:09:42 -0700 | [diff] [blame] | 695 | parser.add_option('--archive_dir', |
| 696 | metavar='PATH', |
Chris Sosa | dbc2008 | 2012-12-10 13:39:11 -0800 | [diff] [blame] | 697 | help='Enables serve-only mode. Serves archived builds only') |
Gilad Arnold | 9714d9b | 2012-10-04 10:09:42 -0700 | [diff] [blame] | 698 | parser.add_option('--board', |
| 699 | help='when pre-generating update, board for latest image') |
| 700 | parser.add_option('--clear_cache', |
Satoru Takabayashi | d733cbe | 2011-11-15 09:36:32 -0800 | [diff] [blame] | 701 | action='store_true', default=False, |
Gilad Arnold | 9714d9b | 2012-10-04 10:09:42 -0700 | [diff] [blame] | 702 | help='clear out all cached updates and exit') |
| 703 | parser.add_option('--critical_update', |
| 704 | action='store_true', default=False, |
| 705 | help='present update payload as critical') |
| 706 | parser.add_option('--data_dir', |
| 707 | metavar='PATH', |
| 708 | default=os.path.dirname(os.path.abspath(sys.argv[0])), |
| 709 | help='writable directory where static lives') |
| 710 | parser.add_option('--exit', |
| 711 | action='store_true', |
| 712 | help='do not start server (yet pregenerate/clear cache)') |
Gilad Arnold | 9714d9b | 2012-10-04 10:09:42 -0700 | [diff] [blame] | 713 | parser.add_option('--for_vm', |
| 714 | dest='vm', action='store_true', |
| 715 | help='update is for a vm image') |
Gilad Arnold | 8318eac | 2012-10-04 12:52:23 -0700 | [diff] [blame] | 716 | parser.add_option('--host_log', |
| 717 | action='store_true', default=False, |
| 718 | help='record history of host update events (/api/hostlog)') |
Gilad Arnold | 9714d9b | 2012-10-04 10:09:42 -0700 | [diff] [blame] | 719 | parser.add_option('--image', |
| 720 | metavar='FILE', |
Chris Sosa | dbc2008 | 2012-12-10 13:39:11 -0800 | [diff] [blame] | 721 | help='Force update using this image. Can only be used when ' |
| 722 | 'not in serve-only mode as it is used to generate a ' |
| 723 | 'payload.') |
Gilad Arnold | 9714d9b | 2012-10-04 10:09:42 -0700 | [diff] [blame] | 724 | parser.add_option('--logfile', |
| 725 | metavar='PATH', |
| 726 | help='log output to this file instead of stdout') |
Gilad Arnold | a564b4b | 2012-10-04 10:32:44 -0700 | [diff] [blame] | 727 | parser.add_option('--max_updates', |
Chris Sosa | 76e44b9 | 2013-01-31 12:11:38 -0800 | [diff] [blame] | 728 | metavar='NUM', default= -1, type='int', |
Gilad Arnold | a564b4b | 2012-10-04 10:32:44 -0700 | [diff] [blame] | 729 | help='maximum number of update checks handled positively ' |
| 730 | '(default: unlimited)') |
Gilad Arnold | 9714d9b | 2012-10-04 10:09:42 -0700 | [diff] [blame] | 731 | parser.add_option('-p', '--pregenerate_update', |
| 732 | action='store_true', default=False, |
Chris Sosa | dbc2008 | 2012-12-10 13:39:11 -0800 | [diff] [blame] | 733 | help='pre-generate update payload. Can only be used when ' |
| 734 | 'not in serve-only mode as it is used to generate a ' |
| 735 | 'payload.') |
Gilad Arnold | 9714d9b | 2012-10-04 10:09:42 -0700 | [diff] [blame] | 736 | parser.add_option('--payload', |
| 737 | metavar='PATH', |
| 738 | help='use update payload from specified directory') |
| 739 | parser.add_option('--port', |
| 740 | default=8080, type='int', |
| 741 | help='port for the dev server to use (default: 8080)') |
| 742 | parser.add_option('--private_key', |
| 743 | metavar='PATH', default=None, |
| 744 | help='path to the private key in pem format') |
| 745 | parser.add_option('--production', |
| 746 | action='store_true', default=False, |
| 747 | help='have the devserver use production values') |
| 748 | parser.add_option('--proxy_port', |
| 749 | metavar='PORT', default=None, type='int', |
| 750 | help='port to have the client connect to (testing support)') |
| 751 | parser.add_option('--remote_payload', |
| 752 | action='store_true', default=False, |
| 753 | help='Payload is being served from a remote machine') |
| 754 | parser.add_option('--src_image', |
| 755 | metavar='PATH', default='', |
| 756 | help='source image for generating delta updates from') |
| 757 | parser.add_option('-t', '--test_image', |
| 758 | action='store_true', |
| 759 | help='whether or not to use test images') |
| 760 | parser.add_option('-u', '--urlbase', |
| 761 | metavar='URL', |
| 762 | help='base URL for update images, other than the devserver') |
Chris Sosa | 7c93136 | 2010-10-11 19:49:01 -0700 | [diff] [blame] | 763 | (options, _) = parser.parse_args() |
rtc@google.com | 21a5ca3 | 2009-11-04 18:23:23 +0000 | [diff] [blame] | 764 | |
Chris Sosa | 7c93136 | 2010-10-11 19:49:01 -0700 | [diff] [blame] | 765 | devserver_dir = os.path.dirname(os.path.abspath(sys.argv[0])) |
| 766 | root_dir = os.path.realpath('%s/../..' % devserver_dir) |
Chris Sosa | 0356d3b | 2010-09-16 15:46:22 -0700 | [diff] [blame] | 767 | serve_only = False |
| 768 | |
Zdenek Behan | 608f46c | 2011-02-19 00:47:16 +0100 | [diff] [blame] | 769 | static_dir = os.path.realpath('%s/static' % options.data_dir) |
| 770 | os.system('mkdir -p %s' % static_dir) |
| 771 | |
Sean O'Connor | 14b6a0a | 2010-03-20 23:23:48 -0700 | [diff] [blame] | 772 | if options.archive_dir: |
Zdenek Behan | 608f46c | 2011-02-19 00:47:16 +0100 | [diff] [blame] | 773 | # TODO(zbehan) Remove legacy support: |
| 774 | # archive_dir is the directory where static/archive will point. |
| 775 | # If this is an absolute path, all is fine. If someone calls this |
| 776 | # using a relative path, that is relative to src/platform/dev/. |
| 777 | # That use case is unmaintainable, but since applications use it |
| 778 | # with =./static, instead of a boolean flag, we'll make this relative |
| 779 | # to devserver_dir to keep these unbroken. For now. |
| 780 | archive_dir = options.archive_dir |
| 781 | if not os.path.isabs(archive_dir): |
Chris Sosa | 47a7d4e | 2012-03-28 11:26:55 -0700 | [diff] [blame] | 782 | archive_dir = os.path.realpath(os.path.join(devserver_dir, archive_dir)) |
Zdenek Behan | 608f46c | 2011-02-19 00:47:16 +0100 | [diff] [blame] | 783 | _PrepareToServeUpdatesOnly(archive_dir, static_dir) |
Zdenek Behan | 6d93e55 | 2011-03-02 22:35:49 +0100 | [diff] [blame] | 784 | static_dir = os.path.realpath(archive_dir) |
Chris Sosa | 0356d3b | 2010-09-16 15:46:22 -0700 | [diff] [blame] | 785 | serve_only = True |
Chris Sosa | 0356d3b | 2010-09-16 15:46:22 -0700 | [diff] [blame] | 786 | |
Don Garrett | f90edf0 | 2010-11-16 17:36:14 -0800 | [diff] [blame] | 787 | cache_dir = os.path.join(static_dir, 'cache') |
Chris Sosa | dbc2008 | 2012-12-10 13:39:11 -0800 | [diff] [blame] | 788 | # If our devserver is only supposed to serve payloads, we shouldn't be mucking |
| 789 | # with the cache at all. If the devserver hadn't previously generated a cache |
| 790 | # and is expected, the caller is using it wrong. |
| 791 | if serve_only: |
| 792 | # Extra check to make sure we're not being called incorrectly. |
| 793 | if (options.clear_cache or options.exit or options.pregenerate_update or |
| 794 | options.board or options.image): |
| 795 | parser.error('Incompatible flags detected for serve_only mode.') |
Don Garrett | f90edf0 | 2010-11-16 17:36:14 -0800 | [diff] [blame] | 796 | |
Chris Sosa | dbc2008 | 2012-12-10 13:39:11 -0800 | [diff] [blame] | 797 | elif os.path.exists(cache_dir): |
| 798 | _CleanCache(cache_dir, options.clear_cache) |
Chris Sosa | 6b8c374 | 2011-01-31 12:12:17 -0800 | [diff] [blame] | 799 | else: |
| 800 | os.makedirs(cache_dir) |
Don Garrett | f90edf0 | 2010-11-16 17:36:14 -0800 | [diff] [blame] | 801 | |
Chris Sosa | dbc2008 | 2012-12-10 13:39:11 -0800 | [diff] [blame] | 802 | _Log('Using cache directory %s' % cache_dir) |
Gilad Arnold | c65330c | 2012-09-20 15:17:48 -0700 | [diff] [blame] | 803 | _Log('Data dir is %s' % options.data_dir) |
| 804 | _Log('Source root is %s' % root_dir) |
| 805 | _Log('Serving from %s' % static_dir) |
rtc@google.com | 21a5ca3 | 2009-11-04 18:23:23 +0000 | [diff] [blame] | 806 | |
Chris Sosa | 6a3697f | 2013-01-29 16:44:43 -0800 | [diff] [blame] | 807 | # We allow global use here to share with cherrypy classes. |
| 808 | # pylint: disable=W0603 |
Chris Sosa | cde6bf4 | 2012-05-31 18:36:39 -0700 | [diff] [blame] | 809 | global updater |
Andrew de los Reyes | 5262080 | 2010-04-12 13:40:07 -0700 | [diff] [blame] | 810 | updater = autoupdate.Autoupdate( |
| 811 | root_dir=root_dir, |
| 812 | static_dir=static_dir, |
Chris Sosa | 0356d3b | 2010-09-16 15:46:22 -0700 | [diff] [blame] | 813 | serve_only=serve_only, |
Andrew de los Reyes | 5262080 | 2010-04-12 13:40:07 -0700 | [diff] [blame] | 814 | urlbase=options.urlbase, |
| 815 | test_image=options.test_image, |
Chris Sosa | 5d342a2 | 2010-09-28 16:54:41 -0700 | [diff] [blame] | 816 | forced_image=options.image, |
Gilad Arnold | 0c9c860 | 2012-10-02 23:58:58 -0700 | [diff] [blame] | 817 | payload_path=options.payload, |
Don Garrett | 0ad0937 | 2010-12-06 16:20:30 -0800 | [diff] [blame] | 818 | proxy_port=options.proxy_port, |
Chris Sosa | 4136e69 | 2010-10-28 23:42:37 -0700 | [diff] [blame] | 819 | src_image=options.src_image, |
Chris Sosa | e67b78f | 2010-11-04 17:33:16 -0700 | [diff] [blame] | 820 | vm=options.vm, |
Chris Sosa | 08d55a2 | 2011-01-19 16:08:02 -0800 | [diff] [blame] | 821 | board=options.board, |
Chris Sosa | 0f1ec84 | 2011-02-14 16:33:22 -0800 | [diff] [blame] | 822 | copy_to_static_root=not options.exit, |
| 823 | private_key=options.private_key, |
Satoru Takabayashi | d733cbe | 2011-11-15 09:36:32 -0800 | [diff] [blame] | 824 | critical_update=options.critical_update, |
Gilad Arnold | 0c9c860 | 2012-10-02 23:58:58 -0700 | [diff] [blame] | 825 | remote_payload=options.remote_payload, |
Gilad Arnold | a564b4b | 2012-10-04 10:32:44 -0700 | [diff] [blame] | 826 | max_updates=options.max_updates, |
Gilad Arnold | 8318eac | 2012-10-04 12:52:23 -0700 | [diff] [blame] | 827 | host_log=options.host_log, |
Chris Sosa | 0f1ec84 | 2011-02-14 16:33:22 -0800 | [diff] [blame] | 828 | ) |
Chris Sosa | 7c93136 | 2010-10-11 19:49:01 -0700 | [diff] [blame] | 829 | |
Chris Sosa | 6a3697f | 2013-01-29 16:44:43 -0800 | [diff] [blame] | 830 | if options.pregenerate_update: |
| 831 | updater.PreGenerateUpdate() |
Chris Sosa | 0356d3b | 2010-09-16 15:46:22 -0700 | [diff] [blame] | 832 | |
Don Garrett | 0c880e2 | 2010-11-17 18:13:37 -0800 | [diff] [blame] | 833 | # If the command line requested after setup, it's time to do it. |
| 834 | if not options.exit: |
Chris Sosa | 66e2d9c | 2012-07-11 14:14:14 -0700 | [diff] [blame] | 835 | # Handle options that must be set globally in cherrypy. |
Chris Sosa | 2f1c41e | 2012-07-10 14:32:33 -0700 | [diff] [blame] | 836 | if options.production: |
Chris Sosa | 66e2d9c | 2012-07-11 14:14:14 -0700 | [diff] [blame] | 837 | cherrypy.config.update({'environment': 'production'}) |
| 838 | if not options.logfile: |
| 839 | cherrypy.config.update({'log.screen': True}) |
| 840 | else: |
| 841 | cherrypy.config.update({'log.error_file': options.logfile, |
| 842 | 'log.access_file': options.logfile}) |
Chris Sosa | 2f1c41e | 2012-07-10 14:32:33 -0700 | [diff] [blame] | 843 | |
Don Garrett | 0c880e2 | 2010-11-17 18:13:37 -0800 | [diff] [blame] | 844 | cherrypy.quickstart(DevServerRoot(), config=_GetConfig(options)) |
Chris Sosa | cde6bf4 | 2012-05-31 18:36:39 -0700 | [diff] [blame] | 845 | |
| 846 | |
| 847 | if __name__ == '__main__': |
| 848 | main() |