| OLD | NEW |
| 1 # Copyright (c) 2010 The Chromium OS Authors. All rights reserved. | 1 # Copyright (c) 2010 The Chromium OS Authors. All rights reserved. |
| 2 # Use of this source code is governed by a BSD-style license that can be | 2 # Use of this source code is governed by a BSD-style license that can be |
| 3 # found in the LICENSE file. | 3 # found in the LICENSE file. |
| 4 | 4 |
| 5 import logging, os, platform, time | 5 import logging, os, platform, re, tempfile, time |
| 6 from autotest_lib.client.common_lib import error | 6 from autotest_lib.client.common_lib import error |
| 7 from autotest_lib.client.common_lib import utils |
| 7 | 8 |
| 8 | 9 |
| 9 class TimeoutError(error.TestError): | 10 class TimeoutError(error.TestError): |
| 10 """Error raised when we time out when waiting on a condition.""" | 11 """Error raised when we time out when waiting on a condition.""" |
| 11 | 12 |
| 12 | 13 |
| 14 class Crossystem(object): |
| 15 """A wrapper for the crossystem utility.""" |
| 16 |
| 17 def __init__(self, client): |
| 18 self.cros_system_data = {} |
| 19 self._client = client |
| 20 |
| 21 def init(self): |
| 22 self.cros_system_data = {} |
| 23 (_, fname) = tempfile.mkstemp() |
| 24 f = open(fname, 'w') |
| 25 self._client.run('crossystem', stdout_tee=f) |
| 26 f.close() |
| 27 text = utils.read_file(fname) |
| 28 for line in text.splitlines(): |
| 29 assignment_string = line.split('#')[0] |
| 30 if not assignment_string.count('='): |
| 31 continue |
| 32 (name, value) = assignment_string.split('=', 1) |
| 33 self.cros_system_data[name.strip()] = value.strip() |
| 34 os.remove(fname) |
| 35 |
| 36 def __getattr__(self, name): |
| 37 """ |
| 38 Retrieve a crosssystem attribute. |
| 39 |
| 40 The call crossystemobject.name() will return the crossystem reported |
| 41 string. |
| 42 """ |
| 43 return lambda : self.cros_system_data[name] |
| 44 |
| 45 |
| 13 def poll_for_condition( | 46 def poll_for_condition( |
| 14 condition, exception=None, timeout=10, sleep_interval=0.1, desc=None): | 47 condition, exception=None, timeout=10, sleep_interval=0.1, desc=None): |
| 15 """Poll until a condition becomes true. | 48 """Poll until a condition becomes true. |
| 16 | 49 |
| 17 condition: function taking no args and returning bool | 50 condition: function taking no args and returning bool |
| 18 exception: exception to throw if condition doesn't become true | 51 exception: exception to throw if condition doesn't become true |
| 19 timeout: maximum number of seconds to wait | 52 timeout: maximum number of seconds to wait |
| 20 sleep_interval: time to sleep between polls | 53 sleep_interval: time to sleep between polls |
| 21 desc: description of default TimeoutError used if 'exception' is None | 54 desc: description of default TimeoutError used if 'exception' is None |
| 22 | 55 |
| (...skipping 47 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 70 which do not appear in the whitelist. | 103 which do not appear in the whitelist. |
| 71 | 104 |
| 72 Arguments: | 105 Arguments: |
| 73 dmesg - string containing raw dmesg buffer | 106 dmesg - string containing raw dmesg buffer |
| 74 message_level - minimum message priority to check | 107 message_level - minimum message priority to check |
| 75 whitelist - messages to ignore | 108 whitelist - messages to ignore |
| 76 | 109 |
| 77 Returns: | 110 Returns: |
| 78 List of unexpected warnings | 111 List of unexpected warnings |
| 79 """ | 112 """ |
| 80 | 113 whitelist_re = re.compile(r'(%s)' % '|'.join(whitelist)) |
| 81 unexpected = [] | 114 unexpected = [] |
| 82 for line in dmesg.splitlines(): | 115 for line in dmesg.splitlines(): |
| 83 if int(line[1]) <= message_level: | 116 if int(line[1]) <= message_level: |
| 84 if not 'used greatest stack depth' in line: | 117 stripped_line = line.split('] ', 1)[1] |
| 85 stripped_line = line.split('] ', 1)[1] | 118 if whitelist_re.search(stripped_line): |
| 86 if not stripped_line in whitelist: | 119 continue |
| 87 unexpected.append(stripped_line) | 120 unexpected.append(stripped_line) |
| 88 return unexpected | 121 return unexpected |
| OLD | NEW |