X-Git-Url: http://git.onelab.eu/?a=blobdiff_plain;f=account.py;h=b2f9714e264c6e5fa1ece13e531075ba5c62d347;hb=ecee05390277f57b02d21ffca0195292bde1defa;hp=ac0231228a21289db0835be49ebc6c01ae9939d8;hpb=cdf732863cd300caf8222c59e3f38256d9e04fca;p=nodemanager.git diff --git a/account.py b/account.py index ac02312..b2f9714 100644 --- a/account.py +++ b/account.py @@ -27,6 +27,7 @@ maximum stack size. import os import pwd, grp import threading +import subprocess import logger import tools @@ -69,12 +70,13 @@ def get(name): finally: name_worker_lock.release() +# xxx strictly speaking this class should not use self.name that in fact +# is accidentally inited by the subclasses constructor... class Account: - def __init__(self, rec): - logger.verbose('account: Initing account %s'%rec['name']) - self.name = rec['name'] + def __init__(self, name): + self.name = name self.keys = '' - self.configure(rec) + logger.verbose('account: Initing account %s'%name) @staticmethod def create(name, vref = None): abstract @@ -121,6 +123,46 @@ class Account: def stop(self): pass def is_running(self): pass + ### this used to be a plain method but because it needs to be invoked by destroy + # which is a static method, they need to become static as well + # needs to be done before sliver starts (checked with vs and lxc) + @staticmethod + def mount_ssh_dir (slicename): return Account._manage_ssh_dir (slicename, do_mount=True) + @staticmethod + def umount_ssh_dir (slicename): return Account._manage_ssh_dir (slicename, do_mount=False) + + # bind mount / umount root side dir to sliver side + @staticmethod + def _manage_ssh_dir (slicename, do_mount): + logger.log ("_manage_ssh_dir, requested to "+("mount" if do_mount else "umount")+" ssh dir for "+ slicename) + try: + root_ssh="/home/%s/.ssh"%slicename + sliver_ssh="/vservers/%s/home/%s/.ssh"%(slicename,slicename) + def is_mounted (root_ssh): + for mount_line in file('/proc/mounts').readlines(): + if mount_line.find (root_ssh)>=0: return True + return False + if do_mount: + # any of both might not exist yet + for path in [root_ssh,sliver_ssh]: + if not os.path.exists (path): + os.mkdir(path) + if not os.path.isdir (path): + raise Exception + if not is_mounted(root_ssh): + command=['mount','--bind','-o','ro',root_ssh,sliver_ssh] + mounted=logger.log_call (command) + msg="OK" if mounted else "WARNING: FAILED" + logger.log("_manage_ssh_dir: mounted %s into slice %s - %s"%(root_ssh,slicename,msg)) + else: + if is_mounted (sliver_ssh): + command=['umount',sliver_ssh] + umounted=logger.log_call(command) + msg="OK" if umounted else "WARNING: FAILED" + logger.log("_manage_ssh_dir: umounted %s - %s"%(sliver_ssh,msg)) + except: + logger.log_exc("_manage_ssh_dir failed",name=slicename) + class Worker: def __init__(self, name): @@ -140,9 +182,9 @@ If still valid, check if running and configure/start if not.""" try: next_class.create(self.name, rec) finally: create_sem.release() if not isinstance(self._acct, next_class): self._acct = next_class(rec) - logger.verbose("account.ensure_created: %s, running=%r"%(self.name,self.is_running())) + logger.verbose("account.Worker.ensure_created: %s, running=%r"%(self.name,self.is_running())) - # reservation_alive is set on reervable nodes, and its value is a boolean + # reservation_alive is set on reservable nodes, and its value is a boolean if 'reservation_alive' in rec: # reservable nodes if rec['reservation_alive']: @@ -191,23 +233,3 @@ If still valid, check if running and configure/start if not.""" except KeyError: return None return shell_acct_class[shell] - # bind mount root side dir to sliver side - # needs to be done before sliver starts, in the vserver case at least - def expose_ssh_dir (self): - try: - root_ssh="/home/%s/.ssh"%self.name - sliver_ssh="/vservers/%s/home/%s/.ssh"%(self.name,self.name) - # any of both might not exist yet - for path in [root_ssh,sliver_ssh]: - if not os.path.exists (path): - os.mkdir(path) - if not os.path.isdir (path): - raise Exception - mounts=file('/proc/mounts').read() - if mounts.find(sliver_ssh)<0: - # xxx perform mount - subprocess.call("mount --bind -o ro %s %s"%(root_ssh,sliver_ssh),shell=True) - logger.log("expose_ssh_dir: %s mounted into slice %s"%(root_ssh,self.name)) - except: - logger.log_exc("expose_ssh_dir with slice %s failed"%self.name) -