U
vf+ @ 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 @/opt/alt/python38/lib64/python3.8/site-packages/cffi/verifier.py_extension_suffixes s r
c C s dd t D S )Nc S s g | ]\}}}|t jkr|qS r )imp C_EXTENSION).0suffix_typer r r
s
z'_extension_suffixes..)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)
isinstanceunicodeencodesuperr 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 sz |j jrtd|j jf || _|| _|s4t| } t||}
|
| | _| j | | | _
| | |
| _ |rz|rxt
dnddtjd d t|| g|j }tjdkr|d}tt|dd d d @ }|d
d }tt|d d d d @ }|d
d }d|| jj||f }t d }|p4t | _ 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' z%d.%d r zutf-8r l 0xLr 0z_cffi_%s_%s%s%sF)$_parserZ_uses_new_featurer ffipreambler flatten_locate_engine_class_vengineZpatch_extension_kwdsflagsmake_relative_tokwds TypeErrorjoinsys version_infor Z _cdefsourcesr hexbinasciicrc32lstriprstripZ
_class_key_get_so_suffixes_caller_dir_pycachetmpdirospathsourcefilenamemodulefilename ext_package _has_source _has_module)r r. r/ rA
modulenamerF tagforce_generic_engineZsource_extensionr3 relative_tor5 Zflattened_kwdsZ
vengine_classkeyZk1Zk2r r r r __init__ sT
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. _lockrG r
_write_sourcer filer r r write_sourceI 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. rO rH r rG rP _compile_moduler r r r compile_moduleR 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. rO rH _locate_modulerG rP rT
_load_libraryrU r r r load_library\ s
zVerifier.load_libraryc C sB t j| j}|ddd }|dr>ttdr>|d d }|S )N.r r Z_dgettotalrefcount)rB rC basenamerE splitendswithhasattrr8 )r r] r r r get_module_namel s
zVerifier.get_module_namec C sN | j s(| jj | j s | W 5 Q R X t| j}| }tj||f| j S r )
rG r. rO rP r Zmaybe_relative_pathrD ra
get_extensionr5 )r Z
sourcenamemodnamer r r rb w s
zVerifier.get_extensionc C s | j jS r )r2 Z_gen_python_modulerU 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 )rB rC r7 )r fndirnamer r r s z-Verifier.make_relative_to..)
rB rC rg copyr ZLIST_OF_FILE_NAMESr listtupler6 )r r5 rL rM lstr rf r r4 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)rB rC isfilerE rF
__import__ ImportError__path__r2 find_modulera r? Z
collect_typesrH )r pkgrC filenamer r r rW s
zVerifier._locate_modulec C s$ || j _z| j W 5 | j `X d S r )r2 Z_fZwrite_source_to_frQ 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) rt r getvaluerB rC existsrD openread _ensure_dirr rG )r rR fZ source_datafpZ
needs_writtenr r r rP 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)rB rC rg rD r compilerb samefilerE OSErrorr{ shutilmoverH )r rA ZoutputfilenameZsamer r r rT s
zVerifier._compile_modulec C s0 | j s
t| jd k r"| j| jS | j S d S r )rH AssertionErrorr3 r2 rY rU r r r rX s
zVerifier._load_library)NNNr& Fr' NN)N)N)r! r" r# rN rS rV rY ra rb rd r4 rW rt rP rT rX 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_ENGINEr8 builtin_module_names
_cffi_backendro Z_backendr& r ZVGenericEnginer ZVCPythonEngine)r. rK r r r r r r r1 s
r1 c C sH t rt S tjd} | r | S tdjj}tj tj
tj |dS )NZ CFFI_TMPDIRr) __pycache__) _TMPDIRrB environgetr8 _getframef_code co_filenamerC abspathr7 rg )resultrs r r r r@ s r@ c C s | a dS )z:Set the temporary directory to use instead of __pycache__.N)r rf 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_build)
r@ rB listdirr r? lower
startswithr_ unlinkrC r7 isdirappend)rA Zkeep_sofilelistr re Z clean_dirdirr r r cleanup_tmpdir s8
r c C s& t } | s"tjdkr dg} ndg} | S )Nwin32z.pydz.so)r
r8 platform)suffixesr r r r? $ s
r? c C s* t j| }|r&t j|s&t | d S r )rB rC rg r makedirs)rs rg r r r r{ / s r{ )NF) r8 rB r; r ior& r r errorr r9 Zimportlib.machineryr r
r StringIOr BytesIOobjectr% r r1 r r@ r r r? r{ r r r r s* (
?