blob: eeb5418b463012f2c3ae9ccf0e37e74fb97cd0de [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
Gilad Arnold55a2a372012-10-02 16:46:3227import json
David Riley2fcb0122017-11-02 18:25:3928import optparse # pylint: disable=deprecated-module
[email protected]ded22402009-10-26 22:36:2129import os
Scott Zawalski4647ce62012-01-03 22:17:2830import re
Simran Basi4baad082013-02-14 21:39:1831import shutil
xixuan52c2fba2016-05-21 00:02:4832import signal
Mandeep Singh Baines38dcdda2012-12-08 01:55:3333import socket
Chris Masone816e38c2012-05-02 19:22:3634import subprocess
J. Richard Barnette3d977b82013-04-23 18:05:1935import sys
Chris Masone816e38c2012-05-02 19:22:3636import tempfile
Dan Shi59ae7092013-06-04 21:37:2737import threading
Gilad Arnoldd5ebaaa2012-10-02 18:52:3838import types
J. Richard Barnette3d977b82013-04-23 18:05:1939from logging import handlers
40
Amin Hassanid4e35392019-10-03 18:02:4441from six.moves import http_client
42
43# pylint: disable=no-name-in-module, import-error
J. Richard Barnette3d977b82013-04-23 18:05:1944import cherrypy
Chris Sosa855b8932013-08-21 20:24:5545from cherrypy import _cplogging as cplogging
Amin Hassanid4e35392019-10-03 18:02:4446from cherrypy.process import plugins
47# pylint: enable=no-name-in-module, import-error
[email protected]ded22402009-10-26 22:36:2148
Achuith Bhandarkar662fb722019-10-31 23:12:4949import autoupdate
50import cherrypy_ext
51import cros_update
52import cros_update_progress
53import health_checker
54
Richard Barnettedf35c322017-08-19 00:02:1355# This must happen before any local modules get a chance to import
56# anything from chromite. Otherwise, really bad things will happen, and
57# you will _not_ understand why.
Congbin Guo3afae6c2019-08-13 23:29:4258import setup_chromite # pylint: disable=unused-import
Achuith Bhandarkar662fb722019-10-31 23:12:4959from chromite.lib.xbuddy import android_build
60from chromite.lib.xbuddy import artifact_info
61from chromite.lib.xbuddy import build_artifact
62from chromite.lib.xbuddy import cherrypy_log_util
63from chromite.lib.xbuddy import common_util
64from chromite.lib.xbuddy import devserver_constants
65from chromite.lib.xbuddy import downloader
66from chromite.lib.xbuddy import xbuddy
Richard Barnettedf35c322017-08-19 00:02:1367
Gilad Arnoldc65330c2012-09-20 22:17:4868
Gilad Arnoldc65330c2012-09-20 22:17:4869# Module-local log function.
Chris Sosa6a3697f2013-01-30 00:44:4370def _Log(message, *args):
Achuith Bhandarkar662fb722019-10-31 23:12:4971 return cherrypy_log_util.LogWithTag('DEVSERVER', message, *args)
Frank Farzan40160872011-12-13 02:39:1872
Chris Sosa417e55d2011-01-26 00:40:4873CACHED_ENTRIES = 12
Don Garrettf90edf02010-11-17 01:36:1474
Simran Basi4baad082013-02-14 21:39:1875TELEMETRY_FOLDER = 'telemetry_src'
76TELEMETRY_DEPS = ['dep-telemetry_dep.tar.bz2',
77 'dep-page_cycler_dep.tar.bz2',
Simran Basi0d078682013-03-22 23:40:0478 'dep-chrome_test.tar.bz2',
79 'dep-perf_data_dep.tar.bz2']
Simran Basi4baad082013-02-14 21:39:1880
Chris Sosa0356d3b2010-09-16 22:46:2281# Sets up global to share between classes.
[email protected]21a5ca32009-11-04 18:23:2382updater = None
[email protected]ded22402009-10-26 22:36:2183
xixuan3d48bff2017-01-31 03:00:0984# Log rotation parameters. These settings correspond to twice a day once
85# devserver is started, with about two weeks (28 backup files) of old logs
86# kept for backup.
J. Richard Barnette3d977b82013-04-23 18:05:1987#
xixuan3d48bff2017-01-31 03:00:0988# For more, see the documentation in standard python library for
J. Richard Barnette3d977b82013-04-23 18:05:1989# logging.handlers.TimedRotatingFileHandler
xixuan3d48bff2017-01-31 03:00:0990_LOG_ROTATION_TIME = 'H'
Congbin Guo3afae6c2019-08-13 23:29:4291_LOG_ROTATION_INTERVAL = 12 # hours
92_LOG_ROTATION_BACKUP = 28 # backup counts
Frank Farzan40160872011-12-13 02:39:1893
xixuan52c2fba2016-05-21 00:02:4894# Auto-update parameters
95
96# Error msg for missing key in CrOS auto-update.
Xixuan Wu32af9f12017-11-13 22:11:4497KEY_ERROR_MSG = 'Key Error in RPC: %s= is required'
xixuan52c2fba2016-05-21 00:02:4898
xixuan52c2fba2016-05-21 00:02:4899
Amin Hassanid4e35392019-10-03 18:02:44100class DevServerError(Exception):
101 """Exception class used by DevServer."""
102
103
Gabe Black3b567202015-09-23 21:07:59104def _canonicalize_archive_url(archive_url):
105 """Canonicalizes archive_url strings.
106
107 Raises:
108 DevserverError: if archive_url is not set.
109 """
110 if archive_url:
111 if not archive_url.startswith('gs://'):
Amin Hassanid4e35392019-10-03 18:02:44112 raise DevServerError(
Congbin Guo4132a272019-08-20 19:32:14113 "Archive URL isn't from Google Storage (%s) ." % archive_url)
Gabe Black3b567202015-09-23 21:07:59114
115 return archive_url.rstrip('/')
116 else:
Amin Hassanid4e35392019-10-03 18:02:44117 raise DevServerError('Must specify an archive_url in the request')
Gabe Black3b567202015-09-23 21:07:59118
119
120def _canonicalize_local_path(local_path):
121 """Canonicalizes |local_path| strings.
122
123 Raises:
124 DevserverError: if |local_path| is not set.
125 """
126 # Restrict staging of local content to only files within the static
127 # directory.
128 local_path = os.path.abspath(local_path)
129 if not local_path.startswith(updater.static_dir):
Amin Hassanid4e35392019-10-03 18:02:44130 raise DevServerError(
Congbin Guo4132a272019-08-20 19:32:14131 'Local path %s must be a subdirectory of the static'
132 ' directory: %s' % (local_path, updater.static_dir))
Gabe Black3b567202015-09-23 21:07:59133
134 return local_path.rstrip('/')
135
136
137def _get_artifacts(kwargs):
138 """Returns a tuple of named and file artifacts given the stage rpc kwargs.
139
140 Raises:
141 DevserverError if no artifacts would be returned.
142 """
143 artifacts = kwargs.get('artifacts')
144 files = kwargs.get('files')
145 if not artifacts and not files:
Amin Hassanid4e35392019-10-03 18:02:44146 raise DevServerError('No artifacts specified.')
Gabe Black3b567202015-09-23 21:07:59147
148 # Note we NEED to coerce files to a string as we get raw unicode from
149 # cherrypy and we treat files as strings elsewhere in the code.
150 return (str(artifacts).split(',') if artifacts else [],
151 str(files).split(',') if files else [])
152
153
Dan Shi61305df2015-10-26 23:52:35154def _is_android_build_request(kwargs):
155 """Check if a devserver call is for Android build, based on the arguments.
156
157 This method exams the request's arguments (os_type) to determine if the
158 request is for Android build. If os_type is set to `android`, returns True.
159 If os_type is not set or has other values, returns False.
160
161 Args:
162 kwargs: Keyword arguments for the request.
163
164 Returns:
165 True if the request is for Android build. False otherwise.
166 """
167 os_type = kwargs.get('os_type', None)
168 return os_type == 'android'
169
170
Gabe Black3b567202015-09-23 21:07:59171def _get_downloader(kwargs):
172 """Returns the downloader based on passed in arguments.
173
174 Args:
Amin Hassani08e42d22019-06-03 07:31:30175 kwargs: Keyword arguments for the request.
Gabe Black3b567202015-09-23 21:07:59176 """
177 local_path = kwargs.get('local_path')
178 if local_path:
179 local_path = _canonicalize_local_path(local_path)
180
181 dl = None
182 if local_path:
Prathmesh Prabhu58d08932018-01-19 23:08:19183 delete_source = _parse_boolean_arg(kwargs, 'delete_source')
184 dl = downloader.LocalDownloader(updater.static_dir, local_path,
185 delete_source=delete_source)
Gabe Black3b567202015-09-23 21:07:59186
Dan Shi61305df2015-10-26 23:52:35187 if not _is_android_build_request(kwargs):
Gabe Black3b567202015-09-23 21:07:59188 archive_url = kwargs.get('archive_url')
189 if not archive_url and not local_path:
Amin Hassanid4e35392019-10-03 18:02:44190 raise DevServerError(
Congbin Guo4132a272019-08-20 19:32:14191 'Requires archive_url or local_path to be specified.')
Gabe Black3b567202015-09-23 21:07:59192 if archive_url and local_path:
Amin Hassanid4e35392019-10-03 18:02:44193 raise DevServerError(
Congbin Guo4132a272019-08-20 19:32:14194 'archive_url and local_path can not both be specified.')
Gabe Black3b567202015-09-23 21:07:59195 if not dl:
196 archive_url = _canonicalize_archive_url(archive_url)
Luis Hector Chavezdca9dd72018-06-12 19:56:30197 dl = downloader.GoogleStorageDownloader(
198 updater.static_dir, archive_url,
199 downloader.GoogleStorageDownloader.GetBuildIdFromArchiveURL(
200 archive_url))
Gabe Black3b567202015-09-23 21:07:59201 elif not dl:
202 target = kwargs.get('target', None)
Dan Shi72b16132015-10-08 19:10:33203 branch = kwargs.get('branch', None)
Dan Shi61305df2015-10-26 23:52:35204 build_id = kwargs.get('build_id', None)
205 if not target or not branch or not build_id:
Amin Hassanid4e35392019-10-03 18:02:44206 raise DevServerError('target, branch, build ID must all be specified for '
207 'downloading Android build.')
Dan Shi72b16132015-10-08 19:10:33208 dl = downloader.AndroidBuildDownloader(updater.static_dir, branch, build_id,
209 target)
Gabe Black3b567202015-09-23 21:07:59210
211 return dl
212
213
214def _get_downloader_and_factory(kwargs):
215 """Returns the downloader and artifact factory based on passed in arguments.
216
217 Args:
Amin Hassani08e42d22019-06-03 07:31:30218 kwargs: Keyword arguments for the request.
Gabe Black3b567202015-09-23 21:07:59219 """
220 artifacts, files = _get_artifacts(kwargs)
221 dl = _get_downloader(kwargs)
222
Achuith Bhandarkar2a1fcd82019-10-18 00:45:58223 if (isinstance(dl, (downloader.GoogleStorageDownloader,
224 downloader.LocalDownloader))):
Gabe Black3b567202015-09-23 21:07:59225 factory_class = build_artifact.ChromeOSArtifactFactory
Dan Shi72b16132015-10-08 19:10:33226 elif isinstance(dl, downloader.AndroidBuildDownloader):
Gabe Black3b567202015-09-23 21:07:59227 factory_class = build_artifact.AndroidArtifactFactory
228 else:
Amin Hassanid4e35392019-10-03 18:02:44229 raise DevServerError(
Congbin Guo4132a272019-08-20 19:32:14230 'Unrecognized value for downloader type: %s' % type(dl))
Gabe Black3b567202015-09-23 21:07:59231
232 factory = factory_class(dl.GetBuildDir(), artifacts, files, dl.GetBuild())
233
234 return dl, factory
235
236
Scott Zawalski4647ce62012-01-03 22:17:28237def _LeadingWhiteSpaceCount(string):
238 """Count the amount of leading whitespace in a string.
239
240 Args:
241 string: The string to count leading whitespace in.
Don Garrettf84631a2014-01-08 02:21:26242
Scott Zawalski4647ce62012-01-03 22:17:28243 Returns:
244 number of white space chars before characters start.
245 """
Gabe Black3b567202015-09-23 21:07:59246 matched = re.match(r'^\s+', string)
Scott Zawalski4647ce62012-01-03 22:17:28247 if matched:
248 return len(matched.group())
249
250 return 0
251
252
253def _PrintDocStringAsHTML(func):
254 """Make a functions docstring somewhat HTML style.
255
256 Args:
257 func: The function to return the docstring from.
Don Garrettf84631a2014-01-08 02:21:26258
Scott Zawalski4647ce62012-01-03 22:17:28259 Returns:
260 A string that is somewhat formated for a web browser.
261 """
262 # TODO(scottz): Make this parse Args/Returns in a prettier way.
263 # Arguments could be bolded and indented etc.
264 html_doc = []
265 for line in func.__doc__.splitlines():
266 leading_space = _LeadingWhiteSpaceCount(line)
267 if leading_space > 0:
Chris Sosa47a7d4e2012-03-28 18:26:55268 line = ' ' * leading_space + line
Scott Zawalski4647ce62012-01-03 22:17:28269
270 html_doc.append('<BR>%s' % line)
271
272 return '\n'.join(html_doc)
273
274
Simran Basief83d6a2014-08-28 21:32:01275def _GetUpdateTimestampHandler(static_dir):
276 """Returns a handler to update directory staged.timestamp.
277
278 This handler resets the stage.timestamp whenever static content is accessed.
279
280 Args:
281 static_dir: Directory from which static content is being staged.
282
283 Returns:
Amin Hassani08e42d22019-06-03 07:31:30284 A cherrypy handler to update the timestamp of accessed content.
Simran Basief83d6a2014-08-28 21:32:01285 """
286 def UpdateTimestampHandler():
287 if not '404' in cherrypy.response.status:
288 build_match = re.match(devserver_constants.STAGED_BUILD_REGEX,
289 cherrypy.request.path_info)
290 if build_match:
291 build_dir = os.path.join(static_dir, build_match.group('build'))
292 downloader.Downloader.TouchTimestampForStaged(build_dir)
293 return UpdateTimestampHandler
294
295
Chris Sosa7c931362010-10-12 02:49:01296def _GetConfig(options):
297 """Returns the configuration for the devserver."""
Mandeep Singh Baines38dcdda2012-12-08 01:55:33298
Mandeep Singh Baines38dcdda2012-12-08 01:55:33299 socket_host = '::'
Yu-Ju Hongc8d4af32013-11-12 23:14:26300 # Fall back to IPv4 when python is not configured with IPv6.
301 if not socket.has_ipv6:
Mandeep Singh Baines38dcdda2012-12-08 01:55:33302 socket_host = '0.0.0.0'
303
Simran Basief83d6a2014-08-28 21:32:01304 # Adds the UpdateTimestampHandler to cherrypy's tools. This tools executes
305 # on the on_end_resource hook. This hook is called once processing is
306 # complete and the response is ready to be returned.
307 cherrypy.tools.update_timestamp = cherrypy.Tool(
308 'on_end_resource', _GetUpdateTimestampHandler(options.static_dir))
309
David Riley2fcb0122017-11-02 18:25:39310 base_config = {
311 'global': {
312 'server.log_request_headers': True,
313 'server.protocol_version': 'HTTP/1.1',
314 'server.socket_host': socket_host,
315 'server.socket_port': int(options.port),
316 'response.timeout': 6000,
317 'request.show_tracebacks': True,
318 'server.socket_timeout': 60,
319 'server.thread_pool': 2,
320 'engine.autoreload.on': False,
321 },
322 '/api': {
323 # Gets rid of cherrypy parsing post file for args.
324 'request.process_request_body': False,
325 },
326 '/build': {
327 'response.timeout': 100000,
328 },
329 '/update': {
330 # Gets rid of cherrypy parsing post file for args.
331 'request.process_request_body': False,
332 'response.timeout': 10000,
333 },
334 # Sets up the static dir for file hosting.
335 '/static': {
336 'tools.staticdir.dir': options.static_dir,
337 'tools.staticdir.on': True,
338 'response.timeout': 10000,
339 'tools.update_timestamp.on': True,
340 },
341 }
Chris Sosa5f118ef2012-07-12 18:37:50342 if options.production:
Alex Miller93beca52013-07-31 02:25:09343 base_config['global'].update({'server.thread_pool': 150})
Scott Zawalski1c5e7cd2012-02-27 18:12:52344
Chris Sosa7c931362010-10-12 02:49:01345 return base_config
[email protected]64244662009-11-12 00:52:08346
Darin Petkove17164a2010-08-11 20:24:41347
Gilad Arnoldd5ebaaa2012-10-02 18:52:38348def _GetRecursiveMemberObject(root, member_list):
349 """Returns an object corresponding to a nested member list.
350
351 Args:
352 root: the root object to search
353 member_list: list of nested members to search
Don Garrettf84631a2014-01-08 02:21:26354
Gilad Arnoldd5ebaaa2012-10-02 18:52:38355 Returns:
356 An object corresponding to the member name list; None otherwise.
357 """
358 for member in member_list:
359 next_root = root.__class__.__dict__.get(member)
360 if not next_root:
361 return None
362 root = next_root
363 return root
364
365
366def _IsExposed(name):
367 """Returns True iff |name| has an `exposed' attribute and it is set."""
368 return hasattr(name, 'exposed') and name.exposed
369
370
Congbin Guo6bc32182019-08-21 00:54:30371def _GetExposedMethod(nested_member):
Gilad Arnoldd5ebaaa2012-10-02 18:52:38372 """Returns a CherryPy-exposed method, if such exists.
373
374 Args:
Gilad Arnoldd5ebaaa2012-10-02 18:52:38375 nested_member: a slash-joined path to the nested member
Don Garrettf84631a2014-01-08 02:21:26376
Gilad Arnoldd5ebaaa2012-10-02 18:52:38377 Returns:
Congbin Guo6bc32182019-08-21 00:54:30378 A function object corresponding to the path defined by |nested_member| from
379 the app root object registered, if the function is exposed; None otherwise.
Gilad Arnoldd5ebaaa2012-10-02 18:52:38380 """
Congbin Guo6bc32182019-08-21 00:54:30381 for app in cherrypy.tree.apps.values():
382 # Use the 'index' function doc as the doc of the app.
383 if nested_member == app.script_name.lstrip('/'):
384 nested_member = 'index'
385
386 method = _GetRecursiveMemberObject(app.root, nested_member.split('/'))
387 if method and isinstance(method, types.FunctionType) and _IsExposed(method):
388 return method
Gilad Arnoldd5ebaaa2012-10-02 18:52:38389
390
Gilad Arnold748c8322012-10-12 16:51:35391def _FindExposedMethods(root, prefix, unlisted=None):
Gilad Arnoldd5ebaaa2012-10-02 18:52:38392 """Finds exposed CherryPy methods.
393
394 Args:
395 root: the root object for searching
396 prefix: slash-joined chain of members leading to current object
397 unlisted: URLs to be excluded regardless of their exposed status
Don Garrettf84631a2014-01-08 02:21:26398
Gilad Arnoldd5ebaaa2012-10-02 18:52:38399 Returns:
400 List of exposed URLs that are not unlisted.
401 """
402 method_list = []
Congbin Guo6bc32182019-08-21 00:54:30403 for member in root.__class__.__dict__.keys():
Gilad Arnoldd5ebaaa2012-10-02 18:52:38404 prefixed_member = prefix + '/' + member if prefix else member
Gilad Arnold748c8322012-10-12 16:51:35405 if unlisted and prefixed_member in unlisted:
Gilad Arnoldd5ebaaa2012-10-02 18:52:38406 continue
407 member_obj = root.__class__.__dict__[member]
408 if _IsExposed(member_obj):
Amin Hassani08e42d22019-06-03 07:31:30409 if isinstance(member_obj, types.FunctionType):
Congbin Guo6bc32182019-08-21 00:54:30410 # Regard the app name as exposed "method" name if it exposed 'index'
411 # function.
412 if prefix and member == 'index':
413 method_list.append(prefix)
414 else:
415 method_list.append(prefixed_member)
Gilad Arnoldd5ebaaa2012-10-02 18:52:38416 else:
417 method_list += _FindExposedMethods(
418 member_obj, prefixed_member, unlisted)
419 return method_list
420
421
xixuan52c2fba2016-05-21 00:02:48422def _check_base_args_for_auto_update(kwargs):
xixuanac89ce82016-12-01 00:48:20423 """Check basic args required for auto-update.
424
425 Args:
426 kwargs: the parameters to be checked.
427
428 Raises:
429 DevServerHTTPError if required parameters don't exist in kwargs.
430 """
xixuan52c2fba2016-05-21 00:02:48431 if 'host_name' not in kwargs:
Amin Hassanid4e35392019-10-03 18:02:44432 raise common_util.DevServerHTTPError(http_client.INTERNAL_SERVER_ERROR,
Amin Hassani08e42d22019-06-03 07:31:30433 KEY_ERROR_MSG % 'host_name')
xixuan52c2fba2016-05-21 00:02:48434
435 if 'build_name' not in kwargs:
Amin Hassanid4e35392019-10-03 18:02:44436 raise common_util.DevServerHTTPError(http_client.INTERNAL_SERVER_ERROR,
Amin Hassani08e42d22019-06-03 07:31:30437 KEY_ERROR_MSG % 'build_name')
xixuan52c2fba2016-05-21 00:02:48438
439
440def _parse_boolean_arg(kwargs, key):
xixuanac89ce82016-12-01 00:48:20441 """Parse boolean arg from kwargs.
442
443 Args:
444 kwargs: the parameters to be checked.
445 key: the key to be parsed.
446
447 Returns:
448 The boolean value of kwargs[key], or False if key doesn't exist in kwargs.
449
450 Raises:
451 DevServerHTTPError if kwargs[key] is not a boolean variable.
452 """
xixuan52c2fba2016-05-21 00:02:48453 if key in kwargs:
454 if kwargs[key] == 'True':
455 return True
456 elif kwargs[key] == 'False':
457 return False
458 else:
459 raise common_util.DevServerHTTPError(
Amin Hassanid4e35392019-10-03 18:02:44460 http_client.INTERNAL_SERVER_ERROR,
xixuan52c2fba2016-05-21 00:02:48461 'The value for key %s is not boolean.' % key)
462 else:
463 return False
464
xixuan447ad9d2017-02-28 22:46:20465
xixuanac89ce82016-12-01 00:48:20466def _parse_string_arg(kwargs, key):
467 """Parse string arg from kwargs.
468
469 Args:
470 kwargs: the parameters to be checked.
471 key: the key to be parsed.
472
473 Returns:
474 The string value of kwargs[key], or None if key doesn't exist in kwargs.
475 """
476 if key in kwargs:
477 return kwargs[key]
478 else:
479 return None
480
xixuan447ad9d2017-02-28 22:46:20481
xixuanac89ce82016-12-01 00:48:20482def _build_uri_from_build_name(build_name):
483 """Get build url from a given build name.
484
485 Args:
486 build_name: the build name to be parsed, whose format is
487 'board/release_version'.
488
489 Returns:
490 The release_archive_url on Google Storage for this build name.
491 """
Amin Hassani08e42d22019-06-03 07:31:30492 # TODO(ahassani): This function doesn't seem to be used anywhere since its
493 # previous use of lib.paygen.gspath was broken and it doesn't seem to be
494 # causing any runtime issues. So deprecate this in the future.
495 tokens = build_name.split('/')
496 return 'gs://chromeos-releases/stable-channel/%s/%s' % (tokens[0], tokens[1])
xixuan52c2fba2016-05-21 00:02:48497
xixuan447ad9d2017-02-28 22:46:20498
499def _clear_process(host_name, pid):
500 """Clear AU process for given hostname and pid.
501
502 This clear includes:
503 1. kill process if it's alive.
504 2. delete the track status file of this process.
505 3. delete the executing log file of this process.
506
507 Args:
508 host_name: the host to execute auto-update.
509 pid: the background auto-update process id.
510 """
511 if cros_update_progress.IsProcessAlive(pid):
512 os.killpg(int(pid), signal.SIGKILL)
513
514 cros_update_progress.DelTrackStatusFile(host_name, pid)
515 cros_update_progress.DelExecuteLogFile(host_name, pid)
516
517
Dale Curtisc9aaf3a2011-08-09 22:47:40518class ApiRoot(object):
519 """RESTful API for Dev Server information."""
520 exposed = True
521
522 @cherrypy.expose
Gilad Arnold286a0062012-01-12 21:47:02523 def hostlog(self, ip):
Gilad Arnold1b908392012-10-05 18:36:27524 """Returns a JSON object containing a log of host event.
525
526 Args:
527 ip: address of host whose event log is requested, or `all'
Don Garrettf84631a2014-01-08 02:21:26528
Gilad Arnold1b908392012-10-05 18:36:27529 Returns:
Amin Hassani7c447852019-09-26 22:01:48530 A JSON dictionary containing all or some of the following fields:
Amin Hassanie7ead902019-10-11 23:42:43531 version: The Chromium OS version the device is running.
532 track: The channel the device is running on.
533 board: The device's board.
534 event_result: The event result of Omaha request.
535 event_type: The event type of Omaha request.
536 previous_version: The Chromium OS version we updated and rebooted from.
537 timestamp: The timestamp the event was received.
Amin Hassani7c447852019-09-26 22:01:48538 See the OmahaEvent class in update_engine/omaha_request_action.h for
539 event type and status code definitions. If the ip does not exist an empty
540 string is returned.
Gilad Arnold1b908392012-10-05 18:36:27541
542 Example URL:
543 http://myhost/api/hostlog?ip=192.168.1.5
544 """
Gilad Arnold286a0062012-01-12 21:47:02545 return updater.HandleHostLogPing(ip)
546
547 @cherrypy.expose
Don Garrettf84631a2014-01-08 02:21:26548 def fileinfo(self, *args):
Gilad Arnold55a2a372012-10-02 16:46:32549 """Returns information about a given staged file.
550
551 Args:
Don Garrettf84631a2014-01-08 02:21:26552 args: path to the file inside the server's static staging directory
553
Gilad Arnold55a2a372012-10-02 16:46:32554 Returns:
555 A JSON encoded dictionary with information about the said file, which may
556 contain the following keys/values:
Gilad Arnold1b908392012-10-05 18:36:27557 size (int): the file size in bytes
Gilad Arnold1b908392012-10-05 18:36:27558 sha256 (string): a base64 encoded SHA256 hash
559
560 Example URL:
561 http://myhost/api/fileinfo/some/path/to/file
Gilad Arnold55a2a372012-10-02 16:46:32562 """
Amin Hassani28df4212019-10-28 17:16:50563 # TODO(ahassani): A better way of doing this is to just return the the
564 # content of the payloads' property file instead. That has all this info
565 # except that the key for sha256 is 'sha256_hex', but still base64 encdoed.
566
Don Garrettf84631a2014-01-08 02:21:26567 file_path = os.path.join(updater.static_dir, *args)
Gilad Arnold55a2a372012-10-02 16:46:32568 if not os.path.exists(file_path):
Amin Hassanid4e35392019-10-03 18:02:44569 raise DevServerError('file not found: %s' % file_path)
Gilad Arnold55a2a372012-10-02 16:46:32570 try:
571 file_size = os.path.getsize(file_path)
Gilad Arnold55a2a372012-10-02 16:46:32572 file_sha256 = common_util.GetFileSha256(file_path)
Amin Hassani469f5702019-10-21 22:35:06573 except os.error as e:
Amin Hassanid4e35392019-10-03 18:02:44574 raise DevServerError(
Congbin Guo4132a272019-08-20 19:32:14575 'failed to get info for file %s: %s' % (file_path, e))
Gilad Arnolde74b3812013-04-22 18:27:38576
Gilad Arnolde74b3812013-04-22 18:27:38577 return json.dumps({
Amin Hassani28df4212019-10-28 17:16:50578 'size': file_size,
579 'sha256': file_sha256,
580 }, sort_keys=True)
Gilad Arnold55a2a372012-10-02 16:46:32581
Chris Sosa76e44b92013-01-31 20:11:38582
David Rochberg7c79a812011-01-19 19:24:45583class DevServerRoot(object):
Chris Sosa7c931362010-10-12 02:49:01584 """The Root Class for the Dev Server.
585
586 CherryPy works as follows:
587 For each method in this class, cherrpy interprets root/path
588 as a call to an instance of DevServerRoot->method_name. For example,
589 a call to http://myhost/build will call build. CherryPy automatically
590 parses http args and places them as keyword arguments in each method.
591 For paths http://myhost/update/dir1/dir2, you can use *args so that
592 cherrypy uses the update method and puts the extra paths in args.
593 """
Gilad Arnoldf8f769f2012-09-24 15:43:01594 # Method names that should not be listed on the index page.
595 _UNLISTED_METHODS = ['index', 'doc']
596
Dale Curtisc9aaf3a2011-08-09 22:47:40597 api = ApiRoot()
Chris Sosa7c931362010-10-12 02:49:01598
Dan Shi59ae7092013-06-04 21:37:27599 # Number of threads that devserver is staging images.
600 _staging_thread_count = 0
601 # Lock used to lock increasing/decreasing count.
602 _staging_thread_count_lock = threading.Lock()
603
joychen3cb228e2013-06-12 19:13:13604 def __init__(self, _xbuddy):
Nick Sanders7dcaa2e2011-08-04 22:20:41605 self._builder = None
Simran Basi4baad082013-02-14 21:39:18606 self._telemetry_lock_dict = common_util.LockDict()
joychen3cb228e2013-06-12 19:13:13607 self._xbuddy = _xbuddy
David Rochberg7c79a812011-01-19 19:24:45608
Congbin Guo3afae6c2019-08-13 23:29:42609 @property
610 def staging_thread_count(self):
611 """Get the staging thread count."""
612 return self._staging_thread_count
Dan Shiafd0e492015-05-27 21:23:51613
Dale Curtisc9aaf3a2011-08-09 22:47:40614 @cherrypy.expose
David Rochberg7c79a812011-01-19 19:24:45615 def build(self, board, pkg, **kwargs):
Chris Sosa7c931362010-10-12 02:49:01616 """Builds the package specified."""
Nick Sanders7dcaa2e2011-08-04 22:20:41617 import builder
618 if self._builder is None:
619 self._builder = builder.Builder()
David Rochberg7c79a812011-01-19 19:24:45620 return self._builder.Build(board, pkg, kwargs)
Chris Sosa7c931362010-10-12 02:49:01621
Dale Curtisc9aaf3a2011-08-09 22:47:40622 @cherrypy.expose
Dan Shif8eb0d12013-08-02 00:52:06623 def is_staged(self, **kwargs):
624 """Check if artifacts have been downloaded.
625
Congbin Guo3afae6c2019-08-13 23:29:42626 Examples:
627 To check if autotest and test_suites are staged:
628 http://devserver_url:<port>/is_staged?archive_url=gs://your_url/path&
629 artifacts=autotest,test_suites
630
Amin Hassani08e42d22019-06-03 07:31:30631 Args:
Chris Sosa6b0c6172013-08-06 00:01:33632 async: True to return without waiting for download to complete.
633 artifacts: Comma separated list of named artifacts to download.
634 These are defined in artifact_info and have their implementation
635 in build_artifact.py.
636 files: Comma separated list of file artifacts to stage. These
637 will be available as is in the corresponding static directory with no
638 custom post-processing.
639
Congbin Guo3afae6c2019-08-13 23:29:42640 Returns:
641 True of all artifacts are staged.
Dan Shif8eb0d12013-08-02 00:52:06642 """
Gabe Black3b567202015-09-23 21:07:59643 dl, factory = _get_downloader_and_factory(kwargs)
Aviv Keshet57d18172016-06-19 03:39:09644 response = str(dl.IsStaged(factory))
645 _Log('Responding to is_staged %s request with %r', kwargs, response)
646 return response
Dan Shi59ae7092013-06-04 21:37:27647
Chris Sosa76e44b92013-01-31 20:11:38648 @cherrypy.expose
Prashanth Ba06d2d22014-03-07 23:35:19649 def list_image_dir(self, **kwargs):
650 """Take an archive url and list the contents in its staged directory.
651
Amin Hassani08e42d22019-06-03 07:31:30652 Examples:
Prashanth Ba06d2d22014-03-07 23:35:19653 To list the contents of where this devserver should have staged
654 gs://image-archive/<board>-release/<build> call:
655 http://devserver_url:<port>/list_image_dir?archive_url=<gs://..>
656
Congbin Guo3afae6c2019-08-13 23:29:42657 Args:
658 archive_url: Google Storage URL for the build.
659
Prashanth Ba06d2d22014-03-07 23:35:19660 Returns:
661 A string with information about the contents of the image directory.
662 """
Gabe Black3b567202015-09-23 21:07:59663 dl = _get_downloader(kwargs)
Prashanth Ba06d2d22014-03-07 23:35:19664 try:
Gabe Black3b567202015-09-23 21:07:59665 image_dir_contents = dl.ListBuildDir()
Prashanth Ba06d2d22014-03-07 23:35:19666 except build_artifact.ArtifactDownloadError as e:
667 return 'Cannot list the contents of staged artifacts. %s' % e
668 if not image_dir_contents:
Gabe Black3b567202015-09-23 21:07:59669 return '%s has not been staged on this devserver.' % dl.DescribeSource()
Prashanth Ba06d2d22014-03-07 23:35:19670 return image_dir_contents
671
672 @cherrypy.expose
Chris Sosa76e44b92013-01-31 20:11:38673 def stage(self, **kwargs):
Gabe Black3b567202015-09-23 21:07:59674 """Downloads and caches build artifacts.
Chris Sosa76e44b92013-01-31 20:11:38675
Gabe Black3b567202015-09-23 21:07:59676 Downloads and caches build artifacts, possibly from a Google Storage URL,
Dan Shi72b16132015-10-08 19:10:33677 or from Android's build server. Returns once these have been downloaded
Gabe Black3b567202015-09-23 21:07:59678 on the devserver. A call to this will attempt to cache non-specified
679 artifacts in the background for the given from the given URL following
680 the principle of spatial locality. Spatial locality of different
Chris Sosa76e44b92013-01-31 20:11:38681 artifacts is explicitly defined in the build_artifact module.
682
683 These artifacts will then be available from the static/ sub-directory of
684 the devserver.
685
Amin Hassani08e42d22019-06-03 07:31:30686 Examples:
Chris Sosa76e44b92013-01-31 20:11:38687 To download the autotest and test suites tarballs:
688 http://devserver_url:<port>/stage?archive_url=gs://your_url/path&
689 artifacts=autotest,test_suites
690 To download the full update payload:
691 http://devserver_url:<port>/stage?archive_url=gs://your_url/path&
692 artifacts=full_payload
Chris Sosa6b0c6172013-08-06 00:01:33693 To download just a file called blah.bin:
694 http://devserver_url:<port>/stage?archive_url=gs://your_url/path&
695 files=blah.bin
Chris Sosa76e44b92013-01-31 20:11:38696
697 For both these examples, one could find these artifacts at:
joychened64b222013-06-21 23:39:34698 http://devserver_url:<port>/static/<relative_path>*
Chris Sosa76e44b92013-01-31 20:11:38699
700 Note for this example, relative path is the archive_url stripped of its
701 basename i.e. path/ in the examples above. Specific example:
702
703 gs://chromeos-image-archive/x86-mario-release/R26-3920.0.0
704
705 Will get staged to:
706
joychened64b222013-06-21 23:39:34707 http://devserver_url:<port>/static/x86-mario-release/R26-3920.0.0
Congbin Guo3afae6c2019-08-13 23:29:42708
709 Args:
710 archive_url: Google Storage URL for the build.
711 local_path: Local path for the build.
712 delete_source: Only meaningful with local_path. bool to indicate if the
713 source files should be deleted. This is especially useful when staging
714 a file locally in resource constrained environments as it allows us to
715 move the relevant files locally instead of copying them.
716 async: True to return without waiting for download to complete.
717 artifacts: Comma separated list of named artifacts to download.
718 These are defined in artifact_info and have their implementation
719 in build_artifact.py.
720 files: Comma separated list of files to stage. These
721 will be available as is in the corresponding static directory with no
722 custom post-processing.
723 clean: True to remove any previously staged artifacts first.
Chris Sosa76e44b92013-01-31 20:11:38724 """
Gabe Black3b567202015-09-23 21:07:59725 dl, factory = _get_downloader_and_factory(kwargs)
726
Dan Shi59ae7092013-06-04 21:37:27727 with DevServerRoot._staging_thread_count_lock:
728 DevServerRoot._staging_thread_count += 1
729 try:
Laurence Goodbyf5c958d2016-01-15 02:23:56730 boolean_string = kwargs.get('clean')
731 clean = xbuddy.XBuddy.ParseBoolean(boolean_string)
732 if clean and os.path.exists(dl.GetBuildDir()):
733 _Log('Removing %s' % dl.GetBuildDir())
734 shutil.rmtree(dl.GetBuildDir())
Achuith Bhandarkar2a1fcd82019-10-18 00:45:58735 is_async = kwargs.get('async', False)
736 dl.Download(factory, is_async=is_async)
Dan Shi59ae7092013-06-04 21:37:27737 finally:
738 with DevServerRoot._staging_thread_count_lock:
739 DevServerRoot._staging_thread_count -= 1
Chris Sosa76e44b92013-01-31 20:11:38740 return 'Success'
Chris Sosacde6bf42012-06-01 01:36:39741
742 @cherrypy.expose
xixuan52c2fba2016-05-21 00:02:48743 def cros_au(self, **kwargs):
744 """Auto-update a CrOS DUT.
745
746 Args:
747 kwargs:
748 host_name: the hostname of the DUT to auto-update.
749 build_name: the build name for update the DUT.
750 force_update: Force an update even if the version installed is the
751 same. Default: False.
752 full_update: If True, do not run stateful update, directly force a full
753 reimage. If False, try stateful update first if the dut is already
754 installed with the same version.
755 async: Whether the auto_update function is ran in the background.
David Rileyee75de22017-11-02 17:48:15756 quick_provision: Whether the quick provision path is attempted first.
xixuan52c2fba2016-05-21 00:02:48757
758 Returns:
759 A tuple includes two elements:
760 a boolean variable represents whether the auto-update process is
761 successfully started.
762 an integer represents the background auto-update process id.
763 """
764 _check_base_args_for_auto_update(kwargs)
765
766 host_name = kwargs['host_name']
767 build_name = kwargs['build_name']
768 force_update = _parse_boolean_arg(kwargs, 'force_update')
769 full_update = _parse_boolean_arg(kwargs, 'full_update')
Achuith Bhandarkar2a1fcd82019-10-18 00:45:58770 is_async = _parse_boolean_arg(kwargs, 'async')
xixuanac89ce82016-12-01 00:48:20771 original_build = _parse_string_arg(kwargs, 'original_build')
David Haddock90e49442017-04-08 02:14:09772 payload_filename = _parse_string_arg(kwargs, 'payload_filename')
David Haddock20559612017-06-29 05:15:08773 clobber_stateful = _parse_boolean_arg(kwargs, 'clobber_stateful')
David Rileyee75de22017-11-02 17:48:15774 quick_provision = _parse_boolean_arg(kwargs, 'quick_provision')
775
776 devserver_url = updater.GetDevserverUrl()
777 static_url = updater.GetStaticUrl()
xixuan52c2fba2016-05-21 00:02:48778
Achuith Bhandarkar2a1fcd82019-10-18 00:45:58779 if is_async:
Amin Hassani469f5702019-10-21 22:35:06780 # Command of running auto-update.
Amin Hassani78520ae2019-10-29 20:26:51781 path = os.path.dirname(os.path.abspath(__file__))
782 execute_file = os.path.join(path, 'cros_update.py')
783 cmd = ['/usr/bin/python', '-u', execute_file, '-d', host_name,
784 '-b', build_name, '--static_dir', updater.static_dir]
xixuanac89ce82016-12-01 00:48:20785
786 # The original_build's format is like: link/3428.210.0
787 # The corresponding release_archive_url's format is like:
788 # gs://chromeos-releases/stable-channel/link/3428.210.0
789 if original_build:
790 release_archive_url = _build_uri_from_build_name(original_build)
791 # First staging the stateful.tgz synchronousely.
Amin Hassani469f5702019-10-21 22:35:06792 self.stage(files='stateful.tgz', is_async=False,
xixuanac89ce82016-12-01 00:48:20793 archive_url=release_archive_url)
Amin Hassani469f5702019-10-21 22:35:06794 cmd += ['--original_build', original_build]
xixuanac89ce82016-12-01 00:48:20795
xixuan52c2fba2016-05-21 00:02:48796 if force_update:
Amin Hassani469f5702019-10-21 22:35:06797 cmd += ['--force_update']
xixuan52c2fba2016-05-21 00:02:48798
799 if full_update:
Amin Hassani469f5702019-10-21 22:35:06800 cmd += ['--full_update']
xixuan52c2fba2016-05-21 00:02:48801
David Haddock90e49442017-04-08 02:14:09802 if payload_filename:
Amin Hassani469f5702019-10-21 22:35:06803 cmd += ['--payload_filename', payload_filename]
David Haddock90e49442017-04-08 02:14:09804
David Haddock20559612017-06-29 05:15:08805 if clobber_stateful:
Amin Hassani469f5702019-10-21 22:35:06806 cmd += ['--clobber_stateful']
David Haddock20559612017-06-29 05:15:08807
David Rileyee75de22017-11-02 17:48:15808 if quick_provision:
Amin Hassani469f5702019-10-21 22:35:06809 cmd += ['--quick_provision']
David Rileyee75de22017-11-02 17:48:15810
811 if devserver_url:
Amin Hassani469f5702019-10-21 22:35:06812 cmd += ['--devserver_url', devserver_url]
David Rileyee75de22017-11-02 17:48:15813
814 if static_url:
Amin Hassani469f5702019-10-21 22:35:06815 cmd += ['--static_url', static_url]
David Rileyee75de22017-11-02 17:48:15816
Amin Hassani78520ae2019-10-29 20:26:51817 p = subprocess.Popen(cmd, preexec_fn=os.setsid)
xixuan2a0970a2016-08-10 19:12:44818 pid = os.getpgid(p.pid)
xixuan52c2fba2016-05-21 00:02:48819
820 # Pre-write status in the track_status_file before the first call of
821 # 'get_au_status' to make sure that the track_status_file exists.
xixuan2a0970a2016-08-10 19:12:44822 progress_tracker = cros_update_progress.AUProgress(host_name, pid)
xixuan52c2fba2016-05-21 00:02:48823 progress_tracker.WriteStatus('CrOS update is just started.')
824
xixuan2a0970a2016-08-10 19:12:44825 return json.dumps((True, pid))
xixuan52c2fba2016-05-21 00:02:48826 else:
827 cros_update_trigger = cros_update.CrOSUpdateTrigger(
xixuanac89ce82016-12-01 00:48:20828 host_name, build_name, updater.static_dir, force_update=force_update,
David Rileyee75de22017-11-02 17:48:15829 full_update=full_update, original_build=original_build,
Amin Hassani78520ae2019-10-29 20:26:51830 payload_filename=payload_filename, quick_provision=quick_provision,
831 devserver_url=devserver_url, static_url=static_url)
xixuan52c2fba2016-05-21 00:02:48832 cros_update_trigger.TriggerAU()
xixuan27d50442017-08-09 17:38:25833 return json.dumps((True, -1))
xixuan52c2fba2016-05-21 00:02:48834
835 @cherrypy.expose
836 def get_au_status(self, **kwargs):
837 """Check if the auto-update task is finished.
838
839 It handles 4 cases:
840 1. If an error exists in the track_status_file, delete the track file and
841 raise it.
842 2. If cros-update process is finished, delete the file and return the
843 success result.
844 3. If the process is not running, delete the track file and raise an error
845 about 'the process is terminated due to unknown reason'.
846 4. If the track_status_file does not exist, kill the process if it exists,
847 and raise the IOError.
848
849 Args:
850 kwargs:
851 host_name: the hostname of the DUT to auto-update.
852 pid: the background process id of cros-update.
853
854 Returns:
xixuan28d99072016-10-06 19:24:16855 A dict with three elements:
xixuan52c2fba2016-05-21 00:02:48856 a boolean variable represents whether the auto-update process is
857 finished.
858 a string represents the current auto-update process status.
859 For example, 'Transfer Devserver/Stateful Update Package'.
xixuan28d99072016-10-06 19:24:16860 a detailed error message paragraph if there exists an Auto-Update
861 error, in which the last line shows the main exception. Empty
862 string otherwise.
xixuan52c2fba2016-05-21 00:02:48863 """
864 if 'host_name' not in kwargs:
Amin Hassanid4e35392019-10-03 18:02:44865 raise common_util.DevServerHTTPError(http_client.INTERNAL_SERVER_ERROR,
Amin Hassani08e42d22019-06-03 07:31:30866 KEY_ERROR_MSG % 'host_name')
xixuan52c2fba2016-05-21 00:02:48867
868 if 'pid' not in kwargs:
Amin Hassanid4e35392019-10-03 18:02:44869 raise common_util.DevServerHTTPError(http_client.INTERNAL_SERVER_ERROR,
Amin Hassani08e42d22019-06-03 07:31:30870 KEY_ERROR_MSG % 'pid')
xixuan52c2fba2016-05-21 00:02:48871
872 host_name = kwargs['host_name']
873 pid = kwargs['pid']
874 progress_tracker = cros_update_progress.AUProgress(host_name, pid)
875
xixuan28d99072016-10-06 19:24:16876 result_dict = {'finished': False, 'status': '', 'detailed_error_msg': ''}
xixuan52c2fba2016-05-21 00:02:48877 try:
878 result = progress_tracker.ReadStatus()
879 if result.startswith(cros_update_progress.ERROR_TAG):
xixuan28d99072016-10-06 19:24:16880 result_dict['detailed_error_msg'] = result[len(
881 cros_update_progress.ERROR_TAG):]
xixuan28681fd2016-11-23 19:13:56882 elif result == cros_update_progress.FINISHED:
xixuan28d99072016-10-06 19:24:16883 result_dict['finished'] = True
884 result_dict['status'] = result
xixuan28681fd2016-11-23 19:13:56885 elif not cros_update_progress.IsProcessAlive(pid):
xixuan28d99072016-10-06 19:24:16886 result_dict['detailed_error_msg'] = (
887 'Cros_update process terminated midway due to unknown reason. '
888 'Last update status was %s' % result)
xixuan28681fd2016-11-23 19:13:56889 else:
890 result_dict['status'] = result
891 except IOError as e:
892 if pid and cros_update_progress.IsProcessAlive(pid):
xixuan2a0970a2016-08-10 19:12:44893 os.killpg(int(pid), signal.SIGKILL)
xixuan52c2fba2016-05-21 00:02:48894
xixuan28681fd2016-11-23 19:13:56895 result_dict['detailed_error_msg'] = str(e)
896
897 return json.dumps(result_dict)
xixuan52c2fba2016-05-21 00:02:48898
899 @cherrypy.expose
David Riley6d5fca02017-10-31 17:35:47900 def post_au_status(self, status, **kwargs):
901 """Updates the status of an auto-update task.
902
903 Callers will need to POST to this URL with a body of MIME-type
904 "multipart/form-data".
905 The body should include a single argument, 'status', containing the
906 AU status to record.
907
908 Args:
909 status: The updated status.
910 kwargs:
911 host_name: the hostname of the DUT to auto-update.
912 pid: the background process id of cros-update.
913 """
914 if 'host_name' not in kwargs:
Amin Hassanid4e35392019-10-03 18:02:44915 raise common_util.DevServerHTTPError(http_client.INTERNAL_SERVER_ERROR,
Amin Hassani08e42d22019-06-03 07:31:30916 KEY_ERROR_MSG % 'host_name')
David Riley6d5fca02017-10-31 17:35:47917
918 if 'pid' not in kwargs:
Amin Hassanid4e35392019-10-03 18:02:44919 raise common_util.DevServerHTTPError(http_client.INTERNAL_SERVER_ERROR,
Amin Hassani08e42d22019-06-03 07:31:30920 KEY_ERROR_MSG % 'pid')
David Riley6d5fca02017-10-31 17:35:47921
922 host_name = kwargs['host_name']
923 pid = kwargs['pid']
David Riley3cea2582017-11-25 06:03:01924 status = status.rstrip()
925 _Log('Recording status for %s (%s): %s' % (host_name, pid, status))
David Riley6d5fca02017-10-31 17:35:47926 progress_tracker = cros_update_progress.AUProgress(host_name, pid)
927
David Riley3cea2582017-11-25 06:03:01928 progress_tracker.WriteStatus(status)
David Riley6d5fca02017-10-31 17:35:47929
930 return 'True'
931
932 @cherrypy.expose
xixuan52c2fba2016-05-21 00:02:48933 def handler_cleanup(self, **kwargs):
xixuan3bc974e2016-10-19 00:21:43934 """Clean track status log and temp directory for CrOS auto-update process.
xixuan52c2fba2016-05-21 00:02:48935
936 Args:
937 kwargs:
938 host_name: the hostname of the DUT to auto-update.
939 pid: the background process id of cros-update.
940 """
941 if 'host_name' not in kwargs:
Amin Hassanid4e35392019-10-03 18:02:44942 raise common_util.DevServerHTTPError(http_client.INTERNAL_SERVER_ERROR,
Amin Hassani08e42d22019-06-03 07:31:30943 KEY_ERROR_MSG % 'host_name')
xixuan52c2fba2016-05-21 00:02:48944
945 if 'pid' not in kwargs:
Amin Hassanid4e35392019-10-03 18:02:44946 raise common_util.DevServerHTTPError(http_client.INTERNAL_SERVER_ERROR,
Amin Hassani08e42d22019-06-03 07:31:30947 KEY_ERROR_MSG % 'pid')
xixuan52c2fba2016-05-21 00:02:48948
949 host_name = kwargs['host_name']
950 pid = kwargs['pid']
951 cros_update_progress.DelTrackStatusFile(host_name, pid)
xixuan3bc974e2016-10-19 00:21:43952 cros_update_progress.DelAUTempDirectory(host_name, pid)
xixuan52c2fba2016-05-21 00:02:48953
954 @cherrypy.expose
955 def kill_au_proc(self, **kwargs):
956 """Kill CrOS auto-update process using given process id.
957
958 Args:
959 kwargs:
960 host_name: Kill all the CrOS auto-update process of this host.
961
962 Returns:
963 True if all processes are killed properly.
964 """
965 if 'host_name' not in kwargs:
Amin Hassanid4e35392019-10-03 18:02:44966 raise common_util.DevServerHTTPError(http_client.INTERNAL_SERVER_ERROR,
Amin Hassani08e42d22019-06-03 07:31:30967 KEY_ERROR_MSG % 'host_name')
xixuan52c2fba2016-05-21 00:02:48968
xixuan447ad9d2017-02-28 22:46:20969 cur_pid = kwargs.get('pid')
970
xixuan52c2fba2016-05-21 00:02:48971 host_name = kwargs['host_name']
xixuan3bc974e2016-10-19 00:21:43972 track_log_list = cros_update_progress.GetAllTrackStatusFileByHostName(
973 host_name)
xixuan52c2fba2016-05-21 00:02:48974 for log in track_log_list:
975 # The track log's full path is: path/host_name_pid.log
976 # Use splitext to remove file extension, then parse pid from the
977 # filename.
Congbin Guo3afae6c2019-08-13 23:29:42978 pid = os.path.splitext(os.path.basename(log))[0][len(host_name) + 1:]
xixuan447ad9d2017-02-28 22:46:20979 _clear_process(host_name, pid)
xixuan52c2fba2016-05-21 00:02:48980
xixuan447ad9d2017-02-28 22:46:20981 if cur_pid:
982 _clear_process(host_name, cur_pid)
xixuan52c2fba2016-05-21 00:02:48983
984 return 'True'
985
986 @cherrypy.expose
987 def collect_cros_au_log(self, **kwargs):
988 """Collect CrOS auto-update log.
989
990 Args:
991 kwargs:
992 host_name: the hostname of the DUT to auto-update.
993 pid: the background process id of cros-update.
994
995 Returns:
David Haddock9f459632017-05-11 21:45:46996 A dictionary containing the execute log file and any hostlog files.
xixuan52c2fba2016-05-21 00:02:48997 """
998 if 'host_name' not in kwargs:
Amin Hassanid4e35392019-10-03 18:02:44999 raise common_util.DevServerHTTPError(http_client.INTERNAL_SERVER_ERROR,
Amin Hassani08e42d22019-06-03 07:31:301000 KEY_ERROR_MSG % 'host_name')
xixuan52c2fba2016-05-21 00:02:481001
1002 if 'pid' not in kwargs:
Amin Hassanid4e35392019-10-03 18:02:441003 raise common_util.DevServerHTTPError(http_client.INTERNAL_SERVER_ERROR,
Amin Hassani08e42d22019-06-03 07:31:301004 KEY_ERROR_MSG % 'pid')
xixuan52c2fba2016-05-21 00:02:481005
1006 host_name = kwargs['host_name']
1007 pid = kwargs['pid']
xixuan3bc974e2016-10-19 00:21:431008
1009 # Fetch the execute log recorded by cros_update_progress.
xixuan1bbfaba2016-10-14 00:53:221010 au_log = cros_update_progress.ReadExecuteLogFile(host_name, pid)
1011 cros_update_progress.DelExecuteLogFile(host_name, pid)
David Haddock9f459632017-05-11 21:45:461012 # Fetch the cros_au host_logs if they exist
1013 au_hostlogs = cros_update_progress.ReadAUHostLogFiles(host_name, pid)
1014 return json.dumps({'cros_au_log': au_log, 'host_logs': au_hostlogs})
xixuan1bbfaba2016-10-14 00:53:221015
xixuan52c2fba2016-05-21 00:02:481016 @cherrypy.expose
Dan Shi2f136862016-02-11 23:38:381017 def locate_file(self, **kwargs):
1018 """Get the path to the given file name.
1019
1020 This method looks up the given file name inside specified build artifacts.
1021 One use case is to help caller to locate an apk file inside a build
1022 artifact. The location of the apk file could be different based on the
1023 branch and target.
1024
1025 Args:
1026 file_name: Name of the file to look for.
1027 artifacts: A list of artifact names to search for the file.
1028
1029 Returns:
1030 Path to the file with the given name. It's relative to the folder for the
1031 build, e.g., DATA/priv-app/sl4a/sl4a.apk
Dan Shi2f136862016-02-11 23:38:381032 """
1033 dl, _ = _get_downloader_and_factory(kwargs)
1034 try:
Joe Brennan1691f8e2017-03-15 22:53:361035 file_name = kwargs['file_name']
Dan Shi2f136862016-02-11 23:38:381036 artifacts = kwargs['artifacts']
1037 except KeyError:
Amin Hassanid4e35392019-10-03 18:02:441038 raise DevServerError(
Congbin Guo4132a272019-08-20 19:32:141039 '`file_name` and `artifacts` are required to search '
1040 'for a file in build artifacts.')
Dan Shi2f136862016-02-11 23:38:381041 build_path = dl.GetBuildDir()
1042 for artifact in artifacts:
1043 # Get the unzipped folder of the artifact. If it's not defined in
1044 # ARTIFACT_UNZIP_FOLDER_MAP, assume the files are unzipped to the build
1045 # directory directly.
1046 folder = artifact_info.ARTIFACT_UNZIP_FOLDER_MAP.get(artifact, '')
1047 artifact_path = os.path.join(build_path, folder)
1048 for root, _, filenames in os.walk(artifact_path):
Joe Brennan1691f8e2017-03-15 22:53:361049 if file_name in set([f for f in filenames]):
Dan Shi2f136862016-02-11 23:38:381050 return os.path.relpath(os.path.join(root, file_name), build_path)
Amin Hassanid4e35392019-10-03 18:02:441051 raise DevServerError(
Congbin Guo4132a272019-08-20 19:32:141052 'File `%s` can not be found in artifacts: %s' % (file_name, artifacts))
Dan Shi2f136862016-02-11 23:38:381053
1054 @cherrypy.expose
Simran Basi4baad082013-02-14 21:39:181055 def setup_telemetry(self, **kwargs):
1056 """Extracts and sets up telemetry
1057
1058 This method goes through the telemetry deps packages, and stages them on
1059 the devserver to be used by the drones and the telemetry tests.
1060
1061 Args:
1062 archive_url: Google Storage URL for the build.
1063
1064 Returns:
1065 Path to the source folder for the telemetry codebase once it is staged.
1066 """
Gabe Black3b567202015-09-23 21:07:591067 dl = _get_downloader(kwargs)
Simran Basi4baad082013-02-14 21:39:181068
Gabe Black3b567202015-09-23 21:07:591069 build_path = dl.GetBuildDir()
Simran Basi4baad082013-02-14 21:39:181070 deps_path = os.path.join(build_path, 'autotest/packages')
1071 telemetry_path = os.path.join(build_path, TELEMETRY_FOLDER)
1072 src_folder = os.path.join(telemetry_path, 'src')
1073
1074 with self._telemetry_lock_dict.lock(telemetry_path):
1075 if os.path.exists(src_folder):
1076 # Telemetry is already fully stage return
1077 return src_folder
1078
1079 common_util.MkDirP(telemetry_path)
1080
1081 # Copy over the required deps tar balls to the telemetry directory.
1082 for dep in TELEMETRY_DEPS:
1083 dep_path = os.path.join(deps_path, dep)
Simran Basi0d078682013-03-22 23:40:041084 if not os.path.exists(dep_path):
1085 # This dep does not exist (could be new), do not extract it.
1086 continue
Simran Basi4baad082013-02-14 21:39:181087 try:
1088 common_util.ExtractTarball(dep_path, telemetry_path)
1089 except common_util.CommonUtilError as e:
1090 shutil.rmtree(telemetry_path)
Amin Hassanid4e35392019-10-03 18:02:441091 raise DevServerError(str(e))
Simran Basi4baad082013-02-14 21:39:181092
1093 # By default all the tarballs extract to test_src but some parts of
1094 # the telemetry code specifically hardcoded to exist inside of 'src'.
1095 test_src = os.path.join(telemetry_path, 'test_src')
1096 try:
1097 shutil.move(test_src, src_folder)
1098 except shutil.Error:
1099 # This can occur if src_folder already exists. Remove and retry move.
1100 shutil.rmtree(src_folder)
Amin Hassanid4e35392019-10-03 18:02:441101 raise DevServerError(
Gabe Black3b567202015-09-23 21:07:591102 'Failure in telemetry setup for build %s. Appears that the '
1103 'test_src to src move failed.' % dl.GetBuild())
Simran Basi4baad082013-02-14 21:39:181104
1105 return src_folder
1106
1107 @cherrypy.expose
Chris Sosa76e44b92013-01-31 20:11:381108 def symbolicate_dump(self, minidump, **kwargs):
Chris Masone816e38c2012-05-02 19:22:361109 """Symbolicates a minidump using pre-downloaded symbols, returns it.
1110
1111 Callers will need to POST to this URL with a body of MIME-type
1112 "multipart/form-data".
1113 The body should include a single argument, 'minidump', containing the
1114 binary-formatted minidump to symbolicate.
1115
Chris Masone816e38c2012-05-02 19:22:361116 Args:
Chris Sosa76e44b92013-01-31 20:11:381117 archive_url: Google Storage URL for the build.
Chris Masone816e38c2012-05-02 19:22:361118 minidump: The binary minidump file to symbolicate.
1119 """
Chris Sosa76e44b92013-01-31 20:11:381120 # Ensure the symbols have been staged.
Dan Shif08fe492016-10-04 21:39:251121 # Try debug.tar.xz first, then debug.tgz
1122 for artifact in (artifact_info.SYMBOLS_ONLY, artifact_info.SYMBOLS):
1123 kwargs['artifacts'] = artifact
1124 dl = _get_downloader(kwargs)
1125
1126 try:
1127 if self.stage(**kwargs) == 'Success':
1128 break
1129 except build_artifact.ArtifactDownloadError:
1130 continue
1131 else:
Amin Hassanid4e35392019-10-03 18:02:441132 raise DevServerError(
Congbin Guo4132a272019-08-20 19:32:141133 'Failed to stage symbols for %s' % dl.DescribeSource())
Chris Sosa76e44b92013-01-31 20:11:381134
Chris Masone816e38c2012-05-02 19:22:361135 to_return = ''
1136 with tempfile.NamedTemporaryFile() as local:
1137 while True:
1138 data = minidump.file.read(8192)
1139 if not data:
1140 break
1141 local.write(data)
Chris Sosa76e44b92013-01-31 20:11:381142
Chris Masone816e38c2012-05-02 19:22:361143 local.flush()
Chris Sosa76e44b92013-01-31 20:11:381144
Gabe Black3b567202015-09-23 21:07:591145 symbols_directory = os.path.join(dl.GetBuildDir(), 'debug', 'breakpad')
Chris Sosa76e44b92013-01-31 20:11:381146
xixuanab744382017-04-27 17:41:271147 # The location of minidump_stackwalk is defined in chromeos-admin.
Chris Sosa76e44b92013-01-31 20:11:381148 stackwalk = subprocess.Popen(
xixuanab744382017-04-27 17:41:271149 ['/usr/local/bin/minidump_stackwalk', local.name, symbols_directory],
Chris Sosa76e44b92013-01-31 20:11:381150 stdout=subprocess.PIPE, stderr=subprocess.PIPE)
1151
Chris Masone816e38c2012-05-02 19:22:361152 to_return, error_text = stackwalk.communicate()
1153 if stackwalk.returncode != 0:
Amin Hassanid4e35392019-10-03 18:02:441154 raise DevServerError(
Congbin Guo4132a272019-08-20 19:32:141155 "Can't generate stack trace: %s (rc=%d)" % (error_text,
1156 stackwalk.returncode))
Chris Masone816e38c2012-05-02 19:22:361157
1158 return to_return
1159
1160 @cherrypy.expose
Don Garrettf84631a2014-01-08 02:21:261161 def latestbuild(self, **kwargs):
Scott Zawalski16954532012-03-20 19:31:361162 """Return a string representing the latest build for a given target.
1163
1164 Args:
1165 target: The build target, typically a combination of the board and the
1166 type of build e.g. x86-mario-release.
1167 milestone: The milestone to filter builds on. E.g. R16. Optional, if not
1168 provided the latest RXX build will be returned.
Don Garrettf84631a2014-01-08 02:21:261169
Scott Zawalski16954532012-03-20 19:31:361170 Returns:
1171 A string representation of the latest build if one exists, i.e.
1172 R19-1993.0.0-a1-b1480.
1173 An empty string if no latest could be found.
1174 """
Don Garrettf84631a2014-01-08 02:21:261175 if not kwargs:
Scott Zawalski16954532012-03-20 19:31:361176 return _PrintDocStringAsHTML(self.latestbuild)
1177
Don Garrettf84631a2014-01-08 02:21:261178 if 'target' not in kwargs:
Amin Hassanid4e35392019-10-03 18:02:441179 raise common_util.DevServerHTTPError(http_client.INTERNAL_SERVER_ERROR,
Amin Hassani08e42d22019-06-03 07:31:301180 'Error: target= is required!')
Dan Shi61305df2015-10-26 23:52:351181
1182 if _is_android_build_request(kwargs):
1183 branch = kwargs.get('branch', None)
1184 target = kwargs.get('target', None)
1185 if not target or not branch:
Amin Hassanid4e35392019-10-03 18:02:441186 raise DevServerError('Both target and branch must be specified to query'
1187 ' for the latest Android build.')
Dan Shi61305df2015-10-26 23:52:351188 return android_build.BuildAccessor.GetLatestBuildID(target, branch)
1189
Scott Zawalski16954532012-03-20 19:31:361190 try:
Gilad Arnoldc65330c2012-09-20 22:17:481191 return common_util.GetLatestBuildVersion(
Don Garrettf84631a2014-01-08 02:21:261192 updater.static_dir, kwargs['target'],
1193 milestone=kwargs.get('milestone'))
Gilad Arnold17fe03d2012-10-02 17:05:011194 except common_util.CommonUtilError as errmsg:
Amin Hassanid4e35392019-10-03 18:02:441195 raise common_util.DevServerHTTPError(http_client.INTERNAL_SERVER_ERROR,
Amin Hassani08e42d22019-06-03 07:31:301196 str(errmsg))
Scott Zawalski16954532012-03-20 19:31:361197
1198 @cherrypy.expose
xixuan7efd0002016-04-14 22:34:011199 def list_suite_controls(self, **kwargs):
1200 """Return a list of contents of all known control files.
1201
1202 Example URL:
1203 To List all control files' content:
1204 http://dev-server/list_suite_controls?suite_name=bvt&
1205 build=daisy_spring-release/R29-4279.0.0
1206
1207 Args:
1208 build: The build i.e. x86-alex-release/R18-1514.0.0-a1-b1450.
1209 suite_name: List the control files belonging to that suite.
1210
1211 Returns:
Dan Shia1cd6522016-04-18 23:07:211212 A dictionary of all control files's path to its content for given suite.
xixuan7efd0002016-04-14 22:34:011213 """
1214 if not kwargs:
1215 return _PrintDocStringAsHTML(self.controlfiles)
1216
1217 if 'build' not in kwargs:
Amin Hassanid4e35392019-10-03 18:02:441218 raise common_util.DevServerHTTPError(http_client.INTERNAL_SERVER_ERROR,
Amin Hassani08e42d22019-06-03 07:31:301219 'Error: build= is required!')
xixuan7efd0002016-04-14 22:34:011220
1221 if 'suite_name' not in kwargs:
Amin Hassanid4e35392019-10-03 18:02:441222 raise common_util.DevServerHTTPError(http_client.INTERNAL_SERVER_ERROR,
Dan Shia1cd6522016-04-18 23:07:211223 'Error: suite_name= is required!')
xixuan7efd0002016-04-14 22:34:011224
1225 control_file_list = [
1226 line.rstrip() for line in common_util.GetControlFileListForSuite(
1227 updater.static_dir, kwargs['build'],
1228 kwargs['suite_name']).splitlines()]
1229
Dan Shia1cd6522016-04-18 23:07:211230 control_file_content_dict = {}
xixuan7efd0002016-04-14 22:34:011231 for control_path in control_file_list:
Dan Shia1cd6522016-04-18 23:07:211232 control_file_content_dict[control_path] = (common_util.GetControlFile(
xixuan7efd0002016-04-14 22:34:011233 updater.static_dir, kwargs['build'], control_path))
1234
Dan Shia1cd6522016-04-18 23:07:211235 return json.dumps(control_file_content_dict)
xixuan7efd0002016-04-14 22:34:011236
1237 @cherrypy.expose
Don Garrettf84631a2014-01-08 02:21:261238 def controlfiles(self, **kwargs):
Scott Zawalski4647ce62012-01-03 22:17:281239 """Return a control file or a list of all known control files.
1240
1241 Example URL:
1242 To List all control files:
beepsbd337242013-07-10 05:44:061243 http://dev-server/controlfiles?suite_name=&build=daisy_spring-release/R29-4279.0.0
1244 To List all control files for, say, the bvt suite:
1245 http://dev-server/controlfiles?suite_name=bvt&build=daisy_spring-release/R29-4279.0.0
Scott Zawalski4647ce62012-01-03 22:17:281246 To return the contents of a path:
Scott Zawalski84a39c92012-01-13 20:12:421247 http://dev-server/controlfiles?board=x86-alex-release&build=R18-1514.0.0&control_path=client/sleeptest/control
Scott Zawalski4647ce62012-01-03 22:17:281248
1249 Args:
Scott Zawalski84a39c92012-01-13 20:12:421250 build: The build i.e. x86-alex-release/R18-1514.0.0-a1-b1450.
Scott Zawalski4647ce62012-01-03 22:17:281251 control_path: If you want the contents of a control file set this
1252 to the path. E.g. client/site_tests/sleeptest/control
1253 Optional, if not provided return a list of control files is returned.
beepsbd337242013-07-10 05:44:061254 suite_name: If control_path is not specified but a suite_name is
1255 specified, list the control files belonging to that suite instead of
1256 all control files. The empty string for suite_name will list all control
1257 files for the build.
Don Garrettf84631a2014-01-08 02:21:261258
Scott Zawalski4647ce62012-01-03 22:17:281259 Returns:
1260 Contents of a control file if control_path is provided.
1261 A list of control files if no control_path is provided.
1262 """
Don Garrettf84631a2014-01-08 02:21:261263 if not kwargs:
Scott Zawalski4647ce62012-01-03 22:17:281264 return _PrintDocStringAsHTML(self.controlfiles)
1265
Don Garrettf84631a2014-01-08 02:21:261266 if 'build' not in kwargs:
Amin Hassanid4e35392019-10-03 18:02:441267 raise common_util.DevServerHTTPError(http_client.INTERNAL_SERVER_ERROR,
Amin Hassani08e42d22019-06-03 07:31:301268 'Error: build= is required!')
Scott Zawalski4647ce62012-01-03 22:17:281269
Don Garrettf84631a2014-01-08 02:21:261270 if 'control_path' not in kwargs:
1271 if 'suite_name' in kwargs and kwargs['suite_name']:
beepsbd337242013-07-10 05:44:061272 return common_util.GetControlFileListForSuite(
Don Garrettf84631a2014-01-08 02:21:261273 updater.static_dir, kwargs['build'], kwargs['suite_name'])
beepsbd337242013-07-10 05:44:061274 else:
1275 return common_util.GetControlFileList(
Don Garrettf84631a2014-01-08 02:21:261276 updater.static_dir, kwargs['build'])
Scott Zawalski4647ce62012-01-03 22:17:281277 else:
Gilad Arnoldc65330c2012-09-20 22:17:481278 return common_util.GetControlFile(
Don Garrettf84631a2014-01-08 02:21:261279 updater.static_dir, kwargs['build'], kwargs['control_path'])
Frank Farzan40160872011-12-13 02:39:181280
1281 @cherrypy.expose
Simran Basi99e63c02014-05-20 17:39:521282 def xbuddy_translate(self, *args, **kwargs):
Yu-Ju Hong1bdb7a92014-04-10 23:02:111283 """Translates an xBuddy path to a real path to artifact if it exists.
1284
1285 Args:
Simran Basi99e63c02014-05-20 17:39:521286 args: An xbuddy path in the form of {local|remote}/build_id/artifact.
1287 Local searches the devserver's static directory. Remote searches a
1288 Google Storage image archive.
1289
1290 Kwargs:
1291 image_dir: Google Storage image archive to search in if requesting a
1292 remote artifact. If none uses the default bucket.
Yu-Ju Hong1bdb7a92014-04-10 23:02:111293
1294 Returns:
Simran Basi99e63c02014-05-20 17:39:521295 String in the format of build_id/artifact as stored on the local server
1296 or in Google Storage.
Yu-Ju Hong1bdb7a92014-04-10 23:02:111297 """
Simran Basi99e63c02014-05-20 17:39:521298 build_id, filename = self._xbuddy.Translate(
Gabe Black3b567202015-09-23 21:07:591299 args, image_dir=kwargs.get('image_dir'))
Yu-Ju Hong1bdb7a92014-04-10 23:02:111300 response = os.path.join(build_id, filename)
1301 _Log('Path translation requested, returning: %s', response)
1302 return response
1303
1304 @cherrypy.expose
joycheneaf4cfc2013-07-02 15:38:571305 def xbuddy(self, *args, **kwargs):
1306 """The full xBuddy call, returns resource specified by path_parts.
joychen3cb228e2013-06-12 19:13:131307
1308 Args:
joycheneaf4cfc2013-07-02 15:38:571309 path_parts: the path following xbuddy/ in the call url is split into the
joychen121fc9b2013-08-02 21:30:301310 components of the path. The path can be understood as
1311 "{local|remote}/build_id/artifact" where build_id is composed of
1312 "board/version."
joycheneaf4cfc2013-07-02 15:38:571313
joychen121fc9b2013-08-02 21:30:301314 The first path element is optional, and can be "remote" or "local"
1315 If local (the default), devserver will not attempt to access Google
1316 Storage, and will only search the static directory for the files.
1317 If remote, devserver will try to obtain the artifact off GS if it's
1318 not found locally.
1319 The board is the familiar board name, optionally suffixed.
1320 The version can be the google storage version number, and may also be
1321 any of a number of xBuddy defined version aliases that will be
1322 translated into the latest built image that fits the description.
1323 Defaults to latest.
1324 The artifact is one of a number of image or artifact aliases used by
1325 xbuddy, defined in xbuddy:ALIASES. Defaults to test.
joycheneaf4cfc2013-07-02 15:38:571326
1327 Kwargs:
Yu-Ju Hong51495eb2013-12-13 01:08:431328 for_update: {true|false}
Amin Hassanie9ffb862019-09-26 00:10:401329 if true, prepares the update payloads for the image,
Yu-Ju Hong51495eb2013-12-13 01:08:431330 and returns the update uri to pass to the
1331 update_engine_client.
joychen3cb228e2013-06-12 19:13:131332 return_dir: {true|false}
1333 if set to true, returns the url to the update.gz
Yu-Ju Hong51495eb2013-12-13 01:08:431334 relative_path: {true|false}
1335 if set to true, returns the relative path to the payload
1336 directory from static_dir.
joychen3cb228e2013-06-12 19:13:131337 Example URL:
joycheneaf4cfc2013-07-02 15:38:571338 http://host:port/xbuddy/x86-generic/R26-4000.0.0/test
joychen3cb228e2013-06-12 19:13:131339 or
joycheneaf4cfc2013-07-02 15:38:571340 http://host:port/xbuddy/x86-generic/R26-4000.0.0/test?return_dir=true
joychen3cb228e2013-06-12 19:13:131341
1342 Returns:
Yu-Ju Hong51495eb2013-12-13 01:08:431343 If |for_update|, returns a redirect to the image or update file
1344 on the devserver. E.g.,
1345 http://host:port/static/archive/x86-generic-release/R26-4000.0.0/
1346 chromium-test-image.bin
1347 If |return_dir|, return a uri to the folder where the artifact is. E.g.,
1348 http://host:port/static/x86-generic-release/R26-4000.0.0/
1349 If |relative_path| is true, return a relative path the folder where the
1350 payloads are. E.g.,
1351 archive/x86-generic-release/R26-4000.0.0
joychen3cb228e2013-06-12 19:13:131352 """
Chris Sosa75490802013-10-01 00:21:451353 boolean_string = kwargs.get('for_update')
1354 for_update = xbuddy.XBuddy.ParseBoolean(boolean_string)
Yu-Ju Hong51495eb2013-12-13 01:08:431355 boolean_string = kwargs.get('return_dir')
1356 return_dir = xbuddy.XBuddy.ParseBoolean(boolean_string)
1357 boolean_string = kwargs.get('relative_path')
1358 relative_path = xbuddy.XBuddy.ParseBoolean(boolean_string)
joychen121fc9b2013-08-02 21:30:301359
Yu-Ju Hong51495eb2013-12-13 01:08:431360 if return_dir and relative_path:
Chris Sosa4b951602014-04-10 03:26:071361 raise common_util.DevServerHTTPError(
Amin Hassanid4e35392019-10-03 18:02:441362 http_client.INTERNAL_SERVER_ERROR,
Amin Hassani08e42d22019-06-03 07:31:301363 'Cannot specify both return_dir and relative_path')
Chris Sosa75490802013-10-01 00:21:451364
1365 # For updates, we optimize downloading of test images.
1366 file_name = None
1367 build_id = None
1368 if for_update:
1369 try:
Yu-Ju Hong1bdb7a92014-04-10 23:02:111370 build_id = self._xbuddy.StageTestArtifactsForUpdate(args)
Chris Sosa75490802013-10-01 00:21:451371 except build_artifact.ArtifactDownloadError:
1372 build_id = None
1373
1374 if not build_id:
1375 build_id, file_name = self._xbuddy.Get(args)
1376
Yu-Ju Hong51495eb2013-12-13 01:08:431377 if for_update:
Amin Hassanie9ffb862019-09-26 00:10:401378 _Log('Payloads requested.')
Yu-Ju Hong51495eb2013-12-13 01:08:431379 # Forces payload to be in cache and symlinked into build_id dir.
Amin Hassanie9ffb862019-09-26 00:10:401380 updater.GetUpdateForLabel(build_id)
Yu-Ju Hong51495eb2013-12-13 01:08:431381
1382 response = None
1383 if return_dir:
1384 response = os.path.join(cherrypy.request.base, 'static', build_id)
1385 _Log('Directory requested, returning: %s', response)
1386 elif relative_path:
1387 response = build_id
1388 _Log('Relative path requested, returning: %s', response)
1389 elif for_update:
1390 response = os.path.join(cherrypy.request.base, 'update', build_id)
1391 _Log('Update URI requested, returning: %s', response)
joychen3cb228e2013-06-12 19:13:131392 else:
Yu-Ju Hong51495eb2013-12-13 01:08:431393 # Redirect to download the payload if no kwargs are set.
joychen121fc9b2013-08-02 21:30:301394 build_id = '/' + os.path.join('static', build_id, file_name)
Yu-Ju Hong51495eb2013-12-13 01:08:431395 _Log('Payload requested, returning: %s', build_id)
joychen121fc9b2013-08-02 21:30:301396 raise cherrypy.HTTPRedirect(build_id, 302)
joychen3cb228e2013-06-12 19:13:131397
Yu-Ju Hong51495eb2013-12-13 01:08:431398 return response
1399
joychen3cb228e2013-06-12 19:13:131400 @cherrypy.expose
1401 def xbuddy_list(self):
1402 """Lists the currently available images & time since last access.
1403
Gilad Arnold452fd272014-02-04 19:09:281404 Returns:
1405 A string representation of a list of tuples [(build_id, time since last
1406 access),...]
joychen3cb228e2013-06-12 19:13:131407 """
1408 return self._xbuddy.List()
1409
1410 @cherrypy.expose
1411 def xbuddy_capacity(self):
Gilad Arnold452fd272014-02-04 19:09:281412 """Returns the number of images cached by xBuddy."""
joychen3cb228e2013-06-12 19:13:131413 return self._xbuddy.Capacity()
1414
1415 @cherrypy.expose
Chris Sosa7c931362010-10-12 02:49:011416 def index(self):
Gilad Arnoldf8f769f2012-09-24 15:43:011417 """Presents a welcome message and documentation links."""
Congbin Guo6bc32182019-08-21 00:54:301418 html_template = (
1419 'Welcome to the Dev Server!<br>\n'
1420 '<br>\n'
1421 'Here are the available methods, click for documentation:<br>\n'
1422 '<br>\n'
1423 '%s')
1424
1425 exposed_methods = []
1426 for app in cherrypy.tree.apps.values():
1427 exposed_methods += _FindExposedMethods(
1428 app.root, app.script_name.lstrip('/'),
1429 unlisted=self._UNLISTED_METHODS)
1430
1431 return html_template % '<br>\n'.join(
1432 ['<a href=doc/%s>%s</a>' % (name, name)
1433 for name in sorted(exposed_methods)])
Gilad Arnoldf8f769f2012-09-24 15:43:011434
1435 @cherrypy.expose
1436 def doc(self, *args):
1437 """Shows the documentation for available methods / URLs.
1438
Amin Hassani08e42d22019-06-03 07:31:301439 Examples:
Gilad Arnoldf8f769f2012-09-24 15:43:011440 http://myhost/doc/update
1441 """
Gilad Arnoldd5ebaaa2012-10-02 18:52:381442 name = '/'.join(args)
Congbin Guo6bc32182019-08-21 00:54:301443 method = _GetExposedMethod(name)
Gilad Arnoldf8f769f2012-09-24 15:43:011444 if not method:
Amin Hassanid4e35392019-10-03 18:02:441445 raise DevServerError("No exposed method named `%s'" % name)
Gilad Arnoldf8f769f2012-09-24 15:43:011446 if not method.__doc__:
Amin Hassanid4e35392019-10-03 18:02:441447 raise DevServerError("No documentation for exposed method `%s'" % name)
Gilad Arnoldf8f769f2012-09-24 15:43:011448 return '<pre>\n%s</pre>' % method.__doc__
Chris Sosa7c931362010-10-12 02:49:011449
Dale Curtisc9aaf3a2011-08-09 22:47:401450 @cherrypy.expose
Chris Sosa7c931362010-10-12 02:49:011451 def update(self, *args):
Gilad Arnoldf8f769f2012-09-24 15:43:011452 """Handles an update check from a Chrome OS client.
1453
1454 The HTTP request should contain the standard Omaha-style XML blob. The URL
1455 line may contain an additional intermediate path to the update payload.
1456
joychen121fc9b2013-08-02 21:30:301457 This request can be handled in one of 4 ways, depending on the devsever
1458 settings and intermediate path.
joychenb0dfe552013-07-30 17:02:061459
Amin Hassanie9ffb862019-09-26 00:10:401460 1. No intermediate path. DEPRECATED
joychen121fc9b2013-08-02 21:30:301461
1462 2. Path explicitly invokes XBuddy
1463 If there is a path given, it can explicitly invoke xbuddy by prefixing it
1464 with 'xbuddy'. This path is then used to acquire an image binary for the
1465 devserver to generate an update payload from. Devserver then serves this
1466 payload.
1467
1468 3. Path is left for the devserver to interpret.
1469 If the path given doesn't explicitly invoke xbuddy, devserver will attempt
1470 to generate a payload from the test image in that directory and serve it.
1471
joychen121fc9b2013-08-02 21:30:301472 Examples:
joychen121fc9b2013-08-02 21:30:301473 2. Explicitly invoke xbuddy
1474 update_engine_client --omaha_url=
1475 http://myhost/update/xbuddy/remote/board/version/dev
1476 This would go to GS to download the dev image for the board, from which
1477 the devserver would generate a payload to serve.
1478
1479 3. Give a path for devserver to interpret
1480 update_engine_client --omaha_url=http://myhost/update/some/random/path
1481 This would attempt, in order to:
1482 a) Generate an update from a test image binary if found in
1483 static_dir/some/random/path.
1484 b) Serve an update payload found in static_dir/some/random/path.
1485 c) Hope that some/random/path takes the form "board/version" and
1486 and attempt to download an update payload for that board/version
1487 from GS.
Gilad Arnoldf8f769f2012-09-24 15:43:011488 """
joychen121fc9b2013-08-02 21:30:301489 label = '/'.join(args)
Gilad Arnold286a0062012-01-12 21:47:021490 body_length = int(cherrypy.request.headers.get('Content-Length', 0))
Chris Sosa7c931362010-10-12 02:49:011491 data = cherrypy.request.rfile.read(body_length)
Chris Sosa7c931362010-10-12 02:49:011492
joychen121fc9b2013-08-02 21:30:301493 return updater.HandleUpdatePing(data, label)
Chris Sosa0356d3b2010-09-16 22:46:221494
Dan Shif5ce2de2013-04-25 23:06:321495
Chris Sosadbc20082012-12-10 21:39:111496def _CleanCache(cache_dir, wipe):
1497 """Wipes any excess cached items in the cache_dir.
1498
1499 Args:
1500 cache_dir: the directory we are wiping from.
1501 wipe: If True, wipe all the contents -- not just the excess.
1502 """
1503 if wipe:
1504 # Clear the cache and exit on error.
1505 cmd = 'rm -rf %s/*' % cache_dir
1506 if os.system(cmd) != 0:
1507 _Log('Failed to clear the cache with %s' % cmd)
1508 sys.exit(1)
1509 else:
1510 # Clear all but the last N cached updates
1511 cmd = ('cd %s; ls -tr | head --lines=-%d | xargs rm -rf' %
1512 (cache_dir, CACHED_ENTRIES))
1513 if os.system(cmd) != 0:
1514 _Log('Failed to clean up old delta cache files with %s' % cmd)
1515 sys.exit(1)
1516
1517
Chris Sosa3ae4dc12013-03-29 18:47:001518def _AddTestingOptions(parser):
1519 group = optparse.OptionGroup(
1520 parser, 'Advanced Testing Options', 'These are used by test scripts and '
1521 'developers writing integration tests utilizing the devserver. They are '
1522 'not intended to be really used outside the scope of someone '
1523 'knowledgable about the test.')
1524 group.add_option('--exit',
1525 action='store_true',
Amin Hassanie9ffb862019-09-26 00:10:401526 help='do not start the server (yet clear cache)')
Chris Sosa3ae4dc12013-03-29 18:47:001527 group.add_option('--host_log',
1528 action='store_true', default=False,
1529 help='record history of host update events (/api/hostlog)')
1530 group.add_option('--max_updates',
Gabe Black3b567202015-09-23 21:07:591531 metavar='NUM', default=-1, type='int',
Chris Sosa3ae4dc12013-03-29 18:47:001532 help='maximum number of update checks handled positively '
1533 '(default: unlimited)')
Chris Sosa3ae4dc12013-03-29 18:47:001534 group.add_option('--proxy_port',
1535 metavar='PORT', default=None, type='int',
1536 help='port to have the client connect to -- basically the '
1537 'devserver lies to the update to tell it to get the payload '
1538 'from a different port that will proxy the request back to '
1539 'the devserver. The proxy must be managed outside the '
1540 'devserver.')
Chris Sosa3ae4dc12013-03-29 18:47:001541 parser.add_option_group(group)
1542
1543
1544def _AddUpdateOptions(parser):
1545 group = optparse.OptionGroup(
1546 parser, 'Autoupdate Options', 'These options can be used to change '
Amin Hassanie9ffb862019-09-26 00:10:401547 'how the devserver serve update payloads. Please '
Chris Sosa3ae4dc12013-03-29 18:47:001548 'note that all of these option affect how a payload is generated and so '
1549 'do not work in archive-only mode.')
Chris Sosa3ae4dc12013-03-29 18:47:001550 group.add_option('--critical_update',
1551 action='store_true', default=False,
1552 help='Present update payload as critical')
Chris Sosa3ae4dc12013-03-29 18:47:001553 group.add_option('--payload',
1554 metavar='PATH',
1555 help='use the update payload from specified directory '
1556 '(update.gz).')
Chris Sosa3ae4dc12013-03-29 18:47:001557 parser.add_option_group(group)
1558
1559
1560def _AddProductionOptions(parser):
1561 group = optparse.OptionGroup(
1562 parser, 'Advanced Server Options', 'These options can be used to changed '
1563 'for advanced server behavior.')
Chris Sosa3ae4dc12013-03-29 18:47:001564 group.add_option('--clear_cache',
1565 action='store_true', default=False,
1566 help='At startup, removes all cached entries from the'
Amin Hassanid4e35392019-10-03 18:02:441567 "devserver's cache.")
Chris Sosa3ae4dc12013-03-29 18:47:001568 group.add_option('--logfile',
1569 metavar='PATH',
1570 help='log output to this file instead of stdout')
Chris Sosa855b8932013-08-21 20:24:551571 group.add_option('--pidfile',
1572 metavar='PATH',
1573 help='path to output a pid file for the server.')
Gilad Arnold11fbef42014-02-10 19:04:131574 group.add_option('--portfile',
1575 metavar='PATH',
1576 help='path to output the port number being served on.')
Chris Sosa3ae4dc12013-03-29 18:47:001577 group.add_option('--production',
1578 action='store_true', default=False,
1579 help='have the devserver use production values when '
1580 'starting up. This includes using more threads and '
1581 'performing less logging.')
1582 parser.add_option_group(group)
1583
1584
Paul Hobbsef4e0702016-06-28 00:01:421585def MakeLogHandler(logfile):
J. Richard Barnette3d977b82013-04-23 18:05:191586 """Create a LogHandler instance used to log all messages."""
1587 hdlr_cls = handlers.TimedRotatingFileHandler
1588 hdlr = hdlr_cls(logfile, when=_LOG_ROTATION_TIME,
xixuan3d48bff2017-01-31 03:00:091589 interval=_LOG_ROTATION_INTERVAL,
J. Richard Barnette3d977b82013-04-23 18:05:191590 backupCount=_LOG_ROTATION_BACKUP)
Chris Sosa855b8932013-08-21 20:24:551591 hdlr.setFormatter(cplogging.logfmt)
J. Richard Barnette3d977b82013-04-23 18:05:191592 return hdlr
1593
1594
Chris Sosacde6bf42012-06-01 01:36:391595def main():
Chris Sosa3ae4dc12013-03-29 18:47:001596 usage = '\n\n'.join(['usage: %prog [options]', __doc__])
Gilad Arnold286a0062012-01-12 21:47:021597 parser = optparse.OptionParser(usage=usage)
joychened64b222013-06-21 23:39:341598
1599 # get directory that the devserver is run from
1600 devserver_dir = os.path.dirname(os.path.abspath(sys.argv[0]))
joychen84d13772013-08-06 16:17:231601 default_static_dir = '%s/static' % devserver_dir
joychened64b222013-06-21 23:39:341602 parser.add_option('--static_dir',
Gilad Arnold9714d9b2012-10-04 17:09:421603 metavar='PATH',
joychen84d13772013-08-06 16:17:231604 default=default_static_dir,
joychened64b222013-06-21 23:39:341605 help='writable static directory')
Gilad Arnold9714d9b2012-10-04 17:09:421606 parser.add_option('--port',
1607 default=8080, type='int',
Gilad Arnoldaf696d12014-02-14 21:13:281608 help=('port for the dev server to use; if zero, binds to '
1609 'an arbitrary available port (default: 8080)'))
Gilad Arnold9714d9b2012-10-04 17:09:421610 parser.add_option('-t', '--test_image',
1611 action='store_true',
joychen121fc9b2013-08-02 21:30:301612 help='Deprecated.')
joychen5260b9a2013-07-16 21:48:011613 parser.add_option('-x', '--xbuddy_manage_builds',
1614 action='store_true',
1615 default=False,
1616 help='If set, allow xbuddy to manage images in'
1617 'build/images.')
Dan Shi72b16132015-10-08 19:10:331618 parser.add_option('-a', '--android_build_credential',
1619 default=None,
1620 help='Path to a json file which contains the credential '
1621 'needed to access Android builds.')
Chris Sosa3ae4dc12013-03-29 18:47:001622 _AddProductionOptions(parser)
1623 _AddUpdateOptions(parser)
1624 _AddTestingOptions(parser)
Chris Sosa7c931362010-10-12 02:49:011625 (options, _) = parser.parse_args()
[email protected]21a5ca32009-11-04 18:23:231626
J. Richard Barnette3d977b82013-04-23 18:05:191627 # Handle options that must be set globally in cherrypy. Do this
1628 # work up front, because calls to _Log() below depend on this
1629 # initialization.
1630 if options.production:
1631 cherrypy.config.update({'environment': 'production'})
1632 if not options.logfile:
1633 cherrypy.config.update({'log.screen': True})
1634 else:
1635 cherrypy.config.update({'log.error_file': '',
1636 'log.access_file': ''})
Paul Hobbsef4e0702016-06-28 00:01:421637 hdlr = MakeLogHandler(options.logfile)
J. Richard Barnette3d977b82013-04-23 18:05:191638 # Pylint can't seem to process these two calls properly
1639 # pylint: disable=E1101
1640 cherrypy.log.access_log.addHandler(hdlr)
1641 cherrypy.log.error_log.addHandler(hdlr)
1642 # pylint: enable=E1101
1643
joychened64b222013-06-21 23:39:341644 # set static_dir, from which everything will be served
joychen84d13772013-08-06 16:17:231645 options.static_dir = os.path.realpath(options.static_dir)
Chris Sosa0356d3b2010-09-16 22:46:221646
joychened64b222013-06-21 23:39:341647 cache_dir = os.path.join(options.static_dir, 'cache')
J. Richard Barnette3d977b82013-04-23 18:05:191648 # If our devserver is only supposed to serve payloads, we shouldn't be
1649 # mucking with the cache at all. If the devserver hadn't previously
1650 # generated a cache and is expected, the caller is using it wrong.
joychen7c2054a2013-07-25 18:14:071651 if os.path.exists(cache_dir):
Chris Sosadbc20082012-12-10 21:39:111652 _CleanCache(cache_dir, options.clear_cache)
Chris Sosa6b8c3742011-01-31 20:12:171653 else:
1654 os.makedirs(cache_dir)
Don Garrettf90edf02010-11-17 01:36:141655
Chris Sosadbc20082012-12-10 21:39:111656 _Log('Using cache directory %s' % cache_dir)
joychened64b222013-06-21 23:39:341657 _Log('Serving from %s' % options.static_dir)
[email protected]21a5ca32009-11-04 18:23:231658
Amin Hassanie9ffb862019-09-26 00:10:401659 _xbuddy = xbuddy.XBuddy(manage_builds=options.xbuddy_manage_builds,
joychen121fc9b2013-08-02 21:30:301660 static_dir=options.static_dir)
Chris Sosa75490802013-10-01 00:21:451661 if options.clear_cache and options.xbuddy_manage_builds:
1662 _xbuddy.CleanCache()
joychen121fc9b2013-08-02 21:30:301663
Chris Sosa6a3697f2013-01-30 00:44:431664 # We allow global use here to share with cherrypy classes.
1665 # pylint: disable=W0603
Chris Sosacde6bf42012-06-01 01:36:391666 global updater
Andrew de los Reyes52620802010-04-12 20:40:071667 updater = autoupdate.Autoupdate(
joychen121fc9b2013-08-02 21:30:301668 _xbuddy,
joychened64b222013-06-21 23:39:341669 static_dir=options.static_dir,
Gilad Arnold0c9c8602012-10-03 06:58:581670 payload_path=options.payload,
Don Garrett0ad09372010-12-07 00:20:301671 proxy_port=options.proxy_port,
Satoru Takabayashid733cbe2011-11-15 17:36:321672 critical_update=options.critical_update,
Gilad Arnolda564b4b2012-10-04 17:32:441673 max_updates=options.max_updates,
Gilad Arnold8318eac2012-10-04 19:52:231674 host_log=options.host_log,
Chris Sosa0f1ec842011-02-15 00:33:221675 )
Chris Sosa7c931362010-10-12 02:49:011676
J. Richard Barnette3d977b82013-04-23 18:05:191677 if options.exit:
1678 return
Chris Sosa2f1c41e2012-07-10 21:32:331679
joychen3cb228e2013-06-12 19:13:131680 dev_server = DevServerRoot(_xbuddy)
Congbin Guo3afae6c2019-08-13 23:29:421681 health_checker_app = health_checker.Root(dev_server, options.static_dir)
joychen3cb228e2013-06-12 19:13:131682
Chris Sosa855b8932013-08-21 20:24:551683 if options.pidfile:
1684 plugins.PIDFile(cherrypy.engine, options.pidfile).subscribe()
1685
Gilad Arnold11fbef42014-02-10 19:04:131686 if options.portfile:
1687 cherrypy_ext.PortFile(cherrypy.engine, options.portfile).subscribe()
1688
Dan Shiafd5c6c2016-01-07 18:27:031689 if (options.android_build_credential and
1690 os.path.exists(options.android_build_credential)):
1691 try:
1692 with open(options.android_build_credential) as f:
1693 android_build.BuildAccessor.credential_info = json.load(f)
1694 except ValueError as e:
1695 _Log('Failed to load the android build credential: %s. Error: %s.' %
1696 (options.android_build_credential, e))
Congbin Guo3afae6c2019-08-13 23:29:421697
1698 cherrypy.tree.mount(health_checker_app, '/check_health',
1699 config=health_checker.get_config())
joychen3cb228e2013-06-12 19:13:131700 cherrypy.quickstart(dev_server, config=_GetConfig(options))
Chris Sosacde6bf42012-06-01 01:36:391701
1702
1703if __name__ == '__main__':
1704 main()