import xmlrpclib
import socket
import os
+import sys
+import resource
import logger
import tools
from config import Config
-from plcapi import PLCAPI
+from plcapi import PLCAPI
+import random
+
+savedargv = sys.argv[:]
parser = optparse.OptionParser()
parser.add_option('-d', '--daemon', action='store_true', dest='daemon', default=False, help='run daemonized')
def GetSlivers(plc):
data = plc.GetSlivers()
+ # net needs access to API for i2 nodes.
for module in modules:
- callback = getattr(module, 'GetSlivers')
- callback(data)
+ if module.__name__ == 'net':
+ module.GetSlivers(plc, data)
+ else:
+ callback = getattr(module, 'GetSlivers')
+ callback(data)
+
+def UpdateHostKey(plc):
+ logger.log('Trying to update ssh host key at PLC...')
+ ssh_host_key = open('/etc/ssh/ssh_host_rsa_key.pub').read().strip()
+ plc.BootUpdateNode(dict(ssh_host_key=ssh_host_key))
+ logger.log('Host key update succeeded')
def run():
try:
print "Warning while writing PID file:", err
# Load and start modules
- for module in ['net', 'proper', 'conf_files', 'sm']:
+ for module in ['net', 'proper', 'conf_files', 'sm', 'bwmon']:
try:
m = __import__(module)
m.start(options, config)
plc = PLCAPI(config.plc_api_uri, config.cacert, session, timeout=options.period/2)
while True:
+ try: UpdateHostKey(plc)
+ except: logger.log_exc()
try: GetSlivers(plc)
except: logger.log_exc()
- time.sleep(options.period)
+ time.sleep(options.period + random.randrange(0,301))
except: logger.log_exc()
-if __name__ == '__main__': run()
+if __name__ == '__main__':
+ stacklim = 512*1024 # 0.5 MiB
+ curlim = resource.getrlimit(resource.RLIMIT_STACK)[0] # soft limit
+ if curlim > stacklim:
+ resource.setrlimit(resource.RLIMIT_STACK, (stacklim, stacklim))
+ # for some reason, doesn't take effect properly without the exec()
+ python = '/usr/bin/python'
+ os.execv(python, [python] + savedargv)
+ run()
else:
# This is for debugging purposes. Open a copy of Python and import nm
tools.as_daemon_thread(run)