Python code coverage for PC/VS8.0/build_ssl.py

#	count	content
1	n/a	# Script for building the _ssl and _hashlib modules for Windows.
2	n/a	# Uses Perl to setup the OpenSSL environment correctly
3	n/a	# and build OpenSSL, then invokes a simple nmake session
4	n/a	# for the actual _ssl.pyd and _hashlib.pyd DLLs.
5	n/a
6	n/a	# THEORETICALLY, you can:
7	n/a	# * Unpack the latest SSL release one level above your main Python source
8	n/a	# directory. It is likely you will already find the zlib library and
9	n/a	# any other external packages there.
10	n/a	# * Install ActivePerl and ensure it is somewhere on your path.
11	n/a	# * Run this script from the PC/VS8.0 directory.
12	n/a	#
13	n/a	# it should configure and build SSL, then build the _ssl and _hashlib
14	n/a	# Python extensions without intervention.
15	n/a
16	n/a	# Modified by Christian Heimes
17	n/a	# Now this script supports pre-generated makefiles and assembly files.
18	n/a	# Developers don't need an installation of Perl anymore to build Python. A svn
19	n/a	# checkout from our svn repository is enough.
20	n/a	#
21	n/a	# In Order to create the files in the case of an update you still need Perl.
22	n/a	# Run build_ssl in this order:
23	n/a	# python.exe build_ssl.py Release x64
24	n/a	# python.exe build_ssl.py Release Win32
25	n/a
26	n/a	import os, sys, re, shutil
27	n/a
28	n/a	# Find all "foo.exe" files on the PATH.
29	n/a	def find_all_on_path(filename, extras = None):
30	n/a	entries = os.environ["PATH"].split(os.pathsep)
31	n/a	ret = []
32	n/a	for p in entries:
33	n/a	fname = os.path.abspath(os.path.join(p, filename))
34	n/a	if os.path.isfile(fname) and fname not in ret:
35	n/a	ret.append(fname)
36	n/a	if extras:
37	n/a	for p in extras:
38	n/a	fname = os.path.abspath(os.path.join(p, filename))
39	n/a	if os.path.isfile(fname) and fname not in ret:
40	n/a	ret.append(fname)
41	n/a	return ret
42	n/a
43	n/a	# Find a suitable Perl installation for OpenSSL.
44	n/a	# cygwin perl does not work. ActivePerl does.
45	n/a	# Being a Perl dummy, the simplest way I can check is if the "Win32" package
46	n/a	# is available.
47	n/a	def find_working_perl(perls):
48	n/a	for perl in perls:
49	n/a	fh = os.popen('"%s" -e "use Win32;"' % perl)
50	n/a	fh.read()
51	n/a	rc = fh.close()
52	n/a	if rc:
53	n/a	continue
54	n/a	return perl
55	n/a	print("Can not find a suitable PERL:")
56	n/a	if perls:
57	n/a	print(" the following perl interpreters were found:")
58	n/a	for p in perls:
59	n/a	print(" ", p)
60	n/a	print(" None of these versions appear suitable for building OpenSSL")
61	n/a	else:
62	n/a	print(" NO perl interpreters were found on this machine at all!")
63	n/a	print(" Please install ActivePerl and ensure it appears on your path")
64	n/a	return None
65	n/a
66	n/a	# Locate the best SSL directory given a few roots to look into.
67	n/a	def find_best_ssl_dir(sources):
68	n/a	candidates = []
69	n/a	for s in sources:
70	n/a	try:
71	n/a	# note: do not abspath s; the build will fail if any
72	n/a	# higher up directory name has spaces in it.
73	n/a	fnames = os.listdir(s)
74	n/a	except OSError:
75	n/a	fnames = []
76	n/a	for fname in fnames:
77	n/a	fqn = os.path.join(s, fname)
78	n/a	if os.path.isdir(fqn) and fname.startswith("openssl-"):
79	n/a	candidates.append(fqn)
80	n/a	# Now we have all the candidates, locate the best.
81	n/a	best_parts = []
82	n/a	best_name = None
83	n/a	for c in candidates:
84	n/a	parts = re.split("[.-]", os.path.basename(c))[1:]
85	n/a	# eg - openssl-0.9.7-beta1 - ignore all "beta" or any other qualifiers
86	n/a	if len(parts) >= 4:
87	n/a	continue
88	n/a	if parts > best_parts:
89	n/a	best_parts = parts
90	n/a	best_name = c
91	n/a	if best_name is not None:
92	n/a	print("Found an SSL directory at '%s'" % (best_name,))
93	n/a	else:
94	n/a	print("Could not find an SSL directory in '%s'" % (sources,))
95	n/a	sys.stdout.flush()
96	n/a	return best_name
97	n/a
98	n/a	def create_makefile64(makefile, m32):
99	n/a	"""Create and fix makefile for 64bit
100	n/a
101	n/a	Replace 32 with 64bit directories
102	n/a	"""
103	n/a	if not os.path.isfile(m32):
104	n/a	return
105	n/a	with open(m32) as fin:
106	n/a	with open(makefile, 'w') as fout:
107	n/a	for line in fin:
108	n/a	line = line.replace("=tmp32", "=tmp64")
109	n/a	line = line.replace("=out32", "=out64")
110	n/a	line = line.replace("=inc32", "=inc64")
111	n/a	# force 64 bit machine
112	n/a	line = line.replace("MKLIB=lib", "MKLIB=lib /MACHINE:X64")
113	n/a	line = line.replace("LFLAGS=", "LFLAGS=/MACHINE:X64 ")
114	n/a	# don't link against the lib on 64bit systems
115	n/a	line = line.replace("bufferoverflowu.lib", "")
116	n/a	fout.write(line)
117	n/a	os.unlink(m32)
118	n/a
119	n/a	def fix_makefile(makefile):
120	n/a	"""Fix some stuff in all makefiles
121	n/a	"""
122	n/a	if not os.path.isfile(makefile):
123	n/a	return
124	n/a	with open(makefile) as fin:
125	n/a	lines = fin.readlines()
126	n/a	with open(makefile, 'w') as fout:
127	n/a	for line in lines:
128	n/a	if line.startswith("PERL="):
129	n/a	continue
130	n/a	if line.startswith("CP="):
131	n/a	line = "CP=copy\n"
132	n/a	if line.startswith("MKDIR="):
133	n/a	line = "MKDIR=mkdir\n"
134	n/a	if line.startswith("CFLAG="):
135	n/a	line = line.strip()
136	n/a	for algo in ("RC5", "MDC2", "IDEA"):
137	n/a	noalgo = " -DOPENSSL_NO_%s" % algo
138	n/a	if noalgo not in line:
139	n/a	line = line + noalgo
140	n/a	line = line + '\n'
141	n/a	fout.write(line)
142	n/a
143	n/a	def run_configure(configure, do_script):
144	n/a	print("perl Configure "+configure+" no-idea no-mdc2")
145	n/a	os.system("perl Configure "+configure+" no-idea no-mdc2")
146	n/a	print(do_script)
147	n/a	os.system(do_script)
148	n/a
149	n/a	def cmp(f1, f2):
150	n/a	bufsize = 1024 * 8
151	n/a	with open(f1, 'rb') as fp1, open(f2, 'rb') as fp2:
152	n/a	while True:
153	n/a	b1 = fp1.read(bufsize)
154	n/a	b2 = fp2.read(bufsize)
155	n/a	if b1 != b2:
156	n/a	return False
157	n/a	if not b1:
158	n/a	return True
159	n/a
160	n/a	def copy(src, dst):
161	n/a	if os.path.isfile(dst) and cmp(src, dst):
162	n/a	return
163	n/a	shutil.copy(src, dst)
164	n/a
165	n/a	def main():
166	n/a	build_all = "-a" in sys.argv
167	n/a	if sys.argv[1] == "Release":
168	n/a	debug = False
169	n/a	elif sys.argv[1] == "Debug":
170	n/a	debug = True
171	n/a	else:
172	n/a	raise ValueError(str(sys.argv))
173	n/a
174	n/a	if sys.argv[2] == "Win32":
175	n/a	arch = "x86"
176	n/a	configure = "VC-WIN32"
177	n/a	do_script = "ms\\do_nasm"
178	n/a	makefile="ms\\nt.mak"
179	n/a	m32 = makefile
180	n/a	dirsuffix = "32"
181	n/a	elif sys.argv[2] == "x64":
182	n/a	arch="amd64"
183	n/a	configure = "VC-WIN64A"
184	n/a	do_script = "ms\\do_win64a"
185	n/a	makefile = "ms\\nt64.mak"
186	n/a	m32 = makefile.replace('64', '')
187	n/a	dirsuffix = "64"
188	n/a	#os.environ["VSEXTCOMP_USECL"] = "MS_OPTERON"
189	n/a	else:
190	n/a	raise ValueError(str(sys.argv))
191	n/a
192	n/a	make_flags = ""
193	n/a	if build_all:
194	n/a	make_flags = "-a"
195	n/a	# perl should be on the path, but we also look in "\perl" and "c:\\perl"
196	n/a	# as "well known" locations
197	n/a	perls = find_all_on_path("perl.exe", ["\\perl\\bin", "C:\\perl\\bin"])
198	n/a	perl = find_working_perl(perls)
199	n/a	if perl:
200	n/a	print("Found a working perl at '%s'" % (perl,))
201	n/a	else:
202	n/a	print("No Perl installation was found. Existing Makefiles are used.")
203	n/a	sys.stdout.flush()
204	n/a	# Look for SSL 3 levels up from PC/VS8.0 - ie, same place zlib etc all live.
205	n/a	ssl_dir = find_best_ssl_dir(("..\\..\\..",))
206	n/a	if ssl_dir is None:
207	n/a	sys.exit(1)
208	n/a
209	n/a	old_cd = os.getcwd()
210	n/a	try:
211	n/a	os.chdir(ssl_dir)
212	n/a	# rebuild makefile when we do the role over from 32 to 64 build
213	n/a	if arch == "amd64" and os.path.isfile(m32) and not os.path.isfile(makefile):
214	n/a	os.unlink(m32)
215	n/a
216	n/a	# If the ssl makefiles do not exist, we invoke Perl to generate them.
217	n/a	# Due to a bug in this script, the makefile sometimes ended up empty
218	n/a	# Force a regeneration if it is.
219	n/a	if not os.path.isfile(makefile) or os.path.getsize(makefile)==0:
220	n/a	if perl is None:
221	n/a	print("Perl is required to build the makefiles!")
222	n/a	sys.exit(1)
223	n/a
224	n/a	print("Creating the makefiles...")
225	n/a	sys.stdout.flush()
226	n/a	# Put our working Perl at the front of our path
227	n/a	os.environ["PATH"] = os.path.dirname(perl) + \
228	n/a	os.pathsep + \
229	n/a	os.environ["PATH"]
230	n/a	run_configure(configure, do_script)
231	n/a	if debug:
232	n/a	print("OpenSSL debug builds aren't supported.")
233	n/a	#if arch=="x86" and debug:
234	n/a	# # the do_masm script in openssl doesn't generate a debug
235	n/a	# # build makefile so we generate it here:
236	n/a	# os.system("perl util\mk1mf.pl debug "+configure+" >"+makefile)
237	n/a
238	n/a	if arch == "amd64":
239	n/a	create_makefile64(makefile, m32)
240	n/a	fix_makefile(makefile)
241	n/a	copy(r"crypto\buildinf.h", r"crypto\buildinf_%s.h" % arch)
242	n/a	copy(r"crypto\opensslconf.h", r"crypto\opensslconf_%s.h" % arch)
243	n/a
244	n/a	# If the assembler files don't exist in tmpXX, copy them there
245	n/a	if perl is None and os.path.exists("asm"+dirsuffix):
246	n/a	if not os.path.exists("tmp"+dirsuffix):
247	n/a	os.mkdir("tmp"+dirsuffix)
248	n/a	for f in os.listdir("asm"+dirsuffix):
249	n/a	if not f.endswith(".asm"): continue
250	n/a	if os.path.isfile(r"tmp%s\%s" % (dirsuffix, f)): continue
251	n/a	shutil.copy(r"asm%s\%s" % (dirsuffix, f), "tmp"+dirsuffix)
252	n/a
253	n/a	# Now run make.
254	n/a	if arch == "amd64":
255	n/a	rc = os.system("ml64 -c -Foms\\uptable.obj ms\\uptable.asm")
256	n/a	if rc:
257	n/a	print("ml64 assembler has failed.")
258	n/a	sys.exit(rc)
259	n/a
260	n/a	copy(r"crypto\buildinf_%s.h" % arch, r"crypto\buildinf.h")
261	n/a	copy(r"crypto\opensslconf_%s.h" % arch, r"crypto\opensslconf.h")
262	n/a
263	n/a	#makeCommand = "nmake /nologo PERL=\"%s\" -f \"%s\"" %(perl, makefile)
264	n/a	makeCommand = "nmake /nologo -f \"%s\"" % makefile
265	n/a	print("Executing ssl makefiles:", makeCommand)
266	n/a	sys.stdout.flush()
267	n/a	rc = os.system(makeCommand)
268	n/a	if rc:
269	n/a	print("Executing "+makefile+" failed")
270	n/a	print(rc)
271	n/a	sys.exit(rc)
272	n/a	finally:
273	n/a	os.chdir(old_cd)
274	n/a	sys.exit(rc)
275	n/a
276	n/a	if __name__=='__main__':
277	n/a	main()