mirror of
https://github.com/ARM-software/devlib.git
synced 2025-02-27 15:07:50 +00:00
390a544a92
Prior to this commit, measurements to be collected were specified via "sites" and "kinds" parameters. This has the limitation that if you wanted measurments of kind X from site A and kind Y from site B, you'd have to specify them as reset(sites=['A', 'B'], kinds=['X', 'Y']) Which would have the effect of also collecting measurments Y for site A and measurments X for site B. This commit adds the option of specifying individual channels, via thier labels, e.g. reset(channels=['A_X', 'B_Y']) so that only the channels you're interested in will be collected.
136 lines
5.1 KiB
Python
136 lines
5.1 KiB
Python
import os
|
|
import re
|
|
import csv
|
|
import tempfile
|
|
from datetime import datetime
|
|
from collections import defaultdict
|
|
from itertools import izip_longest
|
|
|
|
from devlib.instrument import Instrument, MeasurementsCsv, CONTINUOUS
|
|
from devlib.exception import TargetError, HostError
|
|
from devlib.utils.android import ApkInfo
|
|
|
|
|
|
THIS_DIR = os.path.dirname(__file__)
|
|
|
|
NETSTAT_REGEX = re.compile(r'I/(?P<tag>netstats-\d+)\(\s*\d*\): (?P<ts>\d+) '
|
|
r'"(?P<package>[^"]+)" TX: (?P<tx>\S+) RX: (?P<rx>\S+)')
|
|
|
|
|
|
def extract_netstats(filepath, tag=None):
|
|
netstats = []
|
|
with open(filepath) as fh:
|
|
for line in fh:
|
|
match = NETSTAT_REGEX.search(line)
|
|
if not match:
|
|
continue
|
|
if tag and match.group('tag') != tag:
|
|
continue
|
|
netstats.append((match.group('tag'),
|
|
match.group('ts'),
|
|
match.group('package'),
|
|
match.group('tx'),
|
|
match.group('rx')))
|
|
return netstats
|
|
|
|
|
|
def netstats_to_measurements(netstats):
|
|
measurements = defaultdict(list)
|
|
for row in netstats:
|
|
tag, ts, package, tx, rx = row # pylint: disable=unused-variable
|
|
measurements[package + '_tx'].append(tx)
|
|
measurements[package + '_rx'].append(rx)
|
|
return measurements
|
|
|
|
|
|
def write_measurements_csv(measurements, filepath):
|
|
headers = sorted(measurements.keys())
|
|
columns = [measurements[h] for h in headers]
|
|
with open(filepath, 'wb') as wfh:
|
|
writer = csv.writer(wfh)
|
|
writer.writerow(headers)
|
|
writer.writerows(izip_longest(*columns))
|
|
|
|
|
|
class NetstatsInstrument(Instrument):
|
|
|
|
mode = CONTINUOUS
|
|
|
|
def __init__(self, target, apk=None, service='.TrafficMetricsService'):
|
|
"""
|
|
Additional paramerter:
|
|
|
|
:apk: Path to the APK file that contains ``com.arm.devlab.netstats``
|
|
package. If not specified, it will be assumed that an APK with
|
|
name "netstats.apk" is located in the same directory as the
|
|
Python module for the instrument.
|
|
:service: Name of the service to be launched. This service must be
|
|
present in the APK.
|
|
|
|
"""
|
|
if target.os != 'android':
|
|
raise TargetError('netstats insturment only supports Android targets')
|
|
if apk is None:
|
|
apk = os.path.join(THIS_DIR, 'netstats.apk')
|
|
if not os.path.isfile(apk):
|
|
raise HostError('APK for netstats instrument does not exist ({})'.format(apk))
|
|
super(NetstatsInstrument, self).__init__(target)
|
|
self.apk = apk
|
|
self.package = ApkInfo(self.apk).package
|
|
self.service = service
|
|
self.tag = None
|
|
self.command = None
|
|
self.stop_command = 'am kill {}'.format(self.package)
|
|
|
|
for package in self.target.list_packages():
|
|
self.add_channel(package, 'tx')
|
|
self.add_channel(package, 'rx')
|
|
|
|
def setup(self, force=False, *args, **kwargs):
|
|
if self.target.package_is_installed(self.package):
|
|
if force:
|
|
self.logger.debug('Re-installing {} (forced)'.format(self.package))
|
|
self.target.uninstall_package(self.package)
|
|
self.target.install(self.apk)
|
|
else:
|
|
self.logger.debug('{} already present on target'.format(self.package))
|
|
else:
|
|
self.logger.debug('Deploying {} to target'.format(self.package))
|
|
self.target.install(self.apk)
|
|
|
|
def reset(self, sites=None, kinds=None, channels=None, period=None): # pylint: disable=arguments-differ
|
|
super(NetstatsInstrument, self).reset(sites, kinds, channels)
|
|
period_arg, packages_arg = '', ''
|
|
self.tag = 'netstats-{}'.format(datetime.now().strftime('%Y%m%d%H%M%s'))
|
|
tag_arg = ' --es tag {}'.format(self.tag)
|
|
if sites:
|
|
packages_arg = ' --es packages {}'.format(','.join(sites))
|
|
if period:
|
|
period_arg = ' --ei period {}'.format(period)
|
|
self.command = 'am startservice{}{}{} {}/{}'.format(tag_arg,
|
|
period_arg,
|
|
packages_arg,
|
|
self.package,
|
|
self.service)
|
|
self.target.execute(self.stop_command) # ensure the service is not running.
|
|
|
|
def start(self):
|
|
if self.command is None:
|
|
raise RuntimeError('reset() must be called before start()')
|
|
self.target.execute(self.command)
|
|
|
|
def stop(self):
|
|
self.target.execute(self.stop_command)
|
|
|
|
def get_data(self, outfile):
|
|
raw_log_file = tempfile.mktemp()
|
|
self.target.dump_logcat(raw_log_file)
|
|
data = extract_netstats(raw_log_file)
|
|
measurements = netstats_to_measurements(data)
|
|
write_measurements_csv(measurements, outfile)
|
|
os.remove(raw_log_file)
|
|
return MeasurementsCsv(outfile, self.active_channels)
|
|
|
|
def teardown(self):
|
|
self.target.uninstall_package(self.package)
|