summaryrefslogtreecommitdiff
path: root/pjsip-apps/src/swig/importsym.py
blob: 32faf2cce4a7098b61bc0ab6a1639d3aa174a7b0 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
# $Id$
#
# importsym.py: Import C symbol decls (structs, enums, etc) and write them
#               to another file 
#
# Copyright (C)2013 Teluu Inc. (http://www.teluu.com)
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA 
#
import pycparser
from pycparser import c_generator
import sys
import os

def which(program):
	import os
	def is_exe(fpath):
		return os.path.isfile(fpath) and os.access(fpath, os.X_OK)
	
	if sys.platform == 'win32' and not program.endswith(".exe"):
		program += ".exe"

	fpath, fname = os.path.split(program)
	if fpath:
		if is_exe(program):
			return program
	else:
		for path in os.environ["PATH"].split(os.pathsep):
			path = path.strip('"')
			exe_file = os.path.join(path, program)
			if is_exe(exe_file):
				return exe_file
	return None

#
PJ_ROOT_PATH = "../../../"
   
# CPP is needed by pycparser.
CPP_PATH = which("cpp")
if not CPP_PATH:
	print 'Error: need to have cpp in PATH'
	sys.exit(1)

# Hardcoded!
if sys.platform == 'win32':
	PYCPARSER_DIR="C:/devs/tools/pycparser"
elif sys.platform == "linux2":
	PYCPARSER_DIR="/home/bennylp/Desktop/opt/src/pycparser-master"
else:
	PYCPARSER_DIR="/Library/Python/2.7/site-packages/pycparser"

if not os.path.exists(PYCPARSER_DIR + '/utils/fake_libc_include'):
	print "Error: couldn't find pycparser utils in '%s'" % PYPARSER_DIR 
	sys.exit(1)

# Heading, to be placed before the source files
C_HEADING_SECTION = """
#define PJ_AUTOCONF		1
#define jmp_buf			int
#define __attribute__(x)
"""

# CPP (C preprocessor) settings
CPP_CFLAGS   = [
	'-I' + PYCPARSER_DIR + '/utils/fake_libc_include',
	"-I" + PJ_ROOT_PATH + "pjlib/include",
	"-I" + PJ_ROOT_PATH + "pjlib-util/include",
	"-I" + PJ_ROOT_PATH + "pjnath/include",
	"-I" + PJ_ROOT_PATH + "pjmedia/include",
	"-I" + PJ_ROOT_PATH + "pjsip/include"
	]


class SymbolVisitor(pycparser.c_ast.NodeVisitor):
	def __init__(self, names):
		self.nodeDict = {}
		for name in names:
			self.nodeDict[name] = None
		
	def _add(self, node):
		if self.nodeDict.has_key(node.name):
			self.nodeDict[node.name] = node
		
	def visit_Struct(self, node):
		self._add(node)
		
	def visit_Enum(self, node):
		self._add(node)

	def visit_Typename(self, node):
		self._add(node)
		
	def visit_Typedef(self, node):
		self._add(node)

		
TEMP_FILE="tmpsrc.h"
		
class SymbolImporter:
	"""
	Import C selected declarations from C source file and move it
	to another file.
	
	Parameters:
	 - listfile	Path of file containing list of C source file
	                and identifier names to be imported. The format
	                of the listfile is:
	                
	                filename        name1  name2  name3
	                
	                for example:
	                
	                pj/sock_qos.h	pj_qos_type  pj_qos_flag
	                pj/types.h	pj_status_t  PJ_SUCCESS
	"""
	def __init__(self):
		pass
	
	def process(self, listfile, outfile):
		
		# Read listfile
		f = open(listfile)
		lines = f.readlines()
		f.close()
		
		# Process each line in list file, while generating the
		# temporary C file to be processed by pycparser
		f = open(TEMP_FILE, "w")
		f.write(C_HEADING_SECTION)
		names = []
		fcnt = 0
		for line in lines:
			spec = line.split()
			if len(spec) < 2:
				continue
			fcnt += 1
			f.write("#include <%s>\n" % spec[0])
			names.extend(spec[1:])
		f.close()
		print 'Parsing %d symbols from %d files..' % (len(names), fcnt)
		
		# Parse the temporary C file
		ast = pycparser.parse_file(TEMP_FILE, use_cpp=True, cpp_path=CPP_PATH, cpp_args=CPP_CFLAGS)
		os.remove(TEMP_FILE)
		
		# Filter the declarations that we wanted
		print 'Filtering..'
		visitor = SymbolVisitor(names)
		visitor.visit(ast)
	
		# Print symbol declarations to outfile
		print 'Writing declarations..'
		f = open(outfile, 'w')
		f.write("// This file is autogenerated by importsym script, do not modify!\n\n")
		gen = pycparser.c_generator.CGenerator()
		for name in names:
			node = visitor.nodeDict[name]
			if not node:
				print "  ** Warning: declaration for '%s' is not found **" % k
			else:
				print "  writing '%s'.." % name
				output = gen.visit(node) + ";\n\n"
				f.write(output)
		f.close()
		print "Done."


if __name__ == "__main__":
	print "Importing symbols: 'symbols.lst' --> 'symbols.i'"
	si = SymbolImporter()
	si.process("symbols.lst", "symbols.i")
	try:
		os.remove("lextab.py")
	except OSError:
		pass
	try:
		os.remove("yacctab.py")
	except OSError:
		pass