U
e5d/w � @ sR d Z ddlZddlZddlZddlZddlmZmZmZm Z m
Z
ddl m Z m
Z
mZ ddlmZ ddlmZ ddlZejZejZejZejZej ej ej ej fZ ej!dko�ej"dkZ#e#r�d Z$d
Z%d Z&n d Z$d
Z%dZ&ddd�Z'G dd� d�Z(G dd� d�Z)dd� Z*dd� Z+dd� Z,d d � Z-d$d d �Z.e*� Z/e/d k �r>ed!e/ ��G d"d#� d#e �Z0dS )%a distutils.msvc9compiler
Contains MSVCCompiler, an implementation of the abstract CCompiler class
for the Microsoft Visual Studio 2008.
The module is compatible with VS 2005 and VS 2008. You can find legacy support
for older versions of VS in distutils.msvccompiler.
� N)�DistutilsExecError�DistutilsPlatformError� CompileError�LibError� LinkError)� CCompiler�gen_preprocess_options�gen_lib_options)�log)� get_platform�win32l z1Software\Wow6432Node\Microsoft\VisualStudio\%0.1fz5Software\Wow6432Node\Microsoft\Microsoft SDKs\Windowsz,Software\Wow6432Node\Microsoft\.NETFrameworkz%Software\Microsoft\VisualStudio\%0.1fz)Software\Microsoft\Microsoft SDKs\Windowsz Software\Microsoft\.NETFramework�x86Zamd64�r z win-amd64c @ sP e Zd ZdZdd� Zee�Zdd� Zee�Zdd� Zee�Zdd � Ze e�Zd
S ) �Regz2Helper class to read values from the registry
c C s: t D ](}| �||�}|r||kr|| S qt|��d S �N)�HKEYS� read_values�KeyError)�cls�path�key�base�d� r �//usr/lib64/python3.8/distutils/msvc9compiler.py� get_value@ s
z
Reg.get_valuec C sn zt ||�}W n tk
r$ Y dS X g }d}zt||�}W n tk
rT Y qjY nX |�|� |d7 }q.|S )z Return list of registry keys.Nr � )� RegOpenKeyEx�RegError�
RegEnumKey�append)r r r �handle�L�i�kr r r � read_keysH s
z
Reg.read_keysc C s� zt ||�}W n tk
r$ Y dS X i }d}zt||�\}}}W n tk
rZ Y q�Y nX |�� }| �|�|| �|�< |d7 }q.|S )z`Return dict of registry keys and values.
All names are converted to lowercase.
Nr r )r r � RegEnumValue�lower� convert_mbcs) r r r r! r r# �name�value�typer r r r Z s
zReg.read_valuesc C s: t | dd �}|d k r6z |d�} W n tk
r4 Y nX | S )N�decode�mbcs)�getattr� UnicodeError)�sZdecr r r r( p s zReg.convert_mbcsN)
�__name__�
__module__� __qualname__�__doc__r � classmethodr% r r( � staticmethodr r r r r < s r c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
�
MacroExpanderc C s i | _ t| | _| �|� d S r )�macros�VS_BASE�vsbase� load_macros)�self�versionr r r �__init__| s
zMacroExpander.__init__c C s t �||�| jd| < d S )Nz$(%s))r r r8 )r< Zmacror r r r r � set_macro� s zMacroExpander.set_macroc C s | � d| jd d� | � d| jd d� | � dtd� z$|dkrP| � d td
� ntd
��W n tk
rv td ��Y nX |d kr�| � d
| jd� | � dtd� nbd}tD ]X}zt||�}W n tk
r� Y q�Y nX t |d�}t
� |d||f �}|d | j d< q�d S )NZ VCInstallDirz \Setup\VC�
productdirZ VSInstallDirz \Setup\VSZ FrameworkDirZ installroot� @ZFrameworkSDKDirzsdkinstallrootv2.0a Python was built with Visual Studio 2008;
extensions must be built with a compiler than can generate compatible binaries.
Visual Studio 2008 was not found on this system. If you have Cygwin installed,
you can try compiling with MingW32, by passing "-c mingw32" to setup.py.g "@ZFrameworkVersionz clr versionZ
WindowsSdkDirZcurrentinstallfolderz.Software\Microsoft\NET Framework Setup\Productr z%s\%sr= z$(FrameworkVersion))
r? r: �NET_BASEr r � WINSDK_BASEr r r r r r r8 )r< r= �pr �hr r r r r r; � s2 � �
zMacroExpander.load_macrosc C s$ | j �� D ]\}}|�||�}q
|S r )r8 �items�replace)r< r0 r$ �vr r r �sub� s zMacroExpander.subN)r1 r2 r3 r> r? r; rI r r r r r7 z s r7 c C s� d} t j�| �}|dkr dS |t| � }t j|d� �dd�\}}t|dd� �d }|dkrf|d7 }t|d d
� �d }|dkr�d }|dkr�|| S dS )
z�Return the version of MSVC that was used to build Python.
For Python 2.3 and up, the version number is included in
sys.version. For earlier versions, assume the compiler is MSVC 6.
zMSC v.������ N� r ������
� � g $@r )�sysr= �find�len�split�int)�prefixr# r0 �restZ majorVersionZ minorVersionr r r �get_build_version� s rX c C s0 g }| D ]"}t j�|�}||kr|�|� q|S )znReturn a list of normalized paths with duplicates removed.
The current order of paths is maintained.
)�osr �normpathr )�pathsZ
reduced_pathsrD Znpr r r �normalize_and_reduce_paths� s r\ c C s< | � tj�}g }|D ]}||kr|�|� qtj�|�}|S )z8Remove duplicate values of an environment variable.
)rT rY �pathsepr �join)ZvariableZoldListZnewListr# Z newVariabler r r �removeDuplicates� s r_ c C s t | }zt�d| d�}W n" tk
r> t�d� d}Y nX |rPtj�|�s�d| }tj �
|d�}|r�tj�|�r�tj� |tj tj d�}tj�
|�}tj�|�s�t�d| � dS nt�d| � |s�t�d � dS tj� |d
�}tj�|�r�|S t�d � dS ) z�Find the vcvarsall.bat file
At first it tries to find the productdir of VS 2008 in the registry. If
that fails it falls back to the VS90COMNTOOLS env var.
z %s\Setup\VCr@ z%Unable to find productdir in registryNzVS%0.f0COMNTOOLSZVCz%s is not a valid directoryz Env var %s is not set or invalidzNo productdir foundz
vcvarsall.bat� Unable to find vcvarsall.bat)r9 r r r r
�debugrY r �isdir�environ�getr^ �pardir�abspath�isfile)r= r: r@ ZtoolskeyZtoolsdir� vcvarsallr r r �find_vcvarsall� s4
�
ri c C s8 t | �}ddddh}i }|dkr(td��t�d|| � tjd||f tjtjd �}z�|�
� \}}|� � d
krzt|� d ���|� d �}|�
d �D ]d}t�|�}d
|kr�q�|�� }|�
d
d�\} }
| �� } | |kr�|
�tj�r�|
dd� }
t|
�|| < q�W 5 |j�� |j �� X t|�t|�k�r4ttt|�� ����|S )zDLaunch vcvarsall.bat and read the settings from its environment
�include�libZlibpathr Nr` z'Calling 'vcvarsall.bat %s' (version=%s)z
"%s" %s & set)�stdout�stderrr r- �
�=r rJ )ri r r
ra �
subprocess�Popen�PIPErl �closerm Z communicate�waitr, rT r r( �stripr' �endswithrY r] r_ rS �
ValueError�str�list�keys) r= Zarchrh Z interesting�result�popenrl rm �liner r* r r r �query_vcvarsall� s> �
r~ rA z(VC %0.1f is not supported by this modulec
@ s� e Zd ZdZdZi ZdgZdddgZdgZdgZ ee e e Z
d Z d
Z d Z
d Zd
ZZdZd.dd�Zd/dd�Zd0dd�Zd1dd�Zd2dd�Zd3d d �Zd d � Zd d!� Zd"d#� Zd$d%� Zd&d'� Z d(d)� Z d4d*d+�Z d,d-� Z dS )5� MSVCCompilerzwConcrete class that implements an interface to Microsoft Visual C++,
as defined by the CCompiler abstract class.Zmsvcz.cz.ccz.cppz.cxx�.rcz.mcz.resz.objz.libz.dllz%s%sz.exer c C s8 t �| |||� t| _d| _g | _d | _d | _d| _d S )Nz Software\Microsoft\VisualStudioF) r r> �VERSION�_MSVCCompiler__versionZ_MSVCCompiler__root�_MSVCCompiler__paths� plat_name�_MSVCCompiler__arch� initialized)r< �verboseZdry_runZforcer r r r> I s zMSVCCompiler.__init__Nc C s |d krt � }d}||kr(td|f ��dtjkrfdtjkrf| �d�rfd| _d| _d| _d| _d | _ n�|t � ksx|d
kr�t
| }nt
t � d t
| }t t |�}|d �
tj�| _|d
tjd
< |d tjd< t| j�dkr�td| j ��| �d�| _| �d�| _| �d�| _| �d�| _| �d �| _ z(tjd �
d�D ]}| j�|� �q:W n tk
�rf Y nX t| j�| _d�| j�tjd < d | _| jdk�r�dddddg| _ddddddg| _n&ddddd dg| _ddddd ddg| _d dd g| _| jd k�rd dd d!g| _ dg| _ d"| _ d S )#Nr z --plat-name must be one of %sZDISTUTILS_USE_SDKZMSSdkzcl.exezlink.exezlib.exezrc.exezmc.exer �_r rk rj r zxPython was built with %s, and extensions need to be built with the same version of the compiler, but it isn't installed.�;r
z/nologoz/Oxz/MDz/W3z/DNDEBUGz/Odz/MDdz/Z7z/D_DEBUGz/GS-z/DLLz/INCREMENTAL:NO� z/INCREMENTAL:noz/DEBUGT) r r rY rc �find_exe�cc�linkerrk �rc�mc�PLAT_TO_VCVARSr~ r� rT r] r� rS Z_MSVCCompiler__productr r r\ r^ Zpreprocess_optionsr� �compile_options�compile_options_debug�ldflags_sharedr� �ldflags_shared_debugZldflags_staticr� )r< r� Zok_platsZ plat_specZvc_envrD r r r �
initializeT s~ �
�
� � �
�
� �zMSVCCompiler.initialize� c C s� |d kr d}g }|D ]�}t j�|�\}}t j�|�d }|t j�|�d � }|| jkrbtd| ��|rrt j�|�}|| jkr�|� t j�
||| j �� q|| j kr�|� t j�
||| j �� q|� t j�
||| j
�� q|S )Nr� r z Don't know how to compile %s)rY r �splitext�
splitdrive�isabs�src_extensionsr �basename�_rc_extensionsr r^ �
res_extension�_mc_extensions�
obj_extension)r< Zsource_filenamesZ strip_dir�
output_dirZ obj_namesZsrc_namer �extr r r �object_filenames� s.
�
� �z MSVCCompiler.object_filenamesc C sp | j s| �� | �||||||�} | \}}
}} } |p6g }
|
�d� |rT|
�| j� n |
�| j� |
D �]}z| | \}}W n tk
r� Y qdY nX |r�tj �
|�}|| j kr�d| }�nT|| j kr�d| }�n>|| j
k�r<|}d| }z"| �| jg| |g |g � W qd tk
�r6 } z t|��W 5 d }~X Y qdX qdn�|| jk�r�tj �|�}tj �|�}zl| �| jgd|d|g |g � tj �tj �|��\}}tj �||d �}| �| jgd| g |g � W qd tk
�r� } z t|��W 5 d }~X Y qdX qdntd||f ��d | }z&| �| jg|
| ||g | � W qd tk
�rh } z t|��W 5 d }~X Y qdX qd|
S )
Nz/cz/Tcz/Tpz/foz-hz-rr� z"Don't know how to compile %s to %sz/Fo)r� r� Z_setup_compiler �extendr� r� r rY r rf �
_c_extensions�_cpp_extensionsr� �spawnr� r r r� �dirnamer� r� r� r^ r� )r< Zsourcesr� r8 Z include_dirsra �
extra_preargs�extra_postargsZdependsZ compile_info�objectsZpp_optsZbuildZ compile_opts�obj�srcr� Z input_optZ
output_opt�msgZh_dirZrc_dirr r� Zrc_filer r r �compile� s�
�
��
��
��
���
zMSVCCompiler.compilec
C s� | j s| �� | �||�\}}| j||d�}| �||�r�|d| g }|rJz| �| jg| � W q� tk
r� } z t|��W 5 d }~X Y q�X n t �
d|� d S )N)r� �/OUT:�skipping %s (up-to-date)) r� r� �_fix_object_args�library_filename�
_need_linkr� rk r r r
ra ) r< r� Zoutput_libnamer� ra � target_lang�output_filenameZlib_argsr� r r r �create_static_lib s � z MSVCCompiler.create_static_libc
C sT | j s| �� | �||�\}}| �|||�}|\}}}|rL| �dt|� � t| |||�}|d k rptj� ||�}| �
||��rD|t j kr�| r�| j
dd � }q�| jdd � }n| r�| j
}n| j}g }|p�g D ]}|�d| � q�|| | | d| g }tj�|d �} |d k �rLtj�tj�|��\}}tj� | | �|��}|�d| � | �|| |� |
�rl|
|d d�< | �r||�| � | �tj�|�� z| �| jg| � W n, tk
�r� } z t|��W 5 d }~X Y nX | �||�}|d k �rP|\}}d||f }z| �dd d
||g� W n, tk
�r@ } z t|��W 5 d }~X Y nX n t � d |� d S ) Nz5I don't know what to do with 'runtime_library_dirs': r z/EXPORT:r� r z/IMPLIB:z-outputresource:%s;%szmt.exez-nologoz -manifestr� ) r� r� r� Z
_fix_lib_args�warnrx r rY r r^ r� r �
EXECUTABLEr� r� r r� r� r� r� �manifest_setup_ldargsr� Zmkpathr� r� r r �manifest_get_embed_infor
ra ) r< � target_descr� r� r� Z librariesZ library_dirsZruntime_library_dirsZexport_symbolsra r� r� �
build_tempr� Z
fixed_argsZlib_optsZldflagsZ export_optsZsym�ld_argsZdll_nameZdll_extZ implib_filer� ZmfinfoZ
mffilename�mfidZout_argr r r �link6 s� �
� �
��
��
�
zMSVCCompiler.linkc C s, t j�|t j�|�d �}|�d| � d S )Nz .manifest�/MANIFESTFILE:)rY r r^ r� r )r< r� r� r� �
temp_manifestr r r r� � s
�z"MSVCCompiler.manifest_setup_ldargsc C s^ |D ]"}|� d�r|�dd�d } q,qd S |tjkr|)r� z*\s*zI|)�w)
�openrs �read�rer� �DOTALLrI �search�write�OSError)r< Z
manifest_fileZ
manifest_fZ manifest_buf�patternr r r r� � s2
��
z!MSVCCompiler._remove_visual_c_refc C s d| S )Nz /LIBPATH:r �r< �dirr r r �library_dir_option� s z MSVCCompiler.library_dir_optionc C s t d��d S )Nz r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r r + sl
��
W �
�
X �
�
_ +
r )r
)1r4 rY rp rQ r� Zdistutils.errorsr r r r r Zdistutils.ccompilerr r r Z distutilsr
Zdistutils.utilr �winregZ OpenKeyExr ZEnumKeyr Z EnumValuer& �errorr Z
HKEY_USERS�HKEY_CURRENT_USER�HKEY_LOCAL_MACHINEZHKEY_CLASSES_ROOTr �platform�maxsizeZ NATIVE_WIN64r9 rC rB r� r r7 rX r\ r_ ri r~ r� r r r r r � sP ��>. #
)