diff options
Diffstat (limited to 'scripts/update_device.py')
-rwxr-xr-x | scripts/update_device.py | 10 |
1 files changed, 6 insertions, 4 deletions
diff --git a/scripts/update_device.py b/scripts/update_device.py index 49de7a15..64cfbe3b 100755 --- a/scripts/update_device.py +++ b/scripts/update_device.py @@ -276,14 +276,13 @@ def StartServer(ota_filename, serving_range): return t -def AndroidUpdateCommand(ota_filename, payload_url): +def AndroidUpdateCommand(ota_filename, payload_url, extra_headers): """Return the command to run to start the update in the Android device.""" ota = AndroidOTAPackage(ota_filename) headers = ota.properties headers += 'USER_AGENT=Dalvik (something, something)\n' - - # headers += 'POWERWASH=1\n' headers += 'NETWORK_ID=0\n' + headers += extra_headers return ['update_engine_client', '--update', '--follow', '--payload=%s' % payload_url, '--offset=%d' % ota.offset, @@ -360,6 +359,8 @@ def main(): help='Less verbose output') parser.add_argument('--public-key', type=str, default='', help='Override the public key used to verify payload.') + parser.add_argument('--extra-headers', type=str, default='', + help='Extra headers to pass to the device.') args = parser.parse_args() logging.basicConfig( level=logging.WARNING if args.no_verbose else logging.INFO) @@ -423,7 +424,8 @@ def main(): update_cmd = \ OmahaUpdateCommand('http://127.0.0.1:%d/update' % DEVICE_PORT) else: - update_cmd = AndroidUpdateCommand(args.otafile, payload_url) + update_cmd = \ + AndroidUpdateCommand(args.otafile, payload_url, args.extra_headers) cmds.append(['shell', 'su', '0'] + update_cmd) for cmd in cmds: |