1

#!/usr/bin/python

     2

     3

"""

     4

    Analyze WLAN STA logs.

     5

     6

        Jul  3 23:05:04 buffalo-g300n-647682 daemon.info hostapd: wlan0-1: STA aa:bb:cc:dd:ee:ff WPA: group key handshake completed (RSN)

     7

     8

"""

     9

    10

__version__ = '0.1'

    11

    12

import pvl.args

    13

import pvl.syslog.args

    14

import pvl.rrd.hosts

    15

    16

import optparse

    17

import logging; log = logging.getLogger('main')

    18

    19

WLAN_STA_PROG = 'hostapd'

    20

    21

def parse_argv (argv, doc = __doc__) :

    22

    """

    23

        Parse command-line argv, returning (options, args).

    24

    """

    25

    26

    prog = argv.pop(0)

    27

    args = argv

    28

    29

    # optparse

    30

    parser = optparse.OptionParser(

    31

        prog        = prog,

    32

        usage       = '%prog: [options] [<input.txt> [...]]',

    33

        version     = __version__,

    34

        description = doc,

    35

    )

    36

    37

    # common

    38

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

    39

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

    40

    41

    parser.add_option('--interfaces', metavar='PATH',

    42

            help="Load interface/node names from mapping file")

    43

    44

    parser.set_defaults(

    45

    )

    46

    47

    # parse

    48

    options, args = parser.parse_args(args)

    49

    50

    # apply

    51

    pvl.args.apply(options)

    52

    53

    return options, args

    54

    55

import re

    56

    57

class HostapdHandler (object) :

    58

    HOSTAPD_STA_RE = re.compile(r'(?P<wlan>.+?): STA (?P<sta>.+?) (?P<msg>.+)')

    59

    60

    def __init__ (self, interface_map=None) :

    61

        """

    62

            interface_map       - {(hostname, interface): (nodename, wlan)}

    63

        """

    64

        self.interface_map = interface_map

    65

    66

    def parse (self, item) :

    67

        """

    68

            Parse fields from a hostapd syslog message.

    69

        """

    70

    71

        match = self.HOSTAPD_STA_RE.match(item['msg'])

    72

    73

        if not match :

    74

            return None

    75

    76

        return dict(item, **match.groupdict())

    77

    78

    def build_sta (self, match) :

    79

        ap, wlan = match['host'], match['wlan']

    80

    81

        # override?

    82

        if self.interface_map :

    83

            mapping = self.interface_map.get((ap, wlan))

    84

    85

            if mapping :

    86

                ap, wlan = mapping

    87

    88

        build = dict(

    89

                timestamp       = match['timestamp'],

    90

                ap              = ap,

    91

                wlan            = wlan,

    92

                sta             = match['sta'],

    93

                msg             = match['msg'],

    94

        )

    95

    96

    97

        return build

    98

    99

def main (argv) :

   100

    options, args = parse_argv(argv)

   101

   102

    if options.interfaces :

   103

        interfaces = dict(pvl.rrd.hosts.map_interfaces(options, open(options.interfaces)))

   104

    else :

   105

        interfaces = None

   106

   107

    # syslog

   108

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

   109

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

   110

    handler = HostapdHandler(interface_map=interfaces)

   111

   112

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

   113

    for source in syslog.main() :

   114

        for item in source:

   115

            match = handler.parse(item)

   116

   117

            if not match :

   118

                continue

   119

            elif 'sta' in match :

   120

                sta = handler.build_sta(match)

   121

            else :

   122

                continue

   123

   124

            print "{ap:>30}/{wlan:10} {sta:20} : {msg}".format(**sta)

   125

   126

    return 0

   127

   128

if __name__ == '__main__':

   129

    import sys

   130

   131

    sys.exit(main(sys.argv))