Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(184)

Unified Diff: tools/telemetry/telemetry/core/platform/power_monitor/ippet_power_monitor.py

Issue 504333003: [telemetry] Use SetEvent signals instead of web server for IPPET power monitor. (Closed) Base URL: https://chromium.googlesource.com/chromium/src.git@master
Patch Set: Created 6 years, 4 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View side-by-side diff with in-line comments
Download patch
« no previous file with comments | « no previous file | no next file » | no next file with comments »
Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
Index: tools/telemetry/telemetry/core/platform/power_monitor/ippet_power_monitor.py
diff --git a/tools/telemetry/telemetry/core/platform/power_monitor/ippet_power_monitor.py b/tools/telemetry/telemetry/core/platform/power_monitor/ippet_power_monitor.py
index f92f485bdc04cf5feb27335dfd96c80b0496b5a8..d6131178860659f9e8eb32f105904ec227fda42b 100644
--- a/tools/telemetry/telemetry/core/platform/power_monitor/ippet_power_monitor.py
+++ b/tools/telemetry/telemetry/core/platform/power_monitor/ippet_power_monitor.py
@@ -11,11 +11,9 @@ import platform
import re
import shutil
import tempfile
-import urllib2
import zipfile
from telemetry import decorators
-from telemetry.core import util
from telemetry.core.platform import platform_backend
from telemetry.core.platform import power_monitor
from telemetry.util import cloud_storage
@@ -32,6 +30,10 @@ except ImportError:
win32process = None
+START_EVENT = 'ippet_StartEvent'
+QUIT_EVENT = 'ippet_QuitEvent'
+
+
class IppetError(Exception):
pass
@@ -68,7 +70,6 @@ class IppetPowerMonitor(power_monitor.PowerMonitor):
super(IppetPowerMonitor, self).__init__()
self._backend = backend
self._ippet_handle = None
- self._ippet_port = None
self._output_dir = None
def CanMonitorPower(self):
@@ -108,53 +109,63 @@ class IppetPowerMonitor(power_monitor.PowerMonitor):
def StartMonitoringPower(self, browser):
assert not self._ippet_handle, 'Called StartMonitoringPower() twice.'
self._output_dir = tempfile.mkdtemp()
- self._ippet_port = util.GetUnreservedAvailableLocalPort()
parameters = ['-log_dir', self._output_dir,
- '-web_port', str(self._ippet_port),
- '-zip', 'n']
- self._ippet_handle = self._backend.LaunchApplication(
- IppetPath(), parameters, elevate_privilege=True)
+ '-signals', 'START,QUIT',
+ '-enable_web', 'n', '-zip', 'n']
- def IppetServerIsUp():
- try:
- urllib2.urlopen('http://127.0.0.1:%d/ippet' % self._ippet_port,
- timeout=1).close()
- except urllib2.URLError:
- return False
- return True
- util.WaitFor(IppetServerIsUp, timeout=5)
+ try:
+ with contextlib.closing(win32event.CreateEvent(
+ None, True, False, START_EVENT)) as start_event:
+ self._ippet_handle = self._backend.LaunchApplication(
+ IppetPath(), parameters, elevate_privilege=True)
+ wait_code = win32event.WaitForSingleObject(start_event, 5000)
+ if wait_code != win32event.WAIT_OBJECT_0:
+ if wait_code == win32event.WAIT_TIMEOUT:
+ raise IppetError('Timed out waiting for IPPET to start.')
+ else:
+ raise IppetError('Error code %d while waiting for IPPET to start.' %
+ wait_code)
+
+ except: # In case of emergency, don't leave IPPET processes hanging around.
+ if self._ippet_handle:
+ try:
+ exit_code = win32process.GetExitCodeProcess(self._ippet_handle)
+ if exit_code == win32con.STILL_ACTIVE:
+ win32process.TerminateProcess(self._ippet_handle, 0)
+ finally:
+ self._ippet_handle.Close()
+ self._ippet_handle = None
+ raise
def StopMonitoringPower(self):
assert self._ippet_handle, (
'Called StopMonitoringPower() before StartMonitoringPower().')
- # Stop IPPET.
try:
- ippet_quit_url = 'http://127.0.0.1:%d/ippet?cmd=quit' % self._ippet_port
- with contextlib.closing(
- urllib2.urlopen(ippet_quit_url, timeout=5)) as response:
- quit_output = response.read()
- if quit_output != 'quiting\r\n':
- raise IppetError('Failed to quit IPPET: %s' % quit_output.strip())
- wait_return_code = win32event.WaitForSingleObject(self._ippet_handle,
- 20000)
- if wait_return_code != win32event.WAIT_OBJECT_0:
- if wait_return_code == win32event.WAIT_TIMEOUT:
+ # Stop IPPET.
+ with contextlib.closing(win32event.OpenEvent(
+ win32event.EVENT_MODIFY_STATE, False, QUIT_EVENT)) as quit_event:
+ win32event.SetEvent(quit_event)
+
+ # Wait for IPPET process to exit.
+ wait_code = win32event.WaitForSingleObject(self._ippet_handle, 20000)
+ if wait_code != win32event.WAIT_OBJECT_0:
+ if wait_code == win32event.WAIT_TIMEOUT:
raise IppetError('Timed out waiting for IPPET to close.')
else:
raise IppetError('Error code %d while waiting for IPPET to close.' %
- wait_return_code)
- finally:
+ wait_code)
+
+ finally: # If we need to, forcefully kill IPPET.
try:
- ippet_exit_code = win32process.GetExitCodeProcess(self._ippet_handle)
- if ippet_exit_code == win32con.STILL_ACTIVE:
- win32process.TerminateProcess(self._ippet_handle)
+ exit_code = win32process.GetExitCodeProcess(self._ippet_handle)
+ if exit_code == win32con.STILL_ACTIVE:
+ win32process.TerminateProcess(self._ippet_handle, 0)
raise IppetError('IPPET is still running but should have stopped.')
- elif ippet_exit_code != 0:
- raise IppetError('IPPET closed with exit code %d.' % ippet_exit_code)
+ elif exit_code != 0:
+ raise IppetError('IPPET closed with exit code %d.' % exit_code)
finally:
self._ippet_handle.Close()
self._ippet_handle = None
- self._ippet_port = None
# Read IPPET's log file.
log_file = os.path.join(self._output_dir, 'ippet_log_processes.xls')
« no previous file with comments | « no previous file | no next file » | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698