update-oui.py

   1 #!/usr/bin/env python
   2 # -*- coding: utf-8 -*-
   3 #
   4 # update-oui.py -- update netsniff-ng oui.conf from official IEEE OUI list
   5 #
   6 # Copyright (C) 2013 Tobias Klauser <tklauser@distanz.ch>
   7 #
   8 # This program is free software; you can redistribute it and/or modify
   9 # it under the terms of the GNU General Public License version 2 as
  10 # published by the Free Software Foundation.
  11
  12 import os
  13 import sys
  14 import re
  15 import getopt
  16 try:
  17     from urllib2 import urlopen             # Python 2.x
  18 except ImportError:
  19     from urllib.request import urlopen      # Python 3.x
  20
  21 DEFAULT_OUPUT_FILE = "oui.conf"
  22 DEFAULT_OUI_URL = "http://standards.ieee.org/develop/regauth/oui/oui.txt"
  23
  24 OUI_PATTERN = re.compile(b"^([a-zA-Z0-9]{2})-([a-zA-Z0-9]{2})-([a-zA-Z0-9]{2})\s+\(hex\)\s+(.*)$")
  25
  26 def usage():
  27     print("""usage: {0} [OPTION...]
  28 available options:
  29     -f  force overwrite of existing file
  30     -o  set output file (default: {1})
  31     -u  set URL to fetch OUI list from (default: {2})
  32     -h  show this help and exit""".format(os.path.basename(sys.argv[0]),
  33                                           DEFAULT_OUPUT_FILE, DEFAULT_OUI_URL))
  34
  35 def main():
  36     try:
  37         opts, args = getopt.getopt(sys.argv[1:], "fo:u:h")
  38     except getopt.GetoptError as err:
  39         print(str(err))
  40         usage()
  41         sys.exit(-1)
  42
  43     overwrite = False
  44     output_file = DEFAULT_OUPUT_FILE
  45     oui_url = DEFAULT_OUI_URL
  46     for o, a in opts:
  47         if o == '-f':
  48             overwrite = True
  49         elif o == '-o':
  50             output_file = a
  51         elif o == '-u':
  52             oui_url = a
  53         elif o == '-h':
  54             usage()
  55             sys.exit(0)
  56         else:
  57             assert False, "unhandled option"
  58
  59     if not overwrite and os.path.exists(output_file):
  60         print("Eror: output file {} already exists".format(output_file))
  61         sys.exit(-1)
  62
  63     print("Updating OUI information in {} from {}... ".format(output_file, oui_url))
  64
  65     fh_file = open(output_file, 'w')
  66     fh_url = urlopen(oui_url)
  67
  68     n = 0
  69     for line in fh_url:
  70         m = OUI_PATTERN.match(line)
  71         if m:
  72             fh_file.write("0x{}{}{}, {}\n".format(m.group(1), m.group(2), m.group(3), m.group(4)))
  73             n += 1
  74
  75     print("{} OUIs written to {}".format(n, output_file))
  76
  77     fh_url.close()
  78     fh_file.close()
  79
  80 if __name__ == '__main__':
  81     main()