blob: 3b89260816cb42ce6d43ae58c8675712c3de7841 [file] [log] [blame]
David Riley2fcb0122017-11-02 18:25:391#!/usr/bin/env python2
Luis Hector Chavezdca9dd72018-06-12 19:56:302# -*- coding: utf-8 -*-
Chris Sosa781ba6d2012-04-11 19:44:433# Copyright (c) 2009-2012 The Chromium OS Authors. All rights reserved.
[email protected]ded22402009-10-26 22:36:214# 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 18:47:007"""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
Amin Hassanie9ffb862019-09-26 00:10:4011systems.
Chris Sosa3ae4dc12013-03-29 18:47:0012
Amin Hassanie9ffb862019-09-26 00:10:4013The devserver is configured to stage and
Chris Sosa3ae4dc12013-03-29 18:47:0014serve 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
Chris Sosa3ae4dc12013-03-29 18:47:0021For autoupdates, there are many more advanced options that can help specify
22how to update and which payload to give to a requester.
23"""
24
Gabe Black3b567202015-09-23 21:07:5925from __future__ import print_function
Chris Sosa7c931362010-10-12 02:49:0126
Amin Hassanic5af4262019-11-13 21:37:2027import distutils.version # pylint: disable=import-error,no-name-in-module
Gilad Arnold55a2a372012-10-02 16:46:3228import json
David Riley2fcb0122017-11-02 18:25:3929import optparse # pylint: disable=deprecated-module
[email protected]ded22402009-10-26 22:36:2130import os
Scott Zawalski4647ce62012-01-03 22:17:2831import re
Simran Basi4baad082013-02-14 21:39:1832import shutil
xixuan52c2fba2016-05-21 00:02:4833import signal
Mandeep Singh Baines38dcdda2012-12-08 01:55:3334import socket
Chris Masone816e38c2012-05-02 19:22:3635import subprocess
J. Richard Barnette3d977b82013-04-23 18:05:1936import sys
Chris Masone816e38c2012-05-02 19:22:3637import tempfile
Dan Shi59ae7092013-06-04 21:37:2738import threading
Gilad Arnoldd5ebaaa2012-10-02 18:52:3839import types
J. Richard Barnette3d977b82013-04-23 18:05:1940from logging import handlers
41
Amin Hassanid4e35392019-10-03 18:02:4442from six.moves import http_client
43
44# pylint: disable=no-name-in-module, import-error
J. Richard Barnette3d977b82013-04-23 18:05:1945import cherrypy
Chris Sosa855b8932013-08-21 20:24:5546from cherrypy import _cplogging as cplogging
Amin Hassanid4e35392019-10-03 18:02:4447from cherrypy.process import plugins
48# pylint: enable=no-name-in-module, import-error
[email protected]ded22402009-10-26 22:36:2149
Achuith Bhandarkar662fb722019-10-31 23:12:4950import autoupdate
51import cherrypy_ext
52import cros_update
53import cros_update_progress
54import health_checker
55
Richard Barnettedf35c322017-08-19 00:02:1356# 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 23:29:4259import setup_chromite # pylint: disable=unused-import
Achuith Bhandarkar662fb722019-10-31 23:12:4960from chromite.lib.xbuddy import android_build
61from chromite.lib.xbuddy import artifact_info
62from chromite.lib.xbuddy import build_artifact
63from chromite.lib.xbuddy import cherrypy_log_util
64from chromite.lib.xbuddy import common_util
65from chromite.lib.xbuddy import devserver_constants
66from chromite.lib.xbuddy import downloader
67from chromite.lib.xbuddy import xbuddy
Richard Barnettedf35c322017-08-19 00:02:1368
Gilad Arnoldc65330c2012-09-20 22:17:4869
Gilad Arnoldc65330c2012-09-20 22:17:4870# Module-local log function.
Chris Sosa6a3697f2013-01-30 00:44:4371def _Log(message, *args):
Achuith Bhandarkar662fb722019-10-31 23:12:4972 return cherrypy_log_util.LogWithTag('DEVSERVER', message, *args)
Frank Farzan40160872011-12-13 02:39:1873
Chris Sosa417e55d2011-01-26 00:40:4874CACHED_ENTRIES = 12
Don Garrettf90edf02010-11-17 01:36:1475
Simran Basi4baad082013-02-14 21:39:1876TELEMETRY_FOLDER = 'telemetry_src'
77TELEMETRY_DEPS = ['dep-telemetry_dep.tar.bz2',
78 'dep-page_cycler_dep.tar.bz2',
Simran Basi0d078682013-03-22 23:40:0479 'dep-chrome_test.tar.bz2',
80 'dep-perf_data_dep.tar.bz2']
Simran Basi4baad082013-02-14 21:39:1881
Chris Sosa0356d3b2010-09-16 22:46:2282# Sets up global to share between classes.
[email protected]21a5ca32009-11-04 18:23:2383updater = None
[email protected]ded22402009-10-26 22:36:2184
xixuan3d48bff2017-01-31 03:00:0985# Log rotation parameters. These settings correspond to twice a day once
86# devserver is started, with about two weeks (28 backup files) of old logs
87# kept for backup.
J. Richard Barnette3d977b82013-04-23 18:05:1988#
xixuan3d48bff2017-01-31 03:00:0989# For more, see the documentation in standard python library for
J. Richard Barnette3d977b82013-04-23 18:05:1990# logging.handlers.TimedRotatingFileHandler
xixuan3d48bff2017-01-31 03:00:0991_LOG_ROTATION_TIME = 'H'
Congbin Guo3afae6c2019-08-13 23:29:4292_LOG_ROTATION_INTERVAL = 12 # hours
93_LOG_ROTATION_BACKUP = 28 # backup counts
Frank Farzan40160872011-12-13 02:39:1894
xixuan52c2fba2016-05-21 00:02:4895# Auto-update parameters
96
97# Error msg for missing key in CrOS auto-update.
Xixuan Wu32af9f12017-11-13 22:11:4498KEY_ERROR_MSG = 'Key Error in RPC: %s= is required'
xixuan52c2fba2016-05-21 00:02:4899
xixuan52c2fba2016-05-21 00:02:48100
Amin Hassanid4e35392019-10-03 18:02:44101class DevServerError(Exception):
102 """Exception class used by DevServer."""
103
104
Gabe Black3b567202015-09-23 21:07:59105def _canonicalize_archive_url(archive_url):
106 """Canonicalizes archive_url strings.
107
108 Raises:
109 DevserverError: if archive_url is not set.
110 """
111 if archive_url:
112 if not archive_url.startswith('gs://'):
Amin Hassanid4e35392019-10-03 18:02:44113 raise DevServerError(
Congbin Guo4132a272019-08-20 19:32:14114 "Archive URL isn't from Google Storage (%s) ." % archive_url)
Gabe Black3b567202015-09-23 21:07:59115
116 return archive_url.rstrip('/')
117 else:
Amin Hassanid4e35392019-10-03 18:02:44118 raise DevServerError('Must specify an archive_url in the request')
Gabe Black3b567202015-09-23 21:07:59119
120
121def _canonicalize_local_path(local_path):
122 """Canonicalizes |local_path| strings.
123
124 Raises:
125 DevserverError: if |local_path| is not set.
126 """
127 # Restrict staging of local content to only files within the static
128 # directory.
129 local_path = os.path.abspath(local_path)
130 if not local_path.startswith(updater.static_dir):
Amin Hassanid4e35392019-10-03 18:02:44131 raise DevServerError(
Congbin Guo4132a272019-08-20 19:32:14132 'Local path %s must be a subdirectory of the static'
133 ' directory: %s' % (local_path, updater.static_dir))
Gabe Black3b567202015-09-23 21:07:59134
135 return local_path.rstrip('/')
136
137
138def _get_artifacts(kwargs):
139 """Returns a tuple of named and file artifacts given the stage rpc kwargs.
140
141 Raises:
142 DevserverError if no artifacts would be returned.
143 """
144 artifacts = kwargs.get('artifacts')
145 files = kwargs.get('files')
146 if not artifacts and not files:
Amin Hassanid4e35392019-10-03 18:02:44147 raise DevServerError('No artifacts specified.')
Gabe Black3b567202015-09-23 21:07:59148
149 # Note we NEED to coerce files to a string as we get raw unicode from
150 # cherrypy and we treat files as strings elsewhere in the code.
151 return (str(artifacts).split(',') if artifacts else [],
152 str(files).split(',') if files else [])
153
154
Dan Shi61305df2015-10-26 23:52:35155def _is_android_build_request(kwargs):
156 """Check if a devserver call is for Android build, based on the arguments.
157
158 This method exams the request's arguments (os_type) to determine if the
159 request is for Android build. If os_type is set to `android`, returns True.
160 If os_type is not set or has other values, returns False.
161
162 Args:
163 kwargs: Keyword arguments for the request.
164
165 Returns:
166 True if the request is for Android build. False otherwise.
167 """
168 os_type = kwargs.get('os_type', None)
169 return os_type == 'android'
170
171
Gabe Black3b567202015-09-23 21:07:59172def _get_downloader(kwargs):
173 """Returns the downloader based on passed in arguments.
174
175 Args:
Amin Hassani08e42d22019-06-03 07:31:30176 kwargs: Keyword arguments for the request.
Gabe Black3b567202015-09-23 21:07:59177 """
178 local_path = kwargs.get('local_path')
179 if local_path:
180 local_path = _canonicalize_local_path(local_path)
181
182 dl = None
183 if local_path:
Prathmesh Prabhu58d08932018-01-19 23:08:19184 delete_source = _parse_boolean_arg(kwargs, 'delete_source')
185 dl = downloader.LocalDownloader(updater.static_dir, local_path,
186 delete_source=delete_source)
Gabe Black3b567202015-09-23 21:07:59187
Dan Shi61305df2015-10-26 23:52:35188 if not _is_android_build_request(kwargs):
Gabe Black3b567202015-09-23 21:07:59189 archive_url = kwargs.get('archive_url')
190 if not archive_url and not local_path:
Amin Hassanid4e35392019-10-03 18:02:44191 raise DevServerError(
Congbin Guo4132a272019-08-20 19:32:14192 'Requires archive_url or local_path to be specified.')
Gabe Black3b567202015-09-23 21:07:59193 if archive_url and local_path:
Amin Hassanid4e35392019-10-03 18:02:44194 raise DevServerError(
Congbin Guo4132a272019-08-20 19:32:14195 'archive_url and local_path can not both be specified.')
Gabe Black3b567202015-09-23 21:07:59196 if not dl:
197 archive_url = _canonicalize_archive_url(archive_url)
Luis Hector Chavezdca9dd72018-06-12 19:56:30198 dl = downloader.GoogleStorageDownloader(
199 updater.static_dir, archive_url,
200 downloader.GoogleStorageDownloader.GetBuildIdFromArchiveURL(
201 archive_url))
Gabe Black3b567202015-09-23 21:07:59202 elif not dl:
203 target = kwargs.get('target', None)
Dan Shi72b16132015-10-08 19:10:33204 branch = kwargs.get('branch', None)
Dan Shi61305df2015-10-26 23:52:35205 build_id = kwargs.get('build_id', None)
206 if not target or not branch or not build_id:
Amin Hassanid4e35392019-10-03 18:02:44207 raise DevServerError('target, branch, build ID must all be specified for '
208 'downloading Android build.')
Dan Shi72b16132015-10-08 19:10:33209 dl = downloader.AndroidBuildDownloader(updater.static_dir, branch, build_id,
210 target)
Gabe Black3b567202015-09-23 21:07:59211
212 return dl
213
214
215def _get_downloader_and_factory(kwargs):
216 """Returns the downloader and artifact factory based on passed in arguments.
217
218 Args:
Amin Hassani08e42d22019-06-03 07:31:30219 kwargs: Keyword arguments for the request.
Gabe Black3b567202015-09-23 21:07:59220 """
221 artifacts, files = _get_artifacts(kwargs)
222 dl = _get_downloader(kwargs)
223
Achuith Bhandarkar2a1fcd82019-10-18 00:45:58224 if (isinstance(dl, (downloader.GoogleStorageDownloader,
225 downloader.LocalDownloader))):
Gabe Black3b567202015-09-23 21:07:59226 factory_class = build_artifact.ChromeOSArtifactFactory
Dan Shi72b16132015-10-08 19:10:33227 elif isinstance(dl, downloader.AndroidBuildDownloader):
Gabe Black3b567202015-09-23 21:07:59228 factory_class = build_artifact.AndroidArtifactFactory
229 else:
Amin Hassanid4e35392019-10-03 18:02:44230 raise DevServerError(
Congbin Guo4132a272019-08-20 19:32:14231 'Unrecognized value for downloader type: %s' % type(dl))
Gabe Black3b567202015-09-23 21:07:59232
233 factory = factory_class(dl.GetBuildDir(), artifacts, files, dl.GetBuild())
234
235 return dl, factory
236
237
Scott Zawalski4647ce62012-01-03 22:17:28238def _LeadingWhiteSpaceCount(string):
239 """Count the amount of leading whitespace in a string.
240
241 Args:
242 string: The string to count leading whitespace in.
Don Garrettf84631a2014-01-08 02:21:26243
Scott Zawalski4647ce62012-01-03 22:17:28244 Returns:
245 number of white space chars before characters start.
246 """
Gabe Black3b567202015-09-23 21:07:59247 matched = re.match(r'^\s+', string)
Scott Zawalski4647ce62012-01-03 22:17:28248 if matched:
249 return len(matched.group())
250
251 return 0
252
253
254def _PrintDocStringAsHTML(func):
255 """Make a functions docstring somewhat HTML style.
256
257 Args:
258 func: The function to return the docstring from.
Don Garrettf84631a2014-01-08 02:21:26259
Scott Zawalski4647ce62012-01-03 22:17:28260 Returns:
261 A string that is somewhat formated for a web browser.
262 """
263 # TODO(scottz): Make this parse Args/Returns in a prettier way.
264 # Arguments could be bolded and indented etc.
265 html_doc = []
266 for line in func.__doc__.splitlines():
267 leading_space = _LeadingWhiteSpaceCount(line)
268 if leading_space > 0:
Chris Sosa47a7d4e2012-03-28 18:26:55269 line = ' ' * leading_space + line
Scott Zawalski4647ce62012-01-03 22:17:28270
271 html_doc.append('<BR>%s' % line)
272
273 return '\n'.join(html_doc)
274
275
Simran Basief83d6a2014-08-28 21:32:01276def _GetUpdateTimestampHandler(static_dir):
277 """Returns a handler to update directory staged.timestamp.
278
279 This handler resets the stage.timestamp whenever static content is accessed.
280
281 Args:
282 static_dir: Directory from which static content is being staged.
283
284 Returns:
Amin Hassani08e42d22019-06-03 07:31:30285 A cherrypy handler to update the timestamp of accessed content.
Simran Basief83d6a2014-08-28 21:32:01286 """
287 def UpdateTimestampHandler():
288 if not '404' in cherrypy.response.status:
289 build_match = re.match(devserver_constants.STAGED_BUILD_REGEX,
290 cherrypy.request.path_info)
291 if build_match:
292 build_dir = os.path.join(static_dir, build_match.group('build'))
293 downloader.Downloader.TouchTimestampForStaged(build_dir)
294 return UpdateTimestampHandler
295
296
Chris Sosa7c931362010-10-12 02:49:01297def _GetConfig(options):
298 """Returns the configuration for the devserver."""
Mandeep Singh Baines38dcdda2012-12-08 01:55:33299
Mandeep Singh Baines38dcdda2012-12-08 01:55:33300 socket_host = '::'
Yu-Ju Hongc8d4af32013-11-12 23:14:26301 # Fall back to IPv4 when python is not configured with IPv6.
302 if not socket.has_ipv6:
Mandeep Singh Baines38dcdda2012-12-08 01:55:33303 socket_host = '0.0.0.0'
304
Simran Basief83d6a2014-08-28 21:32:01305 # Adds the UpdateTimestampHandler to cherrypy's tools. This tools executes
306 # on the on_end_resource hook. This hook is called once processing is
307 # complete and the response is ready to be returned.
308 cherrypy.tools.update_timestamp = cherrypy.Tool(
309 'on_end_resource', _GetUpdateTimestampHandler(options.static_dir))
310
David Riley2fcb0122017-11-02 18:25:39311 base_config = {
312 'global': {
313 'server.log_request_headers': True,
314 'server.protocol_version': 'HTTP/1.1',
315 'server.socket_host': socket_host,
316 'server.socket_port': int(options.port),
317 'response.timeout': 6000,
318 'request.show_tracebacks': True,
319 'server.socket_timeout': 60,
320 'server.thread_pool': 2,
321 'engine.autoreload.on': False,
322 },
323 '/api': {
324 # Gets rid of cherrypy parsing post file for args.
325 'request.process_request_body': False,
326 },
327 '/build': {
328 'response.timeout': 100000,
329 },
330 '/update': {
331 # Gets rid of cherrypy parsing post file for args.
332 'request.process_request_body': False,
333 'response.timeout': 10000,
334 },
335 # Sets up the static dir for file hosting.
336 '/static': {
337 'tools.staticdir.dir': options.static_dir,
338 'tools.staticdir.on': True,
339 'response.timeout': 10000,
340 'tools.update_timestamp.on': True,
341 },
342 }
Chris Sosa5f118ef2012-07-12 18:37:50343 if options.production:
Alex Miller93beca52013-07-31 02:25:09344 base_config['global'].update({'server.thread_pool': 150})
Scott Zawalski1c5e7cd2012-02-27 18:12:52345
Chris Sosa7c931362010-10-12 02:49:01346 return base_config
[email protected]64244662009-11-12 00:52:08347
Darin Petkove17164a2010-08-11 20:24:41348
Gilad Arnoldd5ebaaa2012-10-02 18:52:38349def _GetRecursiveMemberObject(root, member_list):
350 """Returns an object corresponding to a nested member list.
351
352 Args:
353 root: the root object to search
354 member_list: list of nested members to search
Don Garrettf84631a2014-01-08 02:21:26355
Gilad Arnoldd5ebaaa2012-10-02 18:52:38356 Returns:
357 An object corresponding to the member name list; None otherwise.
358 """
359 for member in member_list:
360 next_root = root.__class__.__dict__.get(member)
361 if not next_root:
362 return None
363 root = next_root
364 return root
365
366
367def _IsExposed(name):
368 """Returns True iff |name| has an `exposed' attribute and it is set."""
369 return hasattr(name, 'exposed') and name.exposed
370
371
Congbin Guo6bc32182019-08-21 00:54:30372def _GetExposedMethod(nested_member):
Gilad Arnoldd5ebaaa2012-10-02 18:52:38373 """Returns a CherryPy-exposed method, if such exists.
374
375 Args:
Gilad Arnoldd5ebaaa2012-10-02 18:52:38376 nested_member: a slash-joined path to the nested member
Don Garrettf84631a2014-01-08 02:21:26377
Gilad Arnoldd5ebaaa2012-10-02 18:52:38378 Returns:
Congbin Guo6bc32182019-08-21 00:54:30379 A function object corresponding to the path defined by |nested_member| from
380 the app root object registered, if the function is exposed; None otherwise.
Gilad Arnoldd5ebaaa2012-10-02 18:52:38381 """
Congbin Guo6bc32182019-08-21 00:54:30382 for app in cherrypy.tree.apps.values():
383 # Use the 'index' function doc as the doc of the app.
384 if nested_member == app.script_name.lstrip('/'):
385 nested_member = 'index'
386
387 method = _GetRecursiveMemberObject(app.root, nested_member.split('/'))
388 if method and isinstance(method, types.FunctionType) and _IsExposed(method):
389 return method
Gilad Arnoldd5ebaaa2012-10-02 18:52:38390
391
Gilad Arnold748c8322012-10-12 16:51:35392def _FindExposedMethods(root, prefix, unlisted=None):
Gilad Arnoldd5ebaaa2012-10-02 18:52:38393 """Finds exposed CherryPy methods.
394
395 Args:
396 root: the root object for searching
397 prefix: slash-joined chain of members leading to current object
398 unlisted: URLs to be excluded regardless of their exposed status
Don Garrettf84631a2014-01-08 02:21:26399
Gilad Arnoldd5ebaaa2012-10-02 18:52:38400 Returns:
401 List of exposed URLs that are not unlisted.
402 """
403 method_list = []
Congbin Guo6bc32182019-08-21 00:54:30404 for member in root.__class__.__dict__.keys():
Gilad Arnoldd5ebaaa2012-10-02 18:52:38405 prefixed_member = prefix + '/' + member if prefix else member
Gilad Arnold748c8322012-10-12 16:51:35406 if unlisted and prefixed_member in unlisted:
Gilad Arnoldd5ebaaa2012-10-02 18:52:38407 continue
408 member_obj = root.__class__.__dict__[member]
409 if _IsExposed(member_obj):
Amin Hassani08e42d22019-06-03 07:31:30410 if isinstance(member_obj, types.FunctionType):
Congbin Guo6bc32182019-08-21 00:54:30411 # Regard the app name as exposed "method" name if it exposed 'index'
412 # function.
413 if prefix and member == 'index':
414 method_list.append(prefix)
415 else:
416 method_list.append(prefixed_member)
Gilad Arnoldd5ebaaa2012-10-02 18:52:38417 else:
418 method_list += _FindExposedMethods(
419 member_obj, prefixed_member, unlisted)
420 return method_list
421
422
xixuan52c2fba2016-05-21 00:02:48423def _check_base_args_for_auto_update(kwargs):
xixuanac89ce82016-12-01 00:48:20424 """Check basic args required for auto-update.
425
426 Args:
427 kwargs: the parameters to be checked.
428
429 Raises:
430 DevServerHTTPError if required parameters don't exist in kwargs.
431 """
xixuan52c2fba2016-05-21 00:02:48432 if 'host_name' not in kwargs:
Amin Hassanid4e35392019-10-03 18:02:44433 raise common_util.DevServerHTTPError(http_client.INTERNAL_SERVER_ERROR,
Amin Hassani08e42d22019-06-03 07:31:30434 KEY_ERROR_MSG % 'host_name')
xixuan52c2fba2016-05-21 00:02:48435
436 if 'build_name' not in kwargs:
Amin Hassanid4e35392019-10-03 18:02:44437 raise common_util.DevServerHTTPError(http_client.INTERNAL_SERVER_ERROR,
Amin Hassani08e42d22019-06-03 07:31:30438 KEY_ERROR_MSG % 'build_name')
xixuan52c2fba2016-05-21 00:02:48439
440
441def _parse_boolean_arg(kwargs, key):
xixuanac89ce82016-12-01 00:48:20442 """Parse boolean arg from kwargs.
443
444 Args:
445 kwargs: the parameters to be checked.
446 key: the key to be parsed.
447
448 Returns:
449 The boolean value of kwargs[key], or False if key doesn't exist in kwargs.
450
451 Raises:
452 DevServerHTTPError if kwargs[key] is not a boolean variable.
453 """
xixuan52c2fba2016-05-21 00:02:48454 if key in kwargs:
455 if kwargs[key] == 'True':
456 return True
457 elif kwargs[key] == 'False':
458 return False
459 else:
460 raise common_util.DevServerHTTPError(
Amin Hassanid4e35392019-10-03 18:02:44461 http_client.INTERNAL_SERVER_ERROR,
xixuan52c2fba2016-05-21 00:02:48462 'The value for key %s is not boolean.' % key)
463 else:
464 return False
465
xixuan447ad9d2017-02-28 22:46:20466
xixuanac89ce82016-12-01 00:48:20467def _parse_string_arg(kwargs, key):
468 """Parse string arg from kwargs.
469
470 Args:
471 kwargs: the parameters to be checked.
472 key: the key to be parsed.
473
474 Returns:
475 The string value of kwargs[key], or None if key doesn't exist in kwargs.
476 """
477 if key in kwargs:
478 return kwargs[key]
479 else:
480 return None
481
xixuan447ad9d2017-02-28 22:46:20482
xixuanac89ce82016-12-01 00:48:20483def _build_uri_from_build_name(build_name):
484 """Get build url from a given build name.
485
486 Args:
487 build_name: the build name to be parsed, whose format is
488 'board/release_version'.
489
490 Returns:
491 The release_archive_url on Google Storage for this build name.
492 """
Amin Hassani08e42d22019-06-03 07:31:30493 # TODO(ahassani): This function doesn't seem to be used anywhere since its
494 # previous use of lib.paygen.gspath was broken and it doesn't seem to be
495 # causing any runtime issues. So deprecate this in the future.
496 tokens = build_name.split('/')
497 return 'gs://chromeos-releases/stable-channel/%s/%s' % (tokens[0], tokens[1])
xixuan52c2fba2016-05-21 00:02:48498
xixuan447ad9d2017-02-28 22:46:20499
500def _clear_process(host_name, pid):
501 """Clear AU process for given hostname and pid.
502
503 This clear includes:
504 1. kill process if it's alive.
505 2. delete the track status file of this process.
506 3. delete the executing log file of this process.
507
508 Args:
509 host_name: the host to execute auto-update.
510 pid: the background auto-update process id.
511 """
512 if cros_update_progress.IsProcessAlive(pid):
513 os.killpg(int(pid), signal.SIGKILL)
514
515 cros_update_progress.DelTrackStatusFile(host_name, pid)
516 cros_update_progress.DelExecuteLogFile(host_name, pid)
517
518
Dale Curtisc9aaf3a2011-08-09 22:47:40519class ApiRoot(object):
520 """RESTful API for Dev Server information."""
521 exposed = True
522
523 @cherrypy.expose
Gilad Arnold286a0062012-01-12 21:47:02524 def hostlog(self, ip):
Gilad Arnold1b908392012-10-05 18:36:27525 """Returns a JSON object containing a log of host event.
526
527 Args:
528 ip: address of host whose event log is requested, or `all'
Don Garrettf84631a2014-01-08 02:21:26529
Gilad Arnold1b908392012-10-05 18:36:27530 Returns:
Amin Hassani7c447852019-09-26 22:01:48531 A JSON dictionary containing all or some of the following fields:
Amin Hassanie7ead902019-10-11 23:42:43532 version: The Chromium OS version the device is running.
533 track: The channel the device is running on.
534 board: The device's board.
535 event_result: The event result of Omaha request.
536 event_type: The event type of Omaha request.
537 previous_version: The Chromium OS version we updated and rebooted from.
538 timestamp: The timestamp the event was received.
Amin Hassani7c447852019-09-26 22:01:48539 See the OmahaEvent class in update_engine/omaha_request_action.h for
540 event type and status code definitions. If the ip does not exist an empty
541 string is returned.
Gilad Arnold1b908392012-10-05 18:36:27542
543 Example URL:
544 http://myhost/api/hostlog?ip=192.168.1.5
545 """
Gilad Arnold286a0062012-01-12 21:47:02546 return updater.HandleHostLogPing(ip)
547
548 @cherrypy.expose
Don Garrettf84631a2014-01-08 02:21:26549 def fileinfo(self, *args):
Gilad Arnold55a2a372012-10-02 16:46:32550 """Returns information about a given staged file.
551
552 Args:
Don Garrettf84631a2014-01-08 02:21:26553 args: path to the file inside the server's static staging directory
554
Gilad Arnold55a2a372012-10-02 16:46:32555 Returns:
556 A JSON encoded dictionary with information about the said file, which may
557 contain the following keys/values:
Gilad Arnold1b908392012-10-05 18:36:27558 size (int): the file size in bytes
Gilad Arnold1b908392012-10-05 18:36:27559 sha256 (string): a base64 encoded SHA256 hash
560
561 Example URL:
562 http://myhost/api/fileinfo/some/path/to/file
Gilad Arnold55a2a372012-10-02 16:46:32563 """
Amin Hassani28df4212019-10-28 17:16:50564 # TODO(ahassani): A better way of doing this is to just return the the
565 # content of the payloads' property file instead. That has all this info
566 # except that the key for sha256 is 'sha256_hex', but still base64 encdoed.
567
Don Garrettf84631a2014-01-08 02:21:26568 file_path = os.path.join(updater.static_dir, *args)
Gilad Arnold55a2a372012-10-02 16:46:32569 if not os.path.exists(file_path):
Amin Hassanid4e35392019-10-03 18:02:44570 raise DevServerError('file not found: %s' % file_path)
Gilad Arnold55a2a372012-10-02 16:46:32571 try:
572 file_size = os.path.getsize(file_path)
Gilad Arnold55a2a372012-10-02 16:46:32573 file_sha256 = common_util.GetFileSha256(file_path)
Amin Hassani469f5702019-10-21 22:35:06574 except os.error as e:
Amin Hassanid4e35392019-10-03 18:02:44575 raise DevServerError(
Congbin Guo4132a272019-08-20 19:32:14576 'failed to get info for file %s: %s' % (file_path, e))
Gilad Arnolde74b3812013-04-22 18:27:38577
Gilad Arnolde74b3812013-04-22 18:27:38578 return json.dumps({
Amin Hassani28df4212019-10-28 17:16:50579 'size': file_size,
580 'sha256': file_sha256,
581 }, sort_keys=True)
Gilad Arnold55a2a372012-10-02 16:46:32582
Chris Sosa76e44b92013-01-31 20:11:38583
David Rochberg7c79a812011-01-19 19:24:45584class DevServerRoot(object):
Chris Sosa7c931362010-10-12 02:49:01585 """The Root Class for the Dev Server.
586
587 CherryPy works as follows:
588 For each method in this class, cherrpy interprets root/path
589 as a call to an instance of DevServerRoot->method_name. For example,
590 a call to http://myhost/build will call build. CherryPy automatically
591 parses http args and places them as keyword arguments in each method.
592 For paths http://myhost/update/dir1/dir2, you can use *args so that
593 cherrypy uses the update method and puts the extra paths in args.
594 """
Gilad Arnoldf8f769f2012-09-24 15:43:01595 # Method names that should not be listed on the index page.
596 _UNLISTED_METHODS = ['index', 'doc']
597
Dale Curtisc9aaf3a2011-08-09 22:47:40598 api = ApiRoot()
Chris Sosa7c931362010-10-12 02:49:01599
Dan Shi59ae7092013-06-04 21:37:27600 # Number of threads that devserver is staging images.
601 _staging_thread_count = 0
602 # Lock used to lock increasing/decreasing count.
603 _staging_thread_count_lock = threading.Lock()
604
joychen3cb228e2013-06-12 19:13:13605 def __init__(self, _xbuddy):
Nick Sanders7dcaa2e2011-08-04 22:20:41606 self._builder = None
Simran Basi4baad082013-02-14 21:39:18607 self._telemetry_lock_dict = common_util.LockDict()
joychen3cb228e2013-06-12 19:13:13608 self._xbuddy = _xbuddy
David Rochberg7c79a812011-01-19 19:24:45609
Congbin Guo3afae6c2019-08-13 23:29:42610 @property
611 def staging_thread_count(self):
612 """Get the staging thread count."""
613 return self._staging_thread_count
Dan Shiafd0e492015-05-27 21:23:51614
Dale Curtisc9aaf3a2011-08-09 22:47:40615 @cherrypy.expose
David Rochberg7c79a812011-01-19 19:24:45616 def build(self, board, pkg, **kwargs):
Chris Sosa7c931362010-10-12 02:49:01617 """Builds the package specified."""
Nick Sanders7dcaa2e2011-08-04 22:20:41618 import builder
619 if self._builder is None:
620 self._builder = builder.Builder()
David Rochberg7c79a812011-01-19 19:24:45621 return self._builder.Build(board, pkg, kwargs)
Chris Sosa7c931362010-10-12 02:49:01622
Dale Curtisc9aaf3a2011-08-09 22:47:40623 @cherrypy.expose
Dan Shif8eb0d12013-08-02 00:52:06624 def is_staged(self, **kwargs):
625 """Check if artifacts have been downloaded.
626
Congbin Guo3afae6c2019-08-13 23:29:42627 Examples:
628 To check if autotest and test_suites are staged:
629 http://devserver_url:<port>/is_staged?archive_url=gs://your_url/path&
630 artifacts=autotest,test_suites
631
Amin Hassani08e42d22019-06-03 07:31:30632 Args:
Chris Sosa6b0c6172013-08-06 00:01:33633 async: True to return without waiting for download to complete.
634 artifacts: Comma separated list of named artifacts to download.
635 These are defined in artifact_info and have their implementation
636 in build_artifact.py.
637 files: Comma separated list of file artifacts to stage. These
638 will be available as is in the corresponding static directory with no
639 custom post-processing.
640
Congbin Guo3afae6c2019-08-13 23:29:42641 Returns:
642 True of all artifacts are staged.
Dan Shif8eb0d12013-08-02 00:52:06643 """
Gabe Black3b567202015-09-23 21:07:59644 dl, factory = _get_downloader_and_factory(kwargs)
Aviv Keshet57d18172016-06-19 03:39:09645 response = str(dl.IsStaged(factory))
646 _Log('Responding to is_staged %s request with %r', kwargs, response)
647 return response
Dan Shi59ae7092013-06-04 21:37:27648
Chris Sosa76e44b92013-01-31 20:11:38649 @cherrypy.expose
Prashanth Ba06d2d22014-03-07 23:35:19650 def list_image_dir(self, **kwargs):
651 """Take an archive url and list the contents in its staged directory.
652
Amin Hassani08e42d22019-06-03 07:31:30653 Examples:
Prashanth Ba06d2d22014-03-07 23:35:19654 To list the contents of where this devserver should have staged
655 gs://image-archive/<board>-release/<build> call:
656 http://devserver_url:<port>/list_image_dir?archive_url=<gs://..>
657
Congbin Guo3afae6c2019-08-13 23:29:42658 Args:
659 archive_url: Google Storage URL for the build.
660
Prashanth Ba06d2d22014-03-07 23:35:19661 Returns:
662 A string with information about the contents of the image directory.
663 """
Gabe Black3b567202015-09-23 21:07:59664 dl = _get_downloader(kwargs)
Prashanth Ba06d2d22014-03-07 23:35:19665 try:
Gabe Black3b567202015-09-23 21:07:59666 image_dir_contents = dl.ListBuildDir()
Prashanth Ba06d2d22014-03-07 23:35:19667 except build_artifact.ArtifactDownloadError as e:
668 return 'Cannot list the contents of staged artifacts. %s' % e
669 if not image_dir_contents:
Gabe Black3b567202015-09-23 21:07:59670 return '%s has not been staged on this devserver.' % dl.DescribeSource()
Prashanth Ba06d2d22014-03-07 23:35:19671 return image_dir_contents
672
673 @cherrypy.expose
Chris Sosa76e44b92013-01-31 20:11:38674 def stage(self, **kwargs):
Gabe Black3b567202015-09-23 21:07:59675 """Downloads and caches build artifacts.
Chris Sosa76e44b92013-01-31 20:11:38676
Gabe Black3b567202015-09-23 21:07:59677 Downloads and caches build artifacts, possibly from a Google Storage URL,
Dan Shi72b16132015-10-08 19:10:33678 or from Android's build server. Returns once these have been downloaded
Gabe Black3b567202015-09-23 21:07:59679 on the devserver. A call to this will attempt to cache non-specified
680 artifacts in the background for the given from the given URL following
681 the principle of spatial locality. Spatial locality of different
Chris Sosa76e44b92013-01-31 20:11:38682 artifacts is explicitly defined in the build_artifact module.
683
684 These artifacts will then be available from the static/ sub-directory of
685 the devserver.
686
Amin Hassani08e42d22019-06-03 07:31:30687 Examples:
Chris Sosa76e44b92013-01-31 20:11:38688 To download the autotest and test suites tarballs:
689 http://devserver_url:<port>/stage?archive_url=gs://your_url/path&
690 artifacts=autotest,test_suites
691 To download the full update payload:
692 http://devserver_url:<port>/stage?archive_url=gs://your_url/path&
693 artifacts=full_payload
Chris Sosa6b0c6172013-08-06 00:01:33694 To download just a file called blah.bin:
695 http://devserver_url:<port>/stage?archive_url=gs://your_url/path&
696 files=blah.bin
Chris Sosa76e44b92013-01-31 20:11:38697
698 For both these examples, one could find these artifacts at:
joychened64b222013-06-21 23:39:34699 http://devserver_url:<port>/static/<relative_path>*
Chris Sosa76e44b92013-01-31 20:11:38700
701 Note for this example, relative path is the archive_url stripped of its
702 basename i.e. path/ in the examples above. Specific example:
703
704 gs://chromeos-image-archive/x86-mario-release/R26-3920.0.0
705
706 Will get staged to:
707
joychened64b222013-06-21 23:39:34708 http://devserver_url:<port>/static/x86-mario-release/R26-3920.0.0
Congbin Guo3afae6c2019-08-13 23:29:42709
710 Args:
711 archive_url: Google Storage URL for the build.
712 local_path: Local path for the build.
713 delete_source: Only meaningful with local_path. bool to indicate if the
714 source files should be deleted. This is especially useful when staging
715 a file locally in resource constrained environments as it allows us to
716 move the relevant files locally instead of copying them.
717 async: True to return without waiting for download to complete.
718 artifacts: Comma separated list of named artifacts to download.
719 These are defined in artifact_info and have their implementation
720 in build_artifact.py.
721 files: Comma separated list of files to stage. These
722 will be available as is in the corresponding static directory with no
723 custom post-processing.
724 clean: True to remove any previously staged artifacts first.
Chris Sosa76e44b92013-01-31 20:11:38725 """
Gabe Black3b567202015-09-23 21:07:59726 dl, factory = _get_downloader_and_factory(kwargs)
727
Dan Shi59ae7092013-06-04 21:37:27728 with DevServerRoot._staging_thread_count_lock:
729 DevServerRoot._staging_thread_count += 1
730 try:
Laurence Goodbyf5c958d2016-01-15 02:23:56731 boolean_string = kwargs.get('clean')
732 clean = xbuddy.XBuddy.ParseBoolean(boolean_string)
733 if clean and os.path.exists(dl.GetBuildDir()):
734 _Log('Removing %s' % dl.GetBuildDir())
735 shutil.rmtree(dl.GetBuildDir())
Achuith Bhandarkar2a1fcd82019-10-18 00:45:58736 is_async = kwargs.get('async', False)
737 dl.Download(factory, is_async=is_async)
Dan Shi59ae7092013-06-04 21:37:27738 finally:
739 with DevServerRoot._staging_thread_count_lock:
740 DevServerRoot._staging_thread_count -= 1
Chris Sosa76e44b92013-01-31 20:11:38741 return 'Success'
Chris Sosacde6bf42012-06-01 01:36:39742
743 @cherrypy.expose
xixuan52c2fba2016-05-21 00:02:48744 def cros_au(self, **kwargs):
745 """Auto-update a CrOS DUT.
746
747 Args:
748 kwargs:
749 host_name: the hostname of the DUT to auto-update.
750 build_name: the build name for update the DUT.
751 force_update: Force an update even if the version installed is the
752 same. Default: False.
753 full_update: If True, do not run stateful update, directly force a full
754 reimage. If False, try stateful update first if the dut is already
755 installed with the same version.
756 async: Whether the auto_update function is ran in the background.
David Rileyee75de22017-11-02 17:48:15757 quick_provision: Whether the quick provision path is attempted first.
xixuan52c2fba2016-05-21 00:02:48758
759 Returns:
760 A tuple includes two elements:
761 a boolean variable represents whether the auto-update process is
762 successfully started.
763 an integer represents the background auto-update process id.
764 """
765 _check_base_args_for_auto_update(kwargs)
766
767 host_name = kwargs['host_name']
768 build_name = kwargs['build_name']
769 force_update = _parse_boolean_arg(kwargs, 'force_update')
770 full_update = _parse_boolean_arg(kwargs, 'full_update')
Achuith Bhandarkar2a1fcd82019-10-18 00:45:58771 is_async = _parse_boolean_arg(kwargs, 'async')
xixuanac89ce82016-12-01 00:48:20772 original_build = _parse_string_arg(kwargs, 'original_build')
David Haddock90e49442017-04-08 02:14:09773 payload_filename = _parse_string_arg(kwargs, 'payload_filename')
David Haddock20559612017-06-29 05:15:08774 clobber_stateful = _parse_boolean_arg(kwargs, 'clobber_stateful')
David Rileyee75de22017-11-02 17:48:15775 quick_provision = _parse_boolean_arg(kwargs, 'quick_provision')
776
777 devserver_url = updater.GetDevserverUrl()
778 static_url = updater.GetStaticUrl()
xixuan52c2fba2016-05-21 00:02:48779
Achuith Bhandarkar2a1fcd82019-10-18 00:45:58780 if is_async:
Amin Hassani469f5702019-10-21 22:35:06781 # Command of running auto-update.
Amin Hassani78520ae2019-10-29 20:26:51782 path = os.path.dirname(os.path.abspath(__file__))
783 execute_file = os.path.join(path, 'cros_update.py')
784 cmd = ['/usr/bin/python', '-u', execute_file, '-d', host_name,
785 '-b', build_name, '--static_dir', updater.static_dir]
xixuanac89ce82016-12-01 00:48:20786
787 # The original_build's format is like: link/3428.210.0
788 # The corresponding release_archive_url's format is like:
789 # gs://chromeos-releases/stable-channel/link/3428.210.0
790 if original_build:
791 release_archive_url = _build_uri_from_build_name(original_build)
792 # First staging the stateful.tgz synchronousely.
Amin Hassani469f5702019-10-21 22:35:06793 self.stage(files='stateful.tgz', is_async=False,
xixuanac89ce82016-12-01 00:48:20794 archive_url=release_archive_url)
Amin Hassani469f5702019-10-21 22:35:06795 cmd += ['--original_build', original_build]
xixuanac89ce82016-12-01 00:48:20796
xixuan52c2fba2016-05-21 00:02:48797 if force_update:
Amin Hassani469f5702019-10-21 22:35:06798 cmd += ['--force_update']
xixuan52c2fba2016-05-21 00:02:48799
800 if full_update:
Amin Hassani469f5702019-10-21 22:35:06801 cmd += ['--full_update']
xixuan52c2fba2016-05-21 00:02:48802
David Haddock90e49442017-04-08 02:14:09803 if payload_filename:
Amin Hassani469f5702019-10-21 22:35:06804 cmd += ['--payload_filename', payload_filename]
David Haddock90e49442017-04-08 02:14:09805
David Haddock20559612017-06-29 05:15:08806 if clobber_stateful:
Amin Hassani469f5702019-10-21 22:35:06807 cmd += ['--clobber_stateful']
David Haddock20559612017-06-29 05:15:08808
David Rileyee75de22017-11-02 17:48:15809 if quick_provision:
Amin Hassani469f5702019-10-21 22:35:06810 cmd += ['--quick_provision']
David Rileyee75de22017-11-02 17:48:15811
812 if devserver_url:
Amin Hassani469f5702019-10-21 22:35:06813 cmd += ['--devserver_url', devserver_url]
David Rileyee75de22017-11-02 17:48:15814
815 if static_url:
Amin Hassani469f5702019-10-21 22:35:06816 cmd += ['--static_url', static_url]
David Rileyee75de22017-11-02 17:48:15817
Amin Hassani78520ae2019-10-29 20:26:51818 p = subprocess.Popen(cmd, preexec_fn=os.setsid)
xixuan2a0970a2016-08-10 19:12:44819 pid = os.getpgid(p.pid)
xixuan52c2fba2016-05-21 00:02:48820
821 # Pre-write status in the track_status_file before the first call of
822 # 'get_au_status' to make sure that the track_status_file exists.
xixuan2a0970a2016-08-10 19:12:44823 progress_tracker = cros_update_progress.AUProgress(host_name, pid)
xixuan52c2fba2016-05-21 00:02:48824 progress_tracker.WriteStatus('CrOS update is just started.')
825
xixuan2a0970a2016-08-10 19:12:44826 return json.dumps((True, pid))
xixuan52c2fba2016-05-21 00:02:48827 else:
828 cros_update_trigger = cros_update.CrOSUpdateTrigger(
xixuanac89ce82016-12-01 00:48:20829 host_name, build_name, updater.static_dir, force_update=force_update,
David Rileyee75de22017-11-02 17:48:15830 full_update=full_update, original_build=original_build,
Amin Hassani78520ae2019-10-29 20:26:51831 payload_filename=payload_filename, quick_provision=quick_provision,
832 devserver_url=devserver_url, static_url=static_url)
xixuan52c2fba2016-05-21 00:02:48833 cros_update_trigger.TriggerAU()
xixuan27d50442017-08-09 17:38:25834 return json.dumps((True, -1))
xixuan52c2fba2016-05-21 00:02:48835
836 @cherrypy.expose
837 def get_au_status(self, **kwargs):
838 """Check if the auto-update task is finished.
839
840 It handles 4 cases:
841 1. If an error exists in the track_status_file, delete the track file and
842 raise it.
843 2. If cros-update process is finished, delete the file and return the
844 success result.
845 3. If the process is not running, delete the track file and raise an error
846 about 'the process is terminated due to unknown reason'.
847 4. If the track_status_file does not exist, kill the process if it exists,
848 and raise the IOError.
849
850 Args:
851 kwargs:
852 host_name: the hostname of the DUT to auto-update.
853 pid: the background process id of cros-update.
854
855 Returns:
xixuan28d99072016-10-06 19:24:16856 A dict with three elements:
xixuan52c2fba2016-05-21 00:02:48857 a boolean variable represents whether the auto-update process is
858 finished.
859 a string represents the current auto-update process status.
860 For example, 'Transfer Devserver/Stateful Update Package'.
xixuan28d99072016-10-06 19:24:16861 a detailed error message paragraph if there exists an Auto-Update
862 error, in which the last line shows the main exception. Empty
863 string otherwise.
xixuan52c2fba2016-05-21 00:02:48864 """
865 if 'host_name' not in kwargs:
Amin Hassanid4e35392019-10-03 18:02:44866 raise common_util.DevServerHTTPError(http_client.INTERNAL_SERVER_ERROR,
Amin Hassani08e42d22019-06-03 07:31:30867 KEY_ERROR_MSG % 'host_name')
xixuan52c2fba2016-05-21 00:02:48868
869 if 'pid' not in kwargs:
Amin Hassanid4e35392019-10-03 18:02:44870 raise common_util.DevServerHTTPError(http_client.INTERNAL_SERVER_ERROR,
Amin Hassani08e42d22019-06-03 07:31:30871 KEY_ERROR_MSG % 'pid')
xixuan52c2fba2016-05-21 00:02:48872
873 host_name = kwargs['host_name']
874 pid = kwargs['pid']
875 progress_tracker = cros_update_progress.AUProgress(host_name, pid)
876
xixuan28d99072016-10-06 19:24:16877 result_dict = {'finished': False, 'status': '', 'detailed_error_msg': ''}
xixuan52c2fba2016-05-21 00:02:48878 try:
879 result = progress_tracker.ReadStatus()
880 if result.startswith(cros_update_progress.ERROR_TAG):
xixuan28d99072016-10-06 19:24:16881 result_dict['detailed_error_msg'] = result[len(
882 cros_update_progress.ERROR_TAG):]
xixuan28681fd2016-11-23 19:13:56883 elif result == cros_update_progress.FINISHED:
xixuan28d99072016-10-06 19:24:16884 result_dict['finished'] = True
885 result_dict['status'] = result
xixuan28681fd2016-11-23 19:13:56886 elif not cros_update_progress.IsProcessAlive(pid):
xixuan28d99072016-10-06 19:24:16887 result_dict['detailed_error_msg'] = (
888 'Cros_update process terminated midway due to unknown reason. '
889 'Last update status was %s' % result)
xixuan28681fd2016-11-23 19:13:56890 else:
891 result_dict['status'] = result
892 except IOError as e:
893 if pid and cros_update_progress.IsProcessAlive(pid):
xixuan2a0970a2016-08-10 19:12:44894 os.killpg(int(pid), signal.SIGKILL)
xixuan52c2fba2016-05-21 00:02:48895
xixuan28681fd2016-11-23 19:13:56896 result_dict['detailed_error_msg'] = str(e)
897
898 return json.dumps(result_dict)
xixuan52c2fba2016-05-21 00:02:48899
900 @cherrypy.expose
David Riley6d5fca02017-10-31 17:35:47901 def post_au_status(self, status, **kwargs):
902 """Updates the status of an auto-update task.
903
904 Callers will need to POST to this URL with a body of MIME-type
905 "multipart/form-data".
906 The body should include a single argument, 'status', containing the
907 AU status to record.
908
909 Args:
910 status: The updated status.
911 kwargs:
912 host_name: the hostname of the DUT to auto-update.
913 pid: the background process id of cros-update.
914 """
915 if 'host_name' not in kwargs:
Amin Hassanid4e35392019-10-03 18:02:44916 raise common_util.DevServerHTTPError(http_client.INTERNAL_SERVER_ERROR,
Amin Hassani08e42d22019-06-03 07:31:30917 KEY_ERROR_MSG % 'host_name')
David Riley6d5fca02017-10-31 17:35:47918
919 if 'pid' not in kwargs:
Amin Hassanid4e35392019-10-03 18:02:44920 raise common_util.DevServerHTTPError(http_client.INTERNAL_SERVER_ERROR,
Amin Hassani08e42d22019-06-03 07:31:30921 KEY_ERROR_MSG % 'pid')
David Riley6d5fca02017-10-31 17:35:47922
923 host_name = kwargs['host_name']
924 pid = kwargs['pid']
David Riley3cea2582017-11-25 06:03:01925 status = status.rstrip()
926 _Log('Recording status for %s (%s): %s' % (host_name, pid, status))
David Riley6d5fca02017-10-31 17:35:47927 progress_tracker = cros_update_progress.AUProgress(host_name, pid)
928
David Riley3cea2582017-11-25 06:03:01929 progress_tracker.WriteStatus(status)
David Riley6d5fca02017-10-31 17:35:47930
931 return 'True'
932
933 @cherrypy.expose
xixuan52c2fba2016-05-21 00:02:48934 def handler_cleanup(self, **kwargs):
xixuan3bc974e2016-10-19 00:21:43935 """Clean track status log and temp directory for CrOS auto-update process.
xixuan52c2fba2016-05-21 00:02:48936
937 Args:
938 kwargs:
939 host_name: the hostname of the DUT to auto-update.
940 pid: the background process id of cros-update.
941 """
942 if 'host_name' not in kwargs:
Amin Hassanid4e35392019-10-03 18:02:44943 raise common_util.DevServerHTTPError(http_client.INTERNAL_SERVER_ERROR,
Amin Hassani08e42d22019-06-03 07:31:30944 KEY_ERROR_MSG % 'host_name')
xixuan52c2fba2016-05-21 00:02:48945
946 if 'pid' not in kwargs:
Amin Hassanid4e35392019-10-03 18:02:44947 raise common_util.DevServerHTTPError(http_client.INTERNAL_SERVER_ERROR,
Amin Hassani08e42d22019-06-03 07:31:30948 KEY_ERROR_MSG % 'pid')
xixuan52c2fba2016-05-21 00:02:48949
950 host_name = kwargs['host_name']
951 pid = kwargs['pid']
952 cros_update_progress.DelTrackStatusFile(host_name, pid)
xixuan3bc974e2016-10-19 00:21:43953 cros_update_progress.DelAUTempDirectory(host_name, pid)
xixuan52c2fba2016-05-21 00:02:48954
955 @cherrypy.expose
956 def kill_au_proc(self, **kwargs):
957 """Kill CrOS auto-update process using given process id.
958
959 Args:
960 kwargs:
961 host_name: Kill all the CrOS auto-update process of this host.
962
963 Returns:
964 True if all processes are killed properly.
965 """
966 if 'host_name' not in kwargs:
Amin Hassanid4e35392019-10-03 18:02:44967 raise common_util.DevServerHTTPError(http_client.INTERNAL_SERVER_ERROR,
Amin Hassani08e42d22019-06-03 07:31:30968 KEY_ERROR_MSG % 'host_name')
xixuan52c2fba2016-05-21 00:02:48969
xixuan447ad9d2017-02-28 22:46:20970 cur_pid = kwargs.get('pid')
971
xixuan52c2fba2016-05-21 00:02:48972 host_name = kwargs['host_name']
xixuan3bc974e2016-10-19 00:21:43973 track_log_list = cros_update_progress.GetAllTrackStatusFileByHostName(
974 host_name)
xixuan52c2fba2016-05-21 00:02:48975 for log in track_log_list:
976 # The track log's full path is: path/host_name_pid.log
977 # Use splitext to remove file extension, then parse pid from the
978 # filename.
Congbin Guo3afae6c2019-08-13 23:29:42979 pid = os.path.splitext(os.path.basename(log))[0][len(host_name) + 1:]
xixuan447ad9d2017-02-28 22:46:20980 _clear_process(host_name, pid)
xixuan52c2fba2016-05-21 00:02:48981
xixuan447ad9d2017-02-28 22:46:20982 if cur_pid:
983 _clear_process(host_name, cur_pid)
xixuan52c2fba2016-05-21 00:02:48984
985 return 'True'
986
987 @cherrypy.expose
988 def collect_cros_au_log(self, **kwargs):
989 """Collect CrOS auto-update log.
990
991 Args:
992 kwargs:
993 host_name: the hostname of the DUT to auto-update.
994 pid: the background process id of cros-update.
995
996 Returns:
David Haddock9f459632017-05-11 21:45:46997 A dictionary containing the execute log file and any hostlog files.
xixuan52c2fba2016-05-21 00:02:48998 """
999 if 'host_name' not in kwargs:
Amin Hassanid4e35392019-10-03 18:02:441000 raise common_util.DevServerHTTPError(http_client.INTERNAL_SERVER_ERROR,
Amin Hassani08e42d22019-06-03 07:31:301001 KEY_ERROR_MSG % 'host_name')
xixuan52c2fba2016-05-21 00:02:481002
1003 if 'pid' not in kwargs:
Amin Hassanid4e35392019-10-03 18:02:441004 raise common_util.DevServerHTTPError(http_client.INTERNAL_SERVER_ERROR,
Amin Hassani08e42d22019-06-03 07:31:301005 KEY_ERROR_MSG % 'pid')
xixuan52c2fba2016-05-21 00:02:481006
1007 host_name = kwargs['host_name']
1008 pid = kwargs['pid']
xixuan3bc974e2016-10-19 00:21:431009
1010 # Fetch the execute log recorded by cros_update_progress.
xixuan1bbfaba2016-10-14 00:53:221011 au_log = cros_update_progress.ReadExecuteLogFile(host_name, pid)
1012 cros_update_progress.DelExecuteLogFile(host_name, pid)
David Haddock9f459632017-05-11 21:45:461013 # Fetch the cros_au host_logs if they exist
1014 au_hostlogs = cros_update_progress.ReadAUHostLogFiles(host_name, pid)
1015 return json.dumps({'cros_au_log': au_log, 'host_logs': au_hostlogs})
xixuan1bbfaba2016-10-14 00:53:221016
xixuan52c2fba2016-05-21 00:02:481017 @cherrypy.expose
Dan Shi2f136862016-02-11 23:38:381018 def locate_file(self, **kwargs):
1019 """Get the path to the given file name.
1020
1021 This method looks up the given file name inside specified build artifacts.
1022 One use case is to help caller to locate an apk file inside a build
1023 artifact. The location of the apk file could be different based on the
1024 branch and target.
1025
1026 Args:
1027 file_name: Name of the file to look for.
1028 artifacts: A list of artifact names to search for the file.
1029
1030 Returns:
1031 Path to the file with the given name. It's relative to the folder for the
1032 build, e.g., DATA/priv-app/sl4a/sl4a.apk
Dan Shi2f136862016-02-11 23:38:381033 """
1034 dl, _ = _get_downloader_and_factory(kwargs)
1035 try:
Joe Brennan1691f8e2017-03-15 22:53:361036 file_name = kwargs['file_name']
Dan Shi2f136862016-02-11 23:38:381037 artifacts = kwargs['artifacts']
1038 except KeyError:
Amin Hassanid4e35392019-10-03 18:02:441039 raise DevServerError(
Congbin Guo4132a272019-08-20 19:32:141040 '`file_name` and `artifacts` are required to search '
1041 'for a file in build artifacts.')
Dan Shi2f136862016-02-11 23:38:381042 build_path = dl.GetBuildDir()
1043 for artifact in artifacts:
1044 # Get the unzipped folder of the artifact. If it's not defined in
1045 # ARTIFACT_UNZIP_FOLDER_MAP, assume the files are unzipped to the build
1046 # directory directly.
1047 folder = artifact_info.ARTIFACT_UNZIP_FOLDER_MAP.get(artifact, '')
1048 artifact_path = os.path.join(build_path, folder)
1049 for root, _, filenames in os.walk(artifact_path):
Joe Brennan1691f8e2017-03-15 22:53:361050 if file_name in set([f for f in filenames]):
Dan Shi2f136862016-02-11 23:38:381051 return os.path.relpath(os.path.join(root, file_name), build_path)
Amin Hassanid4e35392019-10-03 18:02:441052 raise DevServerError(
Congbin Guo4132a272019-08-20 19:32:141053 'File `%s` can not be found in artifacts: %s' % (file_name, artifacts))
Dan Shi2f136862016-02-11 23:38:381054
1055 @cherrypy.expose
Simran Basi4baad082013-02-14 21:39:181056 def setup_telemetry(self, **kwargs):
1057 """Extracts and sets up telemetry
1058
1059 This method goes through the telemetry deps packages, and stages them on
1060 the devserver to be used by the drones and the telemetry tests.
1061
1062 Args:
1063 archive_url: Google Storage URL for the build.
1064
1065 Returns:
1066 Path to the source folder for the telemetry codebase once it is staged.
1067 """
Gabe Black3b567202015-09-23 21:07:591068 dl = _get_downloader(kwargs)
Simran Basi4baad082013-02-14 21:39:181069
Gabe Black3b567202015-09-23 21:07:591070 build_path = dl.GetBuildDir()
Simran Basi4baad082013-02-14 21:39:181071 deps_path = os.path.join(build_path, 'autotest/packages')
1072 telemetry_path = os.path.join(build_path, TELEMETRY_FOLDER)
1073 src_folder = os.path.join(telemetry_path, 'src')
1074
1075 with self._telemetry_lock_dict.lock(telemetry_path):
1076 if os.path.exists(src_folder):
1077 # Telemetry is already fully stage return
1078 return src_folder
1079
1080 common_util.MkDirP(telemetry_path)
1081
1082 # Copy over the required deps tar balls to the telemetry directory.
1083 for dep in TELEMETRY_DEPS:
1084 dep_path = os.path.join(deps_path, dep)
Simran Basi0d078682013-03-22 23:40:041085 if not os.path.exists(dep_path):
1086 # This dep does not exist (could be new), do not extract it.
1087 continue
Simran Basi4baad082013-02-14 21:39:181088 try:
1089 common_util.ExtractTarball(dep_path, telemetry_path)
1090 except common_util.CommonUtilError as e:
1091 shutil.rmtree(telemetry_path)
Amin Hassanid4e35392019-10-03 18:02:441092 raise DevServerError(str(e))
Simran Basi4baad082013-02-14 21:39:181093
1094 # By default all the tarballs extract to test_src but some parts of
1095 # the telemetry code specifically hardcoded to exist inside of 'src'.
1096 test_src = os.path.join(telemetry_path, 'test_src')
1097 try:
1098 shutil.move(test_src, src_folder)
1099 except shutil.Error:
1100 # This can occur if src_folder already exists. Remove and retry move.
1101 shutil.rmtree(src_folder)
Amin Hassanid4e35392019-10-03 18:02:441102 raise DevServerError(
Gabe Black3b567202015-09-23 21:07:591103 'Failure in telemetry setup for build %s. Appears that the '
1104 'test_src to src move failed.' % dl.GetBuild())
Simran Basi4baad082013-02-14 21:39:181105
1106 return src_folder
1107
1108 @cherrypy.expose
Chris Sosa76e44b92013-01-31 20:11:381109 def symbolicate_dump(self, minidump, **kwargs):
Chris Masone816e38c2012-05-02 19:22:361110 """Symbolicates a minidump using pre-downloaded symbols, returns it.
1111
1112 Callers will need to POST to this URL with a body of MIME-type
1113 "multipart/form-data".
1114 The body should include a single argument, 'minidump', containing the
1115 binary-formatted minidump to symbolicate.
1116
Chris Masone816e38c2012-05-02 19:22:361117 Args:
Chris Sosa76e44b92013-01-31 20:11:381118 archive_url: Google Storage URL for the build.
Chris Masone816e38c2012-05-02 19:22:361119 minidump: The binary minidump file to symbolicate.
1120 """
Chris Sosa76e44b92013-01-31 20:11:381121 # Ensure the symbols have been staged.
Dan Shif08fe492016-10-04 21:39:251122 # Try debug.tar.xz first, then debug.tgz
1123 for artifact in (artifact_info.SYMBOLS_ONLY, artifact_info.SYMBOLS):
1124 kwargs['artifacts'] = artifact
1125 dl = _get_downloader(kwargs)
1126
1127 try:
1128 if self.stage(**kwargs) == 'Success':
1129 break
1130 except build_artifact.ArtifactDownloadError:
1131 continue
1132 else:
Amin Hassanid4e35392019-10-03 18:02:441133 raise DevServerError(
Congbin Guo4132a272019-08-20 19:32:141134 'Failed to stage symbols for %s' % dl.DescribeSource())
Chris Sosa76e44b92013-01-31 20:11:381135
Chris Masone816e38c2012-05-02 19:22:361136 to_return = ''
1137 with tempfile.NamedTemporaryFile() as local:
1138 while True:
1139 data = minidump.file.read(8192)
1140 if not data:
1141 break
1142 local.write(data)
Chris Sosa76e44b92013-01-31 20:11:381143
Chris Masone816e38c2012-05-02 19:22:361144 local.flush()
Chris Sosa76e44b92013-01-31 20:11:381145
Gabe Black3b567202015-09-23 21:07:591146 symbols_directory = os.path.join(dl.GetBuildDir(), 'debug', 'breakpad')
Chris Sosa76e44b92013-01-31 20:11:381147
xixuanab744382017-04-27 17:41:271148 # The location of minidump_stackwalk is defined in chromeos-admin.
Chris Sosa76e44b92013-01-31 20:11:381149 stackwalk = subprocess.Popen(
xixuanab744382017-04-27 17:41:271150 ['/usr/local/bin/minidump_stackwalk', local.name, symbols_directory],
Chris Sosa76e44b92013-01-31 20:11:381151 stdout=subprocess.PIPE, stderr=subprocess.PIPE)
1152
Chris Masone816e38c2012-05-02 19:22:361153 to_return, error_text = stackwalk.communicate()
1154 if stackwalk.returncode != 0:
Amin Hassanid4e35392019-10-03 18:02:441155 raise DevServerError(
Congbin Guo4132a272019-08-20 19:32:141156 "Can't generate stack trace: %s (rc=%d)" % (error_text,
1157 stackwalk.returncode))
Chris Masone816e38c2012-05-02 19:22:361158
1159 return to_return
1160
1161 @cherrypy.expose
Don Garrettf84631a2014-01-08 02:21:261162 def latestbuild(self, **kwargs):
Scott Zawalski16954532012-03-20 19:31:361163 """Return a string representing the latest build for a given target.
1164
1165 Args:
1166 target: The build target, typically a combination of the board and the
1167 type of build e.g. x86-mario-release.
1168 milestone: The milestone to filter builds on. E.g. R16. Optional, if not
1169 provided the latest RXX build will be returned.
Don Garrettf84631a2014-01-08 02:21:261170
Scott Zawalski16954532012-03-20 19:31:361171 Returns:
1172 A string representation of the latest build if one exists, i.e.
1173 R19-1993.0.0-a1-b1480.
1174 An empty string if no latest could be found.
1175 """
Don Garrettf84631a2014-01-08 02:21:261176 if not kwargs:
Scott Zawalski16954532012-03-20 19:31:361177 return _PrintDocStringAsHTML(self.latestbuild)
1178
Don Garrettf84631a2014-01-08 02:21:261179 if 'target' not in kwargs:
Amin Hassanid4e35392019-10-03 18:02:441180 raise common_util.DevServerHTTPError(http_client.INTERNAL_SERVER_ERROR,
Amin Hassani08e42d22019-06-03 07:31:301181 'Error: target= is required!')
Dan Shi61305df2015-10-26 23:52:351182
1183 if _is_android_build_request(kwargs):
1184 branch = kwargs.get('branch', None)
1185 target = kwargs.get('target', None)
1186 if not target or not branch:
Amin Hassanid4e35392019-10-03 18:02:441187 raise DevServerError('Both target and branch must be specified to query'
1188 ' for the latest Android build.')
Dan Shi61305df2015-10-26 23:52:351189 return android_build.BuildAccessor.GetLatestBuildID(target, branch)
1190
Scott Zawalski16954532012-03-20 19:31:361191 try:
Gilad Arnoldc65330c2012-09-20 22:17:481192 return common_util.GetLatestBuildVersion(
Don Garrettf84631a2014-01-08 02:21:261193 updater.static_dir, kwargs['target'],
1194 milestone=kwargs.get('milestone'))
Gilad Arnold17fe03d2012-10-02 17:05:011195 except common_util.CommonUtilError as errmsg:
Amin Hassanid4e35392019-10-03 18:02:441196 raise common_util.DevServerHTTPError(http_client.INTERNAL_SERVER_ERROR,
Amin Hassani08e42d22019-06-03 07:31:301197 str(errmsg))
Scott Zawalski16954532012-03-20 19:31:361198
1199 @cherrypy.expose
xixuan7efd0002016-04-14 22:34:011200 def list_suite_controls(self, **kwargs):
1201 """Return a list of contents of all known control files.
1202
1203 Example URL:
1204 To List all control files' content:
1205 http://dev-server/list_suite_controls?suite_name=bvt&
1206 build=daisy_spring-release/R29-4279.0.0
1207
1208 Args:
1209 build: The build i.e. x86-alex-release/R18-1514.0.0-a1-b1450.
1210 suite_name: List the control files belonging to that suite.
1211
1212 Returns:
Dan Shia1cd6522016-04-18 23:07:211213 A dictionary of all control files's path to its content for given suite.
xixuan7efd0002016-04-14 22:34:011214 """
1215 if not kwargs:
1216 return _PrintDocStringAsHTML(self.controlfiles)
1217
1218 if 'build' not in kwargs:
Amin Hassanid4e35392019-10-03 18:02:441219 raise common_util.DevServerHTTPError(http_client.INTERNAL_SERVER_ERROR,
Amin Hassani08e42d22019-06-03 07:31:301220 'Error: build= is required!')
xixuan7efd0002016-04-14 22:34:011221
1222 if 'suite_name' not in kwargs:
Amin Hassanid4e35392019-10-03 18:02:441223 raise common_util.DevServerHTTPError(http_client.INTERNAL_SERVER_ERROR,
Dan Shia1cd6522016-04-18 23:07:211224 'Error: suite_name= is required!')
xixuan7efd0002016-04-14 22:34:011225
1226 control_file_list = [
1227 line.rstrip() for line in common_util.GetControlFileListForSuite(
1228 updater.static_dir, kwargs['build'],
1229 kwargs['suite_name']).splitlines()]
1230
Dan Shia1cd6522016-04-18 23:07:211231 control_file_content_dict = {}
xixuan7efd0002016-04-14 22:34:011232 for control_path in control_file_list:
Dan Shia1cd6522016-04-18 23:07:211233 control_file_content_dict[control_path] = (common_util.GetControlFile(
xixuan7efd0002016-04-14 22:34:011234 updater.static_dir, kwargs['build'], control_path))
1235
Dan Shia1cd6522016-04-18 23:07:211236 return json.dumps(control_file_content_dict)
xixuan7efd0002016-04-14 22:34:011237
1238 @cherrypy.expose
Don Garrettf84631a2014-01-08 02:21:261239 def controlfiles(self, **kwargs):
Scott Zawalski4647ce62012-01-03 22:17:281240 """Return a control file or a list of all known control files.
1241
1242 Example URL:
1243 To List all control files:
beepsbd337242013-07-10 05:44:061244 http://dev-server/controlfiles?suite_name=&build=daisy_spring-release/R29-4279.0.0
1245 To List all control files for, say, the bvt suite:
1246 http://dev-server/controlfiles?suite_name=bvt&build=daisy_spring-release/R29-4279.0.0
Scott Zawalski4647ce62012-01-03 22:17:281247 To return the contents of a path:
Scott Zawalski84a39c92012-01-13 20:12:421248 http://dev-server/controlfiles?board=x86-alex-release&build=R18-1514.0.0&control_path=client/sleeptest/control
Scott Zawalski4647ce62012-01-03 22:17:281249
1250 Args:
Scott Zawalski84a39c92012-01-13 20:12:421251 build: The build i.e. x86-alex-release/R18-1514.0.0-a1-b1450.
Scott Zawalski4647ce62012-01-03 22:17:281252 control_path: If you want the contents of a control file set this
1253 to the path. E.g. client/site_tests/sleeptest/control
1254 Optional, if not provided return a list of control files is returned.
beepsbd337242013-07-10 05:44:061255 suite_name: If control_path is not specified but a suite_name is
1256 specified, list the control files belonging to that suite instead of
1257 all control files. The empty string for suite_name will list all control
1258 files for the build.
Don Garrettf84631a2014-01-08 02:21:261259
Scott Zawalski4647ce62012-01-03 22:17:281260 Returns:
1261 Contents of a control file if control_path is provided.
1262 A list of control files if no control_path is provided.
1263 """
Don Garrettf84631a2014-01-08 02:21:261264 if not kwargs:
Scott Zawalski4647ce62012-01-03 22:17:281265 return _PrintDocStringAsHTML(self.controlfiles)
1266
Don Garrettf84631a2014-01-08 02:21:261267 if 'build' not in kwargs:
Amin Hassanid4e35392019-10-03 18:02:441268 raise common_util.DevServerHTTPError(http_client.INTERNAL_SERVER_ERROR,
Amin Hassani08e42d22019-06-03 07:31:301269 'Error: build= is required!')
Scott Zawalski4647ce62012-01-03 22:17:281270
Don Garrettf84631a2014-01-08 02:21:261271 if 'control_path' not in kwargs:
1272 if 'suite_name' in kwargs and kwargs['suite_name']:
beepsbd337242013-07-10 05:44:061273 return common_util.GetControlFileListForSuite(
Don Garrettf84631a2014-01-08 02:21:261274 updater.static_dir, kwargs['build'], kwargs['suite_name'])
beepsbd337242013-07-10 05:44:061275 else:
1276 return common_util.GetControlFileList(
Don Garrettf84631a2014-01-08 02:21:261277 updater.static_dir, kwargs['build'])
Scott Zawalski4647ce62012-01-03 22:17:281278 else:
Gilad Arnoldc65330c2012-09-20 22:17:481279 return common_util.GetControlFile(
Don Garrettf84631a2014-01-08 02:21:261280 updater.static_dir, kwargs['build'], kwargs['control_path'])
Frank Farzan40160872011-12-13 02:39:181281
1282 @cherrypy.expose
Simran Basi99e63c02014-05-20 17:39:521283 def xbuddy_translate(self, *args, **kwargs):
Yu-Ju Hong1bdb7a92014-04-10 23:02:111284 """Translates an xBuddy path to a real path to artifact if it exists.
1285
1286 Args:
Simran Basi99e63c02014-05-20 17:39:521287 args: An xbuddy path in the form of {local|remote}/build_id/artifact.
1288 Local searches the devserver's static directory. Remote searches a
1289 Google Storage image archive.
1290
1291 Kwargs:
1292 image_dir: Google Storage image archive to search in if requesting a
1293 remote artifact. If none uses the default bucket.
Yu-Ju Hong1bdb7a92014-04-10 23:02:111294
1295 Returns:
Simran Basi99e63c02014-05-20 17:39:521296 String in the format of build_id/artifact as stored on the local server
1297 or in Google Storage.
Yu-Ju Hong1bdb7a92014-04-10 23:02:111298 """
Simran Basi99e63c02014-05-20 17:39:521299 build_id, filename = self._xbuddy.Translate(
Gabe Black3b567202015-09-23 21:07:591300 args, image_dir=kwargs.get('image_dir'))
Yu-Ju Hong1bdb7a92014-04-10 23:02:111301 response = os.path.join(build_id, filename)
1302 _Log('Path translation requested, returning: %s', response)
1303 return response
1304
1305 @cherrypy.expose
joycheneaf4cfc2013-07-02 15:38:571306 def xbuddy(self, *args, **kwargs):
1307 """The full xBuddy call, returns resource specified by path_parts.
joychen3cb228e2013-06-12 19:13:131308
1309 Args:
joycheneaf4cfc2013-07-02 15:38:571310 path_parts: the path following xbuddy/ in the call url is split into the
joychen121fc9b2013-08-02 21:30:301311 components of the path. The path can be understood as
1312 "{local|remote}/build_id/artifact" where build_id is composed of
1313 "board/version."
joycheneaf4cfc2013-07-02 15:38:571314
joychen121fc9b2013-08-02 21:30:301315 The first path element is optional, and can be "remote" or "local"
1316 If local (the default), devserver will not attempt to access Google
1317 Storage, and will only search the static directory for the files.
1318 If remote, devserver will try to obtain the artifact off GS if it's
1319 not found locally.
1320 The board is the familiar board name, optionally suffixed.
1321 The version can be the google storage version number, and may also be
1322 any of a number of xBuddy defined version aliases that will be
1323 translated into the latest built image that fits the description.
1324 Defaults to latest.
1325 The artifact is one of a number of image or artifact aliases used by
1326 xbuddy, defined in xbuddy:ALIASES. Defaults to test.
joycheneaf4cfc2013-07-02 15:38:571327
1328 Kwargs:
Yu-Ju Hong51495eb2013-12-13 01:08:431329 for_update: {true|false}
Amin Hassanie9ffb862019-09-26 00:10:401330 if true, prepares the update payloads for the image,
Yu-Ju Hong51495eb2013-12-13 01:08:431331 and returns the update uri to pass to the
1332 update_engine_client.
joychen3cb228e2013-06-12 19:13:131333 return_dir: {true|false}
1334 if set to true, returns the url to the update.gz
Yu-Ju Hong51495eb2013-12-13 01:08:431335 relative_path: {true|false}
1336 if set to true, returns the relative path to the payload
1337 directory from static_dir.
joychen3cb228e2013-06-12 19:13:131338 Example URL:
joycheneaf4cfc2013-07-02 15:38:571339 http://host:port/xbuddy/x86-generic/R26-4000.0.0/test
joychen3cb228e2013-06-12 19:13:131340 or
joycheneaf4cfc2013-07-02 15:38:571341 http://host:port/xbuddy/x86-generic/R26-4000.0.0/test?return_dir=true
joychen3cb228e2013-06-12 19:13:131342
1343 Returns:
Yu-Ju Hong51495eb2013-12-13 01:08:431344 If |for_update|, returns a redirect to the image or update file
1345 on the devserver. E.g.,
1346 http://host:port/static/archive/x86-generic-release/R26-4000.0.0/
1347 chromium-test-image.bin
1348 If |return_dir|, return a uri to the folder where the artifact is. E.g.,
1349 http://host:port/static/x86-generic-release/R26-4000.0.0/
1350 If |relative_path| is true, return a relative path the folder where the
1351 payloads are. E.g.,
1352 archive/x86-generic-release/R26-4000.0.0
joychen3cb228e2013-06-12 19:13:131353 """
Chris Sosa75490802013-10-01 00:21:451354 boolean_string = kwargs.get('for_update')
1355 for_update = xbuddy.XBuddy.ParseBoolean(boolean_string)
Yu-Ju Hong51495eb2013-12-13 01:08:431356 boolean_string = kwargs.get('return_dir')
1357 return_dir = xbuddy.XBuddy.ParseBoolean(boolean_string)
1358 boolean_string = kwargs.get('relative_path')
1359 relative_path = xbuddy.XBuddy.ParseBoolean(boolean_string)
joychen121fc9b2013-08-02 21:30:301360
Yu-Ju Hong51495eb2013-12-13 01:08:431361 if return_dir and relative_path:
Chris Sosa4b951602014-04-10 03:26:071362 raise common_util.DevServerHTTPError(
Amin Hassanid4e35392019-10-03 18:02:441363 http_client.INTERNAL_SERVER_ERROR,
Amin Hassani08e42d22019-06-03 07:31:301364 'Cannot specify both return_dir and relative_path')
Chris Sosa75490802013-10-01 00:21:451365
1366 # For updates, we optimize downloading of test images.
1367 file_name = None
1368 build_id = None
1369 if for_update:
1370 try:
Yu-Ju Hong1bdb7a92014-04-10 23:02:111371 build_id = self._xbuddy.StageTestArtifactsForUpdate(args)
Chris Sosa75490802013-10-01 00:21:451372 except build_artifact.ArtifactDownloadError:
1373 build_id = None
1374
1375 if not build_id:
1376 build_id, file_name = self._xbuddy.Get(args)
1377
Yu-Ju Hong51495eb2013-12-13 01:08:431378 if for_update:
Amin Hassanie9ffb862019-09-26 00:10:401379 _Log('Payloads requested.')
Yu-Ju Hong51495eb2013-12-13 01:08:431380 # Forces payload to be in cache and symlinked into build_id dir.
Amin Hassanie9ffb862019-09-26 00:10:401381 updater.GetUpdateForLabel(build_id)
Yu-Ju Hong51495eb2013-12-13 01:08:431382
1383 response = None
1384 if return_dir:
1385 response = os.path.join(cherrypy.request.base, 'static', build_id)
1386 _Log('Directory requested, returning: %s', response)
1387 elif relative_path:
1388 response = build_id
1389 _Log('Relative path requested, returning: %s', response)
1390 elif for_update:
1391 response = os.path.join(cherrypy.request.base, 'update', build_id)
1392 _Log('Update URI requested, returning: %s', response)
joychen3cb228e2013-06-12 19:13:131393 else:
Yu-Ju Hong51495eb2013-12-13 01:08:431394 # Redirect to download the payload if no kwargs are set.
joychen121fc9b2013-08-02 21:30:301395 build_id = '/' + os.path.join('static', build_id, file_name)
Yu-Ju Hong51495eb2013-12-13 01:08:431396 _Log('Payload requested, returning: %s', build_id)
joychen121fc9b2013-08-02 21:30:301397 raise cherrypy.HTTPRedirect(build_id, 302)
joychen3cb228e2013-06-12 19:13:131398
Yu-Ju Hong51495eb2013-12-13 01:08:431399 return response
1400
joychen3cb228e2013-06-12 19:13:131401 @cherrypy.expose
1402 def xbuddy_list(self):
1403 """Lists the currently available images & time since last access.
1404
Gilad Arnold452fd272014-02-04 19:09:281405 Returns:
1406 A string representation of a list of tuples [(build_id, time since last
1407 access),...]
joychen3cb228e2013-06-12 19:13:131408 """
1409 return self._xbuddy.List()
1410
1411 @cherrypy.expose
1412 def xbuddy_capacity(self):
Gilad Arnold452fd272014-02-04 19:09:281413 """Returns the number of images cached by xBuddy."""
joychen3cb228e2013-06-12 19:13:131414 return self._xbuddy.Capacity()
1415
1416 @cherrypy.expose
Chris Sosa7c931362010-10-12 02:49:011417 def index(self):
Gilad Arnoldf8f769f2012-09-24 15:43:011418 """Presents a welcome message and documentation links."""
Congbin Guo6bc32182019-08-21 00:54:301419 html_template = (
1420 'Welcome to the Dev Server!<br>\n'
1421 '<br>\n'
1422 'Here are the available methods, click for documentation:<br>\n'
1423 '<br>\n'
1424 '%s')
1425
1426 exposed_methods = []
1427 for app in cherrypy.tree.apps.values():
1428 exposed_methods += _FindExposedMethods(
1429 app.root, app.script_name.lstrip('/'),
1430 unlisted=self._UNLISTED_METHODS)
1431
1432 return html_template % '<br>\n'.join(
1433 ['<a href=doc/%s>%s</a>' % (name, name)
1434 for name in sorted(exposed_methods)])
Gilad Arnoldf8f769f2012-09-24 15:43:011435
1436 @cherrypy.expose
1437 def doc(self, *args):
1438 """Shows the documentation for available methods / URLs.
1439
Amin Hassani08e42d22019-06-03 07:31:301440 Examples:
Gilad Arnoldf8f769f2012-09-24 15:43:011441 http://myhost/doc/update
1442 """
Gilad Arnoldd5ebaaa2012-10-02 18:52:381443 name = '/'.join(args)
Congbin Guo6bc32182019-08-21 00:54:301444 method = _GetExposedMethod(name)
Gilad Arnoldf8f769f2012-09-24 15:43:011445 if not method:
Amin Hassanid4e35392019-10-03 18:02:441446 raise DevServerError("No exposed method named `%s'" % name)
Gilad Arnoldf8f769f2012-09-24 15:43:011447 if not method.__doc__:
Amin Hassanid4e35392019-10-03 18:02:441448 raise DevServerError("No documentation for exposed method `%s'" % name)
Gilad Arnoldf8f769f2012-09-24 15:43:011449 return '<pre>\n%s</pre>' % method.__doc__
Chris Sosa7c931362010-10-12 02:49:011450
Dale Curtisc9aaf3a2011-08-09 22:47:401451 @cherrypy.expose
Chris Sosa7c931362010-10-12 02:49:011452 def update(self, *args):
Gilad Arnoldf8f769f2012-09-24 15:43:011453 """Handles an update check from a Chrome OS client.
1454
1455 The HTTP request should contain the standard Omaha-style XML blob. The URL
1456 line may contain an additional intermediate path to the update payload.
1457
joychen121fc9b2013-08-02 21:30:301458 This request can be handled in one of 4 ways, depending on the devsever
1459 settings and intermediate path.
joychenb0dfe552013-07-30 17:02:061460
Amin Hassanie9ffb862019-09-26 00:10:401461 1. No intermediate path. DEPRECATED
joychen121fc9b2013-08-02 21:30:301462
1463 2. Path explicitly invokes XBuddy
1464 If there is a path given, it can explicitly invoke xbuddy by prefixing it
1465 with 'xbuddy'. This path is then used to acquire an image binary for the
1466 devserver to generate an update payload from. Devserver then serves this
1467 payload.
1468
1469 3. Path is left for the devserver to interpret.
1470 If the path given doesn't explicitly invoke xbuddy, devserver will attempt
1471 to generate a payload from the test image in that directory and serve it.
1472
joychen121fc9b2013-08-02 21:30:301473 Examples:
joychen121fc9b2013-08-02 21:30:301474 2. Explicitly invoke xbuddy
1475 update_engine_client --omaha_url=
1476 http://myhost/update/xbuddy/remote/board/version/dev
1477 This would go to GS to download the dev image for the board, from which
1478 the devserver would generate a payload to serve.
1479
1480 3. Give a path for devserver to interpret
1481 update_engine_client --omaha_url=http://myhost/update/some/random/path
1482 This would attempt, in order to:
1483 a) Generate an update from a test image binary if found in
1484 static_dir/some/random/path.
1485 b) Serve an update payload found in static_dir/some/random/path.
1486 c) Hope that some/random/path takes the form "board/version" and
1487 and attempt to download an update payload for that board/version
1488 from GS.
Gilad Arnoldf8f769f2012-09-24 15:43:011489 """
joychen121fc9b2013-08-02 21:30:301490 label = '/'.join(args)
Gilad Arnold286a0062012-01-12 21:47:021491 body_length = int(cherrypy.request.headers.get('Content-Length', 0))
Chris Sosa7c931362010-10-12 02:49:011492 data = cherrypy.request.rfile.read(body_length)
Chris Sosa7c931362010-10-12 02:49:011493
joychen121fc9b2013-08-02 21:30:301494 return updater.HandleUpdatePing(data, label)
Chris Sosa0356d3b2010-09-16 22:46:221495
Dan Shif5ce2de2013-04-25 23:06:321496
Chris Sosadbc20082012-12-10 21:39:111497def _CleanCache(cache_dir, wipe):
1498 """Wipes any excess cached items in the cache_dir.
1499
1500 Args:
1501 cache_dir: the directory we are wiping from.
1502 wipe: If True, wipe all the contents -- not just the excess.
1503 """
1504 if wipe:
1505 # Clear the cache and exit on error.
1506 cmd = 'rm -rf %s/*' % cache_dir
1507 if os.system(cmd) != 0:
1508 _Log('Failed to clear the cache with %s' % cmd)
1509 sys.exit(1)
1510 else:
1511 # Clear all but the last N cached updates
1512 cmd = ('cd %s; ls -tr | head --lines=-%d | xargs rm -rf' %
1513 (cache_dir, CACHED_ENTRIES))
1514 if os.system(cmd) != 0:
1515 _Log('Failed to clean up old delta cache files with %s' % cmd)
1516 sys.exit(1)
1517
1518
Chris Sosa3ae4dc12013-03-29 18:47:001519def _AddTestingOptions(parser):
1520 group = optparse.OptionGroup(
1521 parser, 'Advanced Testing Options', 'These are used by test scripts and '
1522 'developers writing integration tests utilizing the devserver. They are '
1523 'not intended to be really used outside the scope of someone '
1524 'knowledgable about the test.')
1525 group.add_option('--exit',
1526 action='store_true',
Amin Hassanie9ffb862019-09-26 00:10:401527 help='do not start the server (yet clear cache)')
Chris Sosa3ae4dc12013-03-29 18:47:001528 group.add_option('--host_log',
1529 action='store_true', default=False,
1530 help='record history of host update events (/api/hostlog)')
1531 group.add_option('--max_updates',
Gabe Black3b567202015-09-23 21:07:591532 metavar='NUM', default=-1, type='int',
Chris Sosa3ae4dc12013-03-29 18:47:001533 help='maximum number of update checks handled positively '
1534 '(default: unlimited)')
Chris Sosa3ae4dc12013-03-29 18:47:001535 group.add_option('--proxy_port',
1536 metavar='PORT', default=None, type='int',
1537 help='port to have the client connect to -- basically the '
1538 'devserver lies to the update to tell it to get the payload '
1539 'from a different port that will proxy the request back to '
1540 'the devserver. The proxy must be managed outside the '
1541 'devserver.')
Chris Sosa3ae4dc12013-03-29 18:47:001542 parser.add_option_group(group)
1543
1544
1545def _AddUpdateOptions(parser):
1546 group = optparse.OptionGroup(
1547 parser, 'Autoupdate Options', 'These options can be used to change '
Amin Hassanie9ffb862019-09-26 00:10:401548 'how the devserver serve update payloads. Please '
Chris Sosa3ae4dc12013-03-29 18:47:001549 'note that all of these option affect how a payload is generated and so '
1550 'do not work in archive-only mode.')
Chris Sosa3ae4dc12013-03-29 18:47:001551 group.add_option('--critical_update',
1552 action='store_true', default=False,
1553 help='Present update payload as critical')
Chris Sosa3ae4dc12013-03-29 18:47:001554 group.add_option('--payload',
1555 metavar='PATH',
1556 help='use the update payload from specified directory '
1557 '(update.gz).')
Chris Sosa3ae4dc12013-03-29 18:47:001558 parser.add_option_group(group)
1559
1560
1561def _AddProductionOptions(parser):
1562 group = optparse.OptionGroup(
1563 parser, 'Advanced Server Options', 'These options can be used to changed '
1564 'for advanced server behavior.')
Chris Sosa3ae4dc12013-03-29 18:47:001565 group.add_option('--clear_cache',
1566 action='store_true', default=False,
1567 help='At startup, removes all cached entries from the'
Amin Hassanid4e35392019-10-03 18:02:441568 "devserver's cache.")
Chris Sosa3ae4dc12013-03-29 18:47:001569 group.add_option('--logfile',
1570 metavar='PATH',
1571 help='log output to this file instead of stdout')
Chris Sosa855b8932013-08-21 20:24:551572 group.add_option('--pidfile',
1573 metavar='PATH',
1574 help='path to output a pid file for the server.')
Gilad Arnold11fbef42014-02-10 19:04:131575 group.add_option('--portfile',
1576 metavar='PATH',
1577 help='path to output the port number being served on.')
Chris Sosa3ae4dc12013-03-29 18:47:001578 group.add_option('--production',
1579 action='store_true', default=False,
1580 help='have the devserver use production values when '
1581 'starting up. This includes using more threads and '
1582 'performing less logging.')
1583 parser.add_option_group(group)
1584
1585
Paul Hobbsef4e0702016-06-28 00:01:421586def MakeLogHandler(logfile):
J. Richard Barnette3d977b82013-04-23 18:05:191587 """Create a LogHandler instance used to log all messages."""
1588 hdlr_cls = handlers.TimedRotatingFileHandler
1589 hdlr = hdlr_cls(logfile, when=_LOG_ROTATION_TIME,
xixuan3d48bff2017-01-31 03:00:091590 interval=_LOG_ROTATION_INTERVAL,
J. Richard Barnette3d977b82013-04-23 18:05:191591 backupCount=_LOG_ROTATION_BACKUP)
Chris Sosa855b8932013-08-21 20:24:551592 hdlr.setFormatter(cplogging.logfmt)
J. Richard Barnette3d977b82013-04-23 18:05:191593 return hdlr
1594
1595
Chris Sosacde6bf42012-06-01 01:36:391596def main():
Chris Sosa3ae4dc12013-03-29 18:47:001597 usage = '\n\n'.join(['usage: %prog [options]', __doc__])
Gilad Arnold286a0062012-01-12 21:47:021598 parser = optparse.OptionParser(usage=usage)
joychened64b222013-06-21 23:39:341599
1600 # get directory that the devserver is run from
1601 devserver_dir = os.path.dirname(os.path.abspath(sys.argv[0]))
joychen84d13772013-08-06 16:17:231602 default_static_dir = '%s/static' % devserver_dir
joychened64b222013-06-21 23:39:341603 parser.add_option('--static_dir',
Gilad Arnold9714d9b2012-10-04 17:09:421604 metavar='PATH',
joychen84d13772013-08-06 16:17:231605 default=default_static_dir,
joychened64b222013-06-21 23:39:341606 help='writable static directory')
Gilad Arnold9714d9b2012-10-04 17:09:421607 parser.add_option('--port',
1608 default=8080, type='int',
Gilad Arnoldaf696d12014-02-14 21:13:281609 help=('port for the dev server to use; if zero, binds to '
1610 'an arbitrary available port (default: 8080)'))
Gilad Arnold9714d9b2012-10-04 17:09:421611 parser.add_option('-t', '--test_image',
1612 action='store_true',
joychen121fc9b2013-08-02 21:30:301613 help='Deprecated.')
joychen5260b9a2013-07-16 21:48:011614 parser.add_option('-x', '--xbuddy_manage_builds',
1615 action='store_true',
1616 default=False,
1617 help='If set, allow xbuddy to manage images in'
1618 'build/images.')
Dan Shi72b16132015-10-08 19:10:331619 parser.add_option('-a', '--android_build_credential',
1620 default=None,
1621 help='Path to a json file which contains the credential '
1622 'needed to access Android builds.')
Chris Sosa3ae4dc12013-03-29 18:47:001623 _AddProductionOptions(parser)
1624 _AddUpdateOptions(parser)
1625 _AddTestingOptions(parser)
Chris Sosa7c931362010-10-12 02:49:011626 (options, _) = parser.parse_args()
[email protected]21a5ca32009-11-04 18:23:231627
J. Richard Barnette3d977b82013-04-23 18:05:191628 # Handle options that must be set globally in cherrypy. Do this
1629 # work up front, because calls to _Log() below depend on this
1630 # initialization.
1631 if options.production:
1632 cherrypy.config.update({'environment': 'production'})
1633 if not options.logfile:
1634 cherrypy.config.update({'log.screen': True})
1635 else:
1636 cherrypy.config.update({'log.error_file': '',
1637 'log.access_file': ''})
Paul Hobbsef4e0702016-06-28 00:01:421638 hdlr = MakeLogHandler(options.logfile)
J. Richard Barnette3d977b82013-04-23 18:05:191639 # Pylint can't seem to process these two calls properly
1640 # pylint: disable=E1101
1641 cherrypy.log.access_log.addHandler(hdlr)
1642 cherrypy.log.error_log.addHandler(hdlr)
1643 # pylint: enable=E1101
1644
joychened64b222013-06-21 23:39:341645 # set static_dir, from which everything will be served
joychen84d13772013-08-06 16:17:231646 options.static_dir = os.path.realpath(options.static_dir)
Chris Sosa0356d3b2010-09-16 22:46:221647
joychened64b222013-06-21 23:39:341648 cache_dir = os.path.join(options.static_dir, 'cache')
J. Richard Barnette3d977b82013-04-23 18:05:191649 # If our devserver is only supposed to serve payloads, we shouldn't be
1650 # mucking with the cache at all. If the devserver hadn't previously
1651 # generated a cache and is expected, the caller is using it wrong.
joychen7c2054a2013-07-25 18:14:071652 if os.path.exists(cache_dir):
Chris Sosadbc20082012-12-10 21:39:111653 _CleanCache(cache_dir, options.clear_cache)
Chris Sosa6b8c3742011-01-31 20:12:171654 else:
1655 os.makedirs(cache_dir)
Don Garrettf90edf02010-11-17 01:36:141656
Chris Sosadbc20082012-12-10 21:39:111657 _Log('Using cache directory %s' % cache_dir)
joychened64b222013-06-21 23:39:341658 _Log('Serving from %s' % options.static_dir)
[email protected]21a5ca32009-11-04 18:23:231659
Amin Hassanie9ffb862019-09-26 00:10:401660 _xbuddy = xbuddy.XBuddy(manage_builds=options.xbuddy_manage_builds,
joychen121fc9b2013-08-02 21:30:301661 static_dir=options.static_dir)
Chris Sosa75490802013-10-01 00:21:451662 if options.clear_cache and options.xbuddy_manage_builds:
1663 _xbuddy.CleanCache()
joychen121fc9b2013-08-02 21:30:301664
Chris Sosa6a3697f2013-01-30 00:44:431665 # We allow global use here to share with cherrypy classes.
1666 # pylint: disable=W0603
Chris Sosacde6bf42012-06-01 01:36:391667 global updater
Andrew de los Reyes52620802010-04-12 20:40:071668 updater = autoupdate.Autoupdate(
joychen121fc9b2013-08-02 21:30:301669 _xbuddy,
joychened64b222013-06-21 23:39:341670 static_dir=options.static_dir,
Gilad Arnold0c9c8602012-10-03 06:58:581671 payload_path=options.payload,
Don Garrett0ad09372010-12-07 00:20:301672 proxy_port=options.proxy_port,
Satoru Takabayashid733cbe2011-11-15 17:36:321673 critical_update=options.critical_update,
Gilad Arnolda564b4b2012-10-04 17:32:441674 max_updates=options.max_updates,
Gilad Arnold8318eac2012-10-04 19:52:231675 host_log=options.host_log,
Chris Sosa0f1ec842011-02-15 00:33:221676 )
Chris Sosa7c931362010-10-12 02:49:011677
J. Richard Barnette3d977b82013-04-23 18:05:191678 if options.exit:
1679 return
Chris Sosa2f1c41e2012-07-10 21:32:331680
joychen3cb228e2013-06-12 19:13:131681 dev_server = DevServerRoot(_xbuddy)
Congbin Guo3afae6c2019-08-13 23:29:421682 health_checker_app = health_checker.Root(dev_server, options.static_dir)
joychen3cb228e2013-06-12 19:13:131683
Amin Hassanic5af4262019-11-13 21:37:201684 # Patch CherryPy to support binding to any available port (--port=0) only for
1685 # cherrypy versions smaller or equal to 3.2.2.
1686 #
1687 # TODO(crbug/1006305): Remove this once we have deprecated omaha_devserver.py
1688 # in the autotests as that is the only use case.
1689 #
1690 # pylint: disable=no-member
1691 if (distutils.version.StrictVersion(cherrypy.__version__) <=
1692 distutils.version.StrictVersion('3.2.2')):
1693 cherrypy_ext.ZeroPortPatcher.DoPatch(cherrypy)
1694 # pylint: enable=no-member
1695
Chris Sosa855b8932013-08-21 20:24:551696 if options.pidfile:
1697 plugins.PIDFile(cherrypy.engine, options.pidfile).subscribe()
1698
Gilad Arnold11fbef42014-02-10 19:04:131699 if options.portfile:
1700 cherrypy_ext.PortFile(cherrypy.engine, options.portfile).subscribe()
1701
Dan Shiafd5c6c2016-01-07 18:27:031702 if (options.android_build_credential and
1703 os.path.exists(options.android_build_credential)):
1704 try:
1705 with open(options.android_build_credential) as f:
1706 android_build.BuildAccessor.credential_info = json.load(f)
1707 except ValueError as e:
1708 _Log('Failed to load the android build credential: %s. Error: %s.' %
1709 (options.android_build_credential, e))
Congbin Guo3afae6c2019-08-13 23:29:421710
1711 cherrypy.tree.mount(health_checker_app, '/check_health',
1712 config=health_checker.get_config())
joychen3cb228e2013-06-12 19:13:131713 cherrypy.quickstart(dev_server, config=_GetConfig(options))
Chris Sosacde6bf42012-06-01 01:36:391714
1715
1716if __name__ == '__main__':
1717 main()