From e2dbcb7e5c8ff5bf7ea679a17660f863aac66232 Mon Sep 17 00:00:00 2001
From: Tobias Klauser <tklauser@distanz.ch>
Date: Mon, 7 Jan 2013 22:34:12 +0100
Subject: [PATCH] scripts: update-oui: Add OUI list update script

Instead of having to maually update the oui.conf file used by
netsniff-ng to resolve OUIs to vendor names, provide a script which
fetches the necessary information from any file in the format as
provided by the IEEE at
http://standards.ieee.org/develop/regauth/oui/oui.txt

This is also the default URL the script will fetch the information from.

The script runs with Python 2.x and 3.x and only uses the respective
standard libraries.

Signed-off-by: Tobias Klauser <tklauser@distanz.ch>
---
 scripts/update-oui.py | 81 +++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 81 insertions(+)
 create mode 100755 scripts/update-oui.py

diff --git a/scripts/update-oui.py b/scripts/update-oui.py
new file mode 100755
index 00000000..0b676e1e
--- /dev/null
+++ b/scripts/update-oui.py
@@ -0,0 +1,81 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+#
+# update-oui.py -- update netsniff-ng oui.conf from official IEEE OUI list
+#
+# Copyright (C) 2013 Tobias Klauser <tklauser@distanz.ch>
+#
+# This program is free software; you can redistribute it and/or modify
+# it under the terms of the GNU General Public License version 2 as
+# published by the Free Software Foundation.
+
+import os
+import sys
+import re
+import getopt
+try:
+    from urllib2 import urlopen             # Python 2.x
+except ImportError:
+    from urllib.request import urlopen      # Python 3.x
+
+DEFAULT_OUPUT_FILE = "oui.conf"
+DEFAULT_OUI_URL = "http://standards.ieee.org/develop/regauth/oui/oui.txt"
+
+OUI_PATTERN = re.compile(b"^([a-zA-Z0-9]{2})-([a-zA-Z0-9]{2})-([a-zA-Z0-9]{2})\s+\(hex\)\s+(.*)$")
+
+def usage():
+    print("""usage: {0} [OPTION...]
+available options:
+    -f  force overwrite of existing file
+    -o  set output file (default: {1})
+    -u  set URL to fetch OUI list from (default: {2})
+    -h  show this help and exit""".format(os.path.basename(sys.argv[0]),
+                                          DEFAULT_OUPUT_FILE, DEFAULT_OUI_URL))
+
+def main():
+    try:
+        opts, args = getopt.getopt(sys.argv[1:], "fo:u:h")
+    except getopt.GetoptError as err:
+        print(str(err))
+        usage()
+        sys.exit(-1)
+
+    overwrite = False
+    output_file = DEFAULT_OUPUT_FILE
+    oui_url = DEFAULT_OUI_URL
+    for o, a in opts:
+        if o == '-f':
+            overwrite = True
+        elif o == '-o':
+            output_file = a
+        elif o == '-u':
+            oui_url = a
+        elif o == '-h':
+            usage()
+            sys.exit(0)
+        else:
+            assert False, "unhandled option"
+
+    if not overwrite and os.path.exists(output_file):
+        print("Eror: output file {} already exists".format(output_file))
+        sys.exit(-1)
+
+    print("Updating OUI information in {} from {}... ".format(output_file, oui_url))
+
+    fh_file = open(output_file, 'w')
+    fh_url = urlopen(oui_url)
+
+    n = 0
+    for line in fh_url:
+        m = OUI_PATTERN.match(line)
+        if m:
+            fh_file.write("0x{}{}{}, {}\n".format(m.group(1), m.group(2), m.group(3), m.group(4)))
+            n += 1
+
+    print("{} OUIs written to {}".format(n, output_file))
+
+    fh_url.close()
+    fh_file.close()
+
+if __name__ == '__main__':
+    main()
-- 
2.11.4.GIT