source: fedd/fedd.py @ 830ca65

axis_examplecompt_changesinfo-opsversion-1.30version-2.00version-3.01version-3.02
Last change on this file since 830ca65 was 6a0c9f4, checked in by Ted Faber <faber@…>, 16 years ago

More namespace cleanup

  • Property mode set to 100755
File size: 5.5 KB
RevLine 
[6ff0b91]1#!/usr/local/bin/python
2
[2729e48]3import os,sys
[6ff0b91]4
5from optparse import OptionParser
6
[6a0c9f4]7from fedd import config_parser
[ec4fb42]8from fedd.server import server, xmlrpc_handler, soap_handler
9from fedd.util import fedd_ssl_context
[7454054]10from fedd.deter_impl import new_feddservice
[6ff0b91]11
[6a0c9f4]12from threading import Lock, Thread
[11a08b0]13from signal import signal, pause, SIGINT, SIGTERM
[d199ced]14from select import select, error
[0ea11af]15from time import sleep
[11a08b0]16import logging
[19cc408]17
[6ff0b91]18class 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]36servers_active = True       # Sub-servers run while this is True
[ec4fb42]37servers = [ ]               # server instances instantiated from services
[0ea11af]38servers_lock = Lock()       # Lock to manipulate servers from sub-server threads
[11a08b0]39
40def 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]51def 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]78opts, args = fedd_opts().parse_args()
79
[0ea11af]80# Logging setup
[11a08b0]81flog = logging.getLogger("fedd")
[0ea11af]82ffmt = logging.Formatter("%(asctime)s %(name)s %(message)s",
83        '%d %b %y %H:%M:%S')
[11a08b0]84
85if opts.logfile: fh = logging.FileHandler(opts.logfile)
86else: fh = logging.StreamHandler(sys.stdout)
87
88# The handler will print anything, setting the logger level will affect what
89# gets recorded.
90fh.setLevel(logging.DEBUG)
91
92if opts.debug: flog.setLevel(logging.DEBUG)
93else: flog.setLevel(logging.INFO)
94
95fh.setFormatter(ffmt)
96flog.addHandler(fh)
97
[72ed6e4]98
99
[0ea11af]100# Initialize the implementation
[6ff0b91]101if 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]107else: 
108    sys.exit("--configfile is required")
109
[72ed6e4]110try:
111    impl = new_feddservice(config)
112except 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]117if 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)
120else:
[6ff0b91]121    sys.exit("Must supply certificate file (probably in config)")
122
[2729e48]123if 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]129ctx = None
130while 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]138services = config.get("globals", "services", "23235")
[a97394b]139
[a2da110]140for 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]177servers = [ s for s in servers if s ]
[0ea11af]178
179# Catch signals
[11a08b0]180signal(SIGINT, shutdown)
181signal(SIGTERM, shutdown)
[a97394b]182
[0ea11af]183# Start the servers
[a97394b]184for s in servers:
[0ea11af]185    Thread(target=run_server, args=(s,)).start()
186
187# Main thread waits for signals
188while servers_active:
[d199ced]189    sleep(1.0)
[0ea11af]190
191#Once shutdown starts wait for all the servers to terminate.
192while 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
Note: See TracBrowser for help on using the repository browser.