VirtualBox

source: vbox/trunk/src/VBox/ValidationKit/testboxscript/testboxtasks.py@ 53363

Last change on this file since 53363 was 52776, checked in by vboxsync, 10 years ago

fix OSE

  • Property svn:eol-style set to native
  • Property svn:keywords set to Author Date Id Revision
File size: 33.0 KB
Line 
1# -*- coding: utf-8 -*-
2# $Id: testboxtasks.py 52776 2014-09-17 14:51:43Z vboxsync $
3
4"""
5TestBox Script - Async Tasks.
6"""
7
8__copyright__ = \
9"""
10Copyright (C) 2012-2014 Oracle Corporation
11
12This file is part of VirtualBox Open Source Edition (OSE), as
13available from http://www.virtualbox.org. This file is free software;
14you can redistribute it and/or modify it under the terms of the GNU
15General Public License (GPL) as published by the Free Software
16Foundation, in version 2 as it comes in the "COPYING" file of the
17VirtualBox OSE distribution. VirtualBox OSE is distributed in the
18hope that it will be useful, but WITHOUT ANY WARRANTY of any kind.
19
20The contents of this file may alternatively be used under the terms
21of the Common Development and Distribution License Version 1.0
22(CDDL) only, as it comes in the "COPYING.CDDL" file of the
23VirtualBox OSE distribution, in which case the provisions of the
24CDDL are applicable instead of those of the GPL.
25
26You may elect to license modified versions of this file under the
27terms and conditions of either the GPL or the CDDL or both.
28"""
29__version__ = "$Revision: 52776 $"
30
31
32# Standard python imports.
33from datetime import datetime
34import os
35import re
36import signal;
37import sys
38import subprocess
39import threading
40import time
41
42# Validation Kit imports.
43from common import constants
44from common import utils;
45from common import webutils;
46import testboxcommons
47
48# Figure where we are.
49try: __file__
50except: __file__ = sys.argv[0];
51g_ksTestScriptDir = os.path.dirname(os.path.abspath(__file__));
52
53
54
55class TestBoxBaseTask(object):
56 """
57 Asynchronous task employing a thread to do the actual work.
58 """
59
60 ## Time to wait for a task to terminate.
61 kcSecTerminateTimeout = 60
62
63 def __init__(self, oTestBoxScript, cSecTimeout, fnThreadProc):
64 self._oTestBoxScript = oTestBoxScript;
65 self._cSecTimeout = cSecTimeout;
66 self._tsSecStarted = utils.timestampSecond();
67 self.__oRLock = threading.RLock();
68 self._oCv = threading.Condition(self.__oRLock);
69 self._fRunning = True; # Protected by lock.
70 self._fShouldTerminate = False; # Protected by lock.
71
72 # Spawn the worker thread.
73 self._oThread = threading.Thread(target=fnThreadProc);
74 self._oThread.daemon = True;
75 self._oThread.start();
76
77 def _lock(self):
78 """ Take the CV lock. """
79 self._oCv.acquire();
80
81 def _unlock(self):
82 """ Release the CV lock. """
83 self._oCv.release();
84
85 def _complete(self):
86 """
87 Indicate that the task is complete, waking up the main thread.
88 Usually called at the end of the thread procedure.
89 """
90 self._lock();
91 self._fRunning = False;
92 self._oCv.notifyAll();
93 self._unlock();
94
95 def isRunning(self):
96 """ Check if the task is still running. """
97 self._lock();
98 fRunning = self._fRunning;
99 self._unlock();
100 return fRunning;
101
102 def wait(self, cSecTimeout):
103 """ Wait for the task to complete. """
104 self._lock();
105 fRunning = self._fRunning;
106 if fRunning is True and cSecTimeout > 0:
107 self._oCv.wait(cSecTimeout)
108 self._unlock();
109 return fRunning;
110
111 def terminate(self, cSecTimeout = kcSecTerminateTimeout):
112 """ Terminate the task. """
113 self._lock();
114 self._fShouldTerminate = True;
115 self._unlock();
116
117 return self.wait(cSecTimeout);
118
119 def _shouldTerminate(self):
120 """
121 Returns True if we should terminate, False if not.
122 """
123 self._lock();
124 fShouldTerminate = self._fShouldTerminate == True;
125 self._unlock();
126 return fShouldTerminate;
127
128
129class TestBoxTestDriverTask(TestBoxBaseTask):
130 """
131 Base class for tasks involving test drivers.
132 """
133
134 ## When to flush the backlog of log messages.
135 kcchMaxBackLog = 32768;
136
137 ## The backlog sync time (seconds).
138 kcSecBackLogFlush = 30;
139
140 ## The timeout for the cleanup job (5 mins).
141 kcSecCleanupTimeout = 300;
142 ## The timeout to wait for the abort command before killing it.
143 kcSecAbortTimeout = 300;
144
145 ## The timeout to wait for the final output to be processed.
146 kcSecFinalOutputTimeout = 180;
147 ## The timeout to wait for the abort command output to be processed.
148 kcSecAbortCmdOutputTimeout = 30;
149 ## The timeout to wait for the terminate output to be processed.
150 kcSecTerminateOutputTimeout = 30;
151 ## The timeout to wait for the kill output to be processed.
152 kcSecKillOutputTimeout = 30;
153
154 ## The timeout for talking to the test manager.
155 ksecTestManagerTimeout = 60;
156
157
158 def __init__(self, oTestBoxScript, fnThreadProc, cSecTimeout, idResult, sScriptCmdLine):
159 """
160 Class instance init
161 """
162 # Init our instance data.
163 self._idResult = idResult;
164 self._sScriptCmdLine = sScriptCmdLine;
165 self._oChild = None;
166 self._oBackLogLock = threading.RLock();
167 self._oBackLogFlushLock = threading.RLock();
168 self._asBackLog = [];
169 self._cchBackLog = 0;
170 self._secTsBackLogFlush = utils.timestampSecond();
171
172 # Init super.
173 TestBoxBaseTask.__init__(self, oTestBoxScript, cSecTimeout, fnThreadProc);
174
175 def terminate(self, cSecTimeout = kcSecCleanupTimeout):
176 """ Reimplement with higher default timeout. """
177 return TestBoxBaseTask.terminate(self, cSecTimeout);
178
179 def _logFlush(self, oGivenConnection = None):
180 """
181 Flushes the log to the test manager.
182
183 No exceptions.
184 """
185 fRc = True;
186 self._oBackLogFlushLock.acquire();
187
188 # Grab the current back log.
189 self._oBackLogLock.acquire();
190 asBackLog = self._asBackLog;
191 self._asBackLog = [];
192 self._cchBackLog = 0;
193 self._secTsBackLogFlush = utils.timestampSecond();
194 self._oBackLogLock.release();
195
196 # If there is anything to flush, flush it.
197 if len(asBackLog) > 0:
198 sBody = '';
199 for sLine in asBackLog:
200 sBody += sLine + '\n';
201
202 oConnection = None;
203 try:
204 if oGivenConnection is None:
205 oConnection = self._oTestBoxScript.openTestManagerConnection();
206 oConnection.postRequest(constants.tbreq.LOG_MAIN, {constants.tbreq.LOG_PARAM_BODY: sBody});
207 oConnection.close();
208 else:
209 oGivenConnection.postRequest(constants.tbreq.LOG_MAIN, {constants.tbreq.LOG_PARAM_BODY: sBody});
210 except Exception, oXcpt:
211 testboxcommons.log('_logFlush error: %s' % (oXcpt,));
212 if len(sBody) < self.kcchMaxBackLog * 4:
213 self._oBackLogLock.acquire();
214 asBackLog.extend(self._asBackLog);
215 self._asBackLog = asBackLog;
216 # Don't restore _cchBackLog as there is no point in retrying immediately.
217 self._oBackLogLock.release();
218 if oConnection is not None: # Be kind to apache.
219 try: oConnection.close();
220 except: pass;
221 fRc = False;
222
223 self._oBackLogFlushLock.release();
224 return fRc;
225
226 def flushLogOnConnection(self, oConnection):
227 """
228 Attempts to flush the logon the given connection.
229
230 No exceptions.
231 """
232 return self._logFlush(oConnection);
233
234 def _logInternal(self, sMessage, fPrefix = True, fFlushCheck = False):
235 """
236 Internal logging.
237 Won't flush the backlog, returns a flush indicator so the caller can
238 do it instead.
239 """
240 if fPrefix:
241 try:
242 oNow = datetime.utcnow();
243 sTs = '%02u:%02u:%02u.%06u ' % (oNow.hour, oNow.minute, oNow.second, oNow.microsecond);
244 except Exception, oXcpt:
245 sTs = 'oXcpt=%s ' % (oXcpt);
246 sFullMsg = sTs + sMessage;
247 else:
248 sFullMsg = sMessage;
249
250 self._oBackLogLock.acquire();
251 self._asBackLog.append(sFullMsg);
252 cchBackLog = self._cchBackLog + len(sFullMsg) + 1;
253 self._cchBackLog = cchBackLog;
254 secTsBackLogFlush = self._secTsBackLogFlush;
255 self._oBackLogLock.release();
256
257 testboxcommons.log(sFullMsg);
258 return fFlushCheck \
259 and ( cchBackLog >= self.kcchMaxBackLog \
260 or utils.timestampSecond() - secTsBackLogFlush >= self.kcSecBackLogFlush);
261
262 def _log(self, sMessage):
263 """
264 General logging function, will flush.
265 """
266 if self._logInternal(sMessage, fFlushCheck = True):
267 self._logFlush();
268 return True;
269
270 def _reportDone(self, sResult):
271 """
272 Report EXEC job done to the test manager.
273
274 sResult is a value from constants.result.
275 """
276 ## @todo optimize this to use one server connection.
277
278 #
279 # Log it.
280 #
281 assert sResult in constants.result.g_kasValidResults;
282 self._log('Done %s' % (sResult,));
283
284 #
285 # Report it.
286 #
287 fRc = True;
288 secStart = utils.timestampSecond();
289 while True:
290 self._logFlush(); ## @todo Combine this with EXEC_COMPLETED.
291 oConnection = None;
292 try:
293 oConnection = self._oTestBoxScript.openTestManagerConnection();
294 oConnection.postRequest(constants.tbreq.EXEC_COMPLETED, {constants.tbreq.EXEC_COMPLETED_PARAM_RESULT: sResult});
295 oConnection.close();
296 except Exception, oXcpt:
297 if utils.timestampSecond() - secStart < self.ksecTestManagerTimeout:
298 self._log('_reportDone exception (%s) - retrying...' % (oXcpt,));
299 time.sleep(2);
300 continue;
301 self._log('_reportDone error: %s' % (oXcpt,));
302 if oConnection is not None: # Be kind to apache.
303 try: oConnection.close();
304 except: pass;
305 fRc = False;
306 break;
307
308 #
309 # Mark the task as completed.
310 #
311 self._complete();
312 return fRc;
313
314 def _assembleArguments(self, sAction, fWithInterpreter = True):
315 """
316 Creates an argument array for subprocess.Popen, splitting the
317 sScriptCmdLine like bourne shell would.
318 fWithInterpreter is used (False) when checking that the script exists.
319
320 Returns None on bad input.
321 """
322
323 #
324 # This is a good place to export the test set id to the environment.
325 #
326 os.environ['TESTBOX_TEST_SET_ID'] = str(self._idResult);
327 cTimeoutLeft = utils.timestampSecond() - self._tsSecStarted;
328 cTimeoutLeft = 0 if cTimeoutLeft >= self._cSecTimeout else self._cSecTimeout - cTimeoutLeft;
329 os.environ['TESTBOX_TIMEOUT'] = str(cTimeoutLeft);
330 os.environ['TESTBOX_TIMEOUT_ABS'] = str(self._tsSecStarted + self._cSecTimeout);
331
332 #
333 # Do replacements and split the command line into arguments.
334 #
335 if self._sScriptCmdLine.find('@ACTION@') >= 0:
336 sCmdLine = self._sScriptCmdLine.replace('@ACTION@', sAction);
337 else:
338 sCmdLine = self._sScriptCmdLine + ' ' + sAction;
339 for sVar in [ 'TESTBOX_PATH_BUILDS', 'TESTBOX_PATH_RESOURCES', 'TESTBOX_PATH_SCRATCH', 'TESTBOX_PATH_SCRIPTS',
340 'TESTBOX_PATH_UPLOAD', 'TESTBOX_UUID', 'TESTBOX_REPORTER', 'TESTBOX_ID', 'TESTBOX_TEST_SET_ID',
341 'TESTBOX_TIMEOUT', 'TESTBOX_TIMEOUT_ABS' ]:
342 if sCmdLine.find('${' + sVar + '}') >= 0:
343 sCmdLine = sCmdLine.replace('${' + sVar + '}', os.environ[sVar]);
344
345 asArgs = utils.argsSplit(sCmdLine);
346
347 #
348 # Massage argv[0]:
349 # - Convert portable slashes ('/') to the flavor preferred by the
350 # OS we're currently running on.
351 # - Run python script thru the current python interpreter (important
352 # on systems that doesn't sport native hash-bang script execution).
353 #
354 asArgs[0] = asArgs[0].replace('/', os.path.sep);
355 if not os.path.isabs(asArgs[0]):
356 asArgs[0] = os.path.join(self._oTestBoxScript.getPathScripts(), asArgs[0]);
357
358 if asArgs[0].endswith('.py') and fWithInterpreter:
359 if sys.executable is not None and len(sys.executable) > 0:
360 asArgs.insert(0, sys.executable);
361 else:
362 asArgs.insert(0, 'python');
363
364 return asArgs;
365
366 def _outputThreadProc(self, oChild, oStdOut, sAction):
367 """
368 Thread procedure for the thread that reads the output of the child
369 process. We use a dedicated thread for this purpose since non-blocking
370 I/O may be hard to keep portable according to hints around the web...
371 """
372 oThread = oChild.oOutputThread;
373 while not oThread.fPleaseQuit:
374 # Get a line.
375 try:
376 sLine = oStdOut.readline();
377 except Exception, oXcpt:
378 self._log('child (%s) pipe I/O error: %s' % (sAction, oXcpt,));
379 break;
380
381 # EOF?
382 if len(sLine) == 0:
383 break;
384
385 # Strip trailing new line (DOS and UNIX).
386 if sLine.endswith("\r\n"):
387 sLine = sLine[0:-2];
388 elif sLine.endswith("\n"):
389 sLine = sLine[0:-1];
390
391 # Log it.
392 if self._logInternal(sLine, fPrefix = False, fFlushCheck = True):
393 self._logFlush();
394
395 # Close the stdout pipe in case we were told to get lost.
396 try:
397 oStdOut.close();
398 except Exception, oXcpt:
399 self._log('warning: Exception closing stdout pipe of "%s" child: %s' % (sAction, oXcpt,));
400
401 # This is a bit hacky, but try reap the child so it won't hang as
402 # defunkt during abort/timeout.
403 if oChild.poll() is None:
404 for _ in range(15):
405 time.sleep(0.2);
406 if oChild.poll() is not None:
407 break;
408
409 oChild = None;
410 return None;
411
412 def _spawnChild(self, sAction):
413 """
414 Spawns the child process, returning success indicator + child object.
415 """
416
417 # Argument list.
418 asArgs = self._assembleArguments(sAction)
419 if asArgs is None:
420 self._log('Malformed command line: "%s"' % (self._sScriptCmdLine,));
421 return (False, None);
422
423 # Spawn child.
424 try:
425 oChild = subprocess.Popen(asArgs,
426 shell = False,
427 bufsize = -1,
428 stdout = subprocess.PIPE,
429 stderr = subprocess.STDOUT,
430 cwd = self._oTestBoxScript.getPathSpill(),
431 universal_newlines = True,
432 close_fds = (False if utils.getHostOs() == 'win' else True),
433 preexec_fn = (None if utils.getHostOs() in ['win', 'os2']
434 else os.setsid)); # pylint: disable=E1101
435 except Exception, oXcpt:
436 self._log('Error creating child process %s: %s' % (asArgs, oXcpt));
437 return (False, None);
438
439 oChild.sTestBoxScriptAction = sAction;
440
441 # Start output thread, extending the child object to keep track of it.
442 oChild.oOutputThread = threading.Thread(target=self._outputThreadProc, args=(oChild, oChild.stdout, sAction))
443 oChild.oOutputThread.daemon = True;
444 oChild.oOutputThread.fPleaseQuit = False; # Our extension.
445 oChild.oOutputThread.start();
446
447 return (True, oChild);
448
449 def _monitorChild(self, cSecTimeout, fTryKillCommand = True, oChild = None):
450 """
451 Monitors the child process. If the child executes longer that
452 cSecTimeout allows, we'll terminate it.
453 Returns Success indicator and constants.result value.
454 """
455
456 if oChild is None:
457 oChild = self._oChild;
458
459 iProcGroup = oChild.pid;
460 if utils.getHostOs() in ['win', 'os2'] or iProcGroup <= 0:
461 iProcGroup = -2;
462
463 #
464 # Do timeout processing and check the health of the child.
465 #
466 sResult = constants.result.PASSED;
467 seStarted = utils.timestampSecond();
468 while True:
469 # Check status.
470 iRc = oChild.poll();
471 if iRc is not None:
472 self._log('Child doing "%s" completed with exit code %d' % (oChild.sTestBoxScriptAction, iRc));
473 oChild.oOutputThread.join(self.kcSecFinalOutputTimeout);
474
475 if oChild is self._oChild:
476 self._oChild = None;
477
478 if iRc == constants.rtexitcode.SKIPPED:
479 return (True, constants.result.SKIPPED);
480 if iRc != constants.rtexitcode.SUCCESS:
481 return (False, constants.result.FAILED);
482 return (True, constants.result.PASSED);
483
484 # Check for abort first, since that has less of a stigma.
485 if self._shouldTerminate() is True:
486 sResult = constants.result.ABORTED;
487 break;
488
489 # Check timeout.
490 cSecElapsed = utils.timestampSecond() - seStarted;
491 if cSecElapsed > cSecTimeout:
492 self._log('Timeout: %u secs (limit %u secs)' % (cSecElapsed, cSecTimeout));
493 sResult = constants.result.TIMED_OUT;
494 break;
495
496 # Wait.
497 cSecLeft = cSecTimeout - cSecElapsed;
498 oChild.oOutputThread.join(15 if cSecLeft > 15 else (cSecLeft + 1));
499
500 #
501 # If the child is still alive, try use the abort command to stop it
502 # very gently. This let's the testdriver clean up daemon processes
503 # and such that our code below won't catch.
504 #
505 if fTryKillCommand and oChild.poll() is None:
506 self._log('Attempting to abort child...');
507 (fRc2, oAbortChild) = self._spawnChild('abort');
508 if oAbortChild is not None and fRc2 is True:
509 self._monitorChild(self.kcSecAbortTimeout, False, oAbortChild);
510 oAbortChild = None;
511
512 #
513 # If the child is still alive, try the polite way.
514 #
515 if oChild.poll() is None:
516 self._log('Attempting to terminate child doing "%s"...' % (oChild.sTestBoxScriptAction,));
517
518 if iProcGroup > 0:
519 try:
520 os.killpg(iProcGroup, signal.SIGTERM); # pylint: disable=E1101
521 except Exception, oXcpt:
522 self._log('killpg() failed: %s' % (oXcpt,));
523
524 try:
525 self._oChild.terminate();
526 oChild.oOutputThread.join(self.kcSecTerminateOutputTimeout);
527 except Exception, oXcpt:
528 self._log('terminate() failed: %s' % (oXcpt,));
529
530 #
531 # If the child doesn't respond to polite, kill it. Always do a killpg
532 # should there be any processes left in the group.
533 #
534 if iProcGroup > 0:
535 try:
536 os.killpg(iProcGroup, signal.SIGKILL); # pylint: disable=E1101
537 except Exception, oXcpt:
538 self._log('killpg() failed: %s' % (oXcpt,));
539
540 if oChild.poll() is None:
541 self._log('Attemting to kill child doing "%s"...' % (oChild.sTestBoxScriptAction,));
542 try:
543 self._oChild.kill();
544 oChild.oOutputThread.join(self.kcSecKillOutputTimeout);
545 except Exception, oXcpt:
546 self._log('kill() failed: %s' % (oXcpt,));
547
548 #
549 # Give the whole mess a couple of more seconds to respond in case the
550 # output thread exitted prematurely for some weird reason.
551 #
552 if oChild.poll() is None:
553 time.sleep(2);
554 time.sleep(2);
555 time.sleep(2);
556
557 iRc = oChild.poll();
558 if iRc is not None:
559 self._log('Child doing "%s" aborted with exit code %d' % (oChild.sTestBoxScriptAction, iRc));
560 else:
561 self._log('Child doing "%s" is still running, giving up...' % (oChild.sTestBoxScriptAction,));
562 ## @todo in this case we should probably try reboot the testbox...
563 oChild.oOutputThread.fPleaseQuit = True;
564
565 if oChild is self._oChild:
566 self._oChild = None;
567 return (False, sResult);
568
569 def _terminateChild(self):
570 """
571 Terminates the child forcefully.
572 """
573 if self._oChild is not None:
574 pass;
575
576 def _cleanupAfter(self):
577 """
578 Cleans up after a test failure. (On success, cleanup is implicit.)
579 """
580 assert self._oChild is None;
581
582 #
583 # Tell the script to clean up.
584 #
585 if len(self._sScriptCmdLine) > 0: # can be empty if cleanup crashed.
586 (fRc, self._oChild) = self._spawnChild('cleanup-after');
587 if fRc is True:
588 (fRc, _) = self._monitorChild(self.kcSecCleanupTimeout, False);
589 self._terminateChild();
590 else:
591 fRc = False;
592
593 #
594 # Wipe the stuff clean.
595 #
596 fRc2 = self._oTestBoxScript.reinitScratch(fnLog = self._log);
597
598 return fRc and fRc2;
599
600
601
602class TestBoxCleanupTask(TestBoxTestDriverTask):
603 """
604 Special asynchronous task for cleaning up a stale test when starting the
605 testbox script. It's assumed that the reason for the stale test lies in
606 it causing a panic, reboot, or similar, so we'll also try collect some
607 info about recent system crashes and reboots.
608 """
609
610 def __init__(self, oTestBoxScript):
611 # Read the old state, throwing a fit if it's invalid.
612 sScriptState = oTestBoxScript.getPathState();
613 sScriptCmdLine = self._readStateFile(os.path.join(sScriptState, 'script-cmdline.txt'));
614 sResultId = self._readStateFile(os.path.join(sScriptState, 'result-id.txt'));
615 try:
616 idResult = int(sResultId);
617 if idResult <= 0 or idResult >= 0x7fffffff:
618 raise Exception('');
619 except:
620 raise Exception('Invalid id value "%s" found in %s' % (sResultId, os.path.join(sScriptState, 'result-id.txt')));
621
622 sTestBoxId = self._readStateFile(os.path.join(sScriptState, 'testbox-id.txt'));
623 try:
624 self.idTestBox = int(sTestBoxId);
625 if self.idTestBox <= 0 or self.idTestBox >= 0x7fffffff:
626 raise Exception('');
627 except:
628 raise Exception('Invalid id value "%s" found in %s' % (sTestBoxId, os.path.join(sScriptState, 'testbox-id.txt')));
629 self.sTestBoxName = self._readStateFile(os.path.join(sScriptState, 'testbox-name.txt'));
630
631 # Init super.
632 TestBoxTestDriverTask.__init__(self, oTestBoxScript, self._threadProc, self.kcSecCleanupTimeout,
633 idResult, sScriptCmdLine);
634
635 @staticmethod
636 def _readStateFile(sPath):
637 """
638 Reads a state file, returning a string on success and otherwise raising
639 an exception.
640 """
641 try:
642 oFile = open(sPath, "rb");
643 sStr = oFile.read();
644 oFile.close();
645 return sStr.strip();
646 except Exception, oXcpt:
647 raise Exception('Failed to read "%s": %s' % (sPath, oXcpt));
648
649 def _threadProc(self):
650 """
651 Perform the actual clean up on script startup.
652 """
653
654 #
655 # First make sure we won't repeat this exercise should it turn out to
656 # trigger another reboot/panic/whatever.
657 #
658 sScriptCmdLine = os.path.join(self._oTestBoxScript.getPathState(), 'script-cmdline.txt');
659 try:
660 os.remove(sScriptCmdLine);
661 oFile = open(sScriptCmdLine, 'wb');
662 oFile.close();
663 except Exception, oXcpt:
664 self._log('Error truncating "%s": %s' % (sScriptCmdLine, oXcpt));
665
666 #
667 # Report the incident.
668 #
669 self._log('Seems we rebooted!');
670 self._log('script-cmdline="%s"' % (self._sScriptCmdLine));
671 self._log('result-id=%d' % (self._idResult));
672 self._log('testbox-id=%d' % (self.idTestBox));
673 self._log('testbox-name=%s' % (self.sTestBoxName));
674 self._logFlush();
675
676 # System specific info.
677 sOs = utils.getHostOs();
678 if sOs == 'darwin':
679 self._log('NVRAM Panic Info:\n%s\n' % (self.darwinGetPanicInfo(),));
680
681 self._logFlush();
682 ## @todo Add some special command for reporting this situation so we get something
683 # useful in the event log.
684
685 #
686 # Do the cleaning up.
687 #
688 self._cleanupAfter();
689
690 self._reportDone(constants.result.REBOOTED);
691 return False;
692
693 def darwinGetPanicInfo(self):
694 """
695 Returns a string with the aapl,panic-info content.
696 """
697 # Retriev the info.
698 try:
699 sRawInfo = utils.processOutputChecked(['nvram', 'aapl,panic-info']);
700 except Exception, oXcpt:
701 return 'exception running nvram: %s' % (oXcpt,);
702
703 # Decode (%xx) and decompact it (7-bit -> 8-bit).
704 ahDigits = \
705 {
706 '0': 0, '1': 1, '2': 2, '3': 3, '4': 4, '5': 5, '6': 6, '7': 7,
707 '8': 8, '9': 9, 'a': 10, 'b': 11, 'c': 12, 'd': 13, 'e': 14, 'f': 15,
708 };
709 sInfo = '';
710 off = len('aapl,panic-info') + 1;
711 iBit = 0;
712 bLow = 0;
713
714 while off < len(sRawInfo):
715 # isprint is used to determine whether to %xx or %c it, so we have to
716 # be a little careful before assuming % sequences are hex bytes.
717 if sRawInfo[off] == '%' \
718 and off + 3 <= len(sRawInfo) \
719 and sRawInfo[off + 1] in ahDigits \
720 and sRawInfo[off + 2] in ahDigits:
721 bCur = ahDigits[sRawInfo[off + 1]] * 0x10 + ahDigits[sRawInfo[off + 2]];
722 off += 3;
723 else:
724 bCur = ord(sRawInfo[off]);
725 off += 1;
726
727 sInfo += chr(((bCur & (0x7f >> iBit)) << iBit) | bLow);
728 bLow = bCur >> (7 - iBit);
729
730 if iBit < 6:
731 iBit += 1;
732 else:
733 # Final bit in sequence.
734 sInfo += chr(bLow);
735 bLow = 0;
736 iBit = 0;
737
738 # Expand shorthand.
739 sInfo = sInfo.replace('@', 'com.apple.');
740 sInfo = sInfo.replace('>', 'com.apple.driver.');
741 sInfo = sInfo.replace('|', 'com.apple.iokit.');
742 sInfo = sInfo.replace('$', 'com.apple.security.');
743 sInfo = sInfo.replace('!A', 'Apple');
744 sInfo = sInfo.replace('!a', 'Action');
745 sInfo = sInfo.replace('!B', 'Bluetooth');
746 sInfo = sInfo.replace('!C', 'Controller');
747 sInfo = sInfo.replace('!F', 'Family');
748 sInfo = sInfo.replace('!I', 'Intel');
749 sInfo = sInfo.replace('!U', 'AppleUSB');
750 sInfo = sInfo.replace('!P', 'Profile');
751
752 # Done.
753 return sInfo
754
755
756class TestBoxExecTask(TestBoxTestDriverTask):
757 """
758 Implementation of a asynchronous EXEC task.
759
760 This uses a thread for doing the actual work, i.e. starting and monitoring
761 the child process, processing its output, and more.
762 """
763
764 def __init__(self, oTestBoxScript, idResult, sScriptZips, sScriptCmdLine, cSecTimeout):
765 """
766 Class instance init
767 """
768 # Init our instance data.
769 self._sScriptZips = sScriptZips;
770
771 # Init super.
772 TestBoxTestDriverTask.__init__(self, oTestBoxScript, self._threadProc, cSecTimeout, idResult, sScriptCmdLine);
773
774 @staticmethod
775 def _writeStateFile(sPath, sContent):
776 """
777 Writes a state file, raising an exception on failure.
778 """
779 try:
780 oFile = open(sPath, "wb");
781 oFile.write(sContent);
782 oFile.flush();
783 try: os.fsync(oFile.fileno());
784 except: pass;
785 oFile.close();
786 except Exception, oXcpt:
787 raise Exception('Failed to write "%s": %s' % (sPath, oXcpt));
788 return True;
789
790 def _saveState(self):
791 """
792 Saves the task state on disk so we can launch a TestBoxCleanupTask job
793 if the test should cause system panic or similar.
794
795 Note! May later be extended to support tests that reboots the host.
796 """
797 sScriptState = self._oTestBoxScript.getPathState();
798 try:
799 self._writeStateFile(os.path.join(sScriptState, 'script-cmdline.txt'), self._sScriptCmdLine);
800 self._writeStateFile(os.path.join(sScriptState, 'result-id.txt'), str(self._idResult));
801 self._writeStateFile(os.path.join(sScriptState, 'testbox-id.txt'), str(self._oTestBoxScript.getTestBoxId()));
802 self._writeStateFile(os.path.join(sScriptState, 'testbox-name.txt'), self._oTestBoxScript.getTestBoxName());
803 except Exception, oXcpt:
804 self._log('Failed to write state: %s' % (oXcpt,));
805 return False;
806 return True;
807
808 def _downloadAndUnpackScriptZips(self):
809 """
810 Downloads/copies the script ZIPs into TESTBOX_SCRIPT and unzips them to
811 the same directory.
812
813 Raises no exceptions, returns log + success indicator instead.
814 """
815 sPathScript = self._oTestBoxScript.getPathScripts();
816 asArchives = self._sScriptZips.split(',');
817 for sArchive in asArchives:
818 sArchive = sArchive.strip();
819 if len(sArchive) == 0:
820 continue;
821
822 # Figure the destination name (in scripts).
823 sDstFile = webutils.getFilename(sArchive);
824 if len(sDstFile) < 1 \
825 or re.search('[^a-zA-Z0-9 !#$%&\'()@^_`{}~.-]', sDstFile) is not None: # FAT charset sans 128-255 + '.'.
826 self._log('Malformed script zip filename: %s' % (sArchive,));
827 return False;
828 sDstFile = os.path.join(sPathScript, sDstFile);
829
830 # Do the work.
831 if webutils.downloadFile(sArchive, sDstFile, self._oTestBoxScript.getPathBuilds(), self._log, self._log) is not True:
832 return False;
833 asFiles = utils.unpackFile(sDstFile, sPathScript, self._log, self._log);
834 if asFiles is None:
835 return False;
836
837 # Since zip files doesn't always include mode masks, set the X bit
838 # of all of them so we can execute binaries and hash-bang scripts.
839 for sFile in asFiles:
840 utils.chmodPlusX(sFile);
841
842 return True;
843
844 def _threadProc(self):
845 """
846 Do the work of an EXEC command.
847 """
848
849 sResult = constants.result.PASSED;
850
851 #
852 # Start by preparing the scratch directories.
853 #
854 # Note! Failures at this stage are not treated as real errors since
855 # they may be caused by the previous test and other circumstances
856 # so we don't want to go fail a build because of this.
857 #
858 fRc = self._oTestBoxScript.reinitScratch(self._logInternal);
859 fNeedCleanUp = fRc;
860 if fRc is True:
861 fRc = self._downloadAndUnpackScriptZips();
862 testboxcommons.log2('_threadProc: _downloadAndUnpackScriptZips -> %s' % (fRc,));
863 if fRc is not True:
864 sResult = constants.result.BAD_TESTBOX;
865
866 #
867 # Make sure the script exists.
868 #
869 if fRc is True:
870 sScript = self._assembleArguments('none', fWithInterpreter = False)[0];
871 if not os.path.exists(sScript):
872 self._log('The test driver script "%s" cannot be found.' % (sScript,));
873 sDir = sScript;
874 while len(sDir) > 3:
875 sDir = os.path.dirname(sDir);
876 if os.path.exists(sDir):
877 self._log('First existing parent directory is "%s".' % (sDir,));
878 break;
879 fRc = False;
880
881 if fRc is True:
882 #
883 # Start testdriver script.
884 #
885 fRc = self._saveState();
886 if fRc:
887 (fRc, self._oChild) = self._spawnChild('all');
888 testboxcommons.log2('_threadProc: _spawnChild -> %s, %s' % (fRc, self._oChild));
889 if fRc:
890 (fRc, sResult) = self._monitorChild(self._cSecTimeout);
891 testboxcommons.log2('_threadProc: _monitorChild -> %s' % (fRc,));
892
893 # If the run failed, do explicit cleanup.
894 if fRc is not True:
895 testboxcommons.log2('_threadProc: explicit cleanups...');
896 self._terminateChild();
897 self._cleanupAfter();
898 fNeedCleanUp = False;
899 assert self._oChild is None;
900
901 #
902 # Clean up scratch.
903 #
904 if fNeedCleanUp:
905 if self._oTestBoxScript.reinitScratch(self._logInternal) is not True:
906 self._log('post run reinitScratch failed.');
907 fRc = False;
908
909 #
910 # Report status and everything back to the test manager.
911 #
912 if fRc is False and sResult == constants.result.PASSED:
913 sResult = constants.result.FAILED;
914 self._reportDone(sResult);
915 return fRc;
916
Note: See TracBrowser for help on using the repository browser.

© 2024 Oracle Support Privacy / Do Not Sell My Info Terms of Use Trademark Policy Automated Access Etiquette