forked from osmocom/wireshark
119 lines
3.0 KiB
Python
Executable File
119 lines
3.0 KiB
Python
Executable File
#!/usr/bin/python
|
|
#
|
|
# Looks for registration routines in the protocol dissectors,
|
|
# and assembles C code to call all the routines.
|
|
#
|
|
# This is a Python version of the make-reg-dotc shell script.
|
|
# Running the shell script on Win32 is very very slow because of
|
|
# all the process-launching that goes on --- multiple greps and
|
|
# seds for each input file. I wrote this python version so that
|
|
# less processes would have to be started.
|
|
#
|
|
# $Id: make-reg-dotc.py,v 1.4 2003/07/21 07:05:44 girlich Exp $
|
|
|
|
import os
|
|
import sys
|
|
import re
|
|
|
|
tmp_filename = "register.c-tmp"
|
|
final_filename = "register.c"
|
|
|
|
#
|
|
# The first argument is the directory in which the source files live.
|
|
#
|
|
srcdir = sys.argv[1]
|
|
|
|
#
|
|
# All subsequent arguments are the files to scan.
|
|
#
|
|
files = sys.argv[2:]
|
|
|
|
reg_code = open(tmp_filename, "w")
|
|
|
|
reg_code.write("/* Do not modify this file. */\n")
|
|
reg_code.write("/* It is created automatically by the Makefile. */\n")
|
|
reg_code.write('#include "register.h"\n')
|
|
|
|
# Create the proper list of filenames
|
|
filenames = []
|
|
for file in files:
|
|
if os.path.isfile(file):
|
|
filenames.append(file)
|
|
else:
|
|
filenames.append("%s/%s" % (srcdir, file))
|
|
|
|
|
|
# Look through all files, applying the regex to each line.
|
|
# If the pattern matches, save the "symbol" section to the
|
|
# appropriate array.
|
|
proto_reg = []
|
|
handoff_reg = []
|
|
|
|
# For those that don't know Python, r"" indicates a raw string,
|
|
# devoid of Python escapes.
|
|
proto_regex0 = r"^(?P<symbol>proto_register_[_A-Za-z0-9]+)\s*\([^;]+$"
|
|
proto_regex1 = r"void\s+(?P<symbol>proto_register_[_A-Za-z0-9]+)\s*\([^;]+$"
|
|
|
|
handoff_regex0 = r"^(?P<symbol>proto_reg_handoff_[_A-Za-z0-9]+)\s*\([^;]+$"
|
|
handoff_regex1 = r"void\s+(?P<symbol>proto_reg_handoff_[_A-Za-z0-9]+)\s*\([^;]+$"
|
|
|
|
# This table drives the pattern-matching and symbol-harvesting
|
|
patterns = [
|
|
( proto_reg, re.compile(proto_regex0) ),
|
|
( proto_reg, re.compile(proto_regex1) ),
|
|
( handoff_reg, re.compile(handoff_regex0) ),
|
|
( handoff_reg, re.compile(handoff_regex1) ),
|
|
]
|
|
|
|
# Grep
|
|
for filename in filenames:
|
|
file = open(filename)
|
|
# print "Searching %s" % (filename)
|
|
for line in file.readlines():
|
|
for action in patterns:
|
|
regex = action[1]
|
|
match = regex.search(line)
|
|
if match:
|
|
symbol = match.group("symbol")
|
|
list = action[0]
|
|
list.append(symbol)
|
|
file.close()
|
|
|
|
# Sort the lists to make them pretty
|
|
proto_reg.sort()
|
|
handoff_reg.sort()
|
|
|
|
# Make register_all_protocols()
|
|
reg_code.write("void register_all_protocols(void) {\n")
|
|
|
|
for symbol in proto_reg:
|
|
line = " {extern void %s (void); %s ();}\n" % (symbol, symbol)
|
|
reg_code.write(line)
|
|
|
|
reg_code.write("}\n")
|
|
|
|
|
|
# Make register_all_protocol_handoffs()
|
|
reg_code.write("void register_all_protocol_handoffs(void) {\n")
|
|
|
|
for symbol in handoff_reg:
|
|
line = " {extern void %s (void); %s ();}\n" % (symbol, symbol)
|
|
reg_code.write(line)
|
|
|
|
reg_code.write("}\n")
|
|
|
|
# Close the file
|
|
reg_code.close()
|
|
|
|
# Remove the old final_file if it exists.
|
|
try:
|
|
os.stat(final_filename)
|
|
os.remove(final_filename)
|
|
except OSError:
|
|
pass
|
|
|
|
# Move from tmp file to final file
|
|
os.rename(tmp_filename, final_filename)
|
|
|
|
|