blob: 4f32495906e5fb436eba4f2ae27f2cd1d89ceb58 [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
mbligh0c3548d2008-02-01 18:08:53 +000013from common.check_version import check_python_version
14check_python_version()
15
mblighf7243e12008-03-05 16:01:21 +000016import sys, os, re, server_job, hosts.site_host, utils, traceback, signal
17
mblighc251a1f2008-03-26 17:14:15 +000018
19# send stdin to /dev/null
20dev_null = os.open(os.devnull, os.O_RDONLY)
21os.dup2(dev_null, sys.stdin.fileno())
22os.close(dev_null)
23
24
mblighbb421852008-03-11 22:36:16 +000025class PidFileManager(object):
26 pid_file = None
27
28 def open_pid_file(self, results_dir):
29 pid_file_path = os.path.join(results_dir, '.autoserv_execute')
30 assert not os.path.exists(pid_file_path)
31 self.pid_file = open(pid_file_path, 'w')
32 self.pid_file.write(str(os.getpid()) + '\n')
33 self.pid_file.flush()
34
35
36 def close_pid_file(self, exit_code, signal_code=0):
37 if not self.pid_file:
38 return
39 real_exit_code = (exit_code << 8) | (signal_code & 0xFF)
40 self.pid_file.write(str(real_exit_code) + '\n')
41 self.pid_file.close()
42
43
44pid_file_manager = PidFileManager()
45
46
mblighf7243e12008-03-05 16:01:21 +000047# Create separate process group
48os.setpgrp()
49
mblighbb421852008-03-11 22:36:16 +000050# Implement SIGTERM handler
mblighf7243e12008-03-05 16:01:21 +000051def handle_sigint(signum, frame):
mblighbb421852008-03-11 22:36:16 +000052 pid_file_manager.close_pid_file(1, signal.SIGTERM)
mbligh38c2d032008-03-07 00:26:57 +000053 os.killpg(os.getpgrp(), signal.SIGKILL)
mblighbb421852008-03-11 22:36:16 +000054
mblighf7243e12008-03-05 16:01:21 +000055# Set signal handler
56signal.signal(signal.SIGTERM, handle_sigint)
57
mblighdcd57a82007-07-11 23:06:47 +000058
mblighf1c52842007-10-16 15:21:38 +000059usage = """\
60usage: autoserv
mbligh9e618782008-03-13 15:35:09 +000061 [-h, --help] # This help message
mblighf36243d2007-10-30 15:36:16 +000062 [-m machine,[machine,...]] # list of machines to pass to control file
mbligh842c6592007-11-05 18:27:23 +000063 [-M machines_file] # list of machines (from a file)
mblighf1c52842007-10-16 15:21:38 +000064 [-c] # control file is a client side control
65 [-r resultsdir] # specify results directory (default '.')
mblighf36243d2007-10-30 15:36:16 +000066 [-i] # reinstall machines before running the job
67 [-I] # reinstall machines after running the job
68 [-b] # reboot all specified machines after the job
mbligh18420c22007-10-16 22:27:14 +000069 [-l label] # label for the job (arbitrary string)
mblighf1c52842007-10-16 15:21:38 +000070 [-u user] # username for the job (email address)
mbligh1d42d4e2007-11-05 22:42:00 +000071 [-v] # verify the machines only
72 [-R] # repair the machines
mblighe5cd0fd2008-01-26 23:07:31 +000073 [-n] # no teeing the status to stdout/stderr
mblighbb421852008-03-11 22:36:16 +000074 [-p] # write pidfile (.autoserv_execute)
mblighf1c52842007-10-16 15:21:38 +000075 <control file> # name of the control file to run
76 [args ...] # args to pass through to the control file
mbligh29aa9702007-08-09 22:41:43 +000077"""
78
mbligh9d1c7302007-10-07 18:53:13 +000079args = sys.argv[1:]
mbligh40f122a2007-11-03 23:08:46 +000080parser = utils.AutoservOptionParser(args)
mbligh9d1c7302007-10-07 18:53:13 +000081
mbligh74fc0462007-11-05 20:24:17 +000082# Get a useful value for running 'USER'
83realuser = os.environ.get('USER')
84if not realuser:
85 realuser = 'anonymous'
86
mbligh40f122a2007-11-03 23:08:46 +000087machines = parser.parse_opts_param('-m', None, split = ',')
mbligh842c6592007-11-05 18:27:23 +000088machines_file = parser.parse_opts_param('-M', None)
mbligh40f122a2007-11-03 23:08:46 +000089results = parser.parse_opts_param('-r', os.path.abspath('.'))
mbligh1d42d4e2007-11-05 22:42:00 +000090results = os.path.abspath(results)
mbligh40f122a2007-11-03 23:08:46 +000091label = parser.parse_opts_param('-l', '')
mbligh74fc0462007-11-05 20:24:17 +000092user = parser.parse_opts_param('-u', realuser)
mbligh40f122a2007-11-03 23:08:46 +000093client = parser.parse_opts('-c')
94reboot = parser.parse_opts('-b')
95install_before = parser.parse_opts('-i')
96install_after = parser.parse_opts('-I')
mbligh1d42d4e2007-11-05 22:42:00 +000097verify = parser.parse_opts('-v')
98repair = parser.parse_opts('-R')
mblighe5cd0fd2008-01-26 23:07:31 +000099no_tee = parser.parse_opts('-n')
mbligh9e618782008-03-13 15:35:09 +0000100help = parser.parse_opts('-h') or parser.parse_opts('--help')
mblighbb421852008-03-11 22:36:16 +0000101write_pidfile = parser.parse_opts('-p')
mblighc99add62007-09-27 17:02:58 +0000102
mbligh9e618782008-03-13 15:35:09 +0000103if help is True:
104 print usage
105 sys.exit(0)
mblighc99add62007-09-27 17:02:58 +0000106
mbligh1d42d4e2007-11-05 22:42:00 +0000107if len(parser.args) < 1 and not verify and not repair:
mbligh4ef86232007-10-23 00:09:53 +0000108 print usage
109 sys.exit(-1)
mblighc99add62007-09-27 17:02:58 +0000110
mbligh842c6592007-11-05 18:27:23 +0000111if machines_file:
112 machines = []
113 for m in open(machines_file, 'r').readlines():
114 m = re.sub('#.*', '', m).strip() # remove comments, spaces
115 if m:
116 machines.append(m)
117 print "Read list of machines from file: %s" % machines_file
118 print ','.join(machines)
119
mblighdbf37612007-11-24 19:38:11 +0000120if machines:
121 for machine in machines:
122 if not machine or re.search('\s', machine):
123 print "Invalid machine %s" % str(machine)
124 sys.exit(1)
mbligh535a4542008-02-11 16:26:36 +0000125 machines = list(set(machines))
126 machines.sort()
mblighdbf37612007-11-24 19:38:11 +0000127
mblighe25fd5b2008-01-22 17:23:37 +0000128# We have a control file unless it's just a verify/repair job
129if len(parser.args) > 0:
130 control = parser.args[0]
131else:
132 control = None
mbligh1d42d4e2007-11-05 22:42:00 +0000133
mblighe25fd5b2008-01-22 17:23:37 +0000134job = server_job.server_job(control, parser.args[1:], results, label,
mblighe8b37a92007-12-19 15:54:11 +0000135 user, machines, client)
mblighe5cd0fd2008-01-26 23:07:31 +0000136debug_dir = os.path.join(results, 'debug')
137if no_tee:
138 job.stdout.redirect(os.path.join(debug_dir, 'autoserv.stdout'))
139 job.stderr.redirect(os.path.join(debug_dir, 'autoserv.stderr'))
140else:
141 job.stdout.tee_redirect(os.path.join(debug_dir, 'autoserv.stdout'))
142 job.stderr.tee_redirect(os.path.join(debug_dir, 'autoserv.stderr'))
mblighfaf0cd42007-11-19 16:00:24 +0000143
mblighbb421852008-03-11 22:36:16 +0000144if write_pidfile:
145 pid_file_manager.open_pid_file(results)
mblighe25fd5b2008-01-22 17:23:37 +0000146
mblighbb421852008-03-11 22:36:16 +0000147# run the job
148exit_code = 0
mblighfaf0cd42007-11-19 16:00:24 +0000149try:
mblighbb421852008-03-11 22:36:16 +0000150 if repair:
151 job.repair()
152 elif verify:
153 job.verify()
154 else:
155 job.run(reboot, install_before, install_after)
mblighfaf0cd42007-11-19 16:00:24 +0000156except:
mbligh394ff512008-02-21 16:53:11 +0000157 job.aborted = True
mblighfaf0cd42007-11-19 16:00:24 +0000158 traceback.print_exc()
159
mblighfaf0cd42007-11-19 16:00:24 +0000160if getattr(job, 'aborted', False):
mblighbb421852008-03-11 22:36:16 +0000161 exit_code = 1
162pid_file_manager.close_pid_file(exit_code)
163
164sys.exit(exit_code)