X-Git-Url: http://git.onelab.eu/?a=blobdiff_plain;f=plugins%2Fvsys.py;h=f631d95959a3f07596b8f6d306c41d6a76129738;hb=13a17295e8064a3b61b9dee92b835bb0686d16c2;hp=939158c7145575b8ba74521d42683c27e97a017c;hpb=1029f5fa2d29e3fe4ce222ee5fa580f16abfceeb;p=nodemanager.git diff --git a/plugins/vsys.py b/plugins/vsys.py index 939158c..f631d95 100644 --- a/plugins/vsys.py +++ b/plugins/vsys.py @@ -1,21 +1,24 @@ -# $Id$ -# $URL$ - """vsys configurator. Maintains ACLs and script pipes inside vservers based on slice attributes.""" -import logger import os -from sets import Set +import subprocess + +import logger +import tools VSYSCONF="/etc/vsys.conf" VSYSBKEND="/vsys" -def start(options, config): - pass +def start(): + logger.log("vsys: plugin starting up...") - -def GetSlivers(data): +def GetSlivers(data, config=None, plc=None): """For each sliver with the vsys attribute, set the script ACL, create the vsys directory in the slice, and restart vsys.""" + + if 'slivers' not in data: + logger.log_missing_data("vsys.GetSlivers",'slivers') + return + # Touch ACLs and create dict of available scripts = {} for script in touchAcls(): scripts[script] = [] @@ -23,6 +26,9 @@ def GetSlivers(data): slices = [] _restart = False # Parse attributes and update dict of scripts + if 'slivers' not in data: + logger.log_missing_data("vsys.GetSlivers",'slivers') + return for sliver in data['slivers']: for attribute in sliver['attributes']: if attribute['tagname'] == 'vsys': @@ -32,26 +38,33 @@ def GetSlivers(data): _restart = createVsysDir(sliver['name']) or _restart if attribute['value'] in scripts.keys(): scripts[attribute['value']].append(sliver['name']) - + # Write the conf _restart = writeConf(slices, parseConf()) or _restart # Write out the ACLs if writeAcls(scripts, parseAcls()) or _restart: - logger.log("vsys: restarting vsys service") - logger.log_call("/etc/init.d/vsys", "restart") + restartService() +# check for systemctl, use it if present +def restartService (): + if tools.has_systemctl(): + logger.log("vsys: restarting vsys service through systemctl") + logger.log_call(["systemctl", "restart", "vsys"]) + else: + logger.log("vsys: restarting vsys service through /etc/init.d/vsys") + logger.log_call(["/etc/init.d/vsys", "restart", ]) def createVsysDir(sliver): '''Create /vsys directory in slice. Update vsys conf file.''' - try: + try: os.mkdir("/vservers/%s/vsys" % sliver) return True - except OSError: + except OSError: return False def touchAcls(): - '''Creates empty acl files for scripts. + '''Creates empty acl files for scripts. To be ran in case of new scripts that appear in the backend. Returns list of available scripts.''' acls = [] @@ -64,12 +77,12 @@ def touchAcls(): acls.append(file.replace(".acl", "")) else: scripts.append(file) - for new in (Set(scripts) - Set(acls)): + for new in (set(scripts) - set(acls)): logger.log("vsys: Found new script %s. Writing empty acl." % new) f = open("%s/%s.acl" %(VSYSBKEND, new), "w") f.write("\n") f.close() - + return scripts @@ -82,13 +95,16 @@ def writeAcls(currentscripts, oldscripts): # and length of non intersection along new scripts is not 0, # then dicts are different. for (acl, oldslivers) in oldscripts.iteritems(): - if (len(oldslivers) != len(currentscripts[acl])) or \ - (len(Set(oldslivers) - Set(currentscripts[acl])) != 0): - _restartvsys = True - logger.log("vsys: Updating %s.acl w/ slices %s" % (acl, currentscripts[acl])) - f = open("%s/%s.acl" % (VSYSBKEND, acl), "w") - for slice in currentscripts[acl]: f.write("%s\n" % slice) - f.close() + try: + if (len(oldslivers) != len(currentscripts[acl])) or \ + (len(set(oldslivers) - set(currentscripts[acl])) != 0): + _restartvsys = True + logger.log("vsys: Updating %s.acl w/ slices %s" % (acl, currentscripts[acl])) + f = open("%s/%s.acl" % (VSYSBKEND, acl), "w") + for slice in currentscripts[acl]: f.write("%s\n" % slice) + f.close() + except KeyError: + logger.log("vsys: #:)# Warning,Not a valid Vsys script,%s"%acl) # Trigger a restart return _restartvsys @@ -103,7 +119,7 @@ def parseAcls(): f = open(root+"/"+file,"r+") scriptname = file.replace(".acl", "") scriptacls[scriptname] = [] - for slice in f.readlines(): + for slice in f.readlines(): scriptacls[scriptname].append(slice.rstrip()) f.close() # return what scripts are configured for which slices. @@ -116,7 +132,7 @@ def writeConf(slivers, oldslivers): # and the non intersection of both arrays has length 0, # then the arrays are identical. if (len(slivers) != len(oldslivers)) or \ - (len(Set(oldslivers) - Set(slivers)) != 0): + (len(set(oldslivers) - set(slivers)) != 0): logger.log("vsys: Updating %s" % VSYSCONF) f = open(VSYSCONF,"w") for sliver in slivers: @@ -132,11 +148,33 @@ def parseConf(): '''Parse the vsys conf and return list of slices in conf.''' scriptacls = {} slicesinconf = [] - try: + try: f = open(VSYSCONF) for line in f.readlines(): (path, slice) = line.split() slicesinconf.append(slice) f.close() - except: logger.log_exc() + except: logger.log_exc("vsys: failed parseConf") return slicesinconf + + +# before shutting down slivers, it is safe to first remove them from vsys's scope +# so that we are sure that no dangling open file remains +# this will also restart vsys if needed +def removeSliverFromVsys (sliver): + current_slivers=parseConf() + new_slivers= [ s for s in current_slivers if s != sliver ] + if writeConf (current_slivers, new_slivers): + restartService() + trashVsysHandleInSliver (sliver) + else: + logger.log("vsys.removeSliverFromConf: no need to remove %s"%sliver) + + +def trashVsysHandleInSliver (sliver): + slice_vsys_area = "/vservers/%s/vsys"%sliver + if not os.path.exists(slice_vsys_area): + logger.log("vsys.trashVsysHandleInSliver: no action needed, %s not found"%slice_vsys_area) + return + retcod=subprocess.call([ 'rm', '-rf' , slice_vsys_area]) + logger.log ("vsys.trashVsysHandleInSliver: Removed %s (retcod=%s)"%(slice_vsys_area,retcod))