-# $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] = []
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':
_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
+# keyword being 'start', 'stop' or 'restart'
+def handleService (keyword):
+ if tools.has_systemctl():
+ logger.log("vsys: %s'ing vsys service through systemctl"%keyword)
+ return logger.log_call(["systemctl", "restart", "vsys"])
+ else:
+ logger.log("vsys: %s'ing vsys service through /etc/init.d/vsys"%keyword)
+ return logger.log_call(["/etc/init.d/vsys", keyword])
+def startService(): return handleService ('start')
+def stopService(): return handleService ('stop')
+def restartService(): return handleService ('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 = []
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
# 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
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.
# 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:
'''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 stop vsys if needed (in which case it return True to tell caller to restart vsys once done)
+def removeSliverFromVsys (sliver):
+ current_slivers=parseConf()
+ new_slivers= [ s for s in current_slivers if s != sliver ]
+ if writeConf (current_slivers, new_slivers):
+ stopService()
+ trashVsysHandleInSliver (sliver)
+ return True
+ else:
+ logger.log("vsys.removeSliverFromConf: no need to remove %s"%sliver)
+ return False
+
+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))