U
ل=^+ @ s d dl Z d dlZd dlZd dlZd dlZddlmZ ddlmZ ddlm Z e j
dkrhd dl Z dd Z
nd dlZd d Z
e j
d
krejZnG d d d ejZG d
d deZdZdd Zdadd Zdd Zd ddZdd Zdd ZdS ) N ) __version_verifier_modules__) ffiplatform)VerificationError) r c C s t jjd d S N) importlib machineryEXTENSION_SUFFIXES r r /verifier.py_extension_suffixes s r
c C s dd t D S )Nc S s g | ]\}}}|t jkr|qS r )impZ C_EXTENSION).0suffix_typer r r
<listcomp> s
z'_extension_suffixes.<locals>.<listcomp>)r Z get_suffixesr r r r r
s r c s e Zd Z fddZ ZS )NativeIOc s( t |tr|d}tt| | d S )Nascii)
isinstanceZunicodeencodesuperr write)selfs __class__r r r s
zNativeIO.write)__name__
__module__ __qualname__r
__classcell__r r r r r s r c @ sz e Zd Zd ddZd ddZd d
Zd d Zd
d Zdd Zdd Z dd Z
dd Z dd Z d!ddZ
dd Zd d ZdS )"VerifierN F.cc K sv |j jrtd|j jf || _|| _|s4t| } t||}
|
| | _| j | | | _
| | |
| _ |rz|rxt
dndtjd d t|| g|j }tjdkr|d}tt|dd d d @ }|d
d }tt|d d d d @ }|d
d }d|| jj||f }t d }|p0t | _ t j | j || | _ t j | j || | _!|| _"d| _#d| _$d S )NzKfeature not supported with ffi.verify(), but only with ffi.set_source(): %sz)can't specify both 'modulename' and 'tag' r r zutf-8r l Z0xLr 0z_cffi_%s_%s%s%sF)%Z_parserZ_uses_new_featurer ffipreambler Zflatten_locate_engine_class_vengineZpatch_extension_kwdsflagsmake_relative_tokwds TypeErrorjoinsysversionr Z _cdefsources version_infor hexbinasciiZcrc32lstriprstripZ
_class_key_get_so_suffixes_caller_dir_pycachetmpdirospathsourcefilenamemodulefilename ext_package _has_source _has_module)r r* r+ r< Z
modulenamerA tagforce_generic_engineZsource_extensionr. relative_tor0 Zflattened_kwdsZ
vengine_classkeyZk1Zk2r r r r __init__ sR
zVerifier.__init__c C s8 | j j& | jr |dkr td| | W 5 Q R X dS )zkWrite the C source code. It is produced in 'self.sourcefilename',
which can be tweaked beforehand.Nzsource code already written)r* _lockrB r
_write_sourcer filer r r write_sourceH s
zVerifier.write_sourcec C s< | j j* | jrtd| js&| | W 5 Q R X dS )zWrite the C source code (if not done already) and compile it.
This produces a dynamic link library in 'self.modulefilename'.zmodule already compiledN)r* rI rC r rB rJ _compile_moduler r r r compile_moduleQ s
zVerifier.compile_modulec
C sR | j j@ | js4| | js4| js,| | | W 5 Q R S Q R X dS )a" Get a C module from this Verifier instance.
Returns an instance of a FFILibrary class that behaves like the
objects returned by ffi.dlopen(), but that delegates all
operations to the C module. If necessary, the C code is written
and compiled first.
N)r* rI rC _locate_modulerB rJ rN
_load_libraryrO r r r load_library[ s
zVerifier.load_libraryc C sB t j| j}|ddd }|dr>ttdr>|d d }|S )N.r r Z_dZgettotalrefcount)r= r> basenamer@ splitendswithhasattrr3 )r rV r r r get_module_namek s
zVerifier.get_module_namec C sV t | js0| jj | js&| W 5 Q R X t | j}| }t j ||f| j
S r ) r Z_hack_at_distutilsrB r* rI rJ Zmaybe_relative_pathr? rZ
get_extensionr0 )r Z
sourcenamemodnamer r r r[ v s
zVerifier.get_extensionc C s | j jS r )r- Z_gen_python_modulerO r r r generates_python_module s z Verifier.generates_python_modulec sz |rvt j|rvt j| | }tjD ]J}||kr*|| }t|ttfsZt d|f fdd|D }|||< q*|S )Nz&keyword '%s' should be a list or tuplec s g | ]}t j |qS r )r= r> r2 )r fndirnamer r r s z-Verifier.make_relative_to.<locals>.<listcomp>)
r= r> r` copyr ZLIST_OF_FILE_NAMESr listtupler1 )r r0 rF rG Zlstr r_ r r/ s
zVerifier.make_relative_toc C s t j| jsv| jrJzt| jd d dg}W n tk
r@ Y d S X |j}nd }| j |
|t }|d krpd S || _| j d| _
d S )N__doc__T)r= r> isfiler@ rA
__import__ ImportError__path__r- find_modulerZ r: Z
collect_typesrC )r Zpkgr> filenamer r r rQ s
zVerifier._locate_modulec C s$ || j _z| j W 5 | j `X d S r )r- Z_fZwrite_source_to_frK r r r _write_source_to s zVerifier._write_source_toc C s |d k r| | nt }| | | }tj| jrbt| jd}| |k }W 5 Q R X nd}|rt | j t| jd}|
| W 5 Q R X d| _ d S )NrTw) rk r getvaluer= r> existsr? openread _ensure_dirr rB )r rL fZ source_datafpZ
needs_writtenr r r rJ s
zVerifier._write_sourcec C sp t j| j}t|| }zt|| j}W n t k
rH d}Y nX |sft
| j t || j d| _
d S )NFT)r= r> r` r? r compiler[ samefiler@ OSErrorrr shutilZmoverC )r r< ZoutputfilenameZsamer r r rN s
zVerifier._compile_modulec C s0 | j s
t| jd k r"| j| jS | j S d S r )rC AssertionErrorr. r- rS rO r r r rR s
zVerifier._load_library)NNNr$ Fr% NN)N)N)r r r! rH rM rP rS rZ r[ r] r/ rQ rk rJ rN rR r r r r r# s(
(
r# Fc C s| t rd}|sPdtjkr d}n4z dd l}W n tk
r@ d}Y nX | j|k rPd}|rfddlm} |jS ddlm } |j
S d S )NTZ__pypy__r ?r ) vengine_gen) vengine_cpy) _FORCE_GENERIC_ENGINEr3 builtin_module_names
_cffi_backendrg Z_backendr$ r{ ZVGenericEnginer| ZVCPythonEngine)r* rE r r{ r| r r r r, s
r, c C sH t rt S tjd} | r | S tdjj}tj tj
tj |dS )NZ CFFI_TMPDIRr' __pycache__) _TMPDIRr= environgetr3 _getframef_code co_filenamer> abspathr2 r` )resultrj r r r r; s r; c C s | a dS )z:Set the temporary directory to use instead of __pycache__.N)r r_ r r r
set_tmpdir s r c C s$ | pt } zt| }W n tk
r. Y dS X |r:d}nt d }|D ]\}| drL| |sz| drLzttj
| | W qL tk
r Y qLX qLtj
| dg}|D ]`}zDt|D ]4}tj
||}tj |r| | qt| qW q tk
r Y qX qdS )z}Clean up the temporary directory by removing all files in it
called `_cffi_*.{c,so}` as well as the `build` subdirectory.Nr% r Z_cffi_Zbuild)
r; r= listdirrw r: lower
startswithrX unlinkr> r2 isdirappend)r< Zkeep_soZfilelistr r^ Z clean_dirdirr r r cleanup_tmpdir s8
r c C s& t } | s"tjdkr dg} ndg} | S )NZwin32z.pydz.so)r
r3 platform)suffixesr r r r: $ s
r: c C s* t j| }|r&t j|s&t | d S r )r= r> r` r makedirs)rj r` r r r rr / s rr )NF) r3 r= r7 rx ior$ r r errorr r5 Zimportlib.machineryr r
r StringIOr BytesIOobjectr# r} r, r r; r r r: rr r r r r <module> s* (
?