| #!/usr/bin/env vpython3 |
| # Copyright 2015 The Chromium Authors |
| # Use of this source code is governed by a BSD-style license that can be |
| # found in the LICENSE file. |
| |
| """Utility for reading / writing command-line flag files on device(s).""" |
| |
| |
| import argparse |
| import logging |
| import sys |
| |
| import devil_chromium |
| |
| from devil.android import device_errors |
| from devil.android import device_utils |
| from devil.android import flag_changer |
| from devil.android.tools import script_common |
| from devil.utils import cmd_helper |
| from devil.utils import logging_common |
| |
| |
| def CheckBuildTypeSupportsFlags(device, command_line_flags_file): |
| is_webview = command_line_flags_file == 'webview-command-line' |
| if device.IsUserBuild() and is_webview: |
| raise device_errors.CommandFailedError( |
| 'WebView only respects flags on a userdebug or eng device, yours ' |
| 'is a user build.', device) |
| if device.IsUserBuild(): |
| logging.warning( |
| 'Your device (%s) is a user build; Chrome may or may not pick up ' |
| 'your commandline flags. Check your ' |
| '"command_line_on_non_rooted_enabled" preference, or switch ' |
| 'devices.', device) |
| |
| |
| def main(): |
| parser = argparse.ArgumentParser(description=__doc__) |
| parser.usage = '''%(prog)s --name FILENAME [--device SERIAL] [flags...] |
| |
| No flags: Prints existing command-line file. |
| Empty string: Deletes command-line file. |
| Otherwise: Writes command-line file. |
| |
| ''' |
| parser.add_argument('--name', required=True, |
| help='Name of file where to store flags on the device.') |
| parser.add_argument('-e', '--executable', dest='executable', default='chrome', |
| help='(deprecated) No longer used.') |
| script_common.AddEnvironmentArguments(parser) |
| script_common.AddDeviceArguments(parser) |
| logging_common.AddLoggingArguments(parser) |
| |
| args, remote_args = parser.parse_known_args() |
| devil_chromium.Initialize(adb_path=args.adb_path) |
| logging_common.InitializeLogging(args) |
| |
| devices = device_utils.DeviceUtils.HealthyDevices(device_arg=args.devices, |
| default_retries=0) |
| all_devices = device_utils.DeviceUtils.parallel(devices) |
| |
| if not remote_args: |
| # No args == do not update, just print flags. |
| remote_args = None |
| action = '' |
| elif len(remote_args) == 1 and not remote_args[0]: |
| # Single empty string arg == delete flags |
| remote_args = [] |
| action = 'Deleted command line file. ' |
| else: |
| if remote_args[0] == '--': |
| remote_args.pop(0) |
| action = 'Wrote command line file. ' |
| |
| def update_flags(device): |
| CheckBuildTypeSupportsFlags(device, args.name) |
| changer = flag_changer.FlagChanger(device, args.name) |
| if remote_args is not None: |
| flags = changer.ReplaceFlags(remote_args) |
| else: |
| flags = changer.GetCurrentFlags() |
| return (device, device.build_description, flags) |
| |
| updated_values = all_devices.pMap(update_flags).pGet(None) |
| |
| print('%sCurrent flags (in %s):' % (action, args.name)) |
| for d, desc, flags in updated_values: |
| if flags: |
| # Shell-quote flags for easy copy/paste as new args on the terminal. |
| quoted_flags = ' '.join(cmd_helper.SingleQuote(f) for f in sorted(flags)) |
| else: |
| quoted_flags = '( empty )' |
| print(' %s (%s): %s' % (d, desc, quoted_flags)) |
| |
| return 0 |
| |
| |
| if __name__ == '__main__': |
| sys.exit(main()) |