3 M��Z��@sddlZddlZddlZddlmZmZddlmZddlTdZ dZ dZGd d �d �ZGdd�d�Z Gd d�d�ZGdd�d�ZGdd�d�ZGdd�d�Zejd0kr�ejZnGdd�dej�Zdd�Zd1dd�Zd2dd�Zdd �Zd!d"�Zd#d$�Zd%d&�Zd'd(�Zd3d,d-�Zd.d/�ZdS)4�N�)�ffiplatform�model)�VerificationError)�*i&i'i(c@s&eZdZd dd�Zdd�Zdd�ZdS) � GlobalExprrcCs"||_||_||_||_||_dS)N)�name�address�type_op�size�check_value)�selfrr r rr�r�"/usr/lib64/python3.6/recompiler.py�__init__s zGlobalExpr.__init__cCsd|j|j|jj�|jfS)Nz' { "%s", (void *)%s, %s, (void *)%s },)rr r � as_c_exprr)r rrrrszGlobalExpr.as_c_exprcCsd|jj�|j|jfS)Nz b'%s%s',%d)r �as_python_bytesrr)r rrr�as_python_exprszGlobalExpr.as_python_exprN)rr)�__name__� __module__�__qualname__rrrrrrrrs rc@s,eZdZdd�Zdd�Zdd�Zdd�Zd S) � FieldExprcCs"||_||_||_||_||_dS)N)r�field_offset� field_size�fbitsize� field_type_op)r rrrrrrrrrs zFieldExpr.__init__cCs>dt|j�}d|j|jfd||jfd||jj�fS)N� z { "%s", %s, z %s %s, z %s %s },)�lenrrrrr)r Zspacesrrrr#szFieldExpr.as_c_exprcCst�dS)N)�NotImplementedError)r rrrr)szFieldExpr.as_python_exprcCsD|jjtkrd}n|jjtkr*t|j�}nt�d|jj�||jfS)N�z b'%s%s%s') r�op�OP_NOOP�OP_BITFIELD�format_four_bytesrrrr)r Z size_exprrrr�as_field_python_expr,s zFieldExpr.as_field_python_exprN)rrrrrrr$rrrrrsrc@s$eZdZdd�Zdd�Zdd�ZdS)�StructUnionExprc Cs4||_||_||_||_||_||_||_||_dS)N)r� type_index�flagsr� alignment�comment�first_field_index�c_fields) r rr&r'rr(r)r*r+rrrr8szStructUnionExpr.__init__cCsPd|j|j|jfd|j|jfd|jt|j�f|jrFd|jnddS)Nz { "%s", %d, %s,z %s, %s, z%d, %d z /* %s */ rz},) rr&r'rr(r*rr+r))r rrrrCszStructUnionExpr.as_c_exprcCs>t|jt�}dd�|jD�}dt|j�t|�|jdj|�fS)NcSsg|]}|j��qSr)r$)�.0Zc_fieldrrr� Lsz2StructUnionExpr.as_python_expr..z(b'%s%s%s',%s)�,)�evalr'ZG_FLAGSr+r#r&r�join)r r'Zfields_exprrrrrJs zStructUnionExpr.as_python_exprN)rrrrrrrrrrr%7sr%c@s$eZdZdd�Zdd�Zdd�ZdS)�EnumExprcCs"||_||_||_||_||_dS)N)rr&r�signed�allenums)r rr&rr2r3rrrrUs zEnumExpr.__init__cCsd|j|j|j|j|jfS)Nz1 { "%s", %d, _cffi_prim_int(%s, %s), "%s" },)rr&rr2r3)r rrrr\szEnumExpr.as_c_exprcCsNdtdtd td tdtdtd tdti|j|j f}dt |j�t |�|j|j fS)Nrr��zb'%s%s%s\x00%s')rr)rr)r4r)r4r)r5r)r5r)r6r)r6r)Z PRIM_UINT8Z PRIM_INT8ZPRIM_UINT16Z PRIM_INT16ZPRIM_UINT32Z PRIM_INT32ZPRIM_UINT64Z PRIM_INT64rr2r#r&rr3)r � prim_indexrrrras zEnumExpr.as_python_exprN)rrrrrrrrrrr1Tsr1c@s$eZdZdd�Zdd�Zdd�ZdS)�TypenameExprcCs||_||_dS)N)rr&)r rr&rrrrmszTypenameExpr.__init__cCsd|j|jfS)Nz { "%s", %d },)rr&)r rrrrqszTypenameExpr.as_c_exprcCsdt|j�|jfS)Nzb'%s%s')r#r&r)r rrrrtszTypenameExpr.as_python_exprN)rrrrrrrrrrr8lsr8c@s�eZdZdZd�dd�Zdd�Zdd�Zd d �Zdd�Zd ddddgZ dd�Z d�dd�Zdd�Zdd�Z dd�Zdd�Zdd �Zd!d"�Zd#d$�Zd%d&�Zd'd(�Zd)d*�Zd+d,�Zd-d.�Zd/d0�Zd1d2�Zd3d4�Zd5d6�Zd7d8�Zd9d:�Zd;d<�Zd=d>�Zd?d@�Z d�dBdC�Z!dDdE�Z"dFdG�Z#dHdI�Z$e$Z%dJdK�Z&dLdM�Z'e'Z(dNdO�Z)e)Z*dPdQ�Z+dRdS�Z,dTdU�Z-d�dWdX�Z.dYdZ�Z/d[d\�Z0d]d^�Z1d_d`�Z2d�dadb�Z3dcdd�Z4dedf�Z5dgdh�Z6didj�Z7dkdl�Z8dmdn�Z9dodp�Z:dqdr�Z;dsdt�ZZ?dwdx�Z@dydz�ZAd{d|�ZBd}d~�ZCdd��ZDeDZEZFd�d��ZGd�d��ZHd�d��ZId�d��ZJd�d��ZKd�d��ZLd�d��ZMeMZNeMZOd�d��ZPd�d��ZQd�d��ZReRZSd�d��ZTdAS)�� RecompilerrFcCs||_||_||_t|_dS)N)�ffi�module_name�target_is_python�VERSION_BASE�_version)r r:r;r<rrrr~szRecompiler.__init__cCst|j|�|_dS)N)�maxr>)r Zverrrr� needs_version�szRecompiler.needs_versionc CsPi|_|jd�t|jtd�}g|_x�|D]�}|jr*|j|dksFt�t|j�|j|<|jj|�xZ|j D]P}t |tjtj tjtjtjf�s�t�|j|dkr�t|j�|j|<|jj|�qjW|jjd�q*WxZ|D]R}|jo�|j|dkr�t|j�|j|<|jj|�|jr�|jdk r�|jjd�q�Wd|jj�k�s>t�i|_i|_x@|D]8}t |tj��rnd|j|<nt |tj��rPd|j|<�qPWx.tt|jdd�d��D]\}}||j|<�q�Wx.tt|jdd�d��D]\}}||j|<�q�Wx0|D](}t|d|jj�}|||j|��q�Wx |jD]}t |t��s&t��q&Wt|j�|_dS) NZcollecttype)�keyZEND�LENcSs|jS)N)r)�tprrr��sz/Recompiler.collect_type_table..cSs|jS)N)r)rCrrrrD�sZ_emit_bytecode_)� _typesdict� _generate�sorted�str� cffi_typesZis_raw_function�AssertionErrorr�append�args� isinstancer�VoidType�BasePrimitiveType�PointerType�StructOrUnionOrEnum�FunctionPtrTypeZ is_array_type�length�values�_struct_unions�_enums� StructOrUnion�EnumType� enumerate�getattr� __class__r�CffiOp�tuple)r Z all_declsrC�tp1�i�methodr rrr�collect_type_table�sZ zRecompiler.collect_type_tablecCs�t|tj�s2t|t�r.x|D]}|j|�qWdS||jkr�d|j|<t|tj�rb|j|j��ntt|tj�r�|j dk r�||j jjkr�xN|j �D] \}}}}|j|j|||��q�Wn x|j�D]\}}|j|�q�WdS)N)rMrZBaseTypeByIdentityr]�_do_collect_typerErR�as_raw_functionrW�fldtypesr:�_parser�_included_declarations� enumfields�_field_typeZ _get_items)r rC�xZname1r^�_rrrrb�s zRecompiler._do_collect_typec Cs�|jjjj�}x�t|�D]�\}\}}|jdd�\}}yt|d||f�}Wn tk rjtd|��YnXy||_ |||�Wqt k r�} ztj| |��WYdd} ~ XqXqWdS)Nrrz_generate_cpy_%s_%sz"not implemented in recompile(): %r) r:reZ _declarations�itemsrG�splitrZ�AttributeErrorr�_current_quals� ExceptionrZattach_exception_info) r � step_name�lstrrC�qualsZkindZrealnamer`�errrrF�szRecompiler._generate�global�field�struct_union�enum�typenamecCs:i|_x|jD]}g|j|<qWt�|_|jd�|j�x>|jD]4}|j|}|dkrh|jdd�d�t|�|j|<qBW|jd}x:|jj �D],\}}|t |�ks�t�||j|jks�t�q�Wt |�t |j�ks�t�|jd}x<|j j �D].\}}|t |�k�st�||j|jks�t�q�Wt |�t |j �k�s6t�dS)NZctxrucSs|jS)N)r)�entryrrrrD�sz0Recompiler.collect_step_tables..)rArvrw)�_lsts� ALL_STEPS�set�_seen_struct_unionsrF�_add_missing_struct_unions�sortr]rUrkrrJrrV)r rprqrCr_rrr�collect_step_tables�s* zRecompiler.collect_step_tablesrcCs|jj|d�dS)N� )�_f�write)r Zwhatrrr�_prntszRecompiler._prntcCs:|jr|dkst�|j|�n|dk s*t�|j||�dS)N)r<rJ�write_py_source_to_f�write_c_source_to_f)r �f�preamblerrr�write_source_to_fs zRecompiler.write_source_to_fcCs0ttjjtjjt�|�d�}|j�}|j�|S)N�r)�open�os�pathr0�dirname�__file__� readlines�close)r �filename�g�linesrrr�_rel_readlinesszRecompiler._rel_readlinescCsz||_|j}|jjdk r |d�|jd�}|jd�}|jd�|||d�<|dj|��|jjd�dI}|jjdk �r0|d|jf�|d �|j |jj�|d �|d�|d|f�|d �|d|f�|d�|d|f�|d�|jd�}|jd�}|jd�|||d�<|dj|��|j t�|d�|�||�|�|d�|�|d�tdd�|j j�D��}xLt|j�D]>\}}d} ||k�r�d||j�} |d||j�| f��q�W|j�s�|d�|d�|�t�|_|jd�i} xn|jD]d}|j|}t|�| |<| |dk�r|d||f�x|D]} || j��qBW|d�|��qW|jj�r|d �xr|jjD]f}y|jdd!�\}}Wn(tk �r�td"|j|f��YnX|dk�r�td#��|d$|f��q�W|d%�|d�|�|d&�|d'�x8|jD].}| |dk�r@|d(|�n|d)|��q Wx.|jD]$}|d*k�rZ|d+| ||f��qZW|jj�r�|d,�n|d-�|d.t|j�f�d}|j�r�|dO}|d/|�|d�|�|d0�|d1�|d�|�|d�|d2�|d3|f�|d4�|j�rD|d5�|d6�|d7�|d8|j�|d9�|d:�|d;�|d�|d<�|d=�|d>�|d?�|d@|f�|dA�|dB|f�|dC�|dC�|d �|d2�|dD|f�|d4�|dE|j|jf�|d<�|d�|d2�|dF|f�|d4�|dG|j|jf�|d<�|d�|�|d0�|dH�|d�d|_dS)JNz#define _CFFI_USE_EMBEDDINGz_cffi_include.hz#include "parse_c_type.h" zparse_c_type.hrr�.z#define _CFFI_MODULE_NAME "%s"z1static const char _CFFI_PYTHON_STARTUP_CODE[] = {z0 };z#ifdef PYPY_VERSIONz5# define _CFFI_PYTHON_STARTUP_FUNC _cffi_pypyinit_%sz#elif PY_MAJOR_VERSION >= 3z-# define _CFFI_PYTHON_STARTUP_FUNC PyInit_%sz#elsez*# define _CFFI_PYTHON_STARTUP_FUNC init%sz#endifz_embedding.hz#include "_cffi_errors.h" z_cffi_errors.hz>/************************************************************/zstatic void *_cffi_types[] = {cSsg|]\}}||f�qSrr)r,rCr_rrrr-Gsz2Recompiler.write_c_source_to_f..z // z/* %2d */ %s,%sz 0z};�declrz.static const struct _cffi_%s_s _cffi_%ss[] = {z.static const char * const _cffi_includes[] = {r4zQffi object %r includes %r, but the latter has not been prepared with set_source()zMnot implemented yet: ffi.include() of a Python-based ffi inside a C-based ffiz "%s",z NULLz?static const struct _cffi_type_context_s _cffi_type_context = {z _cffi_types,z _cffi_%ss,z NULL, /* no %ss */ruz %d, /* num_%ss */z _cffi_includes,z NULL, /* no includes */z %d, /* num_types */z %d, /* flags */z#ifdef __GNUC__z?# pragma GCC visibility push(default) /* for -fvisibility= */ZPyMODINIT_FUNCz"_cffi_pypyinit_%s(const void *p[])�{z% if (((intptr_t)p[0]) >= 0x0A03) {zQ _cffi_call_python_org = (void(*)(struct _cffi_externpy_s *, char *))p[1];z }z p[0] = (const void *)0x%x;z p[1] = &_cffi_type_context;z#if PY_MAJOR_VERSION >= 3z return NULL;�}z# ifdef _MSC_VERz PyMODINIT_FUNCz# if PY_MAJOR_VERSION >= 3z% PyInit_%s(void) { return NULL; }z# elsez init%s(void) { }z# endifzPyInit_%s(void)z5 return _cffi_init("%s", 0x%x, &_cffi_type_context);zinit%s(void)z. _cffi_init("%s", 0x%x, &_cffi_type_context);z# pragma GCC visibility pop��)r�r�r:� _embeddingr��indexr0r;rl�_print_string_literal_in_arrayr@�VERSION_EMBEDDED�dictrErkrYrI�_get_c_namerr|�_seen_constantsrFr{rzr�_included_ffis�_assigned_sourcermr� _num_externpyr>)r r�r��prntr�r_Zbase_module_name�typeindex2typer r)Znumsrprqry�ffi_to_include�included_module_name�included_sourcer'rrrr�s zRecompiler.write_c_source_to_fcsbt|t�rd|fSt|ttf�rZ�fdd�|D�}t|�dkrJ|jd�ddj|�fS|j�S)Nzb'%s'csg|]}�j|��qSr)�_to_py)r,�item)r rrr-�sz%Recompiler._to_py..rrz(%s)r.)rMrH�listr]rrKr0r)r riZrepr)r rr��s zRecompiler._to_pyc Cs�||_|j}|d�|d�t|jjp(f�}xzt|�D]n}|jj|}y|jdd�\}}Wn&tk r�td|j|f��YnX|dk r�td��|d||f�q6W|�|d|j f�|d|j f�d|_ t|j�|_d d �|jD�}|d|j dj|��f�td d �|jj�D��} xH|jD]>} |j| }t|�dk�r*| dk�r*|d| |j |�f��q*W|dk�r�|ddjdd �t|�D��f�|d�dS)Nz# auto-generated filezimport _cffi_backendr4zQffi object %r includes %r, but the latter has not been prepared with set_source()zMnot implemented yet: ffi.include() of a C-based ffi inside a Python-based ffizfrom %s import ffi as _ffi%dzffi = _cffi_backend.FFI('%s',z _version = 0x%x,cSsg|]}|j��qSr)r)r,r rrrr-�sz3Recompiler.write_py_source_to_f..z _types = %s,rcSsg|]\}}||f�qSrr)r,rCr_rrrr-�srruz _%ss = %s,z _includes = (%s,),z, cSsg|]}d|�qS)z_ffi%dr)r,r_rrrr-�s�))r�r�rr:r��ranger�rmrr;r>r]rIr�r0r�rErkr{rz)r r�r�Znum_includesr_r�r�r�Z types_lstr�rprqrrrr��sB zRecompiler.write_py_source_to_fcCs |j|S)N)rE)r �typerrr�_gettypenumszRecompiler._gettypenumc Cspd}t|tj�r�|j�r�|j�r<|jdkr.z return NULL;zarg%dzreturn NULLz Py_BEGIN_ALLOW_THREADSz _cffi_restore_errno();cSsg|]}d|�qS)zx%dr)r,r_rrrr-�sz { %s%s(%s); }z _cffi_save_errno();z Py_END_ALLOW_THREADSz (void)self; /* unused */z (void)noarg; /* unused */z return %s;�resultzresult typez Py_INCREF(Py_None);z return Py_None;z#elsecSs t|tj�pt|tj�o|j�S)N)rMrrW� PrimitiveTyper�)r�rrr�need_indirection�sz@Recompiler._generate_cpy_function_decl..need_indirectionFrTz %sx%dz%sx%dz *resultz *result = z%s_cffi_f_%s(%s)z return result;z# define _cffi_f_%s _cffi_d_%sz#endif)r<rJrMrrRr��_generate_cpy_constant_declr�rrLrYrKr�r0�abir�rNr|r�r�r�r��insertZ void_type)r rCrr��numargsZargname� argumentsZcall_argumentsr�r_r��repr_argumentsr��name_and_argumentsZresult_code�argr�r�Zresult_decl�rngr�� differenceZindirectionZ tp_resultrrr�_generate_cpy_function_decl�s� z&Recompiler._generate_cpy_function_declcCs�|jr|jr|j||�dS|j|j�}t|j�}|jrBt}n |dkrPt}n|dkr^t }nt }|jdjt |d|t||�d|d��dS)Nrrrtz _cffi_f_%sz _cffi_d_%s)r)r�r<�_generate_cpy_constant_ctxrErcrrLZOP_DLOPEN_FUNCZOP_CPYTHON_BLTN_NZOP_CPYTHON_BLTN_OZOP_CPYTHON_BLTN_VrzrKrr\)r rCrr&r�Z meth_kindrrr�_generate_cpy_function_ctx#s z%Recompiler._generate_cpy_function_ctxcCsTt|tj�rP|j}|dkr0|jd�}d||f}|j|d||j�}tj||�}|S)Nz...rz_cffi_array_len(((%s)0)->%s)z%s[0])rMrr�rSr�rhr�)r Z tp_structZ field_nameZtp_field� actual_lengthZptr_struct_name�tp_itemrrrrh9s zRecompiler._field_typecCs0|j|�|jr,x|j�D]}|j|�qWdS)N)rbr<�anonymous_struct_fields�_struct_collecttype)r rC�fldtyperrrr�Es zRecompiler._struct_collecttypecCs<|jdkrdS|j}d|f}|d�|d||f�|d�|d�|d�x�|j�D]�\}}}} y||j�sv|dkr�|d|||f�wXx2t|tj�r�|jdks�|jd kr�|j}|d }q�W|d|j dd || d�|f�WqXt k �r} z|dt| ��WYdd} ~ XqXXqXW|d�|d||f�|�dS)Nz_cffi_checkfld_%sZ_CFFI_UNUSED_FNzstatic void %s(%s *p)r�z8 /* only to generate compile-time warnings or errors */z (void)p;rz> (void)((p->%s) | 0); /* check that '%s.%s' is an integer */z...z[0]z { %s = &p->%s; (void)tmp; }z*tmpzfield %r)rrz /* %s */r�z(struct _cffi_align_%s { char x; %s y; };)rdr�rgr�rMrr�rSr�r�rrH)r rCr�r�r�Z checkfuncnameZfnameZftyper�fqualrsrrr�_struct_declLs6 &zRecompiler._struct_declNcCs�|j|}d}g}t|tj�r(|jd�|jdkr@|jd�d}||jjjkr�|dksd||jjjkr�|jdkrpn|j s�t |j��r�n |jd�|jr�|jd�n|jd�d}dj |�p�d }g}|dk�rZ|j} t|j| ��} �x | D�]\}}} }|j|||�}|j|d |j|f�t}| dk�r6t}d| }nL|dk�sZt|tj��r`|jdk�r`d }n"d|dk�rv|jd�n|j|f}|dk�s�| dk�r�d }n,|dk �r�d|j|f}nd|jd�|f}|jt|||| t||j|��q�Wt|jd�}|jdj|�|dk�r$d}d}d}n4|dk �r@d|jf}d}nd|f}d|f}d}nd }d}d}|}|jdjt|j|||||||��|jj |�dS) NZ _CFFI_F_UNIONZ_CFFI_F_OPAQUEZopaqueZ_CFFI_F_CHECK_FIELDSZ_CFFI_F_PACKEDZ_CFFI_F_EXTERNALZexternal�|�0z field '%s.%s'rz %d /* bits */z (size_t)-1zsizeof(((%s)0)->%s)rz"((char *)&((%s)0)->%s) - (char *)0zoffsetof(%s, %s)rruz (size_t)-2r4Zunnamedzsizeof(*(%s)0)z-1 /* unknown alignment */z sizeof(%s)z"offsetof(struct _cffi_align_%s, y)rrv��r�r�)!rErMrZ UnionTyperKrdr:rerf�partial�anyr�Zpackedr0r<r�rgrh�_check_not_opaquerr!r"r�rSr�rr\rrz�extendr%r}r�)r rCr�r�r�r&Zreason_for_not_expandingr'r+Zexpand_anonymous_struct_unionrgZfldnamer�rr�r r�offsetr*Zalignr)rrrr�ms� zRecompiler._struct_ctxcCs>xt|tj�r|j}qWt|tj�r:|jdkr:td|��dS)Nz0%s is of an opaque type (not declared in cdef()))rMrr�r�rWrdr�)r rC�locationrrrr��s zRecompiler._check_not_opaquecCs�t|jj��}|jdd�d�x�|D]�\}}||jkr$|jrJtd|f��|jjd�rx|jdd�j �rx|jdd�}n4|jdkr�|j dkr�d}|j|d�ntd |f��|j|d|�q$WdS) NcSs|dS)Nrr)Ztp_orderrrrrD�sz7Recompiler._add_missing_struct_unions..)rAzDinternal inconsistency: %r is partial but was not seen at this point�$rZ_IO_FILE�FILEzinternal inconsistency: %r) r�rUrkrr}r�rr� startswith�isdigitZ forcenamer�r�)r rqrC�orderr�rrrr~�s z%Recompiler._add_missing_struct_unionscCs|j|�dS)N)r�)r rCrrrr� _generate_cpy_struct_collecttype�sz+Recompiler._generate_cpy_struct_collecttypecCs2|jd�}d|kr"||jdd�fS|d|fSdS)Nrrrj)r�r�)r rCr�rrr� _struct_names�s zRecompiler._struct_namescCs|j|f|j|��dS)N)r�r�)r rCrrrr�_generate_cpy_struct_decl�sz$Recompiler._generate_cpy_struct_declcCs|j|f|j|��dS)N)r�r�)r rCrrrr�_generate_cpy_struct_ctx�sz#Recompiler._generate_cpy_struct_ctxcCs(t|tj�r|j||�n |j|�dS)N)rMrrX�_generate_cpy_enum_collecttyper�)r rCrrrr�#_generate_cpy_anonymous_collecttype�sz.Recompiler._generate_cpy_anonymous_collecttypecCs.t|tj�r|j|�n|j||d|�dS)N�typedef_)rMrrX�_generate_cpy_enum_declr�)r rCrrrr�_generate_cpy_anonymous_decl�sz'Recompiler._generate_cpy_anonymous_declcCs0t|tj�r|j||�n|j||d|�dS)Nr)rMrrX� _enum_ctxr�)r rCrrrr�_generate_cpy_anonymous_ctx�sz&Recompiler._generate_cpy_anonymous_ctx�constcCs||f|jkrtd||f��|jj||f�|j}d||f}|r�|d|�|d�|d|f�|d||f�|dk r�|dkr�d|f}|d |f�|d �|d�|d�n>|dks�t�|d |�|d�|d|jd�|f�|d�|�dS)Nz duplicate declaration of %s '%s'z_cffi_%s_%sz$static int %s(unsigned long long *o)r�z int n = (%s) <= 0;zI *o = (unsigned long long)((%s) | 0); /* check that %s is an integer */rz%dUz" if (!_cffi_check_int(*o, n, %s))z n |= 2;z return n;r�zstatic void %s(char *o)z *(%s)o = %s;r)r�rr�r�rJr�)r �is_intrrC�categoryrr��funcnamerrr�_generate_cpy_consts2 zRecompiler._generate_cpy_constcCs"|j�}|s|jr|j|�dS)N)r�r<rb)r rCrrrrr�"_generate_cpy_constant_collecttype&sz-Recompiler._generate_cpy_constant_collecttypecCs|j�}|j|||�dS)N)r�r )r rCrrrrrr�+sz&Recompiler._generate_cpy_constant_declcCs`|jr|j�rttd�}n$|jr(t}nt}|j|}t||�}|jdjt |d||��dS)Nrrtz_cffi_const_%sr�) r<r�r\�OP_CONSTANT_INTZOP_DLOPEN_CONSTZOP_CONSTANTrErzrKr)r rCrr Z const_kindr&rrrr�/s z%Recompiler._generate_cpy_constant_ctxcCs|j|�dS)N)rb)r rCrrrrr�?sz)Recompiler._generate_cpy_enum_collecttypecCs x|jD]}|jd|�qWdS)NT)�enumeratorsr )r rCr� enumeratorrrrrBsz"Recompiler._generate_cpy_enum_declcCs�|j|}ttd�}|jr"|j�x:t|j|j�D](\}}|jdj t |d|||d��q2W|dk r�d|kr�|jr�d|}d|}n4|j|jg�} |jj | �}tt|jj| d��dk�}d j|j�} |jd j t|j|||| ��dS) Nrrtz_cffi_const_%s)rr�z sizeof(%s)z ((%s)-1) <= 0rr.rwr�r�)rEr\�OP_ENUMr<Zcheck_not_partial�zipr Z enumvaluesrzrKrZbuild_baseinttyper:Zsizeof�int�castr0r1r)r rCr�r&r rZ enumvaluerr2Zbasetpr3rrrrFs" zRecompiler._enum_ctxcCs|j||j��dS)N)rr�)r rCrrrr�_generate_cpy_enum_ctx[sz!Recompiler._generate_cpy_enum_ctxcCsdS)Nr)r rCrrrr�_generate_cpy_macro_collecttypeasz*Recompiler._generate_cpy_macro_collecttypecCs&|dkrd}n|}|jd||d�dS)Nz...T)r)r )r rCrrrrr�_generate_cpy_macro_decldsz#Recompiler._generate_cpy_macro_declcCsT|dkr"|jrtd|f��d}n|}ttd�}|jdjt|d|||d��dS)Nz...zGcannot use the syntax '...' in '#define %s ...' when using the ABI moderrtz_cffi_const_%s)rr�)r<rr\rrzrKr)r rCrrr rrr�_generate_cpy_macro_ctxks z"Recompiler._generate_cpy_macro_ctxcCsFt|tj�rB|j}|dkr$d|f}|j|jd|�}tj||�}|S)Nz...z_cffi_array_len(%s)z%s[0])rMrr�rSr�r�)r rCZglobal_namer�r�rrrr�|s zRecompiler._global_typecCs|j|j||��dS)N)rbr�)r rCrrrr�"_generate_cpy_variable_collecttype�sz-Recompiler._generate_cpy_variable_collecttypecCs�|j}|j||�}t|tj�r4|jdkr4|j}d}nd}d|f}|d|j||jd��|d�|d||f�|d�|�dS) Nr�&z*_cffi_var_%s(void)zstatic )rrr�z return %s(%s);r�) r�r�rMrr�rSr�r�rn)r rCrr�Z ampersandr�rrr�_generate_cpy_variable_decl�s z&Recompiler._generate_cpy_variable_declcCsL|j||�}|j|}|jr"t}nt}|jdjt|d|t||��dS)Nrtz_cffi_var_%s) r�rEr<Z OP_GLOBAL_VARZOP_GLOBAL_VAR_FrzrKrr\)r rCrr&r rrr�_generate_cpy_variable_ctx�s z%Recompiler._generate_cpy_variable_ctxcCst|tj�st�|j|�dS)N)rMrrRrJrb)r rCrrrr�'_generate_cpy_extern_python_collecttype�sz2Recompiler._generate_cpy_extern_python_collecttypecCs|j}t|jtj�rd}nd|}d|jjd|�f}|d|�|d|j||f�|�g}d|}x0t|j�D]"\}} | jd||�} |j | �qtWd j |�}|p�d }d||f}|jdkr�d |}dd�} tt |j�dd�}| |j�r�t|d�}t|jtj��r*d|jjd�||jjd�|f}|d||jj|�f�|d�|d|�|d�xft|j�D]X\}} d|} t| tj��s�| | ��r�d| } tj| �} |d| jd�|d| f��qjW|d|�t|jtj��s�|d|jjd�f�|d�|�|jd7_dS)Nr�zresult of %sz(int)sizeof(%s)rz3static struct _cffi_externpy_s _cffi_externpy__%s =z { "%s.%s", %s };zargument of %sz a%dz, r�z%s(%s)� __stdcallz_cffi_stdcall cSst|tj�o|jdkS)Nzlong double)rMrr�r)rCrrr�may_need_128_bits�sz9Recompiler._extern_python_decl..may_need_128_bitsr6�z!sizeof(%s) > %d ? sizeof(%s) : %dz%s%sr�z char a[%s];z char *p = a;za%drz *(%s)(p + %d) = %s;rz, _cffi_call_python(&_cffi_externpy__%s, p);z return *(%s)p;r�r)r�rMr�rrNr�r;rYrLrKr0r�r?rrWrPr�)r rCrZ tag_and_spacer�Zsize_of_resultr�r�r_r�r�r�r�rZ size_of_arrr�_extern_python_decl�sX "zRecompiler._extern_python_declcCs|j||d�dS)Nzstatic )r)r rCrrrr� _generate_cpy_extern_python_decl�sz+Recompiler._generate_cpy_extern_python_declcCs|j||d�dS)NzCFFI_DLLEXPORT )r)r rCrrrr�#_generate_cpy_dllexport_python_decl�sz.Recompiler._generate_cpy_dllexport_python_declcCs|j||d�dS)Nr)r)r rCrrrr�'_generate_cpy_extern_python_plus_c_decl�sz2Recompiler._generate_cpy_extern_python_plus_c_declcCsR|jrtd��|jrtd��|j|}tt|�}|jdjt |d|||��dS)Nz,cannot use 'extern "Python"' in the ABI modez$a vararg function is extern "Python"rtz&_cffi_externpy__%s) r<rr�rrEr\ZOP_EXTERN_PYTHONrzrKr)r rCrr&r rrr�_generate_cpy_extern_python_ctx�s z*Recompiler._generate_cpy_extern_python_ctxcCsz|j}|d�xf|jd�D]X}|d|j��d}x6|D].}t|�dkrT||�d}|dt|�f7}q8W||�qWdS)Nz=// # NB. this is not a string because of a size limit in MSVCTz// r�Lz%d,)r�� splitlines�rstripr�ord)r �sr��lineZprinted_line�crrrr�s z)Recompiler._print_string_literal_in_arraycCsttt�|j|<dS)N)r\�OP_PRIMITIVEZ PRIM_VOIDrI)r rCr�rrr�_emit_bytecode_VoidTypesz"Recompiler._emit_bytecode_VoidTypecCst|j}tt|�|j|<dS)N)ZPRIMITIVE_TO_INDEXrr\r+rI)r rCr�r7rrr�_emit_bytecode_PrimitiveTypes z'Recompiler._emit_bytecode_PrimitiveTypecCs(d|j|j|jf}tt|�|j|<dS)Nzl_cffi_prim_int(sizeof(%s), ( ((%s)-1) | 0 /* check that %s is an integer type */ ) <= 0))rr\r+rI)r rCr�r(rrr�!_emit_bytecode_UnknownIntegerTypesz,Recompiler._emit_bytecode_UnknownIntegerTypecCs$d|j|jf}tt|�|j|<dS)Nzd_cffi_prim_float(sizeof(%s) * (((%s)1) / 2) * 2 /* integer => 0, float => 1 */ ))rr\r+rI)r rCr�r(rrr�_emit_bytecode_UnknownFloatTypesz*Recompiler._emit_bytecode_UnknownFloatTypecCs�tt|j|j�|j|<|d7}xR|jD]H}|j|}||krht|tj�rX|j ||�ntt |�|j|<|d7}q(Wt|j�}|j dk r�|j dkr�|dO}ntd|j f��tt|�|j|<dS)Nrrr4zabi=%r)r\ZOP_FUNCTIONrEr�rIrLrMrr�r-r!rr�r�rZOP_FUNCTION_END)r rCr�r^Z realindexr'rrr�_emit_bytecode_RawFunctionType#s z)Recompiler._emit_bytecode_RawFunctionTypecCstt|j|j�|j|<dS)N)r\� OP_POINTERrEr�rI)r rCr�rrr�_emit_bytecode_PointerType6sz%Recompiler._emit_bytecode_PointerTypecCs"|j�}tt|j|�|j|<dS)N)rcr\r1rErI)r rCr��rawrrr�_emit_bytecode_FunctionPtrType<sz)Recompiler._emit_bytecode_FunctionPtrTypecCs�|j|j}|jdkr(tt|�|j|<nf|jdkrNtdt|�jdd�f��n@|j|ddksdt �tt |�|j|<tdt|j��|j|d<dS)Nz...zitype %s badly placed: the '...' array length can only be used on global arrays or on fields of structuresz/*...*/rrB)rEr�rSr\Z OP_OPEN_ARRAYrIrrHr�rJZOP_ARRAY)r rCr�Z item_indexrrr�_emit_bytecode_ArrayType@s z#Recompiler._emit_bytecode_ArrayTypecCs|j|}tt|�|j|<dS)N)rUr\ZOP_STRUCT_UNIONrI)r rCr�Zstruct_indexrrr�_emit_bytecode_StructTypeNs z$Recompiler._emit_bytecode_StructTypecCs|j|}tt|�|j|<dS)N)rVr\rrI)r rCr�Z enum_indexrrr�_emit_bytecode_EnumTypeSs z"Recompiler._emit_bytecode_EnumType)F)r)N)NrN)N)Urrrr�rr@rarbrFr{r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�rhr�r�r�r�r~r�Z_generate_cpy_union_collecttyper�r�Z_generate_cpy_union_declr�Z_generate_cpy_union_ctxrrrr rr�r�r�rrrrrrr�rrrrZ*_generate_cpy_dllexport_python_collecttypeZ._generate_cpy_extern_python_plus_c_collecttyperr r!r"r#Z"_generate_cpy_dllexport_python_ctxZ&_generate_cpy_extern_python_plus_c_ctxr�r,r-r.r/r0r2Z_emit_bytecode_ConstPointerTypeZ_emit_bytecode_NamedPointerTyper4r5r6Z_emit_bytecode_UnionTyper7rrrrr9{s� : , 4-#! Q 5 r9�cseZdZ�fdd�Z�ZS)�NativeIOcs(t|t�r|jd�}tt|�j|�dS)N�ascii)rMZunicode�encode�superr9r�)r r()r[rrr�\s zNativeIO.write)rrrr�� __classcell__rr)r[rr9[sr9c Cs|rtd|f�t|||dkd�}|j�|j�t�}|j||�|j�}y@t|d�� }|jt |�d�|krvt �WdQRX|r�td�dSt k �rd|tj�f} t| d��}|j |�WdQRXytj| |�Wn,tk �rtj|�tj| |�YnXd SXdS) Nz generating %s)r<r�rz(already up-to-date)Fz%s.~%d�wT)�printr9rar�r9r��getvaluer��readr�IOErrorr��getpidr��rename�OSError�unlink) r:r;r�Ztarget_file�verboseZ recompilerr��output�f1Ztmp_filerrr�_make_c_or_py_sourceas4 rJFcCs|dk st�t|||||�S)N)rJrJ)r:r;r�Z target_c_filerGrrr� make_c_source}s rKcCst||d||�S)N)rJ)r:r;Ztarget_py_filerGrrr�make_py_source�s rLcCsj|jd�}y$tjtjj|f|dd��Wntk rBYnX|d|7<tjj|f|��|fS)Nr�rr�r�)rlr��makedirsr�r0rE)Z outputdir�modname� extension�partsrrr�_modname_to_file�s $rQcCs*t||�}|j|||f�t|||�|S)N)rZrK�setattr)� patchlist�clsrZnew_meth�oldrrr�_patch_meth�s rVcCs(x"t|�D]\}}}t|||�q WdS)N)�reversedrR)rSrTrZold_methrrr�_unpatch_meths�srXcs\tjdkr(ddlm}t||ddd��tjdkrXddlm}�fd d �}t||d|��dS)N�win32r)�MSVCCompilerZ_remove_visual_c_refcSs|S)Nr)r Z manifest_filerrrrD�sz&_patch_for_embedding..�darwin)� CCompilercs<d|jkr,t|j�|_|jjd�}d|j|<�|f|�|�S)Nz-bundlez-dynamiclib)Z linker_sor�r�)r rL�kwdsr_)�old_link_shared_objectrr�my_link_shared_object�s z3_patch_for_embedding..my_link_shared_objectZlink_shared_object)�sys�platformZdistutils.msvc9compilerrZrVZdistutils.ccompilerr\)rSrZr\r_r)r^r�_patch_for_embedding�s rbcslddlm}�jd�rR�dd ��tjdkr6�d7�ntjdkrJ�d7�n�d 7�t||d �fdd��dS)Nr)� build_extz.*r4rYz.dllr[z.dylibz.soZget_ext_filenamecs�S)Nr)r Zext_name)�targetrrrD�sz#_patch_for_target..r�)Zdistutils.command.build_extrc�endswithr`rarV)rSrdrcr)rdr�_patch_for_target�s rfr�T�.ccKs�t|t�s|jd�}|jr$|j|�|dk �r^|jdk }|rF|j|�|dkrzt|||�\}} |rl|g| } tj j | �}n|}| dkr�|r�d|} nd} tj||f|�}t |||||d�}|�rTg}tj�}zn|r�t|�| dkr�t|| �|�r|dk�rd}nd}td|tj j|�f�tj|�tjd||| �}Wdtj|�t|�X|S||fSn<|dk�rxt||d �\}}t||||d�}|�r�|Sd|fSdS) Nr:z%s.*r)rGr�zthe current directory isz setting the current directory toz%s %rz.py)rMrHr;Z_windows_unicodeZ_apply_windows_unicoder�Z_apply_embedding_fixrQr�r�r0rZ get_extensionrK�getcwdrbrfr?�abspath�chdir�compilerXrL)r:r;r��tmpdirZcall_c_compilerZc_fileZsource_extensionZextradirZcompiler_verboserd�debugr]Z embeddingrPZ ext_c_fileZext�updatedrS�cwd�msg�outputfilenamerjrrr� recompile�sb rrcOs�ddlm}ddl}|tjks,td|f��|jdt|��t|||f|�|�}|j ||�}xFt |j�D]8} | jd�sht |j| �} | t || t��k rht|| | �qhWdd�}||_x6t |�D]*} | jd�r�t|j| �r�t|| t�q�W|jS)Nr)�udirzmodule name conflict: %rrlrjc_st�dS)N)r)rLr]rrr�typeof_disabledsz _verify..typeof_disabled)Ztesting.udirrs�impr`�modulesrJ� setdefaultrHrrZload_dynamic�dirr:r�rZ�objectrRZ_typeof�hasattr�NotImplemented�lib)r:r;r�rLr]rsrurq�moduler�attrrtrrr�_verifys$ r)r8)F)F)r�TNrgNrNN) r�r`�iorrr�errorrZcffi_opcoder=r�r�rrr%r1r8r9�version_info�StringIOr9�BytesIOrJrKrLrQrVrXrbrfrrrrrrr�sHf <