U
ل=^4h � @ 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 �/vengine_gen.py�__init__ s zVGenericEngine.__init__c C s |� d| j� d S )Nr )�
setdefaultr )r
�kwdsr r r �patch_extension_kwds s z#VGenericEngine.patch_extension_kwdsc C sT |D ]J}|| }|d kr t j}|D ]*}tj�||�}tj�|�r"| S q"qd S r )�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 r r �r
r r r �
collect_types% s z VGenericEngine.collect_types� c C s | j �|d � d S )N�
)Z_f�write)r
Zwhatr r r �_prnt( s zVGenericEngine._prntc C s` | j }|t� || jj� | �d� tjdkr\tjdkr>d}nd}| j�� }|d||f � d S )NZdeclZwin32�� ZPyInit_Zinitzvoid %s%s(void) { }
) 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}tj�tj�jj�}|�||���� �d� G � ��fdd�dt
j �� � d�}�j �d|d� |S )NZloadingc s& e Zd Z�Z�jZg Z� fdd�ZdS )z/VGenericEngine.load_library.<locals>.FFILibraryc s � j S r )� _cffi_dirr )�
FFILibraryr r �__dir__T s z7VGenericEngine.load_library.<locals>.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 r6 r r3 r r9 A s zVGenericEngine.load_libraryc C s$ dd� | j jj�� D �}|�� |S )Nc S s g | ]\}\}}||f�qS r r )�.0�key�tpZqualr r r �
<listcomp>^ s
z4VGenericEngine._get_declarations.<locals>.<listcomp>)r
Z_parserZ
_declarations�items�sort)r
Zlstr r r �_get_declarations] s
�z VGenericEngine._get_declarationsc C s� | � � D ]�\}}|�dd�\}}zt| d||f �}W n tk
rV td| ��Y nX z|||� W q tk
r� } zt�||� � W 5 d }~X Y qX qd S )N� r z_generate_gen_%s_%sz not implemented in verify(): %r)rE �split�getattr�AttributeErrorr � Exceptionr �attach_exception_info)r
� step_name�namerA �kind�realname�method�er r r r& c s � �
zVGenericEngine._generatec
K s~ | � � D ]p\}}|�dd�\}}t| d||f �}z||||f|� W q tk
rv } zt�| |� � W 5 d } ~ X Y qX qd S )NrF r z
_%s_gen_%s)rE rG rH rJ r rK )
r
r4 rL r rM rA rN rO rP rQ r r r r: r s zVGenericEngine._loadc C s d S r r �r
rA rM r r r �_generate_nothing| s z VGenericEngine._generate_nothingc K s d S r r �r
rA rM r4 r r r r � _loaded_noop s zVGenericEngine._loaded_noopc st |j r| �d||� d S | j}t|j�}g }t|j�D ].\}}d}t|tj�rRd}|� d||f � q6d| � � fdd�t
|j|�D �} |j }
t|
tj�r�| � d|
�
d � �� tj}
d
�| �p�d } d | } | j� | � |jr�|jd
} nd} d| | | f }
d| � ||
�
|
� �� |d� t|j tj��r0d}nt|j tj��sFd}nd}|d||d
�|�f � |d� |� d S )NFr �*z%sx%dzargument of %sc s g | ]\}}|� d | � ��qS )z %s)�
get_c_name)r? �type�arg��contextr r rB � s �z>VGenericEngine._generate_gen_function_decl.<locals>.<listcomp>r z *rz, Zvoid�
_cffi_f_%srF z %s%s(%s)z result of %s�{z*r = zreturn z %s%s(%s);�})�ellipsis�_generate_gen_constr" �len�args� enumerate�
isinstancer �
StructOrUnion�append�zip�result�insertrW � void_typer r �abiZVoidType)r
rA rM r) ZnumargsZargnames�irX Z indirectionZarglistZtpresult� wrappernamerk ZfuncdeclZ result_coder rZ r �_generate_gen_function_decl� sJ
� z*VGenericEngine._generate_gen_function_declc C sX |j r| �d|||�}�n g }|}tdd� |jD ��sDt|jtj�r�g }t|j�D ]6\} }
t|
tj�r~t� |
�}
|�
| |
f� |�
|
� qR|j} t| tj�r�| j d kr�t d| �
� f ��t� | �} |�d| � |�dd| f� tj} t�t|�| |j �}| j�|�} d| }
|�| |
�}|D ]\} }
| �|| |
|�}�q t|||� t|�j�
|� d S )NFc s s | ]}t |tj�V qd S r )rd r re )r? �typr r r � <genexpr>� s z6VGenericEngine._loaded_gen_function.<locals>.<genexpr>z*'%s' is used as result type, but is opaquer rh r\ )r_ �_load_constant�anyrb rd rh r re rc � PointerTyperf Zfldtypes� TypeError� _get_c_nameri rj ZFunctionPtrType�tupler
�_get_cached_btype�
load_function�_make_struct_wrapper�setattrrX r- )r
rA rM r4 r6 Z newfunctionZ indirections�base_tpZ
indirect_argsrl ro Zindirect_result�BFuncrm r r r �_loaded_gen_function� sJ �
��
�
� z#VGenericEngine._loaded_gen_functionc sP | j j�| j �|�� �dkr4| j �� ��fdd�}n� ���fdd�}||_|S )Nrh c s �� � �}�|f| �� |d S �Nr )�new)rb �res)�BTyper
�oldfuncr r �newfunc� s
z4VGenericEngine._make_struct_wrapper.<locals>.newfuncc s6 | d �� �� � | � �f | �d d � } �| � S )Nr )Znewp)rb )r� r> rl r� r r r� � s .)r
r7 rw Z_cffi_base_type)r
r� rl rA r{ r� r )r� r> r
rl r� r ry � s z#VGenericEngine._make_struct_wrapperc C s | � |d|� d S �N�struct�� _generate_struct_or_union_declrR r r r �_generate_gen_struct_decl� s z(VGenericEngine._generate_gen_struct_declc C s | � |d||� d S r� ��_loading_struct_or_union�r
rA rM r4 r r r �_loading_gen_struct� s z"VGenericEngine._loading_gen_structc K s | � |� d S r ��_loaded_struct_or_unionrT r r r �_loaded_gen_struct� s z!VGenericEngine._loaded_gen_structc C s | � |d|� d S �N�unionr� rR r r r �_generate_gen_union_decl� s z'VGenericEngine._generate_gen_union_declc C s | � |d||� d S r� r� r� r r r �_loading_gen_union s z!VGenericEngine._loading_gen_unionc K s | � |� d S r r� rT 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}|d||f � |d� |d� |d� |�� D ]�\}} }
} t| tj�r�| �� s�|
dkr�|d | � qlz$|d
| jd d | | d
�|f � W ql t k
r� } z|dt
| � � W 5 d } ~ X Y qlX ql|d� | j � |� |d|f � |d� |d| � |d� |d| � |d� |�� D ]h\}} }
} |
dk�rp�qV|d||f � t| tj
��r�| jd k�r�|d| �� � n|d||f � �qV|d� |d� |d� |d� |d |f � |d� |� d S ) Nz_cffi_check_%s_%s�_cffi_layout_%s_%s�%s %szstatic void %s(%s *p)r] 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 */r^ 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" �
enumfieldsrd r �
PrimitiveType�is_integer_typerW r �strr rf � ArrayType�lengthru )
r
rA r* rM Z
checkfuncname�layoutfuncname�cnamer) �fname�ftype�fbitsize�fqualrQ r r r r� s\
��� $
z-VGenericEngine._generate_struct_or_union_declc C s� |j d krd S d||f }| j�d�d }|�||�}g }d} || �}
|
dk rPqd|�|
� | d7 } q>t|tj�r�|jr�|d } |d } |dd d� }
|dd d� }|� � |
|| | f|_
n d||f � � }||f| j |< d S )Nr� zintptr_t(*)(intptr_t)r r � r$ r� )
r� r
�_typeof_lockedrx rf rd r re �partialZ
force_flattenZ fixedlayoutr� r )r
rA r* rM r4 r� r| �function�layoutZnum�xZ totalsizeZtotalalignmentZfieldofsZ fieldsizer� r r r r� 7 s*
z'VGenericEngine._loading_struct_or_unionc
C s |j d krd S | j�|� || jkr�dd� }| j}|�|�}| j�|�\}}||d |�|�d� ||d |�|�d� d}|�� D ]v\}} }
} |
dkr�q�||| |�||�d|f � ||d dkr�|�| �} |||d |�| �d |f � |d7 }q�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.<locals>.checkr zwrong total sizer zwrong total alignmentr� zwrong offset for field %rzwrong size for field %r) r� r
rw r �pop�sizeofZalignofr� Zoffsetof)
r
rA r� r
ZBStructr� r� rl r� r� r� r� ZBFieldr r r r� S s0
�
�
z&VGenericEngine._loaded_struct_or_unionc C s. t |tj�r | �||d� n| �|d|� d S �Nr )rd r �EnumType�_generate_gen_enum_declr� rR r r r � _generate_gen_anonymous_declu s z+VGenericEngine._generate_gen_anonymous_declc C s2 t |tj�r | �|||d� n| �|d||� d S r� )rd r r� �_loading_gen_enumr� r� r r r �_loading_gen_anonymous{ s z%VGenericEngine._loading_gen_anonymousc K s. t |tj�r | j|||f|� n
| �|� d S r )rd r r� �_loaded_gen_enumr� rT r r r �_loaded_gen_anonymous� s z$VGenericEngine._loaded_gen_anonymousN�constc
C s
| j }d||f }| j�|� |d k rX|d| � |d� | �||� |d� |d� n�|r�|d| � |d� |d|f � |d|f � |d� nj|d kr�d
}nd }d } |d kr�t|tj�r�d
} d
}||�d| |f |�� |d� |d||f � |d� |� d S )Nz _cffi_%s_%s�int %s(char *out_error)r] � return 0;r^ 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 rf �_check_int_constant_valuerd r re rW )
r
�is_intrM rA �category� check_valuer) �funcnameZ ampersandZextrar r r r` � s6
z"VGenericEngine._generate_gen_constc C s& t |tj�o|�� }| �|||� d S r )rd r r� r� r` )r
rA rM r� r r r �_generate_gen_constant_decl� s z*VGenericEngine._generate_gen_constant_declc C s d| }|d k r"| � ||� |}n�|r�| j�d�d }| j�d�d } |�| |�}
| j�|�} |
| �} t| d �}|dk r�| s�| j�d�d }
|dd| j�|
� > 7 }nVd}t|tj �r�d | }| j�|�
||��d } |�| |�}
|
� }t|tj �r�|d }|S )
Nz_cffi_const_%sz
long long*r zint(*)(long long*)z long longr � z (*)(void)rV ) �_load_known_int_constantr
r� rx r �intr� rd r re rW )r
r� rA rM r4 r� r� �valuer� r| r� �p�negativeZ BLongLongZ fntypeextrar r r rq � s. z VGenericEngine._load_constantc C sD t |tj�o|�� }| �||||�}t|||� t|�j�|� d S r ) rd r r� r� rq rz rX r- rf )r
rA rM r4 r6 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
rM r� r) r r r r� � s. �
� �z(VGenericEngine._check_int_constant_valuec C sr | j �d�d }| j �d�d }|�||�}| j �|d�}||�dk rn| j �|�}tjdkrft|d�}t|��d S )Nzchar[]r z
int(*)(char*)� r# zutf-8) r
r� rx r �stringr r( r� r )r
r4 r� r� r| r� r� �errorr r r r� � s
z'VGenericEngine._load_known_int_constantc C s |� dd�}d||f S )N�$Z___D_z
_cffi_e_%s_%s)�replace)r
r* rM r r r �_enum_funcname� s z VGenericEngine._enum_funcname�enumc C s� |j r"|jD ]}| �d|� q d S | �||�}| j�|� | j}|d| � |d� t|j|j�D ]\}}| � ||� qb|d� |d� |� d S )NTr� r] r� r^ )
r� � enumeratorsr` r� r rf r" rg �
enumvaluesr� )r
rA rM r* �
enumeratorr� r) � enumvaluer r r r� � s
z&VGenericEngine._generate_gen_enum_declc sL �j r0� ��fdd��jD �}t|��_d�_n��||�}��� |� d S )Nc s g | ]}�� d �|� ��qS )T)rq )r? r� �r4 r
rA r r rB s �z4VGenericEngine._loading_gen_enum.<locals>.<listcomp>T)r� r� rv r� Zpartial_resolvedr� r� )r
rA rM r4 r* r� r� r r� r r� s �
z VGenericEngine._loading_gen_enumc C s8 t |j|j�D ]$\}}t|||� t|�j�|� qd S r )rg r� r� rz rX r- rf )r
rA rM r4 r6 r� r� r r r r� s z VGenericEngine._loaded_gen_enumc C s& |dkrd }n|}| j d||d� d S �N�...T)r� )r` )r
rA rM r� r r r �_generate_gen_macro_decl s z'VGenericEngine._generate_gen_macro_declc C sF |dkrd }n|}| j d||||d�}t|||� t|�j�|� d S r� )rq rz rX r- rf )r
rA rM r4 r6 r� r� r r r �_loaded_gen_macro) s � z VGenericEngine._loaded_gen_macroc C s� t |tj�rx|jdkr\| j}d|f }| j�|� |d| � |d� |d|f � |d� t�|j�}| � d||� n t�|�}| j d||dd � d S )
Nr� �_cffi_sizeof_%szsize_t %s(void)r] z return sizeof(%s);r^ Fr� )r� )
rd r r� r� r" r rf rs �itemr` )r
rA rM r) r� �tp_ptrr r r �_generate_gen_variable_decl6 s
z*VGenericEngine._generate_gen_variable_declc s^ t |tj�r�|jdkr�d|f }| j�d�d }|�||�}|� }| j�|j�} t || j�
| ��\}
} | dkr�t d||jf ��|� |
�}t�
|j�} | �d| ||�}
|jd k r�| j�|�}| j�||
�}
t|||
� t|�j�|� d S d| }| j�|�d|��d }|�||�}|� � � fd d
�}� fd d �}tt|�|t||�� t|�j�|� d S )
Nr� r� zsize_t(*)(void)r z/bad size: %r does not seem to be an array of %sFz _cffi_var_%sz
*(*)(void)c s � d S r~ r r5 �Zptrr r �getterg s z3VGenericEngine._loaded_gen_variable.<locals>.getterc s |� d< d S r~ r )r6 r� r� r r �setteri s z3VGenericEngine._loaded_gen_variable.<locals>.setter)rd r r� r� r
r� rx rw r� �divmodr� r Zresolve_lengthrs rq �castrz rX r- rf rW �property)r
rA rM r4 r6 r� r| 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� )5r0 r1 r2 Z
_class_keyZ_gen_python_moduler r r r r" r, r9 rE r& r: rS rU Z_generate_gen_typedef_declZ_loading_gen_typedefZ_loaded_gen_typedefrn Z_loading_gen_functionr} ry r� r� r� r� r� r� r� r� r� r� r� r� r` r� Z_loading_gen_constantrq r� r� r� r� r� r� r� r� Z_loading_gen_macror� r� Z_loading_gen_variabler� r r r r r sh
-$1 " �
%
r ad
#include <stdio.h>
#include <stddef.h>
#include <stdarg.h>
#include <errno.h>
#include <sys/types.h> /* 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 <malloc.h> /* 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 <stdint.h>
# endif
# if _MSC_VER < 1800 /* MSVC < 2013 */
# ifndef __cplusplus
typedef unsigned char _Bool;
# endif
# endif
#else
# include <stdint.h>
# if (defined (__SVR4) && defined (__sun)) || defined(_AIX) || defined(__hpux)
# include <alloca.h>
# endif
#endif
)
r r r; r r r� r �objectr r% r r r r �<module> s g