3
M��Z4h � @ sH d dl Z d dlZd dlZddlmZ ddlmZ G dd� de�ZdZ dS )� N� )�model)�VerificationErrorc @ s� e Zd ZdZdZdd� Zdd� Zdd� Zd d
� ZdZd d
�Z dd� Z
d[dd�Z 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]dAdB�Z'dCdD� Z(dEdF� Z)dGdH� Z*dIdJ� Z+d^dLdM�Z,d_dNdO�Z-dPdQ� Z.dRdS� Z/eZ0dTdU� Z1dVdW� Z2eZ3dXdY� Z4d;S )`�VGenericEngine�gFc C s || _ |j| _g | _i | _d S )N)�verifier�ffi�export_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)�sys�path�os�join�isfile)r Z module_namer Z so_suffixesZ so_suffix�basename�dirname�filenamer 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_f�write)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 �prnt�prefix�modnamer r r
�write_source_to_f+ s
z VGenericEngine.write_source_to_fr c sj �j j}tjjtj�jj�}|j||���j �d� G � ��fdd�dt
j �� � d�}�j �d|d� |S )NZloadingc s& e Zd Z�Z�jZg Z� fdd�ZdS )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�_load�types�
ModuleType)r �flags�backendr r0 r )r* r/ r r
r3 A s zVGenericEngine.load_libraryc C s$ dd� | j jjj� D �}|j� |S )Nc S s g | ]\}\}}||f�qS r r )�.0�key�tpZqualr r r
�
^ s z4VGenericEngine._get_declarations..)r Z_parserZ
_declarations�items�sort)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? �split�getattr�AttributeErrorr � Exceptionr �attach_exception_info)r � step_name�namer; �kind�realname�method�er 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 s� t |tj�st�|jr(| 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);�})�
isinstancer �FunctionPtrType�AssertionError�ellipsis�_generate_gen_constr �len�args� enumerate�
StructOrUnion�append�zip�result�insertrO � void_typer r �abiZVoidType)r r; rG r% ZnumargsZargnames�irP Z indirectionZarglistZtpresult� wrappernamerd ZfuncdeclZ result_coder )rR r
�_generate_gen_function_decl� sJ
z*VGenericEngine._generate_gen_function_declc C sr t |tj�st�|jr*| jd|||�}�n(g }|}tdd� |jD ��sVt |jtj ��rg }xDt
|j�D ]6\} }
t |
tj �r�tj |
�}
|j | |
f� |j |
� qfW |j} t | tj �r�| j
d kr�td| j� f ��tj | �} |jd| � |jdd| f� tj} tjt|�| |j�}| jj|�} d| }
|j| |
�}x"|D ]\} }
| j|| |
|�}�q4W t|||� t|�jj |� d S )NFc s s | ]}t |tj�V qd S )N)rV r r^ )r9 �typr r r
� � s z6VGenericEngine._loaded_gen_function..z*'%s' is used as result type, but is opaquer ra z
_cffi_f_%s)rV r rW rX rY �_load_constant�anyr\ ra r^ r] � PointerTyper_ Zfldtypes� TypeError� _get_c_namerb rc �tupler �_get_cached_btype�
load_function�_make_struct_wrapper�setattrrP r) )r r; rG r/ r0 Z newfunctionZ indirections�base_tpZ
indirect_argsre rh Zindirect_result�BFuncrf r r r
�_loaded_gen_function� s@
z#VGenericEngine._loaded_gen_functionc sP | j j�| j j|�� �dkr4| j �� ��fdd�}n� ���fdd�}||_|S )Nra c s �j � �}�|f| �� |d S )Nr )�new)r\ �res)�BTyper �oldfuncr r
�newfunc� s
z4VGenericEngine._make_struct_wrapper..newfuncc s6 | d �� �j � | � �f | �d d � } �| � S )Nr )Znewp)r\ )ry r8 re rz r r
r{ � s .)r r1 rp Z_cffi_base_type)r rz re r; rt r{ r )ry r8 r re rz r
rr � s z#VGenericEngine._make_struct_wrapperc C s ||j kst�| j|d|� d S )N�struct)rG rX � _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 )Nr| )�_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 kst�| j|d|� d S )N�union)rG rX 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| tj�r�| 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\}} }
} |
dk�rv�q\|d||f � t| tj
��r�| jd k�r�|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);)�fldnames�stripr �
enumfieldsrV r �
PrimitiveType�is_integer_typerO r �strr r_ � ArrayType�lengthrn )
r r; r& rG Z
checkfuncname�layoutfuncname�cnamer% �fname�ftype�fbitsize�fqualrK r r r
r} sT
&
z-VGenericEngine._generate_struct_or_union_declc C s
|j d krd S d||f }| jjd�d }|j||�}g }d} x(|| �}
|
dk rRP |j|
� | d7 } q@W t|tj�r�|jr�|d } |d } |dd d� }
|dd d� }|j � t
|
�t
|� ko�t
|j �kn s�t �|
|| | 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_lockedrq r_ rV r r^ �partialZ
force_flattenr[ rX Z fixedlayoutr� r
)r r; r& rG r/ r� ru �function�layoutZnum�xZ totalsizeZtotalalignmentZfieldofsZ fieldsizer� r r r
r 7 s.
*z'VGenericEngine._loading_struct_or_unionc
C s |j d krd S | jj|� || jk�rdd� }| j}|j|�}| jj|�\}}||d |j|�d� ||d |j|�d� d}x�|j� D ]v\}} }
} |
dkr�q�||| |j||�d|f � ||d dkr�|j| �} |||d |j| �d |f � |d7 }q�W |t |�k�st
�d S )
Nc S s | |krt d||| f ��d S )Nz'%s (we have %d, but C compiler says %d))r )Z realvalueZ
expectedvalue�msgr 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 rp r
�pop�sizeofZalignofr� Zoffsetofr[ rX )
r r; r� r ZBStructr� r� re r� r� r� r� ZBFieldr r r
r� S s,
z&VGenericEngine._loaded_struct_or_unionc C s. t |tj�r | j||d� n| j|d|� d S )Nr )rV 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 |tj�r | j|||d� n| j|d||� d S )Nr )rV r r� �_loading_gen_enumr )r r; rG r/ r r r
�_loading_gen_anonymous{ s z%VGenericEngine._loading_gen_anonymousc K s. t |tj�r | j|||f|� n
| j|� d S )N)rV r r� �_loaded_gen_enumr� )r r; rG r/ r r r r
�_loaded_gen_anonymous� s z$VGenericEngine._loaded_gen_anonymousN�constc
C sD | j }d||f }| jj|� |d k rl|s.t�|dks:t�|d| � |d� | j||� |d� |d� n�|r�|dks|t�|d| � |d� |d|f � |d |f � |d� n�|d k s�t�|d ks�t�|d
kr�d }nd }d } |dko�t|tj��rd
} d }||jd| |f |�� |d� |d||f � |d� |� d S )Nz _cffi_%s_%sr� zint %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 zconst *z %s%s(void)z return (%s%s);) r r r_ rX �_check_int_constant_valuerV r r^ rO )
r �is_intrG r; �category� check_valuer% �funcnameZ ampersandZextrar r r
rZ � s@
z"VGenericEngine._generate_gen_constc C s& t |tj�o|j� }| j|||� d S )N)rV r r� r� rZ )r r; rG r� r r r
�_generate_gen_constant_decl� s z*VGenericEngine._generate_gen_constant_declc C s d| }|d k r*|st �| j||� |}n�|r�| jjd�d }| jjd�d } |j| |�}
| jj|�} |
| �} t| d �}|dk r�| r�| jjd�d }
|dd| jj|
� > 7 }nd|d ks�t �d}t|t j
�r�d | }| jj|j ||��d } |j| |�}
|
� }t|t j
��r|d }|S )
Nz_cffi_const_%sz
long long*r zint(*)(long long*)z long longr � z (*)(void)rN ) rX �_load_known_int_constantr r� rq rw �intr� rV r r^ rO )r r� r; rG r/ r� r� �valuery ru r� �p�negativeZ BLongLongZ fntypeextrar r r
rj � s2 z VGenericEngine._load_constantc C sD t |tj�o|j� }| j||||�}t|||� t|�jj|� d S )N) rV r r� r� rj rs 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 jd�d }| j jd�d }|j||�}| j j|d�}||�dk rn| j j|�}tjdkrft|d�}t|��d S )Nzchar[]r z
int(*)(char*)� r zutf-8)r ) r r� rq rw �stringr r$ r� r )r r/ r� ry ru 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_funcname�enumc C s� |j r&x|jD ]}| jd|� qW d S | j||�}| jj|� | j}|d| � |d� x&t|j|j�D ]\}}| j ||� qhW |d� |d� |� d S )NTzint %s(char *out_error)rT z return 0;rU )
r� � enumeratorsrZ 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� ��fdd��jD �}t|��_d�_n�j||�}�j� |� d S )Nc s g | ]}�j d �|� ��qS )T)rj )r9 r� )r/ r r; r r
r< s z4VGenericEngine._loading_gen_enum..T)r� r� ro 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|j�D ]$\}}t|||� t|�jj|� qW d S )N)r` r� r� rs 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� )rZ )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� )rj rs 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 |tj�rx|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� )
rV r r� r� r r r_ rl �itemrZ )r r; rG r% r� �tp_ptrr r r
�_generate_gen_variable_decl6 s
z*VGenericEngine._generate_gen_variable_declc s^ t |tj�r�|jdkr�d|f }| jjd�d }|j||�}|� }| jj|j�} t || jj
| ��\}
} | dkr�t 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)rV r r� r� r r� rq rp r� �divmodr� r Zresolve_lengthrl rj �castrs rP r) r_ rO �property)r r; rG r/ r0 r� ru 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_typedefrg Z_loading_gen_functionrv rr r~ r� r� r� r� r� r} r r� r� r� r� rZ r� Z_loading_gen_constantrj 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