blob: b444ec8e57bf6a25a5f4d29054f270cda7c46c2c [file] [log] [blame]
mbligh6203ace2007-10-04 21:54:24 +00001#!/usr/bin/python -u
mblighdcd57a82007-07-11 23:06:47 +00002#
3# Copyright 2007 Google Inc. Released under the GPL v2
4
mblighc8949b82007-07-23 16:33:58 +00005"""
mblighf1c52842007-10-16 15:21:38 +00006Run an control file through the server side engine
mblighdcd57a82007-07-11 23:06:47 +00007"""
8
mblighf1c52842007-10-16 15:21:38 +00009__author__ = """\
10mbligh@google.com (Martin J. Bligh)
mblighc8949b82007-07-23 16:33:58 +000011"""
mblighdcd57a82007-07-11 23:06:47 +000012
mblighf5427bb2008-04-09 15:55:57 +000013import sys, os, re, traceback, signal
mbligh0c3548d2008-02-01 18:08:53 +000014
mblighf5427bb2008-04-09 15:55:57 +000015import common
16from autotest_lib.server import server_job, utils
mblighb7dcc7f2008-06-02 19:34:25 +000017from autotest_lib.server import autoserv_parser
mbligh29aa9702007-08-09 22:41:43 +000018
mbligh9d1c7302007-10-07 18:53:13 +000019
mbligha46678d2008-05-01 20:00:01 +000020class PidFileManager(object):
21 pid_file = None
mbligh74fc0462007-11-05 20:24:17 +000022
mbligha46678d2008-05-01 20:00:01 +000023 def open_pid_file(self, results_dir):
24 pid_file_path = os.path.join(results_dir, '.autoserv_execute')
25 assert not os.path.exists(pid_file_path)
26 self.pid_file = open(pid_file_path, 'w')
27 self.pid_file.write(str(os.getpid()) + '\n')
28 self.pid_file.flush()
mblighc99add62007-09-27 17:02:58 +000029
mblighc99add62007-09-27 17:02:58 +000030
mbligha46678d2008-05-01 20:00:01 +000031 def close_pid_file(self, exit_code, signal_code=0):
32 if not self.pid_file:
33 return
34 real_exit_code = (exit_code << 8) | (signal_code & 0xFF)
35 self.pid_file.write(str(real_exit_code) + '\n')
36 self.pid_file.close()
37 self.pid_file = None
mblighc99add62007-09-27 17:02:58 +000038
mbligh842c6592007-11-05 18:27:23 +000039
mbligha46678d2008-05-01 20:00:01 +000040def run_autoserv(pid_file_manager, results, parser):
41 # send stdin to /dev/null
42 dev_null = os.open(os.devnull, os.O_RDONLY)
43 os.dup2(dev_null, sys.stdin.fileno())
44 os.close(dev_null)
mblighdbf37612007-11-24 19:38:11 +000045
mbligha46678d2008-05-01 20:00:01 +000046 # Create separate process group
47 os.setpgrp()
mbligh1d42d4e2007-11-05 22:42:00 +000048
mbligha46678d2008-05-01 20:00:01 +000049 # Implement SIGTERM handler
50 def handle_sigint(signum, frame):
51 pid_file_manager.close_pid_file(1, signal.SIGTERM)
52 os.killpg(os.getpgrp(), signal.SIGKILL)
mblighfaf0cd42007-11-19 16:00:24 +000053
mbligha46678d2008-05-01 20:00:01 +000054 # Set signal handler
55 signal.signal(signal.SIGTERM, handle_sigint)
mblighe25fd5b2008-01-22 17:23:37 +000056
mbligha46678d2008-05-01 20:00:01 +000057 # Get a useful value for running 'USER'
58 realuser = os.environ.get('USER')
59 if not realuser:
60 realuser = 'anonymous'
61
62 machines = parser.parse_opts_param('-m', None, split = ',')
63 machines_file = parser.parse_opts_param('-M', None)
64 label = parser.parse_opts_param('-l', '')
65 user = parser.parse_opts_param('-u', realuser)
66 client = parser.parse_opts('-c')
67 reboot = parser.parse_opts('-b')
68 install_before = parser.parse_opts('-i')
69 install_after = parser.parse_opts('-I')
70 verify = parser.parse_opts('-v')
71 repair = parser.parse_opts('-R')
72 no_tee = parser.parse_opts('-n')
73 help = parser.parse_opts('-h') or parser.parse_opts('--help')
74 parse_job = parser.parse_opts_param('-P', '')
mbligh1965dfa2008-06-04 19:58:37 +000075 ssh_user = parser.parse_opts_param('--ssh-user', 'root')
76 ssh_port = int(parser.parse_opts_param('--ssh-port', '22'))
77 ssh_pass = parser.parse_opts_param('--ssh-pass', '')
mbligha46678d2008-05-01 20:00:01 +000078
79 if help is True:
80 print usage
81 sys.exit(0)
82
83 if len(parser.args) < 1 and not verify and not repair:
84 print usage
85 sys.exit(1)
86
87 # We have a control file unless it's just a verify/repair job
88 if len(parser.args) > 0:
89 control = parser.args[0]
mblighbb421852008-03-11 22:36:16 +000090 else:
mbligha46678d2008-05-01 20:00:01 +000091 control = None
92
93 if machines_file:
94 machines = []
95 for m in open(machines_file, 'r').readlines():
96 # remove comments, spaces
97 m = re.sub('#.*', '', m).strip()
98 if m:
99 machines.append(m)
100 print "Read list of machines from file: %s" % machines_file
101 print ','.join(machines)
102
103 if machines:
104 for machine in machines:
105 if not machine or re.search('\s', machine):
106 print "Invalid machine %s" % str(machine)
107 sys.exit(1)
108 machines = list(set(machines))
109 machines.sort()
110
111 job = server_job.server_job(control, parser.args[1:], results, label,
mbligh1965dfa2008-06-04 19:58:37 +0000112 user, machines, client, parse_job,
113 ssh_user, ssh_port, ssh_pass)
mbligha46678d2008-05-01 20:00:01 +0000114 debug_dir = os.path.join(results, 'debug')
115 stdout = os.path.join(debug_dir, 'autoserv.stdout')
116 stderr = os.path.join(debug_dir, 'autoserv.stderr')
117 if no_tee:
118 job.stdout.redirect(stdout)
119 job.stderr.redirect(stderr)
120 else:
121 job.stdout.tee_redirect(stdout)
122 job.stderr.tee_redirect(stderr)
123
124 # run the job
125 exit_code = 0
126 try:
127 if repair:
128 job.repair()
129 elif verify:
130 job.verify()
131 else:
132 try:
133 job.run(reboot, install_before, install_after)
134 finally:
135 job.cleanup_parser()
136 except:
137 job.aborted = True
138 traceback.print_exc()
139
140 if getattr(job, 'aborted', False):
141 sys.exit(1)
142
143
144def main():
145 pid_file_manager = PidFileManager()
146
mblighb7dcc7f2008-06-02 19:34:25 +0000147 # grab the parser
148 parser = autoserv_parser.autoserv_parser
mbligha46678d2008-05-01 20:00:01 +0000149
150 results = parser.parse_opts_param('-r', '.')
151 results = os.path.abspath(results)
152 write_pidfile = parser.parse_opts('-p')
153 if write_pidfile:
154 pid_file_manager.open_pid_file(results)
155
156 exit_code = 0
157 try:
mbligh6437ff52008-04-17 15:24:38 +0000158 try:
mbligha46678d2008-05-01 20:00:01 +0000159 run_autoserv(pid_file_manager, results, parser)
160 except SystemExit, e:
161 exit_code = e.code
162 except:
jadmanskia6f2a482008-05-01 20:11:50 +0000163 traceback.print_exc()
mbligha46678d2008-05-01 20:00:01 +0000164 # If we don't know what happened, we'll classify it as
165 # an 'abort' and return 1.
166 exit_code = 1
167 finally:
168 pid_file_manager.close_pid_file(exit_code)
169 sys.exit(exit_code)
mblighfaf0cd42007-11-19 16:00:24 +0000170
mblighbb421852008-03-11 22:36:16 +0000171
mbligha46678d2008-05-01 20:00:01 +0000172if __name__ == '__main__':
173 main()