blob: bdf73e2d708694410e42c63376494a5a172d98a6 [file] [log] [blame]
David Riley2fcb0122017-11-02 11:25:39 -07001#!/usr/bin/env python2
Luis Hector Chavezdca9dd72018-06-12 12:56:30 -07002# -*- coding: utf-8 -*-
Chris Sosa781ba6d2012-04-11 12:44:43 -07003# Copyright (c) 2009-2012 The Chromium OS Authors. All rights reserved.
rtc@google.comded22402009-10-26 22:36:21 +00004# Use of this source code is governed by a BSD-style license that can be
5# found in the LICENSE file.
6
Chris Sosa3ae4dc12013-03-29 11:47:00 -07007"""Chromium OS development server that can be used for all forms of update.
8
9This devserver can be used to perform system-wide autoupdate and update
10of specific portage packages on devices running Chromium OS derived operating
11systems. It mainly operates in two modes:
12
131) archive mode: In this mode, the devserver is configured to stage and
14serve artifacts from Google Storage using the credentials provided to it before
15it is run. The easiest way to understand this is that the devserver is
16functioning as a local cache for artifacts produced and uploaded by build
17servers. Users of this form of devserver can either download the artifacts
18from the devservers static directory OR use the update RPC to perform a
19system-wide autoupdate. Archive mode is always active.
20
212) artifact-generation mode: in this mode, the devserver will attempt to
22generate update payloads and build artifacts when requested. This mode only
23works in the Chromium OS chroot as it uses build tools only present in the
24chroot (emerge, cros_generate_update_payload, etc.). By default, when a device
25requests an update from this form of devserver, the devserver will attempt to
26discover if a more recent build of the board has been built by the developer
Mike Frysingeraa0cb102019-02-25 01:09:19 -050027and generate a payload that the requested system can autoupdate to.
Chris Sosa3ae4dc12013-03-29 11:47:00 -070028
29For autoupdates, there are many more advanced options that can help specify
30how to update and which payload to give to a requester.
31"""
32
Gabe Black3b567202015-09-23 14:07:59 -070033from __future__ import print_function
Chris Sosa7c931362010-10-11 19:49:01 -070034
Amin Hassani08e42d22019-06-03 00:31:30 -070035import httplib
Gilad Arnold55a2a372012-10-02 09:46:32 -070036import json
David Riley2fcb0122017-11-02 11:25:39 -070037import optparse # pylint: disable=deprecated-module
rtc@google.comded22402009-10-26 22:36:21 +000038import os
Scott Zawalski4647ce62012-01-03 17:17:28 -050039import re
Simran Basi4baad082013-02-14 13:39:18 -080040import shutil
xixuan52c2fba2016-05-20 17:02:48 -070041import signal
Mandeep Singh Baines38dcdda2012-12-07 17:55:33 -080042import socket
Chris Masone816e38c2012-05-02 12:22:36 -070043import subprocess
J. Richard Barnette3d977b82013-04-23 11:05:19 -070044import sys
Chris Masone816e38c2012-05-02 12:22:36 -070045import tempfile
Dan Shi59ae7092013-06-04 14:37:27 -070046import threading
Gilad Arnoldd5ebaaa2012-10-02 11:52:38 -070047import types
J. Richard Barnette3d977b82013-04-23 11:05:19 -070048from logging import handlers
49
50import cherrypy
David Riley2fcb0122017-11-02 11:25:39 -070051# pylint: disable=no-name-in-module
Chris Sosa855b8932013-08-21 13:24:55 -070052from cherrypy import _cplogging as cplogging
Congbin Guo3afae6c2019-08-13 16:29:42 -070053from cherrypy.process import plugins # pylint: disable=import-error
David Riley2fcb0122017-11-02 11:25:39 -070054# pylint: enable=no-name-in-module
rtc@google.comded22402009-10-26 22:36:21 +000055
Richard Barnettedf35c322017-08-18 17:02:13 -070056# This must happen before any local modules get a chance to import
57# anything from chromite. Otherwise, really bad things will happen, and
58# you will _not_ understand why.
Congbin Guo3afae6c2019-08-13 16:29:42 -070059import setup_chromite # pylint: disable=unused-import
Richard Barnettedf35c322017-08-18 17:02:13 -070060
Dan Shi2f136862016-02-11 15:38:38 -080061import artifact_info
Congbin Guo3afae6c2019-08-13 16:29:42 -070062import autoupdate
Chris Sosa75490802013-09-30 17:21:45 -070063import build_artifact
Gilad Arnold11fbef42014-02-10 11:04:13 -080064import cherrypy_ext
Gilad Arnoldc65330c2012-09-20 15:17:48 -070065import common_util
Simran Basief83d6a2014-08-28 14:32:01 -070066import devserver_constants
Congbin Guo4132a272019-08-20 12:32:14 -070067import devserver_exceptions
Chris Sosa47a7d4e2012-03-28 11:26:55 -070068import downloader
Congbin Guo3afae6c2019-08-13 16:29:42 -070069import health_checker
Gilad Arnoldc65330c2012-09-20 15:17:48 -070070import log_util
joychen3cb228e2013-06-12 12:13:13 -070071import xbuddy
Gilad Arnoldc65330c2012-09-20 15:17:48 -070072
Gilad Arnoldc65330c2012-09-20 15:17:48 -070073# Module-local log function.
Chris Sosa6a3697f2013-01-29 16:44:43 -080074def _Log(message, *args):
75 return log_util.LogWithTag('DEVSERVER', message, *args)
Chris Sosa0356d3b2010-09-16 15:46:22 -070076
Dan Shi94dcbe82015-06-08 20:51:13 -070077
xixuanac89ce82016-11-30 16:48:20 -080078# Use try-except to skip unneccesary import for simple use case, eg. running
79# devserver on host.
80try:
81 import cros_update
xixuanac89ce82016-11-30 16:48:20 -080082except ImportError as e:
83 _Log('cros_update cannot be imported: %r', e)
84 cros_update = None
xixuana4f4e712017-05-08 15:17:54 -070085
86try:
87 import cros_update_progress
88except ImportError as e:
89 _Log('cros_update_progress cannot be imported: %r', e)
xixuanac89ce82016-11-30 16:48:20 -080090 cros_update_progress = None
91
xixuanac89ce82016-11-30 16:48:20 -080092try:
Dan Shi72b16132015-10-08 12:10:33 -070093 import android_build
94except ImportError as e:
95 # Ignore android_build import failure. This is to support devserver running
96 # inside a ChromeOS device triggered by cros flash. Most ChromeOS test images
97 # do not have google-api-python-client module and they don't need to support
98 # Android updating, therefore, ignore the import failure here.
Dan Shi72b16132015-10-08 12:10:33 -070099 android_build = None
Frank Farzan40160872011-12-12 18:39:18 -0800100
Chris Sosa417e55d2011-01-25 16:40:48 -0800101CACHED_ENTRIES = 12
Don Garrettf90edf02010-11-16 17:36:14 -0800102
Simran Basi4baad082013-02-14 13:39:18 -0800103TELEMETRY_FOLDER = 'telemetry_src'
104TELEMETRY_DEPS = ['dep-telemetry_dep.tar.bz2',
105 'dep-page_cycler_dep.tar.bz2',
Simran Basi0d078682013-03-22 16:40:04 -0700106 'dep-chrome_test.tar.bz2',
107 'dep-perf_data_dep.tar.bz2']
Simran Basi4baad082013-02-14 13:39:18 -0800108
Chris Sosa0356d3b2010-09-16 15:46:22 -0700109# Sets up global to share between classes.
rtc@google.com21a5ca32009-11-04 18:23:23 +0000110updater = None
rtc@google.comded22402009-10-26 22:36:21 +0000111
xixuan3d48bff2017-01-30 19:00:09 -0800112# Log rotation parameters. These settings correspond to twice a day once
113# devserver is started, with about two weeks (28 backup files) of old logs
114# kept for backup.
J. Richard Barnette3d977b82013-04-23 11:05:19 -0700115#
xixuan3d48bff2017-01-30 19:00:09 -0800116# For more, see the documentation in standard python library for
J. Richard Barnette3d977b82013-04-23 11:05:19 -0700117# logging.handlers.TimedRotatingFileHandler
xixuan3d48bff2017-01-30 19:00:09 -0800118_LOG_ROTATION_TIME = 'H'
Congbin Guo3afae6c2019-08-13 16:29:42 -0700119_LOG_ROTATION_INTERVAL = 12 # hours
120_LOG_ROTATION_BACKUP = 28 # backup counts
Frank Farzan40160872011-12-12 18:39:18 -0800121
xixuan52c2fba2016-05-20 17:02:48 -0700122# Auto-update parameters
123
124# Error msg for missing key in CrOS auto-update.
Xixuan Wu32af9f12017-11-13 14:11:44 -0800125KEY_ERROR_MSG = 'Key Error in RPC: %s= is required'
xixuan52c2fba2016-05-20 17:02:48 -0700126
127# Command of running auto-update.
128AUTO_UPDATE_CMD = '/usr/bin/python -u %s -d %s -b %s --static_dir %s'
129
130
Gabe Black3b567202015-09-23 14:07:59 -0700131def _canonicalize_archive_url(archive_url):
132 """Canonicalizes archive_url strings.
133
134 Raises:
135 DevserverError: if archive_url is not set.
136 """
137 if archive_url:
138 if not archive_url.startswith('gs://'):
Congbin Guo4132a272019-08-20 12:32:14 -0700139 raise devserver_exceptions.DevServerError(
140 "Archive URL isn't from Google Storage (%s) ." % archive_url)
Gabe Black3b567202015-09-23 14:07:59 -0700141
142 return archive_url.rstrip('/')
143 else:
Congbin Guo4132a272019-08-20 12:32:14 -0700144 raise devserver_exceptions.DevServerError(
145 "Must specify an archive_url in the request")
Gabe Black3b567202015-09-23 14:07:59 -0700146
147
148def _canonicalize_local_path(local_path):
149 """Canonicalizes |local_path| strings.
150
151 Raises:
152 DevserverError: if |local_path| is not set.
153 """
154 # Restrict staging of local content to only files within the static
155 # directory.
156 local_path = os.path.abspath(local_path)
157 if not local_path.startswith(updater.static_dir):
Congbin Guo4132a272019-08-20 12:32:14 -0700158 raise devserver_exceptions.DevServerError(
159 'Local path %s must be a subdirectory of the static'
160 ' directory: %s' % (local_path, updater.static_dir))
Gabe Black3b567202015-09-23 14:07:59 -0700161
162 return local_path.rstrip('/')
163
164
165def _get_artifacts(kwargs):
166 """Returns a tuple of named and file artifacts given the stage rpc kwargs.
167
168 Raises:
169 DevserverError if no artifacts would be returned.
170 """
171 artifacts = kwargs.get('artifacts')
172 files = kwargs.get('files')
173 if not artifacts and not files:
Congbin Guo4132a272019-08-20 12:32:14 -0700174 raise devserver_exceptions.DevServerError('No artifacts specified.')
Gabe Black3b567202015-09-23 14:07:59 -0700175
176 # Note we NEED to coerce files to a string as we get raw unicode from
177 # cherrypy and we treat files as strings elsewhere in the code.
178 return (str(artifacts).split(',') if artifacts else [],
179 str(files).split(',') if files else [])
180
181
Dan Shi61305df2015-10-26 16:52:35 -0700182def _is_android_build_request(kwargs):
183 """Check if a devserver call is for Android build, based on the arguments.
184
185 This method exams the request's arguments (os_type) to determine if the
186 request is for Android build. If os_type is set to `android`, returns True.
187 If os_type is not set or has other values, returns False.
188
189 Args:
190 kwargs: Keyword arguments for the request.
191
192 Returns:
193 True if the request is for Android build. False otherwise.
194 """
195 os_type = kwargs.get('os_type', None)
196 return os_type == 'android'
197
198
Gabe Black3b567202015-09-23 14:07:59 -0700199def _get_downloader(kwargs):
200 """Returns the downloader based on passed in arguments.
201
202 Args:
Amin Hassani08e42d22019-06-03 00:31:30 -0700203 kwargs: Keyword arguments for the request.
Gabe Black3b567202015-09-23 14:07:59 -0700204 """
205 local_path = kwargs.get('local_path')
206 if local_path:
207 local_path = _canonicalize_local_path(local_path)
208
209 dl = None
210 if local_path:
Prathmesh Prabhu58d08932018-01-19 15:08:19 -0800211 delete_source = _parse_boolean_arg(kwargs, 'delete_source')
212 dl = downloader.LocalDownloader(updater.static_dir, local_path,
213 delete_source=delete_source)
Gabe Black3b567202015-09-23 14:07:59 -0700214
Dan Shi61305df2015-10-26 16:52:35 -0700215 if not _is_android_build_request(kwargs):
Gabe Black3b567202015-09-23 14:07:59 -0700216 archive_url = kwargs.get('archive_url')
217 if not archive_url and not local_path:
Congbin Guo4132a272019-08-20 12:32:14 -0700218 raise devserver_exceptions.DevServerError(
219 'Requires archive_url or local_path to be specified.')
Gabe Black3b567202015-09-23 14:07:59 -0700220 if archive_url and local_path:
Congbin Guo4132a272019-08-20 12:32:14 -0700221 raise devserver_exceptions.DevServerError(
222 'archive_url and local_path can not both be specified.')
Gabe Black3b567202015-09-23 14:07:59 -0700223 if not dl:
224 archive_url = _canonicalize_archive_url(archive_url)
Luis Hector Chavezdca9dd72018-06-12 12:56:30 -0700225 dl = downloader.GoogleStorageDownloader(
226 updater.static_dir, archive_url,
227 downloader.GoogleStorageDownloader.GetBuildIdFromArchiveURL(
228 archive_url))
Gabe Black3b567202015-09-23 14:07:59 -0700229 elif not dl:
230 target = kwargs.get('target', None)
Dan Shi72b16132015-10-08 12:10:33 -0700231 branch = kwargs.get('branch', None)
Dan Shi61305df2015-10-26 16:52:35 -0700232 build_id = kwargs.get('build_id', None)
233 if not target or not branch or not build_id:
Congbin Guo4132a272019-08-20 12:32:14 -0700234 raise devserver_exceptions.DevServerError(
Dan Shi61305df2015-10-26 16:52:35 -0700235 'target, branch, build ID must all be specified for downloading '
236 'Android build.')
Dan Shi72b16132015-10-08 12:10:33 -0700237 dl = downloader.AndroidBuildDownloader(updater.static_dir, branch, build_id,
238 target)
Gabe Black3b567202015-09-23 14:07:59 -0700239
240 return dl
241
242
243def _get_downloader_and_factory(kwargs):
244 """Returns the downloader and artifact factory based on passed in arguments.
245
246 Args:
Amin Hassani08e42d22019-06-03 00:31:30 -0700247 kwargs: Keyword arguments for the request.
Gabe Black3b567202015-09-23 14:07:59 -0700248 """
249 artifacts, files = _get_artifacts(kwargs)
250 dl = _get_downloader(kwargs)
251
252 if (isinstance(dl, downloader.GoogleStorageDownloader) or
253 isinstance(dl, downloader.LocalDownloader)):
254 factory_class = build_artifact.ChromeOSArtifactFactory
Dan Shi72b16132015-10-08 12:10:33 -0700255 elif isinstance(dl, downloader.AndroidBuildDownloader):
Gabe Black3b567202015-09-23 14:07:59 -0700256 factory_class = build_artifact.AndroidArtifactFactory
257 else:
Congbin Guo4132a272019-08-20 12:32:14 -0700258 raise devserver_exceptions.DevServerError(
259 'Unrecognized value for downloader type: %s' % type(dl))
Gabe Black3b567202015-09-23 14:07:59 -0700260
261 factory = factory_class(dl.GetBuildDir(), artifacts, files, dl.GetBuild())
262
263 return dl, factory
264
265
Scott Zawalski4647ce62012-01-03 17:17:28 -0500266def _LeadingWhiteSpaceCount(string):
267 """Count the amount of leading whitespace in a string.
268
269 Args:
270 string: The string to count leading whitespace in.
Don Garrettf84631a2014-01-07 18:21:26 -0800271
Scott Zawalski4647ce62012-01-03 17:17:28 -0500272 Returns:
273 number of white space chars before characters start.
274 """
Gabe Black3b567202015-09-23 14:07:59 -0700275 matched = re.match(r'^\s+', string)
Scott Zawalski4647ce62012-01-03 17:17:28 -0500276 if matched:
277 return len(matched.group())
278
279 return 0
280
281
282def _PrintDocStringAsHTML(func):
283 """Make a functions docstring somewhat HTML style.
284
285 Args:
286 func: The function to return the docstring from.
Don Garrettf84631a2014-01-07 18:21:26 -0800287
Scott Zawalski4647ce62012-01-03 17:17:28 -0500288 Returns:
289 A string that is somewhat formated for a web browser.
290 """
291 # TODO(scottz): Make this parse Args/Returns in a prettier way.
292 # Arguments could be bolded and indented etc.
293 html_doc = []
294 for line in func.__doc__.splitlines():
295 leading_space = _LeadingWhiteSpaceCount(line)
296 if leading_space > 0:
Chris Sosa47a7d4e2012-03-28 11:26:55 -0700297 line = ' ' * leading_space + line
Scott Zawalski4647ce62012-01-03 17:17:28 -0500298
299 html_doc.append('<BR>%s' % line)
300
301 return '\n'.join(html_doc)
302
303
Simran Basief83d6a2014-08-28 14:32:01 -0700304def _GetUpdateTimestampHandler(static_dir):
305 """Returns a handler to update directory staged.timestamp.
306
307 This handler resets the stage.timestamp whenever static content is accessed.
308
309 Args:
310 static_dir: Directory from which static content is being staged.
311
312 Returns:
Amin Hassani08e42d22019-06-03 00:31:30 -0700313 A cherrypy handler to update the timestamp of accessed content.
Simran Basief83d6a2014-08-28 14:32:01 -0700314 """
315 def UpdateTimestampHandler():
316 if not '404' in cherrypy.response.status:
317 build_match = re.match(devserver_constants.STAGED_BUILD_REGEX,
318 cherrypy.request.path_info)
319 if build_match:
320 build_dir = os.path.join(static_dir, build_match.group('build'))
321 downloader.Downloader.TouchTimestampForStaged(build_dir)
322 return UpdateTimestampHandler
323
324
Chris Sosa7c931362010-10-11 19:49:01 -0700325def _GetConfig(options):
326 """Returns the configuration for the devserver."""
Mandeep Singh Baines38dcdda2012-12-07 17:55:33 -0800327
Mandeep Singh Baines38dcdda2012-12-07 17:55:33 -0800328 socket_host = '::'
Yu-Ju Hongc8d4af32013-11-12 15:14:26 -0800329 # Fall back to IPv4 when python is not configured with IPv6.
330 if not socket.has_ipv6:
Mandeep Singh Baines38dcdda2012-12-07 17:55:33 -0800331 socket_host = '0.0.0.0'
332
Simran Basief83d6a2014-08-28 14:32:01 -0700333 # Adds the UpdateTimestampHandler to cherrypy's tools. This tools executes
334 # on the on_end_resource hook. This hook is called once processing is
335 # complete and the response is ready to be returned.
336 cherrypy.tools.update_timestamp = cherrypy.Tool(
337 'on_end_resource', _GetUpdateTimestampHandler(options.static_dir))
338
David Riley2fcb0122017-11-02 11:25:39 -0700339 base_config = {
340 'global': {
341 'server.log_request_headers': True,
342 'server.protocol_version': 'HTTP/1.1',
343 'server.socket_host': socket_host,
344 'server.socket_port': int(options.port),
345 'response.timeout': 6000,
346 'request.show_tracebacks': True,
347 'server.socket_timeout': 60,
348 'server.thread_pool': 2,
349 'engine.autoreload.on': False,
350 },
351 '/api': {
352 # Gets rid of cherrypy parsing post file for args.
353 'request.process_request_body': False,
354 },
355 '/build': {
356 'response.timeout': 100000,
357 },
358 '/update': {
359 # Gets rid of cherrypy parsing post file for args.
360 'request.process_request_body': False,
361 'response.timeout': 10000,
362 },
363 # Sets up the static dir for file hosting.
364 '/static': {
365 'tools.staticdir.dir': options.static_dir,
366 'tools.staticdir.on': True,
367 'response.timeout': 10000,
368 'tools.update_timestamp.on': True,
369 },
370 }
Chris Sosa5f118ef2012-07-12 11:37:50 -0700371 if options.production:
Alex Miller93beca52013-07-30 19:25:09 -0700372 base_config['global'].update({'server.thread_pool': 150})
Scott Zawalski1c5e7cd2012-02-27 13:12:52 -0500373
Chris Sosa7c931362010-10-11 19:49:01 -0700374 return base_config
rtc@google.com64244662009-11-12 00:52:08 +0000375
Darin Petkove17164a2010-08-11 13:24:41 -0700376
Gilad Arnoldd5ebaaa2012-10-02 11:52:38 -0700377def _GetRecursiveMemberObject(root, member_list):
378 """Returns an object corresponding to a nested member list.
379
380 Args:
381 root: the root object to search
382 member_list: list of nested members to search
Don Garrettf84631a2014-01-07 18:21:26 -0800383
Gilad Arnoldd5ebaaa2012-10-02 11:52:38 -0700384 Returns:
385 An object corresponding to the member name list; None otherwise.
386 """
387 for member in member_list:
388 next_root = root.__class__.__dict__.get(member)
389 if not next_root:
390 return None
391 root = next_root
392 return root
393
394
395def _IsExposed(name):
396 """Returns True iff |name| has an `exposed' attribute and it is set."""
397 return hasattr(name, 'exposed') and name.exposed
398
399
Gilad Arnold748c8322012-10-12 09:51:35 -0700400def _GetExposedMethod(root, nested_member, ignored=None):
Gilad Arnoldd5ebaaa2012-10-02 11:52:38 -0700401 """Returns a CherryPy-exposed method, if such exists.
402
403 Args:
404 root: the root object for searching
405 nested_member: a slash-joined path to the nested member
406 ignored: method paths to be ignored
Don Garrettf84631a2014-01-07 18:21:26 -0800407
Gilad Arnoldd5ebaaa2012-10-02 11:52:38 -0700408 Returns:
409 A function object corresponding to the path defined by |member_list| from
410 the |root| object, if the function is exposed and not ignored; None
411 otherwise.
412 """
Gilad Arnold748c8322012-10-12 09:51:35 -0700413 method = (not (ignored and nested_member in ignored) and
Gilad Arnoldd5ebaaa2012-10-02 11:52:38 -0700414 _GetRecursiveMemberObject(root, nested_member.split('/')))
Amin Hassani08e42d22019-06-03 00:31:30 -0700415 if method and isinstance(method, types.FunctionType) and _IsExposed(method):
Gilad Arnoldd5ebaaa2012-10-02 11:52:38 -0700416 return method
417
418
Gilad Arnold748c8322012-10-12 09:51:35 -0700419def _FindExposedMethods(root, prefix, unlisted=None):
Gilad Arnoldd5ebaaa2012-10-02 11:52:38 -0700420 """Finds exposed CherryPy methods.
421
422 Args:
423 root: the root object for searching
424 prefix: slash-joined chain of members leading to current object
425 unlisted: URLs to be excluded regardless of their exposed status
Don Garrettf84631a2014-01-07 18:21:26 -0800426
Gilad Arnoldd5ebaaa2012-10-02 11:52:38 -0700427 Returns:
428 List of exposed URLs that are not unlisted.
429 """
430 method_list = []
431 for member in sorted(root.__class__.__dict__.keys()):
432 prefixed_member = prefix + '/' + member if prefix else member
Gilad Arnold748c8322012-10-12 09:51:35 -0700433 if unlisted and prefixed_member in unlisted:
Gilad Arnoldd5ebaaa2012-10-02 11:52:38 -0700434 continue
435 member_obj = root.__class__.__dict__[member]
436 if _IsExposed(member_obj):
Amin Hassani08e42d22019-06-03 00:31:30 -0700437 if isinstance(member_obj, types.FunctionType):
Gilad Arnoldd5ebaaa2012-10-02 11:52:38 -0700438 method_list.append(prefixed_member)
439 else:
440 method_list += _FindExposedMethods(
441 member_obj, prefixed_member, unlisted)
442 return method_list
443
444
xixuan52c2fba2016-05-20 17:02:48 -0700445def _check_base_args_for_auto_update(kwargs):
xixuanac89ce82016-11-30 16:48:20 -0800446 """Check basic args required for auto-update.
447
448 Args:
449 kwargs: the parameters to be checked.
450
451 Raises:
452 DevServerHTTPError if required parameters don't exist in kwargs.
453 """
xixuan52c2fba2016-05-20 17:02:48 -0700454 if 'host_name' not in kwargs:
Amin Hassani08e42d22019-06-03 00:31:30 -0700455 raise common_util.DevServerHTTPError(httplib.INTERNAL_SERVER_ERROR,
456 KEY_ERROR_MSG % 'host_name')
xixuan52c2fba2016-05-20 17:02:48 -0700457
458 if 'build_name' not in kwargs:
Amin Hassani08e42d22019-06-03 00:31:30 -0700459 raise common_util.DevServerHTTPError(httplib.INTERNAL_SERVER_ERROR,
460 KEY_ERROR_MSG % 'build_name')
xixuan52c2fba2016-05-20 17:02:48 -0700461
462
463def _parse_boolean_arg(kwargs, key):
xixuanac89ce82016-11-30 16:48:20 -0800464 """Parse boolean arg from kwargs.
465
466 Args:
467 kwargs: the parameters to be checked.
468 key: the key to be parsed.
469
470 Returns:
471 The boolean value of kwargs[key], or False if key doesn't exist in kwargs.
472
473 Raises:
474 DevServerHTTPError if kwargs[key] is not a boolean variable.
475 """
xixuan52c2fba2016-05-20 17:02:48 -0700476 if key in kwargs:
477 if kwargs[key] == 'True':
478 return True
479 elif kwargs[key] == 'False':
480 return False
481 else:
482 raise common_util.DevServerHTTPError(
Amin Hassani08e42d22019-06-03 00:31:30 -0700483 httplib.INTERNAL_SERVER_ERROR,
xixuan52c2fba2016-05-20 17:02:48 -0700484 'The value for key %s is not boolean.' % key)
485 else:
486 return False
487
xixuan447ad9d2017-02-28 14:46:20 -0800488
xixuanac89ce82016-11-30 16:48:20 -0800489def _parse_string_arg(kwargs, key):
490 """Parse string arg from kwargs.
491
492 Args:
493 kwargs: the parameters to be checked.
494 key: the key to be parsed.
495
496 Returns:
497 The string value of kwargs[key], or None if key doesn't exist in kwargs.
498 """
499 if key in kwargs:
500 return kwargs[key]
501 else:
502 return None
503
xixuan447ad9d2017-02-28 14:46:20 -0800504
xixuanac89ce82016-11-30 16:48:20 -0800505def _build_uri_from_build_name(build_name):
506 """Get build url from a given build name.
507
508 Args:
509 build_name: the build name to be parsed, whose format is
510 'board/release_version'.
511
512 Returns:
513 The release_archive_url on Google Storage for this build name.
514 """
Amin Hassani08e42d22019-06-03 00:31:30 -0700515 # TODO(ahassani): This function doesn't seem to be used anywhere since its
516 # previous use of lib.paygen.gspath was broken and it doesn't seem to be
517 # causing any runtime issues. So deprecate this in the future.
518 tokens = build_name.split('/')
519 return 'gs://chromeos-releases/stable-channel/%s/%s' % (tokens[0], tokens[1])
xixuan52c2fba2016-05-20 17:02:48 -0700520
xixuan447ad9d2017-02-28 14:46:20 -0800521
522def _clear_process(host_name, pid):
523 """Clear AU process for given hostname and pid.
524
525 This clear includes:
526 1. kill process if it's alive.
527 2. delete the track status file of this process.
528 3. delete the executing log file of this process.
529
530 Args:
531 host_name: the host to execute auto-update.
532 pid: the background auto-update process id.
533 """
534 if cros_update_progress.IsProcessAlive(pid):
535 os.killpg(int(pid), signal.SIGKILL)
536
537 cros_update_progress.DelTrackStatusFile(host_name, pid)
538 cros_update_progress.DelExecuteLogFile(host_name, pid)
539
540
Dale Curtisc9aaf3a2011-08-09 15:47:40 -0700541class ApiRoot(object):
542 """RESTful API for Dev Server information."""
543 exposed = True
544
545 @cherrypy.expose
546 def hostinfo(self, ip):
547 """Returns a JSON dictionary containing information about the given ip.
548
Gilad Arnold1b908392012-10-05 11:36:27 -0700549 Args:
550 ip: address of host whose info is requested
Don Garrettf84631a2014-01-07 18:21:26 -0800551
Gilad Arnold1b908392012-10-05 11:36:27 -0700552 Returns:
553 A JSON dictionary containing all or some of the following fields:
554 last_event_type (int): last update event type received
555 last_event_status (int): last update event status received
556 last_known_version (string): last known version reported in update ping
557 forced_update_label (string): update label to force next update ping to
558 use, set by setnextupdate
559 See the OmahaEvent class in update_engine/omaha_request_action.h for
560 event type and status code definitions. If the ip does not exist an empty
561 string is returned.
Dale Curtisc9aaf3a2011-08-09 15:47:40 -0700562
Gilad Arnold1b908392012-10-05 11:36:27 -0700563 Example URL:
564 http://myhost/api/hostinfo?ip=192.168.1.5
565 """
Dale Curtisc9aaf3a2011-08-09 15:47:40 -0700566 return updater.HandleHostInfoPing(ip)
567
568 @cherrypy.expose
Gilad Arnold286a0062012-01-12 13:47:02 -0800569 def hostlog(self, ip):
Gilad Arnold1b908392012-10-05 11:36:27 -0700570 """Returns a JSON object containing a log of host event.
571
572 Args:
573 ip: address of host whose event log is requested, or `all'
Don Garrettf84631a2014-01-07 18:21:26 -0800574
Gilad Arnold1b908392012-10-05 11:36:27 -0700575 Returns:
576 A JSON encoded list (log) of dictionaries (events), each of which
577 containing a `timestamp' and other event fields, as described under
578 /api/hostinfo.
579
580 Example URL:
581 http://myhost/api/hostlog?ip=192.168.1.5
582 """
Gilad Arnold286a0062012-01-12 13:47:02 -0800583 return updater.HandleHostLogPing(ip)
584
585 @cherrypy.expose
Dale Curtisc9aaf3a2011-08-09 15:47:40 -0700586 def setnextupdate(self, ip):
587 """Allows the response to the next update ping from a host to be set.
588
589 Takes the IP of the host and an update label as normally provided to the
Gilad Arnold1b908392012-10-05 11:36:27 -0700590 /update command.
591 """
Dale Curtisc9aaf3a2011-08-09 15:47:40 -0700592 body_length = int(cherrypy.request.headers['Content-Length'])
593 label = cherrypy.request.rfile.read(body_length)
594
595 if label:
596 label = label.strip()
597 if label:
598 return updater.HandleSetUpdatePing(ip, label)
Amin Hassani08e42d22019-06-03 00:31:30 -0700599 raise common_util.DevServerHTTPError(httplib.BAD_REQUEST,
600 'No label provided.')
Dale Curtisc9aaf3a2011-08-09 15:47:40 -0700601
Gilad Arnold55a2a372012-10-02 09:46:32 -0700602 @cherrypy.expose
Don Garrettf84631a2014-01-07 18:21:26 -0800603 def fileinfo(self, *args):
Gilad Arnold55a2a372012-10-02 09:46:32 -0700604 """Returns information about a given staged file.
605
606 Args:
Don Garrettf84631a2014-01-07 18:21:26 -0800607 args: path to the file inside the server's static staging directory
608
Gilad Arnold55a2a372012-10-02 09:46:32 -0700609 Returns:
610 A JSON encoded dictionary with information about the said file, which may
611 contain the following keys/values:
Gilad Arnold1b908392012-10-05 11:36:27 -0700612 size (int): the file size in bytes
Gilad Arnold1b908392012-10-05 11:36:27 -0700613 sha256 (string): a base64 encoded SHA256 hash
614
615 Example URL:
616 http://myhost/api/fileinfo/some/path/to/file
Gilad Arnold55a2a372012-10-02 09:46:32 -0700617 """
Don Garrettf84631a2014-01-07 18:21:26 -0800618 file_path = os.path.join(updater.static_dir, *args)
Gilad Arnold55a2a372012-10-02 09:46:32 -0700619 if not os.path.exists(file_path):
Congbin Guo4132a272019-08-20 12:32:14 -0700620 raise devserver_exceptions.DevServerError(
621 'file not found: %s' % file_path)
Gilad Arnold55a2a372012-10-02 09:46:32 -0700622 try:
623 file_size = os.path.getsize(file_path)
Gilad Arnold55a2a372012-10-02 09:46:32 -0700624 file_sha256 = common_util.GetFileSha256(file_path)
625 except os.error, e:
Congbin Guo4132a272019-08-20 12:32:14 -0700626 raise devserver_exceptions.DevServerError(
627 'failed to get info for file %s: %s' % (file_path, e))
Gilad Arnolde74b3812013-04-22 11:27:38 -0700628
Gilad Arnolde74b3812013-04-22 11:27:38 -0700629 return json.dumps({
630 autoupdate.Autoupdate.SIZE_ATTR: file_size,
Gilad Arnolde74b3812013-04-22 11:27:38 -0700631 autoupdate.Autoupdate.SHA256_ATTR: file_sha256,
Gilad Arnolde74b3812013-04-22 11:27:38 -0700632 })
Gilad Arnold55a2a372012-10-02 09:46:32 -0700633
Chris Sosa76e44b92013-01-31 12:11:38 -0800634
David Rochberg7c79a812011-01-19 14:24:45 -0500635class DevServerRoot(object):
Chris Sosa7c931362010-10-11 19:49:01 -0700636 """The Root Class for the Dev Server.
637
638 CherryPy works as follows:
639 For each method in this class, cherrpy interprets root/path
640 as a call to an instance of DevServerRoot->method_name. For example,
641 a call to http://myhost/build will call build. CherryPy automatically
642 parses http args and places them as keyword arguments in each method.
643 For paths http://myhost/update/dir1/dir2, you can use *args so that
644 cherrypy uses the update method and puts the extra paths in args.
645 """
Gilad Arnoldf8f769f2012-09-24 08:43:01 -0700646 # Method names that should not be listed on the index page.
647 _UNLISTED_METHODS = ['index', 'doc']
648
Dale Curtisc9aaf3a2011-08-09 15:47:40 -0700649 api = ApiRoot()
Chris Sosa7c931362010-10-11 19:49:01 -0700650
Dan Shi59ae7092013-06-04 14:37:27 -0700651 # Number of threads that devserver is staging images.
652 _staging_thread_count = 0
653 # Lock used to lock increasing/decreasing count.
654 _staging_thread_count_lock = threading.Lock()
655
joychen3cb228e2013-06-12 12:13:13 -0700656 def __init__(self, _xbuddy):
Nick Sanders7dcaa2e2011-08-04 15:20:41 -0700657 self._builder = None
Simran Basi4baad082013-02-14 13:39:18 -0800658 self._telemetry_lock_dict = common_util.LockDict()
joychen3cb228e2013-06-12 12:13:13 -0700659 self._xbuddy = _xbuddy
David Rochberg7c79a812011-01-19 14:24:45 -0500660
Congbin Guo3afae6c2019-08-13 16:29:42 -0700661 @property
662 def staging_thread_count(self):
663 """Get the staging thread count."""
664 return self._staging_thread_count
Dan Shiafd0e492015-05-27 14:23:51 -0700665
Dale Curtisc9aaf3a2011-08-09 15:47:40 -0700666 @cherrypy.expose
David Rochberg7c79a812011-01-19 14:24:45 -0500667 def build(self, board, pkg, **kwargs):
Chris Sosa7c931362010-10-11 19:49:01 -0700668 """Builds the package specified."""
Nick Sanders7dcaa2e2011-08-04 15:20:41 -0700669 import builder
670 if self._builder is None:
671 self._builder = builder.Builder()
David Rochberg7c79a812011-01-19 14:24:45 -0500672 return self._builder.Build(board, pkg, kwargs)
Chris Sosa7c931362010-10-11 19:49:01 -0700673
Dale Curtisc9aaf3a2011-08-09 15:47:40 -0700674 @cherrypy.expose
Dan Shif8eb0d12013-08-01 17:52:06 -0700675 def is_staged(self, **kwargs):
676 """Check if artifacts have been downloaded.
677
Congbin Guo3afae6c2019-08-13 16:29:42 -0700678 Examples:
679 To check if autotest and test_suites are staged:
680 http://devserver_url:<port>/is_staged?archive_url=gs://your_url/path&
681 artifacts=autotest,test_suites
682
Amin Hassani08e42d22019-06-03 00:31:30 -0700683 Args:
Chris Sosa6b0c6172013-08-05 17:01:33 -0700684 async: True to return without waiting for download to complete.
685 artifacts: Comma separated list of named artifacts to download.
686 These are defined in artifact_info and have their implementation
687 in build_artifact.py.
688 files: Comma separated list of file artifacts to stage. These
689 will be available as is in the corresponding static directory with no
690 custom post-processing.
691
Congbin Guo3afae6c2019-08-13 16:29:42 -0700692 Returns:
693 True of all artifacts are staged.
Dan Shif8eb0d12013-08-01 17:52:06 -0700694 """
Gabe Black3b567202015-09-23 14:07:59 -0700695 dl, factory = _get_downloader_and_factory(kwargs)
Aviv Keshet57d18172016-06-18 20:39:09 -0700696 response = str(dl.IsStaged(factory))
697 _Log('Responding to is_staged %s request with %r', kwargs, response)
698 return response
Dan Shi59ae7092013-06-04 14:37:27 -0700699
Chris Sosa76e44b92013-01-31 12:11:38 -0800700 @cherrypy.expose
Prashanth Ba06d2d22014-03-07 15:35:19 -0800701 def list_image_dir(self, **kwargs):
702 """Take an archive url and list the contents in its staged directory.
703
Amin Hassani08e42d22019-06-03 00:31:30 -0700704 Examples:
Prashanth Ba06d2d22014-03-07 15:35:19 -0800705 To list the contents of where this devserver should have staged
706 gs://image-archive/<board>-release/<build> call:
707 http://devserver_url:<port>/list_image_dir?archive_url=<gs://..>
708
Congbin Guo3afae6c2019-08-13 16:29:42 -0700709 Args:
710 archive_url: Google Storage URL for the build.
711
Prashanth Ba06d2d22014-03-07 15:35:19 -0800712 Returns:
713 A string with information about the contents of the image directory.
714 """
Gabe Black3b567202015-09-23 14:07:59 -0700715 dl = _get_downloader(kwargs)
Prashanth Ba06d2d22014-03-07 15:35:19 -0800716 try:
Gabe Black3b567202015-09-23 14:07:59 -0700717 image_dir_contents = dl.ListBuildDir()
Prashanth Ba06d2d22014-03-07 15:35:19 -0800718 except build_artifact.ArtifactDownloadError as e:
719 return 'Cannot list the contents of staged artifacts. %s' % e
720 if not image_dir_contents:
Gabe Black3b567202015-09-23 14:07:59 -0700721 return '%s has not been staged on this devserver.' % dl.DescribeSource()
Prashanth Ba06d2d22014-03-07 15:35:19 -0800722 return image_dir_contents
723
724 @cherrypy.expose
Chris Sosa76e44b92013-01-31 12:11:38 -0800725 def stage(self, **kwargs):
Gabe Black3b567202015-09-23 14:07:59 -0700726 """Downloads and caches build artifacts.
Chris Sosa76e44b92013-01-31 12:11:38 -0800727
Gabe Black3b567202015-09-23 14:07:59 -0700728 Downloads and caches build artifacts, possibly from a Google Storage URL,
Dan Shi72b16132015-10-08 12:10:33 -0700729 or from Android's build server. Returns once these have been downloaded
Gabe Black3b567202015-09-23 14:07:59 -0700730 on the devserver. A call to this will attempt to cache non-specified
731 artifacts in the background for the given from the given URL following
732 the principle of spatial locality. Spatial locality of different
Chris Sosa76e44b92013-01-31 12:11:38 -0800733 artifacts is explicitly defined in the build_artifact module.
734
735 These artifacts will then be available from the static/ sub-directory of
736 the devserver.
737
Amin Hassani08e42d22019-06-03 00:31:30 -0700738 Examples:
Chris Sosa76e44b92013-01-31 12:11:38 -0800739 To download the autotest and test suites tarballs:
740 http://devserver_url:<port>/stage?archive_url=gs://your_url/path&
741 artifacts=autotest,test_suites
742 To download the full update payload:
743 http://devserver_url:<port>/stage?archive_url=gs://your_url/path&
744 artifacts=full_payload
Chris Sosa6b0c6172013-08-05 17:01:33 -0700745 To download just a file called blah.bin:
746 http://devserver_url:<port>/stage?archive_url=gs://your_url/path&
747 files=blah.bin
Chris Sosa76e44b92013-01-31 12:11:38 -0800748
749 For both these examples, one could find these artifacts at:
joychened64b222013-06-21 16:39:34 -0700750 http://devserver_url:<port>/static/<relative_path>*
Chris Sosa76e44b92013-01-31 12:11:38 -0800751
752 Note for this example, relative path is the archive_url stripped of its
753 basename i.e. path/ in the examples above. Specific example:
754
755 gs://chromeos-image-archive/x86-mario-release/R26-3920.0.0
756
757 Will get staged to:
758
joychened64b222013-06-21 16:39:34 -0700759 http://devserver_url:<port>/static/x86-mario-release/R26-3920.0.0
Congbin Guo3afae6c2019-08-13 16:29:42 -0700760
761 Args:
762 archive_url: Google Storage URL for the build.
763 local_path: Local path for the build.
764 delete_source: Only meaningful with local_path. bool to indicate if the
765 source files should be deleted. This is especially useful when staging
766 a file locally in resource constrained environments as it allows us to
767 move the relevant files locally instead of copying them.
768 async: True to return without waiting for download to complete.
769 artifacts: Comma separated list of named artifacts to download.
770 These are defined in artifact_info and have their implementation
771 in build_artifact.py.
772 files: Comma separated list of files to stage. These
773 will be available as is in the corresponding static directory with no
774 custom post-processing.
775 clean: True to remove any previously staged artifacts first.
Chris Sosa76e44b92013-01-31 12:11:38 -0800776 """
Gabe Black3b567202015-09-23 14:07:59 -0700777 dl, factory = _get_downloader_and_factory(kwargs)
778
Dan Shi59ae7092013-06-04 14:37:27 -0700779 with DevServerRoot._staging_thread_count_lock:
780 DevServerRoot._staging_thread_count += 1
781 try:
Laurence Goodbyf5c958d2016-01-14 18:23:56 -0800782 boolean_string = kwargs.get('clean')
783 clean = xbuddy.XBuddy.ParseBoolean(boolean_string)
784 if clean and os.path.exists(dl.GetBuildDir()):
785 _Log('Removing %s' % dl.GetBuildDir())
786 shutil.rmtree(dl.GetBuildDir())
Gabe Black3b567202015-09-23 14:07:59 -0700787 async = kwargs.get('async', False)
788 dl.Download(factory, async=async)
Dan Shi59ae7092013-06-04 14:37:27 -0700789 finally:
790 with DevServerRoot._staging_thread_count_lock:
791 DevServerRoot._staging_thread_count -= 1
Chris Sosa76e44b92013-01-31 12:11:38 -0800792 return 'Success'
Chris Sosacde6bf42012-05-31 18:36:39 -0700793
794 @cherrypy.expose
xixuan52c2fba2016-05-20 17:02:48 -0700795 def cros_au(self, **kwargs):
796 """Auto-update a CrOS DUT.
797
798 Args:
799 kwargs:
800 host_name: the hostname of the DUT to auto-update.
801 build_name: the build name for update the DUT.
802 force_update: Force an update even if the version installed is the
803 same. Default: False.
804 full_update: If True, do not run stateful update, directly force a full
805 reimage. If False, try stateful update first if the dut is already
806 installed with the same version.
807 async: Whether the auto_update function is ran in the background.
David Rileyee75de22017-11-02 10:48:15 -0700808 quick_provision: Whether the quick provision path is attempted first.
xixuan52c2fba2016-05-20 17:02:48 -0700809
810 Returns:
811 A tuple includes two elements:
812 a boolean variable represents whether the auto-update process is
813 successfully started.
814 an integer represents the background auto-update process id.
815 """
816 _check_base_args_for_auto_update(kwargs)
817
818 host_name = kwargs['host_name']
819 build_name = kwargs['build_name']
820 force_update = _parse_boolean_arg(kwargs, 'force_update')
821 full_update = _parse_boolean_arg(kwargs, 'full_update')
822 async = _parse_boolean_arg(kwargs, 'async')
xixuanac89ce82016-11-30 16:48:20 -0800823 original_build = _parse_string_arg(kwargs, 'original_build')
David Haddock90e49442017-04-07 19:14:09 -0700824 payload_filename = _parse_string_arg(kwargs, 'payload_filename')
David Haddock20559612017-06-28 22:15:08 -0700825 clobber_stateful = _parse_boolean_arg(kwargs, 'clobber_stateful')
David Rileyee75de22017-11-02 10:48:15 -0700826 quick_provision = _parse_boolean_arg(kwargs, 'quick_provision')
827
828 devserver_url = updater.GetDevserverUrl()
829 static_url = updater.GetStaticUrl()
xixuan52c2fba2016-05-20 17:02:48 -0700830
831 if async:
832 path = os.path.dirname(os.path.abspath(__file__))
833 execute_file = os.path.join(path, 'cros_update.py')
834 args = (AUTO_UPDATE_CMD % (execute_file, host_name, build_name,
835 updater.static_dir))
xixuanac89ce82016-11-30 16:48:20 -0800836
837 # The original_build's format is like: link/3428.210.0
838 # The corresponding release_archive_url's format is like:
839 # gs://chromeos-releases/stable-channel/link/3428.210.0
840 if original_build:
841 release_archive_url = _build_uri_from_build_name(original_build)
842 # First staging the stateful.tgz synchronousely.
843 self.stage(files='stateful.tgz', async=False,
844 archive_url=release_archive_url)
845 args = ('%s --original_build %s' % (args, original_build))
846
xixuan52c2fba2016-05-20 17:02:48 -0700847 if force_update:
848 args = ('%s --force_update' % args)
849
850 if full_update:
851 args = ('%s --full_update' % args)
852
David Haddock90e49442017-04-07 19:14:09 -0700853 if payload_filename:
854 args = ('%s --payload_filename %s' % (args, payload_filename))
855
David Haddock20559612017-06-28 22:15:08 -0700856 if clobber_stateful:
857 args = ('%s --clobber_stateful' % args)
858
David Rileyee75de22017-11-02 10:48:15 -0700859 if quick_provision:
860 args = ('%s --quick_provision' % args)
861
862 if devserver_url:
863 args = ('%s --devserver_url %s' % (args, devserver_url))
864
865 if static_url:
866 args = ('%s --static_url %s' % (args, static_url))
867
xixuan2a0970a2016-08-10 12:12:44 -0700868 p = subprocess.Popen([args], shell=True, preexec_fn=os.setsid)
869 pid = os.getpgid(p.pid)
xixuan52c2fba2016-05-20 17:02:48 -0700870
871 # Pre-write status in the track_status_file before the first call of
872 # 'get_au_status' to make sure that the track_status_file exists.
xixuan2a0970a2016-08-10 12:12:44 -0700873 progress_tracker = cros_update_progress.AUProgress(host_name, pid)
xixuan52c2fba2016-05-20 17:02:48 -0700874 progress_tracker.WriteStatus('CrOS update is just started.')
875
xixuan2a0970a2016-08-10 12:12:44 -0700876 return json.dumps((True, pid))
xixuan52c2fba2016-05-20 17:02:48 -0700877 else:
878 cros_update_trigger = cros_update.CrOSUpdateTrigger(
xixuanac89ce82016-11-30 16:48:20 -0800879 host_name, build_name, updater.static_dir, force_update=force_update,
David Rileyee75de22017-11-02 10:48:15 -0700880 full_update=full_update, original_build=original_build,
881 quick_provision=quick_provision, devserver_url=devserver_url,
882 static_url=static_url)
xixuan52c2fba2016-05-20 17:02:48 -0700883 cros_update_trigger.TriggerAU()
xixuan27d50442017-08-09 10:38:25 -0700884 return json.dumps((True, -1))
xixuan52c2fba2016-05-20 17:02:48 -0700885
886 @cherrypy.expose
887 def get_au_status(self, **kwargs):
888 """Check if the auto-update task is finished.
889
890 It handles 4 cases:
891 1. If an error exists in the track_status_file, delete the track file and
892 raise it.
893 2. If cros-update process is finished, delete the file and return the
894 success result.
895 3. If the process is not running, delete the track file and raise an error
896 about 'the process is terminated due to unknown reason'.
897 4. If the track_status_file does not exist, kill the process if it exists,
898 and raise the IOError.
899
900 Args:
901 kwargs:
902 host_name: the hostname of the DUT to auto-update.
903 pid: the background process id of cros-update.
904
905 Returns:
xixuan28d99072016-10-06 12:24:16 -0700906 A dict with three elements:
xixuan52c2fba2016-05-20 17:02:48 -0700907 a boolean variable represents whether the auto-update process is
908 finished.
909 a string represents the current auto-update process status.
910 For example, 'Transfer Devserver/Stateful Update Package'.
xixuan28d99072016-10-06 12:24:16 -0700911 a detailed error message paragraph if there exists an Auto-Update
912 error, in which the last line shows the main exception. Empty
913 string otherwise.
xixuan52c2fba2016-05-20 17:02:48 -0700914 """
915 if 'host_name' not in kwargs:
Amin Hassani08e42d22019-06-03 00:31:30 -0700916 raise common_util.DevServerHTTPError(httplib.INTERNAL_SERVER_ERROR,
917 KEY_ERROR_MSG % 'host_name')
xixuan52c2fba2016-05-20 17:02:48 -0700918
919 if 'pid' not in kwargs:
Amin Hassani08e42d22019-06-03 00:31:30 -0700920 raise common_util.DevServerHTTPError(httplib.INTERNAL_SERVER_ERROR,
921 KEY_ERROR_MSG % 'pid')
xixuan52c2fba2016-05-20 17:02:48 -0700922
923 host_name = kwargs['host_name']
924 pid = kwargs['pid']
925 progress_tracker = cros_update_progress.AUProgress(host_name, pid)
926
xixuan28d99072016-10-06 12:24:16 -0700927 result_dict = {'finished': False, 'status': '', 'detailed_error_msg': ''}
xixuan52c2fba2016-05-20 17:02:48 -0700928 try:
929 result = progress_tracker.ReadStatus()
930 if result.startswith(cros_update_progress.ERROR_TAG):
xixuan28d99072016-10-06 12:24:16 -0700931 result_dict['detailed_error_msg'] = result[len(
932 cros_update_progress.ERROR_TAG):]
xixuan28681fd2016-11-23 11:13:56 -0800933 elif result == cros_update_progress.FINISHED:
xixuan28d99072016-10-06 12:24:16 -0700934 result_dict['finished'] = True
935 result_dict['status'] = result
xixuan28681fd2016-11-23 11:13:56 -0800936 elif not cros_update_progress.IsProcessAlive(pid):
xixuan28d99072016-10-06 12:24:16 -0700937 result_dict['detailed_error_msg'] = (
938 'Cros_update process terminated midway due to unknown reason. '
939 'Last update status was %s' % result)
xixuan28681fd2016-11-23 11:13:56 -0800940 else:
941 result_dict['status'] = result
942 except IOError as e:
943 if pid and cros_update_progress.IsProcessAlive(pid):
xixuan2a0970a2016-08-10 12:12:44 -0700944 os.killpg(int(pid), signal.SIGKILL)
xixuan52c2fba2016-05-20 17:02:48 -0700945
xixuan28681fd2016-11-23 11:13:56 -0800946 result_dict['detailed_error_msg'] = str(e)
947
948 return json.dumps(result_dict)
xixuan52c2fba2016-05-20 17:02:48 -0700949
950 @cherrypy.expose
David Riley6d5fca02017-10-31 10:35:47 -0700951 def post_au_status(self, status, **kwargs):
952 """Updates the status of an auto-update task.
953
954 Callers will need to POST to this URL with a body of MIME-type
955 "multipart/form-data".
956 The body should include a single argument, 'status', containing the
957 AU status to record.
958
959 Args:
960 status: The updated status.
961 kwargs:
962 host_name: the hostname of the DUT to auto-update.
963 pid: the background process id of cros-update.
964 """
965 if 'host_name' not in kwargs:
Amin Hassani08e42d22019-06-03 00:31:30 -0700966 raise common_util.DevServerHTTPError(httplib.INTERNAL_SERVER_ERROR,
967 KEY_ERROR_MSG % 'host_name')
David Riley6d5fca02017-10-31 10:35:47 -0700968
969 if 'pid' not in kwargs:
Amin Hassani08e42d22019-06-03 00:31:30 -0700970 raise common_util.DevServerHTTPError(httplib.INTERNAL_SERVER_ERROR,
971 KEY_ERROR_MSG % 'pid')
David Riley6d5fca02017-10-31 10:35:47 -0700972
973 host_name = kwargs['host_name']
974 pid = kwargs['pid']
David Riley3cea2582017-11-24 22:03:01 -0800975 status = status.rstrip()
976 _Log('Recording status for %s (%s): %s' % (host_name, pid, status))
David Riley6d5fca02017-10-31 10:35:47 -0700977 progress_tracker = cros_update_progress.AUProgress(host_name, pid)
978
David Riley3cea2582017-11-24 22:03:01 -0800979 progress_tracker.WriteStatus(status)
David Riley6d5fca02017-10-31 10:35:47 -0700980
981 return 'True'
982
983 @cherrypy.expose
xixuan52c2fba2016-05-20 17:02:48 -0700984 def handler_cleanup(self, **kwargs):
xixuan3bc974e2016-10-18 17:21:43 -0700985 """Clean track status log and temp directory for CrOS auto-update process.
xixuan52c2fba2016-05-20 17:02:48 -0700986
987 Args:
988 kwargs:
989 host_name: the hostname of the DUT to auto-update.
990 pid: the background process id of cros-update.
991 """
992 if 'host_name' not in kwargs:
Amin Hassani08e42d22019-06-03 00:31:30 -0700993 raise common_util.DevServerHTTPError(httplib.INTERNAL_SERVER_ERROR,
994 KEY_ERROR_MSG % 'host_name')
xixuan52c2fba2016-05-20 17:02:48 -0700995
996 if 'pid' not in kwargs:
Amin Hassani08e42d22019-06-03 00:31:30 -0700997 raise common_util.DevServerHTTPError(httplib.INTERNAL_SERVER_ERROR,
998 KEY_ERROR_MSG % 'pid')
xixuan52c2fba2016-05-20 17:02:48 -0700999
1000 host_name = kwargs['host_name']
1001 pid = kwargs['pid']
1002 cros_update_progress.DelTrackStatusFile(host_name, pid)
xixuan3bc974e2016-10-18 17:21:43 -07001003 cros_update_progress.DelAUTempDirectory(host_name, pid)
xixuan52c2fba2016-05-20 17:02:48 -07001004
1005 @cherrypy.expose
1006 def kill_au_proc(self, **kwargs):
1007 """Kill CrOS auto-update process using given process id.
1008
1009 Args:
1010 kwargs:
1011 host_name: Kill all the CrOS auto-update process of this host.
1012
1013 Returns:
1014 True if all processes are killed properly.
1015 """
1016 if 'host_name' not in kwargs:
Amin Hassani08e42d22019-06-03 00:31:30 -07001017 raise common_util.DevServerHTTPError(httplib.INTERNAL_SERVER_ERROR,
1018 KEY_ERROR_MSG % 'host_name')
xixuan52c2fba2016-05-20 17:02:48 -07001019
xixuan447ad9d2017-02-28 14:46:20 -08001020 cur_pid = kwargs.get('pid')
1021
xixuan52c2fba2016-05-20 17:02:48 -07001022 host_name = kwargs['host_name']
xixuan3bc974e2016-10-18 17:21:43 -07001023 track_log_list = cros_update_progress.GetAllTrackStatusFileByHostName(
1024 host_name)
xixuan52c2fba2016-05-20 17:02:48 -07001025 for log in track_log_list:
1026 # The track log's full path is: path/host_name_pid.log
1027 # Use splitext to remove file extension, then parse pid from the
1028 # filename.
Congbin Guo3afae6c2019-08-13 16:29:42 -07001029 pid = os.path.splitext(os.path.basename(log))[0][len(host_name) + 1:]
xixuan447ad9d2017-02-28 14:46:20 -08001030 _clear_process(host_name, pid)
xixuan52c2fba2016-05-20 17:02:48 -07001031
xixuan447ad9d2017-02-28 14:46:20 -08001032 if cur_pid:
1033 _clear_process(host_name, cur_pid)
xixuan52c2fba2016-05-20 17:02:48 -07001034
1035 return 'True'
1036
1037 @cherrypy.expose
1038 def collect_cros_au_log(self, **kwargs):
1039 """Collect CrOS auto-update log.
1040
1041 Args:
1042 kwargs:
1043 host_name: the hostname of the DUT to auto-update.
1044 pid: the background process id of cros-update.
1045
1046 Returns:
David Haddock9f459632017-05-11 14:45:46 -07001047 A dictionary containing the execute log file and any hostlog files.
xixuan52c2fba2016-05-20 17:02:48 -07001048 """
1049 if 'host_name' not in kwargs:
Amin Hassani08e42d22019-06-03 00:31:30 -07001050 raise common_util.DevServerHTTPError(httplib.INTERNAL_SERVER_ERROR,
1051 KEY_ERROR_MSG % 'host_name')
xixuan52c2fba2016-05-20 17:02:48 -07001052
1053 if 'pid' not in kwargs:
Amin Hassani08e42d22019-06-03 00:31:30 -07001054 raise common_util.DevServerHTTPError(httplib.INTERNAL_SERVER_ERROR,
1055 KEY_ERROR_MSG % 'pid')
xixuan52c2fba2016-05-20 17:02:48 -07001056
1057 host_name = kwargs['host_name']
1058 pid = kwargs['pid']
xixuan3bc974e2016-10-18 17:21:43 -07001059
1060 # Fetch the execute log recorded by cros_update_progress.
xixuan1bbfaba2016-10-13 17:53:22 -07001061 au_log = cros_update_progress.ReadExecuteLogFile(host_name, pid)
1062 cros_update_progress.DelExecuteLogFile(host_name, pid)
David Haddock9f459632017-05-11 14:45:46 -07001063 # Fetch the cros_au host_logs if they exist
1064 au_hostlogs = cros_update_progress.ReadAUHostLogFiles(host_name, pid)
1065 return json.dumps({'cros_au_log': au_log, 'host_logs': au_hostlogs})
xixuan1bbfaba2016-10-13 17:53:22 -07001066
xixuan52c2fba2016-05-20 17:02:48 -07001067 @cherrypy.expose
Dan Shi2f136862016-02-11 15:38:38 -08001068 def locate_file(self, **kwargs):
1069 """Get the path to the given file name.
1070
1071 This method looks up the given file name inside specified build artifacts.
1072 One use case is to help caller to locate an apk file inside a build
1073 artifact. The location of the apk file could be different based on the
1074 branch and target.
1075
1076 Args:
1077 file_name: Name of the file to look for.
1078 artifacts: A list of artifact names to search for the file.
1079
1080 Returns:
1081 Path to the file with the given name. It's relative to the folder for the
1082 build, e.g., DATA/priv-app/sl4a/sl4a.apk
Dan Shi2f136862016-02-11 15:38:38 -08001083 """
1084 dl, _ = _get_downloader_and_factory(kwargs)
1085 try:
Joe Brennan1691f8e2017-03-15 15:53:36 -07001086 file_name = kwargs['file_name']
Dan Shi2f136862016-02-11 15:38:38 -08001087 artifacts = kwargs['artifacts']
1088 except KeyError:
Congbin Guo4132a272019-08-20 12:32:14 -07001089 raise devserver_exceptions.DevServerError(
1090 '`file_name` and `artifacts` are required to search '
1091 'for a file in build artifacts.')
Dan Shi2f136862016-02-11 15:38:38 -08001092 build_path = dl.GetBuildDir()
1093 for artifact in artifacts:
1094 # Get the unzipped folder of the artifact. If it's not defined in
1095 # ARTIFACT_UNZIP_FOLDER_MAP, assume the files are unzipped to the build
1096 # directory directly.
1097 folder = artifact_info.ARTIFACT_UNZIP_FOLDER_MAP.get(artifact, '')
1098 artifact_path = os.path.join(build_path, folder)
1099 for root, _, filenames in os.walk(artifact_path):
Joe Brennan1691f8e2017-03-15 15:53:36 -07001100 if file_name in set([f for f in filenames]):
Dan Shi2f136862016-02-11 15:38:38 -08001101 return os.path.relpath(os.path.join(root, file_name), build_path)
Congbin Guo4132a272019-08-20 12:32:14 -07001102 raise devserver_exceptions.DevServerError(
1103 'File `%s` can not be found in artifacts: %s' % (file_name, artifacts))
Dan Shi2f136862016-02-11 15:38:38 -08001104
1105 @cherrypy.expose
Simran Basi4baad082013-02-14 13:39:18 -08001106 def setup_telemetry(self, **kwargs):
1107 """Extracts and sets up telemetry
1108
1109 This method goes through the telemetry deps packages, and stages them on
1110 the devserver to be used by the drones and the telemetry tests.
1111
1112 Args:
1113 archive_url: Google Storage URL for the build.
1114
1115 Returns:
1116 Path to the source folder for the telemetry codebase once it is staged.
1117 """
Gabe Black3b567202015-09-23 14:07:59 -07001118 dl = _get_downloader(kwargs)
Simran Basi4baad082013-02-14 13:39:18 -08001119
Gabe Black3b567202015-09-23 14:07:59 -07001120 build_path = dl.GetBuildDir()
Simran Basi4baad082013-02-14 13:39:18 -08001121 deps_path = os.path.join(build_path, 'autotest/packages')
1122 telemetry_path = os.path.join(build_path, TELEMETRY_FOLDER)
1123 src_folder = os.path.join(telemetry_path, 'src')
1124
1125 with self._telemetry_lock_dict.lock(telemetry_path):
1126 if os.path.exists(src_folder):
1127 # Telemetry is already fully stage return
1128 return src_folder
1129
1130 common_util.MkDirP(telemetry_path)
1131
1132 # Copy over the required deps tar balls to the telemetry directory.
1133 for dep in TELEMETRY_DEPS:
1134 dep_path = os.path.join(deps_path, dep)
Simran Basi0d078682013-03-22 16:40:04 -07001135 if not os.path.exists(dep_path):
1136 # This dep does not exist (could be new), do not extract it.
1137 continue
Simran Basi4baad082013-02-14 13:39:18 -08001138 try:
1139 common_util.ExtractTarball(dep_path, telemetry_path)
1140 except common_util.CommonUtilError as e:
1141 shutil.rmtree(telemetry_path)
Congbin Guo4132a272019-08-20 12:32:14 -07001142 raise devserver_exceptions.DevServerError(str(e))
Simran Basi4baad082013-02-14 13:39:18 -08001143
1144 # By default all the tarballs extract to test_src but some parts of
1145 # the telemetry code specifically hardcoded to exist inside of 'src'.
1146 test_src = os.path.join(telemetry_path, 'test_src')
1147 try:
1148 shutil.move(test_src, src_folder)
1149 except shutil.Error:
1150 # This can occur if src_folder already exists. Remove and retry move.
1151 shutil.rmtree(src_folder)
Congbin Guo4132a272019-08-20 12:32:14 -07001152 raise devserver_exceptions.DevServerError(
Gabe Black3b567202015-09-23 14:07:59 -07001153 'Failure in telemetry setup for build %s. Appears that the '
1154 'test_src to src move failed.' % dl.GetBuild())
Simran Basi4baad082013-02-14 13:39:18 -08001155
1156 return src_folder
1157
1158 @cherrypy.expose
Chris Sosa76e44b92013-01-31 12:11:38 -08001159 def symbolicate_dump(self, minidump, **kwargs):
Chris Masone816e38c2012-05-02 12:22:36 -07001160 """Symbolicates a minidump using pre-downloaded symbols, returns it.
1161
1162 Callers will need to POST to this URL with a body of MIME-type
1163 "multipart/form-data".
1164 The body should include a single argument, 'minidump', containing the
1165 binary-formatted minidump to symbolicate.
1166
Chris Masone816e38c2012-05-02 12:22:36 -07001167 Args:
Chris Sosa76e44b92013-01-31 12:11:38 -08001168 archive_url: Google Storage URL for the build.
Chris Masone816e38c2012-05-02 12:22:36 -07001169 minidump: The binary minidump file to symbolicate.
1170 """
Chris Sosa76e44b92013-01-31 12:11:38 -08001171 # Ensure the symbols have been staged.
Dan Shif08fe492016-10-04 14:39:25 -07001172 # Try debug.tar.xz first, then debug.tgz
1173 for artifact in (artifact_info.SYMBOLS_ONLY, artifact_info.SYMBOLS):
1174 kwargs['artifacts'] = artifact
1175 dl = _get_downloader(kwargs)
1176
1177 try:
1178 if self.stage(**kwargs) == 'Success':
1179 break
1180 except build_artifact.ArtifactDownloadError:
1181 continue
1182 else:
Congbin Guo4132a272019-08-20 12:32:14 -07001183 raise devserver_exceptions.DevServerError(
1184 'Failed to stage symbols for %s' % dl.DescribeSource())
Chris Sosa76e44b92013-01-31 12:11:38 -08001185
Chris Masone816e38c2012-05-02 12:22:36 -07001186 to_return = ''
1187 with tempfile.NamedTemporaryFile() as local:
1188 while True:
1189 data = minidump.file.read(8192)
1190 if not data:
1191 break
1192 local.write(data)
Chris Sosa76e44b92013-01-31 12:11:38 -08001193
Chris Masone816e38c2012-05-02 12:22:36 -07001194 local.flush()
Chris Sosa76e44b92013-01-31 12:11:38 -08001195
Gabe Black3b567202015-09-23 14:07:59 -07001196 symbols_directory = os.path.join(dl.GetBuildDir(), 'debug', 'breakpad')
Chris Sosa76e44b92013-01-31 12:11:38 -08001197
xixuanab744382017-04-27 10:41:27 -07001198 # The location of minidump_stackwalk is defined in chromeos-admin.
Chris Sosa76e44b92013-01-31 12:11:38 -08001199 stackwalk = subprocess.Popen(
xixuanab744382017-04-27 10:41:27 -07001200 ['/usr/local/bin/minidump_stackwalk', local.name, symbols_directory],
Chris Sosa76e44b92013-01-31 12:11:38 -08001201 stdout=subprocess.PIPE, stderr=subprocess.PIPE)
1202
Chris Masone816e38c2012-05-02 12:22:36 -07001203 to_return, error_text = stackwalk.communicate()
1204 if stackwalk.returncode != 0:
Congbin Guo4132a272019-08-20 12:32:14 -07001205 raise devserver_exceptions.DevServerError(
1206 "Can't generate stack trace: %s (rc=%d)" % (error_text,
1207 stackwalk.returncode))
Chris Masone816e38c2012-05-02 12:22:36 -07001208
1209 return to_return
1210
1211 @cherrypy.expose
Don Garrettf84631a2014-01-07 18:21:26 -08001212 def latestbuild(self, **kwargs):
Scott Zawalski16954532012-03-20 15:31:36 -04001213 """Return a string representing the latest build for a given target.
1214
1215 Args:
1216 target: The build target, typically a combination of the board and the
1217 type of build e.g. x86-mario-release.
1218 milestone: The milestone to filter builds on. E.g. R16. Optional, if not
1219 provided the latest RXX build will be returned.
Don Garrettf84631a2014-01-07 18:21:26 -08001220
Scott Zawalski16954532012-03-20 15:31:36 -04001221 Returns:
1222 A string representation of the latest build if one exists, i.e.
1223 R19-1993.0.0-a1-b1480.
1224 An empty string if no latest could be found.
1225 """
Don Garrettf84631a2014-01-07 18:21:26 -08001226 if not kwargs:
Scott Zawalski16954532012-03-20 15:31:36 -04001227 return _PrintDocStringAsHTML(self.latestbuild)
1228
Don Garrettf84631a2014-01-07 18:21:26 -08001229 if 'target' not in kwargs:
Amin Hassani08e42d22019-06-03 00:31:30 -07001230 raise common_util.DevServerHTTPError(httplib.INTERNAL_SERVER_ERROR,
1231 'Error: target= is required!')
Dan Shi61305df2015-10-26 16:52:35 -07001232
1233 if _is_android_build_request(kwargs):
1234 branch = kwargs.get('branch', None)
1235 target = kwargs.get('target', None)
1236 if not target or not branch:
Congbin Guo4132a272019-08-20 12:32:14 -07001237 raise devserver_exceptions.DevServerError(
xixuan52c2fba2016-05-20 17:02:48 -07001238 'Both target and branch must be specified to query for the latest '
1239 'Android build.')
Dan Shi61305df2015-10-26 16:52:35 -07001240 return android_build.BuildAccessor.GetLatestBuildID(target, branch)
1241
Scott Zawalski16954532012-03-20 15:31:36 -04001242 try:
Gilad Arnoldc65330c2012-09-20 15:17:48 -07001243 return common_util.GetLatestBuildVersion(
Don Garrettf84631a2014-01-07 18:21:26 -08001244 updater.static_dir, kwargs['target'],
1245 milestone=kwargs.get('milestone'))
Gilad Arnold17fe03d2012-10-02 10:05:01 -07001246 except common_util.CommonUtilError as errmsg:
Amin Hassani08e42d22019-06-03 00:31:30 -07001247 raise common_util.DevServerHTTPError(httplib.INTERNAL_SERVER_ERROR,
1248 str(errmsg))
Scott Zawalski16954532012-03-20 15:31:36 -04001249
1250 @cherrypy.expose
xixuan7efd0002016-04-14 15:34:01 -07001251 def list_suite_controls(self, **kwargs):
1252 """Return a list of contents of all known control files.
1253
1254 Example URL:
1255 To List all control files' content:
1256 http://dev-server/list_suite_controls?suite_name=bvt&
1257 build=daisy_spring-release/R29-4279.0.0
1258
1259 Args:
1260 build: The build i.e. x86-alex-release/R18-1514.0.0-a1-b1450.
1261 suite_name: List the control files belonging to that suite.
1262
1263 Returns:
Dan Shia1cd6522016-04-18 16:07:21 -07001264 A dictionary of all control files's path to its content for given suite.
xixuan7efd0002016-04-14 15:34:01 -07001265 """
1266 if not kwargs:
1267 return _PrintDocStringAsHTML(self.controlfiles)
1268
1269 if 'build' not in kwargs:
Amin Hassani08e42d22019-06-03 00:31:30 -07001270 raise common_util.DevServerHTTPError(httplib.INTERNAL_SERVER_ERROR,
1271 'Error: build= is required!')
xixuan7efd0002016-04-14 15:34:01 -07001272
1273 if 'suite_name' not in kwargs:
Amin Hassani08e42d22019-06-03 00:31:30 -07001274 raise common_util.DevServerHTTPError(httplib.INTERNAL_SERVER_ERROR,
Dan Shia1cd6522016-04-18 16:07:21 -07001275 'Error: suite_name= is required!')
xixuan7efd0002016-04-14 15:34:01 -07001276
1277 control_file_list = [
1278 line.rstrip() for line in common_util.GetControlFileListForSuite(
1279 updater.static_dir, kwargs['build'],
1280 kwargs['suite_name']).splitlines()]
1281
Dan Shia1cd6522016-04-18 16:07:21 -07001282 control_file_content_dict = {}
xixuan7efd0002016-04-14 15:34:01 -07001283 for control_path in control_file_list:
Dan Shia1cd6522016-04-18 16:07:21 -07001284 control_file_content_dict[control_path] = (common_util.GetControlFile(
xixuan7efd0002016-04-14 15:34:01 -07001285 updater.static_dir, kwargs['build'], control_path))
1286
Dan Shia1cd6522016-04-18 16:07:21 -07001287 return json.dumps(control_file_content_dict)
xixuan7efd0002016-04-14 15:34:01 -07001288
1289 @cherrypy.expose
Don Garrettf84631a2014-01-07 18:21:26 -08001290 def controlfiles(self, **kwargs):
Scott Zawalski4647ce62012-01-03 17:17:28 -05001291 """Return a control file or a list of all known control files.
1292
1293 Example URL:
1294 To List all control files:
beepsbd337242013-07-09 22:44:06 -07001295 http://dev-server/controlfiles?suite_name=&build=daisy_spring-release/R29-4279.0.0
1296 To List all control files for, say, the bvt suite:
1297 http://dev-server/controlfiles?suite_name=bvt&build=daisy_spring-release/R29-4279.0.0
Scott Zawalski4647ce62012-01-03 17:17:28 -05001298 To return the contents of a path:
Scott Zawalski84a39c92012-01-13 15:12:42 -05001299 http://dev-server/controlfiles?board=x86-alex-release&build=R18-1514.0.0&control_path=client/sleeptest/control
Scott Zawalski4647ce62012-01-03 17:17:28 -05001300
1301 Args:
Scott Zawalski84a39c92012-01-13 15:12:42 -05001302 build: The build i.e. x86-alex-release/R18-1514.0.0-a1-b1450.
Scott Zawalski4647ce62012-01-03 17:17:28 -05001303 control_path: If you want the contents of a control file set this
1304 to the path. E.g. client/site_tests/sleeptest/control
1305 Optional, if not provided return a list of control files is returned.
beepsbd337242013-07-09 22:44:06 -07001306 suite_name: If control_path is not specified but a suite_name is
1307 specified, list the control files belonging to that suite instead of
1308 all control files. The empty string for suite_name will list all control
1309 files for the build.
Don Garrettf84631a2014-01-07 18:21:26 -08001310
Scott Zawalski4647ce62012-01-03 17:17:28 -05001311 Returns:
1312 Contents of a control file if control_path is provided.
1313 A list of control files if no control_path is provided.
1314 """
Don Garrettf84631a2014-01-07 18:21:26 -08001315 if not kwargs:
Scott Zawalski4647ce62012-01-03 17:17:28 -05001316 return _PrintDocStringAsHTML(self.controlfiles)
1317
Don Garrettf84631a2014-01-07 18:21:26 -08001318 if 'build' not in kwargs:
Amin Hassani08e42d22019-06-03 00:31:30 -07001319 raise common_util.DevServerHTTPError(httplib.INTERNAL_SERVER_ERROR,
1320 'Error: build= is required!')
Scott Zawalski4647ce62012-01-03 17:17:28 -05001321
Don Garrettf84631a2014-01-07 18:21:26 -08001322 if 'control_path' not in kwargs:
1323 if 'suite_name' in kwargs and kwargs['suite_name']:
beepsbd337242013-07-09 22:44:06 -07001324 return common_util.GetControlFileListForSuite(
Don Garrettf84631a2014-01-07 18:21:26 -08001325 updater.static_dir, kwargs['build'], kwargs['suite_name'])
beepsbd337242013-07-09 22:44:06 -07001326 else:
1327 return common_util.GetControlFileList(
Don Garrettf84631a2014-01-07 18:21:26 -08001328 updater.static_dir, kwargs['build'])
Scott Zawalski4647ce62012-01-03 17:17:28 -05001329 else:
Gilad Arnoldc65330c2012-09-20 15:17:48 -07001330 return common_util.GetControlFile(
Don Garrettf84631a2014-01-07 18:21:26 -08001331 updater.static_dir, kwargs['build'], kwargs['control_path'])
Frank Farzan40160872011-12-12 18:39:18 -08001332
1333 @cherrypy.expose
Simran Basi99e63c02014-05-20 10:39:52 -07001334 def xbuddy_translate(self, *args, **kwargs):
Yu-Ju Hong1bdb7a92014-04-10 16:02:11 -07001335 """Translates an xBuddy path to a real path to artifact if it exists.
1336
1337 Args:
Simran Basi99e63c02014-05-20 10:39:52 -07001338 args: An xbuddy path in the form of {local|remote}/build_id/artifact.
1339 Local searches the devserver's static directory. Remote searches a
1340 Google Storage image archive.
1341
1342 Kwargs:
1343 image_dir: Google Storage image archive to search in if requesting a
1344 remote artifact. If none uses the default bucket.
Yu-Ju Hong1bdb7a92014-04-10 16:02:11 -07001345
1346 Returns:
Simran Basi99e63c02014-05-20 10:39:52 -07001347 String in the format of build_id/artifact as stored on the local server
1348 or in Google Storage.
Yu-Ju Hong1bdb7a92014-04-10 16:02:11 -07001349 """
Simran Basi99e63c02014-05-20 10:39:52 -07001350 build_id, filename = self._xbuddy.Translate(
Gabe Black3b567202015-09-23 14:07:59 -07001351 args, image_dir=kwargs.get('image_dir'))
Yu-Ju Hong1bdb7a92014-04-10 16:02:11 -07001352 response = os.path.join(build_id, filename)
1353 _Log('Path translation requested, returning: %s', response)
1354 return response
1355
1356 @cherrypy.expose
joycheneaf4cfc2013-07-02 08:38:57 -07001357 def xbuddy(self, *args, **kwargs):
1358 """The full xBuddy call, returns resource specified by path_parts.
joychen3cb228e2013-06-12 12:13:13 -07001359
1360 Args:
joycheneaf4cfc2013-07-02 08:38:57 -07001361 path_parts: the path following xbuddy/ in the call url is split into the
joychen121fc9b2013-08-02 14:30:30 -07001362 components of the path. The path can be understood as
1363 "{local|remote}/build_id/artifact" where build_id is composed of
1364 "board/version."
joycheneaf4cfc2013-07-02 08:38:57 -07001365
joychen121fc9b2013-08-02 14:30:30 -07001366 The first path element is optional, and can be "remote" or "local"
1367 If local (the default), devserver will not attempt to access Google
1368 Storage, and will only search the static directory for the files.
1369 If remote, devserver will try to obtain the artifact off GS if it's
1370 not found locally.
1371 The board is the familiar board name, optionally suffixed.
1372 The version can be the google storage version number, and may also be
1373 any of a number of xBuddy defined version aliases that will be
1374 translated into the latest built image that fits the description.
1375 Defaults to latest.
1376 The artifact is one of a number of image or artifact aliases used by
1377 xbuddy, defined in xbuddy:ALIASES. Defaults to test.
joycheneaf4cfc2013-07-02 08:38:57 -07001378
1379 Kwargs:
Yu-Ju Hong51495eb2013-12-12 17:08:43 -08001380 for_update: {true|false}
1381 if true, pregenerates the update payloads for the image,
1382 and returns the update uri to pass to the
1383 update_engine_client.
joychen3cb228e2013-06-12 12:13:13 -07001384 return_dir: {true|false}
1385 if set to true, returns the url to the update.gz
Yu-Ju Hong51495eb2013-12-12 17:08:43 -08001386 relative_path: {true|false}
1387 if set to true, returns the relative path to the payload
1388 directory from static_dir.
joychen3cb228e2013-06-12 12:13:13 -07001389 Example URL:
joycheneaf4cfc2013-07-02 08:38:57 -07001390 http://host:port/xbuddy/x86-generic/R26-4000.0.0/test
joychen3cb228e2013-06-12 12:13:13 -07001391 or
joycheneaf4cfc2013-07-02 08:38:57 -07001392 http://host:port/xbuddy/x86-generic/R26-4000.0.0/test?return_dir=true
joychen3cb228e2013-06-12 12:13:13 -07001393
1394 Returns:
Yu-Ju Hong51495eb2013-12-12 17:08:43 -08001395 If |for_update|, returns a redirect to the image or update file
1396 on the devserver. E.g.,
1397 http://host:port/static/archive/x86-generic-release/R26-4000.0.0/
1398 chromium-test-image.bin
1399 If |return_dir|, return a uri to the folder where the artifact is. E.g.,
1400 http://host:port/static/x86-generic-release/R26-4000.0.0/
1401 If |relative_path| is true, return a relative path the folder where the
1402 payloads are. E.g.,
1403 archive/x86-generic-release/R26-4000.0.0
joychen3cb228e2013-06-12 12:13:13 -07001404 """
Chris Sosa75490802013-09-30 17:21:45 -07001405 boolean_string = kwargs.get('for_update')
1406 for_update = xbuddy.XBuddy.ParseBoolean(boolean_string)
Yu-Ju Hong51495eb2013-12-12 17:08:43 -08001407 boolean_string = kwargs.get('return_dir')
1408 return_dir = xbuddy.XBuddy.ParseBoolean(boolean_string)
1409 boolean_string = kwargs.get('relative_path')
1410 relative_path = xbuddy.XBuddy.ParseBoolean(boolean_string)
joychen121fc9b2013-08-02 14:30:30 -07001411
Yu-Ju Hong51495eb2013-12-12 17:08:43 -08001412 if return_dir and relative_path:
Chris Sosa4b951602014-04-09 20:26:07 -07001413 raise common_util.DevServerHTTPError(
Amin Hassani08e42d22019-06-03 00:31:30 -07001414 httplib.INTERNAL_SERVER_ERROR,
1415 'Cannot specify both return_dir and relative_path')
Chris Sosa75490802013-09-30 17:21:45 -07001416
1417 # For updates, we optimize downloading of test images.
1418 file_name = None
1419 build_id = None
1420 if for_update:
1421 try:
Yu-Ju Hong1bdb7a92014-04-10 16:02:11 -07001422 build_id = self._xbuddy.StageTestArtifactsForUpdate(args)
Chris Sosa75490802013-09-30 17:21:45 -07001423 except build_artifact.ArtifactDownloadError:
1424 build_id = None
1425
1426 if not build_id:
1427 build_id, file_name = self._xbuddy.Get(args)
1428
Yu-Ju Hong51495eb2013-12-12 17:08:43 -08001429 if for_update:
1430 _Log('Payload generation triggered by request')
1431 # Forces payload to be in cache and symlinked into build_id dir.
Chris Sosa75490802013-09-30 17:21:45 -07001432 updater.GetUpdateForLabel(autoupdate.FORCED_UPDATE, build_id,
1433 image_name=file_name)
Yu-Ju Hong51495eb2013-12-12 17:08:43 -08001434
1435 response = None
1436 if return_dir:
1437 response = os.path.join(cherrypy.request.base, 'static', build_id)
1438 _Log('Directory requested, returning: %s', response)
1439 elif relative_path:
1440 response = build_id
1441 _Log('Relative path requested, returning: %s', response)
1442 elif for_update:
1443 response = os.path.join(cherrypy.request.base, 'update', build_id)
1444 _Log('Update URI requested, returning: %s', response)
joychen3cb228e2013-06-12 12:13:13 -07001445 else:
Yu-Ju Hong51495eb2013-12-12 17:08:43 -08001446 # Redirect to download the payload if no kwargs are set.
joychen121fc9b2013-08-02 14:30:30 -07001447 build_id = '/' + os.path.join('static', build_id, file_name)
Yu-Ju Hong51495eb2013-12-12 17:08:43 -08001448 _Log('Payload requested, returning: %s', build_id)
joychen121fc9b2013-08-02 14:30:30 -07001449 raise cherrypy.HTTPRedirect(build_id, 302)
joychen3cb228e2013-06-12 12:13:13 -07001450
Yu-Ju Hong51495eb2013-12-12 17:08:43 -08001451 return response
1452
joychen3cb228e2013-06-12 12:13:13 -07001453 @cherrypy.expose
1454 def xbuddy_list(self):
1455 """Lists the currently available images & time since last access.
1456
Gilad Arnold452fd272014-02-04 11:09:28 -08001457 Returns:
1458 A string representation of a list of tuples [(build_id, time since last
1459 access),...]
joychen3cb228e2013-06-12 12:13:13 -07001460 """
1461 return self._xbuddy.List()
1462
1463 @cherrypy.expose
1464 def xbuddy_capacity(self):
Gilad Arnold452fd272014-02-04 11:09:28 -08001465 """Returns the number of images cached by xBuddy."""
joychen3cb228e2013-06-12 12:13:13 -07001466 return self._xbuddy.Capacity()
1467
1468 @cherrypy.expose
Chris Sosa7c931362010-10-11 19:49:01 -07001469 def index(self):
Gilad Arnoldf8f769f2012-09-24 08:43:01 -07001470 """Presents a welcome message and documentation links."""
Gilad Arnoldf8f769f2012-09-24 08:43:01 -07001471 return ('Welcome to the Dev Server!<br>\n'
1472 '<br>\n'
1473 'Here are the available methods, click for documentation:<br>\n'
1474 '<br>\n'
1475 '%s' %
1476 '<br>\n'.join(
1477 [('<a href=doc/%s>%s</a>' % (name, name))
Gilad Arnoldd5ebaaa2012-10-02 11:52:38 -07001478 for name in _FindExposedMethods(
1479 self, '', unlisted=self._UNLISTED_METHODS)]))
Gilad Arnoldf8f769f2012-09-24 08:43:01 -07001480
1481 @cherrypy.expose
1482 def doc(self, *args):
1483 """Shows the documentation for available methods / URLs.
1484
Amin Hassani08e42d22019-06-03 00:31:30 -07001485 Examples:
Gilad Arnoldf8f769f2012-09-24 08:43:01 -07001486 http://myhost/doc/update
1487 """
Gilad Arnoldd5ebaaa2012-10-02 11:52:38 -07001488 name = '/'.join(args)
1489 method = _GetExposedMethod(self, name)
Gilad Arnoldf8f769f2012-09-24 08:43:01 -07001490 if not method:
Congbin Guo4132a272019-08-20 12:32:14 -07001491 raise devserver_exceptions.DevServerError(
1492 "No exposed method named `%s'" % name)
Gilad Arnoldf8f769f2012-09-24 08:43:01 -07001493 if not method.__doc__:
Congbin Guo4132a272019-08-20 12:32:14 -07001494 raise devserver_exceptions.DevServerError(
1495 "No documentation for exposed method `%s'" % name)
Gilad Arnoldf8f769f2012-09-24 08:43:01 -07001496 return '<pre>\n%s</pre>' % method.__doc__
Chris Sosa7c931362010-10-11 19:49:01 -07001497
Dale Curtisc9aaf3a2011-08-09 15:47:40 -07001498 @cherrypy.expose
Chris Sosa7c931362010-10-11 19:49:01 -07001499 def update(self, *args):
Gilad Arnoldf8f769f2012-09-24 08:43:01 -07001500 """Handles an update check from a Chrome OS client.
1501
1502 The HTTP request should contain the standard Omaha-style XML blob. The URL
1503 line may contain an additional intermediate path to the update payload.
1504
joychen121fc9b2013-08-02 14:30:30 -07001505 This request can be handled in one of 4 ways, depending on the devsever
1506 settings and intermediate path.
joychenb0dfe552013-07-30 10:02:06 -07001507
joychen121fc9b2013-08-02 14:30:30 -07001508 1. No intermediate path
1509 If no intermediate path is given, the default behavior is to generate an
1510 update payload from the latest test image locally built for the board
1511 specified in the xml. Devserver serves the generated payload.
1512
1513 2. Path explicitly invokes XBuddy
1514 If there is a path given, it can explicitly invoke xbuddy by prefixing it
1515 with 'xbuddy'. This path is then used to acquire an image binary for the
1516 devserver to generate an update payload from. Devserver then serves this
1517 payload.
1518
1519 3. Path is left for the devserver to interpret.
1520 If the path given doesn't explicitly invoke xbuddy, devserver will attempt
1521 to generate a payload from the test image in that directory and serve it.
1522
1523 4. The devserver is in a 'forced' mode. TO BE DEPRECATED
1524 This comes from the usage of --forced_payload or --image when starting the
1525 devserver. No matter what path (or no path) gets passed in, devserver will
1526 serve the update payload (--forced_payload) or generate an update payload
1527 from the image (--image).
1528
1529 Examples:
1530 1. No intermediate path
1531 update_engine_client --omaha_url=http://myhost/update
1532 This generates an update payload from the latest test image locally built
1533 for the board specified in the xml.
1534
1535 2. Explicitly invoke xbuddy
1536 update_engine_client --omaha_url=
1537 http://myhost/update/xbuddy/remote/board/version/dev
1538 This would go to GS to download the dev image for the board, from which
1539 the devserver would generate a payload to serve.
1540
1541 3. Give a path for devserver to interpret
1542 update_engine_client --omaha_url=http://myhost/update/some/random/path
1543 This would attempt, in order to:
1544 a) Generate an update from a test image binary if found in
1545 static_dir/some/random/path.
1546 b) Serve an update payload found in static_dir/some/random/path.
1547 c) Hope that some/random/path takes the form "board/version" and
1548 and attempt to download an update payload for that board/version
1549 from GS.
Gilad Arnoldf8f769f2012-09-24 08:43:01 -07001550 """
joychen121fc9b2013-08-02 14:30:30 -07001551 label = '/'.join(args)
Gilad Arnold286a0062012-01-12 13:47:02 -08001552 body_length = int(cherrypy.request.headers.get('Content-Length', 0))
Chris Sosa7c931362010-10-11 19:49:01 -07001553 data = cherrypy.request.rfile.read(body_length)
Chris Sosa7c931362010-10-11 19:49:01 -07001554
joychen121fc9b2013-08-02 14:30:30 -07001555 return updater.HandleUpdatePing(data, label)
Chris Sosa0356d3b2010-09-16 15:46:22 -07001556
Dan Shif5ce2de2013-04-25 16:06:32 -07001557
Chris Sosadbc20082012-12-10 13:39:11 -08001558def _CleanCache(cache_dir, wipe):
1559 """Wipes any excess cached items in the cache_dir.
1560
1561 Args:
1562 cache_dir: the directory we are wiping from.
1563 wipe: If True, wipe all the contents -- not just the excess.
1564 """
1565 if wipe:
1566 # Clear the cache and exit on error.
1567 cmd = 'rm -rf %s/*' % cache_dir
1568 if os.system(cmd) != 0:
1569 _Log('Failed to clear the cache with %s' % cmd)
1570 sys.exit(1)
1571 else:
1572 # Clear all but the last N cached updates
1573 cmd = ('cd %s; ls -tr | head --lines=-%d | xargs rm -rf' %
1574 (cache_dir, CACHED_ENTRIES))
1575 if os.system(cmd) != 0:
1576 _Log('Failed to clean up old delta cache files with %s' % cmd)
1577 sys.exit(1)
1578
1579
Chris Sosa3ae4dc12013-03-29 11:47:00 -07001580def _AddTestingOptions(parser):
1581 group = optparse.OptionGroup(
1582 parser, 'Advanced Testing Options', 'These are used by test scripts and '
1583 'developers writing integration tests utilizing the devserver. They are '
1584 'not intended to be really used outside the scope of someone '
1585 'knowledgable about the test.')
1586 group.add_option('--exit',
1587 action='store_true',
1588 help='do not start the server (yet pregenerate/clear cache)')
1589 group.add_option('--host_log',
1590 action='store_true', default=False,
1591 help='record history of host update events (/api/hostlog)')
1592 group.add_option('--max_updates',
Gabe Black3b567202015-09-23 14:07:59 -07001593 metavar='NUM', default=-1, type='int',
Chris Sosa3ae4dc12013-03-29 11:47:00 -07001594 help='maximum number of update checks handled positively '
1595 '(default: unlimited)')
David Zeuthen52ccd012013-10-31 12:58:26 -07001596 group.add_option('--public_key',
1597 metavar='PATH', default=None,
1598 help='path to the public key in pem format. If this is set '
1599 'the devserver will transmit a base64 encoded version of '
1600 'the content in the Omaha-style XML response.')
Chris Sosa3ae4dc12013-03-29 11:47:00 -07001601 group.add_option('--proxy_port',
1602 metavar='PORT', default=None, type='int',
1603 help='port to have the client connect to -- basically the '
1604 'devserver lies to the update to tell it to get the payload '
1605 'from a different port that will proxy the request back to '
1606 'the devserver. The proxy must be managed outside the '
1607 'devserver.')
Chris Sosa3ae4dc12013-03-29 11:47:00 -07001608 parser.add_option_group(group)
1609
1610
1611def _AddUpdateOptions(parser):
1612 group = optparse.OptionGroup(
1613 parser, 'Autoupdate Options', 'These options can be used to change '
1614 'how the devserver either generates or serve update payloads. Please '
1615 'note that all of these option affect how a payload is generated and so '
1616 'do not work in archive-only mode.')
1617 group.add_option('--board',
1618 help='By default the devserver will create an update '
1619 'payload from the latest image built for the board '
1620 'a device that is requesting an update has. When we '
1621 'pre-generate an update (see below) and we do not specify '
1622 'another update_type option like image or payload, the '
1623 'devserver needs to know the board to generate the latest '
1624 'image for. This is that board.')
1625 group.add_option('--critical_update',
1626 action='store_true', default=False,
1627 help='Present update payload as critical')
Chris Sosa3ae4dc12013-03-29 11:47:00 -07001628 group.add_option('--image',
1629 metavar='FILE',
1630 help='Generate and serve an update using this image to any '
1631 'device that requests an update.')
Chris Sosa3ae4dc12013-03-29 11:47:00 -07001632 group.add_option('--payload',
1633 metavar='PATH',
1634 help='use the update payload from specified directory '
1635 '(update.gz).')
1636 group.add_option('-p', '--pregenerate_update',
1637 action='store_true', default=False,
1638 help='pre-generate the update payload before accepting '
1639 'update requests. Useful to help debug payload generation '
1640 'issues quickly. Also if an update payload will take a '
1641 'long time to generate, a client may timeout if you do not'
1642 'pregenerate the update.')
1643 group.add_option('--src_image',
1644 metavar='PATH', default='',
1645 help='If specified, delta updates will be generated using '
1646 'this image as the source image. Delta updates are when '
1647 'you are updating from a "source image" to a another '
1648 'image.')
1649 parser.add_option_group(group)
1650
1651
1652def _AddProductionOptions(parser):
1653 group = optparse.OptionGroup(
1654 parser, 'Advanced Server Options', 'These options can be used to changed '
1655 'for advanced server behavior.')
Chris Sosa3ae4dc12013-03-29 11:47:00 -07001656 group.add_option('--clear_cache',
1657 action='store_true', default=False,
1658 help='At startup, removes all cached entries from the'
1659 'devserver\'s cache.')
1660 group.add_option('--logfile',
1661 metavar='PATH',
1662 help='log output to this file instead of stdout')
Chris Sosa855b8932013-08-21 13:24:55 -07001663 group.add_option('--pidfile',
1664 metavar='PATH',
1665 help='path to output a pid file for the server.')
Gilad Arnold11fbef42014-02-10 11:04:13 -08001666 group.add_option('--portfile',
1667 metavar='PATH',
1668 help='path to output the port number being served on.')
Chris Sosa3ae4dc12013-03-29 11:47:00 -07001669 group.add_option('--production',
1670 action='store_true', default=False,
1671 help='have the devserver use production values when '
1672 'starting up. This includes using more threads and '
1673 'performing less logging.')
1674 parser.add_option_group(group)
1675
1676
Paul Hobbsef4e0702016-06-27 17:01:42 -07001677def MakeLogHandler(logfile):
J. Richard Barnette3d977b82013-04-23 11:05:19 -07001678 """Create a LogHandler instance used to log all messages."""
1679 hdlr_cls = handlers.TimedRotatingFileHandler
1680 hdlr = hdlr_cls(logfile, when=_LOG_ROTATION_TIME,
xixuan3d48bff2017-01-30 19:00:09 -08001681 interval=_LOG_ROTATION_INTERVAL,
J. Richard Barnette3d977b82013-04-23 11:05:19 -07001682 backupCount=_LOG_ROTATION_BACKUP)
Chris Sosa855b8932013-08-21 13:24:55 -07001683 hdlr.setFormatter(cplogging.logfmt)
J. Richard Barnette3d977b82013-04-23 11:05:19 -07001684 return hdlr
1685
1686
Chris Sosacde6bf42012-05-31 18:36:39 -07001687def main():
Chris Sosa3ae4dc12013-03-29 11:47:00 -07001688 usage = '\n\n'.join(['usage: %prog [options]', __doc__])
Gilad Arnold286a0062012-01-12 13:47:02 -08001689 parser = optparse.OptionParser(usage=usage)
joychened64b222013-06-21 16:39:34 -07001690
1691 # get directory that the devserver is run from
1692 devserver_dir = os.path.dirname(os.path.abspath(sys.argv[0]))
joychen84d13772013-08-06 09:17:23 -07001693 default_static_dir = '%s/static' % devserver_dir
joychened64b222013-06-21 16:39:34 -07001694 parser.add_option('--static_dir',
Gilad Arnold9714d9b2012-10-04 10:09:42 -07001695 metavar='PATH',
joychen84d13772013-08-06 09:17:23 -07001696 default=default_static_dir,
joychened64b222013-06-21 16:39:34 -07001697 help='writable static directory')
Gilad Arnold9714d9b2012-10-04 10:09:42 -07001698 parser.add_option('--port',
1699 default=8080, type='int',
Gilad Arnoldaf696d12014-02-14 13:13:28 -08001700 help=('port for the dev server to use; if zero, binds to '
1701 'an arbitrary available port (default: 8080)'))
Gilad Arnold9714d9b2012-10-04 10:09:42 -07001702 parser.add_option('-t', '--test_image',
1703 action='store_true',
joychen121fc9b2013-08-02 14:30:30 -07001704 help='Deprecated.')
joychen5260b9a2013-07-16 14:48:01 -07001705 parser.add_option('-x', '--xbuddy_manage_builds',
1706 action='store_true',
1707 default=False,
1708 help='If set, allow xbuddy to manage images in'
1709 'build/images.')
Dan Shi72b16132015-10-08 12:10:33 -07001710 parser.add_option('-a', '--android_build_credential',
1711 default=None,
1712 help='Path to a json file which contains the credential '
1713 'needed to access Android builds.')
Chris Sosa3ae4dc12013-03-29 11:47:00 -07001714 _AddProductionOptions(parser)
1715 _AddUpdateOptions(parser)
1716 _AddTestingOptions(parser)
Chris Sosa7c931362010-10-11 19:49:01 -07001717 (options, _) = parser.parse_args()
rtc@google.com21a5ca32009-11-04 18:23:23 +00001718
J. Richard Barnette3d977b82013-04-23 11:05:19 -07001719 # Handle options that must be set globally in cherrypy. Do this
1720 # work up front, because calls to _Log() below depend on this
1721 # initialization.
1722 if options.production:
1723 cherrypy.config.update({'environment': 'production'})
1724 if not options.logfile:
1725 cherrypy.config.update({'log.screen': True})
1726 else:
1727 cherrypy.config.update({'log.error_file': '',
1728 'log.access_file': ''})
Paul Hobbsef4e0702016-06-27 17:01:42 -07001729 hdlr = MakeLogHandler(options.logfile)
J. Richard Barnette3d977b82013-04-23 11:05:19 -07001730 # Pylint can't seem to process these two calls properly
1731 # pylint: disable=E1101
1732 cherrypy.log.access_log.addHandler(hdlr)
1733 cherrypy.log.error_log.addHandler(hdlr)
1734 # pylint: enable=E1101
1735
joychened64b222013-06-21 16:39:34 -07001736 # set static_dir, from which everything will be served
joychen84d13772013-08-06 09:17:23 -07001737 options.static_dir = os.path.realpath(options.static_dir)
Chris Sosa0356d3b2010-09-16 15:46:22 -07001738
joychened64b222013-06-21 16:39:34 -07001739 cache_dir = os.path.join(options.static_dir, 'cache')
J. Richard Barnette3d977b82013-04-23 11:05:19 -07001740 # If our devserver is only supposed to serve payloads, we shouldn't be
1741 # mucking with the cache at all. If the devserver hadn't previously
1742 # generated a cache and is expected, the caller is using it wrong.
joychen7c2054a2013-07-25 11:14:07 -07001743 if os.path.exists(cache_dir):
Chris Sosadbc20082012-12-10 13:39:11 -08001744 _CleanCache(cache_dir, options.clear_cache)
Chris Sosa6b8c3742011-01-31 12:12:17 -08001745 else:
1746 os.makedirs(cache_dir)
Don Garrettf90edf02010-11-16 17:36:14 -08001747
Chris Sosadbc20082012-12-10 13:39:11 -08001748 _Log('Using cache directory %s' % cache_dir)
joychened64b222013-06-21 16:39:34 -07001749 _Log('Serving from %s' % options.static_dir)
rtc@google.com21a5ca32009-11-04 18:23:23 +00001750
joychen121fc9b2013-08-02 14:30:30 -07001751 _xbuddy = xbuddy.XBuddy(options.xbuddy_manage_builds,
1752 options.board,
joychen121fc9b2013-08-02 14:30:30 -07001753 static_dir=options.static_dir)
Chris Sosa75490802013-09-30 17:21:45 -07001754 if options.clear_cache and options.xbuddy_manage_builds:
1755 _xbuddy.CleanCache()
joychen121fc9b2013-08-02 14:30:30 -07001756
Chris Sosa6a3697f2013-01-29 16:44:43 -08001757 # We allow global use here to share with cherrypy classes.
1758 # pylint: disable=W0603
Chris Sosacde6bf42012-05-31 18:36:39 -07001759 global updater
Andrew de los Reyes52620802010-04-12 13:40:07 -07001760 updater = autoupdate.Autoupdate(
joychen121fc9b2013-08-02 14:30:30 -07001761 _xbuddy,
joychened64b222013-06-21 16:39:34 -07001762 static_dir=options.static_dir,
Chris Sosa5d342a22010-09-28 16:54:41 -07001763 forced_image=options.image,
Gilad Arnold0c9c8602012-10-02 23:58:58 -07001764 payload_path=options.payload,
Don Garrett0ad09372010-12-06 16:20:30 -08001765 proxy_port=options.proxy_port,
Chris Sosa4136e692010-10-28 23:42:37 -07001766 src_image=options.src_image,
Chris Sosa08d55a22011-01-19 16:08:02 -08001767 board=options.board,
Chris Sosa0f1ec842011-02-14 16:33:22 -08001768 copy_to_static_root=not options.exit,
David Zeuthen52ccd012013-10-31 12:58:26 -07001769 public_key=options.public_key,
Satoru Takabayashid733cbe2011-11-15 09:36:32 -08001770 critical_update=options.critical_update,
Gilad Arnolda564b4b2012-10-04 10:32:44 -07001771 max_updates=options.max_updates,
Gilad Arnold8318eac2012-10-04 12:52:23 -07001772 host_log=options.host_log,
Chris Sosa0f1ec842011-02-14 16:33:22 -08001773 )
Chris Sosa7c931362010-10-11 19:49:01 -07001774
Chris Sosa6a3697f2013-01-29 16:44:43 -08001775 if options.pregenerate_update:
1776 updater.PreGenerateUpdate()
Chris Sosa0356d3b2010-09-16 15:46:22 -07001777
J. Richard Barnette3d977b82013-04-23 11:05:19 -07001778 if options.exit:
1779 return
Chris Sosa2f1c41e2012-07-10 14:32:33 -07001780
joychen3cb228e2013-06-12 12:13:13 -07001781 dev_server = DevServerRoot(_xbuddy)
Congbin Guo3afae6c2019-08-13 16:29:42 -07001782 health_checker_app = health_checker.Root(dev_server, options.static_dir)
joychen3cb228e2013-06-12 12:13:13 -07001783
Gilad Arnold11fbef42014-02-10 11:04:13 -08001784 # Patch CherryPy to support binding to any available port (--port=0).
1785 cherrypy_ext.ZeroPortPatcher.DoPatch(cherrypy)
1786
Chris Sosa855b8932013-08-21 13:24:55 -07001787 if options.pidfile:
1788 plugins.PIDFile(cherrypy.engine, options.pidfile).subscribe()
1789
Gilad Arnold11fbef42014-02-10 11:04:13 -08001790 if options.portfile:
1791 cherrypy_ext.PortFile(cherrypy.engine, options.portfile).subscribe()
1792
Dan Shiafd5c6c2016-01-07 10:27:03 -08001793 if (options.android_build_credential and
1794 os.path.exists(options.android_build_credential)):
1795 try:
1796 with open(options.android_build_credential) as f:
1797 android_build.BuildAccessor.credential_info = json.load(f)
1798 except ValueError as e:
1799 _Log('Failed to load the android build credential: %s. Error: %s.' %
1800 (options.android_build_credential, e))
Congbin Guo3afae6c2019-08-13 16:29:42 -07001801
1802 cherrypy.tree.mount(health_checker_app, '/check_health',
1803 config=health_checker.get_config())
joychen3cb228e2013-06-12 12:13:13 -07001804 cherrypy.quickstart(dev_server, config=_GetConfig(options))
Chris Sosacde6bf42012-05-31 18:36:39 -07001805
1806
1807if __name__ == '__main__':
1808 main()