diff options
Diffstat (limited to 'chrome/test/pyautolib/mock_pref_pane.py')
| -rw-r--r-- | chrome/test/pyautolib/mock_pref_pane.py | 124 |
1 files changed, 0 insertions, 124 deletions
diff --git a/chrome/test/pyautolib/mock_pref_pane.py b/chrome/test/pyautolib/mock_pref_pane.py deleted file mode 100644 index cbb3a48..0000000 --- a/chrome/test/pyautolib/mock_pref_pane.py +++ /dev/null @@ -1,124 +0,0 @@ -# Copyright (c) 2012 The Chromium Authors. All rights reserved. -# Use of this source code is governed by a BSD-style license that can be -# found in the LICENSE file. - -"""Mock pref pane for testing purpose on Mac.""" - -import Foundation -import os -import signal -import subprocess -import sys -import tempfile -import time - - -class MockPrefPane(object): - """Mock Pref Pane to enable/disable/changepin without system prompt. - - This only applies to Mac. - """ - - def __init__(self): - self._service_name = 'org.chromium.chromoting' - self._real_user_id = os.getuid() - self._config_file = os.path.join(tempfile.gettempdir(), - '%s.json' % self._service_name) - self._tool_script = '/Library/PrivilegedHelperTools/%s.me2me.sh' % \ - self._service_name - - def _GetJobPid(self): - """Gets the org.chromium.chromoting job id.""" - process = subprocess.Popen(['launchctl', 'list'], stdout=subprocess.PIPE) - pid = None - for line in process.stdout: - # Format is: - # 12345 - my.job (if my.job is running, number is job's PID) - # - 0 my.other.job (if my.other.job is not running) - fields = line.strip().split('\t') - if fields[2] == self._service_name and fields[0] != "-": - pid = fields[0] - break - process.wait() - return pid - - def Enable(self): - """Handles what pref pane does for enabling connection.""" - # Elevate privileges, otherwise tool_script executes with EUID != 0. - os.setuid(0) - subprocess.call([self._tool_script, '--enable'], - stdin=open(self._config_file)) - - # Drop privileges, start the launchd job as the logged-in user. - os.setuid(self._real_user_id) - subprocess.call(['launchctl', 'start', self._service_name]) - - # Starting a launchd job is an asynchronous operation that typically takes - # a couple of seconds, so poll until the job has started. - for _ in range(1, 10): - if self._GetJobPid(): - print '*** org.chromium.chromoting is running ***' - break - time.sleep(2) - - def Disable(self): - """Handles what pref pane does for disabling connection.""" - # Elevate privileges, otherwise tool_script executes with EUID != 0. - os.setuid(0) - subprocess.call([self._tool_script, '--disable'], - stdin=open(self._config_file)) - - # Drop privileges, stop the launchd job as the logged-in user. - os.setuid(self._real_user_id) - subprocess.call(['launchctl', 'stop', self._service_name]) - - # Stopping a launchd job is an asynchronous operation that typically takes - # a couple of seconds, so poll until the job has stopped. - for _ in range(1, 10): - if not self._GetJobPid(): - print '*** org.chromium.chromoting is not running ***' - break - time.sleep(2) - - def ChangePin(self): - """Handles what pref pane does for changing pin.""" - # Elevate privileges, otherwise tool_script executes with EUID != 0. - os.setuid(0) - subprocess.call([self._tool_script, '--save-config'], - stdin=open(self._config_file)) - - # Drop privileges and send SIGHUP to org.chromium.chromoting - os.setuid(self._real_user_id) - os.kill(int(self._GetJobPid()), signal.SIGHUP) - - def NotifyWebapp(self): - """Notifies the web app that pref pane operation is done.""" - notif_center = Foundation.NSDistributedNotificationCenter.defaultCenter() - notif_center.postNotificationName_object_userInfo_( - self._service_name + '.update_succeeded', None, None) - - -def Main(): - """Handles the mock pref pane actions.""" - assert sys.platform.startswith('darwin') - - print '*** Started mock pref pane ***' - print '*** EUID=%d, UID=%d ***' % (os.geteuid(), os.getuid()) - - pref_pane = MockPrefPane() - - if sys.argv[1] == 'enable': - pref_pane.Enable() - elif sys.argv[1] == 'disable': - pref_pane.Disable() - elif sys.argv[1] == 'changepin': - pref_pane.ChangePin() - else: - print >>sys.stderr, 'Invalid syntax' - return - - pref_pane.NotifyWebapp() - - -if __name__ == '__main__': - Main()
\ No newline at end of file |