blob: ceaee58a414adfe3b87ef0f54b5df094bb418e4b [file] [log] [blame]
Mike Frysingerd5fcb3a2013-05-30 21:10:50 -04001# 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
7This script need only be used to upload release builds symbols or to debug
8crashes on non-release builds (in which case try to only upload the symbols
Mike Frysinger02e1e072013-11-10 22:11:34 -05009for those executables involved).
10"""
Mike Frysingerd5fcb3a2013-05-30 21:10:50 -040011
Mike Frysingera4fa1e82014-01-15 01:45:56 -050012from __future__ import print_function
13
Mike Frysingerd5fcb3a2013-05-30 21:10:50 -040014import ctypes
Mike Frysinger8ec8c502014-02-10 00:19:13 -050015import datetime
Mike Frysinger02e92402013-11-22 16:22:02 -050016import functools
Mike Frysinger0c0efa22014-02-09 23:32:23 -050017import hashlib
Mike Frysingera4fa1e82014-01-15 01:45:56 -050018import httplib
Mike Frysingerd5fcb3a2013-05-30 21:10:50 -040019import multiprocessing
20import os
Mike Frysinger094a2172013-08-14 12:54:35 -040021import poster
Mike Frysingerd5fcb3a2013-05-30 21:10:50 -040022import random
Mike Frysingerfd355652014-01-23 02:57:48 -050023import socket
Mike Frysingerd5fcb3a2013-05-30 21:10:50 -040024import textwrap
25import tempfile
26import time
Mike Frysinger094a2172013-08-14 12:54:35 -040027import urllib2
Mike Frysingerd41938e2014-02-10 06:37:55 -050028import urlparse
Mike Frysingerd5fcb3a2013-05-30 21:10:50 -040029
Mike Frysinger0c0efa22014-02-09 23:32:23 -050030from chromite.buildbot import constants
Mike Frysingerd41938e2014-02-10 06:37:55 -050031from chromite.lib import cache
Mike Frysingerd5fcb3a2013-05-30 21:10:50 -040032from chromite.lib import commandline
33from chromite.lib import cros_build_lib
Mike Frysingerd41938e2014-02-10 06:37:55 -050034from chromite.lib import gs
35from chromite.lib import osutils
Mike Frysingerd5fcb3a2013-05-30 21:10:50 -040036from chromite.lib import parallel
David Jamesc93e6a4d2014-01-13 11:37:36 -080037from chromite.lib import retry_util
Mike Frysinger0c0efa22014-02-09 23:32:23 -050038from chromite.lib import timeout_util
Mike Frysinger69cb41d2013-08-11 20:08:19 -040039from chromite.scripts import cros_generate_breakpad_symbols
Mike Frysingerd5fcb3a2013-05-30 21:10:50 -040040
Mike Frysinger0c0efa22014-02-09 23:32:23 -050041# 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
46try:
47 import isolateserver
48except ImportError:
49 isolateserver = None
50
Mike Frysingerd5fcb3a2013-05-30 21:10:50 -040051
52# URLs used for uploading symbols.
53OFFICIAL_UPLOAD_URL = 'http://clients2.google.com/cr/symbol'
54STAGING_UPLOAD_URL = 'http://clients2.google.com/cr/staging_symbol'
55
56
57# The crash server rejects files that are this big.
58CRASH_SERVER_FILE_LIMIT = 350 * 1024 * 1024
59# Give ourselves a little breathing room from what the server expects.
60DEFAULT_FILE_LIMIT = CRASH_SERVER_FILE_LIMIT - (10 * 1024 * 1024)
61
62
Mike Frysinger0c0efa22014-02-09 23:32:23 -050063# 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.
70DEDUPE_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.
75DEDUPE_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.
79OFFICIAL_DEDUPE_NAMESPACE = 'chromium-os-upload-symbols'
80STAGING_DEDUPE_NAMESPACE = '%s-staging' % OFFICIAL_DEDUPE_NAMESPACE
81
82
Mike Frysingercd78a082013-06-26 17:13:04 -040083# 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.
85UPLOAD_TIMEOUT = 30 * 60
86
87
Mike Frysingerd5fcb3a2013-05-30 21:10:50 -040088# Sleep for 200ms in between uploads to avoid DoS'ing symbol server.
89DEFAULT_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.
94INITIAL_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).
98MAX_RETRIES = 6
99
Mike Frysingereb753bf2013-11-22 16:05:35 -0500100# 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.
103MAX_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.
123ERROR_WATERMARK = 3.0
124ERROR_ADJUST_FAIL = 1.0
125ERROR_ADJUST_PASS = -0.5
Mike Frysingerd5fcb3a2013-05-30 21:10:50 -0400126
127
Mike Frysinger0c0efa22014-02-09 23:32:23 -0500128def SymUpload(upload_url, sym_item):
Mike Frysinger094a2172013-08-14 12:54:35 -0400129 """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 Frysinger094a2172013-08-14 12:54:35 -0400145 upload_url: The crash URL to POST the |sym_file| to
Mike Frysinger0c0efa22014-02-09 23:32:23 -0500146 sym_item: A SymbolItem containing the path to the breakpad symbol to upload
Mike Frysinger094a2172013-08-14 12:54:35 -0400147 """
Mike Frysinger0c0efa22014-02-09 23:32:23 -0500148 sym_header = sym_item.sym_header
149 sym_file = sym_item.sym_file
Mike Frysinger094a2172013-08-14 12:54:35 -0400150
151 fields = (
152 ('code_file', sym_header.name),
153 ('debug_file', sym_header.name),
154 ('debug_identifier', sym_header.id.replace('-', '')),
Mike Frysingerb8a966b2014-03-19 17:36:18 -0400155 # The product/version fields are used by the server only for statistic
156 # purposes. They do not impact symbolization, so they're safe to set
157 # to any value all the time.
158 # In this case, we use it to help see the load our build system is
159 # placing on the server.
160 # Not sure what to set for the version. Maybe the git sha1 of this file.
161 # Note: the server restricts this to 30 chars.
Mike Frysinger094a2172013-08-14 12:54:35 -0400162 #('version', None),
Mike Frysingerb8a966b2014-03-19 17:36:18 -0400163 ('product', 'ChromeOS'),
Mike Frysinger094a2172013-08-14 12:54:35 -0400164 ('os', sym_header.os),
165 ('cpu', sym_header.cpu),
166 poster.encode.MultipartParam.from_file('symbol_file', sym_file),
167 )
168
169 data, headers = poster.encode.multipart_encode(fields)
170 request = urllib2.Request(upload_url, data, headers)
171 request.add_header('User-agent', 'chromite.upload_symbols')
172 urllib2.urlopen(request, timeout=UPLOAD_TIMEOUT)
Mike Frysingerd5fcb3a2013-05-30 21:10:50 -0400173
174
Mike Frysinger0c0efa22014-02-09 23:32:23 -0500175def TestingSymUpload(upload_url, sym_item):
Mike Frysingerd5fcb3a2013-05-30 21:10:50 -0400176 """A stub version of SymUpload for --testing usage"""
Mike Frysinger0c0efa22014-02-09 23:32:23 -0500177 cmd = ['sym_upload', sym_item.sym_file, upload_url]
Mike Frysingerd5fcb3a2013-05-30 21:10:50 -0400178 # Randomly fail 80% of the time (the retry logic makes this 80%/3 per file).
179 returncode = random.randint(1, 100) <= 80
180 cros_build_lib.Debug('would run (and return %i): %s', returncode,
Matt Tennant7feda352013-12-20 14:03:40 -0800181 cros_build_lib.CmdToStr(cmd))
Mike Frysingerd5fcb3a2013-05-30 21:10:50 -0400182 if returncode:
183 output = 'Failed to send the symbol file.'
184 else:
185 output = 'Successfully sent the symbol file.'
186 result = cros_build_lib.CommandResult(cmd=cmd, error=None, output=output,
187 returncode=returncode)
188 if returncode:
Mike Frysingera4fa1e82014-01-15 01:45:56 -0500189 exceptions = (
Mike Frysingerfd355652014-01-23 02:57:48 -0500190 socket.error('[socket.error] forced test fail'),
Mike Frysingera4fa1e82014-01-15 01:45:56 -0500191 httplib.BadStatusLine('[BadStatusLine] forced test fail'),
192 urllib2.HTTPError(upload_url, 400, '[HTTPError] forced test fail',
193 {}, None),
194 urllib2.URLError('[URLError] forced test fail'),
195 )
196 raise random.choice(exceptions)
Mike Frysingerd5fcb3a2013-05-30 21:10:50 -0400197 else:
198 return result
199
200
Mike Frysingereb753bf2013-11-22 16:05:35 -0500201def ErrorLimitHit(num_errors, watermark_errors):
202 """See if our error limit has been hit
203
204 Args:
205 num_errors: A multiprocessing.Value of the raw number of failures.
206 watermark_errors: A multiprocessing.Value of the current rate of failures.
Mike Frysinger1a736a82013-12-12 01:50:59 -0500207
Mike Frysingereb753bf2013-11-22 16:05:35 -0500208 Returns:
209 True if our error limits have been exceeded.
210 """
211 return ((num_errors is not None and
212 num_errors.value > MAX_TOTAL_ERRORS_FOR_RETRY) or
213 (watermark_errors is not None and
214 watermark_errors.value > ERROR_WATERMARK))
215
216
217def _UpdateCounter(counter, adj):
218 """Update |counter| by |adj|
219
220 Handle atomic updates of |counter|. Also make sure it does not
221 fall below 0.
222
223 Args:
224 counter: A multiprocessing.Value to update
225 adj: The value to add to |counter|
226 """
227 def _Update():
228 clamp = 0 if type(adj) is int else 0.0
229 counter.value = max(clamp, counter.value + adj)
230
231 if hasattr(counter, 'get_lock'):
232 with counter.get_lock():
233 _Update()
234 elif counter is not None:
235 _Update()
236
237
Mike Frysinger0c0efa22014-02-09 23:32:23 -0500238def UploadSymbol(upload_url, sym_item, file_limit=DEFAULT_FILE_LIMIT,
Mike Frysinger02e92402013-11-22 16:22:02 -0500239 sleep=0, num_errors=None, watermark_errors=None,
Mike Frysinger0c0efa22014-02-09 23:32:23 -0500240 failed_queue=None, passed_queue=None):
241 """Upload |sym_item| to |upload_url|
Mike Frysingerd5fcb3a2013-05-30 21:10:50 -0400242
243 Args:
Mike Frysingerd5fcb3a2013-05-30 21:10:50 -0400244 upload_url: The crash server to upload things to
Mike Frysinger0c0efa22014-02-09 23:32:23 -0500245 sym_item: A SymbolItem containing the path to the breakpad symbol to upload
Mike Frysingerd5fcb3a2013-05-30 21:10:50 -0400246 file_limit: The max file size of a symbol file before we try to strip it
247 sleep: Number of seconds to sleep before running
Mike Frysinger69cb41d2013-08-11 20:08:19 -0400248 num_errors: An object to update with the error count (needs a .value member)
Mike Frysingereb753bf2013-11-22 16:05:35 -0500249 watermark_errors: An object to track current error behavior (needs a .value)
Mike Frysinger02e92402013-11-22 16:22:02 -0500250 failed_queue: When a symbol fails, add it to this queue
Mike Frysinger0c0efa22014-02-09 23:32:23 -0500251 passed_queue: When a symbol passes, add it to this queue
Mike Frysinger1a736a82013-12-12 01:50:59 -0500252
Mike Frysinger69cb41d2013-08-11 20:08:19 -0400253 Returns:
254 The number of errors that were encountered.
Mike Frysingerd5fcb3a2013-05-30 21:10:50 -0400255 """
Mike Frysinger0c0efa22014-02-09 23:32:23 -0500256 sym_file = sym_item.sym_file
257 upload_item = sym_item
258
Mike Frysingerd5fcb3a2013-05-30 21:10:50 -0400259 if num_errors is None:
260 num_errors = ctypes.c_int()
Mike Frysingereb753bf2013-11-22 16:05:35 -0500261 if ErrorLimitHit(num_errors, watermark_errors):
Mike Frysingerd5fcb3a2013-05-30 21:10:50 -0400262 # Abandon ship! It's on fire! NOoooooooooooOOOoooooo.
Mike Frysinger7f9be142014-01-15 02:16:42 -0500263 if failed_queue:
264 failed_queue.put(sym_file)
Mike Frysingerd5fcb3a2013-05-30 21:10:50 -0400265 return 0
266
Mike Frysingerd5fcb3a2013-05-30 21:10:50 -0400267 if sleep:
268 # Keeps us from DoS-ing the symbol server.
269 time.sleep(sleep)
270
271 cros_build_lib.Debug('uploading %s' % sym_file)
272
273 # Ideally there'd be a tempfile.SpooledNamedTemporaryFile that we could use.
274 with tempfile.NamedTemporaryFile(prefix='upload_symbols',
275 bufsize=0) as temp_sym_file:
276 if file_limit:
277 # If the symbols size is too big, strip out the call frame info. The CFI
278 # is unnecessary for 32bit x86 targets where the frame pointer is used (as
279 # all of ours have) and it accounts for over half the size of the symbols
280 # uploaded.
281 file_size = os.path.getsize(sym_file)
282 if file_size > file_limit:
283 cros_build_lib.Warning('stripping CFI from %s due to size %s > %s',
284 sym_file, file_size, file_limit)
285 temp_sym_file.writelines([x for x in open(sym_file, 'rb').readlines()
286 if not x.startswith('STACK CFI')])
Mike Frysinger0c0efa22014-02-09 23:32:23 -0500287
288 upload_item = FakeItem(sym_file=temp_sym_file.name,
289 sym_header=sym_item.sym_header)
Mike Frysingerd5fcb3a2013-05-30 21:10:50 -0400290
291 # Hopefully the crash server will let it through. But it probably won't.
292 # Not sure what the best answer is in this case.
Mike Frysinger0c0efa22014-02-09 23:32:23 -0500293 file_size = os.path.getsize(upload_item.sym_file)
Mike Frysingerd5fcb3a2013-05-30 21:10:50 -0400294 if file_size > CRASH_SERVER_FILE_LIMIT:
295 cros_build_lib.PrintBuildbotStepWarnings()
Mike Frysinger02e92402013-11-22 16:22:02 -0500296 cros_build_lib.Warning('upload file %s is awfully large, risking '
297 'rejection by the symbol server (%s > %s)',
298 sym_file, file_size, CRASH_SERVER_FILE_LIMIT)
Mike Frysingerd5fcb3a2013-05-30 21:10:50 -0400299
300 # Upload the symbol file.
Mike Frysingereb753bf2013-11-22 16:05:35 -0500301 success = False
Mike Frysingerd5fcb3a2013-05-30 21:10:50 -0400302 try:
Mike Frysinger9adcfd22013-10-24 12:01:40 -0400303 cros_build_lib.TimedCommand(
David Jamesc93e6a4d2014-01-13 11:37:36 -0800304 retry_util.RetryException,
Mike Frysinger9adcfd22013-10-24 12:01:40 -0400305 (urllib2.HTTPError, urllib2.URLError), MAX_RETRIES, SymUpload,
Mike Frysinger0c0efa22014-02-09 23:32:23 -0500306 upload_url, upload_item, sleep=INITIAL_RETRY_DELAY,
Mike Frysinger9adcfd22013-10-24 12:01:40 -0400307 timed_log_msg='upload of %10i bytes took %%s: %s' %
308 (file_size, os.path.basename(sym_file)))
Mike Frysingereb753bf2013-11-22 16:05:35 -0500309 success = True
Mike Frysinger0c0efa22014-02-09 23:32:23 -0500310
311 if passed_queue:
312 passed_queue.put(sym_item)
Mike Frysinger094a2172013-08-14 12:54:35 -0400313 except urllib2.HTTPError as e:
314 cros_build_lib.Warning('could not upload: %s: HTTP %s: %s',
315 os.path.basename(sym_file), e.code, e.reason)
Mike Frysingerfd355652014-01-23 02:57:48 -0500316 except (urllib2.URLError, httplib.HTTPException, socket.error) as e:
Mike Frysingerc4ab5782013-10-02 18:14:22 -0400317 cros_build_lib.Warning('could not upload: %s: %s',
318 os.path.basename(sym_file), e)
Mike Frysingereb753bf2013-11-22 16:05:35 -0500319 finally:
320 if success:
321 _UpdateCounter(watermark_errors, ERROR_ADJUST_PASS)
322 else:
323 _UpdateCounter(num_errors, 1)
324 _UpdateCounter(watermark_errors, ERROR_ADJUST_FAIL)
Mike Frysinger02e92402013-11-22 16:22:02 -0500325 if failed_queue:
326 failed_queue.put(sym_file)
Mike Frysingerd5fcb3a2013-05-30 21:10:50 -0400327
328 return num_errors.value
329
330
Mike Frysinger0c0efa22014-02-09 23:32:23 -0500331# A dummy class that allows for stubbing in tests and SymUpload.
332FakeItem = cros_build_lib.Collection(
333 'FakeItem', sym_file=None, sym_header=None, content=lambda x: '')
334
335
336# TODO(build): Delete this if check. http://crbug.com/341152
337if isolateserver:
338 class SymbolItem(isolateserver.BufferItem):
339 """Turn a sym_file into an isolateserver.Item"""
340
341 ALGO = hashlib.sha1
342
343 def __init__(self, sym_file):
344 sym_header = cros_generate_breakpad_symbols.ReadSymsHeader(sym_file)
345 super(SymbolItem, self).__init__(str(sym_header), self.ALGO)
346 self.sym_header = sym_header
347 self.sym_file = sym_file
348
349
350def SymbolDeduplicatorNotify(dedupe_namespace, dedupe_queue):
351 """Send a symbol file to the swarming service
352
353 Notify the swarming service of a successful upload. If the notification fails
354 for any reason, we ignore it. We don't care as it just means we'll upload it
355 again later on, and the symbol server will handle that graciously.
356
357 This func runs in a different process from the main one, so we cannot share
358 the storage object. Instead, we create our own. This func stays alive for
359 the life of the process, so we only create one here overall.
360
361 Args:
362 dedupe_namespace: The isolateserver namespace to dedupe uploaded symbols.
363 dedupe_queue: The queue to read SymbolItems from
364 """
365 if dedupe_queue is None:
366 return
367
368 item = None
369 try:
370 storage = isolateserver.get_storage_api(constants.ISOLATESERVER,
371 dedupe_namespace)
372 for item in iter(dedupe_queue.get, None):
373 with timeout_util.Timeout(DEDUPE_TIMEOUT):
374 storage.push(item, item.content(0))
375 except Exception:
376 sym_file = item.sym_file if (item and item.sym_file) else ''
377 cros_build_lib.Warning('posting %s to dedupe server failed',
378 os.path.basename(sym_file), exc_info=True)
379
Mike Frysinger58312e92014-03-18 04:18:36 -0400380 # Keep draining the queue though so it doesn't fill up.
381 while dedupe_queue.get() is not None:
382 continue
383
Mike Frysinger0c0efa22014-02-09 23:32:23 -0500384
385def SymbolDeduplicator(storage, sym_paths):
386 """Filter out symbol files that we've already uploaded
387
388 Using the swarming service, ask it to tell us which symbol files we've already
389 uploaded in previous runs and/or by other bots. If the query fails for any
390 reason, we'll just upload all symbols. This is fine as the symbol server will
391 do the right thing and this phase is purely an optimization.
392
393 This code runs in the main thread which is why we can re-use the existing
394 storage object. Saves us from having to recreate one all the time.
395
396 Args:
397 storage: An isolateserver.StorageApi object
398 sym_paths: List of symbol files to check against the dedupe server
399
400 Returns:
401 List of symbol files that have not been uploaded before
402 """
403 if not sym_paths:
404 return sym_paths
405
406 items = [SymbolItem(x) for x in sym_paths]
407 if storage:
408 try:
409 with timeout_util.Timeout(DEDUPE_TIMEOUT):
410 items = storage.contains(items)
411 except Exception:
412 cros_build_lib.Warning('talking to dedupe server failed', exc_info=True)
413
414 return items
415
416
Mike Frysingerd41938e2014-02-10 06:37:55 -0500417def IsTarball(path):
418 """Guess if this is a tarball based on the filename."""
419 parts = path.split('.')
420 if len(parts) <= 1:
421 return False
422
423 if parts[-1] == 'tar':
424 return True
425
426 if parts[-2] == 'tar':
427 return parts[-1] in ('bz2', 'gz', 'xz')
428
429 return parts[-1] in ('tbz2', 'tbz', 'tgz', 'txz')
430
431
432def SymbolFinder(tempdir, paths):
Mike Frysinger9b2ff5c2013-11-22 10:01:12 -0500433 """Locate symbol files in |paths|
434
435 Args:
Mike Frysingerd41938e2014-02-10 06:37:55 -0500436 tempdir: Path to use for temporary files (caller will clean up).
Mike Frysinger9b2ff5c2013-11-22 10:01:12 -0500437 paths: A list of input paths to walk. Files are returned w/out any checks.
Mike Frysingerd41938e2014-02-10 06:37:55 -0500438 Dirs are searched for files that end in ".sym". Urls are fetched and then
439 processed. Tarballs are unpacked and walked.
Mike Frysinger1a736a82013-12-12 01:50:59 -0500440
Mike Frysinger9b2ff5c2013-11-22 10:01:12 -0500441 Returns:
442 Yield every viable sym file.
443 """
444 for p in paths:
Don Garrett25f309a2014-03-19 14:02:12 -0700445 # Pylint is confused about members of ParseResult.
Don Garrettf8bf7842014-03-20 17:03:42 -0700446
Mike Frysingerd41938e2014-02-10 06:37:55 -0500447 o = urlparse.urlparse(p)
Don Garrettf8bf7842014-03-20 17:03:42 -0700448 if o.scheme: # pylint: disable=E1101
Mike Frysingerd41938e2014-02-10 06:37:55 -0500449 # Support globs of filenames.
450 ctx = gs.GSContext()
451 for p in ctx.LS(p):
452 cros_build_lib.Info('processing files inside %s', p)
453 o = urlparse.urlparse(p)
454 cache_dir = commandline.GetCacheDir()
455 common_path = os.path.join(cache_dir, constants.COMMON_CACHE)
456 tar_cache = cache.TarballCache(common_path)
Don Garrettf8bf7842014-03-20 17:03:42 -0700457 key = ('%s%s' % (o.netloc, o.path)).split('/') # pylint: disable=E1101
Mike Frysingerd41938e2014-02-10 06:37:55 -0500458 # The common cache will not be LRU, removing the need to hold a read
459 # lock on the cached gsutil.
460 ref = tar_cache.Lookup(key)
461 try:
462 ref.SetDefault(p)
463 except cros_build_lib.RunCommandError as e:
464 cros_build_lib.Warning('ignoring %s\n%s', p, e)
465 continue
466 for p in SymbolFinder(tempdir, [ref.path]):
467 yield p
468
469 elif os.path.isdir(p):
Mike Frysinger9b2ff5c2013-11-22 10:01:12 -0500470 for root, _, files in os.walk(p):
471 for f in files:
472 if f.endswith('.sym'):
473 yield os.path.join(root, f)
Mike Frysingerd41938e2014-02-10 06:37:55 -0500474
475 elif IsTarball(p):
476 cros_build_lib.Info('processing files inside %s', p)
477 tardir = tempfile.mkdtemp(dir=tempdir)
478 cache.Untar(os.path.realpath(p), tardir)
479 for p in SymbolFinder(tardir, [tardir]):
480 yield p
481
Mike Frysinger9b2ff5c2013-11-22 10:01:12 -0500482 else:
483 yield p
484
485
Mike Frysinger0c0efa22014-02-09 23:32:23 -0500486def WriteQueueToFile(listing, queue, relpath=None):
487 """Write all the items in |queue| to the |listing|.
488
Mike Frysinger5e6dd712014-03-07 22:21:17 -0500489 Note: The queue must have a sentinel None appended to the end.
490
Mike Frysinger0c0efa22014-02-09 23:32:23 -0500491 Args:
492 listing: Where to write out the list of files.
493 queue: The queue of paths to drain.
494 relpath: If set, write out paths relative to this one.
495 """
496 if not listing:
Mike Frysingera0ddac62014-03-14 10:30:25 -0400497 # Still drain the queue so we make sure the producer has finished
498 # before we return. Otherwise, the queue might get destroyed too
499 # quickly which will trigger a traceback in the producer.
500 while queue.get() is not None:
501 continue
Mike Frysinger0c0efa22014-02-09 23:32:23 -0500502 return
503
504 with cros_build_lib.Open(listing, 'wb+') as f:
Mike Frysinger5e6dd712014-03-07 22:21:17 -0500505 while True:
Mike Frysinger0c0efa22014-02-09 23:32:23 -0500506 path = queue.get()
Mike Frysinger5e6dd712014-03-07 22:21:17 -0500507 if path is None:
508 return
Mike Frysinger0c0efa22014-02-09 23:32:23 -0500509 if relpath:
510 path = os.path.relpath(path, relpath)
511 f.write('%s\n' % path)
512
513
Mike Frysinger9dcf9ae2013-08-10 15:17:09 -0400514def UploadSymbols(board=None, official=False, breakpad_dir=None,
Mike Frysingerd5fcb3a2013-05-30 21:10:50 -0400515 file_limit=DEFAULT_FILE_LIMIT, sleep=DEFAULT_SLEEP_DELAY,
Mike Frysinger8ec8c502014-02-10 00:19:13 -0500516 upload_limit=None, sym_paths=None, failed_list=None,
Mike Frysinger0c0efa22014-02-09 23:32:23 -0500517 root=None, retry=True, dedupe_namespace=None):
Mike Frysingerd5fcb3a2013-05-30 21:10:50 -0400518 """Upload all the generated symbols for |board| to the crash server
519
Mike Frysinger9dcf9ae2013-08-10 15:17:09 -0400520 You can use in a few ways:
521 * pass |board| to locate all of its symbols
522 * pass |breakpad_dir| to upload all the symbols in there
Mike Frysinger9b2ff5c2013-11-22 10:01:12 -0500523 * pass |sym_paths| to upload specific symbols (or dirs of symbols)
Mike Frysinger9dcf9ae2013-08-10 15:17:09 -0400524
Mike Frysingerd5fcb3a2013-05-30 21:10:50 -0400525 Args:
526 board: The board whose symbols we wish to upload
527 official: Use the official symbol server rather than the staging one
528 breakpad_dir: The full path to the breakpad directory where symbols live
529 file_limit: The max file size of a symbol file before we try to strip it
530 sleep: How long to sleep in between uploads
Mike Frysinger8ec8c502014-02-10 00:19:13 -0500531 upload_limit: If set, only upload this many symbols (meant for testing)
Mike Frysinger9b2ff5c2013-11-22 10:01:12 -0500532 sym_paths: Specific symbol files (or dirs of sym files) to upload,
533 otherwise search |breakpad_dir|
Mike Frysinger7f9be142014-01-15 02:16:42 -0500534 failed_list: Write the names of all sym files we did not upload; can be a
535 filename or file-like object.
Mike Frysinger118d2502013-08-19 03:36:56 -0400536 root: The tree to prefix to |breakpad_dir| (if |breakpad_dir| is not set)
Mike Frysinger02e92402013-11-22 16:22:02 -0500537 retry: Whether we should retry failures.
Mike Frysinger0c0efa22014-02-09 23:32:23 -0500538 dedupe_namespace: The isolateserver namespace to dedupe uploaded symbols.
Mike Frysinger1a736a82013-12-12 01:50:59 -0500539
Mike Frysingerd5fcb3a2013-05-30 21:10:50 -0400540 Returns:
Mike Frysinger69cb41d2013-08-11 20:08:19 -0400541 The number of errors that were encountered.
Mike Frysingerd5fcb3a2013-05-30 21:10:50 -0400542 """
Mike Frysinger0c0efa22014-02-09 23:32:23 -0500543 # TODO(build): Delete this assert.
544 assert isolateserver, 'Missing isolateserver import http://crbug.com/341152'
545
Mike Frysingerd5fcb3a2013-05-30 21:10:50 -0400546 if official:
547 upload_url = OFFICIAL_UPLOAD_URL
548 else:
549 cros_build_lib.Warning('unofficial builds upload to the staging server')
550 upload_url = STAGING_UPLOAD_URL
551
Mike Frysinger9b2ff5c2013-11-22 10:01:12 -0500552 if sym_paths:
553 cros_build_lib.Info('uploading specified symbols to %s', upload_url)
Mike Frysinger9dcf9ae2013-08-10 15:17:09 -0400554 else:
555 if breakpad_dir is None:
Mike Frysinger118d2502013-08-19 03:36:56 -0400556 breakpad_dir = os.path.join(
557 root,
558 cros_generate_breakpad_symbols.FindBreakpadDir(board).lstrip('/'))
Mike Frysinger9dcf9ae2013-08-10 15:17:09 -0400559 cros_build_lib.Info('uploading all symbols to %s from %s', upload_url,
560 breakpad_dir)
Mike Frysinger9b2ff5c2013-11-22 10:01:12 -0500561 sym_paths = [breakpad_dir]
Mike Frysingerd5fcb3a2013-05-30 21:10:50 -0400562
Mike Frysinger0c0efa22014-02-09 23:32:23 -0500563 # We use storage_query to ask the server about existing symbols. The
564 # storage_notify_proc process is used to post updates to the server. We
565 # cannot safely share the storage object between threads/processes, but
566 # we also want to minimize creating new ones as each object has to init
567 # new state (like server connections).
568 if dedupe_namespace:
569 dedupe_limit = DEDUPE_LIMIT
570 dedupe_queue = multiprocessing.Queue()
571 storage_query = isolateserver.get_storage_api(constants.ISOLATESERVER,
572 dedupe_namespace)
573 else:
574 dedupe_limit = 1
575 dedupe_queue = storage_query = None
576 # Can't use parallel.BackgroundTaskRunner because that'll create multiple
577 # processes and we want only one the whole time (see comment above).
578 storage_notify_proc = multiprocessing.Process(
579 target=SymbolDeduplicatorNotify, args=(dedupe_namespace, dedupe_queue))
580
Mike Frysingerd5fcb3a2013-05-30 21:10:50 -0400581 bg_errors = multiprocessing.Value('i')
Mike Frysingereb753bf2013-11-22 16:05:35 -0500582 watermark_errors = multiprocessing.Value('f')
Mike Frysinger02e92402013-11-22 16:22:02 -0500583 failed_queue = multiprocessing.Queue()
584 uploader = functools.partial(
Mike Frysinger0c0efa22014-02-09 23:32:23 -0500585 UploadSymbol, upload_url, file_limit=file_limit, sleep=sleep,
586 num_errors=bg_errors, watermark_errors=watermark_errors,
587 failed_queue=failed_queue, passed_queue=dedupe_queue)
Mike Frysingerd5fcb3a2013-05-30 21:10:50 -0400588
Mike Frysinger8ec8c502014-02-10 00:19:13 -0500589 start_time = datetime.datetime.now()
590 Counters = cros_build_lib.Collection(
Mike Frysinger0c0efa22014-02-09 23:32:23 -0500591 'Counters', upload_limit=upload_limit, uploaded_count=0, deduped_count=0)
Mike Frysinger8ec8c502014-02-10 00:19:13 -0500592 counters = Counters()
593
Mike Frysinger0c0efa22014-02-09 23:32:23 -0500594 def _Upload(queue, counters, files):
595 if not files:
596 return
597
598 missing_count = 0
599 for item in SymbolDeduplicator(storage_query, files):
Mike Frysinger0c0efa22014-02-09 23:32:23 -0500600 missing_count += 1
Mike Frysingerd42e5f02014-03-14 11:19:37 -0400601
602 if counters.upload_limit == 0:
603 continue
604
Mike Frysinger0c0efa22014-02-09 23:32:23 -0500605 queue.put((item,))
606 counters.uploaded_count += 1
607 if counters.upload_limit is not None:
608 counters.upload_limit -= 1
609
610 counters.deduped_count += (len(files) - missing_count)
611
Mike Frysinger13870082014-03-14 10:41:20 -0400612 try:
Mike Frysingerd41938e2014-02-10 06:37:55 -0500613 storage_notify_proc.start()
Mike Frysinger02e92402013-11-22 16:22:02 -0500614
Mike Frysinger13870082014-03-14 10:41:20 -0400615 with osutils.TempDir(prefix='upload_symbols.') as tempdir:
616 # For the first run, we collect the symbols that failed. If the
617 # overall failure rate was low, we'll retry them on the second run.
618 for retry in (retry, False):
619 # We need to limit ourselves to one upload at a time to avoid the server
620 # kicking in DoS protection. See these bugs for more details:
621 # http://crbug.com/209442
622 # http://crbug.com/212496
623 with parallel.BackgroundTaskRunner(uploader, processes=1) as queue:
Mike Frysingerd41938e2014-02-10 06:37:55 -0500624 dedupe_list = []
Mike Frysinger13870082014-03-14 10:41:20 -0400625 for sym_file in SymbolFinder(tempdir, sym_paths):
626 dedupe_list.append(sym_file)
627 dedupe_len = len(dedupe_list)
628 if dedupe_len < dedupe_limit:
629 if (counters.upload_limit is None or
630 dedupe_len < counters.upload_limit):
631 continue
Mike Frysinger02e92402013-11-22 16:22:02 -0500632
Mike Frysinger1010a892014-03-14 11:24:17 -0400633 # We check the counter before _Upload so that we don't keep talking
634 # to the dedupe server. Otherwise, we end up sending one symbol at
635 # a time to it and that slows things down a lot.
636 if counters.upload_limit == 0:
637 break
638
Mike Frysinger13870082014-03-14 10:41:20 -0400639 _Upload(queue, counters, dedupe_list)
640 dedupe_list = []
641 _Upload(queue, counters, dedupe_list)
Mike Frysingerd41938e2014-02-10 06:37:55 -0500642
Mike Frysinger13870082014-03-14 10:41:20 -0400643 # See if we need to retry, and if we haven't failed too many times yet.
644 if not retry or ErrorLimitHit(bg_errors, watermark_errors):
Mike Frysinger5e6dd712014-03-07 22:21:17 -0500645 break
Mike Frysinger5e6dd712014-03-07 22:21:17 -0500646
Mike Frysinger13870082014-03-14 10:41:20 -0400647 sym_paths = []
648 failed_queue.put(None)
649 while True:
650 sym_path = failed_queue.get()
651 if sym_path is None:
652 break
653 sym_paths.append(sym_path)
Mike Frysinger02e92402013-11-22 16:22:02 -0500654
Mike Frysinger13870082014-03-14 10:41:20 -0400655 if sym_paths:
656 cros_build_lib.Warning('retrying %i symbols', len(sym_paths))
657 if counters.upload_limit is not None:
658 counters.upload_limit += len(sym_paths)
659 # Decrement the error count in case we recover in the second pass.
660 assert bg_errors.value >= len(sym_paths), \
661 'more failed files than errors?'
662 bg_errors.value -= len(sym_paths)
663 else:
664 # No failed symbols, so just return now.
665 break
Mike Frysinger7f9be142014-01-15 02:16:42 -0500666
Mike Frysinger13870082014-03-14 10:41:20 -0400667 # If the user has requested it, save all the symbol files that we failed to
668 # upload to a listing file. This should help with recovery efforts later.
669 failed_queue.put(None)
670 WriteQueueToFile(failed_list, failed_queue, breakpad_dir)
671
672 finally:
673 if dedupe_queue:
674 dedupe_queue.put(None)
675 storage_notify_proc.join()
Mike Frysinger0c0efa22014-02-09 23:32:23 -0500676
677 cros_build_lib.Info('uploaded %i symbols (%i were deduped) which took: %s',
678 counters.uploaded_count, counters.deduped_count,
Mike Frysinger8ec8c502014-02-10 00:19:13 -0500679 datetime.datetime.now() - start_time)
680
Mike Frysinger9b2ff5c2013-11-22 10:01:12 -0500681 return bg_errors.value
Mike Frysingerd5fcb3a2013-05-30 21:10:50 -0400682
683
Mike Frysingerd5fcb3a2013-05-30 21:10:50 -0400684def main(argv):
Mike Frysinger0c0efa22014-02-09 23:32:23 -0500685 # TODO(build): Delete this assert.
686 assert isolateserver, 'Missing isolateserver import http://crbug.com/341152'
687
Mike Frysingerd5fcb3a2013-05-30 21:10:50 -0400688 parser = commandline.ArgumentParser(description=__doc__)
689
Mike Frysingerd41938e2014-02-10 06:37:55 -0500690 parser.add_argument('sym_paths', type='path_or_uri', nargs='*', default=None,
691 help='symbol file or directory or URL or tarball')
Mike Frysingerd5fcb3a2013-05-30 21:10:50 -0400692 parser.add_argument('--board', default=None,
693 help='board to build packages for')
694 parser.add_argument('--breakpad_root', type='path', default=None,
695 help='root directory for breakpad symbols')
696 parser.add_argument('--official_build', action='store_true', default=False,
697 help='point to official symbol server')
698 parser.add_argument('--regenerate', action='store_true', default=False,
699 help='regenerate all symbols')
Mike Frysinger8ec8c502014-02-10 00:19:13 -0500700 parser.add_argument('--upload-limit', type=int, default=None,
Mike Frysingerd5fcb3a2013-05-30 21:10:50 -0400701 help='only upload # number of symbols')
702 parser.add_argument('--strip_cfi', type=int,
703 default=CRASH_SERVER_FILE_LIMIT - (10 * 1024 * 1024),
704 help='strip CFI data for files above this size')
Mike Frysinger7f9be142014-01-15 02:16:42 -0500705 parser.add_argument('--failed-list', type='path',
706 help='where to save a list of failed symbols')
Mike Frysinger0c0efa22014-02-09 23:32:23 -0500707 parser.add_argument('--dedupe', action='store_true', default=False,
708 help='use the swarming service to avoid re-uploading')
Mike Frysingerd5fcb3a2013-05-30 21:10:50 -0400709 parser.add_argument('--testing', action='store_true', default=False,
710 help='run in testing mode')
711 parser.add_argument('--yes', action='store_true', default=False,
712 help='answer yes to all prompts')
713
714 opts = parser.parse_args(argv)
Mike Frysinger90e49ca2014-01-14 14:42:07 -0500715 opts.Freeze()
Mike Frysingerd5fcb3a2013-05-30 21:10:50 -0400716
Mike Frysinger9b2ff5c2013-11-22 10:01:12 -0500717 if opts.sym_paths:
Mike Frysinger9dcf9ae2013-08-10 15:17:09 -0400718 if opts.regenerate:
719 cros_build_lib.Die('--regenerate may not be used with specific files')
720 else:
721 if opts.board is None:
722 cros_build_lib.Die('--board is required')
Mike Frysingerd5fcb3a2013-05-30 21:10:50 -0400723
724 if opts.breakpad_root and opts.regenerate:
725 cros_build_lib.Die('--regenerate may not be used with --breakpad_root')
726
727 if opts.testing:
728 # TODO(build): Kill off --testing mode once unittests are up-to-snuff.
729 cros_build_lib.Info('running in testing mode')
730 # pylint: disable=W0601,W0603
731 global INITIAL_RETRY_DELAY, SymUpload, DEFAULT_SLEEP_DELAY
732 INITIAL_RETRY_DELAY = DEFAULT_SLEEP_DELAY = 0
733 SymUpload = TestingSymUpload
734
Mike Frysinger0c0efa22014-02-09 23:32:23 -0500735 dedupe_namespace = None
736 if opts.dedupe:
737 if opts.official_build and not opts.testing:
738 dedupe_namespace = OFFICIAL_DEDUPE_NAMESPACE
739 else:
740 dedupe_namespace = STAGING_DEDUPE_NAMESPACE
741
Mike Frysingerd5fcb3a2013-05-30 21:10:50 -0400742 if not opts.yes:
Mike Frysingerc5de9602014-02-09 02:42:36 -0500743 prolog = '\n'.join(textwrap.wrap(textwrap.dedent("""
Mike Frysingerd5fcb3a2013-05-30 21:10:50 -0400744 Uploading symbols for an entire Chromium OS build is really only
745 necessary for release builds and in a few cases for developers
746 to debug problems. It will take considerable time to run. For
747 developer debugging purposes, consider instead passing specific
748 files to upload.
Mike Frysingerc5de9602014-02-09 02:42:36 -0500749 """), 80)).strip()
Mike Frysingerd5fcb3a2013-05-30 21:10:50 -0400750 if not cros_build_lib.BooleanPrompt(
751 prompt='Are you sure you want to upload all build symbols',
Mike Frysingerc5de9602014-02-09 02:42:36 -0500752 default=False, prolog=prolog):
Mike Frysingerd5fcb3a2013-05-30 21:10:50 -0400753 cros_build_lib.Die('better safe than sorry')
754
755 ret = 0
756 if opts.regenerate:
Mike Frysinger69cb41d2013-08-11 20:08:19 -0400757 ret += cros_generate_breakpad_symbols.GenerateBreakpadSymbols(
758 opts.board, breakpad_dir=opts.breakpad_root)
Mike Frysingerd5fcb3a2013-05-30 21:10:50 -0400759
760 ret += UploadSymbols(opts.board, official=opts.official_build,
761 breakpad_dir=opts.breakpad_root,
762 file_limit=opts.strip_cfi, sleep=DEFAULT_SLEEP_DELAY,
Mike Frysinger8ec8c502014-02-10 00:19:13 -0500763 upload_limit=opts.upload_limit, sym_paths=opts.sym_paths,
Mike Frysinger0c0efa22014-02-09 23:32:23 -0500764 failed_list=opts.failed_list,
765 dedupe_namespace=dedupe_namespace)
Mike Frysingerd5fcb3a2013-05-30 21:10:50 -0400766 if ret:
767 cros_build_lib.Error('encountered %i problem(s)', ret)
768 # Since exit(status) gets masked, clamp it to 1 so we don't inadvertently
769 # return 0 in case we are a multiple of the mask.
770 ret = 1
771
772 return ret
Mike Frysinger094a2172013-08-14 12:54:35 -0400773
774
775# We need this to run once per process. Do it at module import time as that
776# will let us avoid doing it inline at function call time (see SymUpload) as
777# that func might be called by the multiprocessing module which means we'll
778# do the opener logic multiple times overall. Plus, if you're importing this
779# module, it's a pretty good chance that you're going to need this.
780poster.streaminghttp.register_openers()