mirror of
https://github.com/SpacehuhnTech/esp8266_deauther.git
synced 2025-12-15 03:00:08 +01:00
82 lines
2.3 KiB
Python
82 lines
2.3 KiB
Python
#/usr/bin/env python
|
|
"""This scripts downloads the last OUI manufaturer file from the Whireshark
|
|
project and converts it to esp8266_deauther format"""
|
|
|
|
import argparse
|
|
from urllib.request import urlopen
|
|
|
|
WS_MANUF_FILE_URL = "https://code.wireshark.org/review/gitweb?p=wireshark.git;a=blob_plain;f=manuf"
|
|
|
|
def parse_options():
|
|
"""Parses command line options"""
|
|
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument("-o", "--output", help="Output file name (e.g oui.h)")
|
|
parser.add_argument("-u", "--url", help="Wireshark oui/manuf file url")
|
|
|
|
opt = parser.parse_args()
|
|
|
|
return opt
|
|
|
|
def generate_oui_h(url, filename):
|
|
"""Generates the vendors/oui file"""
|
|
|
|
if url:
|
|
data = urlopen(url)
|
|
else:
|
|
data = urlopen(WS_MANUF_FILE_URL)
|
|
|
|
out = """#ifndef oui_h
|
|
#define oui_h
|
|
/*
|
|
Based on Wireshark manufacturer database
|
|
source: https://www.wireshark.org/tools/oui-lookup.html
|
|
Wireshark is released under the GNU General Public License version 2
|
|
*/
|
|
|
|
const static uint8_t data_vendors[] PROGMEM = {///*
|
|
"""
|
|
|
|
for line in data:
|
|
line = line.decode()
|
|
|
|
# Skipping empty lines and comments
|
|
if line.startswith('#') or line.startswith('\n'):
|
|
continue
|
|
|
|
mac, short_desc, *rest = line.strip().split('\t')
|
|
|
|
# Limiting short_desc to 8 chars
|
|
short_desc = short_desc[0:8]
|
|
|
|
# Convert to ascii
|
|
short_desc = short_desc.encode("ascii", "ignore").decode()
|
|
|
|
mac_octects = len(mac.split(':'))
|
|
if mac_octects == 6:
|
|
continue
|
|
else:
|
|
# Convert to esp8266_deauther format
|
|
short_desc = short_desc.ljust(8, '\0')
|
|
hex_sdesc = ", 0x".join("{:02x}".format(ord(c)) for c in short_desc)
|
|
|
|
(oc1, oc2, oc3) = mac.split(':')
|
|
|
|
out = out + (" 0x{}, 0x{}, 0x{}, 0x{},\n".format(oc1.upper(), oc2.upper(), oc3.upper(),
|
|
hex_sdesc.upper().replace('X', 'x')))
|
|
|
|
out = out[:-2] # Removing last comma
|
|
out = out + "\n};\n#endif"
|
|
|
|
# Saving to file
|
|
if filename:
|
|
with open(filename, 'w') as out_file:
|
|
out_file.write("%s" % out)
|
|
else:
|
|
print(out)
|
|
|
|
if __name__ == "__main__":
|
|
options = parse_options()
|
|
generate_oui_h(options.url, options.output)
|
|
|