]> git.proxmox.com Git - mirror_ifupdown2.git/blobdiff - pkg/scheduler.py
Fix upperiface check when ifdown is run with -a
[mirror_ifupdown2.git] / pkg / scheduler.py
index 28a3382dc17552cab546932eb26b84695af71b08..6fc19c5bf96a9941033aa48f1940da1495dd159c 100644 (file)
@@ -29,14 +29,27 @@ class ifaceScheduler():
 
     supports scheduling of interfaces serially in plain interface list
     or dependency graph format.
+
+    Algo:
+        - run topological sort on the iface objects
+        - In the sorted iface object list, pick up interfaces with no parents
+        and run ops on them and their children. 
+        - If operation is up and user gave interface list (ie not all)
+        option, also see if there were upper-devices and run ops on them. 
+        - if operation is down, dont down the interface if it still
+        has upperifaces present. The down operation is executed when the
+        last upperiface goes away. If force option is set, this rule does not
+        apply.
+        - run ops calls addon modules run operation passing the iface
+        object and op to each module.
+        - ops are [pre-up, up, post-up, pre-down, down,
+                   post-down, query-running, query-check]
     """
 
     _STATE_CHECK = True
 
-    token_pool = None
-
     @classmethod
-    def run_iface_op(cls, ifupdownobj, ifaceobj, op, cenv):
+    def run_iface_op(cls, ifupdownobj, ifaceobj, op, cenv=None):
         """ Runs sub operation on an interface """
         ifacename = ifaceobj.name
 
@@ -45,16 +58,9 @@ class ifaceScheduler():
             (ifaceobj.status == ifaceStatus.SUCCESS)):
             ifupdownobj.logger.debug('%s: already in state %s' %(ifacename, op))
             return
-
-        # first run ifupdownobj handlers
-        handler = ifupdownobj.ops_handlers.get(op)
-        if handler:
-            if not ifaceobj.addr_method or (ifaceobj.addr_method and
-                    ifaceobj.addr_method != 'manual'):
-                handler(ifupdownobj, ifaceobj)
-
         if not ifupdownobj.ADDONS_ENABLE: return
-
+        if op == 'query-checkcurr':
+            query_ifaceobj=ifupdownobj.create_n_save_ifaceobjcurr(ifaceobj)
         for mname in ifupdownobj.module_ops.get(op):
             m = ifupdownobj.modules.get(mname)
             err = 0
@@ -67,8 +73,7 @@ class ifaceScheduler():
                         if (ifaceobj.priv_flags & ifupdownobj.NOCONFIG):
                             continue
                         ifupdownobj.logger.debug(msg)
-                        m.run(ifaceobj, op,
-                              query_ifaceobj=ifupdownobj.create_n_save_ifaceobjcurr(ifaceobj))
+                        m.run(ifaceobj, op, query_ifaceobj)
                     else:
                         ifupdownobj.logger.debug(msg)
                         m.run(ifaceobj, op)
@@ -94,60 +99,83 @@ class ifaceScheduler():
                     ifupdownobj.log_error(str(e))
 
     @classmethod
-    def run_iface_ops(cls, ifupdownobj, ifaceobj, ops):
-        """ Runs all operations on an interface """
-        ifacename = ifaceobj.name
+    def run_iface_list_ops(cls, ifupdownobj, ifaceobjs, ops):
+        """ Runs all operations on a list of interface
+            configurations for the same interface
+        """
         # minor optimization. If operation is 'down', proceed only
         # if interface exists in the system
-        if 'down' in ops[0] and not ifupdownobj.link_exists(ifacename):
-            ifupdownobj.logger.info('%s: does not exist' %ifacename)
+        ifacename = ifaceobjs[0].name
+        if ('down' in ops[0] and
+                not ifupdownobj.link_exists(ifacename)):
+            ifupdownobj.logger.debug('%s: does not exist' %ifacename)
+            # run posthook before you get out of here, so that
+            # appropriate cleanup is done
+            posthookfunc = ifupdownobj.sched_hooks.get('posthook')
+            if posthookfunc:
+                for ifaceobj in ifaceobjs:
+                    ifaceobj.status = ifaceStatus.SUCCESS
+                    posthookfunc(ifupdownobj, ifaceobj, 'down')
             return 
-        cenv=None
-        if ifupdownobj.COMPAT_EXEC_SCRIPTS:
-            # For backward compatibility generate env variables
-            # for attributes
-            cenv = ifupdownobj.generate_running_env(ifaceobj, ops[0])
-        map(lambda op: cls.run_iface_op(ifupdownobj, ifaceobj, op, cenv), ops)
-        posthookfunc = ifupdownobj.sched_hooks.get('posthook')
-        if posthookfunc:
-            posthookfunc(ifupdownobj, ifaceobj)
-
+        for op in ops:
+            # first run ifupdownobj handlers. This is good enough
+            # for the first object in the list
+            handler = ifupdownobj.ops_handlers.get(op)
+            if handler:
+                if (not ifaceobjs[0].addr_method or
+                    (ifaceobjs[0].addr_method and
+                    ifaceobjs[0].addr_method != 'manual')):
+                    handler(ifupdownobj, ifaceobjs[0])
+            for ifaceobj in ifaceobjs:
+                cls.run_iface_op(ifupdownobj, ifaceobj, op,
+                    cenv=ifupdownobj.generate_running_env(ifaceobj, op)
+                        if ifupdownobj.COMPAT_EXEC_SCRIPTS else None) 
+                posthookfunc = ifupdownobj.sched_hooks.get('posthook')
+                if posthookfunc:
+                    posthookfunc(ifupdownobj, ifaceobj, op)
 
     @classmethod
-    def _check_upperifaces(cls, ifupdownobj, ifaceobj, ops, parent, followdependents=False):
-        """ Check if conflicting upper ifaces are around and warn if required
+    def _check_upperifaces(cls, ifupdownobj, ifaceobj, ops, parent,
+                           followdependents=False):
+        """ Check if upperifaces are hanging off us and help caller decide
+        if he can proceed with the ops on this device
 
-        Returns False if this interface needs to be skipped, else return True """
+        Returns True or False indicating the caller to proceed with the
+        operation.
+        """
+        # proceed only for down operation
+        if 'down' not in ops[0]:
+            return True
 
-        if 'up' in ops[0] and followdependents:
+        if (ifupdownobj.FORCE or
+                not ifupdownobj.ADDONS_ENABLE or
+                (not ifupdownobj.is_ifaceobj_noconfig(ifaceobj) and
+                ifupdownobj.config.get('warn_on_ifdown', '0') == '0' and
+                not ifupdownobj.ALL)):
             return True
 
-        ifacename = ifaceobj.name
-        # Deal with upperdevs first
         ulist = ifaceobj.upperifaces
-        if ulist:
-            tmpulist = ([u for u in ulist if u != parent] if parent
-                            else ulist)
-            if not tmpulist:
-                return True
-            if 'down' in ops[0]:
-                # XXX: This is expensive. Find a cheaper way to do this 
-                # if any of the upperdevs are present,
-                # dont down this interface
-                for u in tmpulist:
-                    if ifupdownobj.link_exists(u):
-                        if not ifupdownobj.FORCE and not ifupdownobj.ALL:
-                            ifupdownobj.logger.warn('%s: ' %ifacename +
-                                    ' skip interface down,' +
-                                    ' upperiface %s still around' %u)
-                            return False
-            elif 'up' in ops[0] and not ifupdownobj.ALL:
-                # For 'up', just warn that there is an upperdev which is
-                # probably not up
-                for u in tmpulist:
-                    if not ifupdownobj.link_exists(u):
-                        ifupdownobj.logger.warn('%s: upper iface %s '
-                                %(ifacename, u) + 'does not exist')
+        if not ulist:
+            return True
+
+        # Get the list of upper ifaces other than the parent
+        tmpulist = ([u for u in ulist if u != parent] if parent
+                    else ulist)
+        if not tmpulist:
+            return True
+        # XXX: This is expensive. Find a cheaper way to do this.
+        # if any of the upperdevs are present,
+        # return false to the caller to skip this interface
+        for u in tmpulist:
+            if ifupdownobj.link_exists(u):
+                if not ifupdownobj.ALL:
+                    if ifupdownobj.is_ifaceobj_noconfig(ifaceobj):
+                        ifupdownobj.logger.info('%s: skipping interface down,'
+                            %ifaceobj.name + ' upperiface %s still around ' %u)
+                    else:
+                        ifupdownobj.logger.warn('%s: skipping interface down,'
+                            %ifaceobj.name + ' upperiface %s still around ' %u)
+                return False
         return True
 
     @classmethod
@@ -162,18 +190,20 @@ class ifaceScheduler():
             raise Exception('%s: not found' %ifacename)
 
         for ifaceobj in ifaceobjs:
-            if not cls._check_upperifaces(ifupdownobj, ifaceobj, ops, parent,
-                                          followdependents):
+            if not cls._check_upperifaces(ifupdownobj, ifaceobj,
+                                          ops, parent, followdependents):
                 return
-            if order == ifaceSchedulerFlags.INORDER:
-                # If inorder, run the iface first and then its dependents
-                cls.run_iface_ops(ifupdownobj, ifaceobj, ops)
 
+        # If inorder, run the iface first and then its dependents
+        if order == ifaceSchedulerFlags.INORDER:
+            cls.run_iface_list_ops(ifupdownobj, ifaceobjs, ops)
+
+        for ifaceobj in ifaceobjs:
             # Run lowerifaces or dependents
             dlist = ifaceobj.lowerifaces
             if dlist:
-                ifupdownobj.logger.debug('%s:' %ifacename +
-                    ' found dependents: %s' %str(dlist))
+                ifupdownobj.logger.debug('%s: found dependents %s'
+                            %(ifacename, str(dlist)))
                 try:
                     if not followdependents:
                         # XXX: this is yet another extra step,
@@ -182,12 +212,11 @@ class ifaceScheduler():
                         # not follow dependents, we must follow the ones
                         # that dont have user given config. Because we own them
                         new_dlist = [d for d in dlist
-                                     if ifupdownobj.is_iface_noconfig(d)]
+                                    if ifupdownobj.is_iface_noconfig(d)]
                         if new_dlist:
                             cls.run_iface_list(ifupdownobj, new_dlist, ops,
-                                                ifacename, order,
-                                                followdependents,
-                                                continueonfailure=False)
+                                           ifacename, order, followdependents,
+                                           continueonfailure=False)
                     else:
                         cls.run_iface_list(ifupdownobj, dlist, ops,
                                             ifacename, order,
@@ -199,10 +228,10 @@ class ifaceScheduler():
                     else:
                         # Dont bring the iface up if children did not come up
                         ifaceobj.set_state_n_status(ifaceState.NEW,
-                                                    ifaceStatus.ERROR)
+                                                ifaceStatus.ERROR)
                         raise
-            if order == ifaceSchedulerFlags.POSTORDER:
-                cls.run_iface_ops(ifupdownobj, ifaceobj, ops)
+        if order == ifaceSchedulerFlags.POSTORDER:
+            cls.run_iface_list_ops(ifupdownobj, ifaceobjs, ops)
 
     @classmethod
     def run_iface_list(cls, ifupdownobj, ifacenames,
@@ -224,8 +253,7 @@ class ifaceScheduler():
                     if (ifupdownobj.ignore_error(str(e))):
                         pass
                     else:
-                        raise Exception('error running iface %s (%s)'
-                                %(ifacename, str(e)))
+                        raise Exception('%s : (%s)' %(ifacename, str(e)))
 
     @classmethod
     def run_iface_graph_upper(cls, ifupdownobj, ifacename, ops, parent=None,
@@ -237,16 +265,15 @@ class ifaceScheduler():
         if not ifaceobjs:
             raise Exception('%s: not found' %ifacename)
 
+        if not skip_root:
+            # run the iface first and then its upperifaces
+            cls.run_iface_list_ops(ifupdownobj, ifaceobjs, ops)
         for ifaceobj in ifaceobjs:
-            if not skip_root:
-                # run the iface first and then its upperifaces
-                cls.run_iface_ops(ifupdownobj, ifaceobj, ops)
-
             # Run upperifaces
             ulist = ifaceobj.upperifaces
             if ulist:
-                ifupdownobj.logger.debug('%s:' %ifacename +
-                    ' found upperifaces: %s' %str(ulist))
+                ifupdownobj.logger.debug('%s: found upperifaces %s'
+                                            %(ifacename, str(ulist)))
                 try:
                     cls.run_iface_list_upper(ifupdownobj, ulist, ops,
                                             ifacename,
@@ -278,8 +305,7 @@ class ifaceScheduler():
                     if (ifupdownobj.ignore_error(str(e))):
                         pass
                     else:
-                        raise Exception('error running iface %s (%s)'
-                                %(ifacename, str(e)))
+                        raise Exception('%s : (%s)' %(ifacename, str(e)))
 
     @classmethod
     def sched_ifaces(cls, ifupdownobj, ifacenames, ops,
@@ -301,10 +327,18 @@ class ifaceScheduler():
         """
 
         if not ifupdownobj.ALL or not followdependents or len(ifacenames) == 1:
