Python code coverage for Lib/packaging/compiler/msvc9compiler.py

#	count	content
1	n/a	"""CCompiler implementation for the Microsoft Visual Studio 2008 compiler.
2	n/a
3	n/a	The MSVCCompiler class is compatible with VS 2005 and VS 2008. Legacy
4	n/a	support for older versions of VS are in the msvccompiler module.
5	n/a	"""
6	n/a
7	n/a	# Written by Perry Stoll
8	n/a	# hacked by Robin Becker and Thomas Heller to do a better job of
9	n/a	# finding DevStudio (through the registry)
10	n/a	# ported to VS2005 and VS 2008 by Christian Heimes
11	n/a	import os
12	n/a	import subprocess
13	n/a	import sys
14	n/a	import re
15	n/a
16	n/a	from packaging.errors import (PackagingExecError, PackagingPlatformError,
17	n/a	CompileError, LibError, LinkError)
18	n/a	from packaging.compiler.ccompiler import CCompiler
19	n/a	from packaging.compiler import gen_lib_options
20	n/a	from packaging import logger
21	n/a	from packaging.util import get_platform
22	n/a
23	n/a	import winreg
24	n/a
25	n/a	RegOpenKeyEx = winreg.OpenKeyEx
26	n/a	RegEnumKey = winreg.EnumKey
27	n/a	RegEnumValue = winreg.EnumValue
28	n/a	RegError = winreg.error
29	n/a
30	n/a	HKEYS = (winreg.HKEY_USERS,
31	n/a	winreg.HKEY_CURRENT_USER,
32	n/a	winreg.HKEY_LOCAL_MACHINE,
33	n/a	winreg.HKEY_CLASSES_ROOT)
34	n/a
35	n/a	VS_BASE = r"Software\Microsoft\VisualStudio\%0.1f"
36	n/a	WINSDK_BASE = r"Software\Microsoft\Microsoft SDKs\Windows"
37	n/a	NET_BASE = r"Software\Microsoft\.NETFramework"
38	n/a
39	n/a	# A map keyed by get_platform() return values to values accepted by
40	n/a	# 'vcvarsall.bat'. Note a cross-compile may combine these (eg, 'x86_amd64' is
41	n/a	# the param to cross-compile on x86 targetting amd64.)
42	n/a	PLAT_TO_VCVARS = {
43	n/a	'win32' : 'x86',
44	n/a	'win-amd64' : 'amd64',
45	n/a	'win-ia64' : 'ia64',
46	n/a	}
47	n/a
48	n/a
49	n/a	class Reg:
50	n/a	"""Helper class to read values from the registry
51	n/a	"""
52	n/a
53	n/a	def get_value(cls, path, key):
54	n/a	for base in HKEYS:
55	n/a	d = cls.read_values(base, path)
56	n/a	if d and key in d:
57	n/a	return d[key]
58	n/a	raise KeyError(key)
59	n/a	get_value = classmethod(get_value)
60	n/a
61	n/a	def read_keys(cls, base, key):
62	n/a	"""Return list of registry keys."""
63	n/a	try:
64	n/a	handle = RegOpenKeyEx(base, key)
65	n/a	except RegError:
66	n/a	return None
67	n/a	L = []
68	n/a	i = 0
69	n/a	while True:
70	n/a	try:
71	n/a	k = RegEnumKey(handle, i)
72	n/a	except RegError:
73	n/a	break
74	n/a	L.append(k)
75	n/a	i += 1
76	n/a	return L
77	n/a	read_keys = classmethod(read_keys)
78	n/a
79	n/a	def read_values(cls, base, key):
80	n/a	"""Return dict of registry keys and values.
81	n/a
82	n/a	All names are converted to lowercase.
83	n/a	"""
84	n/a	try:
85	n/a	handle = RegOpenKeyEx(base, key)
86	n/a	except RegError:
87	n/a	return None
88	n/a	d = {}
89	n/a	i = 0
90	n/a	while True:
91	n/a	try:
92	n/a	name, value, type = RegEnumValue(handle, i)
93	n/a	except RegError:
94	n/a	break
95	n/a	name = name.lower()
96	n/a	d[cls.convert_mbcs(name)] = cls.convert_mbcs(value)
97	n/a	i += 1
98	n/a	return d
99	n/a	read_values = classmethod(read_values)
100	n/a
101	n/a	def convert_mbcs(s):
102	n/a	dec = getattr(s, "decode", None)
103	n/a	if dec is not None:
104	n/a	try:
105	n/a	s = dec("mbcs")
106	n/a	except UnicodeError:
107	n/a	pass
108	n/a	return s
109	n/a	convert_mbcs = staticmethod(convert_mbcs)
110	n/a
111	n/a	class MacroExpander:
112	n/a
113	n/a	def __init__(self, version):
114	n/a	self.macros = {}
115	n/a	self.vsbase = VS_BASE % version
116	n/a	self.load_macros(version)
117	n/a
118	n/a	def set_macro(self, macro, path, key):
119	n/a	self.macros["$(%s)" % macro] = Reg.get_value(path, key)
120	n/a
121	n/a	def load_macros(self, version):
122	n/a	self.set_macro("VCInstallDir", self.vsbase + r"\Setup\VC", "productdir")
123	n/a	self.set_macro("VSInstallDir", self.vsbase + r"\Setup\VS", "productdir")
124	n/a	self.set_macro("FrameworkDir", NET_BASE, "installroot")
125	n/a	try:
126	n/a	if version >= 8.0:
127	n/a	self.set_macro("FrameworkSDKDir", NET_BASE,
128	n/a	"sdkinstallrootv2.0")
129	n/a	else:
130	n/a	raise KeyError("sdkinstallrootv2.0")
131	n/a	except KeyError:
132	n/a	raise PackagingPlatformError(
133	n/a	"""Python was built with Visual Studio 2008; extensions must be built with a
134	n/a	compiler than can generate compatible binaries. Visual Studio 2008 was not
135	n/a	found on this system. If you have Cygwin installed, you can try compiling
136	n/a	with MingW32, by passing "-c mingw32" to pysetup.""")
137	n/a
138	n/a	if version >= 9.0:
139	n/a	self.set_macro("FrameworkVersion", self.vsbase, "clr version")
140	n/a	self.set_macro("WindowsSdkDir", WINSDK_BASE, "currentinstallfolder")
141	n/a	else:
142	n/a	p = r"Software\Microsoft\NET Framework Setup\Product"
143	n/a	for base in HKEYS:
144	n/a	try:
145	n/a	h = RegOpenKeyEx(base, p)
146	n/a	except RegError:
147	n/a	continue
148	n/a	key = RegEnumKey(h, 0)
149	n/a	d = Reg.get_value(base, r"%s\%s" % (p, key))
150	n/a	self.macros["$(FrameworkVersion)"] = d["version"]
151	n/a
152	n/a	def sub(self, s):
153	n/a	for k, v in self.macros.items():
154	n/a	s = s.replace(k, v)
155	n/a	return s
156	n/a
157	n/a	def get_build_version():
158	n/a	"""Return the version of MSVC that was used to build Python.
159	n/a
160	n/a	For Python 2.3 and up, the version number is included in
161	n/a	sys.version. For earlier versions, assume the compiler is MSVC 6.
162	n/a	"""
163	n/a	prefix = "MSC v."
164	n/a	i = sys.version.find(prefix)
165	n/a	if i == -1:
166	n/a	return 6
167	n/a	i = i + len(prefix)
168	n/a	s, rest = sys.version[i:].split(" ", 1)
169	n/a	majorVersion = int(s[:-2]) - 6
170	n/a	minorVersion = int(s[2:3]) / 10.0
171	n/a	# I don't think paths are affected by minor version in version 6
172	n/a	if majorVersion == 6:
173	n/a	minorVersion = 0
174	n/a	if majorVersion >= 6:
175	n/a	return majorVersion + minorVersion
176	n/a	# else we don't know what version of the compiler this is
177	n/a	return None
178	n/a
179	n/a	def normalize_and_reduce_paths(paths):
180	n/a	"""Return a list of normalized paths with duplicates removed.
181	n/a
182	n/a	The current order of paths is maintained.
183	n/a	"""
184	n/a	# Paths are normalized so things like: /a and /a/ aren't both preserved.
185	n/a	reduced_paths = []
186	n/a	for p in paths:
187	n/a	np = os.path.normpath(p)
188	n/a	# XXX(nnorwitz): O(n**2), if reduced_paths gets long perhaps use a set.
189	n/a	if np not in reduced_paths:
190	n/a	reduced_paths.append(np)
191	n/a	return reduced_paths
192	n/a
193	n/a	def removeDuplicates(variable):
194	n/a	"""Remove duplicate values of an environment variable.
195	n/a	"""
196	n/a	oldList = variable.split(os.pathsep)
197	n/a	newList = []
198	n/a	for i in oldList:
199	n/a	if i not in newList:
200	n/a	newList.append(i)
201	n/a	newVariable = os.pathsep.join(newList)
202	n/a	return newVariable
203	n/a
204	n/a	def find_vcvarsall(version):
205	n/a	"""Find the vcvarsall.bat file
206	n/a
207	n/a	At first it tries to find the productdir of VS 2008 in the registry. If
208	n/a	that fails it falls back to the VS90COMNTOOLS env var.
209	n/a	"""
210	n/a	vsbase = VS_BASE % version
211	n/a	try:
212	n/a	productdir = Reg.get_value(r"%s\Setup\VC" % vsbase,
213	n/a	"productdir")
214	n/a	except KeyError:
215	n/a	logger.debug("Unable to find productdir in registry")
216	n/a	productdir = None
217	n/a
218	n/a	if not productdir or not os.path.isdir(productdir):
219	n/a	toolskey = "VS%0.f0COMNTOOLS" % version
220	n/a	toolsdir = os.environ.get(toolskey, None)
221	n/a
222	n/a	if toolsdir and os.path.isdir(toolsdir):
223	n/a	productdir = os.path.join(toolsdir, os.pardir, os.pardir, "VC")
224	n/a	productdir = os.path.abspath(productdir)
225	n/a	if not os.path.isdir(productdir):
226	n/a	logger.debug("%s is not a valid directory", productdir)
227	n/a	return None
228	n/a	else:
229	n/a	logger.debug("env var %s is not set or invalid", toolskey)
230	n/a	if not productdir:
231	n/a	logger.debug("no productdir found")
232	n/a	return None
233	n/a	vcvarsall = os.path.join(productdir, "vcvarsall.bat")
234	n/a	if os.path.isfile(vcvarsall):
235	n/a	return vcvarsall
236	n/a	logger.debug("unable to find vcvarsall.bat")
237	n/a	return None
238	n/a
239	n/a	def query_vcvarsall(version, arch="x86"):
240	n/a	"""Launch vcvarsall.bat and read the settings from its environment
241	n/a	"""
242	n/a	vcvarsall = find_vcvarsall(version)
243	n/a	interesting = set(("include", "lib", "libpath", "path"))
244	n/a	result = {}
245	n/a
246	n/a	if vcvarsall is None:
247	n/a	raise PackagingPlatformError("Unable to find vcvarsall.bat")
248	n/a	logger.debug("calling 'vcvarsall.bat %s' (version=%s)", arch, version)
249	n/a	popen = subprocess.Popen('"%s" %s & set' % (vcvarsall, arch),
250	n/a	stdout=subprocess.PIPE,
251	n/a	stderr=subprocess.PIPE)
252	n/a
253	n/a	stdout, stderr = popen.communicate()
254	n/a	if popen.wait() != 0:
255	n/a	raise PackagingPlatformError(stderr.decode("mbcs"))
256	n/a
257	n/a	stdout = stdout.decode("mbcs")
258	n/a	for line in stdout.split("\n"):
259	n/a	line = Reg.convert_mbcs(line)
260	n/a	if '=' not in line:
261	n/a	continue
262	n/a	line = line.strip()
263	n/a	key, value = line.split('=', 1)
264	n/a	key = key.lower()
265	n/a	if key in interesting:
266	n/a	if value.endswith(os.pathsep):
267	n/a	value = value[:-1]
268	n/a	result[key] = removeDuplicates(value)
269	n/a
270	n/a	if len(result) != len(interesting):
271	n/a	raise ValueError(str(list(result)))
272	n/a
273	n/a	return result
274	n/a
275	n/a	# More globals
276	n/a	VERSION = get_build_version()
277	n/a	if VERSION < 8.0:
278	n/a	raise PackagingPlatformError("VC %0.1f is not supported by this module" % VERSION)
279	n/a	# MACROS = MacroExpander(VERSION)
280	n/a
281	n/a	class MSVCCompiler(CCompiler) :
282	n/a	"""Concrete class that implements an interface to Microsoft Visual C++,
283	n/a	as defined by the CCompiler abstract class."""
284	n/a
285	n/a	name = 'msvc'
286	n/a	description = 'Microsoft Visual C++'
287	n/a
288	n/a	# Just set this so CCompiler's constructor doesn't barf. We currently
289	n/a	# don't use the 'set_executables()' bureaucracy provided by CCompiler,
290	n/a	# as it really isn't necessary for this sort of single-compiler class.
291	n/a	# Would be nice to have a consistent interface with UnixCCompiler,
292	n/a	# though, so it's worth thinking about.
293	n/a	executables = {}
294	n/a
295	n/a	# Private class data (need to distinguish C from C++ source for compiler)
296	n/a	_c_extensions = ['.c']
297	n/a	_cpp_extensions = ['.cc', '.cpp', '.cxx']
298	n/a	_rc_extensions = ['.rc']
299	n/a	_mc_extensions = ['.mc']
300	n/a
301	n/a	# Needed for the filename generation methods provided by the
302	n/a	# base class, CCompiler.
303	n/a	src_extensions = (_c_extensions + _cpp_extensions +
304	n/a	_rc_extensions + _mc_extensions)
305	n/a	res_extension = '.res'
306	n/a	obj_extension = '.obj'
307	n/a	static_lib_extension = '.lib'
308	n/a	shared_lib_extension = '.dll'
309	n/a	static_lib_format = shared_lib_format = '%s%s'
310	n/a	exe_extension = '.exe'
311	n/a
312	n/a	def __init__(self, dry_run=False, force=False):
313	n/a	super(MSVCCompiler, self).__init__(dry_run, force)
314	n/a	self.__version = VERSION
315	n/a	self.__root = r"Software\Microsoft\VisualStudio"
316	n/a	# self.__macros = MACROS
317	n/a	self.__paths = []
318	n/a	# target platform (.plat_name is consistent with 'bdist')
319	n/a	self.plat_name = None
320	n/a	self.__arch = None # deprecated name
321	n/a	self.initialized = False
322	n/a
323	n/a	def initialize(self, plat_name=None):
324	n/a	# multi-init means we would need to check platform same each time...
325	n/a	assert not self.initialized, "don't init multiple times"
326	n/a	if plat_name is None:
327	n/a	plat_name = get_platform()
328	n/a	# sanity check for platforms to prevent obscure errors later.
329	n/a	ok_plats = 'win32', 'win-amd64', 'win-ia64'
330	n/a	if plat_name not in ok_plats:
331	n/a	raise PackagingPlatformError("--plat-name must be one of %s" %
332	n/a	(ok_plats,))
333	n/a
334	n/a	if "DISTUTILS_USE_SDK" in os.environ and "MSSdk" in os.environ and self.find_exe("cl.exe"):
335	n/a	# Assume that the SDK set up everything alright; don't try to be
336	n/a	# smarter
337	n/a	self.cc = "cl.exe"
338	n/a	self.linker = "link.exe"
339	n/a	self.lib = "lib.exe"
340	n/a	self.rc = "rc.exe"
341	n/a	self.mc = "mc.exe"
342	n/a	else:
343	n/a	# On x86, 'vcvars32.bat amd64' creates an env that doesn't work;
344	n/a	# to cross compile, you use 'x86_amd64'.
345	n/a	# On AMD64, 'vcvars32.bat amd64' is a native build env; to cross
346	n/a	# compile use 'x86' (ie, it runs the x86 compiler directly)
347	n/a	# No idea how itanium handles this, if at all.
348	n/a	if plat_name == get_platform() or plat_name == 'win32':
349	n/a	# native build or cross-compile to win32
350	n/a	plat_spec = PLAT_TO_VCVARS[plat_name]
351	n/a	else:
352	n/a	# cross compile from win32 -> some 64bit
353	n/a	plat_spec = PLAT_TO_VCVARS[get_platform()] + '_' + \
354	n/a	PLAT_TO_VCVARS[plat_name]
355	n/a
356	n/a	vc_env = query_vcvarsall(VERSION, plat_spec)
357	n/a
358	n/a	# take care to only use strings in the environment.
359	n/a	self.__paths = vc_env['path'].split(os.pathsep)
360	n/a	os.environ['lib'] = vc_env['lib']
361	n/a	os.environ['include'] = vc_env['include']
362	n/a
363	n/a	if len(self.__paths) == 0:
364	n/a	raise PackagingPlatformError("Python was built with %s, "
365	n/a	"and extensions need to be built with the same "
366	n/a	"version of the compiler, but it isn't installed."
367	n/a	% self.__product)
368	n/a
369	n/a	self.cc = self.find_exe("cl.exe")
370	n/a	self.linker = self.find_exe("link.exe")
371	n/a	self.lib = self.find_exe("lib.exe")
372	n/a	self.rc = self.find_exe("rc.exe") # resource compiler
373	n/a	self.mc = self.find_exe("mc.exe") # message compiler
374	n/a	#self.set_path_env_var('lib')
375	n/a	#self.set_path_env_var('include')
376	n/a
377	n/a	# extend the MSVC path with the current path
378	n/a	try:
379	n/a	for p in os.environ['path'].split(';'):
380	n/a	self.__paths.append(p)
381	n/a	except KeyError:
382	n/a	pass
383	n/a	self.__paths = normalize_and_reduce_paths(self.__paths)
384	n/a	os.environ['path'] = ";".join(self.__paths)
385	n/a
386	n/a	self.preprocess_options = None
387	n/a	if self.__arch == "x86":
388	n/a	self.compile_options = [ '/nologo', '/Ox', '/MD', '/W3',
389	n/a	'/DNDEBUG']
390	n/a	self.compile_options_debug = ['/nologo', '/Od', '/MDd', '/W3',
391	n/a	'/Z7', '/D_DEBUG']
392	n/a	else:
393	n/a	# Win64
394	n/a	self.compile_options = [ '/nologo', '/Ox', '/MD', '/W3', '/GS-' ,
395	n/a	'/DNDEBUG']
396	n/a	self.compile_options_debug = ['/nologo', '/Od', '/MDd', '/W3', '/GS-',
397	n/a	'/Z7', '/D_DEBUG']
398	n/a
399	n/a	self.ldflags_shared = ['/DLL', '/nologo', '/INCREMENTAL:NO']
400	n/a	if self.__version >= 7:
401	n/a	self.ldflags_shared_debug = [
402	n/a	'/DLL', '/nologo', '/INCREMENTAL:no', '/DEBUG', '/pdb:None'
403	n/a	]
404	n/a	self.ldflags_static = [ '/nologo']
405	n/a
406	n/a	self.initialized = True
407	n/a
408	n/a	# -- Worker methods ------------------------------------------------
409	n/a
410	n/a	def object_filenames(self,
411	n/a	source_filenames,
412	n/a	strip_dir=False,
413	n/a	output_dir=''):
414	n/a	# Copied from ccompiler.py, extended to return .res as 'object'-file
415	n/a	# for .rc input file
416	n/a	if output_dir is None: output_dir = ''
417	n/a	obj_names = []
418	n/a	for src_name in source_filenames:
419	n/a	base, ext = os.path.splitext(src_name)
420	n/a	base = os.path.splitdrive(base)[1] # Chop off the drive
421	n/a	base = base[os.path.isabs(base):] # If abs, chop off leading /
422	n/a	if ext not in self.src_extensions:
423	n/a	# Better to raise an exception instead of silently continuing
424	n/a	# and later complain about sources and targets having
425	n/a	# different lengths
426	n/a	raise CompileError("Don't know how to compile %s" % src_name)
427	n/a	if strip_dir:
428	n/a	base = os.path.basename(base)
429	n/a	if ext in self._rc_extensions:
430	n/a	obj_names.append(os.path.join(output_dir,
431	n/a	base + self.res_extension))
432	n/a	elif ext in self._mc_extensions:
433	n/a	obj_names.append(os.path.join(output_dir,
434	n/a	base + self.res_extension))
435	n/a	else:
436	n/a	obj_names.append(os.path.join(output_dir,
437	n/a	base + self.obj_extension))
438	n/a	return obj_names
439	n/a
440	n/a
441	n/a	def compile(self, sources,
442	n/a	output_dir=None, macros=None, include_dirs=None, debug=False,
443	n/a	extra_preargs=None, extra_postargs=None, depends=None):
444	n/a
445	n/a	if not self.initialized:
446	n/a	self.initialize()
447	n/a	compile_info = self._setup_compile(output_dir, macros, include_dirs,
448	n/a	sources, depends, extra_postargs)
449	n/a	macros, objects, extra_postargs, pp_opts, build = compile_info
450	n/a
451	n/a	compile_opts = extra_preargs or []
452	n/a	compile_opts.append('/c')
453	n/a	if debug:
454	n/a	compile_opts.extend(self.compile_options_debug)
455	n/a	else:
456	n/a	compile_opts.extend(self.compile_options)
457	n/a
458	n/a	for obj in objects:
459	n/a	try:
460	n/a	src, ext = build[obj]
461	n/a	except KeyError:
462	n/a	continue
463	n/a	if debug:
464	n/a	# pass the full pathname to MSVC in debug mode,
465	n/a	# this allows the debugger to find the source file
466	n/a	# without asking the user to browse for it
467	n/a	src = os.path.abspath(src)
468	n/a
469	n/a	if ext in self._c_extensions:
470	n/a	input_opt = "/Tc" + src
471	n/a	elif ext in self._cpp_extensions:
472	n/a	input_opt = "/Tp" + src
473	n/a	elif ext in self._rc_extensions:
474	n/a	# compile .RC to .RES file
475	n/a	input_opt = src
476	n/a	output_opt = "/fo" + obj
477	n/a	try:
478	n/a	self.spawn([self.rc] + pp_opts +
479	n/a	[output_opt] + [input_opt])
480	n/a	except PackagingExecError as msg:
481	n/a	raise CompileError(msg)
482	n/a	continue
483	n/a	elif ext in self._mc_extensions:
484	n/a	# Compile .MC to .RC file to .RES file.
485	n/a	# * '-h dir' specifies the directory for the
486	n/a	# generated include file
487	n/a	# * '-r dir' specifies the target directory of the
488	n/a	# generated RC file and the binary message resource
489	n/a	# it includes
490	n/a	#
491	n/a	# For now (since there are no options to change this),
492	n/a	# we use the source-directory for the include file and
493	n/a	# the build directory for the RC file and message
494	n/a	# resources. This works at least for win32all.
495	n/a	h_dir = os.path.dirname(src)
496	n/a	rc_dir = os.path.dirname(obj)
497	n/a	try:
498	n/a	# first compile .MC to .RC and .H file
499	n/a	self.spawn([self.mc] +
500	n/a	['-h', h_dir, '-r', rc_dir] + [src])
501	n/a	base, _ = os.path.splitext(os.path.basename(src))
502	n/a	rc_file = os.path.join(rc_dir, base + '.rc')
503	n/a	# then compile .RC to .RES file
504	n/a	self.spawn([self.rc] +
505	n/a	["/fo" + obj] + [rc_file])
506	n/a
507	n/a	except PackagingExecError as msg:
508	n/a	raise CompileError(msg)
509	n/a	continue
510	n/a	else:
511	n/a	# how to handle this file?
512	n/a	raise CompileError("Don't know how to compile %s to %s"
513	n/a	% (src, obj))
514	n/a
515	n/a	output_opt = "/Fo" + obj
516	n/a	try:
517	n/a	self.spawn([self.cc] + compile_opts + pp_opts +
518	n/a	[input_opt, output_opt] +
519	n/a	extra_postargs)
520	n/a	except PackagingExecError as msg:
521	n/a	raise CompileError(msg)
522	n/a
523	n/a	return objects
524	n/a
525	n/a
526	n/a	def create_static_lib(self,
527	n/a	objects,
528	n/a	output_libname,
529	n/a	output_dir=None,
530	n/a	debug=False,
531	n/a	target_lang=None):
532	n/a
533	n/a	if not self.initialized:
534	n/a	self.initialize()
535	n/a	objects, output_dir = self._fix_object_args(objects, output_dir)
536	n/a	output_filename = self.library_filename(output_libname,
537	n/a	output_dir=output_dir)
538	n/a
539	n/a	if self._need_link(objects, output_filename):
540	n/a	lib_args = objects + ['/OUT:' + output_filename]
541	n/a	if debug:
542	n/a	pass # XXX what goes here?
543	n/a	try:
544	n/a	self.spawn([self.lib] + lib_args)
545	n/a	except PackagingExecError as msg:
546	n/a	raise LibError(msg)
547	n/a	else:
548	n/a	logger.debug("skipping %s (up-to-date)", output_filename)
549	n/a
550	n/a
551	n/a	def link(self, target_desc, objects, output_filename, output_dir=None,
552	n/a	libraries=None, library_dirs=None, runtime_library_dirs=None,
553	n/a	export_symbols=None, debug=False, extra_preargs=None,
554	n/a	extra_postargs=None, build_temp=None, target_lang=None):
555	n/a	if not self.initialized:
556	n/a	self.initialize()
557	n/a	objects, output_dir = self._fix_object_args(objects, output_dir)
558	n/a	fixed_args = self._fix_lib_args(libraries, library_dirs,
559	n/a	runtime_library_dirs)
560	n/a	libraries, library_dirs, runtime_library_dirs = fixed_args
561	n/a
562	n/a	if runtime_library_dirs:
563	n/a	self.warn("don't know what to do with 'runtime_library_dirs': "
564	n/a	+ str(runtime_library_dirs))
565	n/a
566	n/a	lib_opts = gen_lib_options(self,
567	n/a	library_dirs, runtime_library_dirs,
568	n/a	libraries)
569	n/a	if output_dir is not None:
570	n/a	output_filename = os.path.join(output_dir, output_filename)
571	n/a
572	n/a	if self._need_link(objects, output_filename):
573	n/a	if target_desc == CCompiler.EXECUTABLE:
574	n/a	if debug:
575	n/a	ldflags = self.ldflags_shared_debug[1:]
576	n/a	else:
577	n/a	ldflags = self.ldflags_shared[1:]
578	n/a	else:
579	n/a	if debug:
580	n/a	ldflags = self.ldflags_shared_debug
581	n/a	else:
582	n/a	ldflags = self.ldflags_shared
583	n/a
584	n/a	export_opts = []
585	n/a	for sym in (export_symbols or []):
586	n/a	export_opts.append("/EXPORT:" + sym)
587	n/a
588	n/a	ld_args = (ldflags + lib_opts + export_opts +
589	n/a	objects + ['/OUT:' + output_filename])
590	n/a
591	n/a	# The MSVC linker generates .lib and .exp files, which cannot be
592	n/a	# suppressed by any linker switches. The .lib files may even be
593	n/a	# needed! Make sure they are generated in the temporary build
594	n/a	# directory. Since they have different names for debug and release
595	n/a	# builds, they can go into the same directory.
596	n/a	build_temp = os.path.dirname(objects[0])
597	n/a	if export_symbols is not None:
598	n/a	dll_name, dll_ext = os.path.splitext(
599	n/a	os.path.basename(output_filename))
600	n/a	implib_file = os.path.join(
601	n/a	build_temp,
602	n/a	self.library_filename(dll_name))
603	n/a	ld_args.append('/IMPLIB:' + implib_file)
604	n/a
605	n/a	# Embedded manifests are recommended - see MSDN article titled
606	n/a	# "How to: Embed a Manifest Inside a C/C++ Application"
607	n/a	# (currently at http://msdn2.microsoft.com/en-us/library/ms235591(VS.80).aspx)
608	n/a	# Ask the linker to generate the manifest in the temp dir, so
609	n/a	# we can embed it later.
610	n/a	temp_manifest = os.path.join(
611	n/a	build_temp,
612	n/a	os.path.basename(output_filename) + ".manifest")
613	n/a	ld_args.append('/MANIFESTFILE:' + temp_manifest)
614	n/a
615	n/a	if extra_preargs:
616	n/a	ld_args[:0] = extra_preargs
617	n/a	if extra_postargs:
618	n/a	ld_args.extend(extra_postargs)
619	n/a
620	n/a	self.mkpath(os.path.dirname(output_filename))
621	n/a	try:
622	n/a	self.spawn([self.linker] + ld_args)
623	n/a	except PackagingExecError as msg:
624	n/a	raise LinkError(msg)
625	n/a
626	n/a	# embed the manifest
627	n/a	# XXX - this is somewhat fragile - if mt.exe fails, distutils
628	n/a	# will still consider the DLL up-to-date, but it will not have a
629	n/a	# manifest. Maybe we should link to a temp file? OTOH, that
630	n/a	# implies a build environment error that shouldn't go undetected.
631	n/a	if target_desc == CCompiler.EXECUTABLE:
632	n/a	mfid = 1
633	n/a	else:
634	n/a	mfid = 2
635	n/a	self._remove_visual_c_ref(temp_manifest)
636	n/a	out_arg = '-outputresource:%s;%s' % (output_filename, mfid)
637	n/a	if self.__version < 10:
638	n/a	try:
639	n/a	self.spawn(['mt.exe', '-nologo', '-manifest',
640	n/a	temp_manifest, out_arg])
641	n/a	except PackagingExecError as msg:
642	n/a	raise LinkError(msg)
643	n/a	else:
644	n/a	logger.debug("skipping %s (up-to-date)", output_filename)
645	n/a
646	n/a	def _remove_visual_c_ref(self, manifest_file):
647	n/a	try:
648	n/a	# Remove references to the Visual C runtime, so they will
649	n/a	# fall through to the Visual C dependency of Python.exe.
650	n/a	# This way, when installed for a restricted user (e.g.
651	n/a	# runtimes are not in WinSxS folder, but in Python's own
652	n/a	# folder), the runtimes do not need to be in every folder
653	n/a	# with .pyd's.
654	n/a	with open(manifest_file) as manifest_f:
655	n/a	manifest_buf = manifest_f.read()
656	n/a	pattern = re.compile(
657	n/a	r"""<assemblyIdentity.*?name=("\|')Microsoft\."""\
658	n/a	r"""VC\d{2}\.CRT("\|').*?(/>\|</assemblyIdentity>)""",
659	n/a	re.DOTALL)
660	n/a	manifest_buf = re.sub(pattern, "", manifest_buf)
661	n/a	pattern = "<dependentAssembly>\s*</dependentAssembly>"
662	n/a	manifest_buf = re.sub(pattern, "", manifest_buf)
663	n/a	with open(manifest_file, 'w') as manifest_f:
664	n/a	manifest_f.write(manifest_buf)
665	n/a	except IOError:
666	n/a	pass
667	n/a
668	n/a	# -- Miscellaneous methods -----------------------------------------
669	n/a	# These are all used by the 'gen_lib_options() function, in
670	n/a	# ccompiler.py.
671	n/a
672	n/a	def library_dir_option(self, dir):
673	n/a	return "/LIBPATH:" + dir
674	n/a
675	n/a	def runtime_library_dir_option(self, dir):
676	n/a	raise PackagingPlatformError(
677	n/a	"don't know how to set runtime library search path for MSVC++")
678	n/a
679	n/a	def library_option(self, lib):
680	n/a	return self.library_filename(lib)
681	n/a
682	n/a
683	n/a	def find_library_file(self, dirs, lib, debug=False):
684	n/a	# Prefer a debugging library if found (and requested), but deal
685	n/a	# with it if we don't have one.
686	n/a	if debug:
687	n/a	try_names = [lib + "_d", lib]
688	n/a	else:
689	n/a	try_names = [lib]
690	n/a	for dir in dirs:
691	n/a	for name in try_names:
692	n/a	libfile = os.path.join(dir, self.library_filename(name))
693	n/a	if os.path.exists(libfile):
694	n/a	return libfile
695	n/a	else:
696	n/a	# Oops, didn't find it in any of 'dirs'
697	n/a	return None
698	n/a
699	n/a	# Helper methods for using the MSVC registry settings
700	n/a
701	n/a	def find_exe(self, exe):
702	n/a	"""Return path to an MSVC executable program.
703	n/a
704	n/a	Tries to find the program in several places: first, one of the
705	n/a	MSVC program search paths from the registry; next, the directories
706	n/a	in the PATH environment variable. If any of those work, return an
707	n/a	absolute path that is known to exist. If none of them work, just
708	n/a	return the original program name, 'exe'.
709	n/a	"""
710	n/a	for p in self.__paths:
711	n/a	fn = os.path.join(os.path.abspath(p), exe)
712	n/a	if os.path.isfile(fn):
713	n/a	return fn
714	n/a
715	n/a	# didn't find it; try existing path
716	n/a	for p in os.environ['Path'].split(';'):
717	n/a	fn = os.path.join(os.path.abspath(p),exe)
718	n/a	if os.path.isfile(fn):
719	n/a	return fn
720	n/a
721	n/a	return exe