3
MZ4h @ sH d dl Z d dlZd dlZddlmZ ddlmZ G dd deZdZ dS ) N )model)VerificationErrorc @ s e Zd ZdZdZdd Zdd Zdd Zd d
ZdZd d
Z dd Z
d[ddZ dd Z dd Z
dd Zdd Zdd ZeZeZeZd d ZeZd d Zd!d" Zd#d$ Zd%d& Zd'd( Zd)d* Zd+d, Z d-d. Z d/d0 Z d1d2 Z d3d4 Z d5d6 Z!d7d8 Z"d9d: Z#d\d=d>Z$d?d@ Z%eZ&d]dAdBZ'dCdD Z(dEdF Z)dGdH Z*dIdJ Z+d^dLdMZ,d_dNdOZ-dPdQ Z.dRdS Z/eZ0dTdU Z1dVdW Z2eZ3dXdY Z4d;S )`VGenericEnginegFc C s || _ |j| _g | _i | _d S )N)verifierffiexport_symbols _struct_pending_verification)selfr r #/usr/lib64/python3.6/vengine_gen.py__init__ s zVGenericEngine.__init__c C s |j d| j d S )Nr )
setdefaultr )r kwdsr r r
patch_extension_kwds s z#VGenericEngine.patch_extension_kwdsc C sT xN|D ]F}|| }|d kr t j}x*|D ]"}tjj||}tjj|r&|S q&W qW d S )N)syspathosjoinisfile)r Z module_namer Z so_suffixesZ so_suffixbasenamedirnamefilenamer r r
find_module s
zVGenericEngine.find_modulec C s d S )Nr )r r r r
collect_types% s z VGenericEngine.collect_types c C s | j j|d d S )N
)Z_fwrite)r Zwhatr r r
_prnt( s zVGenericEngine._prntc C s` | j }|t || jj | jd tjdkr\tjdkr>d}nd}| jj }|d||f d S )NZdeclZwin32 ZPyInit_Zinitzvoid %s%s(void) { }
)r ) r cffimod_headerr Zpreamble _generater platform version_infoZget_module_name)r prntprefixmodnamer r r
write_source_to_f+ s
z VGenericEngine.write_source_to_fr c sj j j}tjjtjjj}|j||j d G fdddt
j d}j d|d |S )NZloadingc s& e Zd ZZjZg Z fddZdS )z/VGenericEngine.load_library..FFILibraryc s j S )N) _cffi_dir)r )
FFILibraryr r
__dir__T s z7VGenericEngine.load_library..FFILibrary.__dir__N)__name__
__module__ __qualname__Z_cffi_generic_moduler Z _cffi_ffir) r+ r )r* moduler r r
r* P s r* r Zloaded)library) r _backendr r r curdirr Zmodulefilename load_library_loadtypes
ModuleType)r flagsbackendr r0 r )r* r/ r r
r3 A s zVGenericEngine.load_libraryc C s$ dd | j jjj D }|j |S )Nc S s g | ]\}\}}||fqS r r ).0keytpZqualr r r
^ s z4VGenericEngine._get_declarations..)r Z_parserZ
_declarationsitemssort)r Zlstr r r
_get_declarations] s z VGenericEngine._get_declarationsc C s x| j D ]\}}|jdd\}}yt| d||f }W n tk
rX td| Y nX y||| W q
tk
r } ztj|| W Y d d }~X q
X q
W d S )N r z_generate_gen_%s_%sz not implemented in verify(): %r)r? splitgetattrAttributeErrorr Exceptionr attach_exception_info)r step_namenamer; kindrealnamemethoder r r
r" c s zVGenericEngine._generatec
K s x|| j D ]p\}}|jdd\}}t| d||f }y||||f| W q
tk
rx } ztj| | W Y d d } ~ X q
X q
W d S )Nr@ r z
_%s_gen_%s)r? rA rB rD r rE )
r r/ rF r rG r; rH rI rJ rK r r r
r4 r s zVGenericEngine._loadc C s d S )Nr )r r; rG r r r
_generate_nothing| s z VGenericEngine._generate_nothingc K s d S )Nr )r r; rG r/ r r r r
_loaded_noop s zVGenericEngine._loaded_noopc sx |j r| jd|| d S | j}t|j}g }xVGenericEngine._generate_gen_function_decl..r z *rz, voidz
_cffi_f_%sr@ z %s%s(%s)z result of %s{z*r = zreturn z %s%s(%s);})ellipsis_generate_gen_constr lenargs enumerate
isinstancer
StructOrUnionappendzipresultinsertrO void_typer r abiZVoidType)r r; rG r% ZnumargsZargnamesirP Z indirectionZarglistZtpresult wrappernamerb ZfuncdeclZ result_coder )rR r
_generate_gen_function_decl sH
z*VGenericEngine._generate_gen_function_declc C s` |j r| jd|||}n&g }|}tdd |jD sDt|jtjrg }xDt|jD ]6\} }
t|
tjrtj |
}
|j
| |
f |j
|
qTW |j} t| tjr| j d krt d| j
f tj | } |jd| |jdd| f tj} tjt|| |j }| jj|} d| }
|j| |
}x"|D ]\} }
| j|| |
|}q"W t||| t|jj
| d S )NFc s s | ]}t |tjV qd S )N)r[ r r\ )r9 typr r r
s z6VGenericEngine._loaded_gen_function..z*'%s' is used as result type, but is opaquer r_ z
_cffi_f_%s)rV _load_constantanyrY r[ r_ r r\ rZ PointerTyper] Zfldtypes TypeError _get_c_namer` ra ZFunctionPtrTypetupler _get_cached_btype
load_function_make_struct_wrappersetattrrP r) )r r; rG r/ r0 Z newfunctionZ indirectionsbase_tpZ
indirect_argsrc rf Zindirect_resultBFuncrd r r r
_loaded_gen_function s>
z#VGenericEngine._loaded_gen_functionc sP | j j| j j| dkr4| j fdd}n fdd}||_|S )Nr_ c s j }|f| |d S )Nr )new)rY res)BTyper oldfuncr r
newfunc s
z4VGenericEngine._make_struct_wrapper..newfuncc s6 | d j | f | d d } | S )Nr )Znewp)rY )rw r8 rc rx r r
ry s .)r r1 rn Z_cffi_base_type)r rx rc r; rr ry r )rw r8 r rc rx r
rp s z#VGenericEngine._make_struct_wrapperc C s | j |d| d S )Nstruct) _generate_struct_or_union_decl)r r; rG r r r
_generate_gen_struct_decl s z(VGenericEngine._generate_gen_struct_declc C s | j |d|| d S )Nrz )_loading_struct_or_union)r r; rG r/ r r r
_loading_gen_struct s z"VGenericEngine._loading_gen_structc K s | j | d S )N)_loaded_struct_or_union)r r; rG r/ r r r r
_loaded_gen_struct s z!VGenericEngine._loaded_gen_structc C s | j |d| d S )Nunion)r{ )r r; rG r r r
_generate_gen_union_decl s z'VGenericEngine._generate_gen_union_declc C s | j |d|| d S )Nr )r} )r r; rG r/ r r r
_loading_gen_union s z!VGenericEngine._loading_gen_unionc K s | j | d S )N)r )r r; rG r/ r r r r
_loaded_gen_union s z VGenericEngine._loaded_gen_unionc
C s |j d krd S d||f }d||f }d||f j }| j}|d||f |d |d |d x|j D ]\}} }
} t| tjr| j s|
dkr|d | qny$|d
| jd d | | d
|f W qn t k
r } z|dt
| W Y d d } ~ X qnX qnW |d | j j | |d|f |d |d| |d |d| |d xt|j D ]h\}} }
} |
dkrvq\|d||f t| tj
r| jd kr|d| j n|d||f q\W |d |d |d |d |d |f |d | d S ) Nz_cffi_check_%s_%sz_cffi_layout_%s_%sz%s %szstatic void %s(%s *p)rT z8 /* only to generate compile-time warnings or errors */z
(void)p;r z (void)((p->%s) << 1);z { %s = &p->%s; (void)tmp; }z*tmpzfield %r)Zqualsz
/* %s */rU zintptr_t %s(intptr_t i)z, struct _cffi_aligncheck { char x; %s y; };z static intptr_t nums[] = {z sizeof(%s),z) offsetof(struct _cffi_aligncheck, y),z offsetof(%s, %s),z 0, /* %s */z sizeof(((%s *)0)->%s),z -1z };z return nums[i];z3 /* the next line is not executed, but compiled */z %s(0);)fldnamesstripr
enumfieldsr[ r
PrimitiveTypeis_integer_typerO r strr r] ArrayTypelengthrl )
r r; r& rG Z
checkfuncnamelayoutfuncnamecnamer% fnameftypefbitsizefqualrK r r r
r{ sT
&
z-VGenericEngine._generate_struct_or_union_declc C s |j d krd S d||f }| jjdd }|j||}g }d} x(|| }
|
dk rRP |j|
| d7 } q@W t|tjr|jr|d } |d } |dd d }
|dd d }|j |
|| | f|_
n d||f j }||f| j |< d S )Nz_cffi_layout_%s_%szintptr_t(*)(intptr_t)r r r z%s %s)
r r _typeof_lockedro r] r[ r r\ partialZ
force_flattenZ fixedlayoutr r
)r r; r& rG r/ r rs functionlayoutZnumxZ totalsizeZtotalalignmentZfieldofsZ fieldsizer r r r
r} 7 s,
z'VGenericEngine._loading_struct_or_unionc
C s |j d krd S | jj| || jkrdd }| j}|j|}| jj|\}}||d |j|d ||d |j|d d}x|j D ]v\}} }
} |
dkrq||| |j||d|f ||d dkr|j| } |||d |j| d |f |d7 }qW d S )
Nc S s | |krt d||| f d S )Nz'%s (we have %d, but C compiler says %d))r )Z realvalueZ
expectedvaluemsgr r r
checkZ s z5VGenericEngine._loaded_struct_or_union..checkr zwrong total sizer zwrong total alignmentr zwrong offset for field %rzwrong size for field %r) r r rn r
popsizeofZalignofr Zoffsetof)
r r; r r ZBStructr r rc r r r r ZBFieldr r r
r S s,
z&VGenericEngine._loaded_struct_or_unionc C s. t |tjr | j||d n| j|d| d S )Nr )r[ r EnumType_generate_gen_enum_declr{ )r r; rG r r r
_generate_gen_anonymous_declu s z+VGenericEngine._generate_gen_anonymous_declc C s2 t |tjr | j|||d n| j|d|| d S )Nr )r[ r r _loading_gen_enumr} )r r; rG r/ r r r
_loading_gen_anonymous{ s z%VGenericEngine._loading_gen_anonymousc K s. t |tjr | j|||f| n
| j| d S )N)r[ r r _loaded_gen_enumr )r r; rG r/ r r r r
_loaded_gen_anonymous s z$VGenericEngine._loaded_gen_anonymousNconstc
C s
| j }d||f }| jj| |d k rX|d| |d | j|| |d |d n|r|d| |d |d|f |d|f |d nj|d krd
}nd }d } |d krt|tjrd
} d
}||jd| |f | |d |d||f |d | d S )Nz _cffi_%s_%szint %s(char *out_error)rT z return 0;rU z int %s(long long *out_value)z *out_value = (long long)(%s);z return (%s) <= 0;var&r r zconst *z %s%s(void)z return (%s%s);)r r r] _check_int_constant_valuer[ r r\ rO )
r is_intrG r; category check_valuer% funcnameZ ampersandZextrar r r
rW s6
z"VGenericEngine._generate_gen_constc C s& t |tjo|j }| j||| d S )N)r[ r r r rW )r r; rG r r r r
_generate_gen_constant_decl s z*VGenericEngine._generate_gen_constant_declc C s d| }|d k r"| j || |}n|r| jjdd }| jjdd } |j| |}
| jj|} |
| } t| d }|dk r| r| jjdd }
|dd| jj|
> 7 }nXd}t|tj rd | }| jj|j
||d } |j| |}
|
}t|tj r|d }|S )
Nz_cffi_const_%sz
long long*r zint(*)(long long*)z long longr z (*)(void)rN ) _load_known_int_constantr r ro ru intr r[ r r\ rO )r r r; rG r/ r r valuerw rs r pnegativeZ BLongLongZ fntypeextrar r r
rh s. z VGenericEngine._load_constantc C sD t |tjo|j }| j||||}t||| t|jj| d S )N) r[ r r r rh rq rP r) r] )r r; rG r/ r0 r r r r r
_loaded_gen_constant s z#VGenericEngine._loaded_gen_constantc C s | j }|dkr"|d|||f n|d|||f |d |d| |d| |d |d| |d |d
|d d |f |d |d
d S )Nr z& if ((%s) > 0 || (long)(%s) != %dL) {z1 if ((%s) <= 0 || (unsigned long)(%s) != %dUL) {z char buf[64];z if ((%s) <= 0)z) sprintf(buf, "%%ld", (long)(%s));z elsez2 sprintf(buf, "%%lu", (unsigned long)(%s));z: sprintf(out_error, "%s has the real value %s, not %s",z "%s", buf, "%d");d z return -1;z })r )r rG r r% r r r
r s z(VGenericEngine._check_int_constant_valuec C sr | j jdd }| j jdd }|j||}| j j|d}||dk rn| j j|}tjdkrft|d}t|d S )Nzchar[]r z
int(*)(char*) r zutf-8)r ) r r ro ru stringr r$ r r )r r/ r rw rs r r errorr r r
r s
z'VGenericEngine._load_known_int_constantc C s |j dd}d||f S )N$Z___D_z
_cffi_e_%s_%s)replace)r r& rG r r r
_enum_funcname s z VGenericEngine._enum_funcnameenumc C s |j r&x|jD ]}| jd| qW d S | j||}| jj| | j}|d| |d x&t|j|jD ]\}}| j || qhW |d |d | d S )NTzint %s(char *out_error)rT z return 0;rU )
r enumeratorsrW r r r] r r^
enumvaluesr )r r; rG r&
enumeratorr r% enumvaluer r r
r s z&VGenericEngine._generate_gen_enum_declc sL j r0 fddjD }t|_d_nj||}j | d S )Nc s g | ]}j d | qS )T)rh )r9 r )r/ r r; r r
r< s z4VGenericEngine._loading_gen_enum..T)r r rm r Zpartial_resolvedr r )r r; rG r/ r& r r r )r/ r r; r
r s
z VGenericEngine._loading_gen_enumc C s< x6t |j|jD ]$\}}t||| t|jj| qW d S )N)r^ r r rq rP r) r] )r r; rG r/ r0 r r r r r
r s z VGenericEngine._loaded_gen_enumc C s& |dkrd }n|}| j d||d d S )Nz...T)r )rW )r r; rG r r r r
_generate_gen_macro_decl s z'VGenericEngine._generate_gen_macro_declc C sF |dkrd }n|}| j d||||d}t||| t|jj| d S )Nz...T)r )rh rq rP r) r] )r r; rG r/ r0 r r r r r
_loaded_gen_macro) s z VGenericEngine._loaded_gen_macroc C s t |tjrx|jdkr\| j}d|f }| jj| |d| |d |d|f |d tj|j}| j d|| n tj|}| j d||dd d S )
Nz...z_cffi_sizeof_%szsize_t %s(void)rT z return sizeof(%s);rU Fr )r )
r[ r r r r r r] rj itemrW )r r; rG r% r tp_ptrr r r
_generate_gen_variable_decl6 s
z*VGenericEngine._generate_gen_variable_declc s^ t |tjr|jdkrd|f }| jjdd }|j||}| }| jj|j} t || jj
| \}
} | dkrt d||jf |j |
}tj
|j} | jd| ||}
|jd k r| jj|}| jj||
}
t|||
t|jj| d S d| }| jj|jd|d }|j||}| fd d
} fd d }tt||t|| t|jj| d S )
Nz...z_cffi_sizeof_%szsize_t(*)(void)r z/bad size: %r does not seem to be an array of %sFz _cffi_var_%sz
*(*)(void)c s d S )Nr r )r0 )ptrr r
getterg s z3VGenericEngine._loaded_gen_variable..getterc s | d< d S )Nr r )r0 r )r r r
setteri s z3VGenericEngine._loaded_gen_variable..setter)r[ r r r r r ro rn r divmodr r Zresolve_lengthrj rh castrq rP r) r] rO property)r r; rG r/ r0 r rs r sizeZ BItemTyper restr r ZBArrayr r r )r r
_loaded_gen_variableH s:
z#VGenericEngine._loaded_gen_variable)r )r )Nr N)N)r )r )5r, r- r. Z
_class_keyZ_gen_python_moduler r r r r r( r3 r? r" r4 rL rM Z_generate_gen_typedef_declZ_loading_gen_typedefZ_loaded_gen_typedefre Z_loading_gen_functionrt rp r| r~ r r r r r{ r} r r r r rW r Z_loading_gen_constantrh r r r r r r r r Z_loading_gen_macror r Z_loading_gen_variabler r r r r
r sf
-$1 "
$
r ad
#include
#include
#include
#include
#include /* XXX for ssize_t on some platforms */
/* this block of #ifs should be kept exactly identical between
c/_cffi_backend.c, cffi/vengine_cpy.py, cffi/vengine_gen.py
and cffi/_cffi_include.h */
#if defined(_MSC_VER)
# include /* for alloca() */
# if _MSC_VER < 1600 /* MSVC < 2010 */
typedef __int8 int8_t;
typedef __int16 int16_t;
typedef __int32 int32_t;
typedef __int64 int64_t;
typedef unsigned __int8 uint8_t;
typedef unsigned __int16 uint16_t;
typedef unsigned __int32 uint32_t;
typedef unsigned __int64 uint64_t;
typedef __int8 int_least8_t;
typedef __int16 int_least16_t;
typedef __int32 int_least32_t;
typedef __int64 int_least64_t;
typedef unsigned __int8 uint_least8_t;
typedef unsigned __int16 uint_least16_t;
typedef unsigned __int32 uint_least32_t;
typedef unsigned __int64 uint_least64_t;
typedef __int8 int_fast8_t;
typedef __int16 int_fast16_t;
typedef __int32 int_fast32_t;
typedef __int64 int_fast64_t;
typedef unsigned __int8 uint_fast8_t;
typedef unsigned __int16 uint_fast16_t;
typedef unsigned __int32 uint_fast32_t;
typedef unsigned __int64 uint_fast64_t;
typedef __int64 intmax_t;
typedef unsigned __int64 uintmax_t;
# else
# include
# endif
# if _MSC_VER < 1800 /* MSVC < 2013 */
# ifndef __cplusplus
typedef unsigned char _Bool;
# endif
# endif
#else
# include
# if (defined (__SVR4) && defined (__sun)) || defined(_AIX) || defined(__hpux)
# include
# endif
#endif
)
r r r5 r r r r objectr r! r r r r
s