1

#!/usr/bin/env python

     2

     3

"""

     4

    Monitor DHCP use.

     5

"""

     6

     7

__version__ = '0.1'

     8

     9

import pvl.args

    10

import pvl.syslog.args

    11

    12

import pvl.verkko.db as db

    13

import pvl.dhcp.syslog

    14

import pvl.dhcp.hosts

    15

    16

import logging, optparse

    17

    18

log = logging.getLogger('main')

    19

    20

# name of process in syslog

    21

DHCP_SYSLOG_PROG = 'dhcpd'

    22

    23

def parse_options (argv) :

    24

    """

    25

        Parse command-line arguments.

    26

    """

    27

    28

    prog = argv[0]

    29

    30

    parser = optparse.OptionParser(

    31

            prog        = prog,

    32

            usage       = '%prog: [options]',

    33

            version     = __version__,

    34

    35

            # module docstring

    36

            description = __doc__,

    37

    )

    38

    39

    # options

    40

    parser.add_option_group(pvl.args.parser(parser))

    41

    42

    ## syslog

    43

    parser.add_option_group(pvl.syslog.args.parser(parser, prog=DHCP_SYSLOG_PROG))

    44

    45

    ## XXX: networks

    46

    parser.add_option('--network',              metavar='NET', action='append',

    47

            help="Filter leases by network prefix as plugin instance")

    48

    49

    parser.add_option('--gateway',              metavar='GW/IFACE', action='append',

    50

            help="Filter messages by gateway/interface as plugin instance")

    51

    52

    ## hosts

    53

    parser.add_option('--database',             metavar='URI',

    54

            help="Track hosts in given database")

    55

    56

    parser.add_option('--create',               action='store_true',

    57

            help="Initialize database")

    58

    59

    # defaults

    60

    parser.set_defaults(

    61

    62

    )

    63

    64

    # parse

    65

    options, args = parser.parse_args(argv[1:])

    66

    67

    # apply

    68

    pvl.args.apply(options, prog)

    69

    70

    if not options.database :

    71

        parser.error("Missing required option: --database")

    72

    73

    return options, args

    74

    75

def main (argv) :

    76

    options, args = parse_options(argv)

    77

    78

    # db

    79

    if not options.database :

    80

        log.error("No database given")

    81

        return 1

    82

    83

    log.info("Open up database: %s", options.database)

    84

    db = pvl.verkko.db.Database(options.database)

    85

    hosts_db = pvl.dhcp.hosts.DHCPHostsDatabase(db)

    86

    87

    if options.create :

    88

        hosts_db.create()

    89

    90

    # syslog

    91

    log.info("Open up syslog...")

    92

    syslog = pvl.syslog.args.apply(options)

    93

    parser = pvl.dhcp.syslog.DHCPSyslogParser()

    94

    95

    log.info("Enter mainloop...")

    96

    for source in syslog.main() :

    97

        # parse dhcp messages from syslog

    98

        for host in parser(source) :

    99

            log.debug("%s: %s", source, host)

   100

   101

            hosts_db(host)

   102

   103

    # done

   104

    return 0

   105

   106

if __name__ == '__main__':

   107

    import sys

   108

   109

    sys.exit(main(sys.argv))