Python code coverage for Modules/_ctypes/libffi/generate-darwin-source-and-headers.py

#	count	content
1	n/a	#!/usr/bin/env python
2	n/a	import subprocess
3	n/a	import os
4	n/a	import errno
5	n/a	import collections
6	n/a	import glob
7	n/a	import argparse
8	n/a
9	n/a	class Platform(object):
10	n/a	pass
11	n/a
12	n/a	class simulator_platform(Platform):
13	n/a	directory = 'darwin_ios'
14	n/a	sdk = 'iphonesimulator'
15	n/a	arch = 'i386'
16	n/a	triple = 'i386-apple-darwin11'
17	n/a	version_min = '-miphoneos-version-min=5.1.1'
18	n/a
19	n/a	prefix = "#ifdef __i386__\n\n"
20	n/a	suffix = "\n\n#endif"
21	n/a	src_dir = 'x86'
22	n/a	src_files = ['darwin.S', 'win32.S', 'ffi.c']
23	n/a
24	n/a
25	n/a	class simulator64_platform(Platform):
26	n/a	directory = 'darwin_ios'
27	n/a	sdk = 'iphonesimulator'
28	n/a	arch = 'x86_64'
29	n/a	triple = 'x86_64-apple-darwin13'
30	n/a	version_min = '-miphoneos-version-min=7.0'
31	n/a
32	n/a	prefix = "#ifdef __x86_64__\n\n"
33	n/a	suffix = "\n\n#endif"
34	n/a	src_dir = 'x86'
35	n/a	src_files = ['darwin64.S', 'ffi64.c']
36	n/a
37	n/a
38	n/a	class device_platform(Platform):
39	n/a	directory = 'darwin_ios'
40	n/a	sdk = 'iphoneos'
41	n/a	arch = 'armv7'
42	n/a	triple = 'arm-apple-darwin11'
43	n/a	version_min = '-miphoneos-version-min=5.1.1'
44	n/a
45	n/a	prefix = "#ifdef __arm__\n\n"
46	n/a	suffix = "\n\n#endif"
47	n/a	src_dir = 'arm'
48	n/a	src_files = ['sysv.S', 'trampoline.S', 'ffi.c']
49	n/a
50	n/a
51	n/a	class device64_platform(Platform):
52	n/a	directory = 'darwin_ios'
53	n/a	sdk = 'iphoneos'
54	n/a	arch = 'arm64'
55	n/a	triple = 'aarch64-apple-darwin13'
56	n/a	version_min = '-miphoneos-version-min=7.0'
57	n/a
58	n/a	prefix = "#ifdef __arm64__\n\n"
59	n/a	suffix = "\n\n#endif"
60	n/a	src_dir = 'aarch64'
61	n/a	src_files = ['sysv.S', 'ffi.c']
62	n/a
63	n/a
64	n/a	class desktop32_platform(Platform):
65	n/a	directory = 'darwin_osx'
66	n/a	sdk = 'macosx'
67	n/a	arch = 'i386'
68	n/a	triple = 'i386-apple-darwin10'
69	n/a	version_min = '-mmacosx-version-min=10.6'
70	n/a	src_dir = 'x86'
71	n/a	src_files = ['darwin.S', 'win32.S', 'ffi.c']
72	n/a
73	n/a	prefix = "#ifdef __i386__\n\n"
74	n/a	suffix = "\n\n#endif"
75	n/a
76	n/a
77	n/a	class desktop64_platform(Platform):
78	n/a	directory = 'darwin_osx'
79	n/a	sdk = 'macosx'
80	n/a	arch = 'x86_64'
81	n/a	triple = 'x86_64-apple-darwin10'
82	n/a	version_min = '-mmacosx-version-min=10.6'
83	n/a
84	n/a	prefix = "#ifdef __x86_64__\n\n"
85	n/a	suffix = "\n\n#endif"
86	n/a	src_dir = 'x86'
87	n/a	src_files = ['darwin64.S', 'ffi64.c']
88	n/a
89	n/a
90	n/a	def mkdir_p(path):
91	n/a	try:
92	n/a	os.makedirs(path)
93	n/a	except OSError as exc: # Python >2.5
94	n/a	if exc.errno == errno.EEXIST:
95	n/a	pass
96	n/a	else:
97	n/a	raise
98	n/a
99	n/a
100	n/a	def move_file(src_dir, dst_dir, filename, file_suffix=None, prefix='', suffix=''):
101	n/a	mkdir_p(dst_dir)
102	n/a	out_filename = filename
103	n/a
104	n/a	if file_suffix:
105	n/a	split_name = os.path.splitext(filename)
106	n/a	out_filename = "%s_%s%s" % (split_name[0], file_suffix, split_name[1])
107	n/a
108	n/a	with open(os.path.join(src_dir, filename)) as in_file:
109	n/a	with open(os.path.join(dst_dir, out_filename), 'w') as out_file:
110	n/a	if prefix:
111	n/a	out_file.write(prefix)
112	n/a
113	n/a	out_file.write(in_file.read())
114	n/a
115	n/a	if suffix:
116	n/a	out_file.write(suffix)
117	n/a
118	n/a
119	n/a	def list_files(src_dir, pattern=None, filelist=None):
120	n/a	if pattern: filelist = glob.iglob(os.path.join(src_dir, pattern))
121	n/a	for file in filelist:
122	n/a	yield os.path.basename(file)
123	n/a
124	n/a
125	n/a	def copy_files(src_dir, dst_dir, pattern=None, filelist=None, file_suffix=None, prefix=None, suffix=None):
126	n/a	for filename in list_files(src_dir, pattern=pattern, filelist=filelist):
127	n/a	move_file(src_dir, dst_dir, filename, file_suffix=file_suffix, prefix=prefix, suffix=suffix)
128	n/a
129	n/a
130	n/a	def copy_src_platform_files(platform):
131	n/a	src_dir = os.path.join('src', platform.src_dir)
132	n/a	dst_dir = os.path.join(platform.directory, 'src', platform.src_dir)
133	n/a	copy_files(src_dir, dst_dir, filelist=platform.src_files, file_suffix=platform.arch, prefix=platform.prefix, suffix=platform.suffix)
134	n/a
135	n/a
136	n/a	def build_target(platform, platform_headers):
137	n/a	def xcrun_cmd(cmd):
138	n/a	return 'xcrun -sdk %s %s -arch %s' % (platform.sdk, cmd, platform.arch)
139	n/a
140	n/a	tag='%s-%s' % (platform.sdk, platform.arch)
141	n/a	build_dir = 'build_%s' % tag
142	n/a	mkdir_p(build_dir)
143	n/a	env = dict(CC=xcrun_cmd('clang'),
144	n/a	LD=xcrun_cmd('ld'),
145	n/a	CFLAGS='%s' % (platform.version_min))
146	n/a	working_dir = os.getcwd()
147	n/a	try:
148	n/a	os.chdir(build_dir)
149	n/a	subprocess.check_call(['../configure', '-host', platform.triple], env=env)
150	n/a	finally:
151	n/a	os.chdir(working_dir)
152	n/a
153	n/a	for src_dir in [build_dir, os.path.join(build_dir, 'include')]:
154	n/a	copy_files(src_dir,
155	n/a	os.path.join(platform.directory, 'include'),
156	n/a	pattern='*.h',
157	n/a	file_suffix=platform.arch,
158	n/a	prefix=platform.prefix,
159	n/a	suffix=platform.suffix)
160	n/a
161	n/a	for filename in list_files(src_dir, pattern='*.h'):
162	n/a	platform_headers[filename].add((platform.prefix, platform.arch, platform.suffix))
163	n/a
164	n/a
165	n/a	def make_tramp():
166	n/a	with open('src/arm/trampoline.S', 'w') as tramp_out:
167	n/a	p = subprocess.Popen(['bash', 'src/arm/gentramp.sh'], stdout=tramp_out)
168	n/a	p.wait()
169	n/a
170	n/a
171	n/a	def generate_source_and_headers(generate_osx=True, generate_ios=True):
172	n/a	copy_files('src', 'darwin_common/src', pattern='*.c')
173	n/a	copy_files('include', 'darwin_common/include', pattern='*.h')
174	n/a
175	n/a	if generate_ios:
176	n/a	make_tramp()
177	n/a	copy_src_platform_files(simulator_platform)
178	n/a	copy_src_platform_files(simulator64_platform)
179	n/a	copy_src_platform_files(device_platform)
180	n/a	copy_src_platform_files(device64_platform)
181	n/a	if generate_osx:
182	n/a	copy_src_platform_files(desktop32_platform)
183	n/a	copy_src_platform_files(desktop64_platform)
184	n/a
185	n/a	platform_headers = collections.defaultdict(set)
186	n/a
187	n/a	if generate_ios:
188	n/a	build_target(simulator_platform, platform_headers)
189	n/a	build_target(simulator64_platform, platform_headers)
190	n/a	build_target(device_platform, platform_headers)
191	n/a	build_target(device64_platform, platform_headers)
192	n/a	if generate_osx:
193	n/a	build_target(desktop32_platform, platform_headers)
194	n/a	build_target(desktop64_platform, platform_headers)
195	n/a
196	n/a	mkdir_p('darwin_common/include')
197	n/a	for header_name, tag_tuples in platform_headers.iteritems():
198	n/a	basename, suffix = os.path.splitext(header_name)
199	n/a	with open(os.path.join('darwin_common/include', header_name), 'w') as header:
200	n/a	for tag_tuple in tag_tuples:
201	n/a	header.write('%s#include <%s_%s%s>\n%s\n' % (tag_tuple[0], basename, tag_tuple[1], suffix, tag_tuple[2]))
202	n/a
203	n/a	if __name__ == '__main__':
204	n/a	parser = argparse.ArgumentParser()
205	n/a	parser.add_argument('--only-ios', action='store_true', default=False)
206	n/a	parser.add_argument('--only-osx', action='store_true', default=False)
207	n/a	args = parser.parse_args()
208	n/a
209	n/a	generate_source_and_headers(generate_osx=not args.only_ios, generate_ios=not args.only_osx)