mbligh | 6203ace | 2007-10-04 21:54:24 +0000 | [diff] [blame] | 1 | #!/usr/bin/python -u |
mbligh | 1ffd5dc | 2008-11-25 13:24:05 +0000 | [diff] [blame] | 2 | # Copyright 2007-2008 Martin J. Bligh <mbligh@google.com>, Google Inc. |
mbligh | 82648e5 | 2008-11-20 16:54:25 +0000 | [diff] [blame] | 3 | # Released under the GPL v2 |
mbligh | dcd57a8 | 2007-07-11 23:06:47 +0000 | [diff] [blame] | 4 | |
mbligh | c8949b8 | 2007-07-23 16:33:58 +0000 | [diff] [blame] | 5 | """ |
Aviv Keshet | de6bb19 | 2013-01-30 16:17:22 -0800 | [diff] [blame] | 6 | Run a control file through the server side engine |
mbligh | dcd57a8 | 2007-07-11 23:06:47 +0000 | [diff] [blame] | 7 | """ |
mbligh | 1ffd5dc | 2008-11-25 13:24:05 +0000 | [diff] [blame] | 8 | |
mbligh | a5f5e54 | 2009-12-30 16:57:49 +0000 | [diff] [blame] | 9 | import sys, os, re, traceback, signal, time, logging, getpass |
mbligh | 1ffd5dc | 2008-11-25 13:24:05 +0000 | [diff] [blame] | 10 | |
mbligh | f5427bb | 2008-04-09 15:55:57 +0000 | [diff] [blame] | 11 | import common |
mbligh | 9ff89cd | 2009-09-03 20:28:17 +0000 | [diff] [blame] | 12 | |
Dan Shi | a1ecd5c | 2013-06-06 11:21:31 -0700 | [diff] [blame] | 13 | from autotest_lib.client.common_lib import control_data |
| 14 | from autotest_lib.client.common_lib import global_config |
| 15 | require_atfork = global_config.global_config.get_config_value( |
mbligh | cb8cb33 | 2009-09-03 21:08:56 +0000 | [diff] [blame] | 16 | 'AUTOSERV', 'require_atfork_module', type=bool, default=True) |
| 17 | |
Dan Shi | a1ecd5c | 2013-06-06 11:21:31 -0700 | [diff] [blame] | 18 | |
mbligh | cb8cb33 | 2009-09-03 21:08:56 +0000 | [diff] [blame] | 19 | try: |
| 20 | import atfork |
| 21 | atfork.monkeypatch_os_fork_functions() |
| 22 | import atfork.stdlib_fixer |
| 23 | # Fix the Python standard library for threading+fork safety with its |
| 24 | # internal locks. http://code.google.com/p/python-atfork/ |
| 25 | import warnings |
| 26 | warnings.filterwarnings('ignore', 'logging module already imported') |
| 27 | atfork.stdlib_fixer.fix_logging_module() |
| 28 | except ImportError, e: |
| 29 | from autotest_lib.client.common_lib import global_config |
| 30 | if global_config.global_config.get_config_value( |
| 31 | 'AUTOSERV', 'require_atfork_module', type=bool, default=False): |
| 32 | print >>sys.stderr, 'Please run utils/build_externals.py' |
| 33 | print e |
| 34 | sys.exit(1) |
mbligh | 9ff89cd | 2009-09-03 20:28:17 +0000 | [diff] [blame] | 35 | |
Dan Shi | a1ecd5c | 2013-06-06 11:21:31 -0700 | [diff] [blame] | 36 | from autotest_lib.server import frontend |
showard | 75cdfee | 2009-06-10 17:40:41 +0000 | [diff] [blame] | 37 | from autotest_lib.server import server_logging_config |
showard | 043c62a | 2009-06-10 19:48:57 +0000 | [diff] [blame] | 38 | from autotest_lib.server import server_job, utils, autoserv_parser, autotest |
Dan Shi | a1ecd5c | 2013-06-06 11:21:31 -0700 | [diff] [blame] | 39 | from autotest_lib.server import utils as server_utils |
| 40 | |
showard | 75cdfee | 2009-06-10 17:40:41 +0000 | [diff] [blame] | 41 | from autotest_lib.client.common_lib import pidfile, logging_manager |
Dan Shi | a1ecd5c | 2013-06-06 11:21:31 -0700 | [diff] [blame] | 42 | from autotest_lib.site_utils.graphite import stats |
mbligh | 92c0fc2 | 2008-11-20 16:52:23 +0000 | [diff] [blame] | 43 | |
Alex Miller | f1af17e | 2013-01-09 22:50:32 -0800 | [diff] [blame] | 44 | def log_alarm(signum, frame): |
| 45 | logging.error("Received SIGALARM. Ignoring and continuing on.") |
Alex Miller | 0528d6f | 2013-01-11 10:49:48 -0800 | [diff] [blame] | 46 | sys.exit(1) |
Alex Miller | f1af17e | 2013-01-09 22:50:32 -0800 | [diff] [blame] | 47 | |
mbligh | a46678d | 2008-05-01 20:00:01 +0000 | [diff] [blame] | 48 | def run_autoserv(pid_file_manager, results, parser): |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 49 | # send stdin to /dev/null |
| 50 | dev_null = os.open(os.devnull, os.O_RDONLY) |
| 51 | os.dup2(dev_null, sys.stdin.fileno()) |
| 52 | os.close(dev_null) |
mbligh | dbf3761 | 2007-11-24 19:38:11 +0000 | [diff] [blame] | 53 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 54 | # Create separate process group |
| 55 | os.setpgrp() |
mbligh | 1d42d4e | 2007-11-05 22:42:00 +0000 | [diff] [blame] | 56 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 57 | # Implement SIGTERM handler |
mbligh | c229956 | 2009-07-02 19:00:36 +0000 | [diff] [blame] | 58 | def handle_sigterm(signum, frame): |
Simran Basi | 9d9b729 | 2013-10-16 16:44:07 -0700 | [diff] [blame] | 59 | logging.debug('Received SIGTERM') |
mbligh | ff7d61f | 2008-12-22 14:53:35 +0000 | [diff] [blame] | 60 | if pid_file_manager: |
| 61 | pid_file_manager.close_file(1, signal.SIGTERM) |
Simran Basi | 49e21e6 | 2013-10-17 12:40:33 -0700 | [diff] [blame] | 62 | logging.debug('Finished writing to pid_file. Killing process.') |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 63 | os.killpg(os.getpgrp(), signal.SIGKILL) |
mbligh | faf0cd4 | 2007-11-19 16:00:24 +0000 | [diff] [blame] | 64 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 65 | # Set signal handler |
mbligh | c229956 | 2009-07-02 19:00:36 +0000 | [diff] [blame] | 66 | signal.signal(signal.SIGTERM, handle_sigterm) |
mbligh | a46678d | 2008-05-01 20:00:01 +0000 | [diff] [blame] | 67 | |
David Rochberg | 8a60d1e | 2011-02-01 14:22:07 -0500 | [diff] [blame] | 68 | # Ignore SIGTTOU's generated by output from forked children. |
| 69 | signal.signal(signal.SIGTTOU, signal.SIG_IGN) |
| 70 | |
Alex Miller | f1af17e | 2013-01-09 22:50:32 -0800 | [diff] [blame] | 71 | # If we received a SIGALARM, let's be loud about it. |
| 72 | signal.signal(signal.SIGALRM, log_alarm) |
| 73 | |
mbligh | a5f5e54 | 2009-12-30 16:57:49 +0000 | [diff] [blame] | 74 | # Server side tests that call shell scripts often depend on $USER being set |
| 75 | # but depending on how you launch your autotest scheduler it may not be set. |
| 76 | os.environ['USER'] = getpass.getuser() |
| 77 | |
mbligh | cce191f | 2008-09-19 20:31:03 +0000 | [diff] [blame] | 78 | if parser.options.machines: |
| 79 | machines = parser.options.machines.replace(',', ' ').strip().split() |
| 80 | else: |
| 81 | machines = [] |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 82 | machines_file = parser.options.machines_file |
mbligh | b2bea30 | 2008-07-24 20:25:57 +0000 | [diff] [blame] | 83 | label = parser.options.label |
mbligh | 374f341 | 2009-05-13 21:29:45 +0000 | [diff] [blame] | 84 | group_name = parser.options.group_name |
mbligh | b2bea30 | 2008-07-24 20:25:57 +0000 | [diff] [blame] | 85 | user = parser.options.user |
| 86 | client = parser.options.client |
| 87 | server = parser.options.server |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 88 | install_before = parser.options.install_before |
mbligh | b2bea30 | 2008-07-24 20:25:57 +0000 | [diff] [blame] | 89 | install_after = parser.options.install_after |
| 90 | verify = parser.options.verify |
| 91 | repair = parser.options.repair |
showard | 45ae819 | 2008-11-05 19:32:53 +0000 | [diff] [blame] | 92 | cleanup = parser.options.cleanup |
Alex Miller | cb79ba7 | 2013-05-29 14:43:00 -0700 | [diff] [blame] | 93 | provision = parser.options.provision |
Dan Shi | 07e09af | 2013-04-12 09:31:29 -0700 | [diff] [blame] | 94 | reset = parser.options.reset |
mbligh | b2bea30 | 2008-07-24 20:25:57 +0000 | [diff] [blame] | 95 | no_tee = parser.options.no_tee |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 96 | parse_job = parser.options.parse_job |
mbligh | e7d9c60 | 2009-07-02 19:02:33 +0000 | [diff] [blame] | 97 | execution_tag = parser.options.execution_tag |
| 98 | if not execution_tag: |
| 99 | execution_tag = parse_job |
jadmanski | fbc1f0a | 2008-07-09 14:12:54 +0000 | [diff] [blame] | 100 | host_protection = parser.options.host_protection |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 101 | ssh_user = parser.options.ssh_user |
| 102 | ssh_port = parser.options.ssh_port |
| 103 | ssh_pass = parser.options.ssh_pass |
jadmanski | def0c3c | 2009-03-25 20:07:10 +0000 | [diff] [blame] | 104 | collect_crashinfo = parser.options.collect_crashinfo |
mbligh | e0cbc91 | 2010-03-11 18:03:07 +0000 | [diff] [blame] | 105 | control_filename = parser.options.control_filename |
Scott Zawalski | 91493c8 | 2013-01-25 16:15:20 -0500 | [diff] [blame] | 106 | test_retry = parser.options.test_retry |
beeps | cb6f1e2 | 2013-06-28 19:14:10 -0700 | [diff] [blame] | 107 | verify_job_repo_url = parser.options.verify_job_repo_url |
Christopher Wiley | f594c5e | 2013-07-03 18:25:30 -0700 | [diff] [blame] | 108 | skip_crash_collection = parser.options.skip_crash_collection |
Aviv Keshet | 18ee314 | 2013-08-12 15:01:51 -0700 | [diff] [blame] | 109 | ssh_verbosity = int(parser.options.ssh_verbosity) |
Fang Deng | 6cc20de | 2013-09-06 15:47:32 -0700 | [diff] [blame] | 110 | ssh_options = parser.options.ssh_options |
mbligh | a46678d | 2008-05-01 20:00:01 +0000 | [diff] [blame] | 111 | |
mbligh | b2bea30 | 2008-07-24 20:25:57 +0000 | [diff] [blame] | 112 | # can't be both a client and a server side test |
| 113 | if client and server: |
Eric Li | 861b2d5 | 2011-02-04 14:50:35 -0800 | [diff] [blame] | 114 | parser.parser.error("Can not specify a test as both server and client!") |
mbligh | b2bea30 | 2008-07-24 20:25:57 +0000 | [diff] [blame] | 115 | |
Alex Miller | cb79ba7 | 2013-05-29 14:43:00 -0700 | [diff] [blame] | 116 | if provision and client: |
| 117 | parser.parser.error("Cannot specify provisioning and client!") |
| 118 | |
| 119 | is_special_task = (verify or repair or cleanup or collect_crashinfo or |
Dan Shi | 07e09af | 2013-04-12 09:31:29 -0700 | [diff] [blame] | 120 | provision or reset) |
Alex Miller | cb79ba7 | 2013-05-29 14:43:00 -0700 | [diff] [blame] | 121 | if len(parser.args) < 1 and not is_special_task: |
Eric Li | 861b2d5 | 2011-02-04 14:50:35 -0800 | [diff] [blame] | 122 | parser.parser.error("Missing argument: control file") |
mbligh | a46678d | 2008-05-01 20:00:01 +0000 | [diff] [blame] | 123 | |
Aviv Keshet | 18ee314 | 2013-08-12 15:01:51 -0700 | [diff] [blame] | 124 | if ssh_verbosity > 0: |
| 125 | # ssh_verbosity is an integer between 0 and 3, inclusive |
| 126 | ssh_verbosity_flag = '-' + 'v' * ssh_verbosity |
Fang Deng | d1c2b73 | 2013-08-20 12:59:46 -0700 | [diff] [blame] | 127 | else: |
| 128 | ssh_verbosity_flag = '' |
Aviv Keshet | 18ee314 | 2013-08-12 15:01:51 -0700 | [diff] [blame] | 129 | |
showard | 45ae819 | 2008-11-05 19:32:53 +0000 | [diff] [blame] | 130 | # We have a control file unless it's just a verify/repair/cleanup job |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 131 | if len(parser.args) > 0: |
| 132 | control = parser.args[0] |
| 133 | else: |
| 134 | control = None |
mbligh | a46678d | 2008-05-01 20:00:01 +0000 | [diff] [blame] | 135 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 136 | if machines_file: |
| 137 | machines = [] |
| 138 | for m in open(machines_file, 'r').readlines(): |
| 139 | # remove comments, spaces |
| 140 | m = re.sub('#.*', '', m).strip() |
| 141 | if m: |
| 142 | machines.append(m) |
| 143 | print "Read list of machines from file: %s" % machines_file |
| 144 | print ','.join(machines) |
mbligh | a46678d | 2008-05-01 20:00:01 +0000 | [diff] [blame] | 145 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 146 | if machines: |
| 147 | for machine in machines: |
| 148 | if not machine or re.search('\s', machine): |
Eric Li | 861b2d5 | 2011-02-04 14:50:35 -0800 | [diff] [blame] | 149 | parser.parser.error("Invalid machine: %s" % str(machine)) |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 150 | machines = list(set(machines)) |
| 151 | machines.sort() |
mbligh | a46678d | 2008-05-01 20:00:01 +0000 | [diff] [blame] | 152 | |
mbligh | 374f341 | 2009-05-13 21:29:45 +0000 | [diff] [blame] | 153 | if group_name and len(machines) < 2: |
Eric Li | 861b2d5 | 2011-02-04 14:50:35 -0800 | [diff] [blame] | 154 | parser.parser.error("-G %r may only be supplied with more than one machine." |
mbligh | 374f341 | 2009-05-13 21:29:45 +0000 | [diff] [blame] | 155 | % group_name) |
mbligh | 374f341 | 2009-05-13 21:29:45 +0000 | [diff] [blame] | 156 | |
Christopher Wiley | 8a91f23 | 2013-07-09 11:02:27 -0700 | [diff] [blame] | 157 | kwargs = {'group_name': group_name, 'tag': execution_tag, |
| 158 | 'disable_sysinfo': parser.options.disable_sysinfo} |
mbligh | e0cbc91 | 2010-03-11 18:03:07 +0000 | [diff] [blame] | 159 | if control_filename: |
| 160 | kwargs['control_filename'] = control_filename |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 161 | job = server_job.server_job(control, parser.args[1:], results, label, |
| 162 | user, machines, client, parse_job, |
Fang Deng | d1c2b73 | 2013-08-20 12:59:46 -0700 | [diff] [blame] | 163 | ssh_user, ssh_port, ssh_pass, |
Aviv Keshet | c5947fa | 2013-09-04 14:06:29 -0700 | [diff] [blame] | 164 | ssh_verbosity_flag, ssh_options, |
| 165 | test_retry, **kwargs) |
showard | 75cdfee | 2009-06-10 17:40:41 +0000 | [diff] [blame] | 166 | job.logging.start_logging() |
mbligh | 4608b00 | 2010-01-05 18:22:35 +0000 | [diff] [blame] | 167 | job.init_parser() |
mbligh | a46678d | 2008-05-01 20:00:01 +0000 | [diff] [blame] | 168 | |
mbligh | 161fe6f | 2008-06-19 16:26:04 +0000 | [diff] [blame] | 169 | # perform checks |
| 170 | job.precheck() |
| 171 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 172 | # run the job |
| 173 | exit_code = 0 |
| 174 | try: |
mbligh | 332000a | 2009-06-08 16:47:28 +0000 | [diff] [blame] | 175 | try: |
| 176 | if repair: |
| 177 | job.repair(host_protection) |
| 178 | elif verify: |
| 179 | job.verify() |
Alex Miller | cb79ba7 | 2013-05-29 14:43:00 -0700 | [diff] [blame] | 180 | elif provision: |
| 181 | job.provision(provision) |
Dan Shi | 07e09af | 2013-04-12 09:31:29 -0700 | [diff] [blame] | 182 | elif reset: |
| 183 | job.reset() |
mbligh | 332000a | 2009-06-08 16:47:28 +0000 | [diff] [blame] | 184 | else: |
jadmanski | def0c3c | 2009-03-25 20:07:10 +0000 | [diff] [blame] | 185 | job.run(cleanup, install_before, install_after, |
beeps | cb6f1e2 | 2013-06-28 19:14:10 -0700 | [diff] [blame] | 186 | verify_job_repo_url=verify_job_repo_url, |
Christopher Wiley | f594c5e | 2013-07-03 18:25:30 -0700 | [diff] [blame] | 187 | only_collect_crashinfo=collect_crashinfo, |
| 188 | skip_crash_collection=skip_crash_collection) |
mbligh | 332000a | 2009-06-08 16:47:28 +0000 | [diff] [blame] | 189 | finally: |
| 190 | while job.hosts: |
| 191 | host = job.hosts.pop() |
| 192 | host.close() |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 193 | except: |
jadmanski | 27b37ea | 2008-10-29 23:54:31 +0000 | [diff] [blame] | 194 | exit_code = 1 |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 195 | traceback.print_exc() |
mbligh | a46678d | 2008-05-01 20:00:01 +0000 | [diff] [blame] | 196 | |
mbligh | ff7d61f | 2008-12-22 14:53:35 +0000 | [diff] [blame] | 197 | if pid_file_manager: |
| 198 | pid_file_manager.num_tests_failed = job.num_tests_failed |
| 199 | pid_file_manager.close_file(exit_code) |
jadmanski | e0dffc3 | 2008-12-15 17:30:30 +0000 | [diff] [blame] | 200 | job.cleanup_parser() |
showard | 21baa45 | 2008-10-21 00:08:39 +0000 | [diff] [blame] | 201 | |
jadmanski | 27b37ea | 2008-10-29 23:54:31 +0000 | [diff] [blame] | 202 | sys.exit(exit_code) |
mbligh | a46678d | 2008-05-01 20:00:01 +0000 | [diff] [blame] | 203 | |
| 204 | |
| 205 | def main(): |
Dan Shi | a1ecd5c | 2013-06-06 11:21:31 -0700 | [diff] [blame] | 206 | # White list of tests with run time measurement enabled. |
| 207 | measure_run_time_tests_names = global_config.global_config.get_config_value( |
| 208 | 'AUTOSERV', 'measure_run_time_tests', type=str) |
| 209 | if measure_run_time_tests_names: |
| 210 | measure_run_time_tests = [t.strip() for t in |
| 211 | measure_run_time_tests_names.split(',')] |
| 212 | else: |
| 213 | measure_run_time_tests = [] |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 214 | # grab the parser |
| 215 | parser = autoserv_parser.autoserv_parser |
mbligh | a5cb406 | 2009-02-17 15:53:39 +0000 | [diff] [blame] | 216 | parser.parse_args() |
mbligh | a46678d | 2008-05-01 20:00:01 +0000 | [diff] [blame] | 217 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 218 | if len(sys.argv) == 1: |
| 219 | parser.parser.print_help() |
| 220 | sys.exit(1) |
mbligh | a6f1308 | 2008-06-05 23:53:46 +0000 | [diff] [blame] | 221 | |
showard | 75cdfee | 2009-06-10 17:40:41 +0000 | [diff] [blame] | 222 | if parser.options.no_logging: |
| 223 | results = None |
| 224 | else: |
| 225 | results = parser.options.results |
mbligh | 80e1eba | 2008-11-19 00:26:18 +0000 | [diff] [blame] | 226 | if not results: |
| 227 | results = 'results.' + time.strftime('%Y-%m-%d-%H.%M.%S') |
| 228 | results = os.path.abspath(results) |
showard | 566d3c0 | 2010-01-12 18:57:01 +0000 | [diff] [blame] | 229 | resultdir_exists = False |
| 230 | for filename in ('control.srv', 'status.log', '.autoserv_execute'): |
| 231 | if os.path.exists(os.path.join(results, filename)): |
| 232 | resultdir_exists = True |
mbligh | 4608b00 | 2010-01-05 18:22:35 +0000 | [diff] [blame] | 233 | if not parser.options.use_existing_results and resultdir_exists: |
mbligh | 80e1eba | 2008-11-19 00:26:18 +0000 | [diff] [blame] | 234 | error = "Error: results directory already exists: %s\n" % results |
| 235 | sys.stderr.write(error) |
| 236 | sys.exit(1) |
mbligh | a788dc4 | 2009-03-26 21:10:16 +0000 | [diff] [blame] | 237 | |
| 238 | # Now that we certified that there's no leftover results dir from |
| 239 | # previous jobs, lets create the result dir since the logging system |
| 240 | # needs to create the log file in there. |
| 241 | if not os.path.isdir(results): |
| 242 | os.makedirs(results) |
showard | 75cdfee | 2009-06-10 17:40:41 +0000 | [diff] [blame] | 243 | |
| 244 | logging_manager.configure_logging( |
| 245 | server_logging_config.ServerLoggingConfig(), results_dir=results, |
showard | 10d8417 | 2009-06-18 23:16:50 +0000 | [diff] [blame] | 246 | use_console=not parser.options.no_tee, |
| 247 | verbose=parser.options.verbose, |
| 248 | no_console_prefix=parser.options.no_console_prefix) |
showard | 75cdfee | 2009-06-10 17:40:41 +0000 | [diff] [blame] | 249 | if results: |
mbligh | a788dc4 | 2009-03-26 21:10:16 +0000 | [diff] [blame] | 250 | logging.info("Results placed in %s" % results) |
mbligh | 1071763 | 2008-11-19 00:21:57 +0000 | [diff] [blame] | 251 | |
mbligh | 4608b00 | 2010-01-05 18:22:35 +0000 | [diff] [blame] | 252 | # wait until now to perform this check, so it get properly logged |
| 253 | if parser.options.use_existing_results and not resultdir_exists: |
| 254 | logging.error("No existing results directory found: %s", results) |
| 255 | sys.exit(1) |
| 256 | |
Aviv Keshet | 5c40ec6 | 2013-08-20 12:11:12 -0700 | [diff] [blame] | 257 | logging.debug('autoserv command was: %s', ' '.join(sys.argv)) |
mbligh | 4608b00 | 2010-01-05 18:22:35 +0000 | [diff] [blame] | 258 | |
mbligh | 80e1eba | 2008-11-19 00:26:18 +0000 | [diff] [blame] | 259 | if parser.options.write_pidfile: |
mbligh | 4608b00 | 2010-01-05 18:22:35 +0000 | [diff] [blame] | 260 | pid_file_manager = pidfile.PidFileManager(parser.options.pidfile_label, |
| 261 | results) |
jadmanski | d5ab8c5 | 2008-12-03 16:27:07 +0000 | [diff] [blame] | 262 | pid_file_manager.open_file() |
mbligh | ff7d61f | 2008-12-22 14:53:35 +0000 | [diff] [blame] | 263 | else: |
| 264 | pid_file_manager = None |
mbligh | a46678d | 2008-05-01 20:00:01 +0000 | [diff] [blame] | 265 | |
jadmanski | f22fea8 | 2008-11-26 20:57:07 +0000 | [diff] [blame] | 266 | autotest.BaseAutotest.set_install_in_tmpdir( |
| 267 | parser.options.install_in_tmpdir) |
| 268 | |
Dan Shi | a1ecd5c | 2013-06-06 11:21:31 -0700 | [diff] [blame] | 269 | timer = None |
| 270 | try: |
| 271 | # Take the first argument as control file name, get the test name from |
| 272 | # the control file. If the test name exists in the list of tests with |
| 273 | # run time measurement enabled, start a timer to begin measurement. |
| 274 | if (len(parser.args) > 0 and parser.args[0] != '' and |
| 275 | parser.options.machines): |
Dan Shi | bbc1613 | 2013-07-09 16:23:59 -0700 | [diff] [blame] | 276 | try: |
| 277 | test_name = control_data.parse_control(parser.args[0], |
| 278 | raise_warnings=True).name |
| 279 | except control_data.ControlVariableException: |
| 280 | logging.debug('Failed to retrieve test name from control file.') |
| 281 | test_name = None |
Dan Shi | a1ecd5c | 2013-06-06 11:21:31 -0700 | [diff] [blame] | 282 | if test_name in measure_run_time_tests: |
| 283 | machines = parser.options.machines.replace(',', ' ' |
| 284 | ).strip().split() |
| 285 | afe = frontend.AFE() |
| 286 | board = server_utils.get_board_from_afe(machines[0], afe) |
| 287 | timer = stats.Timer('autoserv_run_time.%s.%s' % |
| 288 | (board, test_name)) |
| 289 | timer.start() |
| 290 | except control_data.ControlVariableException as e: |
| 291 | logging.error(str(e)) |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 292 | exit_code = 0 |
| 293 | try: |
| 294 | try: |
mbligh | 1071763 | 2008-11-19 00:21:57 +0000 | [diff] [blame] | 295 | run_autoserv(pid_file_manager, results, parser) |
Aviv Keshet | 5c40ec6 | 2013-08-20 12:11:12 -0700 | [diff] [blame] | 296 | except SystemExit as e: |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 297 | exit_code = e.code |
Aviv Keshet | 5c40ec6 | 2013-08-20 12:11:12 -0700 | [diff] [blame] | 298 | if exit_code: |
| 299 | logging.exception(e) |
| 300 | except Exception as e: |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 301 | # If we don't know what happened, we'll classify it as |
| 302 | # an 'abort' and return 1. |
Aviv Keshet | 5c40ec6 | 2013-08-20 12:11:12 -0700 | [diff] [blame] | 303 | logging.exception(e) |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 304 | exit_code = 1 |
| 305 | finally: |
mbligh | ff7d61f | 2008-12-22 14:53:35 +0000 | [diff] [blame] | 306 | if pid_file_manager: |
| 307 | pid_file_manager.close_file(exit_code) |
Dan Shi | a1ecd5c | 2013-06-06 11:21:31 -0700 | [diff] [blame] | 308 | if timer: |
| 309 | timer.stop() |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 310 | sys.exit(exit_code) |
mbligh | faf0cd4 | 2007-11-19 16:00:24 +0000 | [diff] [blame] | 311 | |
mbligh | bb42185 | 2008-03-11 22:36:16 +0000 | [diff] [blame] | 312 | |
mbligh | a46678d | 2008-05-01 20:00:01 +0000 | [diff] [blame] | 313 | if __name__ == '__main__': |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 314 | main() |