David Riley | 2fcb012 | 2017-11-02 11:25:39 -0700 | [diff] [blame] | 1 | #!/usr/bin/env python2 |
Luis Hector Chavez | dca9dd7 | 2018-06-12 12:56:30 -0700 | [diff] [blame] | 2 | # -*- coding: utf-8 -*- |
Chris Sosa | 781ba6d | 2012-04-11 12:44:43 -0700 | [diff] [blame] | 3 | # Copyright (c) 2009-2012 The Chromium OS Authors. All rights reserved. |
rtc@google.com | ded2240 | 2009-10-26 22:36:21 +0000 | [diff] [blame] | 4 | # Use of this source code is governed by a BSD-style license that can be |
| 5 | # found in the LICENSE file. |
| 6 | |
Chris Sosa | 3ae4dc1 | 2013-03-29 11:47:00 -0700 | [diff] [blame] | 7 | """Chromium OS development server that can be used for all forms of update. |
| 8 | |
| 9 | This devserver can be used to perform system-wide autoupdate and update |
| 10 | of specific portage packages on devices running Chromium OS derived operating |
Amin Hassani | e9ffb86 | 2019-09-25 17:10:40 -0700 | [diff] [blame] | 11 | systems. |
Chris Sosa | 3ae4dc1 | 2013-03-29 11:47:00 -0700 | [diff] [blame] | 12 | |
Amin Hassani | e9ffb86 | 2019-09-25 17:10:40 -0700 | [diff] [blame] | 13 | The devserver is configured to stage and |
Chris Sosa | 3ae4dc1 | 2013-03-29 11:47:00 -0700 | [diff] [blame] | 14 | serve artifacts from Google Storage using the credentials provided to it before |
| 15 | it is run. The easiest way to understand this is that the devserver is |
| 16 | functioning as a local cache for artifacts produced and uploaded by build |
| 17 | servers. Users of this form of devserver can either download the artifacts |
| 18 | from the devservers static directory OR use the update RPC to perform a |
| 19 | system-wide autoupdate. Archive mode is always active. |
| 20 | |
Chris Sosa | 3ae4dc1 | 2013-03-29 11:47:00 -0700 | [diff] [blame] | 21 | For autoupdates, there are many more advanced options that can help specify |
| 22 | how to update and which payload to give to a requester. |
| 23 | """ |
| 24 | |
Gabe Black | 3b56720 | 2015-09-23 14:07:59 -0700 | [diff] [blame] | 25 | from __future__ import print_function |
Chris Sosa | 7c93136 | 2010-10-11 19:49:01 -0700 | [diff] [blame] | 26 | |
Amin Hassani | c5af426 | 2019-11-13 13:37:20 -0800 | [diff] [blame] | 27 | import distutils.version # pylint: disable=import-error,no-name-in-module |
Gilad Arnold | 55a2a37 | 2012-10-02 09:46:32 -0700 | [diff] [blame] | 28 | import json |
David Riley | 2fcb012 | 2017-11-02 11:25:39 -0700 | [diff] [blame] | 29 | import optparse # pylint: disable=deprecated-module |
rtc@google.com | ded2240 | 2009-10-26 22:36:21 +0000 | [diff] [blame] | 30 | import os |
Scott Zawalski | 4647ce6 | 2012-01-03 17:17:28 -0500 | [diff] [blame] | 31 | import re |
Simran Basi | 4baad08 | 2013-02-14 13:39:18 -0800 | [diff] [blame] | 32 | import shutil |
xixuan | 52c2fba | 2016-05-20 17:02:48 -0700 | [diff] [blame] | 33 | import signal |
Mandeep Singh Baines | 38dcdda | 2012-12-07 17:55:33 -0800 | [diff] [blame] | 34 | import socket |
Chris Masone | 816e38c | 2012-05-02 12:22:36 -0700 | [diff] [blame] | 35 | import subprocess |
J. Richard Barnette | 3d977b8 | 2013-04-23 11:05:19 -0700 | [diff] [blame] | 36 | import sys |
Chris Masone | 816e38c | 2012-05-02 12:22:36 -0700 | [diff] [blame] | 37 | import tempfile |
Dan Shi | 59ae709 | 2013-06-04 14:37:27 -0700 | [diff] [blame] | 38 | import threading |
Gilad Arnold | d5ebaaa | 2012-10-02 11:52:38 -0700 | [diff] [blame] | 39 | import types |
J. Richard Barnette | 3d977b8 | 2013-04-23 11:05:19 -0700 | [diff] [blame] | 40 | from logging import handlers |
| 41 | |
Amin Hassani | d4e3539 | 2019-10-03 11:02:44 -0700 | [diff] [blame] | 42 | from six.moves import http_client |
| 43 | |
| 44 | # pylint: disable=no-name-in-module, import-error |
J. Richard Barnette | 3d977b8 | 2013-04-23 11:05:19 -0700 | [diff] [blame] | 45 | import cherrypy |
Chris Sosa | 855b893 | 2013-08-21 13:24:55 -0700 | [diff] [blame] | 46 | from cherrypy import _cplogging as cplogging |
Amin Hassani | d4e3539 | 2019-10-03 11:02:44 -0700 | [diff] [blame] | 47 | from cherrypy.process import plugins |
| 48 | # pylint: enable=no-name-in-module, import-error |
rtc@google.com | ded2240 | 2009-10-26 22:36:21 +0000 | [diff] [blame] | 49 | |
Achuith Bhandarkar | 662fb72 | 2019-10-31 16:12:49 -0700 | [diff] [blame] | 50 | import autoupdate |
| 51 | import cherrypy_ext |
Achuith Bhandarkar | 662fb72 | 2019-10-31 16:12:49 -0700 | [diff] [blame] | 52 | import health_checker |
| 53 | |
Richard Barnette | df35c32 | 2017-08-18 17:02:13 -0700 | [diff] [blame] | 54 | # This must happen before any local modules get a chance to import |
| 55 | # anything from chromite. Otherwise, really bad things will happen, and |
| 56 | # you will _not_ understand why. |
Congbin Guo | 3afae6c | 2019-08-13 16:29:42 -0700 | [diff] [blame] | 57 | import setup_chromite # pylint: disable=unused-import |
Amin Hassani | e427e21 | 2019-10-28 11:04:27 -0700 | [diff] [blame] | 58 | from chromite.lib import cros_update_progress |
Achuith Bhandarkar | 662fb72 | 2019-10-31 16:12:49 -0700 | [diff] [blame] | 59 | from chromite.lib.xbuddy import android_build |
| 60 | from chromite.lib.xbuddy import artifact_info |
| 61 | from chromite.lib.xbuddy import build_artifact |
| 62 | from chromite.lib.xbuddy import cherrypy_log_util |
| 63 | from chromite.lib.xbuddy import common_util |
| 64 | from chromite.lib.xbuddy import devserver_constants |
| 65 | from chromite.lib.xbuddy import downloader |
| 66 | from chromite.lib.xbuddy import xbuddy |
Gilad Arnold | c65330c | 2012-09-20 15:17:48 -0700 | [diff] [blame] | 67 | |
Gilad Arnold | c65330c | 2012-09-20 15:17:48 -0700 | [diff] [blame] | 68 | # Module-local log function. |
Chris Sosa | 6a3697f | 2013-01-29 16:44:43 -0800 | [diff] [blame] | 69 | def _Log(message, *args): |
Achuith Bhandarkar | 662fb72 | 2019-10-31 16:12:49 -0700 | [diff] [blame] | 70 | return cherrypy_log_util.LogWithTag('DEVSERVER', message, *args) |
Frank Farzan | 4016087 | 2011-12-12 18:39:18 -0800 | [diff] [blame] | 71 | |
Chris Sosa | 417e55d | 2011-01-25 16:40:48 -0800 | [diff] [blame] | 72 | CACHED_ENTRIES = 12 |
Don Garrett | f90edf0 | 2010-11-16 17:36:14 -0800 | [diff] [blame] | 73 | |
Simran Basi | 4baad08 | 2013-02-14 13:39:18 -0800 | [diff] [blame] | 74 | TELEMETRY_FOLDER = 'telemetry_src' |
| 75 | TELEMETRY_DEPS = ['dep-telemetry_dep.tar.bz2', |
| 76 | 'dep-page_cycler_dep.tar.bz2', |
Simran Basi | 0d07868 | 2013-03-22 16:40:04 -0700 | [diff] [blame] | 77 | 'dep-chrome_test.tar.bz2', |
| 78 | 'dep-perf_data_dep.tar.bz2'] |
Simran Basi | 4baad08 | 2013-02-14 13:39:18 -0800 | [diff] [blame] | 79 | |
Chris Sosa | 0356d3b | 2010-09-16 15:46:22 -0700 | [diff] [blame] | 80 | # Sets up global to share between classes. |
rtc@google.com | 21a5ca3 | 2009-11-04 18:23:23 +0000 | [diff] [blame] | 81 | updater = None |
rtc@google.com | ded2240 | 2009-10-26 22:36:21 +0000 | [diff] [blame] | 82 | |
xixuan | 3d48bff | 2017-01-30 19:00:09 -0800 | [diff] [blame] | 83 | # Log rotation parameters. These settings correspond to twice a day once |
| 84 | # devserver is started, with about two weeks (28 backup files) of old logs |
| 85 | # kept for backup. |
J. Richard Barnette | 3d977b8 | 2013-04-23 11:05:19 -0700 | [diff] [blame] | 86 | # |
xixuan | 3d48bff | 2017-01-30 19:00:09 -0800 | [diff] [blame] | 87 | # For more, see the documentation in standard python library for |
J. Richard Barnette | 3d977b8 | 2013-04-23 11:05:19 -0700 | [diff] [blame] | 88 | # logging.handlers.TimedRotatingFileHandler |
xixuan | 3d48bff | 2017-01-30 19:00:09 -0800 | [diff] [blame] | 89 | _LOG_ROTATION_TIME = 'H' |
Congbin Guo | 3afae6c | 2019-08-13 16:29:42 -0700 | [diff] [blame] | 90 | _LOG_ROTATION_INTERVAL = 12 # hours |
| 91 | _LOG_ROTATION_BACKUP = 28 # backup counts |
Frank Farzan | 4016087 | 2011-12-12 18:39:18 -0800 | [diff] [blame] | 92 | |
xixuan | 52c2fba | 2016-05-20 17:02:48 -0700 | [diff] [blame] | 93 | # Auto-update parameters |
| 94 | |
| 95 | # Error msg for missing key in CrOS auto-update. |
Xixuan Wu | 32af9f1 | 2017-11-13 14:11:44 -0800 | [diff] [blame] | 96 | KEY_ERROR_MSG = 'Key Error in RPC: %s= is required' |
xixuan | 52c2fba | 2016-05-20 17:02:48 -0700 | [diff] [blame] | 97 | |
Sanika Kulkarni | d4496fd | 2020-02-04 17:26:25 -0800 | [diff] [blame] | 98 | # Error msg for deprecated RPC usage. |
| 99 | DEPRECATED_RPC_ERROR_MSG = ('The %s RPC has been deprecated. Usage of this ' |
| 100 | 'RPC is discouraged. Please go to ' |
| 101 | 'go/devserver-deprecation for more information.') |
| 102 | |
xixuan | 52c2fba | 2016-05-20 17:02:48 -0700 | [diff] [blame] | 103 | |
Amin Hassani | d4e3539 | 2019-10-03 11:02:44 -0700 | [diff] [blame] | 104 | class DevServerError(Exception): |
| 105 | """Exception class used by DevServer.""" |
| 106 | |
| 107 | |
Sanika Kulkarni | d4496fd | 2020-02-04 17:26:25 -0800 | [diff] [blame] | 108 | class DeprecatedRPCError(DevServerError): |
| 109 | """Exception class used when an RPC is deprecated but is still being used.""" |
| 110 | |
| 111 | def __init__(self, rpc_name): |
| 112 | """Constructor for DeprecatedRPCError class. |
| 113 | |
| 114 | :param rpc_name: (str) name of the RPC that has been deprecated. |
| 115 | """ |
| 116 | super(DeprecatedRPCError, self).__init__(DEPRECATED_RPC_ERROR_MSG % rpc_name) |
| 117 | self.rpc_name = rpc_name |
| 118 | |
| 119 | |
Amin Hassani | 722e096 | 2019-11-15 15:45:31 -0800 | [diff] [blame] | 120 | class DevServerHTTPError(cherrypy.HTTPError): |
| 121 | """Exception class to log the HTTPResponse before routing it to cherrypy.""" |
| 122 | def __init__(self, status, message): |
| 123 | """CherryPy error with logging. |
| 124 | |
| 125 | Args: |
| 126 | status: HTTPResponse status. |
| 127 | message: Message associated with the response. |
| 128 | """ |
| 129 | cherrypy.HTTPError.__init__(self, status, message) |
| 130 | _Log('HTTPError status: %s message: %s', status, message) |
| 131 | |
| 132 | |
Gabe Black | 3b56720 | 2015-09-23 14:07:59 -0700 | [diff] [blame] | 133 | def _canonicalize_archive_url(archive_url): |
| 134 | """Canonicalizes archive_url strings. |
| 135 | |
| 136 | Raises: |
| 137 | DevserverError: if archive_url is not set. |
| 138 | """ |
| 139 | if archive_url: |
| 140 | if not archive_url.startswith('gs://'): |
Amin Hassani | d4e3539 | 2019-10-03 11:02:44 -0700 | [diff] [blame] | 141 | raise DevServerError( |
Congbin Guo | 4132a27 | 2019-08-20 12:32:14 -0700 | [diff] [blame] | 142 | "Archive URL isn't from Google Storage (%s) ." % archive_url) |
Gabe Black | 3b56720 | 2015-09-23 14:07:59 -0700 | [diff] [blame] | 143 | |
| 144 | return archive_url.rstrip('/') |
| 145 | else: |
Amin Hassani | d4e3539 | 2019-10-03 11:02:44 -0700 | [diff] [blame] | 146 | raise DevServerError('Must specify an archive_url in the request') |
Gabe Black | 3b56720 | 2015-09-23 14:07:59 -0700 | [diff] [blame] | 147 | |
| 148 | |
| 149 | def _canonicalize_local_path(local_path): |
| 150 | """Canonicalizes |local_path| strings. |
| 151 | |
| 152 | Raises: |
| 153 | DevserverError: if |local_path| is not set. |
| 154 | """ |
| 155 | # Restrict staging of local content to only files within the static |
| 156 | # directory. |
| 157 | local_path = os.path.abspath(local_path) |
| 158 | if not local_path.startswith(updater.static_dir): |
Amin Hassani | d4e3539 | 2019-10-03 11:02:44 -0700 | [diff] [blame] | 159 | raise DevServerError( |
Congbin Guo | 4132a27 | 2019-08-20 12:32:14 -0700 | [diff] [blame] | 160 | 'Local path %s must be a subdirectory of the static' |
| 161 | ' directory: %s' % (local_path, updater.static_dir)) |
Gabe Black | 3b56720 | 2015-09-23 14:07:59 -0700 | [diff] [blame] | 162 | |
| 163 | return local_path.rstrip('/') |
| 164 | |
| 165 | |
| 166 | def _get_artifacts(kwargs): |
| 167 | """Returns a tuple of named and file artifacts given the stage rpc kwargs. |
| 168 | |
| 169 | Raises: |
| 170 | DevserverError if no artifacts would be returned. |
| 171 | """ |
| 172 | artifacts = kwargs.get('artifacts') |
| 173 | files = kwargs.get('files') |
| 174 | if not artifacts and not files: |
Amin Hassani | d4e3539 | 2019-10-03 11:02:44 -0700 | [diff] [blame] | 175 | raise DevServerError('No artifacts specified.') |
Gabe Black | 3b56720 | 2015-09-23 14:07:59 -0700 | [diff] [blame] | 176 | |
| 177 | # Note we NEED to coerce files to a string as we get raw unicode from |
| 178 | # cherrypy and we treat files as strings elsewhere in the code. |
| 179 | return (str(artifacts).split(',') if artifacts else [], |
| 180 | str(files).split(',') if files else []) |
| 181 | |
| 182 | |
Dan Shi | 61305df | 2015-10-26 16:52:35 -0700 | [diff] [blame] | 183 | def _is_android_build_request(kwargs): |
| 184 | """Check if a devserver call is for Android build, based on the arguments. |
| 185 | |
| 186 | This method exams the request's arguments (os_type) to determine if the |
| 187 | request is for Android build. If os_type is set to `android`, returns True. |
| 188 | If os_type is not set or has other values, returns False. |
| 189 | |
| 190 | Args: |
| 191 | kwargs: Keyword arguments for the request. |
| 192 | |
| 193 | Returns: |
| 194 | True if the request is for Android build. False otherwise. |
| 195 | """ |
| 196 | os_type = kwargs.get('os_type', None) |
| 197 | return os_type == 'android' |
| 198 | |
| 199 | |
Gabe Black | 3b56720 | 2015-09-23 14:07:59 -0700 | [diff] [blame] | 200 | def _get_downloader(kwargs): |
| 201 | """Returns the downloader based on passed in arguments. |
| 202 | |
| 203 | Args: |
Amin Hassani | 08e42d2 | 2019-06-03 00:31:30 -0700 | [diff] [blame] | 204 | kwargs: Keyword arguments for the request. |
Gabe Black | 3b56720 | 2015-09-23 14:07:59 -0700 | [diff] [blame] | 205 | """ |
| 206 | local_path = kwargs.get('local_path') |
| 207 | if local_path: |
| 208 | local_path = _canonicalize_local_path(local_path) |
| 209 | |
| 210 | dl = None |
| 211 | if local_path: |
Prathmesh Prabhu | 58d0893 | 2018-01-19 15:08:19 -0800 | [diff] [blame] | 212 | delete_source = _parse_boolean_arg(kwargs, 'delete_source') |
| 213 | dl = downloader.LocalDownloader(updater.static_dir, local_path, |
| 214 | delete_source=delete_source) |
Gabe Black | 3b56720 | 2015-09-23 14:07:59 -0700 | [diff] [blame] | 215 | |
Dan Shi | 61305df | 2015-10-26 16:52:35 -0700 | [diff] [blame] | 216 | if not _is_android_build_request(kwargs): |
Gabe Black | 3b56720 | 2015-09-23 14:07:59 -0700 | [diff] [blame] | 217 | archive_url = kwargs.get('archive_url') |
| 218 | if not archive_url and not local_path: |
Amin Hassani | d4e3539 | 2019-10-03 11:02:44 -0700 | [diff] [blame] | 219 | raise DevServerError( |
Congbin Guo | 4132a27 | 2019-08-20 12:32:14 -0700 | [diff] [blame] | 220 | 'Requires archive_url or local_path to be specified.') |
Gabe Black | 3b56720 | 2015-09-23 14:07:59 -0700 | [diff] [blame] | 221 | if archive_url and local_path: |
Amin Hassani | d4e3539 | 2019-10-03 11:02:44 -0700 | [diff] [blame] | 222 | raise DevServerError( |
Congbin Guo | 4132a27 | 2019-08-20 12:32:14 -0700 | [diff] [blame] | 223 | 'archive_url and local_path can not both be specified.') |
Gabe Black | 3b56720 | 2015-09-23 14:07:59 -0700 | [diff] [blame] | 224 | if not dl: |
| 225 | archive_url = _canonicalize_archive_url(archive_url) |
Luis Hector Chavez | dca9dd7 | 2018-06-12 12:56:30 -0700 | [diff] [blame] | 226 | dl = downloader.GoogleStorageDownloader( |
| 227 | updater.static_dir, archive_url, |
| 228 | downloader.GoogleStorageDownloader.GetBuildIdFromArchiveURL( |
| 229 | archive_url)) |
Gabe Black | 3b56720 | 2015-09-23 14:07:59 -0700 | [diff] [blame] | 230 | elif not dl: |
| 231 | target = kwargs.get('target', None) |
Dan Shi | 72b1613 | 2015-10-08 12:10:33 -0700 | [diff] [blame] | 232 | branch = kwargs.get('branch', None) |
Dan Shi | 61305df | 2015-10-26 16:52:35 -0700 | [diff] [blame] | 233 | build_id = kwargs.get('build_id', None) |
| 234 | if not target or not branch or not build_id: |
Amin Hassani | d4e3539 | 2019-10-03 11:02:44 -0700 | [diff] [blame] | 235 | raise DevServerError('target, branch, build ID must all be specified for ' |
| 236 | 'downloading Android build.') |
Dan Shi | 72b1613 | 2015-10-08 12:10:33 -0700 | [diff] [blame] | 237 | dl = downloader.AndroidBuildDownloader(updater.static_dir, branch, build_id, |
| 238 | target) |
Gabe Black | 3b56720 | 2015-09-23 14:07:59 -0700 | [diff] [blame] | 239 | |
| 240 | return dl |
| 241 | |
| 242 | |
| 243 | def _get_downloader_and_factory(kwargs): |
| 244 | """Returns the downloader and artifact factory based on passed in arguments. |
| 245 | |
| 246 | Args: |
Amin Hassani | 08e42d2 | 2019-06-03 00:31:30 -0700 | [diff] [blame] | 247 | kwargs: Keyword arguments for the request. |
Gabe Black | 3b56720 | 2015-09-23 14:07:59 -0700 | [diff] [blame] | 248 | """ |
| 249 | artifacts, files = _get_artifacts(kwargs) |
| 250 | dl = _get_downloader(kwargs) |
| 251 | |
Achuith Bhandarkar | 2a1fcd8 | 2019-10-17 17:45:58 -0700 | [diff] [blame] | 252 | if (isinstance(dl, (downloader.GoogleStorageDownloader, |
| 253 | downloader.LocalDownloader))): |
Gabe Black | 3b56720 | 2015-09-23 14:07:59 -0700 | [diff] [blame] | 254 | factory_class = build_artifact.ChromeOSArtifactFactory |
Dan Shi | 72b1613 | 2015-10-08 12:10:33 -0700 | [diff] [blame] | 255 | elif isinstance(dl, downloader.AndroidBuildDownloader): |
Gabe Black | 3b56720 | 2015-09-23 14:07:59 -0700 | [diff] [blame] | 256 | factory_class = build_artifact.AndroidArtifactFactory |
| 257 | else: |
Amin Hassani | d4e3539 | 2019-10-03 11:02:44 -0700 | [diff] [blame] | 258 | raise DevServerError( |
Congbin Guo | 4132a27 | 2019-08-20 12:32:14 -0700 | [diff] [blame] | 259 | 'Unrecognized value for downloader type: %s' % type(dl)) |
Gabe Black | 3b56720 | 2015-09-23 14:07:59 -0700 | [diff] [blame] | 260 | |
| 261 | factory = factory_class(dl.GetBuildDir(), artifacts, files, dl.GetBuild()) |
| 262 | |
| 263 | return dl, factory |
| 264 | |
| 265 | |
Scott Zawalski | 4647ce6 | 2012-01-03 17:17:28 -0500 | [diff] [blame] | 266 | def _LeadingWhiteSpaceCount(string): |
| 267 | """Count the amount of leading whitespace in a string. |
| 268 | |
| 269 | Args: |
| 270 | string: The string to count leading whitespace in. |
Don Garrett | f84631a | 2014-01-07 18:21:26 -0800 | [diff] [blame] | 271 | |
Scott Zawalski | 4647ce6 | 2012-01-03 17:17:28 -0500 | [diff] [blame] | 272 | Returns: |
| 273 | number of white space chars before characters start. |
| 274 | """ |
Gabe Black | 3b56720 | 2015-09-23 14:07:59 -0700 | [diff] [blame] | 275 | matched = re.match(r'^\s+', string) |
Scott Zawalski | 4647ce6 | 2012-01-03 17:17:28 -0500 | [diff] [blame] | 276 | if matched: |
| 277 | return len(matched.group()) |
| 278 | |
| 279 | return 0 |
| 280 | |
| 281 | |
| 282 | def _PrintDocStringAsHTML(func): |
| 283 | """Make a functions docstring somewhat HTML style. |
| 284 | |
| 285 | Args: |
| 286 | func: The function to return the docstring from. |
Don Garrett | f84631a | 2014-01-07 18:21:26 -0800 | [diff] [blame] | 287 | |
Scott Zawalski | 4647ce6 | 2012-01-03 17:17:28 -0500 | [diff] [blame] | 288 | Returns: |
| 289 | A string that is somewhat formated for a web browser. |
| 290 | """ |
| 291 | # TODO(scottz): Make this parse Args/Returns in a prettier way. |
| 292 | # Arguments could be bolded and indented etc. |
| 293 | html_doc = [] |
| 294 | for line in func.__doc__.splitlines(): |
| 295 | leading_space = _LeadingWhiteSpaceCount(line) |
| 296 | if leading_space > 0: |
Chris Sosa | 47a7d4e | 2012-03-28 11:26:55 -0700 | [diff] [blame] | 297 | line = ' ' * leading_space + line |
Scott Zawalski | 4647ce6 | 2012-01-03 17:17:28 -0500 | [diff] [blame] | 298 | |
| 299 | html_doc.append('<BR>%s' % line) |
| 300 | |
| 301 | return '\n'.join(html_doc) |
| 302 | |
| 303 | |
Simran Basi | ef83d6a | 2014-08-28 14:32:01 -0700 | [diff] [blame] | 304 | def _GetUpdateTimestampHandler(static_dir): |
| 305 | """Returns a handler to update directory staged.timestamp. |
| 306 | |
| 307 | This handler resets the stage.timestamp whenever static content is accessed. |
| 308 | |
| 309 | Args: |
| 310 | static_dir: Directory from which static content is being staged. |
| 311 | |
| 312 | Returns: |
Amin Hassani | 08e42d2 | 2019-06-03 00:31:30 -0700 | [diff] [blame] | 313 | A cherrypy handler to update the timestamp of accessed content. |
Simran Basi | ef83d6a | 2014-08-28 14:32:01 -0700 | [diff] [blame] | 314 | """ |
| 315 | def UpdateTimestampHandler(): |
| 316 | if not '404' in cherrypy.response.status: |
| 317 | build_match = re.match(devserver_constants.STAGED_BUILD_REGEX, |
| 318 | cherrypy.request.path_info) |
| 319 | if build_match: |
| 320 | build_dir = os.path.join(static_dir, build_match.group('build')) |
| 321 | downloader.Downloader.TouchTimestampForStaged(build_dir) |
| 322 | return UpdateTimestampHandler |
| 323 | |
| 324 | |
Chris Sosa | 7c93136 | 2010-10-11 19:49:01 -0700 | [diff] [blame] | 325 | def _GetConfig(options): |
| 326 | """Returns the configuration for the devserver.""" |
Mandeep Singh Baines | 38dcdda | 2012-12-07 17:55:33 -0800 | [diff] [blame] | 327 | |
Mandeep Singh Baines | 38dcdda | 2012-12-07 17:55:33 -0800 | [diff] [blame] | 328 | socket_host = '::' |
Yu-Ju Hong | c8d4af3 | 2013-11-12 15:14:26 -0800 | [diff] [blame] | 329 | # Fall back to IPv4 when python is not configured with IPv6. |
| 330 | if not socket.has_ipv6: |
Mandeep Singh Baines | 38dcdda | 2012-12-07 17:55:33 -0800 | [diff] [blame] | 331 | socket_host = '0.0.0.0' |
| 332 | |
Simran Basi | ef83d6a | 2014-08-28 14:32:01 -0700 | [diff] [blame] | 333 | # Adds the UpdateTimestampHandler to cherrypy's tools. This tools executes |
| 334 | # on the on_end_resource hook. This hook is called once processing is |
| 335 | # complete and the response is ready to be returned. |
| 336 | cherrypy.tools.update_timestamp = cherrypy.Tool( |
| 337 | 'on_end_resource', _GetUpdateTimestampHandler(options.static_dir)) |
| 338 | |
David Riley | 2fcb012 | 2017-11-02 11:25:39 -0700 | [diff] [blame] | 339 | base_config = { |
| 340 | 'global': { |
| 341 | 'server.log_request_headers': True, |
| 342 | 'server.protocol_version': 'HTTP/1.1', |
| 343 | 'server.socket_host': socket_host, |
| 344 | 'server.socket_port': int(options.port), |
| 345 | 'response.timeout': 6000, |
| 346 | 'request.show_tracebacks': True, |
| 347 | 'server.socket_timeout': 60, |
| 348 | 'server.thread_pool': 2, |
| 349 | 'engine.autoreload.on': False, |
| 350 | }, |
| 351 | '/api': { |
| 352 | # Gets rid of cherrypy parsing post file for args. |
| 353 | 'request.process_request_body': False, |
| 354 | }, |
| 355 | '/build': { |
| 356 | 'response.timeout': 100000, |
| 357 | }, |
| 358 | '/update': { |
| 359 | # Gets rid of cherrypy parsing post file for args. |
| 360 | 'request.process_request_body': False, |
| 361 | 'response.timeout': 10000, |
| 362 | }, |
| 363 | # Sets up the static dir for file hosting. |
| 364 | '/static': { |
| 365 | 'tools.staticdir.dir': options.static_dir, |
| 366 | 'tools.staticdir.on': True, |
| 367 | 'response.timeout': 10000, |
| 368 | 'tools.update_timestamp.on': True, |
| 369 | }, |
| 370 | } |
Chris Sosa | 5f118ef | 2012-07-12 11:37:50 -0700 | [diff] [blame] | 371 | if options.production: |
Alex Miller | 93beca5 | 2013-07-30 19:25:09 -0700 | [diff] [blame] | 372 | base_config['global'].update({'server.thread_pool': 150}) |
Scott Zawalski | 1c5e7cd | 2012-02-27 13:12:52 -0500 | [diff] [blame] | 373 | |
Chris Sosa | 7c93136 | 2010-10-11 19:49:01 -0700 | [diff] [blame] | 374 | return base_config |
rtc@google.com | 6424466 | 2009-11-12 00:52:08 +0000 | [diff] [blame] | 375 | |
Darin Petkov | e17164a | 2010-08-11 13:24:41 -0700 | [diff] [blame] | 376 | |
Gilad Arnold | d5ebaaa | 2012-10-02 11:52:38 -0700 | [diff] [blame] | 377 | def _GetRecursiveMemberObject(root, member_list): |
| 378 | """Returns an object corresponding to a nested member list. |
| 379 | |
| 380 | Args: |
| 381 | root: the root object to search |
| 382 | member_list: list of nested members to search |
Don Garrett | f84631a | 2014-01-07 18:21:26 -0800 | [diff] [blame] | 383 | |
Gilad Arnold | d5ebaaa | 2012-10-02 11:52:38 -0700 | [diff] [blame] | 384 | Returns: |
| 385 | An object corresponding to the member name list; None otherwise. |
| 386 | """ |
| 387 | for member in member_list: |
| 388 | next_root = root.__class__.__dict__.get(member) |
| 389 | if not next_root: |
| 390 | return None |
| 391 | root = next_root |
| 392 | return root |
| 393 | |
| 394 | |
| 395 | def _IsExposed(name): |
| 396 | """Returns True iff |name| has an `exposed' attribute and it is set.""" |
| 397 | return hasattr(name, 'exposed') and name.exposed |
| 398 | |
| 399 | |
Congbin Guo | 6bc3218 | 2019-08-20 17:54:30 -0700 | [diff] [blame] | 400 | def _GetExposedMethod(nested_member): |
Gilad Arnold | d5ebaaa | 2012-10-02 11:52:38 -0700 | [diff] [blame] | 401 | """Returns a CherryPy-exposed method, if such exists. |
| 402 | |
| 403 | Args: |
Gilad Arnold | d5ebaaa | 2012-10-02 11:52:38 -0700 | [diff] [blame] | 404 | nested_member: a slash-joined path to the nested member |
Don Garrett | f84631a | 2014-01-07 18:21:26 -0800 | [diff] [blame] | 405 | |
Gilad Arnold | d5ebaaa | 2012-10-02 11:52:38 -0700 | [diff] [blame] | 406 | Returns: |
Congbin Guo | 6bc3218 | 2019-08-20 17:54:30 -0700 | [diff] [blame] | 407 | A function object corresponding to the path defined by |nested_member| from |
| 408 | the app root object registered, if the function is exposed; None otherwise. |
Gilad Arnold | d5ebaaa | 2012-10-02 11:52:38 -0700 | [diff] [blame] | 409 | """ |
Congbin Guo | 6bc3218 | 2019-08-20 17:54:30 -0700 | [diff] [blame] | 410 | for app in cherrypy.tree.apps.values(): |
| 411 | # Use the 'index' function doc as the doc of the app. |
| 412 | if nested_member == app.script_name.lstrip('/'): |
| 413 | nested_member = 'index' |
| 414 | |
| 415 | method = _GetRecursiveMemberObject(app.root, nested_member.split('/')) |
| 416 | if method and isinstance(method, types.FunctionType) and _IsExposed(method): |
| 417 | return method |
Gilad Arnold | d5ebaaa | 2012-10-02 11:52:38 -0700 | [diff] [blame] | 418 | |
| 419 | |
Gilad Arnold | 748c832 | 2012-10-12 09:51:35 -0700 | [diff] [blame] | 420 | def _FindExposedMethods(root, prefix, unlisted=None): |
Gilad Arnold | d5ebaaa | 2012-10-02 11:52:38 -0700 | [diff] [blame] | 421 | """Finds exposed CherryPy methods. |
| 422 | |
| 423 | Args: |
| 424 | root: the root object for searching |
| 425 | prefix: slash-joined chain of members leading to current object |
| 426 | unlisted: URLs to be excluded regardless of their exposed status |
Don Garrett | f84631a | 2014-01-07 18:21:26 -0800 | [diff] [blame] | 427 | |
Gilad Arnold | d5ebaaa | 2012-10-02 11:52:38 -0700 | [diff] [blame] | 428 | Returns: |
| 429 | List of exposed URLs that are not unlisted. |
| 430 | """ |
| 431 | method_list = [] |
Congbin Guo | 6bc3218 | 2019-08-20 17:54:30 -0700 | [diff] [blame] | 432 | for member in root.__class__.__dict__.keys(): |
Gilad Arnold | d5ebaaa | 2012-10-02 11:52:38 -0700 | [diff] [blame] | 433 | prefixed_member = prefix + '/' + member if prefix else member |
Gilad Arnold | 748c832 | 2012-10-12 09:51:35 -0700 | [diff] [blame] | 434 | if unlisted and prefixed_member in unlisted: |
Gilad Arnold | d5ebaaa | 2012-10-02 11:52:38 -0700 | [diff] [blame] | 435 | continue |
| 436 | member_obj = root.__class__.__dict__[member] |
| 437 | if _IsExposed(member_obj): |
Amin Hassani | 08e42d2 | 2019-06-03 00:31:30 -0700 | [diff] [blame] | 438 | if isinstance(member_obj, types.FunctionType): |
Congbin Guo | 6bc3218 | 2019-08-20 17:54:30 -0700 | [diff] [blame] | 439 | # Regard the app name as exposed "method" name if it exposed 'index' |
| 440 | # function. |
| 441 | if prefix and member == 'index': |
| 442 | method_list.append(prefix) |
| 443 | else: |
| 444 | method_list.append(prefixed_member) |
Gilad Arnold | d5ebaaa | 2012-10-02 11:52:38 -0700 | [diff] [blame] | 445 | else: |
| 446 | method_list += _FindExposedMethods( |
| 447 | member_obj, prefixed_member, unlisted) |
| 448 | return method_list |
| 449 | |
| 450 | |
xixuan | 52c2fba | 2016-05-20 17:02:48 -0700 | [diff] [blame] | 451 | def _check_base_args_for_auto_update(kwargs): |
xixuan | ac89ce8 | 2016-11-30 16:48:20 -0800 | [diff] [blame] | 452 | """Check basic args required for auto-update. |
| 453 | |
| 454 | Args: |
| 455 | kwargs: the parameters to be checked. |
| 456 | |
| 457 | Raises: |
| 458 | DevServerHTTPError if required parameters don't exist in kwargs. |
| 459 | """ |
xixuan | 52c2fba | 2016-05-20 17:02:48 -0700 | [diff] [blame] | 460 | if 'host_name' not in kwargs: |
Amin Hassani | 722e096 | 2019-11-15 15:45:31 -0800 | [diff] [blame] | 461 | raise DevServerHTTPError(http_client.INTERNAL_SERVER_ERROR, |
| 462 | KEY_ERROR_MSG % 'host_name') |
xixuan | 52c2fba | 2016-05-20 17:02:48 -0700 | [diff] [blame] | 463 | |
| 464 | if 'build_name' not in kwargs: |
Amin Hassani | 722e096 | 2019-11-15 15:45:31 -0800 | [diff] [blame] | 465 | raise DevServerHTTPError(http_client.INTERNAL_SERVER_ERROR, |
| 466 | KEY_ERROR_MSG % 'build_name') |
xixuan | 52c2fba | 2016-05-20 17:02:48 -0700 | [diff] [blame] | 467 | |
| 468 | |
| 469 | def _parse_boolean_arg(kwargs, key): |
xixuan | ac89ce8 | 2016-11-30 16:48:20 -0800 | [diff] [blame] | 470 | """Parse boolean arg from kwargs. |
| 471 | |
| 472 | Args: |
| 473 | kwargs: the parameters to be checked. |
| 474 | key: the key to be parsed. |
| 475 | |
| 476 | Returns: |
| 477 | The boolean value of kwargs[key], or False if key doesn't exist in kwargs. |
| 478 | |
| 479 | Raises: |
| 480 | DevServerHTTPError if kwargs[key] is not a boolean variable. |
| 481 | """ |
xixuan | 52c2fba | 2016-05-20 17:02:48 -0700 | [diff] [blame] | 482 | if key in kwargs: |
| 483 | if kwargs[key] == 'True': |
| 484 | return True |
| 485 | elif kwargs[key] == 'False': |
| 486 | return False |
| 487 | else: |
Amin Hassani | 722e096 | 2019-11-15 15:45:31 -0800 | [diff] [blame] | 488 | raise DevServerHTTPError(http_client.INTERNAL_SERVER_ERROR, |
| 489 | 'The value for key %s is not boolean.' % key) |
xixuan | 52c2fba | 2016-05-20 17:02:48 -0700 | [diff] [blame] | 490 | else: |
| 491 | return False |
| 492 | |
xixuan | 447ad9d | 2017-02-28 14:46:20 -0800 | [diff] [blame] | 493 | |
xixuan | ac89ce8 | 2016-11-30 16:48:20 -0800 | [diff] [blame] | 494 | def _parse_string_arg(kwargs, key): |
| 495 | """Parse string arg from kwargs. |
| 496 | |
| 497 | Args: |
| 498 | kwargs: the parameters to be checked. |
| 499 | key: the key to be parsed. |
| 500 | |
| 501 | Returns: |
| 502 | The string value of kwargs[key], or None if key doesn't exist in kwargs. |
| 503 | """ |
| 504 | if key in kwargs: |
| 505 | return kwargs[key] |
| 506 | else: |
| 507 | return None |
| 508 | |
xixuan | 447ad9d | 2017-02-28 14:46:20 -0800 | [diff] [blame] | 509 | |
xixuan | ac89ce8 | 2016-11-30 16:48:20 -0800 | [diff] [blame] | 510 | def _build_uri_from_build_name(build_name): |
| 511 | """Get build url from a given build name. |
| 512 | |
| 513 | Args: |
| 514 | build_name: the build name to be parsed, whose format is |
| 515 | 'board/release_version'. |
| 516 | |
| 517 | Returns: |
| 518 | The release_archive_url on Google Storage for this build name. |
| 519 | """ |
Amin Hassani | 08e42d2 | 2019-06-03 00:31:30 -0700 | [diff] [blame] | 520 | # TODO(ahassani): This function doesn't seem to be used anywhere since its |
| 521 | # previous use of lib.paygen.gspath was broken and it doesn't seem to be |
| 522 | # causing any runtime issues. So deprecate this in the future. |
| 523 | tokens = build_name.split('/') |
| 524 | return 'gs://chromeos-releases/stable-channel/%s/%s' % (tokens[0], tokens[1]) |
xixuan | 52c2fba | 2016-05-20 17:02:48 -0700 | [diff] [blame] | 525 | |
xixuan | 447ad9d | 2017-02-28 14:46:20 -0800 | [diff] [blame] | 526 | |
| 527 | def _clear_process(host_name, pid): |
| 528 | """Clear AU process for given hostname and pid. |
| 529 | |
| 530 | This clear includes: |
| 531 | 1. kill process if it's alive. |
| 532 | 2. delete the track status file of this process. |
| 533 | 3. delete the executing log file of this process. |
| 534 | |
| 535 | Args: |
| 536 | host_name: the host to execute auto-update. |
| 537 | pid: the background auto-update process id. |
| 538 | """ |
| 539 | if cros_update_progress.IsProcessAlive(pid): |
| 540 | os.killpg(int(pid), signal.SIGKILL) |
| 541 | |
| 542 | cros_update_progress.DelTrackStatusFile(host_name, pid) |
| 543 | cros_update_progress.DelExecuteLogFile(host_name, pid) |
| 544 | |
| 545 | |
Sanika Kulkarni | d4496fd | 2020-02-04 17:26:25 -0800 | [diff] [blame] | 546 | def is_deprecated_server(): |
| 547 | """Gets whether the devserver has deprecated RPCs.""" |
| 548 | return cherrypy.config.get('infra_removal', False) |
| 549 | |
| 550 | |
Dale Curtis | c9aaf3a | 2011-08-09 15:47:40 -0700 | [diff] [blame] | 551 | class ApiRoot(object): |
| 552 | """RESTful API for Dev Server information.""" |
| 553 | exposed = True |
| 554 | |
| 555 | @cherrypy.expose |
Gilad Arnold | 286a006 | 2012-01-12 13:47:02 -0800 | [diff] [blame] | 556 | def hostlog(self, ip): |
Gilad Arnold | 1b90839 | 2012-10-05 11:36:27 -0700 | [diff] [blame] | 557 | """Returns a JSON object containing a log of host event. |
| 558 | |
| 559 | Args: |
| 560 | ip: address of host whose event log is requested, or `all' |
Don Garrett | f84631a | 2014-01-07 18:21:26 -0800 | [diff] [blame] | 561 | |
Gilad Arnold | 1b90839 | 2012-10-05 11:36:27 -0700 | [diff] [blame] | 562 | Returns: |
Amin Hassani | 7c44785 | 2019-09-26 15:01:48 -0700 | [diff] [blame] | 563 | A JSON dictionary containing all or some of the following fields: |
Amin Hassani | e7ead90 | 2019-10-11 16:42:43 -0700 | [diff] [blame] | 564 | version: The Chromium OS version the device is running. |
| 565 | track: The channel the device is running on. |
| 566 | board: The device's board. |
| 567 | event_result: The event result of Omaha request. |
| 568 | event_type: The event type of Omaha request. |
| 569 | previous_version: The Chromium OS version we updated and rebooted from. |
| 570 | timestamp: The timestamp the event was received. |
Amin Hassani | 7c44785 | 2019-09-26 15:01:48 -0700 | [diff] [blame] | 571 | See the OmahaEvent class in update_engine/omaha_request_action.h for |
| 572 | event type and status code definitions. If the ip does not exist an empty |
| 573 | string is returned. |
Gilad Arnold | 1b90839 | 2012-10-05 11:36:27 -0700 | [diff] [blame] | 574 | |
| 575 | Example URL: |
| 576 | http://myhost/api/hostlog?ip=192.168.1.5 |
| 577 | """ |
Sanika Kulkarni | d4496fd | 2020-02-04 17:26:25 -0800 | [diff] [blame] | 578 | if is_deprecated_server(): |
| 579 | raise DeprecatedRPCError('hostlog') |
Gilad Arnold | 286a006 | 2012-01-12 13:47:02 -0800 | [diff] [blame] | 580 | return updater.HandleHostLogPing(ip) |
| 581 | |
| 582 | @cherrypy.expose |
Don Garrett | f84631a | 2014-01-07 18:21:26 -0800 | [diff] [blame] | 583 | def fileinfo(self, *args): |
Gilad Arnold | 55a2a37 | 2012-10-02 09:46:32 -0700 | [diff] [blame] | 584 | """Returns information about a given staged file. |
| 585 | |
| 586 | Args: |
Don Garrett | f84631a | 2014-01-07 18:21:26 -0800 | [diff] [blame] | 587 | args: path to the file inside the server's static staging directory |
| 588 | |
Gilad Arnold | 55a2a37 | 2012-10-02 09:46:32 -0700 | [diff] [blame] | 589 | Returns: |
| 590 | A JSON encoded dictionary with information about the said file, which may |
| 591 | contain the following keys/values: |
Gilad Arnold | 1b90839 | 2012-10-05 11:36:27 -0700 | [diff] [blame] | 592 | size (int): the file size in bytes |
Gilad Arnold | 1b90839 | 2012-10-05 11:36:27 -0700 | [diff] [blame] | 593 | sha256 (string): a base64 encoded SHA256 hash |
| 594 | |
| 595 | Example URL: |
| 596 | http://myhost/api/fileinfo/some/path/to/file |
Gilad Arnold | 55a2a37 | 2012-10-02 09:46:32 -0700 | [diff] [blame] | 597 | """ |
Sanika Kulkarni | d4496fd | 2020-02-04 17:26:25 -0800 | [diff] [blame] | 598 | if is_deprecated_server(): |
| 599 | raise DeprecatedRPCError('fileinfo') |
| 600 | |
Amin Hassani | 28df421 | 2019-10-28 10:16:50 -0700 | [diff] [blame] | 601 | # TODO(ahassani): A better way of doing this is to just return the the |
| 602 | # content of the payloads' property file instead. That has all this info |
| 603 | # except that the key for sha256 is 'sha256_hex', but still base64 encdoed. |
| 604 | |
Don Garrett | f84631a | 2014-01-07 18:21:26 -0800 | [diff] [blame] | 605 | file_path = os.path.join(updater.static_dir, *args) |
Gilad Arnold | 55a2a37 | 2012-10-02 09:46:32 -0700 | [diff] [blame] | 606 | if not os.path.exists(file_path): |
Amin Hassani | d4e3539 | 2019-10-03 11:02:44 -0700 | [diff] [blame] | 607 | raise DevServerError('file not found: %s' % file_path) |
Gilad Arnold | 55a2a37 | 2012-10-02 09:46:32 -0700 | [diff] [blame] | 608 | try: |
| 609 | file_size = os.path.getsize(file_path) |
Gilad Arnold | 55a2a37 | 2012-10-02 09:46:32 -0700 | [diff] [blame] | 610 | file_sha256 = common_util.GetFileSha256(file_path) |
Amin Hassani | 469f570 | 2019-10-21 15:35:06 -0700 | [diff] [blame] | 611 | except os.error as e: |
Amin Hassani | d4e3539 | 2019-10-03 11:02:44 -0700 | [diff] [blame] | 612 | raise DevServerError( |
Congbin Guo | 4132a27 | 2019-08-20 12:32:14 -0700 | [diff] [blame] | 613 | 'failed to get info for file %s: %s' % (file_path, e)) |
Gilad Arnold | e74b381 | 2013-04-22 11:27:38 -0700 | [diff] [blame] | 614 | |
Gilad Arnold | e74b381 | 2013-04-22 11:27:38 -0700 | [diff] [blame] | 615 | return json.dumps({ |
Amin Hassani | 28df421 | 2019-10-28 10:16:50 -0700 | [diff] [blame] | 616 | 'size': file_size, |
| 617 | 'sha256': file_sha256, |
| 618 | }, sort_keys=True) |
Gilad Arnold | 55a2a37 | 2012-10-02 09:46:32 -0700 | [diff] [blame] | 619 | |
Chris Sosa | 76e44b9 | 2013-01-31 12:11:38 -0800 | [diff] [blame] | 620 | |
David Rochberg | 7c79a81 | 2011-01-19 14:24:45 -0500 | [diff] [blame] | 621 | class DevServerRoot(object): |
Chris Sosa | 7c93136 | 2010-10-11 19:49:01 -0700 | [diff] [blame] | 622 | """The Root Class for the Dev Server. |
| 623 | |
| 624 | CherryPy works as follows: |
| 625 | For each method in this class, cherrpy interprets root/path |
| 626 | as a call to an instance of DevServerRoot->method_name. For example, |
| 627 | a call to http://myhost/build will call build. CherryPy automatically |
| 628 | parses http args and places them as keyword arguments in each method. |
| 629 | For paths http://myhost/update/dir1/dir2, you can use *args so that |
| 630 | cherrypy uses the update method and puts the extra paths in args. |
| 631 | """ |
Gilad Arnold | f8f769f | 2012-09-24 08:43:01 -0700 | [diff] [blame] | 632 | # Method names that should not be listed on the index page. |
| 633 | _UNLISTED_METHODS = ['index', 'doc'] |
| 634 | |
Dale Curtis | c9aaf3a | 2011-08-09 15:47:40 -0700 | [diff] [blame] | 635 | api = ApiRoot() |
Chris Sosa | 7c93136 | 2010-10-11 19:49:01 -0700 | [diff] [blame] | 636 | |
Dan Shi | 59ae709 | 2013-06-04 14:37:27 -0700 | [diff] [blame] | 637 | # Number of threads that devserver is staging images. |
| 638 | _staging_thread_count = 0 |
| 639 | # Lock used to lock increasing/decreasing count. |
| 640 | _staging_thread_count_lock = threading.Lock() |
| 641 | |
joychen | 3cb228e | 2013-06-12 12:13:13 -0700 | [diff] [blame] | 642 | def __init__(self, _xbuddy): |
Nick Sanders | 7dcaa2e | 2011-08-04 15:20:41 -0700 | [diff] [blame] | 643 | self._builder = None |
Simran Basi | 4baad08 | 2013-02-14 13:39:18 -0800 | [diff] [blame] | 644 | self._telemetry_lock_dict = common_util.LockDict() |
joychen | 3cb228e | 2013-06-12 12:13:13 -0700 | [diff] [blame] | 645 | self._xbuddy = _xbuddy |
David Rochberg | 7c79a81 | 2011-01-19 14:24:45 -0500 | [diff] [blame] | 646 | |
Congbin Guo | 3afae6c | 2019-08-13 16:29:42 -0700 | [diff] [blame] | 647 | @property |
| 648 | def staging_thread_count(self): |
| 649 | """Get the staging thread count.""" |
| 650 | return self._staging_thread_count |
Dan Shi | afd0e49 | 2015-05-27 14:23:51 -0700 | [diff] [blame] | 651 | |
Dale Curtis | c9aaf3a | 2011-08-09 15:47:40 -0700 | [diff] [blame] | 652 | @cherrypy.expose |
David Rochberg | 7c79a81 | 2011-01-19 14:24:45 -0500 | [diff] [blame] | 653 | def build(self, board, pkg, **kwargs): |
Chris Sosa | 7c93136 | 2010-10-11 19:49:01 -0700 | [diff] [blame] | 654 | """Builds the package specified.""" |
Sanika Kulkarni | d4496fd | 2020-02-04 17:26:25 -0800 | [diff] [blame] | 655 | if is_deprecated_server(): |
| 656 | raise DeprecatedRPCError('build') |
| 657 | |
Nick Sanders | 7dcaa2e | 2011-08-04 15:20:41 -0700 | [diff] [blame] | 658 | import builder |
| 659 | if self._builder is None: |
| 660 | self._builder = builder.Builder() |
David Rochberg | 7c79a81 | 2011-01-19 14:24:45 -0500 | [diff] [blame] | 661 | return self._builder.Build(board, pkg, kwargs) |
Chris Sosa | 7c93136 | 2010-10-11 19:49:01 -0700 | [diff] [blame] | 662 | |
Dale Curtis | c9aaf3a | 2011-08-09 15:47:40 -0700 | [diff] [blame] | 663 | @cherrypy.expose |
Dan Shi | f8eb0d1 | 2013-08-01 17:52:06 -0700 | [diff] [blame] | 664 | def is_staged(self, **kwargs): |
| 665 | """Check if artifacts have been downloaded. |
| 666 | |
Congbin Guo | 3afae6c | 2019-08-13 16:29:42 -0700 | [diff] [blame] | 667 | Examples: |
| 668 | To check if autotest and test_suites are staged: |
| 669 | http://devserver_url:<port>/is_staged?archive_url=gs://your_url/path& |
| 670 | artifacts=autotest,test_suites |
| 671 | |
Amin Hassani | 08e42d2 | 2019-06-03 00:31:30 -0700 | [diff] [blame] | 672 | Args: |
Chris Sosa | 6b0c617 | 2013-08-05 17:01:33 -0700 | [diff] [blame] | 673 | async: True to return without waiting for download to complete. |
| 674 | artifacts: Comma separated list of named artifacts to download. |
| 675 | These are defined in artifact_info and have their implementation |
| 676 | in build_artifact.py. |
| 677 | files: Comma separated list of file artifacts to stage. These |
| 678 | will be available as is in the corresponding static directory with no |
| 679 | custom post-processing. |
| 680 | |
Congbin Guo | 3afae6c | 2019-08-13 16:29:42 -0700 | [diff] [blame] | 681 | Returns: |
| 682 | True of all artifacts are staged. |
Dan Shi | f8eb0d1 | 2013-08-01 17:52:06 -0700 | [diff] [blame] | 683 | """ |
Gabe Black | 3b56720 | 2015-09-23 14:07:59 -0700 | [diff] [blame] | 684 | dl, factory = _get_downloader_and_factory(kwargs) |
Aviv Keshet | 57d1817 | 2016-06-18 20:39:09 -0700 | [diff] [blame] | 685 | response = str(dl.IsStaged(factory)) |
| 686 | _Log('Responding to is_staged %s request with %r', kwargs, response) |
| 687 | return response |
Dan Shi | 59ae709 | 2013-06-04 14:37:27 -0700 | [diff] [blame] | 688 | |
Chris Sosa | 76e44b9 | 2013-01-31 12:11:38 -0800 | [diff] [blame] | 689 | @cherrypy.expose |
Prashanth B | a06d2d2 | 2014-03-07 15:35:19 -0800 | [diff] [blame] | 690 | def list_image_dir(self, **kwargs): |
| 691 | """Take an archive url and list the contents in its staged directory. |
| 692 | |
Amin Hassani | 08e42d2 | 2019-06-03 00:31:30 -0700 | [diff] [blame] | 693 | Examples: |
Prashanth B | a06d2d2 | 2014-03-07 15:35:19 -0800 | [diff] [blame] | 694 | To list the contents of where this devserver should have staged |
| 695 | gs://image-archive/<board>-release/<build> call: |
| 696 | http://devserver_url:<port>/list_image_dir?archive_url=<gs://..> |
| 697 | |
Congbin Guo | 3afae6c | 2019-08-13 16:29:42 -0700 | [diff] [blame] | 698 | Args: |
| 699 | archive_url: Google Storage URL for the build. |
| 700 | |
Prashanth B | a06d2d2 | 2014-03-07 15:35:19 -0800 | [diff] [blame] | 701 | Returns: |
| 702 | A string with information about the contents of the image directory. |
| 703 | """ |
Gabe Black | 3b56720 | 2015-09-23 14:07:59 -0700 | [diff] [blame] | 704 | dl = _get_downloader(kwargs) |
Prashanth B | a06d2d2 | 2014-03-07 15:35:19 -0800 | [diff] [blame] | 705 | try: |
Gabe Black | 3b56720 | 2015-09-23 14:07:59 -0700 | [diff] [blame] | 706 | image_dir_contents = dl.ListBuildDir() |
Prashanth B | a06d2d2 | 2014-03-07 15:35:19 -0800 | [diff] [blame] | 707 | except build_artifact.ArtifactDownloadError as e: |
| 708 | return 'Cannot list the contents of staged artifacts. %s' % e |
| 709 | if not image_dir_contents: |
Gabe Black | 3b56720 | 2015-09-23 14:07:59 -0700 | [diff] [blame] | 710 | return '%s has not been staged on this devserver.' % dl.DescribeSource() |
Prashanth B | a06d2d2 | 2014-03-07 15:35:19 -0800 | [diff] [blame] | 711 | return image_dir_contents |
| 712 | |
| 713 | @cherrypy.expose |
Chris Sosa | 76e44b9 | 2013-01-31 12:11:38 -0800 | [diff] [blame] | 714 | def stage(self, **kwargs): |
Gabe Black | 3b56720 | 2015-09-23 14:07:59 -0700 | [diff] [blame] | 715 | """Downloads and caches build artifacts. |
Chris Sosa | 76e44b9 | 2013-01-31 12:11:38 -0800 | [diff] [blame] | 716 | |
Gabe Black | 3b56720 | 2015-09-23 14:07:59 -0700 | [diff] [blame] | 717 | Downloads and caches build artifacts, possibly from a Google Storage URL, |
Dan Shi | 72b1613 | 2015-10-08 12:10:33 -0700 | [diff] [blame] | 718 | or from Android's build server. Returns once these have been downloaded |
Gabe Black | 3b56720 | 2015-09-23 14:07:59 -0700 | [diff] [blame] | 719 | on the devserver. A call to this will attempt to cache non-specified |
| 720 | artifacts in the background for the given from the given URL following |
| 721 | the principle of spatial locality. Spatial locality of different |
Chris Sosa | 76e44b9 | 2013-01-31 12:11:38 -0800 | [diff] [blame] | 722 | artifacts is explicitly defined in the build_artifact module. |
| 723 | |
| 724 | These artifacts will then be available from the static/ sub-directory of |
| 725 | the devserver. |
| 726 | |
Amin Hassani | 08e42d2 | 2019-06-03 00:31:30 -0700 | [diff] [blame] | 727 | Examples: |
Chris Sosa | 76e44b9 | 2013-01-31 12:11:38 -0800 | [diff] [blame] | 728 | To download the autotest and test suites tarballs: |
| 729 | http://devserver_url:<port>/stage?archive_url=gs://your_url/path& |
| 730 | artifacts=autotest,test_suites |
| 731 | To download the full update payload: |
| 732 | http://devserver_url:<port>/stage?archive_url=gs://your_url/path& |
| 733 | artifacts=full_payload |
Chris Sosa | 6b0c617 | 2013-08-05 17:01:33 -0700 | [diff] [blame] | 734 | To download just a file called blah.bin: |
| 735 | http://devserver_url:<port>/stage?archive_url=gs://your_url/path& |
| 736 | files=blah.bin |
Chris Sosa | 76e44b9 | 2013-01-31 12:11:38 -0800 | [diff] [blame] | 737 | |
| 738 | For both these examples, one could find these artifacts at: |
joychen | ed64b22 | 2013-06-21 16:39:34 -0700 | [diff] [blame] | 739 | http://devserver_url:<port>/static/<relative_path>* |
Chris Sosa | 76e44b9 | 2013-01-31 12:11:38 -0800 | [diff] [blame] | 740 | |
| 741 | Note for this example, relative path is the archive_url stripped of its |
| 742 | basename i.e. path/ in the examples above. Specific example: |
| 743 | |
| 744 | gs://chromeos-image-archive/x86-mario-release/R26-3920.0.0 |
| 745 | |
| 746 | Will get staged to: |
| 747 | |
joychen | ed64b22 | 2013-06-21 16:39:34 -0700 | [diff] [blame] | 748 | http://devserver_url:<port>/static/x86-mario-release/R26-3920.0.0 |
Congbin Guo | 3afae6c | 2019-08-13 16:29:42 -0700 | [diff] [blame] | 749 | |
| 750 | Args: |
| 751 | archive_url: Google Storage URL for the build. |
| 752 | local_path: Local path for the build. |
| 753 | delete_source: Only meaningful with local_path. bool to indicate if the |
| 754 | source files should be deleted. This is especially useful when staging |
| 755 | a file locally in resource constrained environments as it allows us to |
| 756 | move the relevant files locally instead of copying them. |
| 757 | async: True to return without waiting for download to complete. |
| 758 | artifacts: Comma separated list of named artifacts to download. |
| 759 | These are defined in artifact_info and have their implementation |
| 760 | in build_artifact.py. |
| 761 | files: Comma separated list of files to stage. These |
| 762 | will be available as is in the corresponding static directory with no |
| 763 | custom post-processing. |
| 764 | clean: True to remove any previously staged artifacts first. |
Chris Sosa | 76e44b9 | 2013-01-31 12:11:38 -0800 | [diff] [blame] | 765 | """ |
Gabe Black | 3b56720 | 2015-09-23 14:07:59 -0700 | [diff] [blame] | 766 | dl, factory = _get_downloader_and_factory(kwargs) |
| 767 | |
Dan Shi | 59ae709 | 2013-06-04 14:37:27 -0700 | [diff] [blame] | 768 | with DevServerRoot._staging_thread_count_lock: |
| 769 | DevServerRoot._staging_thread_count += 1 |
| 770 | try: |
Laurence Goodby | f5c958d | 2016-01-14 18:23:56 -0800 | [diff] [blame] | 771 | boolean_string = kwargs.get('clean') |
| 772 | clean = xbuddy.XBuddy.ParseBoolean(boolean_string) |
| 773 | if clean and os.path.exists(dl.GetBuildDir()): |
| 774 | _Log('Removing %s' % dl.GetBuildDir()) |
| 775 | shutil.rmtree(dl.GetBuildDir()) |
Achuith Bhandarkar | 2a1fcd8 | 2019-10-17 17:45:58 -0700 | [diff] [blame] | 776 | is_async = kwargs.get('async', False) |
| 777 | dl.Download(factory, is_async=is_async) |
Dan Shi | 59ae709 | 2013-06-04 14:37:27 -0700 | [diff] [blame] | 778 | finally: |
| 779 | with DevServerRoot._staging_thread_count_lock: |
| 780 | DevServerRoot._staging_thread_count -= 1 |
Chris Sosa | 76e44b9 | 2013-01-31 12:11:38 -0800 | [diff] [blame] | 781 | return 'Success' |
Chris Sosa | cde6bf4 | 2012-05-31 18:36:39 -0700 | [diff] [blame] | 782 | |
| 783 | @cherrypy.expose |
xixuan | 52c2fba | 2016-05-20 17:02:48 -0700 | [diff] [blame] | 784 | def cros_au(self, **kwargs): |
| 785 | """Auto-update a CrOS DUT. |
| 786 | |
| 787 | Args: |
| 788 | kwargs: |
| 789 | host_name: the hostname of the DUT to auto-update. |
| 790 | build_name: the build name for update the DUT. |
| 791 | force_update: Force an update even if the version installed is the |
| 792 | same. Default: False. |
| 793 | full_update: If True, do not run stateful update, directly force a full |
| 794 | reimage. If False, try stateful update first if the dut is already |
| 795 | installed with the same version. |
| 796 | async: Whether the auto_update function is ran in the background. |
David Riley | ee75de2 | 2017-11-02 10:48:15 -0700 | [diff] [blame] | 797 | quick_provision: Whether the quick provision path is attempted first. |
xixuan | 52c2fba | 2016-05-20 17:02:48 -0700 | [diff] [blame] | 798 | |
| 799 | Returns: |
| 800 | A tuple includes two elements: |
| 801 | a boolean variable represents whether the auto-update process is |
| 802 | successfully started. |
| 803 | an integer represents the background auto-update process id. |
| 804 | """ |
| 805 | _check_base_args_for_auto_update(kwargs) |
| 806 | |
| 807 | host_name = kwargs['host_name'] |
| 808 | build_name = kwargs['build_name'] |
| 809 | force_update = _parse_boolean_arg(kwargs, 'force_update') |
| 810 | full_update = _parse_boolean_arg(kwargs, 'full_update') |
Achuith Bhandarkar | 2a1fcd8 | 2019-10-17 17:45:58 -0700 | [diff] [blame] | 811 | is_async = _parse_boolean_arg(kwargs, 'async') |
xixuan | ac89ce8 | 2016-11-30 16:48:20 -0800 | [diff] [blame] | 812 | original_build = _parse_string_arg(kwargs, 'original_build') |
David Haddock | 90e4944 | 2017-04-07 19:14:09 -0700 | [diff] [blame] | 813 | payload_filename = _parse_string_arg(kwargs, 'payload_filename') |
David Haddock | 2055961 | 2017-06-28 22:15:08 -0700 | [diff] [blame] | 814 | clobber_stateful = _parse_boolean_arg(kwargs, 'clobber_stateful') |
David Riley | ee75de2 | 2017-11-02 10:48:15 -0700 | [diff] [blame] | 815 | quick_provision = _parse_boolean_arg(kwargs, 'quick_provision') |
| 816 | |
| 817 | devserver_url = updater.GetDevserverUrl() |
| 818 | static_url = updater.GetStaticUrl() |
xixuan | 52c2fba | 2016-05-20 17:02:48 -0700 | [diff] [blame] | 819 | |
Mike Frysinger | 88e277f | 2020-03-02 17:20:46 -0500 | [diff] [blame] | 820 | # Command of running auto-update. |
| 821 | cmd = ['cros_update', '--hostname', host_name, '-b', build_name, |
| 822 | '--static_dir', updater.static_dir] |
| 823 | |
| 824 | # The original_build's format is like: link/3428.210.0 |
| 825 | # The corresponding release_archive_url's format is like: |
| 826 | # gs://chromeos-releases/stable-channel/link/3428.210.0 |
| 827 | if original_build: |
| 828 | release_archive_url = _build_uri_from_build_name(original_build) |
| 829 | # First staging the stateful.tgz synchronousely. |
| 830 | self.stage(files='stateful.tgz', is_async=False, |
| 831 | archive_url=release_archive_url) |
| 832 | cmd += ['--original_build', original_build] |
| 833 | |
| 834 | if force_update: |
| 835 | cmd += ['--force_update'] |
| 836 | |
| 837 | if full_update: |
| 838 | cmd += ['--full_update'] |
| 839 | |
| 840 | if payload_filename: |
| 841 | cmd += ['--payload_filename', payload_filename] |
| 842 | |
| 843 | if clobber_stateful: |
| 844 | cmd += ['--clobber_stateful'] |
| 845 | |
| 846 | if quick_provision: |
| 847 | cmd += ['--quick_provision'] |
| 848 | |
| 849 | if devserver_url: |
| 850 | cmd += ['--devserver_url', devserver_url] |
| 851 | |
| 852 | if static_url: |
| 853 | cmd += ['--static_url', static_url] |
| 854 | |
Achuith Bhandarkar | 2a1fcd8 | 2019-10-17 17:45:58 -0700 | [diff] [blame] | 855 | if is_async: |
Amin Hassani | 78520ae | 2019-10-29 13:26:51 -0700 | [diff] [blame] | 856 | p = subprocess.Popen(cmd, preexec_fn=os.setsid) |
xixuan | 2a0970a | 2016-08-10 12:12:44 -0700 | [diff] [blame] | 857 | pid = os.getpgid(p.pid) |
xixuan | 52c2fba | 2016-05-20 17:02:48 -0700 | [diff] [blame] | 858 | |
| 859 | # Pre-write status in the track_status_file before the first call of |
| 860 | # 'get_au_status' to make sure that the track_status_file exists. |
xixuan | 2a0970a | 2016-08-10 12:12:44 -0700 | [diff] [blame] | 861 | progress_tracker = cros_update_progress.AUProgress(host_name, pid) |
xixuan | 52c2fba | 2016-05-20 17:02:48 -0700 | [diff] [blame] | 862 | progress_tracker.WriteStatus('CrOS update is just started.') |
| 863 | |
xixuan | 2a0970a | 2016-08-10 12:12:44 -0700 | [diff] [blame] | 864 | return json.dumps((True, pid)) |
xixuan | 52c2fba | 2016-05-20 17:02:48 -0700 | [diff] [blame] | 865 | else: |
Mike Frysinger | 88e277f | 2020-03-02 17:20:46 -0500 | [diff] [blame] | 866 | subprocess.check_call(cmd) |
xixuan | 27d5044 | 2017-08-09 10:38:25 -0700 | [diff] [blame] | 867 | return json.dumps((True, -1)) |
xixuan | 52c2fba | 2016-05-20 17:02:48 -0700 | [diff] [blame] | 868 | |
| 869 | @cherrypy.expose |
| 870 | def get_au_status(self, **kwargs): |
| 871 | """Check if the auto-update task is finished. |
| 872 | |
| 873 | It handles 4 cases: |
| 874 | 1. If an error exists in the track_status_file, delete the track file and |
| 875 | raise it. |
| 876 | 2. If cros-update process is finished, delete the file and return the |
| 877 | success result. |
| 878 | 3. If the process is not running, delete the track file and raise an error |
| 879 | about 'the process is terminated due to unknown reason'. |
| 880 | 4. If the track_status_file does not exist, kill the process if it exists, |
| 881 | and raise the IOError. |
| 882 | |
| 883 | Args: |
| 884 | kwargs: |
| 885 | host_name: the hostname of the DUT to auto-update. |
| 886 | pid: the background process id of cros-update. |
| 887 | |
| 888 | Returns: |
xixuan | 28d9907 | 2016-10-06 12:24:16 -0700 | [diff] [blame] | 889 | A dict with three elements: |
xixuan | 52c2fba | 2016-05-20 17:02:48 -0700 | [diff] [blame] | 890 | a boolean variable represents whether the auto-update process is |
| 891 | finished. |
| 892 | a string represents the current auto-update process status. |
| 893 | For example, 'Transfer Devserver/Stateful Update Package'. |
xixuan | 28d9907 | 2016-10-06 12:24:16 -0700 | [diff] [blame] | 894 | a detailed error message paragraph if there exists an Auto-Update |
| 895 | error, in which the last line shows the main exception. Empty |
| 896 | string otherwise. |
xixuan | 52c2fba | 2016-05-20 17:02:48 -0700 | [diff] [blame] | 897 | """ |
| 898 | if 'host_name' not in kwargs: |
Amin Hassani | 722e096 | 2019-11-15 15:45:31 -0800 | [diff] [blame] | 899 | raise DevServerHTTPError(http_client.INTERNAL_SERVER_ERROR, |
| 900 | KEY_ERROR_MSG % 'host_name') |
xixuan | 52c2fba | 2016-05-20 17:02:48 -0700 | [diff] [blame] | 901 | |
| 902 | if 'pid' not in kwargs: |
Amin Hassani | 722e096 | 2019-11-15 15:45:31 -0800 | [diff] [blame] | 903 | raise DevServerHTTPError(http_client.INTERNAL_SERVER_ERROR, |
| 904 | KEY_ERROR_MSG % 'pid') |
xixuan | 52c2fba | 2016-05-20 17:02:48 -0700 | [diff] [blame] | 905 | |
| 906 | host_name = kwargs['host_name'] |
| 907 | pid = kwargs['pid'] |
| 908 | progress_tracker = cros_update_progress.AUProgress(host_name, pid) |
| 909 | |
xixuan | 28d9907 | 2016-10-06 12:24:16 -0700 | [diff] [blame] | 910 | result_dict = {'finished': False, 'status': '', 'detailed_error_msg': ''} |
xixuan | 52c2fba | 2016-05-20 17:02:48 -0700 | [diff] [blame] | 911 | try: |
| 912 | result = progress_tracker.ReadStatus() |
| 913 | if result.startswith(cros_update_progress.ERROR_TAG): |
xixuan | 28d9907 | 2016-10-06 12:24:16 -0700 | [diff] [blame] | 914 | result_dict['detailed_error_msg'] = result[len( |
| 915 | cros_update_progress.ERROR_TAG):] |
xixuan | 28681fd | 2016-11-23 11:13:56 -0800 | [diff] [blame] | 916 | elif result == cros_update_progress.FINISHED: |
xixuan | 28d9907 | 2016-10-06 12:24:16 -0700 | [diff] [blame] | 917 | result_dict['finished'] = True |
| 918 | result_dict['status'] = result |
xixuan | 28681fd | 2016-11-23 11:13:56 -0800 | [diff] [blame] | 919 | elif not cros_update_progress.IsProcessAlive(pid): |
xixuan | 28d9907 | 2016-10-06 12:24:16 -0700 | [diff] [blame] | 920 | result_dict['detailed_error_msg'] = ( |
| 921 | 'Cros_update process terminated midway due to unknown reason. ' |
| 922 | 'Last update status was %s' % result) |
xixuan | 28681fd | 2016-11-23 11:13:56 -0800 | [diff] [blame] | 923 | else: |
| 924 | result_dict['status'] = result |
| 925 | except IOError as e: |
| 926 | if pid and cros_update_progress.IsProcessAlive(pid): |
xixuan | 2a0970a | 2016-08-10 12:12:44 -0700 | [diff] [blame] | 927 | os.killpg(int(pid), signal.SIGKILL) |
xixuan | 52c2fba | 2016-05-20 17:02:48 -0700 | [diff] [blame] | 928 | |
xixuan | 28681fd | 2016-11-23 11:13:56 -0800 | [diff] [blame] | 929 | result_dict['detailed_error_msg'] = str(e) |
| 930 | |
| 931 | return json.dumps(result_dict) |
xixuan | 52c2fba | 2016-05-20 17:02:48 -0700 | [diff] [blame] | 932 | |
| 933 | @cherrypy.expose |
David Riley | 6d5fca0 | 2017-10-31 10:35:47 -0700 | [diff] [blame] | 934 | def post_au_status(self, status, **kwargs): |
| 935 | """Updates the status of an auto-update task. |
| 936 | |
| 937 | Callers will need to POST to this URL with a body of MIME-type |
| 938 | "multipart/form-data". |
| 939 | The body should include a single argument, 'status', containing the |
| 940 | AU status to record. |
| 941 | |
| 942 | Args: |
| 943 | status: The updated status. |
| 944 | kwargs: |
| 945 | host_name: the hostname of the DUT to auto-update. |
| 946 | pid: the background process id of cros-update. |
| 947 | """ |
| 948 | if 'host_name' not in kwargs: |
Amin Hassani | 722e096 | 2019-11-15 15:45:31 -0800 | [diff] [blame] | 949 | raise DevServerHTTPError(http_client.INTERNAL_SERVER_ERROR, |
| 950 | KEY_ERROR_MSG % 'host_name') |
David Riley | 6d5fca0 | 2017-10-31 10:35:47 -0700 | [diff] [blame] | 951 | |
| 952 | if 'pid' not in kwargs: |
Amin Hassani | 722e096 | 2019-11-15 15:45:31 -0800 | [diff] [blame] | 953 | raise DevServerHTTPError(http_client.INTERNAL_SERVER_ERROR, |
| 954 | KEY_ERROR_MSG % 'pid') |
David Riley | 6d5fca0 | 2017-10-31 10:35:47 -0700 | [diff] [blame] | 955 | |
| 956 | host_name = kwargs['host_name'] |
| 957 | pid = kwargs['pid'] |
David Riley | 3cea258 | 2017-11-24 22:03:01 -0800 | [diff] [blame] | 958 | status = status.rstrip() |
| 959 | _Log('Recording status for %s (%s): %s' % (host_name, pid, status)) |
David Riley | 6d5fca0 | 2017-10-31 10:35:47 -0700 | [diff] [blame] | 960 | progress_tracker = cros_update_progress.AUProgress(host_name, pid) |
| 961 | |
David Riley | 3cea258 | 2017-11-24 22:03:01 -0800 | [diff] [blame] | 962 | progress_tracker.WriteStatus(status) |
David Riley | 6d5fca0 | 2017-10-31 10:35:47 -0700 | [diff] [blame] | 963 | |
| 964 | return 'True' |
| 965 | |
| 966 | @cherrypy.expose |
xixuan | 52c2fba | 2016-05-20 17:02:48 -0700 | [diff] [blame] | 967 | def handler_cleanup(self, **kwargs): |
xixuan | 3bc974e | 2016-10-18 17:21:43 -0700 | [diff] [blame] | 968 | """Clean track status log and temp directory for CrOS auto-update process. |
xixuan | 52c2fba | 2016-05-20 17:02:48 -0700 | [diff] [blame] | 969 | |
| 970 | Args: |
| 971 | kwargs: |
| 972 | host_name: the hostname of the DUT to auto-update. |
| 973 | pid: the background process id of cros-update. |
| 974 | """ |
| 975 | if 'host_name' not in kwargs: |
Amin Hassani | 722e096 | 2019-11-15 15:45:31 -0800 | [diff] [blame] | 976 | raise DevServerHTTPError(http_client.INTERNAL_SERVER_ERROR, |
| 977 | KEY_ERROR_MSG % 'host_name') |
xixuan | 52c2fba | 2016-05-20 17:02:48 -0700 | [diff] [blame] | 978 | |
| 979 | if 'pid' not in kwargs: |
Amin Hassani | 722e096 | 2019-11-15 15:45:31 -0800 | [diff] [blame] | 980 | raise DevServerHTTPError(http_client.INTERNAL_SERVER_ERROR, |
| 981 | KEY_ERROR_MSG % 'pid') |
xixuan | 52c2fba | 2016-05-20 17:02:48 -0700 | [diff] [blame] | 982 | |
| 983 | host_name = kwargs['host_name'] |
| 984 | pid = kwargs['pid'] |
| 985 | cros_update_progress.DelTrackStatusFile(host_name, pid) |
xixuan | 3bc974e | 2016-10-18 17:21:43 -0700 | [diff] [blame] | 986 | cros_update_progress.DelAUTempDirectory(host_name, pid) |
xixuan | 52c2fba | 2016-05-20 17:02:48 -0700 | [diff] [blame] | 987 | |
| 988 | @cherrypy.expose |
| 989 | def kill_au_proc(self, **kwargs): |
| 990 | """Kill CrOS auto-update process using given process id. |
| 991 | |
| 992 | Args: |
| 993 | kwargs: |
| 994 | host_name: Kill all the CrOS auto-update process of this host. |
| 995 | |
| 996 | Returns: |
| 997 | True if all processes are killed properly. |
| 998 | """ |
| 999 | if 'host_name' not in kwargs: |
Amin Hassani | 722e096 | 2019-11-15 15:45:31 -0800 | [diff] [blame] | 1000 | raise DevServerHTTPError(http_client.INTERNAL_SERVER_ERROR, |
| 1001 | KEY_ERROR_MSG % 'host_name') |
xixuan | 52c2fba | 2016-05-20 17:02:48 -0700 | [diff] [blame] | 1002 | |
xixuan | 447ad9d | 2017-02-28 14:46:20 -0800 | [diff] [blame] | 1003 | cur_pid = kwargs.get('pid') |
| 1004 | |
xixuan | 52c2fba | 2016-05-20 17:02:48 -0700 | [diff] [blame] | 1005 | host_name = kwargs['host_name'] |
xixuan | 3bc974e | 2016-10-18 17:21:43 -0700 | [diff] [blame] | 1006 | track_log_list = cros_update_progress.GetAllTrackStatusFileByHostName( |
| 1007 | host_name) |
xixuan | 52c2fba | 2016-05-20 17:02:48 -0700 | [diff] [blame] | 1008 | for log in track_log_list: |
| 1009 | # The track log's full path is: path/host_name_pid.log |
| 1010 | # Use splitext to remove file extension, then parse pid from the |
| 1011 | # filename. |
Congbin Guo | 3afae6c | 2019-08-13 16:29:42 -0700 | [diff] [blame] | 1012 | pid = os.path.splitext(os.path.basename(log))[0][len(host_name) + 1:] |
xixuan | 447ad9d | 2017-02-28 14:46:20 -0800 | [diff] [blame] | 1013 | _clear_process(host_name, pid) |
xixuan | 52c2fba | 2016-05-20 17:02:48 -0700 | [diff] [blame] | 1014 | |
xixuan | 447ad9d | 2017-02-28 14:46:20 -0800 | [diff] [blame] | 1015 | if cur_pid: |
| 1016 | _clear_process(host_name, cur_pid) |
xixuan | 52c2fba | 2016-05-20 17:02:48 -0700 | [diff] [blame] | 1017 | |
| 1018 | return 'True' |
| 1019 | |
| 1020 | @cherrypy.expose |
| 1021 | def collect_cros_au_log(self, **kwargs): |
| 1022 | """Collect CrOS auto-update log. |
| 1023 | |
| 1024 | Args: |
| 1025 | kwargs: |
| 1026 | host_name: the hostname of the DUT to auto-update. |
| 1027 | pid: the background process id of cros-update. |
| 1028 | |
| 1029 | Returns: |
David Haddock | 9f45963 | 2017-05-11 14:45:46 -0700 | [diff] [blame] | 1030 | A dictionary containing the execute log file and any hostlog files. |
xixuan | 52c2fba | 2016-05-20 17:02:48 -0700 | [diff] [blame] | 1031 | """ |
| 1032 | if 'host_name' not in kwargs: |
Amin Hassani | 722e096 | 2019-11-15 15:45:31 -0800 | [diff] [blame] | 1033 | raise DevServerHTTPError(http_client.INTERNAL_SERVER_ERROR, |
| 1034 | KEY_ERROR_MSG % 'host_name') |
xixuan | 52c2fba | 2016-05-20 17:02:48 -0700 | [diff] [blame] | 1035 | |
| 1036 | if 'pid' not in kwargs: |
Amin Hassani | 722e096 | 2019-11-15 15:45:31 -0800 | [diff] [blame] | 1037 | raise DevServerHTTPError(http_client.INTERNAL_SERVER_ERROR, |
| 1038 | KEY_ERROR_MSG % 'pid') |
xixuan | 52c2fba | 2016-05-20 17:02:48 -0700 | [diff] [blame] | 1039 | |
| 1040 | host_name = kwargs['host_name'] |
| 1041 | pid = kwargs['pid'] |
xixuan | 3bc974e | 2016-10-18 17:21:43 -0700 | [diff] [blame] | 1042 | |
| 1043 | # Fetch the execute log recorded by cros_update_progress. |
xixuan | 1bbfaba | 2016-10-13 17:53:22 -0700 | [diff] [blame] | 1044 | au_log = cros_update_progress.ReadExecuteLogFile(host_name, pid) |
| 1045 | cros_update_progress.DelExecuteLogFile(host_name, pid) |
David Haddock | 9f45963 | 2017-05-11 14:45:46 -0700 | [diff] [blame] | 1046 | # Fetch the cros_au host_logs if they exist |
| 1047 | au_hostlogs = cros_update_progress.ReadAUHostLogFiles(host_name, pid) |
| 1048 | return json.dumps({'cros_au_log': au_log, 'host_logs': au_hostlogs}) |
xixuan | 1bbfaba | 2016-10-13 17:53:22 -0700 | [diff] [blame] | 1049 | |
xixuan | 52c2fba | 2016-05-20 17:02:48 -0700 | [diff] [blame] | 1050 | @cherrypy.expose |
Dan Shi | 2f13686 | 2016-02-11 15:38:38 -0800 | [diff] [blame] | 1051 | def locate_file(self, **kwargs): |
| 1052 | """Get the path to the given file name. |
| 1053 | |
| 1054 | This method looks up the given file name inside specified build artifacts. |
| 1055 | One use case is to help caller to locate an apk file inside a build |
| 1056 | artifact. The location of the apk file could be different based on the |
| 1057 | branch and target. |
| 1058 | |
| 1059 | Args: |
| 1060 | file_name: Name of the file to look for. |
| 1061 | artifacts: A list of artifact names to search for the file. |
| 1062 | |
| 1063 | Returns: |
| 1064 | Path to the file with the given name. It's relative to the folder for the |
| 1065 | build, e.g., DATA/priv-app/sl4a/sl4a.apk |
Dan Shi | 2f13686 | 2016-02-11 15:38:38 -0800 | [diff] [blame] | 1066 | """ |
Sanika Kulkarni | d4496fd | 2020-02-04 17:26:25 -0800 | [diff] [blame] | 1067 | if is_deprecated_server(): |
| 1068 | raise DeprecatedRPCError('locate_file') |
| 1069 | |
Dan Shi | 2f13686 | 2016-02-11 15:38:38 -0800 | [diff] [blame] | 1070 | dl, _ = _get_downloader_and_factory(kwargs) |
| 1071 | try: |
Joe Brennan | 1691f8e | 2017-03-15 15:53:36 -0700 | [diff] [blame] | 1072 | file_name = kwargs['file_name'] |
Dan Shi | 2f13686 | 2016-02-11 15:38:38 -0800 | [diff] [blame] | 1073 | artifacts = kwargs['artifacts'] |
| 1074 | except KeyError: |
Amin Hassani | d4e3539 | 2019-10-03 11:02:44 -0700 | [diff] [blame] | 1075 | raise DevServerError( |
Congbin Guo | 4132a27 | 2019-08-20 12:32:14 -0700 | [diff] [blame] | 1076 | '`file_name` and `artifacts` are required to search ' |
| 1077 | 'for a file in build artifacts.') |
Dan Shi | 2f13686 | 2016-02-11 15:38:38 -0800 | [diff] [blame] | 1078 | build_path = dl.GetBuildDir() |
| 1079 | for artifact in artifacts: |
| 1080 | # Get the unzipped folder of the artifact. If it's not defined in |
| 1081 | # ARTIFACT_UNZIP_FOLDER_MAP, assume the files are unzipped to the build |
| 1082 | # directory directly. |
| 1083 | folder = artifact_info.ARTIFACT_UNZIP_FOLDER_MAP.get(artifact, '') |
| 1084 | artifact_path = os.path.join(build_path, folder) |
| 1085 | for root, _, filenames in os.walk(artifact_path): |
Joe Brennan | 1691f8e | 2017-03-15 15:53:36 -0700 | [diff] [blame] | 1086 | if file_name in set([f for f in filenames]): |
Dan Shi | 2f13686 | 2016-02-11 15:38:38 -0800 | [diff] [blame] | 1087 | return os.path.relpath(os.path.join(root, file_name), build_path) |
Amin Hassani | d4e3539 | 2019-10-03 11:02:44 -0700 | [diff] [blame] | 1088 | raise DevServerError( |
Congbin Guo | 4132a27 | 2019-08-20 12:32:14 -0700 | [diff] [blame] | 1089 | 'File `%s` can not be found in artifacts: %s' % (file_name, artifacts)) |
Dan Shi | 2f13686 | 2016-02-11 15:38:38 -0800 | [diff] [blame] | 1090 | |
| 1091 | @cherrypy.expose |
Simran Basi | 4baad08 | 2013-02-14 13:39:18 -0800 | [diff] [blame] | 1092 | def setup_telemetry(self, **kwargs): |
| 1093 | """Extracts and sets up telemetry |
| 1094 | |
| 1095 | This method goes through the telemetry deps packages, and stages them on |
| 1096 | the devserver to be used by the drones and the telemetry tests. |
| 1097 | |
| 1098 | Args: |
| 1099 | archive_url: Google Storage URL for the build. |
| 1100 | |
| 1101 | Returns: |
| 1102 | Path to the source folder for the telemetry codebase once it is staged. |
| 1103 | """ |
Gabe Black | 3b56720 | 2015-09-23 14:07:59 -0700 | [diff] [blame] | 1104 | dl = _get_downloader(kwargs) |
Simran Basi | 4baad08 | 2013-02-14 13:39:18 -0800 | [diff] [blame] | 1105 | |
Gabe Black | 3b56720 | 2015-09-23 14:07:59 -0700 | [diff] [blame] | 1106 | build_path = dl.GetBuildDir() |
Simran Basi | 4baad08 | 2013-02-14 13:39:18 -0800 | [diff] [blame] | 1107 | deps_path = os.path.join(build_path, 'autotest/packages') |
| 1108 | telemetry_path = os.path.join(build_path, TELEMETRY_FOLDER) |
| 1109 | src_folder = os.path.join(telemetry_path, 'src') |
| 1110 | |
| 1111 | with self._telemetry_lock_dict.lock(telemetry_path): |
| 1112 | if os.path.exists(src_folder): |
| 1113 | # Telemetry is already fully stage return |
| 1114 | return src_folder |
| 1115 | |
| 1116 | common_util.MkDirP(telemetry_path) |
| 1117 | |
| 1118 | # Copy over the required deps tar balls to the telemetry directory. |
| 1119 | for dep in TELEMETRY_DEPS: |
| 1120 | dep_path = os.path.join(deps_path, dep) |
Simran Basi | 0d07868 | 2013-03-22 16:40:04 -0700 | [diff] [blame] | 1121 | if not os.path.exists(dep_path): |
| 1122 | # This dep does not exist (could be new), do not extract it. |
| 1123 | continue |
Simran Basi | 4baad08 | 2013-02-14 13:39:18 -0800 | [diff] [blame] | 1124 | try: |
| 1125 | common_util.ExtractTarball(dep_path, telemetry_path) |
| 1126 | except common_util.CommonUtilError as e: |
| 1127 | shutil.rmtree(telemetry_path) |
Amin Hassani | d4e3539 | 2019-10-03 11:02:44 -0700 | [diff] [blame] | 1128 | raise DevServerError(str(e)) |
Simran Basi | 4baad08 | 2013-02-14 13:39:18 -0800 | [diff] [blame] | 1129 | |
| 1130 | # By default all the tarballs extract to test_src but some parts of |
| 1131 | # the telemetry code specifically hardcoded to exist inside of 'src'. |
| 1132 | test_src = os.path.join(telemetry_path, 'test_src') |
| 1133 | try: |
| 1134 | shutil.move(test_src, src_folder) |
| 1135 | except shutil.Error: |
| 1136 | # This can occur if src_folder already exists. Remove and retry move. |
| 1137 | shutil.rmtree(src_folder) |
Amin Hassani | d4e3539 | 2019-10-03 11:02:44 -0700 | [diff] [blame] | 1138 | raise DevServerError( |
Gabe Black | 3b56720 | 2015-09-23 14:07:59 -0700 | [diff] [blame] | 1139 | 'Failure in telemetry setup for build %s. Appears that the ' |
| 1140 | 'test_src to src move failed.' % dl.GetBuild()) |
Simran Basi | 4baad08 | 2013-02-14 13:39:18 -0800 | [diff] [blame] | 1141 | |
| 1142 | return src_folder |
| 1143 | |
| 1144 | @cherrypy.expose |
Chris Sosa | 76e44b9 | 2013-01-31 12:11:38 -0800 | [diff] [blame] | 1145 | def symbolicate_dump(self, minidump, **kwargs): |
Chris Masone | 816e38c | 2012-05-02 12:22:36 -0700 | [diff] [blame] | 1146 | """Symbolicates a minidump using pre-downloaded symbols, returns it. |
| 1147 | |
| 1148 | Callers will need to POST to this URL with a body of MIME-type |
| 1149 | "multipart/form-data". |
| 1150 | The body should include a single argument, 'minidump', containing the |
| 1151 | binary-formatted minidump to symbolicate. |
| 1152 | |
Chris Masone | 816e38c | 2012-05-02 12:22:36 -0700 | [diff] [blame] | 1153 | Args: |
Chris Sosa | 76e44b9 | 2013-01-31 12:11:38 -0800 | [diff] [blame] | 1154 | archive_url: Google Storage URL for the build. |
Chris Masone | 816e38c | 2012-05-02 12:22:36 -0700 | [diff] [blame] | 1155 | minidump: The binary minidump file to symbolicate. |
| 1156 | """ |
Sanika Kulkarni | d4496fd | 2020-02-04 17:26:25 -0800 | [diff] [blame] | 1157 | if is_deprecated_server(): |
| 1158 | raise DeprecatedRPCError('symbolicate_dump') |
| 1159 | |
Chris Sosa | 76e44b9 | 2013-01-31 12:11:38 -0800 | [diff] [blame] | 1160 | # Ensure the symbols have been staged. |
Dan Shi | f08fe49 | 2016-10-04 14:39:25 -0700 | [diff] [blame] | 1161 | # Try debug.tar.xz first, then debug.tgz |
| 1162 | for artifact in (artifact_info.SYMBOLS_ONLY, artifact_info.SYMBOLS): |
| 1163 | kwargs['artifacts'] = artifact |
| 1164 | dl = _get_downloader(kwargs) |
| 1165 | |
| 1166 | try: |
| 1167 | if self.stage(**kwargs) == 'Success': |
| 1168 | break |
| 1169 | except build_artifact.ArtifactDownloadError: |
| 1170 | continue |
| 1171 | else: |
Amin Hassani | d4e3539 | 2019-10-03 11:02:44 -0700 | [diff] [blame] | 1172 | raise DevServerError( |
Congbin Guo | 4132a27 | 2019-08-20 12:32:14 -0700 | [diff] [blame] | 1173 | 'Failed to stage symbols for %s' % dl.DescribeSource()) |
Chris Sosa | 76e44b9 | 2013-01-31 12:11:38 -0800 | [diff] [blame] | 1174 | |
Chris Masone | 816e38c | 2012-05-02 12:22:36 -0700 | [diff] [blame] | 1175 | to_return = '' |
| 1176 | with tempfile.NamedTemporaryFile() as local: |
| 1177 | while True: |
| 1178 | data = minidump.file.read(8192) |
| 1179 | if not data: |
| 1180 | break |
| 1181 | local.write(data) |
Chris Sosa | 76e44b9 | 2013-01-31 12:11:38 -0800 | [diff] [blame] | 1182 | |
Chris Masone | 816e38c | 2012-05-02 12:22:36 -0700 | [diff] [blame] | 1183 | local.flush() |
Chris Sosa | 76e44b9 | 2013-01-31 12:11:38 -0800 | [diff] [blame] | 1184 | |
Gabe Black | 3b56720 | 2015-09-23 14:07:59 -0700 | [diff] [blame] | 1185 | symbols_directory = os.path.join(dl.GetBuildDir(), 'debug', 'breakpad') |
Chris Sosa | 76e44b9 | 2013-01-31 12:11:38 -0800 | [diff] [blame] | 1186 | |
xixuan | ab74438 | 2017-04-27 10:41:27 -0700 | [diff] [blame] | 1187 | # The location of minidump_stackwalk is defined in chromeos-admin. |
Chris Sosa | 76e44b9 | 2013-01-31 12:11:38 -0800 | [diff] [blame] | 1188 | stackwalk = subprocess.Popen( |
xixuan | ab74438 | 2017-04-27 10:41:27 -0700 | [diff] [blame] | 1189 | ['/usr/local/bin/minidump_stackwalk', local.name, symbols_directory], |
Chris Sosa | 76e44b9 | 2013-01-31 12:11:38 -0800 | [diff] [blame] | 1190 | stdout=subprocess.PIPE, stderr=subprocess.PIPE) |
| 1191 | |
Chris Masone | 816e38c | 2012-05-02 12:22:36 -0700 | [diff] [blame] | 1192 | to_return, error_text = stackwalk.communicate() |
| 1193 | if stackwalk.returncode != 0: |
Amin Hassani | d4e3539 | 2019-10-03 11:02:44 -0700 | [diff] [blame] | 1194 | raise DevServerError( |
Congbin Guo | 4132a27 | 2019-08-20 12:32:14 -0700 | [diff] [blame] | 1195 | "Can't generate stack trace: %s (rc=%d)" % (error_text, |
| 1196 | stackwalk.returncode)) |
Chris Masone | 816e38c | 2012-05-02 12:22:36 -0700 | [diff] [blame] | 1197 | |
| 1198 | return to_return |
| 1199 | |
| 1200 | @cherrypy.expose |
Don Garrett | f84631a | 2014-01-07 18:21:26 -0800 | [diff] [blame] | 1201 | def latestbuild(self, **kwargs): |
Scott Zawalski | 1695453 | 2012-03-20 15:31:36 -0400 | [diff] [blame] | 1202 | """Return a string representing the latest build for a given target. |
| 1203 | |
| 1204 | Args: |
| 1205 | target: The build target, typically a combination of the board and the |
| 1206 | type of build e.g. x86-mario-release. |
| 1207 | milestone: The milestone to filter builds on. E.g. R16. Optional, if not |
| 1208 | provided the latest RXX build will be returned. |
Don Garrett | f84631a | 2014-01-07 18:21:26 -0800 | [diff] [blame] | 1209 | |
Scott Zawalski | 1695453 | 2012-03-20 15:31:36 -0400 | [diff] [blame] | 1210 | Returns: |
| 1211 | A string representation of the latest build if one exists, i.e. |
| 1212 | R19-1993.0.0-a1-b1480. |
| 1213 | An empty string if no latest could be found. |
| 1214 | """ |
Don Garrett | f84631a | 2014-01-07 18:21:26 -0800 | [diff] [blame] | 1215 | if not kwargs: |
Scott Zawalski | 1695453 | 2012-03-20 15:31:36 -0400 | [diff] [blame] | 1216 | return _PrintDocStringAsHTML(self.latestbuild) |
| 1217 | |
Don Garrett | f84631a | 2014-01-07 18:21:26 -0800 | [diff] [blame] | 1218 | if 'target' not in kwargs: |
Amin Hassani | 722e096 | 2019-11-15 15:45:31 -0800 | [diff] [blame] | 1219 | raise DevServerHTTPError(http_client.INTERNAL_SERVER_ERROR, |
| 1220 | 'Error: target= is required!') |
Dan Shi | 61305df | 2015-10-26 16:52:35 -0700 | [diff] [blame] | 1221 | |
| 1222 | if _is_android_build_request(kwargs): |
| 1223 | branch = kwargs.get('branch', None) |
| 1224 | target = kwargs.get('target', None) |
| 1225 | if not target or not branch: |
Amin Hassani | d4e3539 | 2019-10-03 11:02:44 -0700 | [diff] [blame] | 1226 | raise DevServerError('Both target and branch must be specified to query' |
| 1227 | ' for the latest Android build.') |
Dan Shi | 61305df | 2015-10-26 16:52:35 -0700 | [diff] [blame] | 1228 | return android_build.BuildAccessor.GetLatestBuildID(target, branch) |
| 1229 | |
Scott Zawalski | 1695453 | 2012-03-20 15:31:36 -0400 | [diff] [blame] | 1230 | try: |
Gilad Arnold | c65330c | 2012-09-20 15:17:48 -0700 | [diff] [blame] | 1231 | return common_util.GetLatestBuildVersion( |
Don Garrett | f84631a | 2014-01-07 18:21:26 -0800 | [diff] [blame] | 1232 | updater.static_dir, kwargs['target'], |
| 1233 | milestone=kwargs.get('milestone')) |
Gilad Arnold | 17fe03d | 2012-10-02 10:05:01 -0700 | [diff] [blame] | 1234 | except common_util.CommonUtilError as errmsg: |
Amin Hassani | 722e096 | 2019-11-15 15:45:31 -0800 | [diff] [blame] | 1235 | raise DevServerHTTPError(http_client.INTERNAL_SERVER_ERROR, |
| 1236 | str(errmsg)) |
Scott Zawalski | 1695453 | 2012-03-20 15:31:36 -0400 | [diff] [blame] | 1237 | |
| 1238 | @cherrypy.expose |
xixuan | 7efd000 | 2016-04-14 15:34:01 -0700 | [diff] [blame] | 1239 | def list_suite_controls(self, **kwargs): |
| 1240 | """Return a list of contents of all known control files. |
| 1241 | |
| 1242 | Example URL: |
| 1243 | To List all control files' content: |
| 1244 | http://dev-server/list_suite_controls?suite_name=bvt& |
| 1245 | build=daisy_spring-release/R29-4279.0.0 |
| 1246 | |
| 1247 | Args: |
| 1248 | build: The build i.e. x86-alex-release/R18-1514.0.0-a1-b1450. |
| 1249 | suite_name: List the control files belonging to that suite. |
| 1250 | |
| 1251 | Returns: |
Dan Shi | a1cd652 | 2016-04-18 16:07:21 -0700 | [diff] [blame] | 1252 | A dictionary of all control files's path to its content for given suite. |
xixuan | 7efd000 | 2016-04-14 15:34:01 -0700 | [diff] [blame] | 1253 | """ |
Sanika Kulkarni | d4496fd | 2020-02-04 17:26:25 -0800 | [diff] [blame] | 1254 | if is_deprecated_server(): |
| 1255 | raise DeprecatedRPCError('list_suite_controls') |
| 1256 | |
xixuan | 7efd000 | 2016-04-14 15:34:01 -0700 | [diff] [blame] | 1257 | if not kwargs: |
| 1258 | return _PrintDocStringAsHTML(self.controlfiles) |
| 1259 | |
| 1260 | if 'build' not in kwargs: |
Amin Hassani | 722e096 | 2019-11-15 15:45:31 -0800 | [diff] [blame] | 1261 | raise DevServerHTTPError(http_client.INTERNAL_SERVER_ERROR, |
| 1262 | 'Error: build= is required!') |
xixuan | 7efd000 | 2016-04-14 15:34:01 -0700 | [diff] [blame] | 1263 | |
| 1264 | if 'suite_name' not in kwargs: |
Amin Hassani | 722e096 | 2019-11-15 15:45:31 -0800 | [diff] [blame] | 1265 | raise DevServerHTTPError(http_client.INTERNAL_SERVER_ERROR, |
| 1266 | 'Error: suite_name= is required!') |
xixuan | 7efd000 | 2016-04-14 15:34:01 -0700 | [diff] [blame] | 1267 | |
| 1268 | control_file_list = [ |
| 1269 | line.rstrip() for line in common_util.GetControlFileListForSuite( |
| 1270 | updater.static_dir, kwargs['build'], |
| 1271 | kwargs['suite_name']).splitlines()] |
| 1272 | |
Dan Shi | a1cd652 | 2016-04-18 16:07:21 -0700 | [diff] [blame] | 1273 | control_file_content_dict = {} |
xixuan | 7efd000 | 2016-04-14 15:34:01 -0700 | [diff] [blame] | 1274 | for control_path in control_file_list: |
Dan Shi | a1cd652 | 2016-04-18 16:07:21 -0700 | [diff] [blame] | 1275 | control_file_content_dict[control_path] = (common_util.GetControlFile( |
xixuan | 7efd000 | 2016-04-14 15:34:01 -0700 | [diff] [blame] | 1276 | updater.static_dir, kwargs['build'], control_path)) |
| 1277 | |
Dan Shi | a1cd652 | 2016-04-18 16:07:21 -0700 | [diff] [blame] | 1278 | return json.dumps(control_file_content_dict) |
xixuan | 7efd000 | 2016-04-14 15:34:01 -0700 | [diff] [blame] | 1279 | |
| 1280 | @cherrypy.expose |
Don Garrett | f84631a | 2014-01-07 18:21:26 -0800 | [diff] [blame] | 1281 | def controlfiles(self, **kwargs): |
Scott Zawalski | 4647ce6 | 2012-01-03 17:17:28 -0500 | [diff] [blame] | 1282 | """Return a control file or a list of all known control files. |
| 1283 | |
| 1284 | Example URL: |
| 1285 | To List all control files: |
beeps | bd33724 | 2013-07-09 22:44:06 -0700 | [diff] [blame] | 1286 | http://dev-server/controlfiles?suite_name=&build=daisy_spring-release/R29-4279.0.0 |
| 1287 | To List all control files for, say, the bvt suite: |
| 1288 | http://dev-server/controlfiles?suite_name=bvt&build=daisy_spring-release/R29-4279.0.0 |
Scott Zawalski | 4647ce6 | 2012-01-03 17:17:28 -0500 | [diff] [blame] | 1289 | To return the contents of a path: |
Scott Zawalski | 84a39c9 | 2012-01-13 15:12:42 -0500 | [diff] [blame] | 1290 | http://dev-server/controlfiles?board=x86-alex-release&build=R18-1514.0.0&control_path=client/sleeptest/control |
Scott Zawalski | 4647ce6 | 2012-01-03 17:17:28 -0500 | [diff] [blame] | 1291 | |
| 1292 | Args: |
Scott Zawalski | 84a39c9 | 2012-01-13 15:12:42 -0500 | [diff] [blame] | 1293 | build: The build i.e. x86-alex-release/R18-1514.0.0-a1-b1450. |
Scott Zawalski | 4647ce6 | 2012-01-03 17:17:28 -0500 | [diff] [blame] | 1294 | control_path: If you want the contents of a control file set this |
| 1295 | to the path. E.g. client/site_tests/sleeptest/control |
| 1296 | Optional, if not provided return a list of control files is returned. |
beeps | bd33724 | 2013-07-09 22:44:06 -0700 | [diff] [blame] | 1297 | suite_name: If control_path is not specified but a suite_name is |
| 1298 | specified, list the control files belonging to that suite instead of |
| 1299 | all control files. The empty string for suite_name will list all control |
| 1300 | files for the build. |
Don Garrett | f84631a | 2014-01-07 18:21:26 -0800 | [diff] [blame] | 1301 | |
Scott Zawalski | 4647ce6 | 2012-01-03 17:17:28 -0500 | [diff] [blame] | 1302 | Returns: |
| 1303 | Contents of a control file if control_path is provided. |
| 1304 | A list of control files if no control_path is provided. |
| 1305 | """ |
Sanika Kulkarni | d4496fd | 2020-02-04 17:26:25 -0800 | [diff] [blame] | 1306 | if is_deprecated_server(): |
| 1307 | raise DeprecatedRPCError('controlfiles') |
| 1308 | |
Don Garrett | f84631a | 2014-01-07 18:21:26 -0800 | [diff] [blame] | 1309 | if not kwargs: |
Scott Zawalski | 4647ce6 | 2012-01-03 17:17:28 -0500 | [diff] [blame] | 1310 | return _PrintDocStringAsHTML(self.controlfiles) |
| 1311 | |
Don Garrett | f84631a | 2014-01-07 18:21:26 -0800 | [diff] [blame] | 1312 | if 'build' not in kwargs: |
Amin Hassani | 722e096 | 2019-11-15 15:45:31 -0800 | [diff] [blame] | 1313 | raise DevServerHTTPError(http_client.INTERNAL_SERVER_ERROR, |
| 1314 | 'Error: build= is required!') |
Scott Zawalski | 4647ce6 | 2012-01-03 17:17:28 -0500 | [diff] [blame] | 1315 | |
Don Garrett | f84631a | 2014-01-07 18:21:26 -0800 | [diff] [blame] | 1316 | if 'control_path' not in kwargs: |
| 1317 | if 'suite_name' in kwargs and kwargs['suite_name']: |
beeps | bd33724 | 2013-07-09 22:44:06 -0700 | [diff] [blame] | 1318 | return common_util.GetControlFileListForSuite( |
Don Garrett | f84631a | 2014-01-07 18:21:26 -0800 | [diff] [blame] | 1319 | updater.static_dir, kwargs['build'], kwargs['suite_name']) |
beeps | bd33724 | 2013-07-09 22:44:06 -0700 | [diff] [blame] | 1320 | else: |
| 1321 | return common_util.GetControlFileList( |
Don Garrett | f84631a | 2014-01-07 18:21:26 -0800 | [diff] [blame] | 1322 | updater.static_dir, kwargs['build']) |
Scott Zawalski | 4647ce6 | 2012-01-03 17:17:28 -0500 | [diff] [blame] | 1323 | else: |
Gilad Arnold | c65330c | 2012-09-20 15:17:48 -0700 | [diff] [blame] | 1324 | return common_util.GetControlFile( |
Don Garrett | f84631a | 2014-01-07 18:21:26 -0800 | [diff] [blame] | 1325 | updater.static_dir, kwargs['build'], kwargs['control_path']) |
Frank Farzan | 4016087 | 2011-12-12 18:39:18 -0800 | [diff] [blame] | 1326 | |
| 1327 | @cherrypy.expose |
Simran Basi | 99e63c0 | 2014-05-20 10:39:52 -0700 | [diff] [blame] | 1328 | def xbuddy_translate(self, *args, **kwargs): |
Yu-Ju Hong | 1bdb7a9 | 2014-04-10 16:02:11 -0700 | [diff] [blame] | 1329 | """Translates an xBuddy path to a real path to artifact if it exists. |
| 1330 | |
| 1331 | Args: |
Simran Basi | 99e63c0 | 2014-05-20 10:39:52 -0700 | [diff] [blame] | 1332 | args: An xbuddy path in the form of {local|remote}/build_id/artifact. |
| 1333 | Local searches the devserver's static directory. Remote searches a |
| 1334 | Google Storage image archive. |
| 1335 | |
| 1336 | Kwargs: |
| 1337 | image_dir: Google Storage image archive to search in if requesting a |
| 1338 | remote artifact. If none uses the default bucket. |
Yu-Ju Hong | 1bdb7a9 | 2014-04-10 16:02:11 -0700 | [diff] [blame] | 1339 | |
| 1340 | Returns: |
Simran Basi | 99e63c0 | 2014-05-20 10:39:52 -0700 | [diff] [blame] | 1341 | String in the format of build_id/artifact as stored on the local server |
| 1342 | or in Google Storage. |
Yu-Ju Hong | 1bdb7a9 | 2014-04-10 16:02:11 -0700 | [diff] [blame] | 1343 | """ |
Sanika Kulkarni | d4496fd | 2020-02-04 17:26:25 -0800 | [diff] [blame] | 1344 | if is_deprecated_server(): |
| 1345 | raise DeprecatedRPCError('xbuddy_translate') |
| 1346 | |
Simran Basi | 99e63c0 | 2014-05-20 10:39:52 -0700 | [diff] [blame] | 1347 | build_id, filename = self._xbuddy.Translate( |
Gabe Black | 3b56720 | 2015-09-23 14:07:59 -0700 | [diff] [blame] | 1348 | args, image_dir=kwargs.get('image_dir')) |
Yu-Ju Hong | 1bdb7a9 | 2014-04-10 16:02:11 -0700 | [diff] [blame] | 1349 | response = os.path.join(build_id, filename) |
| 1350 | _Log('Path translation requested, returning: %s', response) |
| 1351 | return response |
| 1352 | |
| 1353 | @cherrypy.expose |
joychen | eaf4cfc | 2013-07-02 08:38:57 -0700 | [diff] [blame] | 1354 | def xbuddy(self, *args, **kwargs): |
| 1355 | """The full xBuddy call, returns resource specified by path_parts. |
joychen | 3cb228e | 2013-06-12 12:13:13 -0700 | [diff] [blame] | 1356 | |
| 1357 | Args: |
joychen | eaf4cfc | 2013-07-02 08:38:57 -0700 | [diff] [blame] | 1358 | path_parts: the path following xbuddy/ in the call url is split into the |
joychen | 121fc9b | 2013-08-02 14:30:30 -0700 | [diff] [blame] | 1359 | components of the path. The path can be understood as |
| 1360 | "{local|remote}/build_id/artifact" where build_id is composed of |
| 1361 | "board/version." |
joychen | eaf4cfc | 2013-07-02 08:38:57 -0700 | [diff] [blame] | 1362 | |
joychen | 121fc9b | 2013-08-02 14:30:30 -0700 | [diff] [blame] | 1363 | The first path element is optional, and can be "remote" or "local" |
| 1364 | If local (the default), devserver will not attempt to access Google |
| 1365 | Storage, and will only search the static directory for the files. |
| 1366 | If remote, devserver will try to obtain the artifact off GS if it's |
| 1367 | not found locally. |
| 1368 | The board is the familiar board name, optionally suffixed. |
| 1369 | The version can be the google storage version number, and may also be |
| 1370 | any of a number of xBuddy defined version aliases that will be |
| 1371 | translated into the latest built image that fits the description. |
| 1372 | Defaults to latest. |
| 1373 | The artifact is one of a number of image or artifact aliases used by |
| 1374 | xbuddy, defined in xbuddy:ALIASES. Defaults to test. |
joychen | eaf4cfc | 2013-07-02 08:38:57 -0700 | [diff] [blame] | 1375 | |
| 1376 | Kwargs: |
Yu-Ju Hong | 51495eb | 2013-12-12 17:08:43 -0800 | [diff] [blame] | 1377 | for_update: {true|false} |
Amin Hassani | e9ffb86 | 2019-09-25 17:10:40 -0700 | [diff] [blame] | 1378 | if true, prepares the update payloads for the image, |
Yu-Ju Hong | 51495eb | 2013-12-12 17:08:43 -0800 | [diff] [blame] | 1379 | and returns the update uri to pass to the |
| 1380 | update_engine_client. |
joychen | 3cb228e | 2013-06-12 12:13:13 -0700 | [diff] [blame] | 1381 | return_dir: {true|false} |
| 1382 | if set to true, returns the url to the update.gz |
Yu-Ju Hong | 51495eb | 2013-12-12 17:08:43 -0800 | [diff] [blame] | 1383 | relative_path: {true|false} |
| 1384 | if set to true, returns the relative path to the payload |
| 1385 | directory from static_dir. |
joychen | 3cb228e | 2013-06-12 12:13:13 -0700 | [diff] [blame] | 1386 | Example URL: |
joychen | eaf4cfc | 2013-07-02 08:38:57 -0700 | [diff] [blame] | 1387 | http://host:port/xbuddy/x86-generic/R26-4000.0.0/test |
joychen | 3cb228e | 2013-06-12 12:13:13 -0700 | [diff] [blame] | 1388 | or |
joychen | eaf4cfc | 2013-07-02 08:38:57 -0700 | [diff] [blame] | 1389 | http://host:port/xbuddy/x86-generic/R26-4000.0.0/test?return_dir=true |
joychen | 3cb228e | 2013-06-12 12:13:13 -0700 | [diff] [blame] | 1390 | |
| 1391 | Returns: |
Yu-Ju Hong | 51495eb | 2013-12-12 17:08:43 -0800 | [diff] [blame] | 1392 | If |for_update|, returns a redirect to the image or update file |
| 1393 | on the devserver. E.g., |
| 1394 | http://host:port/static/archive/x86-generic-release/R26-4000.0.0/ |
| 1395 | chromium-test-image.bin |
| 1396 | If |return_dir|, return a uri to the folder where the artifact is. E.g., |
| 1397 | http://host:port/static/x86-generic-release/R26-4000.0.0/ |
| 1398 | If |relative_path| is true, return a relative path the folder where the |
| 1399 | payloads are. E.g., |
| 1400 | archive/x86-generic-release/R26-4000.0.0 |
joychen | 3cb228e | 2013-06-12 12:13:13 -0700 | [diff] [blame] | 1401 | """ |
Sanika Kulkarni | d4496fd | 2020-02-04 17:26:25 -0800 | [diff] [blame] | 1402 | if is_deprecated_server(): |
| 1403 | raise DeprecatedRPCError('xbuddy') |
| 1404 | |
Chris Sosa | 7549080 | 2013-09-30 17:21:45 -0700 | [diff] [blame] | 1405 | boolean_string = kwargs.get('for_update') |
| 1406 | for_update = xbuddy.XBuddy.ParseBoolean(boolean_string) |
Yu-Ju Hong | 51495eb | 2013-12-12 17:08:43 -0800 | [diff] [blame] | 1407 | boolean_string = kwargs.get('return_dir') |
| 1408 | return_dir = xbuddy.XBuddy.ParseBoolean(boolean_string) |
| 1409 | boolean_string = kwargs.get('relative_path') |
| 1410 | relative_path = xbuddy.XBuddy.ParseBoolean(boolean_string) |
joychen | 121fc9b | 2013-08-02 14:30:30 -0700 | [diff] [blame] | 1411 | |
Yu-Ju Hong | 51495eb | 2013-12-12 17:08:43 -0800 | [diff] [blame] | 1412 | if return_dir and relative_path: |
Amin Hassani | 722e096 | 2019-11-15 15:45:31 -0800 | [diff] [blame] | 1413 | raise DevServerHTTPError( |
Amin Hassani | d4e3539 | 2019-10-03 11:02:44 -0700 | [diff] [blame] | 1414 | http_client.INTERNAL_SERVER_ERROR, |
Amin Hassani | 08e42d2 | 2019-06-03 00:31:30 -0700 | [diff] [blame] | 1415 | 'Cannot specify both return_dir and relative_path') |
Chris Sosa | 7549080 | 2013-09-30 17:21:45 -0700 | [diff] [blame] | 1416 | |
| 1417 | # For updates, we optimize downloading of test images. |
| 1418 | file_name = None |
| 1419 | build_id = None |
| 1420 | if for_update: |
| 1421 | try: |
Yu-Ju Hong | 1bdb7a9 | 2014-04-10 16:02:11 -0700 | [diff] [blame] | 1422 | build_id = self._xbuddy.StageTestArtifactsForUpdate(args) |
Chris Sosa | 7549080 | 2013-09-30 17:21:45 -0700 | [diff] [blame] | 1423 | except build_artifact.ArtifactDownloadError: |
| 1424 | build_id = None |
| 1425 | |
| 1426 | if not build_id: |
| 1427 | build_id, file_name = self._xbuddy.Get(args) |
| 1428 | |
Yu-Ju Hong | 51495eb | 2013-12-12 17:08:43 -0800 | [diff] [blame] | 1429 | if for_update: |
Amin Hassani | e9ffb86 | 2019-09-25 17:10:40 -0700 | [diff] [blame] | 1430 | _Log('Payloads requested.') |
Yu-Ju Hong | 51495eb | 2013-12-12 17:08:43 -0800 | [diff] [blame] | 1431 | # Forces payload to be in cache and symlinked into build_id dir. |
Amin Hassani | e9ffb86 | 2019-09-25 17:10:40 -0700 | [diff] [blame] | 1432 | updater.GetUpdateForLabel(build_id) |
Yu-Ju Hong | 51495eb | 2013-12-12 17:08:43 -0800 | [diff] [blame] | 1433 | |
| 1434 | response = None |
| 1435 | if return_dir: |
| 1436 | response = os.path.join(cherrypy.request.base, 'static', build_id) |
| 1437 | _Log('Directory requested, returning: %s', response) |
| 1438 | elif relative_path: |
| 1439 | response = build_id |
| 1440 | _Log('Relative path requested, returning: %s', response) |
| 1441 | elif for_update: |
| 1442 | response = os.path.join(cherrypy.request.base, 'update', build_id) |
| 1443 | _Log('Update URI requested, returning: %s', response) |
joychen | 3cb228e | 2013-06-12 12:13:13 -0700 | [diff] [blame] | 1444 | else: |
Yu-Ju Hong | 51495eb | 2013-12-12 17:08:43 -0800 | [diff] [blame] | 1445 | # Redirect to download the payload if no kwargs are set. |
joychen | 121fc9b | 2013-08-02 14:30:30 -0700 | [diff] [blame] | 1446 | build_id = '/' + os.path.join('static', build_id, file_name) |
Yu-Ju Hong | 51495eb | 2013-12-12 17:08:43 -0800 | [diff] [blame] | 1447 | _Log('Payload requested, returning: %s', build_id) |
joychen | 121fc9b | 2013-08-02 14:30:30 -0700 | [diff] [blame] | 1448 | raise cherrypy.HTTPRedirect(build_id, 302) |
joychen | 3cb228e | 2013-06-12 12:13:13 -0700 | [diff] [blame] | 1449 | |
Yu-Ju Hong | 51495eb | 2013-12-12 17:08:43 -0800 | [diff] [blame] | 1450 | return response |
| 1451 | |
joychen | 3cb228e | 2013-06-12 12:13:13 -0700 | [diff] [blame] | 1452 | @cherrypy.expose |
| 1453 | def xbuddy_list(self): |
| 1454 | """Lists the currently available images & time since last access. |
| 1455 | |
Gilad Arnold | 452fd27 | 2014-02-04 11:09:28 -0800 | [diff] [blame] | 1456 | Returns: |
| 1457 | A string representation of a list of tuples [(build_id, time since last |
| 1458 | access),...] |
joychen | 3cb228e | 2013-06-12 12:13:13 -0700 | [diff] [blame] | 1459 | """ |
Sanika Kulkarni | d4496fd | 2020-02-04 17:26:25 -0800 | [diff] [blame] | 1460 | if is_deprecated_server(): |
| 1461 | raise DeprecatedRPCError('xbuddy') |
| 1462 | |
joychen | 3cb228e | 2013-06-12 12:13:13 -0700 | [diff] [blame] | 1463 | return self._xbuddy.List() |
| 1464 | |
| 1465 | @cherrypy.expose |
| 1466 | def xbuddy_capacity(self): |
Gilad Arnold | 452fd27 | 2014-02-04 11:09:28 -0800 | [diff] [blame] | 1467 | """Returns the number of images cached by xBuddy.""" |
Sanika Kulkarni | d4496fd | 2020-02-04 17:26:25 -0800 | [diff] [blame] | 1468 | if is_deprecated_server(): |
| 1469 | raise DeprecatedRPCError('xbuddy_capacity') |
| 1470 | |
joychen | 3cb228e | 2013-06-12 12:13:13 -0700 | [diff] [blame] | 1471 | return self._xbuddy.Capacity() |
| 1472 | |
| 1473 | @cherrypy.expose |
Chris Sosa | 7c93136 | 2010-10-11 19:49:01 -0700 | [diff] [blame] | 1474 | def index(self): |
Gilad Arnold | f8f769f | 2012-09-24 08:43:01 -0700 | [diff] [blame] | 1475 | """Presents a welcome message and documentation links.""" |
Sanika Kulkarni | d4496fd | 2020-02-04 17:26:25 -0800 | [diff] [blame] | 1476 | if is_deprecated_server(): |
| 1477 | raise DeprecatedRPCError('index') |
| 1478 | |
Congbin Guo | 6bc3218 | 2019-08-20 17:54:30 -0700 | [diff] [blame] | 1479 | html_template = ( |
| 1480 | 'Welcome to the Dev Server!<br>\n' |
| 1481 | '<br>\n' |
| 1482 | 'Here are the available methods, click for documentation:<br>\n' |
| 1483 | '<br>\n' |
| 1484 | '%s') |
| 1485 | |
| 1486 | exposed_methods = [] |
| 1487 | for app in cherrypy.tree.apps.values(): |
| 1488 | exposed_methods += _FindExposedMethods( |
| 1489 | app.root, app.script_name.lstrip('/'), |
| 1490 | unlisted=self._UNLISTED_METHODS) |
| 1491 | |
| 1492 | return html_template % '<br>\n'.join( |
| 1493 | ['<a href=doc/%s>%s</a>' % (name, name) |
| 1494 | for name in sorted(exposed_methods)]) |
Gilad Arnold | f8f769f | 2012-09-24 08:43:01 -0700 | [diff] [blame] | 1495 | |
| 1496 | @cherrypy.expose |
| 1497 | def doc(self, *args): |
| 1498 | """Shows the documentation for available methods / URLs. |
| 1499 | |
Amin Hassani | 08e42d2 | 2019-06-03 00:31:30 -0700 | [diff] [blame] | 1500 | Examples: |
Gilad Arnold | f8f769f | 2012-09-24 08:43:01 -0700 | [diff] [blame] | 1501 | http://myhost/doc/update |
| 1502 | """ |
Sanika Kulkarni | d4496fd | 2020-02-04 17:26:25 -0800 | [diff] [blame] | 1503 | if is_deprecated_server(): |
| 1504 | raise DeprecatedRPCError('doc') |
| 1505 | |
Gilad Arnold | d5ebaaa | 2012-10-02 11:52:38 -0700 | [diff] [blame] | 1506 | name = '/'.join(args) |
Congbin Guo | 6bc3218 | 2019-08-20 17:54:30 -0700 | [diff] [blame] | 1507 | method = _GetExposedMethod(name) |
Gilad Arnold | f8f769f | 2012-09-24 08:43:01 -0700 | [diff] [blame] | 1508 | if not method: |
Amin Hassani | d4e3539 | 2019-10-03 11:02:44 -0700 | [diff] [blame] | 1509 | raise DevServerError("No exposed method named `%s'" % name) |
Gilad Arnold | f8f769f | 2012-09-24 08:43:01 -0700 | [diff] [blame] | 1510 | if not method.__doc__: |
Amin Hassani | d4e3539 | 2019-10-03 11:02:44 -0700 | [diff] [blame] | 1511 | raise DevServerError("No documentation for exposed method `%s'" % name) |
Gilad Arnold | f8f769f | 2012-09-24 08:43:01 -0700 | [diff] [blame] | 1512 | return '<pre>\n%s</pre>' % method.__doc__ |
Chris Sosa | 7c93136 | 2010-10-11 19:49:01 -0700 | [diff] [blame] | 1513 | |
Dale Curtis | c9aaf3a | 2011-08-09 15:47:40 -0700 | [diff] [blame] | 1514 | @cherrypy.expose |
Amin Hassani | 6eec879 | 2020-01-09 14:06:48 -0800 | [diff] [blame] | 1515 | def update(self, *args, **kwargs): |
Gilad Arnold | f8f769f | 2012-09-24 08:43:01 -0700 | [diff] [blame] | 1516 | """Handles an update check from a Chrome OS client. |
| 1517 | |
| 1518 | The HTTP request should contain the standard Omaha-style XML blob. The URL |
| 1519 | line may contain an additional intermediate path to the update payload. |
| 1520 | |
joychen | 121fc9b | 2013-08-02 14:30:30 -0700 | [diff] [blame] | 1521 | This request can be handled in one of 4 ways, depending on the devsever |
| 1522 | settings and intermediate path. |
joychen | b0dfe55 | 2013-07-30 10:02:06 -0700 | [diff] [blame] | 1523 | |
Amin Hassani | e9ffb86 | 2019-09-25 17:10:40 -0700 | [diff] [blame] | 1524 | 1. No intermediate path. DEPRECATED |
joychen | 121fc9b | 2013-08-02 14:30:30 -0700 | [diff] [blame] | 1525 | |
| 1526 | 2. Path explicitly invokes XBuddy |
| 1527 | If there is a path given, it can explicitly invoke xbuddy by prefixing it |
| 1528 | with 'xbuddy'. This path is then used to acquire an image binary for the |
| 1529 | devserver to generate an update payload from. Devserver then serves this |
| 1530 | payload. |
| 1531 | |
| 1532 | 3. Path is left for the devserver to interpret. |
| 1533 | If the path given doesn't explicitly invoke xbuddy, devserver will attempt |
| 1534 | to generate a payload from the test image in that directory and serve it. |
| 1535 | |
joychen | 121fc9b | 2013-08-02 14:30:30 -0700 | [diff] [blame] | 1536 | Examples: |
joychen | 121fc9b | 2013-08-02 14:30:30 -0700 | [diff] [blame] | 1537 | 2. Explicitly invoke xbuddy |
| 1538 | update_engine_client --omaha_url= |
| 1539 | http://myhost/update/xbuddy/remote/board/version/dev |
| 1540 | This would go to GS to download the dev image for the board, from which |
| 1541 | the devserver would generate a payload to serve. |
| 1542 | |
| 1543 | 3. Give a path for devserver to interpret |
| 1544 | update_engine_client --omaha_url=http://myhost/update/some/random/path |
| 1545 | This would attempt, in order to: |
| 1546 | a) Generate an update from a test image binary if found in |
| 1547 | static_dir/some/random/path. |
| 1548 | b) Serve an update payload found in static_dir/some/random/path. |
| 1549 | c) Hope that some/random/path takes the form "board/version" and |
| 1550 | and attempt to download an update payload for that board/version |
| 1551 | from GS. |
Gilad Arnold | f8f769f | 2012-09-24 08:43:01 -0700 | [diff] [blame] | 1552 | """ |
joychen | 121fc9b | 2013-08-02 14:30:30 -0700 | [diff] [blame] | 1553 | label = '/'.join(args) |
Gilad Arnold | 286a006 | 2012-01-12 13:47:02 -0800 | [diff] [blame] | 1554 | body_length = int(cherrypy.request.headers.get('Content-Length', 0)) |
Chris Sosa | 7c93136 | 2010-10-11 19:49:01 -0700 | [diff] [blame] | 1555 | data = cherrypy.request.rfile.read(body_length) |
Amin Hassani | 6aa075c | 2020-02-21 18:36:44 +0000 | [diff] [blame] | 1556 | |
Amin Hassani | 6eec879 | 2020-01-09 14:06:48 -0800 | [diff] [blame] | 1557 | return updater.HandleUpdatePing(data, label, **kwargs) |
Chris Sosa | 0356d3b | 2010-09-16 15:46:22 -0700 | [diff] [blame] | 1558 | |
Dan Shi | f5ce2de | 2013-04-25 16:06:32 -0700 | [diff] [blame] | 1559 | |
Chris Sosa | dbc2008 | 2012-12-10 13:39:11 -0800 | [diff] [blame] | 1560 | def _CleanCache(cache_dir, wipe): |
| 1561 | """Wipes any excess cached items in the cache_dir. |
| 1562 | |
| 1563 | Args: |
| 1564 | cache_dir: the directory we are wiping from. |
| 1565 | wipe: If True, wipe all the contents -- not just the excess. |
| 1566 | """ |
| 1567 | if wipe: |
| 1568 | # Clear the cache and exit on error. |
| 1569 | cmd = 'rm -rf %s/*' % cache_dir |
| 1570 | if os.system(cmd) != 0: |
| 1571 | _Log('Failed to clear the cache with %s' % cmd) |
| 1572 | sys.exit(1) |
| 1573 | else: |
| 1574 | # Clear all but the last N cached updates |
| 1575 | cmd = ('cd %s; ls -tr | head --lines=-%d | xargs rm -rf' % |
| 1576 | (cache_dir, CACHED_ENTRIES)) |
| 1577 | if os.system(cmd) != 0: |
| 1578 | _Log('Failed to clean up old delta cache files with %s' % cmd) |
| 1579 | sys.exit(1) |
| 1580 | |
| 1581 | |
Chris Sosa | 3ae4dc1 | 2013-03-29 11:47:00 -0700 | [diff] [blame] | 1582 | def _AddTestingOptions(parser): |
| 1583 | group = optparse.OptionGroup( |
| 1584 | parser, 'Advanced Testing Options', 'These are used by test scripts and ' |
| 1585 | 'developers writing integration tests utilizing the devserver. They are ' |
| 1586 | 'not intended to be really used outside the scope of someone ' |
| 1587 | 'knowledgable about the test.') |
| 1588 | group.add_option('--exit', |
| 1589 | action='store_true', |
Amin Hassani | e9ffb86 | 2019-09-25 17:10:40 -0700 | [diff] [blame] | 1590 | help='do not start the server (yet clear cache)') |
Chris Sosa | 3ae4dc1 | 2013-03-29 11:47:00 -0700 | [diff] [blame] | 1591 | group.add_option('--host_log', |
| 1592 | action='store_true', default=False, |
| 1593 | help='record history of host update events (/api/hostlog)') |
| 1594 | group.add_option('--max_updates', |
Gabe Black | 3b56720 | 2015-09-23 14:07:59 -0700 | [diff] [blame] | 1595 | metavar='NUM', default=-1, type='int', |
Chris Sosa | 3ae4dc1 | 2013-03-29 11:47:00 -0700 | [diff] [blame] | 1596 | help='maximum number of update checks handled positively ' |
| 1597 | '(default: unlimited)') |
Chris Sosa | 3ae4dc1 | 2013-03-29 11:47:00 -0700 | [diff] [blame] | 1598 | group.add_option('--proxy_port', |
| 1599 | metavar='PORT', default=None, type='int', |
| 1600 | help='port to have the client connect to -- basically the ' |
| 1601 | 'devserver lies to the update to tell it to get the payload ' |
| 1602 | 'from a different port that will proxy the request back to ' |
| 1603 | 'the devserver. The proxy must be managed outside the ' |
| 1604 | 'devserver.') |
Chris Sosa | 3ae4dc1 | 2013-03-29 11:47:00 -0700 | [diff] [blame] | 1605 | parser.add_option_group(group) |
| 1606 | |
| 1607 | |
| 1608 | def _AddUpdateOptions(parser): |
| 1609 | group = optparse.OptionGroup( |
| 1610 | parser, 'Autoupdate Options', 'These options can be used to change ' |
Amin Hassani | e9ffb86 | 2019-09-25 17:10:40 -0700 | [diff] [blame] | 1611 | 'how the devserver serve update payloads. Please ' |
Chris Sosa | 3ae4dc1 | 2013-03-29 11:47:00 -0700 | [diff] [blame] | 1612 | 'note that all of these option affect how a payload is generated and so ' |
| 1613 | 'do not work in archive-only mode.') |
Amin Hassani | 6eec879 | 2020-01-09 14:06:48 -0800 | [diff] [blame] | 1614 | # TODO(crbug/1004487): Deprecate critical_update. |
Chris Sosa | 3ae4dc1 | 2013-03-29 11:47:00 -0700 | [diff] [blame] | 1615 | group.add_option('--critical_update', |
| 1616 | action='store_true', default=False, |
| 1617 | help='Present update payload as critical') |
Chris Sosa | 3ae4dc1 | 2013-03-29 11:47:00 -0700 | [diff] [blame] | 1618 | group.add_option('--payload', |
| 1619 | metavar='PATH', |
| 1620 | help='use the update payload from specified directory ' |
| 1621 | '(update.gz).') |
Chris Sosa | 3ae4dc1 | 2013-03-29 11:47:00 -0700 | [diff] [blame] | 1622 | parser.add_option_group(group) |
| 1623 | |
| 1624 | |
| 1625 | def _AddProductionOptions(parser): |
| 1626 | group = optparse.OptionGroup( |
| 1627 | parser, 'Advanced Server Options', 'These options can be used to changed ' |
| 1628 | 'for advanced server behavior.') |
Chris Sosa | 3ae4dc1 | 2013-03-29 11:47:00 -0700 | [diff] [blame] | 1629 | group.add_option('--clear_cache', |
| 1630 | action='store_true', default=False, |
| 1631 | help='At startup, removes all cached entries from the' |
Amin Hassani | d4e3539 | 2019-10-03 11:02:44 -0700 | [diff] [blame] | 1632 | "devserver's cache.") |
Chris Sosa | 3ae4dc1 | 2013-03-29 11:47:00 -0700 | [diff] [blame] | 1633 | group.add_option('--logfile', |
| 1634 | metavar='PATH', |
| 1635 | help='log output to this file instead of stdout') |
Chris Sosa | 855b893 | 2013-08-21 13:24:55 -0700 | [diff] [blame] | 1636 | group.add_option('--pidfile', |
| 1637 | metavar='PATH', |
| 1638 | help='path to output a pid file for the server.') |
Gilad Arnold | 11fbef4 | 2014-02-10 11:04:13 -0800 | [diff] [blame] | 1639 | group.add_option('--portfile', |
| 1640 | metavar='PATH', |
| 1641 | help='path to output the port number being served on.') |
Chris Sosa | 3ae4dc1 | 2013-03-29 11:47:00 -0700 | [diff] [blame] | 1642 | group.add_option('--production', |
| 1643 | action='store_true', default=False, |
| 1644 | help='have the devserver use production values when ' |
| 1645 | 'starting up. This includes using more threads and ' |
| 1646 | 'performing less logging.') |
| 1647 | parser.add_option_group(group) |
| 1648 | |
| 1649 | |
Paul Hobbs | ef4e070 | 2016-06-27 17:01:42 -0700 | [diff] [blame] | 1650 | def MakeLogHandler(logfile): |
J. Richard Barnette | 3d977b8 | 2013-04-23 11:05:19 -0700 | [diff] [blame] | 1651 | """Create a LogHandler instance used to log all messages.""" |
| 1652 | hdlr_cls = handlers.TimedRotatingFileHandler |
| 1653 | hdlr = hdlr_cls(logfile, when=_LOG_ROTATION_TIME, |
xixuan | 3d48bff | 2017-01-30 19:00:09 -0800 | [diff] [blame] | 1654 | interval=_LOG_ROTATION_INTERVAL, |
J. Richard Barnette | 3d977b8 | 2013-04-23 11:05:19 -0700 | [diff] [blame] | 1655 | backupCount=_LOG_ROTATION_BACKUP) |
Chris Sosa | 855b893 | 2013-08-21 13:24:55 -0700 | [diff] [blame] | 1656 | hdlr.setFormatter(cplogging.logfmt) |
J. Richard Barnette | 3d977b8 | 2013-04-23 11:05:19 -0700 | [diff] [blame] | 1657 | return hdlr |
| 1658 | |
| 1659 | |
Chris Sosa | cde6bf4 | 2012-05-31 18:36:39 -0700 | [diff] [blame] | 1660 | def main(): |
Chris Sosa | 3ae4dc1 | 2013-03-29 11:47:00 -0700 | [diff] [blame] | 1661 | usage = '\n\n'.join(['usage: %prog [options]', __doc__]) |
Gilad Arnold | 286a006 | 2012-01-12 13:47:02 -0800 | [diff] [blame] | 1662 | parser = optparse.OptionParser(usage=usage) |
joychen | ed64b22 | 2013-06-21 16:39:34 -0700 | [diff] [blame] | 1663 | |
| 1664 | # get directory that the devserver is run from |
| 1665 | devserver_dir = os.path.dirname(os.path.abspath(sys.argv[0])) |
joychen | 84d1377 | 2013-08-06 09:17:23 -0700 | [diff] [blame] | 1666 | default_static_dir = '%s/static' % devserver_dir |
joychen | ed64b22 | 2013-06-21 16:39:34 -0700 | [diff] [blame] | 1667 | parser.add_option('--static_dir', |
Gilad Arnold | 9714d9b | 2012-10-04 10:09:42 -0700 | [diff] [blame] | 1668 | metavar='PATH', |
joychen | 84d1377 | 2013-08-06 09:17:23 -0700 | [diff] [blame] | 1669 | default=default_static_dir, |
joychen | ed64b22 | 2013-06-21 16:39:34 -0700 | [diff] [blame] | 1670 | help='writable static directory') |
Gilad Arnold | 9714d9b | 2012-10-04 10:09:42 -0700 | [diff] [blame] | 1671 | parser.add_option('--port', |
| 1672 | default=8080, type='int', |
Gilad Arnold | af696d1 | 2014-02-14 13:13:28 -0800 | [diff] [blame] | 1673 | help=('port for the dev server to use; if zero, binds to ' |
| 1674 | 'an arbitrary available port (default: 8080)')) |
Gilad Arnold | 9714d9b | 2012-10-04 10:09:42 -0700 | [diff] [blame] | 1675 | parser.add_option('-t', '--test_image', |
| 1676 | action='store_true', |
joychen | 121fc9b | 2013-08-02 14:30:30 -0700 | [diff] [blame] | 1677 | help='Deprecated.') |
joychen | 5260b9a | 2013-07-16 14:48:01 -0700 | [diff] [blame] | 1678 | parser.add_option('-x', '--xbuddy_manage_builds', |
| 1679 | action='store_true', |
| 1680 | default=False, |
| 1681 | help='If set, allow xbuddy to manage images in' |
| 1682 | 'build/images.') |
Dan Shi | 72b1613 | 2015-10-08 12:10:33 -0700 | [diff] [blame] | 1683 | parser.add_option('-a', '--android_build_credential', |
| 1684 | default=None, |
| 1685 | help='Path to a json file which contains the credential ' |
| 1686 | 'needed to access Android builds.') |
Sanika Kulkarni | d4496fd | 2020-02-04 17:26:25 -0800 | [diff] [blame] | 1687 | parser.add_option('--infra_removal', |
| 1688 | action='store_true', default=False, |
| 1689 | help='If option is present, some RPCs will be disabled to ' |
| 1690 | 'help with infra removal efforts. See ' |
| 1691 | 'go/devserver-deprecation') |
Chris Sosa | 3ae4dc1 | 2013-03-29 11:47:00 -0700 | [diff] [blame] | 1692 | _AddProductionOptions(parser) |
| 1693 | _AddUpdateOptions(parser) |
| 1694 | _AddTestingOptions(parser) |
Chris Sosa | 7c93136 | 2010-10-11 19:49:01 -0700 | [diff] [blame] | 1695 | (options, _) = parser.parse_args() |
rtc@google.com | 21a5ca3 | 2009-11-04 18:23:23 +0000 | [diff] [blame] | 1696 | |
J. Richard Barnette | 3d977b8 | 2013-04-23 11:05:19 -0700 | [diff] [blame] | 1697 | # Handle options that must be set globally in cherrypy. Do this |
| 1698 | # work up front, because calls to _Log() below depend on this |
| 1699 | # initialization. |
| 1700 | if options.production: |
| 1701 | cherrypy.config.update({'environment': 'production'}) |
Sanika Kulkarni | d4496fd | 2020-02-04 17:26:25 -0800 | [diff] [blame] | 1702 | cherrypy.config.update({'infra_removal': options.infra_removal}) |
J. Richard Barnette | 3d977b8 | 2013-04-23 11:05:19 -0700 | [diff] [blame] | 1703 | if not options.logfile: |
| 1704 | cherrypy.config.update({'log.screen': True}) |
| 1705 | else: |
| 1706 | cherrypy.config.update({'log.error_file': '', |
| 1707 | 'log.access_file': ''}) |
Paul Hobbs | ef4e070 | 2016-06-27 17:01:42 -0700 | [diff] [blame] | 1708 | hdlr = MakeLogHandler(options.logfile) |
J. Richard Barnette | 3d977b8 | 2013-04-23 11:05:19 -0700 | [diff] [blame] | 1709 | # Pylint can't seem to process these two calls properly |
| 1710 | # pylint: disable=E1101 |
| 1711 | cherrypy.log.access_log.addHandler(hdlr) |
| 1712 | cherrypy.log.error_log.addHandler(hdlr) |
| 1713 | # pylint: enable=E1101 |
| 1714 | |
joychen | ed64b22 | 2013-06-21 16:39:34 -0700 | [diff] [blame] | 1715 | # set static_dir, from which everything will be served |
joychen | 84d1377 | 2013-08-06 09:17:23 -0700 | [diff] [blame] | 1716 | options.static_dir = os.path.realpath(options.static_dir) |
Chris Sosa | 0356d3b | 2010-09-16 15:46:22 -0700 | [diff] [blame] | 1717 | |
joychen | ed64b22 | 2013-06-21 16:39:34 -0700 | [diff] [blame] | 1718 | cache_dir = os.path.join(options.static_dir, 'cache') |
J. Richard Barnette | 3d977b8 | 2013-04-23 11:05:19 -0700 | [diff] [blame] | 1719 | # If our devserver is only supposed to serve payloads, we shouldn't be |
| 1720 | # mucking with the cache at all. If the devserver hadn't previously |
| 1721 | # generated a cache and is expected, the caller is using it wrong. |
joychen | 7c2054a | 2013-07-25 11:14:07 -0700 | [diff] [blame] | 1722 | if os.path.exists(cache_dir): |
Chris Sosa | dbc2008 | 2012-12-10 13:39:11 -0800 | [diff] [blame] | 1723 | _CleanCache(cache_dir, options.clear_cache) |
Chris Sosa | 6b8c374 | 2011-01-31 12:12:17 -0800 | [diff] [blame] | 1724 | else: |
| 1725 | os.makedirs(cache_dir) |
Don Garrett | f90edf0 | 2010-11-16 17:36:14 -0800 | [diff] [blame] | 1726 | |
Chris Sosa | dbc2008 | 2012-12-10 13:39:11 -0800 | [diff] [blame] | 1727 | _Log('Using cache directory %s' % cache_dir) |
joychen | ed64b22 | 2013-06-21 16:39:34 -0700 | [diff] [blame] | 1728 | _Log('Serving from %s' % options.static_dir) |
rtc@google.com | 21a5ca3 | 2009-11-04 18:23:23 +0000 | [diff] [blame] | 1729 | |
Amin Hassani | e9ffb86 | 2019-09-25 17:10:40 -0700 | [diff] [blame] | 1730 | _xbuddy = xbuddy.XBuddy(manage_builds=options.xbuddy_manage_builds, |
joychen | 121fc9b | 2013-08-02 14:30:30 -0700 | [diff] [blame] | 1731 | static_dir=options.static_dir) |
Chris Sosa | 7549080 | 2013-09-30 17:21:45 -0700 | [diff] [blame] | 1732 | if options.clear_cache and options.xbuddy_manage_builds: |
| 1733 | _xbuddy.CleanCache() |
joychen | 121fc9b | 2013-08-02 14:30:30 -0700 | [diff] [blame] | 1734 | |
Chris Sosa | 6a3697f | 2013-01-29 16:44:43 -0800 | [diff] [blame] | 1735 | # We allow global use here to share with cherrypy classes. |
| 1736 | # pylint: disable=W0603 |
Chris Sosa | cde6bf4 | 2012-05-31 18:36:39 -0700 | [diff] [blame] | 1737 | global updater |
Andrew de los Reyes | 5262080 | 2010-04-12 13:40:07 -0700 | [diff] [blame] | 1738 | updater = autoupdate.Autoupdate( |
joychen | 121fc9b | 2013-08-02 14:30:30 -0700 | [diff] [blame] | 1739 | _xbuddy, |
joychen | ed64b22 | 2013-06-21 16:39:34 -0700 | [diff] [blame] | 1740 | static_dir=options.static_dir, |
Gilad Arnold | 0c9c860 | 2012-10-02 23:58:58 -0700 | [diff] [blame] | 1741 | payload_path=options.payload, |
Don Garrett | 0ad0937 | 2010-12-06 16:20:30 -0800 | [diff] [blame] | 1742 | proxy_port=options.proxy_port, |
Satoru Takabayashi | d733cbe | 2011-11-15 09:36:32 -0800 | [diff] [blame] | 1743 | critical_update=options.critical_update, |
Gilad Arnold | a564b4b | 2012-10-04 10:32:44 -0700 | [diff] [blame] | 1744 | max_updates=options.max_updates, |
Gilad Arnold | 8318eac | 2012-10-04 12:52:23 -0700 | [diff] [blame] | 1745 | host_log=options.host_log, |
Chris Sosa | 0f1ec84 | 2011-02-14 16:33:22 -0800 | [diff] [blame] | 1746 | ) |
Chris Sosa | 7c93136 | 2010-10-11 19:49:01 -0700 | [diff] [blame] | 1747 | |
J. Richard Barnette | 3d977b8 | 2013-04-23 11:05:19 -0700 | [diff] [blame] | 1748 | if options.exit: |
| 1749 | return |
Chris Sosa | 2f1c41e | 2012-07-10 14:32:33 -0700 | [diff] [blame] | 1750 | |
joychen | 3cb228e | 2013-06-12 12:13:13 -0700 | [diff] [blame] | 1751 | dev_server = DevServerRoot(_xbuddy) |
Congbin Guo | 3afae6c | 2019-08-13 16:29:42 -0700 | [diff] [blame] | 1752 | health_checker_app = health_checker.Root(dev_server, options.static_dir) |
joychen | 3cb228e | 2013-06-12 12:13:13 -0700 | [diff] [blame] | 1753 | |
Amin Hassani | c5af426 | 2019-11-13 13:37:20 -0800 | [diff] [blame] | 1754 | # Patch CherryPy to support binding to any available port (--port=0) only for |
| 1755 | # cherrypy versions smaller or equal to 3.2.2. |
| 1756 | # |
| 1757 | # TODO(crbug/1006305): Remove this once we have deprecated omaha_devserver.py |
| 1758 | # in the autotests as that is the only use case. |
| 1759 | # |
| 1760 | # pylint: disable=no-member |
| 1761 | if (distutils.version.StrictVersion(cherrypy.__version__) <= |
| 1762 | distutils.version.StrictVersion('3.2.2')): |
| 1763 | cherrypy_ext.ZeroPortPatcher.DoPatch(cherrypy) |
| 1764 | # pylint: enable=no-member |
| 1765 | |
Chris Sosa | 855b893 | 2013-08-21 13:24:55 -0700 | [diff] [blame] | 1766 | if options.pidfile: |
| 1767 | plugins.PIDFile(cherrypy.engine, options.pidfile).subscribe() |
| 1768 | |
Gilad Arnold | 11fbef4 | 2014-02-10 11:04:13 -0800 | [diff] [blame] | 1769 | if options.portfile: |
| 1770 | cherrypy_ext.PortFile(cherrypy.engine, options.portfile).subscribe() |
| 1771 | |
Dan Shi | afd5c6c | 2016-01-07 10:27:03 -0800 | [diff] [blame] | 1772 | if (options.android_build_credential and |
| 1773 | os.path.exists(options.android_build_credential)): |
| 1774 | try: |
| 1775 | with open(options.android_build_credential) as f: |
| 1776 | android_build.BuildAccessor.credential_info = json.load(f) |
| 1777 | except ValueError as e: |
| 1778 | _Log('Failed to load the android build credential: %s. Error: %s.' % |
| 1779 | (options.android_build_credential, e)) |
Congbin Guo | 3afae6c | 2019-08-13 16:29:42 -0700 | [diff] [blame] | 1780 | |
| 1781 | cherrypy.tree.mount(health_checker_app, '/check_health', |
| 1782 | config=health_checker.get_config()) |
joychen | 3cb228e | 2013-06-12 12:13:13 -0700 | [diff] [blame] | 1783 | cherrypy.quickstart(dev_server, config=_GetConfig(options)) |
Chris Sosa | cde6bf4 | 2012-05-31 18:36:39 -0700 | [diff] [blame] | 1784 | |
| 1785 | |
| 1786 | if __name__ == '__main__': |
| 1787 | main() |