class IMRegisterServerMoab(object):
def __init__(self):
super(IMRegisterServerMoab, self).__init__()
self.numparams = 5 # prefix,name,os,arch,machine
self.prefix = ""
self.name = ""
self.operatingsystem = ""
self.arch = ""
self.machine = ""
# load from config file
self._registerConf = IMServerConf()
self._registerConf.load_registerServerMoabConfig()
self.port = self._registerConf.getMoabPort()
self.moabInstallPath = self._registerConf.getMoabInstallPath()
self.log_filename = self._registerConf.getLogMoab()
# self.timeToRestartMoab = self._registerConf.getTimeToRestartMoab() #time that we wait to get the moab scheduler restarted (mschedctl -R)
self.logLevel = self._registerConf.getLogLevelMoab()
self._ca_certs = self._registerConf.getCaCertsMoab()
self._certfile = self._registerConf.getCertFileMoab()
self._keyfile = self._registerConf.getKeyFileMoab()
print "\nReading Configuration file from " + self._registerConf.getConfigFile() + "\n"
self.logger = self.setup_logger()
def setup_logger(self):
# Setup logging
logger = logging.getLogger("RegisterMoab")
logger.setLevel(self.logLevel)
formatter = logging.Formatter("%(asctime)s - %(name)s - %(levelname)s - %(message)s")
handler = logging.FileHandler(self.log_filename)
handler.setLevel(self.logLevel)
handler.setFormatter(formatter)
logger.addHandler(handler)
logger.propagate = False # Do not propagate to others
return logger
def start(self):
sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
sock.bind(("", self.port))
sock.listen(1)
self.logger.info("Starting Server on port " + str(self.port))
while True:
newsocket, fromaddr = sock.accept()
connstream = 0
try:
connstream = ssl.wrap_socket(
newsocket,
server_side=True,
ca_certs=self._ca_certs,
cert_reqs=ssl.CERT_REQUIRED,
certfile=self._certfile,
keyfile=self._keyfile,
ssl_version=ssl.PROTOCOL_TLSv1,
)
self.process_client(connstream)
except ssl.SSLError:
self.logger.error("Unsuccessful connection attempt from: " + repr(fromaddr))
finally:
if connstream is ssl.SSLSocket:
connstream.shutdown(socket.SHUT_RDWR)
connstream.close()
def process_client(self, connstream):
self.logger.info("Accepted new connection")
# receive the message
data = connstream.read(2048)
params = data.split(",")
# params[0] is prefix
# params[1] is name
# params[2] is operating system
# params[3] is arch
# params[4] is machine (india, minicluster..)
self.prefix = params[0]
self.name = params[1]
self.operatingsystem = params[2]
self.arch = params[3]
self.machine = params[4]
if len(params) != self.numparams:
msg = "ERROR: incorrect message"
self.errormsg(connstream, msg)
return
if self.prefix == "list":
moabimageslist = ""
f = open(self.moabInstallPath + "/tools/msm/images.txt")
for i in f:
if not re.search("^#", i):
moabimageslist += i.split()[0] + ","
moabimageslist = moabimageslist.rstrip(",")
self.logger.debug("Moab image list: " + moabimageslist)
connstream.write(moabimageslist)
connstream.shutdown(socket.SHUT_RDWR)
connstream.close()
self.logger.info("Image Register Moab (list) DONE")
elif self.prefix == "infosites":
self.logger.debug("Information Moab Site: " + str(self.machine))
connstream.write("True")
connstream.shutdown(socket.SHUT_RDWR)
connstream.close()
self.logger.info("Image Register Moab (info sites) DONE")
elif self.prefix == "remove":
self.logger.debug("Remove image from Moab")
cmd = "sudo sed -i '/^" + self.name + "/d' /opt/moab/tools/msm/images.txt"
self.logger.info(cmd)
status = os.system(cmd)
if status != 0:
msg = "ERROR: removing image from image.txt file"
self.logger.debug(msg)
self.errormsg(connstream, msg)
return
else:
connstream.write("OK")
connstream.shutdown(socket.SHUT_RDWR)
connstream.close()
cmd = "sudo " + self.moabInstallPath + "/bin/mschedctl -R"
self.runCmd(cmd)
self.logger.info("Image Register Moab DONE")
else:
moabstring = ""
if self.machine == "minicluster":
moabstring = (
'echo "'
+ self.prefix
+ self.operatingsystem
+ ""
+ self.name
+ " "
+ self.arch
+ " "
+ self.prefix
+ self.operatingsystem
+ ""
+ self.name
+ ' compute netboot" | sudo tee -a '
+ self.moabInstallPath
+ "/tools/msm/images.txt > /dev/null"
)
# moabstring = 'echo \"' + prefix + operatingsystem + '' + name + ' ' + arch + ' boottarget ' + prefix + operatingsystem + '' + name + ' netboot\" >> ' + moabInstallPath + '/tools/msm/images.txt'
elif self.machine == "india":
moabstring = (
'echo "'
+ self.prefix
+ self.operatingsystem
+ ""
+ self.name
+ " "
+ self.arch
+ " boottarget "
+ self.prefix
+ self.operatingsystem
+ ""
+ self.name
+ ' netboot" | sudo tee -a '
+ self.moabInstallPath
+ "/tools/msm/images.txt > /dev/null"
)
# This message inster the line in the images.txt file
self.logger.debug(moabstring)
status = os.system(moabstring)
if status != 0:
msg = "ERROR: including image name in image.txt file"
self.logger.debug(msg)
self.errormsg(connstream, msg)
return
else:
connstream.write("OK")
connstream.shutdown(socket.SHUT_RDWR)
connstream.close()
cmd = "sudo " + self.moabInstallPath + "/bin/mschedctl -R"
status = self.runCmd(cmd)
self.logger.info("Image Register Moab DONE")
"""
if not os.path.isfile('/tmp/image-register-fork.lock'):
os.system('touch /tmp/image-register-fork.lock')
child_pid = os.fork()
if child_pid == 0:
self.logger.debug("Child Process: PID# %s" % os.getpid())
time.sleep(self.timeToRestartMoab)
cmd = 'mschedctl -R'
status = self.runCmd(cmd)
os.system('rm -f /tmp/image-register-fork.lock')
else:
self.logger.debug("Parent Process: PID# %s" % os.getpid())
"""
def errormsg(self, connstream, msg):
self.logger.error(msg)
try:
connstream.write(msg)
connstream.shutdown(socket.SHUT_RDWR)
connstream.close()
except:
self.logger.debug("In errormsg: " + str(sys.exc_info()))
self.logger.info("Image Register Moab DONE")
def runCmd(self, cmd):
cmdLog = logging.getLogger("RegisterMoab.exec")
cmdLog.debug(cmd)
p = Popen(cmd.split(" "), stdout=PIPE, stderr=PIPE)
std = p.communicate()
status = 0
if len(std[0]) > 0:
cmdLog.debug("stdout: " + std[0])
cmdLog.debug("stderr: " + std[1])
if p.returncode != 0:
cmdLog.error("Command: " + cmd + " failed, status: " + str(p.returncode) + " --- " + std[1])
status = 1
# sys.exit(p.returncode)
return status
