[6ff0b91] | 1 | #!/usr/local/bin/python |
---|
| 2 | |
---|
[2729e48] | 3 | import os,sys |
---|
[6ff0b91] | 4 | |
---|
| 5 | from optparse import OptionParser |
---|
| 6 | |
---|
[6a0c9f4] | 7 | from fedd import config_parser |
---|
[ec4fb42] | 8 | from fedd.server import server, xmlrpc_handler, soap_handler |
---|
| 9 | from fedd.util import fedd_ssl_context |
---|
[7454054] | 10 | from fedd.deter_impl import new_feddservice |
---|
[6ff0b91] | 11 | |
---|
[6a0c9f4] | 12 | from threading import Lock, Thread |
---|
[11a08b0] | 13 | from signal import signal, pause, SIGINT, SIGTERM |
---|
[d199ced] | 14 | from select import select, error |
---|
[0ea11af] | 15 | from time import sleep |
---|
[11a08b0] | 16 | import logging |
---|
[19cc408] | 17 | |
---|
[6ff0b91] | 18 | class fedd_opts(OptionParser): |
---|
| 19 | """Encapsulate option processing in this class, rather than in main""" |
---|
| 20 | def __init__(self): |
---|
| 21 | OptionParser.__init__(self, usage="%prog [opts] (--help for details)", |
---|
| 22 | version="0.1") |
---|
| 23 | |
---|
[a2da110] | 24 | self.set_defaults(logfile=None, debug=0) |
---|
[6ff0b91] | 25 | |
---|
| 26 | self.add_option("-d", "--debug", action="count", dest="debug", |
---|
| 27 | help="Set debug. Repeat for more information") |
---|
| 28 | self.add_option("-f", "--configfile", action="store", |
---|
[a2da110] | 29 | default="/usr/local/etc/fedd.conf", |
---|
[6ff0b91] | 30 | dest="configfile", help="Configuration file (required)") |
---|
[11a08b0] | 31 | self.add_option("-l", "--logfile", action="store", dest="logfile", |
---|
| 32 | help="File to send log messages to") |
---|
[6ff0b91] | 33 | self.add_option("--trace", action="store_const", dest="tracefile", |
---|
| 34 | const=sys.stderr, help="Print SOAP exchange to stderr") |
---|
| 35 | |
---|
[0ea11af] | 36 | servers_active = True # Sub-servers run while this is True |
---|
[ec4fb42] | 37 | servers = [ ] # server instances instantiated from services |
---|
[0ea11af] | 38 | servers_lock = Lock() # Lock to manipulate servers from sub-server threads |
---|
[11a08b0] | 39 | |
---|
| 40 | def shutdown(sig, frame): |
---|
[0ea11af] | 41 | """ |
---|
| 42 | On a signal, stop running sub-servers. |
---|
| 43 | |
---|
| 44 | This is connected to signals below |
---|
| 45 | """ |
---|
[11a08b0] | 46 | global servers_active, flog |
---|
[d199ced] | 47 | |
---|
[11a08b0] | 48 | servers_active = False |
---|
| 49 | flog.info("Received signal %d, shutting down" % sig); |
---|
| 50 | |
---|
[a97394b] | 51 | def run_server(s): |
---|
[0ea11af] | 52 | """ |
---|
| 53 | Operate a subserver, shutting down when servers_active is false. |
---|
| 54 | |
---|
| 55 | Each server (that is host/port/transport triple) has a thread running this |
---|
| 56 | function, so each can handle requests independently. They all call in to |
---|
| 57 | the same implementation, which must manage its own synchronization. |
---|
| 58 | """ |
---|
[11a08b0] | 59 | global servers_active # Not strictly needed: servers_active is only read |
---|
[0ea11af] | 60 | global servers # List of active servers |
---|
| 61 | global servers_lock # Lock to manipulate servers |
---|
[11a08b0] | 62 | |
---|
[0ea11af] | 63 | while servers_active: |
---|
[d199ced] | 64 | try: |
---|
| 65 | i, o, e = select((s,), (), (), 1.0) |
---|
| 66 | if s in i: s.handle_request() |
---|
| 67 | except error: |
---|
| 68 | # The select call seems to get interrupted by signals as well as |
---|
| 69 | # the main thread. This essentially ignores signals in this |
---|
| 70 | # thread. |
---|
| 71 | pass |
---|
[a97394b] | 72 | |
---|
[0ea11af] | 73 | # Done. Remove us from the list |
---|
| 74 | servers_lock.acquire() |
---|
| 75 | servers.remove(s) |
---|
| 76 | servers_lock.release() |
---|
[a97394b] | 77 | |
---|
[6ff0b91] | 78 | opts, args = fedd_opts().parse_args() |
---|
| 79 | |
---|
[0ea11af] | 80 | # Logging setup |
---|
[11a08b0] | 81 | flog = logging.getLogger("fedd") |
---|
[0ea11af] | 82 | ffmt = logging.Formatter("%(asctime)s %(name)s %(message)s", |
---|
| 83 | '%d %b %y %H:%M:%S') |
---|
[11a08b0] | 84 | |
---|
| 85 | if opts.logfile: fh = logging.FileHandler(opts.logfile) |
---|
| 86 | else: fh = logging.StreamHandler(sys.stdout) |
---|
| 87 | |
---|
| 88 | # The handler will print anything, setting the logger level will affect what |
---|
| 89 | # gets recorded. |
---|
| 90 | fh.setLevel(logging.DEBUG) |
---|
| 91 | |
---|
| 92 | if opts.debug: flog.setLevel(logging.DEBUG) |
---|
| 93 | else: flog.setLevel(logging.INFO) |
---|
| 94 | |
---|
| 95 | fh.setFormatter(ffmt) |
---|
| 96 | flog.addHandler(fh) |
---|
| 97 | |
---|
[72ed6e4] | 98 | |
---|
| 99 | |
---|
[0ea11af] | 100 | # Initialize the implementation |
---|
[6ff0b91] | 101 | if opts.configfile != None: |
---|
| 102 | try: |
---|
[ec4fb42] | 103 | config= config_parser() |
---|
[72ed6e4] | 104 | config.read(opts.configfile) |
---|
[ec4fb42] | 105 | except Exception, e: |
---|
| 106 | sys.exit("Cannot parse config file: %s" % e) |
---|
[6ff0b91] | 107 | else: |
---|
| 108 | sys.exit("--configfile is required") |
---|
| 109 | |
---|
[72ed6e4] | 110 | try: |
---|
| 111 | impl = new_feddservice(config) |
---|
| 112 | except RuntimeError, e: |
---|
| 113 | str = getattr(e, 'desc', None) or getattr(e,'message', None) or \ |
---|
| 114 | "No message" |
---|
| 115 | sys.exit("Error configuring fedd: %s" % str) |
---|
| 116 | |
---|
[2729e48] | 117 | if impl.cert_file: |
---|
| 118 | if not os.access(impl.cert_file, os.R_OK): |
---|
| 119 | sys.exit("Cannot read certificate file: %s" % impl.cert_file) |
---|
| 120 | else: |
---|
[6ff0b91] | 121 | sys.exit("Must supply certificate file (probably in config)") |
---|
| 122 | |
---|
[2729e48] | 123 | if impl.trusted_certs: |
---|
| 124 | if not os.access(impl.trusted_certs, os.R_OK): |
---|
| 125 | sys.exit("Cannot read trusted certificate file: %s" % \ |
---|
| 126 | impl.trusted_certs) |
---|
| 127 | |
---|
[0ea11af] | 128 | # Create the SSL credentials |
---|
[2106ed1] | 129 | ctx = None |
---|
| 130 | while ctx == None: |
---|
[6ff0b91] | 131 | try: |
---|
| 132 | ctx = fedd_ssl_context(impl.cert_file, impl.trusted_certs, |
---|
| 133 | password=impl.cert_pwd) |
---|
[2729e48] | 134 | except Exception, e: |
---|
[2106ed1] | 135 | if str(e) != "bad decrypt" or impl.cert_pwd != None: |
---|
[6ff0b91] | 136 | raise |
---|
| 137 | |
---|
[a2da110] | 138 | services = config.get("globals", "services", "23235") |
---|
[a97394b] | 139 | |
---|
[a2da110] | 140 | for s in services.split(","): |
---|
| 141 | s = s.strip() |
---|
| 142 | colons = s.count(":") |
---|
| 143 | try: |
---|
| 144 | if colons == 0: |
---|
| 145 | p = int(s) |
---|
| 146 | h = '' |
---|
| 147 | t = 'soap' |
---|
| 148 | elif colons == 1: |
---|
| 149 | p, t = s.split(":") |
---|
| 150 | p = int(p) |
---|
| 151 | h = '' |
---|
| 152 | elif colons == 2: |
---|
| 153 | h, p, t = s.split(":") |
---|
| 154 | p = int(p) |
---|
| 155 | else: |
---|
| 156 | flog.error("Invalid service specification %s ignored." % s) |
---|
| 157 | continue |
---|
| 158 | except ValueError: |
---|
| 159 | flog.error("Error converting port to integer in %s: spec ignored" % s) |
---|
| 160 | continue |
---|
[a97394b] | 161 | |
---|
[a2da110] | 162 | t = t.lower() |
---|
| 163 | try: |
---|
| 164 | if t == 'soap': |
---|
[ec4fb42] | 165 | servers.append(server((h, p), soap_handler, ctx, impl)) |
---|
[a2da110] | 166 | elif t == 'xmlrpc': |
---|
[ec4fb42] | 167 | servers.append(server((h, p), xmlrpc_handler, ctx, impl)) |
---|
[a2da110] | 168 | else: |
---|
| 169 | flog.error("Invalid transport specification (%s) in service %s" % \ |
---|
| 170 | (t, s)) |
---|
| 171 | continue |
---|
| 172 | except socket_error, e: |
---|
| 173 | flog.error("Cannot create server for %s: %s" % (s, e[1])) |
---|
| 174 | continue |
---|
[11a08b0] | 175 | |
---|
[0ea11af] | 176 | # Make sure that there are no malformed servers in the list |
---|
[a2da110] | 177 | servers = [ s for s in servers if s ] |
---|
[0ea11af] | 178 | |
---|
| 179 | # Catch signals |
---|
[11a08b0] | 180 | signal(SIGINT, shutdown) |
---|
| 181 | signal(SIGTERM, shutdown) |
---|
[a97394b] | 182 | |
---|
[0ea11af] | 183 | # Start the servers |
---|
[a97394b] | 184 | for s in servers: |
---|
[0ea11af] | 185 | Thread(target=run_server, args=(s,)).start() |
---|
| 186 | |
---|
| 187 | # Main thread waits for signals |
---|
| 188 | while servers_active: |
---|
[d199ced] | 189 | sleep(1.0) |
---|
[0ea11af] | 190 | |
---|
| 191 | #Once shutdown starts wait for all the servers to terminate. |
---|
| 192 | while True: |
---|
| 193 | servers_lock.acquire() |
---|
| 194 | if len(servers) == 0: |
---|
| 195 | servers_lock.release() |
---|
| 196 | flog.info("All servers exited. Terminating") |
---|
| 197 | sys.exit(0) |
---|
| 198 | servers_lock.release() |
---|
| 199 | sleep(1) |
---|
[11a08b0] | 200 | |
---|