+            # If there is any interface that does exist, maybe it is a
+            # logical interface and we have to followupperifaces
+            followupperifaces = (True if
+                                    [i for i in ifacenames
+                                        if not ifupdownobj.link_exists(i)]
+                                        else False)
             cls.run_iface_list(ifupdownobj, ifacenames, ops,
                                   parent=None,order=order,
                                   followdependents=followdependents)
-            if not ifupdownobj.ALL and followdependents and 'up' in ops[0]:
+            if (not ifupdownobj.ALL and
+                    (followdependents or followupperifaces) and
+                    'up' in ops[0]):
                 # If user had given a set of interfaces to bring up
                 # try and execute 'up' on the upperifaces
                 ifupdownobj.logger.info('running upperifaces if available')
@@ -313,15 +347,22 @@ class ifaceScheduler():
                                          skip_root=True)
                 cls._STATE_CHECK = True
             return
-        run_queue = []
 
+        if ifupdownobj.config.get('skip_ifacesort', '0') == '1':
+            # This is a backdoor to skip sorting of interfaces, if required
+            cls.run_iface_list(ifupdownobj, ifacenames, ops,
+                                  parent=None,order=order,
+                                  followdependents=followdependents)
+            return
+
+        run_queue = []
         # Get a sorted list of all interfaces
         if not indegrees:
             indegrees = OrderedDict()
             for ifacename in dependency_graph.keys():
                 indegrees[ifacename] = ifupdownobj.get_iface_refcnt(ifacename)
         sorted_ifacenames = graph.topological_sort_graphs_all(dependency_graph,
-                                                          dict(indegrees))
+                                                          indegrees)
         ifupdownobj.logger.debug('sorted ifacenames %s : '
                                  %str(sorted_ifacenames))
 
