2
0
mirror of https://github.com/fork-maintainers/iceraven-browser synced 2024-11-15 18:12:54 +00:00
iceraven-browser/tools/setup-startup-profiling.py
Michael Comella 4657afc4ba No issue: switch subcommand names to deactivate/activate.
These make more sense than push/remove if you don't understand the
underlying mechanism.
2021-01-14 10:25:28 -08:00

83 lines
2.7 KiB
Python
Executable File

#!/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
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('Startup profiling enabled on all future start ups, possibly even after reinstall. Call script with `deactivate` to disable it.')
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()