daemon: Integrate checking for an existing pidfile into daemonize_start().
[openvswitch] / tests / test-daemon.py
1 # Copyright (c) 2010, 2011 Nicira Networks.
2 #
3 # Licensed under the Apache License, Version 2.0 (the "License");
4 # you may not use this file except in compliance with the License.
5 # You may obtain a copy of the License at:
6 #
7 #     http://www.apache.org/licenses/LICENSE-2.0
8 #
9 # Unless required by applicable law or agreed to in writing, software
10 # distributed under the License is distributed on an "AS IS" BASIS,
11 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12 # See the License for the specific language governing permissions and
13 # limitations under the License.
14
15 import getopt
16 import signal
17 import sys
18 import time
19
20 import ovs.daemon
21 import ovs.util
22
23 def handler(signum, frame):
24     raise Exception("Signal handler called with %d" % signum)
25
26 def main(argv):
27
28     signal.signal(signal.SIGHUP, handler)
29
30     try:
31         options, args = getopt.gnu_getopt(
32             argv[1:], 'b', ["bail", "help"] + ovs.daemon.LONG_OPTIONS)
33     except getopt.GetoptError, geo:
34         sys.stderr.write("%s: %s\n" % (ovs.util.PROGRAM_NAME, geo.msg))
35         sys.exit(1)
36
37     bail = False
38     for key, value in options:
39         if key == '--help':
40             usage()
41         elif key in ['-b', '--bail']:
42             bail = True
43         elif not ovs.daemon.parse_opt(key, value):
44             sys.stderr.write("%s: unhandled option %s\n"
45                              % (ovs.util.PROGRAM_NAME, key))
46             sys.exit(1)
47
48     ovs.daemon.daemonize_start()
49     if bail:
50         sys.stderr.write("%s: exiting after daemonize_start() as requested\n"
51                          % ovs.util.PROGRAM_NAME)
52         sys.exit(1)
53     ovs.daemon.daemonize_complete()
54
55     while True:
56         time.sleep(1)
57
58 def usage():
59     sys.stdout.write("""\
60 %s: Open vSwitch daemonization test program for Python
61 usage: %s [OPTIONS]
62 """ % ovs.util.PROGRAM_NAME)
63     ovs.daemon.usage()
64     sys.stdout.write("""
65 Other options:
66   -h, --help              display this help message
67   -b, --bail              exit with an error after daemonize_start()
68 """)
69     sys.exit(0)
70
71 if __name__ == '__main__':
72     try:
73         main(sys.argv)
74     except SystemExit:
75         # Let system.exit() calls complete normally
76         raise
77     except:
78         sys.exit(ovs.daemon.RESTART_EXIT_CODE)