Some fixes + enhancements

[mirror_ifupdown2.git] / sbin / ifupdown

#!/usr/bin/python

import sys
import os
import re
import argparse
from ifupdown.ifupdown_main import *

import logging

lockfile="/run/network/.lock"
logger = None

def run(args, op):
    logger.debug('args = %s' %str(args))

    try:
        logger.debug('creating ifupdown object ..')
        if op == 'up' or op == 'down':
            ifupdown_handle = ifupdown_main(force=args.force,
                                            dryrun=args.noact,
                                            nodepends=args.nodepends,
                                            perfmode=args.perfmode,
                                            njobs=args.jobs)
        elif op == 'query':
            ifupdown_handle = ifupdown_main(dryrun=args.noact,
                                            nodepends=args.nodepends,
                                            perfmode=args.perfmode)

        iflist = args.iflist
        if len(args.iflist) == 0:
            iflist = None

        logger.debug('calling %s' %op + ' for all interfaces ..')
        if op == 'up':
            ifupdown_handle.up(args.all, args.CLASS, iflist)
        elif op == 'down':
            ifupdown_handle.down(args.all, args.CLASS, iflist)
        elif op == 'query':
            if args.curstate == True:
                qstate='curr'
            elif args.presumedstate == True:
                qstate='presumed'
            elif args.presumedstatedetailed == True:
                qstate='presumeddetailed'
            else:
                qstate=None
            ifupdown_handle.query(args.all, args.CLASS, iflist,
                                  query_state=qstate)
    except:
        raise

def init(args):
    global logger

    log_level = logging.WARNING

    if args.verbose == True:
        log_level = logging.INFO

    if args.debug == True:
        log_level = logging.DEBUG

    try:
        logging.basicConfig(level=log_level,
            format='%(levelname)s: %(message)s')
        logger = logging.getLogger('ifupdown')
    except:
        raise


def deinit():
    {}

def update_argparser(argparser):

    argparser.add_argument('iflist', metavar='IFACE',
                nargs='*', help='interfaces list')
    argparser.add_argument('-a', '--all', action='store_true',
                help='process all interfaces marked \"auto\"')
    argparser.add_argument('-n', '--no-act', dest='noact',
                action='store_true', help='print out what would happen,' +
                'but don\'t do it')
    argparser.add_argument('-v', '--verbose', dest='verbose',
                action='store_true', help='verbose')
    argparser.add_argument('-d', '--debug', dest='debug',
                action='store_true',
                help='output debug info')
    argparser.add_argument('-q', '--quiet', dest='quiet',
                action='store_true',
                help=argparse.SUPPRESS)
    argparser.add_argument('--allow', dest='CLASS', 
                help='ignore non-\"allow-CLASS\" interfaces')
    argparser.add_argument('--nodepends', dest='nodepends',
                action='store_true', help=argparse.SUPPRESS)
    argparser.add_argument('--performance-mode', dest='perfmode',
                action='store_true', help=argparse.SUPPRESS)


def update_ifupdown_argparser(argparser):
    argparser.add_argument('-f', '--force', dest='force',
                action='store_true',
                help='force run all operations')
    argparser.add_argument('-j', '--jobs', dest='jobs', type=int,
                default=-1, choices=range(1,12), help=argparse.SUPPRESS)

def update_ifup_argparser(argparser):
    update_ifupdown_argparser(argparser)

def update_ifdown_argparser(argparser):
    update_ifupdown_argparser(argparser)

def update_ifquery_argparser(argparser):
    group = argparser.add_mutually_exclusive_group(required=False)
    group.add_argument('-r', '--running-state', dest='curstate',
                       action='store_true',
                       help='query running state of an interface')

    # presumed-state is state maintained by ifupdown
    group.add_argument('--presumed-state', dest='presumedstate',
                       action='store_true', help=argparse.SUPPRESS)

    # presumed-state-detailed prints all details about the object stored
    # by ifupdown
    group.add_argument('--presumed-state-detailed',
                       dest='presumedstatedetailed',
                       action='store_true', help=argparse.SUPPRESS)

def parse_args(argsv, op):
    descr = 'interface management'

    argparser = argparse.ArgumentParser(description=descr)
    update_argparser(argparser)
    if op == 'up':
        update_ifup_argparser(argparser)
    elif op == 'down':
        update_ifdown_argparser(argparser)
    elif op == 'query':
        update_ifquery_argparser(argparser)

    return argparser.parse_args(argsv)

def main(argv):
    """ main function """
    try:
        op = None
        if re.search(r'ifup', argv[0]) != None:
            op = 'up'
        elif re.search(r'ifdown', argv[0]) != None:
            op = 'down'
        elif re.search(r'ifquery', argv[0]) != None:
            op = 'query'
        else:
            print ('Unexpected executable.' +
                   ' Should be \'ifup\' or \'ifdown\' or \'ifquery\'')
            exit(1)

        # Command line arg parser
        args = parse_args(argv[1:], op)
        if len(args.iflist) > 0 and args.all is True:
            print 'iflist and all are mutually exclusive'
            exit(1)

        init(args)
        run(args, op)
    except Exception, e:
        if str(e) == '':
            exit(1)

        if args.debug == True:
            raise
        else:
            logger.error(str(e))
        exit(1)
    finally:
        deinit()

if __name__ == "__main__":

    if not os.geteuid() == 0:
        print 'Error: Must be root to run this command'
        exit(1)

    """
    XXX: Cannot use this. A spawned dhclient process can hold the lock
    if not utilities.lockFile(lockfile):
        print 'Another instance of this program is already running.'
        exit(0)
    """

    main(sys.argv)