Newer
Older
import logging
import logging.handlers
import ConfigParser
import signal
from optparse import OptionParser
from pwd import getpwnam
from grp import getgrnam
from daemon import DaemonContext
from daemon.pidlockfile import PIDLockFile
'daemonize': False,
'user': '',
'group': '',
'pidfile': '',
'umask': '0177',
'port': 1984,
'verbosity': 0,
'account_db': None, # None = mandatory option
'interface': '127.0.0.1',
'ssl_cert': None,
'ssl_key': None,
}
def run_server(options):
# Setup logging facility:
level = logging.ERROR
verb = int(options['verbosity'])
if verb:
if verb == 1:
level = logging.WARNING
elif verb == 2:
level = logging.INFO
else:
level = logging.DEBUG
logger = logging.getLogger()
logger.setLevel(level)
if options['stdout']:
handler = logging.StreamHandler()
else:
facility = logging.handlers.SysLogHandler.LOG_DAEMON
handler = logging.handlers.SysLogHandler(address='/dev/log',
facility=facility)
fmt = logging.Formatter('cc-server: %(levelname)s %(message)s')
handler.setFormatter(fmt)
logger.addHandler(handler)
server = CCServer(conf_dir=options['account_db'], port=int(options['port']),
address=options['interface'], keyfile=options['ssl_key'],
certfile=options['ssl_cert'])
def shutdown_handler(signum, frame):
'''
Handler called when SIGINT is emitted.
'''
server.manager.shutdown()
logging.info('Server properly exited by SIGINT')
sys.exit(0)
signal.signal(signal.SIGINT, shutdown_handler)
try:
server.run()
except Exception as err:
import traceback
traceback.print_exc(file=sys.stdout)
logging.critical('Server fail: %s' % err)
sys.exit(3)
if __name__ == '__main__':
op = OptionParser(version='%%prog v%s' % __version__)
op.add_option('-c', '--config', default=DEFAULT_CONFIG_FILE,
help='configuration file (default: %default)')
op.add_option('-d', '--daemonize', action='store_true',
help='run as a daemon')
op.add_option('-f', '--foreground', action='store_false', dest='daemonize',
help='don\'t run as a daemon')
op.add_option('-p', '--pidfile', help='write pidfile to the path')
op.add_option('-k', '--umask', help='set the umask of the process')
op.add_option('-u', '--user', help='run as user')
op.add_option('-g', '--group', help='run as group')
op.add_option('-s', '--stdout', action='store_true', default=False,
help='log in stdout instead of syslog')
cliopts, args = op.parse_args()
# Reading the config file:
config = ConfigParser.SafeConfigParser()
config.read(cliopts.config)
try:
options = dict(config.items('server'))
except ConfigParser.NoSectionError:
sys.stderr.write("Configuration error: 'server' section must exist "
"in '%s'\n" % cliopts.config)
sys.exit(1)
# Applying default config file options:
for opt, default in DEFAULT_CONFIGURATION.iteritems():
if opt not in options or not options[opt]:
if default is None:
sys.stderr.write("Configuration error: you must specify '%s' "
"option in '%s' !\n" % (opt, cliopts.config))
sys.exit(1)
else:
options[opt] = default
# Merge cli options and .conf file options:
for opt in ('daemonize', 'pidfile', 'umask', 'user', 'group', 'stdout'):
if getattr(cliopts, opt) is not None:
options[opt] = getattr(cliopts, opt)
# Create option set for the daemonization process:
daemon_opts = {}
daemonize = options['daemonize']
if isinstance(daemonize, str):
daemon_opts['detach_process'] = daemonize
daemon_opts['umask'] = int(options['umask'], 8)
if options['user']:
if options['user'].isdigit():
daemon_opts['uid'] = int(options['user'])
else:
daemon_opts['uid'] = getpwnam(options['user']).pw_uid
if options['group']:
if options['group'].isdigit():
daemon_opts['gid'] = int(options['group'])
daemon_opts['gid'] = getgrnam(options['group']).gr_gid
if not daemonize:
daemon_opts['stderr'] = sys.stderr
daemon_opts['stdout'] = sys.stderr
# I've to write myself the pidfile because the daemon library write it
# after the privilege downgrade:
if options['pidfile']:
pidfile = open(cliopts.pidfile, 'w')
daemon_opts['files_preserve'] = [pidfile]
pidfile = None
with DaemonContext(**daemon_opts):
if pidfile is not None:
pidfile.write('%s' % os.getpid())
pidfile.flush()
@atexit.register
def clean_pidfile():
pidfile.seek(0)
pidfile.truncate()
pidfile.flush()