80 lines
3.0 KiB
Python
Executable File
80 lines
3.0 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
#
|
|
# make-init-lua.py
|
|
#
|
|
# By Gerald Combs <gerald@wireshark.org>
|
|
# Based on make-init-lua.pl by Luis E. Garcia Onatnon <luis.ontanon@gmail.com>
|
|
#
|
|
# Wireshark - Network traffic analyzer
|
|
# By Gerald Combs <gerald@wireshark.org>
|
|
# Copyright 1998 Gerald Combs
|
|
#
|
|
# SPDX-License-Identifier: GPL-2.0-or-later
|
|
|
|
import argparse
|
|
import os
|
|
import re
|
|
|
|
from enum import Enum
|
|
from string import Template
|
|
|
|
def main():
|
|
parser = argparse.ArgumentParser(description="Generate the registration macros for Lua code.")
|
|
parser.add_argument("template", metavar='template', help="template file")
|
|
parser.add_argument("output", metavar='output', help="output file")
|
|
args = parser.parse_args()
|
|
|
|
this_dir = os.path.dirname(__file__)
|
|
src_root = os.path.join(this_dir, '..', '..')
|
|
|
|
replacements = {
|
|
'WTAP_ENCAPS': '-- Wiretap encapsulations XXX\nwtap_encaps = {',
|
|
'WTAP_TSPRECS': '-- Wiretap timestamp precision types\nwtap_tsprecs = {',
|
|
'WTAP_COMMENT_TYPES': '-- Wiretap file comment types\nwtap_comments = {',
|
|
'WTAP_REC_TYPES': '-- Wiretap record_types\nwtap_rec_types = {',
|
|
'WTAP_PRESENCE_FLAGS': '-- Wiretap presence flags\nwtap_presence_flags = {',
|
|
}
|
|
|
|
with open(args.template, encoding='utf-8') as tmpl_f:
|
|
template = Template(tmpl_f.read())
|
|
|
|
wtap_encaps = []
|
|
wtap_tsprecs = []
|
|
wtap_comment_types = []
|
|
wtap_rec_types = []
|
|
wtap_presence_flags = []
|
|
with open(os.path.join(src_root, 'wiretap', 'wtap.h'), encoding='utf-8') as wtap_f:
|
|
for line in wtap_f:
|
|
m = re.search(r'#define WTAP_ENCAP_([A-Z0-9_]+)\s+(-?\d+)', line)
|
|
if m:
|
|
wtap_encaps.append(f'\n\t["{m.group(1)}"] = {m.group(2)}')
|
|
|
|
m = re.search(r'#define WTAP_TSPREC_([A-Z0-9_]+)\s+(\d+)', line)
|
|
if m:
|
|
wtap_tsprecs.append(f'\n\t["{m.group(1)}"] = {m.group(2)}')
|
|
|
|
m = re.search(r'#define WTAP_COMMENT_([A-Z0-9_]+)\s+(0x\d+)', line)
|
|
if m:
|
|
wtap_comment_types.append(f'\n\t["{m.group(1)}"] = {m.group(2)}')
|
|
|
|
m = re.search(r'#define REC_TYPE_([A-Z0-9_]+)\s+(\d+)\s+\/\*\*<([^\*]+)\*\/', line)
|
|
if m:
|
|
wtap_rec_types.append(f'\n\t["{m.group(1)}"] = {m.group(2)}, --{m.group(3)}')
|
|
|
|
m = re.search(r'#define WTAP_HAS_([A-Z0-9_]+)\s+(0x\d+)\s+\/\*\*<([^\*]+)\*\/', line)
|
|
if m:
|
|
wtap_presence_flags.append(f'\n\t["{m.group(1)}"] = {int(m.group(2), 16)}, --{m.group(3)}')
|
|
|
|
replacements['WTAP_ENCAPS'] += ','.join(wtap_encaps) + '\n}\nwtap = wtap_encaps -- for bw compatibility\n'
|
|
replacements['WTAP_TSPRECS'] += ','.join(wtap_tsprecs) + '\n}\n'
|
|
replacements['WTAP_COMMENT_TYPES'] += ','.join(wtap_comment_types) + '\n}\n'
|
|
replacements['WTAP_REC_TYPES'] += ''.join(wtap_rec_types) + '\n}\n'
|
|
replacements['WTAP_PRESENCE_FLAGS'] += ''.join(wtap_presence_flags) + '\n}\n'
|
|
|
|
with open(args.output, mode='w', encoding='utf-8') as out_f:
|
|
out_f.write(template.substitute(replacements))
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|