X-Git-Url: http://git.onelab.eu/?a=blobdiff_plain;f=python%2Fvserver.py;h=c6a1cc7e3c2a60c644e098ab26372478c0fc4d2b;hb=87bb0cc580b61e62b56942ab621e0c23d6fac976;hp=55d78f044209ec79104fb79ed75c787c501cec36;hpb=a6e1134d8bdb41b0146cdac83e4e82909e74cb47;p=util-vserver-pl.git

diff --git a/python/vserver.py b/python/vserver.py
index 55d78f0..c6a1cc7 100644
--- a/python/vserver.py
+++ b/python/vserver.py
@@ -133,8 +133,9 @@ class VServerConfig:
 
 class VServer:
 
-    INITSCRIPTS = [('/etc/rc.vinit', 'start'),
-                   ('/etc/rc.d/rc', '%(runlevel)d')]
+    # 2010 June 21 - Thierry 
+    # the slice initscript now gets invoked through rc - see sliver_vs.py in nodemanager
+    INITSCRIPTS = [('/etc/rc.d/rc', '%(runlevel)d')]
 
     def __init__(self, name, vm_id = None, vm_running = None, logfile=None):
 
@@ -154,12 +155,16 @@ class VServer:
         self.logfile = logfile
 
     # inspired from nodemanager's logger
+    def log_in_file (self, fd, msg):
+        if not msg: msg="\n"
+        if not msg.endswith('\n'): msg += '\n'
+        os.write(fd, '%s: %s' % (time.asctime(time.gmtime()), msg))
+
     def log(self,msg):
         if self.logfile:
             try:
                 fd = os.open(self.logfile,os.O_WRONLY | os.O_CREAT | os.O_APPEND, 0600)
-                if not msg.endswith('\n'): msg += '\n'
-                os.write(fd, '%s: %s' % (time.asctime(time.gmtime()), msg))
+                self.log_in_file(fd,msg)
                 os.close(fd)
             except:
                 print '%s: (%s failed to open) %s'%(time.asctime(time.gmtime()),self.logfile,msg)
@@ -261,13 +266,13 @@ class VServer:
         os.chroot(self.dir)
         os.chdir("/")
 
-    def chroot_call(self, fn, *args):
+    def chroot_call(self, fn, *args, **kwargs):
         cwd_fd = os.open(".", os.O_RDONLY)
         try:
             root_fd = os.open("/", os.O_RDONLY)
             try:
                 self.__do_chroot()
-                result = fn(*args)
+                result = fn(*args, **kwargs)
             finally:
                 os.fchdir(root_fd)
                 os.chroot(".")
@@ -372,17 +377,48 @@ class VServer:
     def enter(self):
         subprocess.call("/usr/sbin/vserver %s enter" % self.name, shell=True)
 
+    # detach the process that triggers the initscripts
+    # after http://code.activestate.com/recipes/278731/
     def start(self, runlevel = 3):
-        if (os.fork() != 0):
+        if os.fork() != 0:
             # Parent should just return.
             self.vm_running = True
             return
         else:
-            # child process
+            os.setsid()
+            # first child process: fork again
+            if os.fork() != 0:
+                os._exit(0)	# Exit parent (the first child) of the second child.
+            # the grandson is the working one
+            os.chdir('/')
+            os.umask(0)
             try:
-                subprocess.call("/usr/sbin/vserver %s start" % self.name, 
-                                shell=True)
-            # we get here due to an exception in the top-level child process
+                # start the vserver
+                subprocess.call(["/usr/sbin/vserver",self.name,"start"])
+
+                # execute initscripts
+                for cmd_to_expand in self.INITSCRIPTS:
+                    # enter vserver context
+                    expand = { 'runlevel': runlevel,
+                               'name': self.name, }
+                    cmd = [ x % expand for x in cmd_to_expand ]
+                    cmd_name = os.path.basename(cmd[0])
+                    cmd_file = "/vservers/" + self.name + cmd[0]
+                    if not os.path.isfile(cmd_file):
+                        self.log("WARNING: could not find %s for %s" % (cmd_file, self.name))
+                        break
+                    self.log("executing %r" % cmd)
+                    try:
+                        logname='/vservers/%s/var/log/%s'%(self.name,cmd_name)
+                        log_fd=os.open(logname,os.O_WRONLY | os.O_CREAT | os.O_APPEND, 0600)
+                        self.log_in_file(log_fd,"Running %r into %s"%(cmd,logname))
+                        self.chroot_call(subprocess.call,cmd,
+                                         stdout=log_fd,stderr=subprocess.STDOUT,
+                                         close_fds=True)
+                    except:  self.log(traceback.format_exc())
+                    finally: os.close(log_fd)
+
+            # we get here due to an exception in the grandson process
             except Exception, ex:
                 self.log(traceback.format_exc())
             os._exit(0)