You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
85 lines
2.8 KiB
Python
85 lines
2.8 KiB
Python
#!/usr/bin/env python3
|
|
# This Source Code Form is subject to the terms of the Mozilla Public
|
|
# License, v. 2.0. If a copy of the MPL was not distributed with this
|
|
# file, You can obtain one at https://mozilla.org/MPL/2.0/.
|
|
|
|
"""
|
|
A script to set up startup profiling with the Firefox Profiler. See
|
|
https://profiler.firefox.com/docs/#/./guide-remote-profiling?id=startup-profiling
|
|
for more information.
|
|
"""
|
|
|
|
import os
|
|
import sys
|
|
import tempfile
|
|
from subprocess import run
|
|
|
|
SCRIPT_NAME=os.path.basename(__file__)
|
|
|
|
PATH_PREFIX='/data/local/tmp'
|
|
|
|
GV_CONFIG=b'''env:
|
|
MOZ_PROFILER_STARTUP: 1
|
|
MOZ_PROFILER_STARTUP_INTERVAL: 5
|
|
MOZ_PROFILER_STARTUP_FEATURES: threads,js,stackwalk,leaf,screenshots,ipcmessages,java,cpu
|
|
MOZ_PROFILER_STARTUP_FILTERS: GeckoMain,Compositor,Renderer,IPDL Background
|
|
'''
|
|
|
|
def print_usage_and_exit():
|
|
print('USAGE: ./{} [activate|deactivate] [nightly|beta|release|debug]'.format(SCRIPT_NAME), file=sys.stderr)
|
|
print('example: ./{} activate nightly'.format(SCRIPT_NAME), file=sys.stderr)
|
|
sys.exit(1)
|
|
|
|
def push(id, filename):
|
|
config = tempfile.NamedTemporaryFile(delete=False)
|
|
try:
|
|
# I think the file needs to be closed to save its contents for adb push to
|
|
# work correctly so we close it here and later delete it manually.
|
|
with config.file as f:
|
|
f.write(GV_CONFIG)
|
|
|
|
print('Pushing {} to device.'.format(filename))
|
|
run(['adb', 'push', config.name, os.path.join(PATH_PREFIX, filename)])
|
|
run(['adb', 'shell', 'am', 'set-debug-app', '--persistent', id])
|
|
print('\nStartup profiling enabled on all future start ups, possibly even after reinstall.')
|
|
print('Call script with `deactivate` to disable it.')
|
|
print('DON\'T FORGET TO ENABLE \'Remote debugging via USB\' IN THE APP SETTINGS!')
|
|
finally:
|
|
os.remove(config.name)
|
|
|
|
def remove(filename):
|
|
print('Removing {} from device.'.format(filename))
|
|
run(['adb', 'shell', 'rm', PATH_PREFIX + '/' + filename])
|
|
run(['adb', 'shell', 'am', 'clear-debug-app'])
|
|
|
|
def convert_channel_to_id(channel):
|
|
# Users might want to use custom app IDs in the future
|
|
# but we don't know that's a need yet: let's add it if it is.
|
|
mapping = {
|
|
'release': 'org.mozilla.firefox',
|
|
'beta': 'org.mozilla.firefox_beta',
|
|
'nightly': 'org.mozilla.fenix',
|
|
'debug': 'org.mozilla.fenix.debug'
|
|
}
|
|
return mapping[channel]
|
|
|
|
def main():
|
|
try:
|
|
cmd = sys.argv[1]
|
|
channel = sys.argv[2]
|
|
id = convert_channel_to_id(channel)
|
|
except (IndexError, KeyError) as e:
|
|
print_usage_and_exit()
|
|
|
|
filename = id + '-geckoview-config.yaml'
|
|
|
|
if cmd == 'activate':
|
|
push(id, filename)
|
|
elif cmd == 'deactivate':
|
|
remove(filename)
|
|
else:
|
|
print_usage_and_exit()
|
|
|
|
if __name__ == '__main__':
|
|
main()
|