nm.py

   1 #!/usr/bin/python
   2
   3 """Node Manager"""
   4
   5 import optparse
   6 import time
   7 import xmlrpclib
   8 import socket
   9 import os
  10 import sys
  11 import resource
  12
  13 import logger
  14 import tools
  15
  16 from config import Config
  17 from plcapi import PLCAPI
  18
  19
  20 savedargv = sys.argv[:]
  21
  22 parser = optparse.OptionParser()
  23 parser.add_option('-d', '--daemon', action='store_true', dest='daemon', default=False, help='run daemonized')
  24 parser.add_option('-s', '--startup', action='store_true', dest='startup', default=False, help='run all sliver startup scripts')
  25 parser.add_option('-f', '--config', action='store', dest='config', default='/etc/planetlab/plc_config', help='PLC configuration file')
  26 parser.add_option('-k', '--session', action='store', dest='session', default='/etc/planetlab/session', help='API session key (or file)')
  27 parser.add_option('-p', '--period', action='store', dest='period', default=600, help='Polling interval (sec)')
  28 (options, args) = parser.parse_args()
  29
  30 modules = []
  31
  32 def GetSlivers(plc):
  33     data = plc.GetSlivers()
  34     # net needs access to API for i2 nodes.
  35     for module in modules:
  36         if module.__name__ == 'net':
  37             module.GetSlivers(plc, data)
  38         else:
  39             callback = getattr(module, 'GetSlivers')
  40             callback(data)
  41
  42 def run():
  43     try:
  44         if options.daemon: tools.daemon()
  45
  46         # Load /etc/planetlab/plc_config
  47         config = Config(options.config)
  48
  49         try:
  50             other_pid = tools.pid_file()
  51             if other_pid != None:
  52                 print """There might be another instance of the node manager running as pid %d.  If this is not the case, please remove the pid file %s""" % (other_pid, tools.PID_FILE)
  53                 return
  54         except OSError, err:
  55             print "Warning while writing PID file:", err
  56
  57         # Load and start modules
  58         for module in ['net', 'proper', 'conf_files', 'sm', 'bwmon']:
  59             try:
  60                 m = __import__(module)
  61                 m.start(options, config)
  62                 modules.append(m)
  63             except ImportError, err:
  64                 print "Warning while loading module %s:" % module, err
  65
  66         # Load /etc/planetlab/session
  67         if os.path.exists(options.session):
  68             session = file(options.session).read().strip()
  69         else:
  70             session = options.session
  71
  72         # Initialize XML-RPC client
  73         plc = PLCAPI(config.plc_api_uri, config.cacert, session, timeout=options.period/2)
  74
  75         while True:
  76             try: GetSlivers(plc)
  77             except: logger.log_exc()
  78             time.sleep(options.period)
  79     except: logger.log_exc()
  80
  81
  82 if __name__ == '__main__':
  83     stacklim = 512*1024  # 0.5 MiB
  84     curlim = resource.getrlimit(resource.RLIMIT_STACK)[0]  # soft limit
  85     if curlim > stacklim:
  86         resource.setrlimit(resource.RLIMIT_STACK, (stacklim, stacklim))
  87         # for some reason, doesn't take effect properly without the exec()
  88         python = '/usr/bin/python'
  89         os.execv(python, [python] + savedargv)
  90     run()
  91 else:
  92     # This is for debugging purposes.  Open a copy of Python and import nm
  93     tools.as_daemon_thread(run)