Mike Frysinger | d5fcb3a | 2013-05-30 21:10:50 -0400 | [diff] [blame] | 1 | # Copyright (c) 2013 The Chromium OS Authors. All rights reserved. |
| 2 | # Use of this source code is governed by a BSD-style license that can be |
| 3 | # found in the LICENSE file. |
| 4 | |
| 5 | """Upload all debug symbols required for crash reporting purposes. |
| 6 | |
| 7 | This script need only be used to upload release builds symbols or to debug |
| 8 | crashes on non-release builds (in which case try to only upload the symbols |
Mike Frysinger | 02e1e07 | 2013-11-10 22:11:34 -0500 | [diff] [blame] | 9 | for those executables involved). |
| 10 | """ |
Mike Frysinger | d5fcb3a | 2013-05-30 21:10:50 -0400 | [diff] [blame] | 11 | |
Mike Frysinger | a4fa1e8 | 2014-01-15 01:45:56 -0500 | [diff] [blame] | 12 | from __future__ import print_function |
| 13 | |
Mike Frysinger | d5fcb3a | 2013-05-30 21:10:50 -0400 | [diff] [blame] | 14 | import ctypes |
Mike Frysinger | 8ec8c50 | 2014-02-10 00:19:13 -0500 | [diff] [blame] | 15 | import datetime |
Mike Frysinger | 02e9240 | 2013-11-22 16:22:02 -0500 | [diff] [blame] | 16 | import functools |
Mike Frysinger | 0c0efa2 | 2014-02-09 23:32:23 -0500 | [diff] [blame] | 17 | import hashlib |
Mike Frysinger | a4fa1e8 | 2014-01-15 01:45:56 -0500 | [diff] [blame] | 18 | import httplib |
Mike Frysinger | d5fcb3a | 2013-05-30 21:10:50 -0400 | [diff] [blame] | 19 | import multiprocessing |
| 20 | import os |
Mike Frysinger | 094a217 | 2013-08-14 12:54:35 -0400 | [diff] [blame] | 21 | import poster |
Mike Frysinger | d5fcb3a | 2013-05-30 21:10:50 -0400 | [diff] [blame] | 22 | import random |
Mike Frysinger | fd35565 | 2014-01-23 02:57:48 -0500 | [diff] [blame] | 23 | import socket |
Mike Frysinger | d5fcb3a | 2013-05-30 21:10:50 -0400 | [diff] [blame] | 24 | import textwrap |
| 25 | import tempfile |
| 26 | import time |
Mike Frysinger | 094a217 | 2013-08-14 12:54:35 -0400 | [diff] [blame] | 27 | import urllib2 |
Mike Frysinger | d41938e | 2014-02-10 06:37:55 -0500 | [diff] [blame^] | 28 | import urlparse |
Mike Frysinger | d5fcb3a | 2013-05-30 21:10:50 -0400 | [diff] [blame] | 29 | |
Mike Frysinger | 0c0efa2 | 2014-02-09 23:32:23 -0500 | [diff] [blame] | 30 | from chromite.buildbot import constants |
Mike Frysinger | d41938e | 2014-02-10 06:37:55 -0500 | [diff] [blame^] | 31 | from chromite.lib import cache |
Mike Frysinger | d5fcb3a | 2013-05-30 21:10:50 -0400 | [diff] [blame] | 32 | from chromite.lib import commandline |
| 33 | from chromite.lib import cros_build_lib |
Mike Frysinger | d41938e | 2014-02-10 06:37:55 -0500 | [diff] [blame^] | 34 | from chromite.lib import gs |
| 35 | from chromite.lib import osutils |
Mike Frysinger | d5fcb3a | 2013-05-30 21:10:50 -0400 | [diff] [blame] | 36 | from chromite.lib import parallel |
David James | c93e6a4d | 2014-01-13 11:37:36 -0800 | [diff] [blame] | 37 | from chromite.lib import retry_util |
Mike Frysinger | 0c0efa2 | 2014-02-09 23:32:23 -0500 | [diff] [blame] | 38 | from chromite.lib import timeout_util |
Mike Frysinger | 69cb41d | 2013-08-11 20:08:19 -0400 | [diff] [blame] | 39 | from chromite.scripts import cros_generate_breakpad_symbols |
Mike Frysinger | d5fcb3a | 2013-05-30 21:10:50 -0400 | [diff] [blame] | 40 | |
Mike Frysinger | 0c0efa2 | 2014-02-09 23:32:23 -0500 | [diff] [blame] | 41 | # Needs to be after chromite imports. |
| 42 | # TODO(build): When doing the initial buildbot bootstrap, we won't have any |
| 43 | # other repos available. So ignore isolateserver imports. But buildbot will |
| 44 | # re-exec itself once it has done a full repo sync and then the module will |
| 45 | # be available -- it isn't needed that early. http://crbug.com/341152 |
| 46 | try: |
| 47 | import isolateserver |
| 48 | except ImportError: |
| 49 | isolateserver = None |
| 50 | |
Mike Frysinger | d5fcb3a | 2013-05-30 21:10:50 -0400 | [diff] [blame] | 51 | |
| 52 | # URLs used for uploading symbols. |
| 53 | OFFICIAL_UPLOAD_URL = 'http://clients2.google.com/cr/symbol' |
| 54 | STAGING_UPLOAD_URL = 'http://clients2.google.com/cr/staging_symbol' |
| 55 | |
| 56 | |
| 57 | # The crash server rejects files that are this big. |
| 58 | CRASH_SERVER_FILE_LIMIT = 350 * 1024 * 1024 |
| 59 | # Give ourselves a little breathing room from what the server expects. |
| 60 | DEFAULT_FILE_LIMIT = CRASH_SERVER_FILE_LIMIT - (10 * 1024 * 1024) |
| 61 | |
| 62 | |
Mike Frysinger | 0c0efa2 | 2014-02-09 23:32:23 -0500 | [diff] [blame] | 63 | # The batch limit when talking to the dedup server. We avoid sending one at a |
| 64 | # time as the round trip overhead will dominate. Conversely, we avoid sending |
| 65 | # all at once so we can start uploading symbols asap -- the symbol server is a |
| 66 | # bit slow and will take longer than anything else. |
| 67 | # TODO: A better algorithm would be adaptive. If we have more than one symbol |
| 68 | # in the upload queue waiting, we could send more symbols to the dedupe server |
| 69 | # at a time. |
| 70 | DEDUPE_LIMIT = 100 |
| 71 | |
| 72 | # How long to wait for the server to respond with the results. Note that the |
| 73 | # larger the limit above, the larger this will need to be. So we give it ~1 |
| 74 | # second per item max. |
| 75 | DEDUPE_TIMEOUT = DEDUPE_LIMIT |
| 76 | |
| 77 | # The unique namespace in the dedupe server that only we use. Helps avoid |
| 78 | # collisions with all the hashed values and unrelated content. |
| 79 | OFFICIAL_DEDUPE_NAMESPACE = 'chromium-os-upload-symbols' |
| 80 | STAGING_DEDUPE_NAMESPACE = '%s-staging' % OFFICIAL_DEDUPE_NAMESPACE |
| 81 | |
| 82 | |
Mike Frysinger | cd78a08 | 2013-06-26 17:13:04 -0400 | [diff] [blame] | 83 | # How long to wait (in seconds) for a single upload to complete. This has |
| 84 | # to allow for symbols that are up to CRASH_SERVER_FILE_LIMIT in size. |
| 85 | UPLOAD_TIMEOUT = 30 * 60 |
| 86 | |
| 87 | |
Mike Frysinger | d5fcb3a | 2013-05-30 21:10:50 -0400 | [diff] [blame] | 88 | # Sleep for 200ms in between uploads to avoid DoS'ing symbol server. |
| 89 | DEFAULT_SLEEP_DELAY = 0.2 |
| 90 | |
| 91 | |
| 92 | # Number of seconds to wait before retrying an upload. The delay will double |
| 93 | # for each subsequent retry of the same symbol file. |
| 94 | INITIAL_RETRY_DELAY = 1 |
| 95 | |
| 96 | # Allow up to 7 attempts to upload a symbol file (total delay may be |
| 97 | # 1+2+4+8+16+32=63 seconds). |
| 98 | MAX_RETRIES = 6 |
| 99 | |
Mike Frysinger | eb753bf | 2013-11-22 16:05:35 -0500 | [diff] [blame] | 100 | # Number of total errors, before uploads are no longer attempted. |
| 101 | # This is used to avoid lots of errors causing unreasonable delays. |
| 102 | # See the related, but independent, error values below. |
| 103 | MAX_TOTAL_ERRORS_FOR_RETRY = 30 |
| 104 | |
| 105 | # A watermark of transient errors which we allow recovery from. If we hit |
| 106 | # errors infrequently, overall we're probably doing fine. For example, if |
| 107 | # we have one failure every 100 passes, then we probably don't want to fail |
| 108 | # right away. But if we hit a string of failures in a row, we want to abort. |
| 109 | # |
| 110 | # The watermark starts at 0 (and can never go below that). When this error |
| 111 | # level is exceeded, we stop uploading. When a failure happens, we add the |
| 112 | # fail adjustment, and when an upload succeeds, we add the pass adjustment. |
| 113 | # We want to penalize failures more so that we ramp up when there is a string |
| 114 | # of them, but then slowly back off as things start working. |
| 115 | # |
| 116 | # A quick example: |
| 117 | # 0.0: Starting point. |
| 118 | # 0.0: Upload works, so add -0.5, and then clamp to 0. |
| 119 | # 1.0: Upload fails, so add 1.0. |
| 120 | # 2.0: Upload fails, so add 1.0. |
| 121 | # 1.5: Upload works, so add -0.5. |
| 122 | # 1.0: Upload works, so add -0.5. |
| 123 | ERROR_WATERMARK = 3.0 |
| 124 | ERROR_ADJUST_FAIL = 1.0 |
| 125 | ERROR_ADJUST_PASS = -0.5 |
Mike Frysinger | d5fcb3a | 2013-05-30 21:10:50 -0400 | [diff] [blame] | 126 | |
| 127 | |
Mike Frysinger | 0c0efa2 | 2014-02-09 23:32:23 -0500 | [diff] [blame] | 128 | def SymUpload(upload_url, sym_item): |
Mike Frysinger | 094a217 | 2013-08-14 12:54:35 -0400 | [diff] [blame] | 129 | """Upload a symbol file to a HTTP server |
| 130 | |
| 131 | The upload is a multipart/form-data POST with the following parameters: |
| 132 | code_file: the basename of the module, e.g. "app" |
| 133 | code_identifier: the module file's identifier |
| 134 | debug_file: the basename of the debugging file, e.g. "app" |
| 135 | debug_identifier: the debug file's identifier, usually consisting of |
| 136 | the guid and age embedded in the pdb, e.g. |
| 137 | "11111111BBBB3333DDDD555555555555F" |
| 138 | version: the file version of the module, e.g. "1.2.3.4" |
| 139 | product: HTTP-friendly product name |
| 140 | os: the operating system that the module was built for |
| 141 | cpu: the CPU that the module was built for |
| 142 | symbol_file: the contents of the breakpad-format symbol file |
| 143 | |
| 144 | Args: |
Mike Frysinger | 094a217 | 2013-08-14 12:54:35 -0400 | [diff] [blame] | 145 | upload_url: The crash URL to POST the |sym_file| to |
Mike Frysinger | 0c0efa2 | 2014-02-09 23:32:23 -0500 | [diff] [blame] | 146 | sym_item: A SymbolItem containing the path to the breakpad symbol to upload |
Mike Frysinger | 094a217 | 2013-08-14 12:54:35 -0400 | [diff] [blame] | 147 | """ |
Mike Frysinger | 0c0efa2 | 2014-02-09 23:32:23 -0500 | [diff] [blame] | 148 | sym_header = sym_item.sym_header |
| 149 | sym_file = sym_item.sym_file |
Mike Frysinger | 094a217 | 2013-08-14 12:54:35 -0400 | [diff] [blame] | 150 | |
| 151 | fields = ( |
| 152 | ('code_file', sym_header.name), |
| 153 | ('debug_file', sym_header.name), |
| 154 | ('debug_identifier', sym_header.id.replace('-', '')), |
| 155 | # Should we set these fields? They aren't critical, but it might be nice? |
| 156 | # We'd have to figure out what file this symbol is coming from and what |
| 157 | # package provides it ... |
| 158 | #('version', None), |
| 159 | #('product', 'ChromeOS'), |
| 160 | ('os', sym_header.os), |
| 161 | ('cpu', sym_header.cpu), |
| 162 | poster.encode.MultipartParam.from_file('symbol_file', sym_file), |
| 163 | ) |
| 164 | |
| 165 | data, headers = poster.encode.multipart_encode(fields) |
| 166 | request = urllib2.Request(upload_url, data, headers) |
| 167 | request.add_header('User-agent', 'chromite.upload_symbols') |
| 168 | urllib2.urlopen(request, timeout=UPLOAD_TIMEOUT) |
Mike Frysinger | d5fcb3a | 2013-05-30 21:10:50 -0400 | [diff] [blame] | 169 | |
| 170 | |
Mike Frysinger | 0c0efa2 | 2014-02-09 23:32:23 -0500 | [diff] [blame] | 171 | def TestingSymUpload(upload_url, sym_item): |
Mike Frysinger | d5fcb3a | 2013-05-30 21:10:50 -0400 | [diff] [blame] | 172 | """A stub version of SymUpload for --testing usage""" |
Mike Frysinger | 0c0efa2 | 2014-02-09 23:32:23 -0500 | [diff] [blame] | 173 | cmd = ['sym_upload', sym_item.sym_file, upload_url] |
Mike Frysinger | d5fcb3a | 2013-05-30 21:10:50 -0400 | [diff] [blame] | 174 | # Randomly fail 80% of the time (the retry logic makes this 80%/3 per file). |
| 175 | returncode = random.randint(1, 100) <= 80 |
| 176 | cros_build_lib.Debug('would run (and return %i): %s', returncode, |
Matt Tennant | 7feda35 | 2013-12-20 14:03:40 -0800 | [diff] [blame] | 177 | cros_build_lib.CmdToStr(cmd)) |
Mike Frysinger | d5fcb3a | 2013-05-30 21:10:50 -0400 | [diff] [blame] | 178 | if returncode: |
| 179 | output = 'Failed to send the symbol file.' |
| 180 | else: |
| 181 | output = 'Successfully sent the symbol file.' |
| 182 | result = cros_build_lib.CommandResult(cmd=cmd, error=None, output=output, |
| 183 | returncode=returncode) |
| 184 | if returncode: |
Mike Frysinger | a4fa1e8 | 2014-01-15 01:45:56 -0500 | [diff] [blame] | 185 | exceptions = ( |
Mike Frysinger | fd35565 | 2014-01-23 02:57:48 -0500 | [diff] [blame] | 186 | socket.error('[socket.error] forced test fail'), |
Mike Frysinger | a4fa1e8 | 2014-01-15 01:45:56 -0500 | [diff] [blame] | 187 | httplib.BadStatusLine('[BadStatusLine] forced test fail'), |
| 188 | urllib2.HTTPError(upload_url, 400, '[HTTPError] forced test fail', |
| 189 | {}, None), |
| 190 | urllib2.URLError('[URLError] forced test fail'), |
| 191 | ) |
| 192 | raise random.choice(exceptions) |
Mike Frysinger | d5fcb3a | 2013-05-30 21:10:50 -0400 | [diff] [blame] | 193 | else: |
| 194 | return result |
| 195 | |
| 196 | |
Mike Frysinger | eb753bf | 2013-11-22 16:05:35 -0500 | [diff] [blame] | 197 | def ErrorLimitHit(num_errors, watermark_errors): |
| 198 | """See if our error limit has been hit |
| 199 | |
| 200 | Args: |
| 201 | num_errors: A multiprocessing.Value of the raw number of failures. |
| 202 | watermark_errors: A multiprocessing.Value of the current rate of failures. |
Mike Frysinger | 1a736a8 | 2013-12-12 01:50:59 -0500 | [diff] [blame] | 203 | |
Mike Frysinger | eb753bf | 2013-11-22 16:05:35 -0500 | [diff] [blame] | 204 | Returns: |
| 205 | True if our error limits have been exceeded. |
| 206 | """ |
| 207 | return ((num_errors is not None and |
| 208 | num_errors.value > MAX_TOTAL_ERRORS_FOR_RETRY) or |
| 209 | (watermark_errors is not None and |
| 210 | watermark_errors.value > ERROR_WATERMARK)) |
| 211 | |
| 212 | |
| 213 | def _UpdateCounter(counter, adj): |
| 214 | """Update |counter| by |adj| |
| 215 | |
| 216 | Handle atomic updates of |counter|. Also make sure it does not |
| 217 | fall below 0. |
| 218 | |
| 219 | Args: |
| 220 | counter: A multiprocessing.Value to update |
| 221 | adj: The value to add to |counter| |
| 222 | """ |
| 223 | def _Update(): |
| 224 | clamp = 0 if type(adj) is int else 0.0 |
| 225 | counter.value = max(clamp, counter.value + adj) |
| 226 | |
| 227 | if hasattr(counter, 'get_lock'): |
| 228 | with counter.get_lock(): |
| 229 | _Update() |
| 230 | elif counter is not None: |
| 231 | _Update() |
| 232 | |
| 233 | |
Mike Frysinger | 0c0efa2 | 2014-02-09 23:32:23 -0500 | [diff] [blame] | 234 | def UploadSymbol(upload_url, sym_item, file_limit=DEFAULT_FILE_LIMIT, |
Mike Frysinger | 02e9240 | 2013-11-22 16:22:02 -0500 | [diff] [blame] | 235 | sleep=0, num_errors=None, watermark_errors=None, |
Mike Frysinger | 0c0efa2 | 2014-02-09 23:32:23 -0500 | [diff] [blame] | 236 | failed_queue=None, passed_queue=None): |
| 237 | """Upload |sym_item| to |upload_url| |
Mike Frysinger | d5fcb3a | 2013-05-30 21:10:50 -0400 | [diff] [blame] | 238 | |
| 239 | Args: |
Mike Frysinger | d5fcb3a | 2013-05-30 21:10:50 -0400 | [diff] [blame] | 240 | upload_url: The crash server to upload things to |
Mike Frysinger | 0c0efa2 | 2014-02-09 23:32:23 -0500 | [diff] [blame] | 241 | sym_item: A SymbolItem containing the path to the breakpad symbol to upload |
Mike Frysinger | d5fcb3a | 2013-05-30 21:10:50 -0400 | [diff] [blame] | 242 | file_limit: The max file size of a symbol file before we try to strip it |
| 243 | sleep: Number of seconds to sleep before running |
Mike Frysinger | 69cb41d | 2013-08-11 20:08:19 -0400 | [diff] [blame] | 244 | num_errors: An object to update with the error count (needs a .value member) |
Mike Frysinger | eb753bf | 2013-11-22 16:05:35 -0500 | [diff] [blame] | 245 | watermark_errors: An object to track current error behavior (needs a .value) |
Mike Frysinger | 02e9240 | 2013-11-22 16:22:02 -0500 | [diff] [blame] | 246 | failed_queue: When a symbol fails, add it to this queue |
Mike Frysinger | 0c0efa2 | 2014-02-09 23:32:23 -0500 | [diff] [blame] | 247 | passed_queue: When a symbol passes, add it to this queue |
Mike Frysinger | 1a736a8 | 2013-12-12 01:50:59 -0500 | [diff] [blame] | 248 | |
Mike Frysinger | 69cb41d | 2013-08-11 20:08:19 -0400 | [diff] [blame] | 249 | Returns: |
| 250 | The number of errors that were encountered. |
Mike Frysinger | d5fcb3a | 2013-05-30 21:10:50 -0400 | [diff] [blame] | 251 | """ |
Mike Frysinger | 0c0efa2 | 2014-02-09 23:32:23 -0500 | [diff] [blame] | 252 | sym_file = sym_item.sym_file |
| 253 | upload_item = sym_item |
| 254 | |
Mike Frysinger | d5fcb3a | 2013-05-30 21:10:50 -0400 | [diff] [blame] | 255 | if num_errors is None: |
| 256 | num_errors = ctypes.c_int() |
Mike Frysinger | eb753bf | 2013-11-22 16:05:35 -0500 | [diff] [blame] | 257 | if ErrorLimitHit(num_errors, watermark_errors): |
Mike Frysinger | d5fcb3a | 2013-05-30 21:10:50 -0400 | [diff] [blame] | 258 | # Abandon ship! It's on fire! NOoooooooooooOOOoooooo. |
Mike Frysinger | 7f9be14 | 2014-01-15 02:16:42 -0500 | [diff] [blame] | 259 | if failed_queue: |
| 260 | failed_queue.put(sym_file) |
Mike Frysinger | d5fcb3a | 2013-05-30 21:10:50 -0400 | [diff] [blame] | 261 | return 0 |
| 262 | |
Mike Frysinger | d5fcb3a | 2013-05-30 21:10:50 -0400 | [diff] [blame] | 263 | if sleep: |
| 264 | # Keeps us from DoS-ing the symbol server. |
| 265 | time.sleep(sleep) |
| 266 | |
| 267 | cros_build_lib.Debug('uploading %s' % sym_file) |
| 268 | |
| 269 | # Ideally there'd be a tempfile.SpooledNamedTemporaryFile that we could use. |
| 270 | with tempfile.NamedTemporaryFile(prefix='upload_symbols', |
| 271 | bufsize=0) as temp_sym_file: |
| 272 | if file_limit: |
| 273 | # If the symbols size is too big, strip out the call frame info. The CFI |
| 274 | # is unnecessary for 32bit x86 targets where the frame pointer is used (as |
| 275 | # all of ours have) and it accounts for over half the size of the symbols |
| 276 | # uploaded. |
| 277 | file_size = os.path.getsize(sym_file) |
| 278 | if file_size > file_limit: |
| 279 | cros_build_lib.Warning('stripping CFI from %s due to size %s > %s', |
| 280 | sym_file, file_size, file_limit) |
| 281 | temp_sym_file.writelines([x for x in open(sym_file, 'rb').readlines() |
| 282 | if not x.startswith('STACK CFI')]) |
Mike Frysinger | 0c0efa2 | 2014-02-09 23:32:23 -0500 | [diff] [blame] | 283 | |
| 284 | upload_item = FakeItem(sym_file=temp_sym_file.name, |
| 285 | sym_header=sym_item.sym_header) |
Mike Frysinger | d5fcb3a | 2013-05-30 21:10:50 -0400 | [diff] [blame] | 286 | |
| 287 | # Hopefully the crash server will let it through. But it probably won't. |
| 288 | # Not sure what the best answer is in this case. |
Mike Frysinger | 0c0efa2 | 2014-02-09 23:32:23 -0500 | [diff] [blame] | 289 | file_size = os.path.getsize(upload_item.sym_file) |
Mike Frysinger | d5fcb3a | 2013-05-30 21:10:50 -0400 | [diff] [blame] | 290 | if file_size > CRASH_SERVER_FILE_LIMIT: |
| 291 | cros_build_lib.PrintBuildbotStepWarnings() |
Mike Frysinger | 02e9240 | 2013-11-22 16:22:02 -0500 | [diff] [blame] | 292 | cros_build_lib.Warning('upload file %s is awfully large, risking ' |
| 293 | 'rejection by the symbol server (%s > %s)', |
| 294 | sym_file, file_size, CRASH_SERVER_FILE_LIMIT) |
Mike Frysinger | d5fcb3a | 2013-05-30 21:10:50 -0400 | [diff] [blame] | 295 | |
| 296 | # Upload the symbol file. |
Mike Frysinger | eb753bf | 2013-11-22 16:05:35 -0500 | [diff] [blame] | 297 | success = False |
Mike Frysinger | d5fcb3a | 2013-05-30 21:10:50 -0400 | [diff] [blame] | 298 | try: |
Mike Frysinger | 9adcfd2 | 2013-10-24 12:01:40 -0400 | [diff] [blame] | 299 | cros_build_lib.TimedCommand( |
David James | c93e6a4d | 2014-01-13 11:37:36 -0800 | [diff] [blame] | 300 | retry_util.RetryException, |
Mike Frysinger | 9adcfd2 | 2013-10-24 12:01:40 -0400 | [diff] [blame] | 301 | (urllib2.HTTPError, urllib2.URLError), MAX_RETRIES, SymUpload, |
Mike Frysinger | 0c0efa2 | 2014-02-09 23:32:23 -0500 | [diff] [blame] | 302 | upload_url, upload_item, sleep=INITIAL_RETRY_DELAY, |
Mike Frysinger | 9adcfd2 | 2013-10-24 12:01:40 -0400 | [diff] [blame] | 303 | timed_log_msg='upload of %10i bytes took %%s: %s' % |
| 304 | (file_size, os.path.basename(sym_file))) |
Mike Frysinger | eb753bf | 2013-11-22 16:05:35 -0500 | [diff] [blame] | 305 | success = True |
Mike Frysinger | 0c0efa2 | 2014-02-09 23:32:23 -0500 | [diff] [blame] | 306 | |
| 307 | if passed_queue: |
| 308 | passed_queue.put(sym_item) |
Mike Frysinger | 094a217 | 2013-08-14 12:54:35 -0400 | [diff] [blame] | 309 | except urllib2.HTTPError as e: |
| 310 | cros_build_lib.Warning('could not upload: %s: HTTP %s: %s', |
| 311 | os.path.basename(sym_file), e.code, e.reason) |
Mike Frysinger | fd35565 | 2014-01-23 02:57:48 -0500 | [diff] [blame] | 312 | except (urllib2.URLError, httplib.HTTPException, socket.error) as e: |
Mike Frysinger | c4ab578 | 2013-10-02 18:14:22 -0400 | [diff] [blame] | 313 | cros_build_lib.Warning('could not upload: %s: %s', |
| 314 | os.path.basename(sym_file), e) |
Mike Frysinger | eb753bf | 2013-11-22 16:05:35 -0500 | [diff] [blame] | 315 | finally: |
| 316 | if success: |
| 317 | _UpdateCounter(watermark_errors, ERROR_ADJUST_PASS) |
| 318 | else: |
| 319 | _UpdateCounter(num_errors, 1) |
| 320 | _UpdateCounter(watermark_errors, ERROR_ADJUST_FAIL) |
Mike Frysinger | 02e9240 | 2013-11-22 16:22:02 -0500 | [diff] [blame] | 321 | if failed_queue: |
| 322 | failed_queue.put(sym_file) |
Mike Frysinger | d5fcb3a | 2013-05-30 21:10:50 -0400 | [diff] [blame] | 323 | |
| 324 | return num_errors.value |
| 325 | |
| 326 | |
Mike Frysinger | 0c0efa2 | 2014-02-09 23:32:23 -0500 | [diff] [blame] | 327 | # A dummy class that allows for stubbing in tests and SymUpload. |
| 328 | FakeItem = cros_build_lib.Collection( |
| 329 | 'FakeItem', sym_file=None, sym_header=None, content=lambda x: '') |
| 330 | |
| 331 | |
| 332 | # TODO(build): Delete this if check. http://crbug.com/341152 |
| 333 | if isolateserver: |
| 334 | class SymbolItem(isolateserver.BufferItem): |
| 335 | """Turn a sym_file into an isolateserver.Item""" |
| 336 | |
| 337 | ALGO = hashlib.sha1 |
| 338 | |
| 339 | def __init__(self, sym_file): |
| 340 | sym_header = cros_generate_breakpad_symbols.ReadSymsHeader(sym_file) |
| 341 | super(SymbolItem, self).__init__(str(sym_header), self.ALGO) |
| 342 | self.sym_header = sym_header |
| 343 | self.sym_file = sym_file |
| 344 | |
| 345 | |
| 346 | def SymbolDeduplicatorNotify(dedupe_namespace, dedupe_queue): |
| 347 | """Send a symbol file to the swarming service |
| 348 | |
| 349 | Notify the swarming service of a successful upload. If the notification fails |
| 350 | for any reason, we ignore it. We don't care as it just means we'll upload it |
| 351 | again later on, and the symbol server will handle that graciously. |
| 352 | |
| 353 | This func runs in a different process from the main one, so we cannot share |
| 354 | the storage object. Instead, we create our own. This func stays alive for |
| 355 | the life of the process, so we only create one here overall. |
| 356 | |
| 357 | Args: |
| 358 | dedupe_namespace: The isolateserver namespace to dedupe uploaded symbols. |
| 359 | dedupe_queue: The queue to read SymbolItems from |
| 360 | """ |
| 361 | if dedupe_queue is None: |
| 362 | return |
| 363 | |
| 364 | item = None |
| 365 | try: |
| 366 | storage = isolateserver.get_storage_api(constants.ISOLATESERVER, |
| 367 | dedupe_namespace) |
| 368 | for item in iter(dedupe_queue.get, None): |
| 369 | with timeout_util.Timeout(DEDUPE_TIMEOUT): |
| 370 | storage.push(item, item.content(0)) |
| 371 | except Exception: |
| 372 | sym_file = item.sym_file if (item and item.sym_file) else '' |
| 373 | cros_build_lib.Warning('posting %s to dedupe server failed', |
| 374 | os.path.basename(sym_file), exc_info=True) |
| 375 | |
| 376 | |
| 377 | def SymbolDeduplicator(storage, sym_paths): |
| 378 | """Filter out symbol files that we've already uploaded |
| 379 | |
| 380 | Using the swarming service, ask it to tell us which symbol files we've already |
| 381 | uploaded in previous runs and/or by other bots. If the query fails for any |
| 382 | reason, we'll just upload all symbols. This is fine as the symbol server will |
| 383 | do the right thing and this phase is purely an optimization. |
| 384 | |
| 385 | This code runs in the main thread which is why we can re-use the existing |
| 386 | storage object. Saves us from having to recreate one all the time. |
| 387 | |
| 388 | Args: |
| 389 | storage: An isolateserver.StorageApi object |
| 390 | sym_paths: List of symbol files to check against the dedupe server |
| 391 | |
| 392 | Returns: |
| 393 | List of symbol files that have not been uploaded before |
| 394 | """ |
| 395 | if not sym_paths: |
| 396 | return sym_paths |
| 397 | |
| 398 | items = [SymbolItem(x) for x in sym_paths] |
| 399 | if storage: |
| 400 | try: |
| 401 | with timeout_util.Timeout(DEDUPE_TIMEOUT): |
| 402 | items = storage.contains(items) |
| 403 | except Exception: |
| 404 | cros_build_lib.Warning('talking to dedupe server failed', exc_info=True) |
| 405 | |
| 406 | return items |
| 407 | |
| 408 | |
Mike Frysinger | d41938e | 2014-02-10 06:37:55 -0500 | [diff] [blame^] | 409 | def IsTarball(path): |
| 410 | """Guess if this is a tarball based on the filename.""" |
| 411 | parts = path.split('.') |
| 412 | if len(parts) <= 1: |
| 413 | return False |
| 414 | |
| 415 | if parts[-1] == 'tar': |
| 416 | return True |
| 417 | |
| 418 | if parts[-2] == 'tar': |
| 419 | return parts[-1] in ('bz2', 'gz', 'xz') |
| 420 | |
| 421 | return parts[-1] in ('tbz2', 'tbz', 'tgz', 'txz') |
| 422 | |
| 423 | |
| 424 | def SymbolFinder(tempdir, paths): |
Mike Frysinger | 9b2ff5c | 2013-11-22 10:01:12 -0500 | [diff] [blame] | 425 | """Locate symbol files in |paths| |
| 426 | |
| 427 | Args: |
Mike Frysinger | d41938e | 2014-02-10 06:37:55 -0500 | [diff] [blame^] | 428 | tempdir: Path to use for temporary files (caller will clean up). |
Mike Frysinger | 9b2ff5c | 2013-11-22 10:01:12 -0500 | [diff] [blame] | 429 | paths: A list of input paths to walk. Files are returned w/out any checks. |
Mike Frysinger | d41938e | 2014-02-10 06:37:55 -0500 | [diff] [blame^] | 430 | Dirs are searched for files that end in ".sym". Urls are fetched and then |
| 431 | processed. Tarballs are unpacked and walked. |
Mike Frysinger | 1a736a8 | 2013-12-12 01:50:59 -0500 | [diff] [blame] | 432 | |
Mike Frysinger | 9b2ff5c | 2013-11-22 10:01:12 -0500 | [diff] [blame] | 433 | Returns: |
| 434 | Yield every viable sym file. |
| 435 | """ |
| 436 | for p in paths: |
Mike Frysinger | d41938e | 2014-02-10 06:37:55 -0500 | [diff] [blame^] | 437 | o = urlparse.urlparse(p) |
| 438 | if o.scheme: |
| 439 | # Support globs of filenames. |
| 440 | ctx = gs.GSContext() |
| 441 | for p in ctx.LS(p): |
| 442 | cros_build_lib.Info('processing files inside %s', p) |
| 443 | o = urlparse.urlparse(p) |
| 444 | cache_dir = commandline.GetCacheDir() |
| 445 | common_path = os.path.join(cache_dir, constants.COMMON_CACHE) |
| 446 | tar_cache = cache.TarballCache(common_path) |
| 447 | key = ('%s%s' % (o.netloc, o.path)).split('/') |
| 448 | # The common cache will not be LRU, removing the need to hold a read |
| 449 | # lock on the cached gsutil. |
| 450 | ref = tar_cache.Lookup(key) |
| 451 | try: |
| 452 | ref.SetDefault(p) |
| 453 | except cros_build_lib.RunCommandError as e: |
| 454 | cros_build_lib.Warning('ignoring %s\n%s', p, e) |
| 455 | continue |
| 456 | for p in SymbolFinder(tempdir, [ref.path]): |
| 457 | yield p |
| 458 | |
| 459 | elif os.path.isdir(p): |
Mike Frysinger | 9b2ff5c | 2013-11-22 10:01:12 -0500 | [diff] [blame] | 460 | for root, _, files in os.walk(p): |
| 461 | for f in files: |
| 462 | if f.endswith('.sym'): |
| 463 | yield os.path.join(root, f) |
Mike Frysinger | d41938e | 2014-02-10 06:37:55 -0500 | [diff] [blame^] | 464 | |
| 465 | elif IsTarball(p): |
| 466 | cros_build_lib.Info('processing files inside %s', p) |
| 467 | tardir = tempfile.mkdtemp(dir=tempdir) |
| 468 | cache.Untar(os.path.realpath(p), tardir) |
| 469 | for p in SymbolFinder(tardir, [tardir]): |
| 470 | yield p |
| 471 | |
Mike Frysinger | 9b2ff5c | 2013-11-22 10:01:12 -0500 | [diff] [blame] | 472 | else: |
| 473 | yield p |
| 474 | |
| 475 | |
Mike Frysinger | 0c0efa2 | 2014-02-09 23:32:23 -0500 | [diff] [blame] | 476 | def WriteQueueToFile(listing, queue, relpath=None): |
| 477 | """Write all the items in |queue| to the |listing|. |
| 478 | |
| 479 | Args: |
| 480 | listing: Where to write out the list of files. |
| 481 | queue: The queue of paths to drain. |
| 482 | relpath: If set, write out paths relative to this one. |
| 483 | """ |
| 484 | if not listing: |
| 485 | return |
| 486 | |
| 487 | with cros_build_lib.Open(listing, 'wb+') as f: |
| 488 | while not queue.empty(): |
| 489 | path = queue.get() |
| 490 | if relpath: |
| 491 | path = os.path.relpath(path, relpath) |
| 492 | f.write('%s\n' % path) |
| 493 | |
| 494 | |
Mike Frysinger | 9dcf9ae | 2013-08-10 15:17:09 -0400 | [diff] [blame] | 495 | def UploadSymbols(board=None, official=False, breakpad_dir=None, |
Mike Frysinger | d5fcb3a | 2013-05-30 21:10:50 -0400 | [diff] [blame] | 496 | file_limit=DEFAULT_FILE_LIMIT, sleep=DEFAULT_SLEEP_DELAY, |
Mike Frysinger | 8ec8c50 | 2014-02-10 00:19:13 -0500 | [diff] [blame] | 497 | upload_limit=None, sym_paths=None, failed_list=None, |
Mike Frysinger | 0c0efa2 | 2014-02-09 23:32:23 -0500 | [diff] [blame] | 498 | root=None, retry=True, dedupe_namespace=None): |
Mike Frysinger | d5fcb3a | 2013-05-30 21:10:50 -0400 | [diff] [blame] | 499 | """Upload all the generated symbols for |board| to the crash server |
| 500 | |
Mike Frysinger | 9dcf9ae | 2013-08-10 15:17:09 -0400 | [diff] [blame] | 501 | You can use in a few ways: |
| 502 | * pass |board| to locate all of its symbols |
| 503 | * pass |breakpad_dir| to upload all the symbols in there |
Mike Frysinger | 9b2ff5c | 2013-11-22 10:01:12 -0500 | [diff] [blame] | 504 | * pass |sym_paths| to upload specific symbols (or dirs of symbols) |
Mike Frysinger | 9dcf9ae | 2013-08-10 15:17:09 -0400 | [diff] [blame] | 505 | |
Mike Frysinger | d5fcb3a | 2013-05-30 21:10:50 -0400 | [diff] [blame] | 506 | Args: |
| 507 | board: The board whose symbols we wish to upload |
| 508 | official: Use the official symbol server rather than the staging one |
| 509 | breakpad_dir: The full path to the breakpad directory where symbols live |
| 510 | file_limit: The max file size of a symbol file before we try to strip it |
| 511 | sleep: How long to sleep in between uploads |
Mike Frysinger | 8ec8c50 | 2014-02-10 00:19:13 -0500 | [diff] [blame] | 512 | upload_limit: If set, only upload this many symbols (meant for testing) |
Mike Frysinger | 9b2ff5c | 2013-11-22 10:01:12 -0500 | [diff] [blame] | 513 | sym_paths: Specific symbol files (or dirs of sym files) to upload, |
| 514 | otherwise search |breakpad_dir| |
Mike Frysinger | 7f9be14 | 2014-01-15 02:16:42 -0500 | [diff] [blame] | 515 | failed_list: Write the names of all sym files we did not upload; can be a |
| 516 | filename or file-like object. |
Mike Frysinger | 118d250 | 2013-08-19 03:36:56 -0400 | [diff] [blame] | 517 | root: The tree to prefix to |breakpad_dir| (if |breakpad_dir| is not set) |
Mike Frysinger | 02e9240 | 2013-11-22 16:22:02 -0500 | [diff] [blame] | 518 | retry: Whether we should retry failures. |
Mike Frysinger | 0c0efa2 | 2014-02-09 23:32:23 -0500 | [diff] [blame] | 519 | dedupe_namespace: The isolateserver namespace to dedupe uploaded symbols. |
Mike Frysinger | 1a736a8 | 2013-12-12 01:50:59 -0500 | [diff] [blame] | 520 | |
Mike Frysinger | d5fcb3a | 2013-05-30 21:10:50 -0400 | [diff] [blame] | 521 | Returns: |
Mike Frysinger | 69cb41d | 2013-08-11 20:08:19 -0400 | [diff] [blame] | 522 | The number of errors that were encountered. |
Mike Frysinger | d5fcb3a | 2013-05-30 21:10:50 -0400 | [diff] [blame] | 523 | """ |
Mike Frysinger | 0c0efa2 | 2014-02-09 23:32:23 -0500 | [diff] [blame] | 524 | # TODO(build): Delete this assert. |
| 525 | assert isolateserver, 'Missing isolateserver import http://crbug.com/341152' |
| 526 | |
Mike Frysinger | d5fcb3a | 2013-05-30 21:10:50 -0400 | [diff] [blame] | 527 | if official: |
| 528 | upload_url = OFFICIAL_UPLOAD_URL |
| 529 | else: |
| 530 | cros_build_lib.Warning('unofficial builds upload to the staging server') |
| 531 | upload_url = STAGING_UPLOAD_URL |
| 532 | |
Mike Frysinger | 9b2ff5c | 2013-11-22 10:01:12 -0500 | [diff] [blame] | 533 | if sym_paths: |
| 534 | cros_build_lib.Info('uploading specified symbols to %s', upload_url) |
Mike Frysinger | 9dcf9ae | 2013-08-10 15:17:09 -0400 | [diff] [blame] | 535 | else: |
| 536 | if breakpad_dir is None: |
Mike Frysinger | 118d250 | 2013-08-19 03:36:56 -0400 | [diff] [blame] | 537 | breakpad_dir = os.path.join( |
| 538 | root, |
| 539 | cros_generate_breakpad_symbols.FindBreakpadDir(board).lstrip('/')) |
Mike Frysinger | 9dcf9ae | 2013-08-10 15:17:09 -0400 | [diff] [blame] | 540 | cros_build_lib.Info('uploading all symbols to %s from %s', upload_url, |
| 541 | breakpad_dir) |
Mike Frysinger | 9b2ff5c | 2013-11-22 10:01:12 -0500 | [diff] [blame] | 542 | sym_paths = [breakpad_dir] |
Mike Frysinger | d5fcb3a | 2013-05-30 21:10:50 -0400 | [diff] [blame] | 543 | |
Mike Frysinger | 0c0efa2 | 2014-02-09 23:32:23 -0500 | [diff] [blame] | 544 | # We use storage_query to ask the server about existing symbols. The |
| 545 | # storage_notify_proc process is used to post updates to the server. We |
| 546 | # cannot safely share the storage object between threads/processes, but |
| 547 | # we also want to minimize creating new ones as each object has to init |
| 548 | # new state (like server connections). |
| 549 | if dedupe_namespace: |
| 550 | dedupe_limit = DEDUPE_LIMIT |
| 551 | dedupe_queue = multiprocessing.Queue() |
| 552 | storage_query = isolateserver.get_storage_api(constants.ISOLATESERVER, |
| 553 | dedupe_namespace) |
| 554 | else: |
| 555 | dedupe_limit = 1 |
| 556 | dedupe_queue = storage_query = None |
| 557 | # Can't use parallel.BackgroundTaskRunner because that'll create multiple |
| 558 | # processes and we want only one the whole time (see comment above). |
| 559 | storage_notify_proc = multiprocessing.Process( |
| 560 | target=SymbolDeduplicatorNotify, args=(dedupe_namespace, dedupe_queue)) |
| 561 | |
Mike Frysinger | d5fcb3a | 2013-05-30 21:10:50 -0400 | [diff] [blame] | 562 | bg_errors = multiprocessing.Value('i') |
Mike Frysinger | eb753bf | 2013-11-22 16:05:35 -0500 | [diff] [blame] | 563 | watermark_errors = multiprocessing.Value('f') |
Mike Frysinger | 02e9240 | 2013-11-22 16:22:02 -0500 | [diff] [blame] | 564 | failed_queue = multiprocessing.Queue() |
| 565 | uploader = functools.partial( |
Mike Frysinger | 0c0efa2 | 2014-02-09 23:32:23 -0500 | [diff] [blame] | 566 | UploadSymbol, upload_url, file_limit=file_limit, sleep=sleep, |
| 567 | num_errors=bg_errors, watermark_errors=watermark_errors, |
| 568 | failed_queue=failed_queue, passed_queue=dedupe_queue) |
Mike Frysinger | d5fcb3a | 2013-05-30 21:10:50 -0400 | [diff] [blame] | 569 | |
Mike Frysinger | 8ec8c50 | 2014-02-10 00:19:13 -0500 | [diff] [blame] | 570 | start_time = datetime.datetime.now() |
| 571 | Counters = cros_build_lib.Collection( |
Mike Frysinger | 0c0efa2 | 2014-02-09 23:32:23 -0500 | [diff] [blame] | 572 | 'Counters', upload_limit=upload_limit, uploaded_count=0, deduped_count=0) |
Mike Frysinger | 8ec8c50 | 2014-02-10 00:19:13 -0500 | [diff] [blame] | 573 | counters = Counters() |
| 574 | |
Mike Frysinger | 0c0efa2 | 2014-02-09 23:32:23 -0500 | [diff] [blame] | 575 | def _Upload(queue, counters, files): |
| 576 | if not files: |
| 577 | return |
| 578 | |
| 579 | missing_count = 0 |
| 580 | for item in SymbolDeduplicator(storage_query, files): |
| 581 | if counters.upload_limit == 0: |
| 582 | break |
| 583 | |
| 584 | missing_count += 1 |
| 585 | queue.put((item,)) |
| 586 | counters.uploaded_count += 1 |
| 587 | if counters.upload_limit is not None: |
| 588 | counters.upload_limit -= 1 |
| 589 | |
| 590 | counters.deduped_count += (len(files) - missing_count) |
| 591 | |
Mike Frysinger | d41938e | 2014-02-10 06:37:55 -0500 | [diff] [blame^] | 592 | with osutils.TempDir(prefix='upload_symbols.') as tempdir: |
| 593 | storage_notify_proc.start() |
| 594 | # For the first run, we collect the symbols that failed. If the |
| 595 | # overall failure rate was low, we'll retry them on the second run. |
| 596 | for retry in (retry, False): |
| 597 | # We need to limit ourselves to one upload at a time to avoid the server |
| 598 | # kicking in DoS protection. See these bugs for more details: |
| 599 | # http://crbug.com/209442 |
| 600 | # http://crbug.com/212496 |
| 601 | with parallel.BackgroundTaskRunner(uploader, processes=1) as queue: |
Mike Frysinger | 0c0efa2 | 2014-02-09 23:32:23 -0500 | [diff] [blame] | 602 | dedupe_list = [] |
Mike Frysinger | d41938e | 2014-02-10 06:37:55 -0500 | [diff] [blame^] | 603 | for sym_file in SymbolFinder(tempdir, sym_paths): |
| 604 | dedupe_list.append(sym_file) |
| 605 | dedupe_len = len(dedupe_list) |
| 606 | if dedupe_len < dedupe_limit: |
| 607 | if (counters.upload_limit is None or |
| 608 | dedupe_len < counters.upload_limit): |
| 609 | continue |
Mike Frysinger | 02e9240 | 2013-11-22 16:22:02 -0500 | [diff] [blame] | 610 | |
Mike Frysinger | d41938e | 2014-02-10 06:37:55 -0500 | [diff] [blame^] | 611 | _Upload(queue, counters, dedupe_list) |
| 612 | dedupe_list = [] |
| 613 | _Upload(queue, counters, dedupe_list) |
Mike Frysinger | 02e9240 | 2013-11-22 16:22:02 -0500 | [diff] [blame] | 614 | |
Mike Frysinger | d41938e | 2014-02-10 06:37:55 -0500 | [diff] [blame^] | 615 | # See if we need to retry, and if we haven't failed too many times yet. |
| 616 | if not retry or ErrorLimitHit(bg_errors, watermark_errors): |
| 617 | break |
| 618 | |
| 619 | sym_paths = [] |
| 620 | while not failed_queue.empty(): |
| 621 | sym_paths.append(failed_queue.get()) |
| 622 | if sym_paths: |
| 623 | cros_build_lib.Warning('retrying %i symbols', len(sym_paths)) |
| 624 | if counters.upload_limit is not None: |
| 625 | counters.upload_limit += len(sym_paths) |
| 626 | # Decrement the error count in case we recover in the second pass. |
| 627 | assert bg_errors.value >= len(sym_paths), \ |
| 628 | 'more failed files than errors?' |
| 629 | bg_errors.value -= len(sym_paths) |
| 630 | else: |
| 631 | # No failed symbols, so just return now. |
| 632 | break |
Mike Frysinger | 02e9240 | 2013-11-22 16:22:02 -0500 | [diff] [blame] | 633 | |
Mike Frysinger | 7f9be14 | 2014-01-15 02:16:42 -0500 | [diff] [blame] | 634 | # If the user has requested it, save all the symbol files that we failed to |
Mike Frysinger | 0c0efa2 | 2014-02-09 23:32:23 -0500 | [diff] [blame] | 635 | # upload to a listing file. This should help with recovery efforts later. |
| 636 | WriteQueueToFile(failed_list, failed_queue, breakpad_dir) |
Mike Frysinger | 7f9be14 | 2014-01-15 02:16:42 -0500 | [diff] [blame] | 637 | |
Mike Frysinger | 0c0efa2 | 2014-02-09 23:32:23 -0500 | [diff] [blame] | 638 | if dedupe_queue: |
| 639 | dedupe_queue.put(None) |
| 640 | dedupe_queue.close() |
| 641 | storage_notify_proc.join() |
| 642 | |
| 643 | cros_build_lib.Info('uploaded %i symbols (%i were deduped) which took: %s', |
| 644 | counters.uploaded_count, counters.deduped_count, |
Mike Frysinger | 8ec8c50 | 2014-02-10 00:19:13 -0500 | [diff] [blame] | 645 | datetime.datetime.now() - start_time) |
| 646 | |
Mike Frysinger | 9b2ff5c | 2013-11-22 10:01:12 -0500 | [diff] [blame] | 647 | return bg_errors.value |
Mike Frysinger | d5fcb3a | 2013-05-30 21:10:50 -0400 | [diff] [blame] | 648 | |
| 649 | |
Mike Frysinger | d5fcb3a | 2013-05-30 21:10:50 -0400 | [diff] [blame] | 650 | def main(argv): |
Mike Frysinger | 0c0efa2 | 2014-02-09 23:32:23 -0500 | [diff] [blame] | 651 | # TODO(build): Delete this assert. |
| 652 | assert isolateserver, 'Missing isolateserver import http://crbug.com/341152' |
| 653 | |
Mike Frysinger | d5fcb3a | 2013-05-30 21:10:50 -0400 | [diff] [blame] | 654 | parser = commandline.ArgumentParser(description=__doc__) |
| 655 | |
Mike Frysinger | d41938e | 2014-02-10 06:37:55 -0500 | [diff] [blame^] | 656 | parser.add_argument('sym_paths', type='path_or_uri', nargs='*', default=None, |
| 657 | help='symbol file or directory or URL or tarball') |
Mike Frysinger | d5fcb3a | 2013-05-30 21:10:50 -0400 | [diff] [blame] | 658 | parser.add_argument('--board', default=None, |
| 659 | help='board to build packages for') |
| 660 | parser.add_argument('--breakpad_root', type='path', default=None, |
| 661 | help='root directory for breakpad symbols') |
| 662 | parser.add_argument('--official_build', action='store_true', default=False, |
| 663 | help='point to official symbol server') |
| 664 | parser.add_argument('--regenerate', action='store_true', default=False, |
| 665 | help='regenerate all symbols') |
Mike Frysinger | 8ec8c50 | 2014-02-10 00:19:13 -0500 | [diff] [blame] | 666 | parser.add_argument('--upload-limit', type=int, default=None, |
Mike Frysinger | d5fcb3a | 2013-05-30 21:10:50 -0400 | [diff] [blame] | 667 | help='only upload # number of symbols') |
| 668 | parser.add_argument('--strip_cfi', type=int, |
| 669 | default=CRASH_SERVER_FILE_LIMIT - (10 * 1024 * 1024), |
| 670 | help='strip CFI data for files above this size') |
Mike Frysinger | 7f9be14 | 2014-01-15 02:16:42 -0500 | [diff] [blame] | 671 | parser.add_argument('--failed-list', type='path', |
| 672 | help='where to save a list of failed symbols') |
Mike Frysinger | 0c0efa2 | 2014-02-09 23:32:23 -0500 | [diff] [blame] | 673 | parser.add_argument('--dedupe', action='store_true', default=False, |
| 674 | help='use the swarming service to avoid re-uploading') |
Mike Frysinger | d5fcb3a | 2013-05-30 21:10:50 -0400 | [diff] [blame] | 675 | parser.add_argument('--testing', action='store_true', default=False, |
| 676 | help='run in testing mode') |
| 677 | parser.add_argument('--yes', action='store_true', default=False, |
| 678 | help='answer yes to all prompts') |
| 679 | |
| 680 | opts = parser.parse_args(argv) |
Mike Frysinger | 90e49ca | 2014-01-14 14:42:07 -0500 | [diff] [blame] | 681 | opts.Freeze() |
Mike Frysinger | d5fcb3a | 2013-05-30 21:10:50 -0400 | [diff] [blame] | 682 | |
Mike Frysinger | 9b2ff5c | 2013-11-22 10:01:12 -0500 | [diff] [blame] | 683 | if opts.sym_paths: |
Mike Frysinger | 9dcf9ae | 2013-08-10 15:17:09 -0400 | [diff] [blame] | 684 | if opts.regenerate: |
| 685 | cros_build_lib.Die('--regenerate may not be used with specific files') |
| 686 | else: |
| 687 | if opts.board is None: |
| 688 | cros_build_lib.Die('--board is required') |
Mike Frysinger | d5fcb3a | 2013-05-30 21:10:50 -0400 | [diff] [blame] | 689 | |
| 690 | if opts.breakpad_root and opts.regenerate: |
| 691 | cros_build_lib.Die('--regenerate may not be used with --breakpad_root') |
| 692 | |
| 693 | if opts.testing: |
| 694 | # TODO(build): Kill off --testing mode once unittests are up-to-snuff. |
| 695 | cros_build_lib.Info('running in testing mode') |
| 696 | # pylint: disable=W0601,W0603 |
| 697 | global INITIAL_RETRY_DELAY, SymUpload, DEFAULT_SLEEP_DELAY |
| 698 | INITIAL_RETRY_DELAY = DEFAULT_SLEEP_DELAY = 0 |
| 699 | SymUpload = TestingSymUpload |
| 700 | |
Mike Frysinger | 0c0efa2 | 2014-02-09 23:32:23 -0500 | [diff] [blame] | 701 | dedupe_namespace = None |
| 702 | if opts.dedupe: |
| 703 | if opts.official_build and not opts.testing: |
| 704 | dedupe_namespace = OFFICIAL_DEDUPE_NAMESPACE |
| 705 | else: |
| 706 | dedupe_namespace = STAGING_DEDUPE_NAMESPACE |
| 707 | |
Mike Frysinger | d5fcb3a | 2013-05-30 21:10:50 -0400 | [diff] [blame] | 708 | if not opts.yes: |
Mike Frysinger | c5de960 | 2014-02-09 02:42:36 -0500 | [diff] [blame] | 709 | prolog = '\n'.join(textwrap.wrap(textwrap.dedent(""" |
Mike Frysinger | d5fcb3a | 2013-05-30 21:10:50 -0400 | [diff] [blame] | 710 | Uploading symbols for an entire Chromium OS build is really only |
| 711 | necessary for release builds and in a few cases for developers |
| 712 | to debug problems. It will take considerable time to run. For |
| 713 | developer debugging purposes, consider instead passing specific |
| 714 | files to upload. |
Mike Frysinger | c5de960 | 2014-02-09 02:42:36 -0500 | [diff] [blame] | 715 | """), 80)).strip() |
Mike Frysinger | d5fcb3a | 2013-05-30 21:10:50 -0400 | [diff] [blame] | 716 | if not cros_build_lib.BooleanPrompt( |
| 717 | prompt='Are you sure you want to upload all build symbols', |
Mike Frysinger | c5de960 | 2014-02-09 02:42:36 -0500 | [diff] [blame] | 718 | default=False, prolog=prolog): |
Mike Frysinger | d5fcb3a | 2013-05-30 21:10:50 -0400 | [diff] [blame] | 719 | cros_build_lib.Die('better safe than sorry') |
| 720 | |
| 721 | ret = 0 |
| 722 | if opts.regenerate: |
Mike Frysinger | 69cb41d | 2013-08-11 20:08:19 -0400 | [diff] [blame] | 723 | ret += cros_generate_breakpad_symbols.GenerateBreakpadSymbols( |
| 724 | opts.board, breakpad_dir=opts.breakpad_root) |
Mike Frysinger | d5fcb3a | 2013-05-30 21:10:50 -0400 | [diff] [blame] | 725 | |
| 726 | ret += UploadSymbols(opts.board, official=opts.official_build, |
| 727 | breakpad_dir=opts.breakpad_root, |
| 728 | file_limit=opts.strip_cfi, sleep=DEFAULT_SLEEP_DELAY, |
Mike Frysinger | 8ec8c50 | 2014-02-10 00:19:13 -0500 | [diff] [blame] | 729 | upload_limit=opts.upload_limit, sym_paths=opts.sym_paths, |
Mike Frysinger | 0c0efa2 | 2014-02-09 23:32:23 -0500 | [diff] [blame] | 730 | failed_list=opts.failed_list, |
| 731 | dedupe_namespace=dedupe_namespace) |
Mike Frysinger | d5fcb3a | 2013-05-30 21:10:50 -0400 | [diff] [blame] | 732 | if ret: |
| 733 | cros_build_lib.Error('encountered %i problem(s)', ret) |
| 734 | # Since exit(status) gets masked, clamp it to 1 so we don't inadvertently |
| 735 | # return 0 in case we are a multiple of the mask. |
| 736 | ret = 1 |
| 737 | |
| 738 | return ret |
Mike Frysinger | 094a217 | 2013-08-14 12:54:35 -0400 | [diff] [blame] | 739 | |
| 740 | |
| 741 | # We need this to run once per process. Do it at module import time as that |
| 742 | # will let us avoid doing it inline at function call time (see SymUpload) as |
| 743 | # that func might be called by the multiprocessing module which means we'll |
| 744 | # do the opener logic multiple times overall. Plus, if you're importing this |
| 745 | # module, it's a pretty good chance that you're going to need this. |
| 746 | poster.streaminghttp.register_openers() |