@@ -337,251 +378,3 @@ class ifaceScheduler():
         cls.run_iface_list(ifupdownobj, run_queue, ops,
                                   parent=None,order=order,
                                   followdependents=followdependents)
-
-    @classmethod
-    def run_iface(cls, ifupdownobj, ifacename, ops):
-        """ Runs operation on an interface """
-
-        ifaceobjs = ifupdownobj.get_ifaceobjs(ifacename)
-        for i in ifaceobjs:
-            cls.run_iface_ops(ifupdownobj, i, ops)
-
-    @classmethod
-    def run_iface_list_op(cls, ifupdownobj, ifacenames, op,
-                          sorted_by_dependency=False):
-        """ Runs interface list through sub operation handler. """
-
-        ifupdownobj.logger.debug('running operation %s on all given interfaces'
-                                 %op)
-        iface_run_queue = deque(ifacenames)
-        for i in range(0, len(iface_run_queue)):
-            if op.endswith('up'):
-                # XXX: simplify this
-                if sorted_by_dependency:
-                    ifacename = iface_run_queue.pop()
-                else:
-                    ifacename = iface_run_queue.popleft()
-            else:
-                if sorted_by_dependency:
-                    ifacename = iface_run_queue.popleft()
-                else:
-                    ifacename = iface_run_queue.pop()
-
-            try:
-                ifaceobjs = ifupdownobj.get_ifaceobjs(ifacename)
-                for ifaceobj in ifaceobjs:
-                    cenv = ifupdownobj.generate_running_env(ifaceobj, op)
-                    cls.run_iface_op(ifupdownobj, ifaceobj, op, cenv)
-            except Exception, e:
-                ifupdownobj.log_error(str(e))
-
-    @classmethod
-    def run_iface_list_ops(cls, ifupdownobj, ifacenames, ops,
-                           sorted_by_dependency=False):
-        """ Runs interface list through sub operations handler
-
-        Unlike run_iface_list, this method executes a sub operation on the
-        entire interface list before proceeding to the next sub-operation.
-        ie operation 'pre-up' is run through the entire interface list before
-        'up'
-        """
-        # Each sub operation has a module list
-        [cls.run_iface_list_op(ifupdownobj, ifacenames, op,
-                sorted_by_dependency) for op in ops]
-
-    @classmethod
-    def run_iface_dependency_graphs_sorted(cls, ifupdownobj,
-                                           dependency_graphs,
-                                           ops, indegrees=None,
-                                           graphsortall=False):
-        """ runs interface dependency graph by topologically sorting the interfaces """
-
-        if indegrees is None:
-            indegrees = OrderedDict()
-            for ifacename in dependency_graphs.keys():
-                indegrees[ifacename] = ifupdownobj.get_iface_refcnt(ifacename)
-
-        ifupdownobj.logger.debug('indegree array :')
-        ifupdownobj.logger.debug(ifupdownobj.pp.pformat(indegrees))
-
-        try:
-            ifupdownobj.logger.debug('calling topological sort on the graph ' +
-                                      '...')
-            if graphsortall:
-                sorted_ifacenames = graph.topological_sort_graphs_all(
-                                            dependency_graphs, indegrees)
-            else:
-                sorted_ifacenames = graph.topological_sort_graphs(
-                                            dependency_graphs, indegrees)
-        except Exception:
-            raise
-
-        ifupdownobj.logger.debug('sorted iface list = %s' %sorted_ifacenames)
-        cls.run_iface_list_ops(ifupdownobj, sorted_ifacenames, ops,
-                               sorted_by_dependency=True)
-
-
-    """ Methods to execute interfaces in parallel """
-    @classmethod
-    def init_tokens(cls, count):
-        cls.token_pool = BoundedSemaphore(count)
-
-    @classmethod
-    def accquire_token(cls, logprefix=''):
-        cls.token_pool.acquire()
-
-    @classmethod
-    def release_token(cls, logprefix=''):
-        cls.token_pool.release()
-
-    @classmethod
-    def run_iface_parallel(cls, ifupdownobj, ifacename, op):
-        """ Configures interface in parallel.
-        
-        Executes all its direct dependents in parallel
-        
-        """
-
-        ifupdownobj.logger.debug('%s:' %ifacename + ' %s' %op)
-        cls.accquire_token(iface)
-
-        # Each iface can have a list of objects
-        ifaceobjs = ifupdownobj.get_ifaceobjs(ifacename)
-        if ifaceobjs is None:
-            ifupdownobj.logger.warning('%s: ' %ifacename + 'not found')
-            cls.release_token(ifacename)
-            return -1
-
-        for ifaceobj in ifaceobjs:
-            # Run dependents
-            dlist = ifaceobj.lowerifaces
-            if dlist:
-                ifupdownobj.logger.debug('%s:' %ifacename +
-                    ' found dependents: %s' %str(dlist))
-                try:
-                    cls.release_token(ifacename)
-                    cls.run_iface_list_parallel(ifacename, ifupdownobj,
-                                                 dlist, op)
-                    cls.accquire_token(ifacename)
-                except Exception, e:
-                    if ifupdownobj.ignore_error(str(e)):
-                        pass
-                    else:
-                        # Dont bring the iface up if children did not come up
-                        ifupdownobj.logger.debug('%s:' %ifacename +
-                            ' there was an error bringing %s' %op +
-                            ' dependents (%s)', str(e))
-                        ifupdownobj.set_iface_state(ifaceobj,
-                            ifaceState.from_str(ops[0]),
-                            ifaceStatus.ERROR)
-                        return -1
-
-            # Run all sub operations sequentially
-            try:
-                ifupdownobj.logger.debug('%s:' %ifacename +
-                                         ' running sub-operations')
-                cls.run_iface_ops(ifupdownobj, ifaceobj, op)
-            except Exception, e:
-                ifupdownobj.logger.error('%s:' %ifacename +
-                    ' error running sub operations (%s)' %str(e))
-
-        cls.release_token(ifacename)
-
-    @classmethod
-    def run_iface_list_parallel(cls, parent, ifupdownobj, ifacenames, op):
-        """ Runs interface list in parallel """
-
-        running_threads = OrderedDict()
-        err = 0
-
-        for ifacename in ifacenames:
-            try:
-                cls.accquire_token(parent)
-                running_threads[ifacename] = Thread(None,
-                    cls.run_iface_parallel, ifacename,
-                    args=(ifupdownobj, ifacename, op))
-                running_threads[ifacename].start()
-                cls.release_token(parent)
-            except Exception, e:
-                cls.release_token(parent)
-                if ifupdownobj.ignore_error(str(e)):
-                    pass
-                else:
-                    raise Exception('error starting thread for iface %s'
-                            %ifacename)
-
-
-        ifupdownobj.logger.debug('%s ' %parent +
-                                 'waiting for all the threads ...')
-        for ifacename, t  in running_threads.items():
-            t.join()
-            if ifupdownobj.get_iface_status(ifacename) != ifaceStatus.SUCCESS:
-                err += 1
-
-        return err
-
-    @classmethod
-    def run_iface_graphs_parallel(cls, parent, ifupdownobj, ifacenames, op):
-        """ Runs iface graphs in parallel """
-
-        running_threads = OrderedDict()
-        err = 0
-
-        for ifacename in ifacenames:
-            try:
-                cls.accquire_graph_token(parent)
-                running_threads[ifacename] = Thread(None,
-                    cls.run_iface_parallel, ifacename,
-                    args=(ifupdownobj, ifacename, op))
-                running_threads[ifacename].start()
-                cls.release_graph_token(parent)
-            except Exception, e:
-                cls.release_graph_token(parent)
-                if ifupdownobj.ignore_error(str(e)):
-                    pass
-                else:
-                    raise Exception('error starting thread for iface %s'
-                            %ifacename)
-
-        ifupdownobj.logger.info('%s ' %parent +
-                                'waiting for all the threads ...')
-        for ifacename, t in running_threads.items():
-            t.join()
-            # Check status of thread
-            # XXX: Check all objs
-            if ifupdownobj.get_iface_status(ifacename) != ifaceStatus.SUCCESS:
-                err += 1
-        return err
-
-    @classmethod
-    def run_iface_dependency_graph_parallel(cls, ifupdownobj, dependency_graph,
-                                            operation):
-        """ Runs iface dependeny graph in parallel.
-        
-        arguments:
-        ifupdownobj -- ifupdown object (used for getting and updating iface
-                                        object state)
-        dependency_graph -- dependency graph with 
-        operation -- 'up' or 'down' or 'query'
-
-        """
-
-        ifupdownobj.logger.debug('running dependency graph in parallel ..')
-        run_queue = []
-        # Build a list of ifaces that dont have any dependencies
-        for ifacename in dependency_graph.keys():
-            if ifupdownobj.get_iface_refcnt(ifacename) == 0:
-                run_queue.append(ifacename)
-
-        ifupdownobj.logger.debug('graph roots (interfaces that dont'
-                    ' have dependents):' + ' %s' %str(run_queue))
-        cls.init_tokens(ifupdownobj.get_njobs())
-        return cls.run_iface_list_parallel('main', ifupdownobj, run_queue,
-                                            operation)
-
-        # OR
-        # Run one graph at a time
-        #for iface in run_queue:
-        #    self.run_iface_list_parallel('main', ifupdownobj, [iface],
-        #            operation)
-