forked from acouzens/open5gs
script to add prefix ASN structure and constant for avoiding conflict
global variable
This commit is contained in:
parent
6da9c34afb
commit
581ac3f577
|
@ -0,0 +1,160 @@
|
|||
#
|
||||
# Copyright (c) 2017, CellWire Group
|
||||
# All rights reserved.
|
||||
#
|
||||
# Redistribution and use in source and binary forms, with or without
|
||||
# modification, are permitted provided that the following conditions are met:
|
||||
#
|
||||
# 1. Redistributions of source code must retain the above copyright notice, this
|
||||
# list of conditions and the following disclaimer.
|
||||
# 2. Redistributions in binary form must reproduce the above copyright notice,
|
||||
# this list of conditions and the following disclaimer in the documentation
|
||||
# and/or other materials provided with the distribution.
|
||||
#
|
||||
# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
|
||||
# ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
|
||||
# WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
|
||||
# DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR
|
||||
# ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
|
||||
# (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
|
||||
# LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
|
||||
# ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
|
||||
# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
|
||||
# SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
||||
#
|
||||
import re, os, sys, string
|
||||
import datetime
|
||||
import getopt
|
||||
import getpass
|
||||
|
||||
version = "0.1.0"
|
||||
|
||||
lines = ""
|
||||
|
||||
in_filename = ""
|
||||
out_filename = './S1AP-PDU.asn'
|
||||
|
||||
FAIL = '\033[91m'
|
||||
WARN = '\033[93m'
|
||||
ENDC = '\033[0m'
|
||||
|
||||
def printFail(string):
|
||||
sys.stderr.write(FAIL + string + ENDC + "\n")
|
||||
|
||||
def printWarning(string):
|
||||
print WARN + string + ENDC
|
||||
|
||||
def printDebug(string):
|
||||
if verbosity > 0:
|
||||
print string
|
||||
|
||||
def outputHeaderToFile(f, in_filename):
|
||||
now = datetime.datetime.now()
|
||||
f.write("""/*
|
||||
* Copyright (c) 2017, CellWire Group
|
||||
* All rights reserved.
|
||||
*
|
||||
* Redistribution and use in source and binary forms, with or without
|
||||
* modification, are permitted provided that the following conditions are met:
|
||||
*
|
||||
* 1. Redistributions of source code must retain the above copyright notice, this
|
||||
* list of conditions and the following disclaimer.
|
||||
* 2. Redistributions in binary form must reproduce the above copyright notice,
|
||||
* this list of conditions and the following disclaimer in the documentation
|
||||
* and/or other materials provided with the distribution.
|
||||
*
|
||||
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
|
||||
* ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
|
||||
* WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
|
||||
* DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR
|
||||
* ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
|
||||
* (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
|
||||
* LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
|
||||
* ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
|
||||
* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
|
||||
* SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
||||
*/
|
||||
|
||||
""")
|
||||
f.write("/*******************************************************************************\n")
|
||||
f.write(" * This file had been created by asn1prefix.py script v%s\n" % (version))
|
||||
f.write(" * Please do not modify this file but regenerate it via script.\n")
|
||||
f.write(" * Created on: %s by %s\n * from %s\n" % (str(now), getpass.getuser(), in_filename))
|
||||
f.write(" ******************************************************************************/\n")
|
||||
|
||||
def usage():
|
||||
print "Python adding prefix for asn1 v%s" % (version)
|
||||
print "Usage: python asn1prefix.py [options]"
|
||||
print "Available options:"
|
||||
print "-d Enable script debug"
|
||||
print "-f [file] Input file to parse"
|
||||
print "-o [file] Output file for result"
|
||||
print "-h Print this help and return"
|
||||
|
||||
try:
|
||||
opts, args = getopt.getopt(sys.argv[1:], "i:ho:", ["input", "help", "output"])
|
||||
except getopt.GetoptError as err:
|
||||
# print help information and exit:
|
||||
usage()
|
||||
sys.exit(2)
|
||||
|
||||
for o, a in opts:
|
||||
if o in ("-i", "--input"):
|
||||
in_filename = a
|
||||
if o in ("-o", "--output"):
|
||||
out_filename = a
|
||||
if o in ("-h", "--help"):
|
||||
usage()
|
||||
sys.exit(2)
|
||||
|
||||
asnType = '(INTEGER|'
|
||||
asnType += 'BIT\sSTRING|'
|
||||
asnType += 'ENUMERATED|'
|
||||
asnType += 'CHOICE|'
|
||||
asnType += 'OCTET\sSTRING|'
|
||||
asnType += 'TBCD-STRING|'
|
||||
asnType += 'PrintableString|'
|
||||
asnType += 'SEQUENCE)'
|
||||
|
||||
structReplacements = {}
|
||||
constantReplacements = {}
|
||||
|
||||
in_file = open(in_filename, 'r')
|
||||
for line in in_file:
|
||||
# Removing any comment
|
||||
if line.find('--') >= 0:
|
||||
line = line[:line.find('--')]
|
||||
# Removing any carriage return
|
||||
lines += re.sub('\r', '', line)
|
||||
|
||||
# Extracting S1AP-PDU-Contents
|
||||
for i in re.findall(r'^([\w-]+)\s*S1AP-PROTOCOL-', lines, re.MULTILINE):
|
||||
structReplacements[i] = 'S1ap-' + i
|
||||
for i in re.findall(r'([\w-]+)\s*::=\s+E-RAB-IE-Container', lines, re.MULTILINE):
|
||||
structReplacements[i] = 'S1ap-' + i
|
||||
# Extracting S1AP-IEs
|
||||
for i in re.findall(r'([\w-]+)\s*::=\s*' + asnType, lines, re.MULTILINE):
|
||||
structReplacements[i[0]] = 'S1ap-' + i[0]
|
||||
|
||||
# Removeing 'S1AP-PDU'
|
||||
del structReplacements['S1AP-PDU']
|
||||
|
||||
# Extracting S1AP-Constant
|
||||
for i in re.findall(r'([\w-]+)\s*INTEGER\s*::=', lines, re.MULTILINE):
|
||||
constantReplacements[i] = 's1ap-' + i
|
||||
|
||||
in_file.seek(0)
|
||||
|
||||
structPattern = re.compile(r'\b(' + '|'.join(structReplacements.keys()) + r')\b')
|
||||
constantPattern = re.compile(r'\b(' + '|'.join(constantReplacements.keys()) + r')\b')
|
||||
|
||||
out_file = open(out_filename, 'w')
|
||||
|
||||
for line in in_file:
|
||||
line = structPattern.sub(lambda x: structReplacements[x.group()], line)
|
||||
line = constantPattern.sub(lambda x: constantReplacements[x.group()], line)
|
||||
line = line.replace('id-S1ap', 'id')
|
||||
out_file.write(line)
|
||||
|
||||
in_file.close()
|
||||
out_file.close()
|
Loading…
Reference in New Issue