OLD | NEW |
1 # Copyright (C) 2010 Google Inc. All rights reserved. | 1 # Copyright (C) 2010 Google Inc. All rights reserved. |
2 # | 2 # |
3 # Redistribution and use in source and binary forms, with or without | 3 # Redistribution and use in source and binary forms, with or without |
4 # modification, are permitted provided that the following conditions are | 4 # modification, are permitted provided that the following conditions are |
5 # met: | 5 # met: |
6 # | 6 # |
7 # * Redistributions of source code must retain the above copyright | 7 # * Redistributions of source code must retain the above copyright |
8 # notice, this list of conditions and the following disclaimer. | 8 # notice, this list of conditions and the following disclaimer. |
9 # * Redistributions in binary form must reproduce the above | 9 # * Redistributions in binary form must reproduce the above |
10 # copyright notice, this list of conditions and the following disclaimer | 10 # copyright notice, this list of conditions and the following disclaimer |
(...skipping 136 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
147 def _clean_up_dummy_home_dir(self): | 147 def _clean_up_dummy_home_dir(self): |
148 """Cleans up the dummy dir and resets the HOME environment variable.""" | 148 """Cleans up the dummy dir and resets the HOME environment variable.""" |
149 dummy_home = self.host.environ['HOME'] | 149 dummy_home = self.host.environ['HOME'] |
150 assert dummy_home != self._original_home | 150 assert dummy_home != self._original_home |
151 self._filesystem.rmtree(dummy_home) | 151 self._filesystem.rmtree(dummy_home) |
152 self.host.environ['HOME'] = self._original_home | 152 self.host.environ['HOME'] = self._original_home |
153 | 153 |
154 def _start_xvfb(self): | 154 def _start_xvfb(self): |
155 display = self._find_display() | 155 display = self._find_display() |
156 if not display: | 156 if not display: |
157 _log.warn('Failed to find a free display to start Xvfb.') | 157 _log.critical('Failed to find a free display to start Xvfb.') |
158 return | 158 return |
159 | 159 |
160 _log.info('Starting Xvfb with display "%s".', display) | 160 _log.info('Starting Xvfb with display "%s".', display) |
161 self._xvfb_stdout = tempfile.NamedTemporaryFile(delete=False) | 161 self._xvfb_stdout = tempfile.NamedTemporaryFile(delete=False) |
162 self._xvfb_stderr = tempfile.NamedTemporaryFile(delete=False) | 162 self._xvfb_stderr = tempfile.NamedTemporaryFile(delete=False) |
163 self._xvfb_process = self.host.executive.popen( | 163 self._xvfb_process = self.host.executive.popen( |
164 ['Xvfb', display, '-screen', '0', '1280x800x24', '-ac', '-dpi', '96'
], | 164 ['Xvfb', display, '-screen', '0', '1280x800x24', '-ac', '-dpi', '96'
], |
165 stdout=self._xvfb_stdout, stderr=self._xvfb_stderr) | 165 stdout=self._xvfb_stdout, stderr=self._xvfb_stderr) |
166 | 166 |
167 # By setting DISPLAY here, the individual worker processes will | 167 # By setting DISPLAY here, the individual worker processes will |
168 # get the right DISPLAY. Note, if this environment could be passed | 168 # get the right DISPLAY. Note, if this environment could be passed |
169 # when creating workers, then we wouldn't need to modify DISPLAY here. | 169 # when creating workers, then we wouldn't need to modify DISPLAY here. |
170 self._original_display = self.host.environ.get('DISPLAY') | 170 self._original_display = self.host.environ.get('DISPLAY') |
171 self.host.environ['DISPLAY'] = display | 171 self.host.environ['DISPLAY'] = display |
172 | 172 |
173 # The poll() method will return None if the process has not terminated: | 173 # Check that xvfb has started correctly via probing using xdpyinfo. |
| 174 # While xvfb is running, the poll() method will return None; |
174 # https://docs.python.org/2/library/subprocess.html#subprocess.Popen.pol
l | 175 # https://docs.python.org/2/library/subprocess.html#subprocess.Popen.pol
l |
175 if self._xvfb_process.poll() is not None: | |
176 _log.warn('Failed to start Xvfb on display "%s."', display) | |
177 self._stop_xvfb() | |
178 | |
179 start_time = self.host.time() | 176 start_time = self.host.time() |
180 while self.host.time() - start_time < self.XVFB_START_TIMEOUT: | 177 while self.host.time() - start_time < self.XVFB_START_TIMEOUT or self._x
vfb_process.poll() is not None: |
181 # We don't explicitly set the display, as we want to check the | 178 # We don't explicitly set the display, as we want to check the |
182 # environment value. | 179 # environment value. |
183 exit_code = self.host.executive.run_command( | 180 exit_code = self.host.executive.run_command( |
184 ['xdpyinfo'], return_exit_code=True) | 181 ['xdpyinfo'], return_exit_code=True) |
185 if exit_code == 0: | 182 if exit_code == 0: |
186 _log.info('Successfully started Xvfb with display "%s".', displa
y) | 183 _log.info('Successfully started Xvfb with display "%s".', displa
y) |
187 return | 184 return |
188 _log.warn('xdpyinfo check failed with exit code %s while starting Xv
fb on "%s".', exit_code, display) | 185 _log.warn('xdpyinfo check failed with exit code %s while starting Xv
fb on "%s".', exit_code, display) |
189 self.host.sleep(0.1) | 186 self.host.sleep(0.1) |
190 _log.fatal('Failed to start Xvfb on display "%s" (xdpyinfo check failed)
.', display) | 187 |
| 188 retcode = self._xvfb_process.poll() |
191 self._stop_xvfb() | 189 self._stop_xvfb() |
| 190 _log.critical('Failed to start Xvfb on display "%s" (xvfb retcode: %r).'
, display, retcode) |
192 | 191 |
193 def _find_display(self): | 192 def _find_display(self): |
194 """Tries to find a free X display, looping if necessary.""" | 193 """Tries to find a free X display, looping if necessary.""" |
195 # The "xvfb-run" command uses :99 by default. | 194 # The "xvfb-run" command uses :99 by default. |
196 for display_number in range(99, 120): | 195 for display_number in range(99, 120): |
197 display = ':%d' % display_number | 196 display = ':%d' % display_number |
198 exit_code = self.host.executive.run_command( | 197 exit_code = self.host.executive.run_command( |
199 ['xdpyinfo', '-display', display], return_exit_code=True) | 198 ['xdpyinfo', '-display', display], return_exit_code=True) |
200 if exit_code == 1: | 199 if exit_code == 1: |
201 return display | 200 return display |
(...skipping 16 matching lines...) Expand all Loading... |
218 self.host.filesystem.remove(self._xvfb_stdout.name) | 217 self.host.filesystem.remove(self._xvfb_stdout.name) |
219 if self._xvfb_stderr and self.host.filesystem.exists(self._xvfb_stderr.n
ame): | 218 if self._xvfb_stderr and self.host.filesystem.exists(self._xvfb_stderr.n
ame): |
220 for line in self.host.filesystem.read_text_file(self._xvfb_stderr.na
me).splitlines(): | 219 for line in self.host.filesystem.read_text_file(self._xvfb_stderr.na
me).splitlines(): |
221 _log.warn('Xvfb stderr: %s', line) | 220 _log.warn('Xvfb stderr: %s', line) |
222 self.host.filesystem.remove(self._xvfb_stderr.name) | 221 self.host.filesystem.remove(self._xvfb_stderr.name) |
223 self._xvfb_stdout = self._xvfb_stderr = self._xvfb_process = None | 222 self._xvfb_stdout = self._xvfb_stderr = self._xvfb_process = None |
224 | 223 |
225 def _path_to_driver(self, target=None): | 224 def _path_to_driver(self, target=None): |
226 binary_name = self.driver_name() | 225 binary_name = self.driver_name() |
227 return self._build_path_with_target(target, binary_name) | 226 return self._build_path_with_target(target, binary_name) |
OLD | NEW |