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 s� t |tj�st�|jr(| �d||� d S | j}t|j�}g }t |j�D ].\}}d}t |tj
�rbd}|� d||f � qFd| � � 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
��r@d}nt |j
tj��sVd}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);�})�
isinstancer �FunctionPtrType�AssertionError�ellipsis�_generate_gen_constr" �len�args� enumerate�
StructOrUnion�append�zip�result�insertrW � void_typer r �abiZVoidType)r
rA rM r) ZnumargsZargnames�irX Z indirectionZarglistZtpresult� wrappernamerm ZfuncdeclZ result_coder rZ r �_generate_gen_function_decl� sL
� z*VGenericEngine._generate_gen_function_declc C sj t |tj�st�|jr*| �d|||�}�n g }|}tdd� |jD ��sVt |jtj ��r
g }t
|j�D ]6\} }
t |
tj �r�t� |
�}
|� | |
f� |� |
� qd|j} t | tj �r�| j
d kr�td| �� 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 )r_ r rg )r? �typr r r � <genexpr>� s z6VGenericEngine._loaded_gen_function.<locals>.<genexpr>z*'%s' is used as result type, but is opaquer rj r\ )r_ r r` ra rb �_load_constant�anyre rj rg rf � PointerTyperh Zfldtypes� TypeError� _get_c_namerk rl �tupler
�_get_cached_btype�
load_function�_make_struct_wrapper�setattrrX r- )r
rA rM r4 r6 Z newfunctionZ indirections�base_tpZ
indirect_argsrn rq Zindirect_result�BFuncro r r r �_loaded_gen_function� sL �
��
�
� z#VGenericEngine._loaded_gen_functionc sP | j j�| j �|�� �dkr4| j �� ��fdd�}n� ���fdd�}||_|S )Nrj c s �� � �}�|f| �� |d S �Nr )�new)re �res)�BTyper
�oldfuncr r �newfunc� s
z4VGenericEngine._make_struct_wrapper.<locals>.newfuncc s6 | d �� �� � | � �f | �d d � } �| � S )Nr )Znewp)re )r� r> rn r� r r r� � s .)r
r7 ry Z_cffi_base_type)r
r� rn rA r} r� r )r� r> r
rn r� r r{ � s z#VGenericEngine._make_struct_wrapperc C s ||j kst�| �|d|� d S �N�struct�rM ra � _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 ||j kst�| �|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" �
enumfieldsr_ r �
PrimitiveType�is_integer_typerW r �strr rh � ArrayType�lengthrw )
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� }|� � t
|
�t
|� kr�t
|j �ks�n t �|
|| | f|_ n d||f �
� }||f| j|< d S )Nr� zintptr_t(*)(intptr_t)r r � r$ r� )r� r
�_typeof_lockedrz rh r_ r rg �partialZ
force_flattenrd ra Z 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�|� || jk�rdd� }| j}|�|�}| j�|�\}}||d |�|�d� ||d |�|�d� d}|�� D ]v\}} }
} |
dkr�q�||| |�||�d|f � ||d dkr�|�| �} |||d |�| �d |f � |d7 }q�|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.<locals>.checkr zwrong total sizer zwrong total alignmentr� zwrong offset for field %rzwrong size for field %r) r� r
ry r �pop�sizeofZalignofr� Zoffsetofrd ra )
r
rA r� r
ZBStructr� r� rn 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 )r_ 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� )r_ 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 )r_ r r� �_loaded_gen_enumr� rT r r r �_loaded_gen_anonymous� s z$VGenericEngine._loaded_gen_anonymousN�constc
C sF | j }d||f }| j�|� |d k rl|s.t�|dks:t�|d| � |d� | �||� |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 } |dk�rt|tj��rd
} d }||�d| |f |�� |d� |d||f � |d� |� d S )Nz _cffi_%s_%sr� �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 zconst *z %s%s(void)z return (%s%s);) r" r rh ra �_check_int_constant_valuer_ r rg rW )
r
�is_intrM rA �category� check_valuer) �funcnameZ ampersandZextrar r r rc � s@
z"VGenericEngine._generate_gen_constc C s& t |tj�o|�� }| �|||� d S r )r_ r r� r� rc )r
rA rM r� r r r �_generate_gen_constant_decl� s z*VGenericEngine._generate_gen_constant_declc C s d| }|d k r*|st �| �||� |}n�|r�| j�d�d }| j�d�d } |�| |�}
| j�|�} |
| �} t| d �}|dk r�| s�| j�d�d }
|dd| j�|
� > 7 }nd|d ks�t �d}t|t j
�r�d | }| j�|� ||��d } |�| |�}
|
� }t|t j
��r|d }|S )
Nz_cffi_const_%sz
long long*r zint(*)(long long*)z long longr � z (*)(void)rV ) ra �_load_known_int_constantr
r� rz r� �intr� r_ r rg rW )r
r� rA rM r4 r� r� �valuer� r~ r� �p�negativeZ BLongLongZ fntypeextrar r r rs � s2 z VGenericEngine._load_constantc C sD t |tj�o|�� }| �||||�}t|||� t|�j�|� d S r ) r_ r r� r� rs r| rX r- rh )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� rz 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� � enumeratorsrc r� r rh r" ri �
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)rs )r? r� �r4 r
rA r r rB s �z4VGenericEngine._loading_gen_enum.<locals>.<listcomp>T)r� r� rx 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 )ri r� r� r| rX r- rh )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� )rc )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� )rs r| rX r- rh )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� )
r_ r r� r� r" r rh ru �itemrc )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)r_ r r� r� r
r� rz ry r� �divmodr� r Zresolve_lengthru rs �castr| rX r- rh 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_typedefrp Z_loading_gen_functionr r{ r� r� r� r� r� r� r� r� r� r� r� r� rc r� Z_loading_gen_constantrs 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