76 lines
2.1 KiB
Python
Executable file
76 lines
2.1 KiB
Python
Executable file
#!/usr/bin/python
|
||
|
||
import sys
|
||
import re
|
||
|
||
header = '''// This file was extracted from the TCG Published
|
||
// Trusted Platform Module Library
|
||
// Part 3: Commands
|
||
// Family "2.0"
|
||
// Level 00 Revision 01.16
|
||
// October 30, 2014
|
||
|
||
'''
|
||
|
||
head_spaces = re.compile('^\s*[0-9]+\s{0,4}')
|
||
source_lines = open(sys.argv[1], 'r').read().splitlines()
|
||
|
||
def strip_line_num(line):
|
||
line = head_spaces.sub('', line)
|
||
return line
|
||
|
||
def postprocess_lines(buffer):
|
||
# get rid of heading line numbers and spaces.
|
||
buffer = [head_spaces.sub('', x) for x in buffer]
|
||
|
||
# Drop the file level conditional compilation statement.
|
||
for i in range(len(buffer)):
|
||
if buffer[i].startswith('#include'):
|
||
continue
|
||
if buffer[i].startswith(
|
||
'#ifdef TPM_CC') and buffer[-1].startswith(
|
||
'#endif // CC_'):
|
||
buffer = buffer[:i] + buffer[i + 1:-1]
|
||
break
|
||
return header + '\n'.join(buffer) + '\n'
|
||
|
||
text = []
|
||
for line in source_lines:
|
||
text.append(line)
|
||
if line == '' and text[-2].startswith(' ') and text[-5] == '':
|
||
text = text[:-5]
|
||
|
||
func_file = None
|
||
func_name = ''
|
||
prev_num = 0
|
||
line_buffer = []
|
||
output_buffer = []
|
||
for line in text:
|
||
f = re.match('^\s*[0-9]+\.[0-9]+\s+(\S+)$', line)
|
||
if f:
|
||
func_name = re.sub('^TPM2_', '', f.groups(0)[0])
|
||
|
||
num = re.match('^\s*([0-9]+)[$ ]', line + ' ')
|
||
if num:
|
||
line_num = int(num.groups(0)[0])
|
||
if line_num == 1:
|
||
# this is the first line of a file
|
||
if func_file:
|
||
func_file.write(postprocess_lines(output_buffer))
|
||
func_file.close()
|
||
func_file = open('%s.c' % func_name, 'w')
|
||
output_buffer = [line,]
|
||
prev_num = 1
|
||
line_buffer = []
|
||
continue
|
||
if line_num == prev_num + 1:
|
||
if line_buffer:
|
||
output_buffer.append('\n'.join(line_buffer))
|
||
line_buffer = []
|
||
output_buffer.append(line)
|
||
prev_num = line_num
|
||
continue
|
||
line_buffer.append('//' + line)
|
||
|
||
func_file.write(postprocess_lines(output_buffer))
|
||
func_file.close()
|