# Copyright 2016 Google Inc. # # Use of this source code is governed by a BSD-style license that can be # found in the LICENSE file. from __future__ import print_function from _hardware import Hardware import sys import time class HardwareAndroid(Hardware): def __init__(self, adb): Hardware.__init__(self) self.warmup_time = 5 self._adb = adb self._is_root = self._adb.attempt_root() if self._is_root: self._adb.remount() self._initial_airplane_mode = None self._initial_location_providers = None self._initial_ASLR = None def __enter__(self): # turn on airplane mode. self._initial_airplane_mode = \ self._adb.check('settings get global airplane_mode_on') self._adb.shell('settings put global airplane_mode_on 1') # disable GPS. self._initial_location_providers = \ self._adb.check('settings get secure location_providers_allowed') self._initial_location_providers = \ self._initial_location_providers.replace(',', ' ') self._adb.shell('''\ for PROVIDER in %s; do settings put secure location_providers_allowed -$PROVIDER done''' % self._initial_location_providers) if self._is_root: # disable bluetooth, wifi, and mobile data. # TODO: can we query these initial values? self._adb.shell('''\ service call bluetooth_manager 8 && svc wifi disable && svc data disable''') # kill the gui. self._adb.shell('''\ setprop ctl.stop media && setprop ctl.stop zygote && setprop ctl.stop surfaceflinger && setprop ctl.stop drm''') # disable ASLR. self._initial_ASLR = \ self._adb.check('cat /proc/sys/kernel/randomize_va_space') self._adb.shell('echo 0 > /proc/sys/kernel/randomize_va_space') else: print("WARNING: no adb root access; results may be unreliable.", file=sys.stderr) return Hardware.__enter__(self) def __exit__(self, exception_type, exception_value, traceback): Hardware.__exit__(self, exception_type, exception_value, traceback) if self._is_root: # restore ASLR. self._adb.shell('echo %s > /proc/sys/kernel/randomize_va_space' % self._initial_ASLR) # revive the gui. self._adb.shell('''\ setprop ctl.start drm && setprop ctl.start surfaceflinger && setprop ctl.start zygote && setprop ctl.start media''') else: # restore GPS (doesn't seem to work if we killed the gui). self._adb.shell('''\ for PROVIDER in %s; do settings put secure location_providers_allowed +$PROVIDER done''' % self._initial_location_providers) # restore airplane mode (doesn't seem to work if we killed the gui). self._adb.shell('settings put global airplane_mode_on %s' % self._initial_airplane_mode) def sanity_check(self): Hardware.sanity_check(self) def print_debug_diagnostics(self): # search for and print thermal trip points that may have been exceeded. self._adb.shell('''\ THERMALDIR=/sys/class/thermal if [ ! -d $THERMALDIR ]; then exit fi for ZONE in $(cd $THERMALDIR; echo thermal_zone*); do cd $THERMALDIR/$ZONE if [ ! -e mode ] || grep -Fxqv enabled mode || [ ! -e trip_point_0_temp ]; then continue fi TEMP=$(cat temp) TRIPPOINT=trip_point_0_temp if [ $TEMP -le $(cat $TRIPPOINT) ]; then echo "$ZONE ($(cat type)): temp=$TEMP <= $TRIPPOINT=$(cat $TRIPPOINT)" 1>&2 else let i=1 while [ -e trip_point_${i}_temp ] && [ $TEMP -gt $(cat trip_point_${i}_temp) ]; do TRIPPOINT=trip_point_${i}_temp let i=i+1 done echo "$ZONE ($(cat type)): temp=$TEMP > $TRIPPOINT=$(cat $TRIPPOINT)" 1>&2 fi done''') Hardware.print_debug_diagnostics(self) def sleep(self, sleeptime): Hardware.sleep(self, sleeptime)