Current File : //lib64/python3.9/site-packages/cffi/__pycache__/recompiler.cpython-39.opt-1.pyc
a

<�`8��@s0ddlZddlZddlZddlmZmZddlmZddlTdZ	dZ
dZejd	kpdej
d
kpdej
dkZGdd
�d
�ZGdd�d�ZGdd�d�ZGdd�d�ZGdd�d�ZGdd�d�Zej
dkr�ejZnGdd�dej�Zdd�Zd1dd�Zd2d d!�Zd"d#�Zd$d%�Zd&d'�Zd(d)�Zd*d+�Zd3d/d0�Z dS)4�N�)�ffiplatform�model)�VerificationError)�*i&i'i(�win32)�r)r�c@s&eZdZd	dd�Zdd�Zdd�ZdS)
�
GlobalExprrcCs"||_||_||_||_||_dS�N)�name�address�type_op�size�check_value)�selfrr
rrr�r�5/usr/lib64/python3.9/site-packages/cffi/recompiler.py�__init__s
zGlobalExpr.__init__cCsd|j|j|j��|jfS)Nz'  { "%s", (void *)%s, %s, (void *)%s },)rr
r�	as_c_exprr�rrrrrs�zGlobalExpr.as_c_exprcCsd|j��|j|jfS)Nz
b'%s%s',%d)r�as_python_bytesrrrrrr�as_python_exprs�zGlobalExpr.as_python_exprN)rr��__name__�
__module__�__qualname__rrrrrrrr
s
r
c@s,eZdZdd�Zdd�Zdd�Zdd�Zd	S)
�	FieldExprcCs"||_||_||_||_||_dSr)r�field_offset�
field_size�fbitsize�
field_type_op)rrrrr r!rrrrs
zFieldExpr.__init__cCs>dt|j�}d|j|jfd||jfd||j��fS)N� z  { "%s", %s,
z     %s   %s,
z     %s   %s },)�lenrrrr!r)r�spacesrrrr&s��zFieldExpr.as_c_exprcCst�dSr)�NotImplementedErrorrrrrr,szFieldExpr.as_python_exprcCsD|jjtkrd}n|jjtkr*t|j�}nt�d|j��||jfS)N�z	b'%s%s%s')	r!�op�OP_NOOP�OP_BITFIELD�format_four_bytesr r%rr)rZ	size_exprrrr�as_field_python_expr/s
�zFieldExpr.as_field_python_exprN)rrrrrrr+rrrrrsrc@s$eZdZdd�Zdd�Zdd�ZdS)�StructUnionExprc		Cs4||_||_||_||_||_||_||_||_dSr)r�
type_index�flagsr�	alignment�comment�first_field_index�c_fields)	rrr-r.rr/r0r1r2rrrr;szStructUnionExpr.__init__cCsPd|j|j|jfd|j|jfd|jt|j�f|jrFd|jnddS)Nz  { "%s", %d, %s,z
    %s, %s, z%d, %d z	/* %s */ r&z},)	rr-r.rr/r1r#r2r0rrrrrFs����zStructUnionExpr.as_c_exprcCs>t|jt�}dd�|jD�}dt|j�t|�|jd�|�fS)NcSsg|]}|���qSr)r+)�.0Zc_fieldrrr�
<listcomp>Os�z2StructUnionExpr.as_python_expr.<locals>.<listcomp>z(b'%s%s%s',%s)�,)�evalr.ZG_FLAGSr2r*r-r�join)rr.Zfields_exprrrrrMs��zStructUnionExpr.as_python_exprNrrrrrr,:sr,c@s$eZdZdd�Zdd�Zdd�ZdS)�EnumExprcCs"||_||_||_||_||_dSr�rr-r�signed�allenums)rrr-rr:r;rrrrXs
zEnumExpr.__init__cCsd|j|j|j|j|jfS)Nz1  { "%s", %d, _cffi_prim_int(%s, %s),
    "%s" },r9rrrrr_s
��zEnumExpr.as_c_exprc	Cs@ttttttttd�|j|j	f}dt
|j�t
|�|j|j
fS)N))rr)rr)�r)r<r)�r)r=r)�r)r>rzb'%s%s%s\x00%s')Z
PRIM_UINT8Z	PRIM_INT8ZPRIM_UINT16Z
PRIM_INT16ZPRIM_UINT32Z
PRIM_INT32ZPRIM_UINT64Z
PRIM_INT64rr:r*r-rr;)r�
prim_indexrrrrds�
�
�zEnumExpr.as_python_exprNrrrrrr8Wsr8c@s$eZdZdd�Zdd�Zdd�ZdS)�TypenameExprcCs||_||_dSr�rr-)rrr-rrrrpszTypenameExpr.__init__cCsd|j|jfS)Nz  { "%s", %d },rArrrrrtszTypenameExpr.as_c_exprcCsdt|j�|jfS)Nzb'%s%s')r*r-rrrrrrwszTypenameExpr.as_python_exprNrrrrrr@osr@c@s�eZdZdZd�dd�Zdd�Zdd�Zd	d
�Zdd�Zd
d�Z	gd�Z
dd�Zd�dd�Zdd�Z
dd�Zdd�Zdd�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<�Z d=d>�Z!d�d@dA�Z"dBdC�Z#dDdE�Z$dFdG�Z%e%Z&dHdI�Z'dJdK�Z(e(Z)dLdM�Z*e*Z+dNdO�Z,dPdQ�Z-dRdS�Z.d�dUdV�Z/dWdX�Z0dYdZ�Z1d[d\�Z2d]d^�Z3d�d_d`�Z4dadb�Z5dcdd�Z6dedf�Z7dgdh�Z8didj�Z9dkdl�Z:dmdn�Z;dodp�Z<dqdr�Z=dsdt�Z>e>Z?Z@dudv�ZAdwdx�ZBdydz�ZCd{d|�ZDd}d~�ZEeEZFZGdd��ZHd�d��ZId�d��ZJd�d��ZKd�d��ZLd�d��ZMd�d��ZNeNZOeNZPd�d��ZQd�d��ZRd�d��ZSeSZTd�d��ZUd?S)��
RecompilerrFcCs||_||_||_t|_dSr)�ffi�module_name�target_is_python�VERSION_BASE�_version)rrCrDrErrrr�szRecompiler.__init__cCst|j|�|_dSr)�maxrG)rZverrrr�
needs_version�szRecompiler.needs_versioncCs�i|_|�d�t|jtd�}g|_|D]h}|jr(t|j�|j|<|j�|�|jD].}|j|durvt|j�|j|<|j�|�qT|j�d�q(|D]P}|js�|j|dur�t|j�|j|<|j�|�|j	r�|j
dur�|j�d�q�i|_i|_|D]4}t
|tj��rd|j|<q�t
|tj�r�d|j|<q�tt|jdd�d��D]\}}||j|<�qDtt|jdd�d��D]\}}||j|<�qp|D](}t|d|jj�}|||j|��q�|jD]}�q�t|j�|_dS)	NZcollecttype��keyZENDZLENcSs|jSr�r��tprrr�<lambda>��z/Recompiler.collect_type_table.<locals>.<lambda>cSs|jSrrLrMrrrrO�rPZ_emit_bytecode_)�
_typesdict�	_generate�sorted�str�
cffi_typesZis_raw_functionr#�append�argsZ
is_array_type�length�_struct_unions�_enums�
isinstancer�
StructOrUnion�EnumType�	enumerate�getattr�	__class__r�tuple)rZ	all_declsrN�tp1�i�methodr'rrr�collect_type_table�sP

��
zRecompiler.collect_type_tablecCs|j}|�|�Sr)rE�
enumfields)rrNZexpand_anonymous_struct_unionrrr�_enum_fields�szRecompiler._enum_fieldscCs�t|tj�s.t|t�r*|D]}|�|�qdS||jvr�d|j|<t|tj�r^|�|���nnt|tj�r�|j	dur�||j
jjvr�|�
|�D] \}}}}|�|�|||��q�n|��D]\}}|�|�q�dSr)r[rZBaseTypeByIdentityra�_do_collect_typerQ�FunctionPtrType�as_raw_functionr\�fldtypesrC�_parser�_included_declarationsrg�_field_typeZ
_get_items)rrN�xZname1rb�_rrrrh�s"



�zRecompiler._do_collect_typec
Cs�|jjj��}t|�D]�\}\}}|�dd�\}}zt|d||f�}Wntyftd|��Yn0z||_	|||�Wqt
y�}	zt�|	|��WYd}	~	qd}	~	00qdS)Nr"rz_generate_cpy_%s_%sz"not implemented in recompile(): %r)
rCrlZ
_declarations�itemsrS�splitr_�AttributeErrorr�_current_quals�	ExceptionrZattach_exception_info)
r�	step_name�lstrrN�quals�kindZrealnamerd�errrrR�s"��
zRecompiler._generate)�global�field�struct_union�enum�typenamecCs�i|_|jD]}g|j|<qt�|_|�d�|��|jD]4}|j|}|dkrb|jdd�d�t|�|j|<q<|jd}|j�	�D]\}}q�|jd}|j
�	�D]\}}q�dS)N�ctxr|cSs|jSrrL)�entryrrrrOrPz0Recompiler.collect_step_tables.<locals>.<lambda>rJr}r~)�_lsts�	ALL_STEPS�set�_seen_struct_unionsrR�_add_missing_struct_unions�sortrarYrqrZ)rrvrwrNrcrrr�collect_step_tables�s$





zRecompiler.collect_step_tablesr&cCs|j�|d�dS)N�
)�_f�write)rZwhatrrr�_prntszRecompiler._prntcCs"|jr|�|�n|�||�dSr)rE�write_py_source_to_f�write_c_source_to_f)r�f�preamblerrr�write_source_to_fszRecompiler.write_source_to_fcCs0ttj�tj�t�|�d�}|��}|��|S)N�r)�open�os�pathr7�dirname�__file__�	readlines�close)r�filename�g�linesrrr�_rel_readlines!szRecompiler._rel_readlinesc	Csl||_|j}|jjdur |d�ts,|d�|�d�}|�d�}|�d�|||d�<|d�|��|j�	d�d	}|jjdu�r<|d
|jf�|d�|�
|jj�|d�|d
�|d|f�|d�|d|f�|d�|d|f�|d�|�d�}|�d�}|�d�|||d�<|d�|��|�t�|d�|�||�|�|d�|�|d�t
dd�|j��D��}t|j�D]>\}}d}	||v�r�d||��}	|d||��|	f��q�|j�s�|d�|d�|�t�|_|�d�i}
|jD]`}|j|}t|�|
|<|
|d k�r
|d!||f�|D]}
||
����qF|d�|��q
|jj�r|d"�|jjD]d}z|jdd#�\}}Wn&t�y�td$|j|f��Yn0|du�r�td%��|d&|f��q�|d'�|d�|�|d(�|d)�|jD].}|
|d k�r8|d*|�n|d+|��q|jD]$}|d,k�rN|d-|
||f��qN|jj�r�|d.�n|d/�|d0t|j�f�d }|j�r�|dO}|d1|�|d�|�|d2�|d3�|d�|�|d
�|d4�|d5|f�|d6�|j�r6|d7�|d8�|d9�|d:|j�|d;�|d<�|d=�|d�|d>�|d?�|d@�|dA�|dB|f�|dC�|dD|f�|dE�|dE�|d�|d4�|dF|f�|d6�|dG|j|jf�|d>�|d�|d4�|dH|f�|d6�|dI|j|jf�|d>�|d�|�|d2�|dJ�|d�d|_dS)KNz#define _CFFI_USE_EMBEDDINGz#define _CFFI_NO_LIMITED_APIz_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_%s�#elsez*# define _CFFI_PYTHON_STARTUP_FUNC  init%s�#endifz_embedding.hz#include "_cffi_errors.h"
z_cffi_errors.hz>/************************************************************/zstatic void *_cffi_types[] = {cSsg|]\}}||f�qSrr�r3rNrcrrrr4WrPz2Recompiler.write_c_source_to_f.<locals>.<listcomp>z // z/* %2d */ %s,%sz  0z};�declrz.static const struct _cffi_%s_s _cffi_%ss[] = {z.static const char * const _cffi_includes[] = {r<�Qffi 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 */r|z  %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 >= 3�    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);zinit%s(void)z.  _cffi_init("%s", 0x%x, &_cffi_type_context);z#  pragma GCC visibility pop) r�r�rC�
_embedding�USE_LIMITED_APIr��indexr7rDrr�_print_string_literal_in_arrayrI�VERSION_EMBEDDED�dictrQrqr^rU�_get_c_namerr��_seen_constantsrRr�r�r#�_included_ffis�_assigned_sourcersr�
_num_externpyrG)rr�r��prntr�rcZbase_module_name�typeindex2typer'r0Znumsrvrwr��ffi_to_include�included_module_name�included_sourcer.rrrr�'s2

���






�

�
���

�




��zRecompiler.write_c_source_to_fcsbt|t�rd|fSt|ttf�rZ�fdd�|D�}t|�dkrJ|�d�dd�|�fS|��S)Nzb'%s'csg|]}��|��qSr)�_to_py)r3�itemrrrr4�rPz%Recompiler._to_py.<locals>.<listcomp>rr&z(%s)r5)r[rT�listrar#rVr7r)rroZreprrrr��s


zRecompiler._to_pyc	Cs�||_|j}|d�|d�t|jjp(d�}t|�D]l}|jj|}z|jdd�\}}Wn$ty~td|j|f��Yn0|dur�td��|d||f�q4|�|d|j	f�|d	|j
f�d|_
t|j�|_d
d�|jD�}|d|�
d
�|��f�tdd�|j��D��}	|jD]>}
|j|
}t|�dk�r"|
dk�r"|d|
|�
|�f��q"|dk�r�|dd�dd�t|�D��f�|d�dS)Nz# auto-generated filezimport _cffi_backendrr<r�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|]}|���qSr)r)r3r'rrrr4�rPz3Recompiler.write_py_source_to_f.<locals>.<listcomp>z    _types = %s,r&cSsg|]\}}||f�qSrrr�rrrr4rPrr|z    _%ss = %s,z    _includes = (%s,),�, cSsg|]}d|�qS)z_ffi%dr�r3rcrrrr4rP�))r�r�r#rCr��ranger�rsrrDrGrarUr�r7r�rQrqr�r�)rr�r�Znum_includesrcr�r�r�Z	types_lstr�rvrwrrrr��sN�
���
�


�zRecompiler.write_py_source_to_fcCs
|j|Sr)rQ)r�typerrr�_gettypenumszRecompiler._gettypenumc	Csnd}t|tj�r�|��s�|��r:|jdkr:d}d|j}nPt|tj�rXd|�d�f}n2|�d�}d||j�dd�f}|d	vr�|�	t
�d
}n�t|tj�r�|�||||�dSt|tj
�s�t|tj�r�|�d||�|�|f�|�d|�dSt|tj��r$d
|�d�}d|�|�}d}nt|��|�d||||f�|�d||�d�|f�|�d|�dS)Nr&�_BoolZ_cffi_to_c_intz, %sz(%s)_cffi_to_c_doublez(%s)_cffi_to_c_%sr"rp�Zchar16_tZchar32_tz-1z6  if (_cffi_to_c((char *)&%s, _cffi_type(%d), %s) < 0)z    %s;z(%s)_cffi_to_c_pointerz, _cffi_type(%d)ZNULLz  %s = %s(%s%s);z'  if (%s == (%s)%s && PyErr_Occurred()))r[r�BasePrimitiveType�is_complex_type�is_integer_typer�UnknownFloatType�
get_c_name�replacerI�VERSION_CHAR16CHAR32�PointerType�"_convert_funcarg_to_c_ptr_or_arrayZStructOrUnionOrEnumr�r�rir%)	rrN�fromvar�tovar�errcodeZextraargZ	converter�cnameZerrvaluerrr�_convert_funcarg_to_csJ
�
�
���z Recompiler._convert_funcarg_to_ccCs.t|tj�r*|�d�|�d�|�d�dS)NzPy_ssize_t datasizez-struct _cffi_freeme_s *large_args_free = NULLzIif (large_args_free != NULL) _cffi_free_array_arguments(large_args_free);)r[rr��add)rrN�	localvars�	freelinesrrr�_extra_local_variablesCs

z!Recompiler._extra_local_variablescCs�|�d�|�d|�|�||f�|�d�|�d||�d�f�|�d|�|�||f�|�d�|�d|�|�d	�dS)
Nz1  datasize = _cffi_prepare_pointer_call_argument(z(      _cffi_type(%d), %s, (char **)&%s);z  if (datasize != 0) {zI    %s = ((size_t)datasize) <= 640 ? (%s)alloca((size_t)datasize) : NULL;r&zF    if (_cffi_convert_array_argument(_cffi_type(%d), %s, (char **)&%s,z,            datasize, &large_args_free) < 0)z	      %s;z  })r�r�r�)rrNr�r�r�rrrr�Js
�

���
z-Recompiler._convert_funcarg_to_c_ptr_or_arraycCsJt|tj�r�|��r,|jdkr,d||jfSt|tj�rBd|fS|jdkr�|��s�|j�dd�}|dvrt|�t	�d||fSd	||�
|�fSn�t|tjtjf�r�d
||�
|�fSt|tj
�r�d
||�
t�|j��fSt|tj��r|jdu�rtd|��|f��d||�
|�fSt|tj��r>d	||�
|�fSt|��dS)
Nr�z_cffi_from_c_int(%s, %s)z_cffi_from_c_double(%s)�long doubler"rpr�z_cffi_from_c_%s(%s)z/_cffi_from_c_deref((char *)&%s, _cffi_type(%d))z0_cffi_from_c_pointer((char *)%s, _cffi_type(%d))z!'%s' is used as %s, but is opaquez0_cffi_from_c_struct((char *)&%s, _cffi_type(%d)))r[rr�r�rr�r�r�rIr�r�r�ri�	ArrayTyper�r\Zfldnames�	TypeErrorr�r]r%)rrN�var�contextr�rrr�_convert_expr_from_cXsD


�
���
�
�zRecompiler._convert_expr_from_ccCs|�|d|f�S)Nz
(*(%s *)0))�_global_type�rrNrrrr�
_typedef_type{szRecompiler._typedef_typecCs|�|�||��dSr)rhr�r�rrr�!_generate_cpy_typedef_collecttype~sz,Recompiler._generate_cpy_typedef_collecttypecCsdSrrr�rrr�_generate_cpy_typedef_decl�sz%Recompiler._generate_cpy_typedef_declcCs$|j|}|jd�t||��dS)Nr)rQr�rVr@)rrNrr-rrr�_typedef_ctx�s
zRecompiler._typedef_ctxcCsf|�||�}|�||�t|dd�dkr<|j||jdd�n&t|tj�rb|j|j|jj|j|d�dS)N�originZunknown_type)�
approxname)r��	named_ptr)	r�r�r_�_struct_ctxrr[rZNamedPointerType�totyper�rrr�_generate_cpy_typedef_ctx�s�z$Recompiler._generate_cpy_typedef_ctxcCs(|�|���|jr$|js$|�|�dSr)rhrj�ellipsisrEr�rrr�"_generate_cpy_function_collecttype�sz-Recompiler._generate_cpy_function_collecttypec	Cs|jr|�||�dS|j}t|j�}|dkr4d}n|dkrBd}nd}g}g}d|}t|j�D],\}	}
|�|
�d|	|��|�d|	�q`d	�|�}|p�d
}|j	r�|j	d}nd}d
|||f}
|d|j
�|
�f�|d�d	�|�}d}t|j
tj
��rd}|d|||f�|d�|d�|d�|d||f�|d�d|}t|j�D]&\}	}
|
�d|	|�}|d|��qXt�}t�}|jD]}
|�|
||��q�t|�D]}|d|f��q�t|j
tj
��sd}d|}d|j
�d|�}||�|d�nd}d}t|j�dk�r~tt|j��}|D]}	|d|	��q0|�|d|t|�t|�d	�dd�|D��f�|d�|�t|j�D](\}	}
|�|
d |	d|	d!�|��q�|d"�|d#�d$d�tt|j��D�}d	�|�}|d%|||f�|d&�|d'�|�|d(�|dk�r,|d)�|�rl|d*|�|j
d+d,��|D]}|d-|��qN|d.�n(|D]}|d-|��qp|d/�|d0�|d�|d1�d2d3�}d4}g}g}d|}t|j�D]P\}	}
d}||
��r�d5}d6}|
�d7||	f|�}|�|�|�d8||	f��q�|j
}||��r^d|}|�d9|�}|�d|�tj}d}d:}d6}|�r�d	�|�}|�pvd
}d;|||f}
|d|�|
�f�|d�|�r�||�d	�|�}|d%|||f�|�r�|d<�|d�n|d=||f�|d>�|�dS)?NrZnoargrZarg0rW�argument of %sz x%d�x%dr��voidr"r&z%s_cffi_d_%s(%s)z	static %sr�zreturn z  %s%s(%s);r�z#ifndef PYPY_VERSIONzstatic PyObject *z(_cffi_f_%s(PyObject *self, PyObject *%s)z  %s;z	result = �result of %sz resultz  PyObject *pyresult;z  PyObject *arg%d;z1  if (!PyArg_UnpackTuple(args, "%s", %d, %d, %s))cSsg|]}d|�qS)z&arg%drr�rrrr4�rPz:Recompiler._generate_cpy_function_decl.<locals>.<listcomp>r�zarg%dzreturn NULLz  Py_BEGIN_ALLOW_THREADSz  _cffi_restore_errno();cSsg|]}d|�qS)r�rr�rrrr4�rPz  { %s%s(%s); }z  _cffi_save_errno();z  Py_END_ALLOW_THREADSz  (void)self; /* unused */z  (void)noarg; /* unused */z  pyresult = %s;�resultzresult typez  z  return pyresult;z  Py_INCREF(Py_None);z  return Py_None;r�cSs t|tj�pt|tj�o|��Sr)r[rr\�
PrimitiveTyper�)r�rrr�need_indirections�z@Recompiler._generate_cpy_function_decl.<locals>.need_indirectionFrTz %sx%dz%sx%dz *resultz
*result = z%s_cffi_f_%s(%s)z  return result;z#  define _cffi_f_%s _cffi_d_%sr�)r��_generate_cpy_constant_declr�r#rWr^rVr�r7�abir�r[r�VoidTyper�r�rSr�r�r��insertZ	void_type)rrNrr��numargs�argname�	argumentsZcall_argumentsr�rcr��repr_argumentsr��name_and_argumentsZresult_code�argr�r�r�Zresult_decl�rngZfreeliner��
differenceZindirectionZ	tp_resultrrr�_generate_cpy_function_decl�s




��


�





�

z&Recompiler._generate_cpy_function_declcCs�|jr|js|�||�dS|j|��}t|j�}|jr@t}n |dkrNt}n|dkr\t	}nt
}|jd�t
|d|t||�d|d��dS)Nrrr{z
_cffi_f_%sz
_cffi_d_%s)r)r�rE�_generate_cpy_constant_ctxrQrjr#rWZOP_DLOPEN_FUNCZOP_CPYTHON_BLTN_NZOP_CPYTHON_BLTN_OZOP_CPYTHON_BLTN_Vr�rVr
�CffiOp)rrNrr-r�Z	meth_kindrrr�_generate_cpy_function_ctx;s$


��z%Recompiler._generate_cpy_function_ctxcCsTt|tj�rP|j}|dkr0|�d�}d||f}|�|d||j�}t�||�}|S)N�...rz_cffi_array_len(((%s)0)->%s)�%s[0])r[rr�rXr�rnr�)rZ	tp_struct�
field_nameZtp_field�
actual_lengthZptr_struct_name�tp_itemrrrrnQs
��zRecompiler._field_typecCs,|�|�|jr(|��D]}|�|�qdSr)rhrE�anonymous_struct_fields�_struct_collecttype)rrN�fldtyperrrr]s
zRecompiler._struct_collecttypecCsB|jdurdS|j}d|f}|d�|d||f�|d�|d�|d�|�|�D]�\}}}}	z�|��sv|dkr�|dkr�|d	|||f�WqXt|tj�r�|jdus�|jd
kr�|j}|d}q�|d|j	d
d||	d�|f�WqXt
�y}
z|dt|
��WYd}
~
qXd}
~
00qX|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;rr&z>  (void)((p->%s) | 0);  /* check that '%s.%s' is an integer */rz[0]z  { %s = &p->%s; (void)tmp; }z*tmpzfield %r�rxz
  /* %s */r�z(struct _cffi_align_%s { char x; %s y; };)rkr�rgr�r[rr�rXr�r�rrT)rrNr�r�r�Z
checkfuncname�fnameZftyper �fqualrzrrr�_struct_decldsB

���
�(zRecompiler._struct_declNcCs�|j|}d}g}t|tj�r(|�d�|jdur@|�d�d}||jjjvr�|dusd||jjjvr�|jdurpn|j	s�t
|���r�n
|�d�|jr�|jdkr�t
d||jf��|�d�n|�d�d	}d
�|�p�d}g}|du�rht|�|��}	|	D�]\}
}}}
|�||
|�}|�|d|j|
f�t}|d
k�rFt}d|}nL|du�sjt|tj��rp|jdu�rpd}n"d|du�r�|�d�n|j|
f}|du�s�|d
k�r�d}n,|du�r�d|j|
f}nd|�d�|
f}|�t|
|||t||j|���q�t|jd�}|jd�|�|du�r2d}d}d}n4|du�rNd|jf}d}nd|f}d|f}d}nd}d}d}|}|jd�t|j|||||||��|j� |�dS)NZ
_CFFI_F_UNIONZ_CFFI_F_OPAQUEZopaqueZ_CFFI_F_CHECK_FIELDSrz�%r is declared with 'pack=%r'; only 0 or 1 are supported in API mode (try to use "...;", which does not require a 'pack' declaration)Z_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)r&r|z
(size_t)-2���Zunnamedzsizeof(*(%s)0)z-1 /* unknown alignment */�
sizeof(%s)z"offsetof(struct _cffi_align_%s, y)r�r})!rQr[rZ	UnionTyperVrkrCrlrm�partial�anyrZpackedr%r7r�rgrn�_check_not_opaquerr(r)r�rXr�rr�r#r��extendr,r�r�)rrNr�r�r�r-Zreason_for_not_expandingr.r2rfZfldnamerr r
r'r�offsetr1Zalignr0rrrr��s�



��


��

�



���
�
��




��zRecompiler._struct_ctxcCs:t|tj�r|j}qt|tj�r6|jdur6td|��dS)Nz0%s is of an opaque type (not declared in cdef()))r[rr�r�r\rkr�)rrN�locationrrrr�s�zRecompiler._check_not_opaquecCs�t|j���}|jdd�d�|D]�\}}||jvr"|jrHtd|f��|j�d�rv|jdd��	�rv|jdd�}n4|jdkr�|j
dkr�d}|�|d�ntd	|f��|�|d|�q"dS)
NcSs|dS)Nrr)Ztp_orderrrrrO�rPz7Recompiler._add_missing_struct_unions.<locals>.<lambda>rJzDinternal inconsistency: %r is partial but was not seen at this point�$rZ_IO_FILE�FILEzinternal inconsistency: %r)
r�rYrqr�r�rr%r�
startswith�isdigitZ	forcenamer�r�)rrwrN�orderr�rrrr��s"
��z%Recompiler._add_missing_struct_unionscCs|�|�dSr)rr�rrr� _generate_cpy_struct_collecttype�sz+Recompiler._generate_cpy_struct_collecttypecCs2|�d�}d|vr"||�dd�fS|d|fSdS)Nr&r"rp)r�r�)rrNr�rrr�
_struct_names�s
zRecompiler._struct_namescCs|j|g|�|��R�dSr)rrr�rrr�_generate_cpy_struct_declsz$Recompiler._generate_cpy_struct_declcCs|j|g|�|��R�dSr)r�rr�rrr�_generate_cpy_struct_ctx	sz#Recompiler._generate_cpy_struct_ctxcCs(t|tj�r|�||�n
|�|�dSr)r[rr]�_generate_cpy_enum_collecttyperr�rrr�#_generate_cpy_anonymous_collecttypesz.Recompiler._generate_cpy_anonymous_collecttypecCs.t|tj�r|�|�n|�||d|�dS�NZtypedef_)r[rr]�_generate_cpy_enum_declrr�rrr�_generate_cpy_anonymous_declsz'Recompiler._generate_cpy_anonymous_declcCs0t|tj�r|�||�n|�||d|�dSr!)r[rr]�	_enum_ctxr�r�rrr�_generate_cpy_anonymous_ctxsz&Recompiler._generate_cpy_anonymous_ctx�constcCs�||f|jvrtd||f��|j�||f�|j}d||f}|r�|d|�|d�|d|f�|d||f�|dur�|dkr�d|f}|d	|f�|d
�|d�|d�n2|d
|�|d�|d|�d�|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�r�)r�is_intrrN�categoryrr��funcnamerrr�_generate_cpy_const&s4
��

zRecompiler._generate_cpy_constcCs |��}|r|jr|�|�dSr)r�rErh�rrNrr'rrr�"_generate_cpy_constant_collecttypeDs
z-Recompiler._generate_cpy_constant_collecttypecCs|��}|�|||�dSr)r�r*r+rrrr�Isz&Recompiler._generate_cpy_constant_declcCs^|js|��rttd�}n$|jr&t}nt}|j|}t||�}|jd�t	|d||��dS)Nr�r{�_cffi_const_%s)
rEr�r��OP_CONSTANT_INTZOP_DLOPEN_CONSTZOP_CONSTANTrQr�rVr
)rrNrrZ
const_kindr-rrrr�Ms


�z%Recompiler._generate_cpy_constant_ctxcCs|�|�dSr�rhr�rrrr]sz)Recompiler._generate_cpy_enum_collecttypecCs|jD]}|�d|�qdS)NT)�enumeratorsr*)rrNr�
enumeratorrrrr"`s
z"Recompiler._generate_cpy_enum_declc	Cs�|j|}ttd�}|jr"|��t|j|j�D](\}}|jd�	t
|d|||d��q0|dur�d|vr�|js�d|}d|}n4|�|jg�}	|j�
|	�}tt|j�|	d��dk�}d	�|j�}
|jd
�	t|j||||
��dS)Nr�r{r-�rrrz
((%s)-1) <= 0rr5r~)rQr��OP_ENUMrEZcheck_not_partial�zipr0Z
enumvaluesr�rVr
Zbuild_baseinttyperCZsizeof�int�castr7r8r)rrNr�r-rr1Z	enumvaluerr:Zbasetpr;rrrr$ds(


��

�zRecompiler._enum_ctxcCs|�||���dSr)r$r�r�rrr�_generate_cpy_enum_ctxysz!Recompiler._generate_cpy_enum_ctxcCsdSrrr�rrr�_generate_cpy_macro_collecttypesz*Recompiler._generate_cpy_macro_collecttypecCs&|dkrd}n|}|jd||d�dS)NrTr2)r*)rrNrrrrr�_generate_cpy_macro_decl�sz#Recompiler._generate_cpy_macro_declcCsT|dkr"|jrtd|f��d}n|}ttd�}|jd�t|d|||d��dS)NrzGcannot use the syntax '...' in '#define %s ...' when using the ABI moder�r{r-r2)rErr�r.r�rVr
)rrNrrrrrr�_generate_cpy_macro_ctx�s��

��z"Recompiler._generate_cpy_macro_ctxcCsFt|tj�rB|j}|dkr$d|f}|�|jd|�}t�||�}|S)Nrz_cffi_array_len(%s)r)r[rr�rXr�r�)rrNZglobal_namerrrrrr��s
zRecompiler._global_typecCs|�|�||��dSr)rhr�r�rrr�"_generate_cpy_variable_collecttype�sz-Recompiler._generate_cpy_variable_collecttypecCs�|j}|�||�}t|tj�r4|jdur4|j}d}nd}d|f}|d|j||jd��|d�|d||f�|d�|�dS)	Nr&�&z*_cffi_var_%s(void)�static rr�z  return %s(%s);r�)	r�r�r[rr�rXr�r�rt)rrNrr�Z	ampersandr�rrr�_generate_cpy_variable_decl�s	
z&Recompiler._generate_cpy_variable_declcCsL|�||�}|j|}|jr"t}nt}|jd�t|d|t||���dS)Nr{z_cffi_var_%s)	r�rQrEZ
OP_GLOBAL_VARZOP_GLOBAL_VAR_Fr�rVr
r�)rrNrr-r'rrr�_generate_cpy_variable_ctx�s

�z%Recompiler._generate_cpy_variable_ctxcCs|�|�dSrr/r�rrr�'_generate_cpy_extern_python_collecttype�sz2Recompiler._generate_cpy_extern_python_collecttypecCs|j}t|jtj�rd}nd|}d|j�d|�f}|d|�|d|j||f�|�g}d|}t|j�D]"\}}	|	�d||�}
|�	|
�qrd	�
|�}|p�d
}d||f}|jdkr�d
|}dd�}
tt
|j�dd�}|
|j�r�t|d�}t|jtj��r&d|j�d�||j�d�|f}|d||j�|�f�|d�|d|�|d�t|j�D]X\}}	d|}
t|	tj��s�|
|	��r�d|
}
t�|	�}	|d|	�d�|d|
f��qd|d|�t|jtj��s�|d|j�d�f�|d�|�|jd7_dS)Nr
r�z(int)sizeof(%s)r&z3static struct _cffi_externpy_s _cffi_externpy__%s =z  { "%s.%s", %s, 0, 0 };r�z a%dr�r�z%s(%s)�	__stdcallz_cffi_stdcall cSst|tj�o|jdkS)Nr�)r[rr�rrMrrr�may_need_128_bits�s�z9Recompiler._extern_python_decl.<locals>.may_need_128_bitsr>�z!sizeof(%s) > %d ? sizeof(%s) : %dz%s%sr�z
  char a[%s];z  char *p = a;za%dr<z  *(%s)(p + %d) = %s;rz,  _cffi_call_python(&_cffi_externpy__%s, p);z  return *(%s)p;r�r)r�r[r�rr�r�rDr^rWrVr7r�rHr#r\r�r�)rrNrZ
tag_and_spacer�Zsize_of_resultr�r�rcr�r�r�r�rBZ	size_of_arrr�_extern_python_decl�sb��



��
 zRecompiler._extern_python_declcCs|�||d�dS)Nr=�rDr�rrr� _generate_cpy_extern_python_declsz+Recompiler._generate_cpy_extern_python_declcCs|�||d�dS)NzCFFI_DLLEXPORT rEr�rrr�#_generate_cpy_dllexport_python_decl
sz.Recompiler._generate_cpy_dllexport_python_declcCs|�||d�dS)Nr&rEr�rrr�'_generate_cpy_extern_python_plus_c_decl
sz2Recompiler._generate_cpy_extern_python_plus_c_declcCsR|jrtd��|jrtd��|j|}tt|�}|jd�t	|d|||��dS)Nz,cannot use 'extern "Python"' in the ABI modez$a vararg function is extern "Python"r{z&_cffi_externpy__%s)
rErr�r%rQr�ZOP_EXTERN_PYTHONr�rVr
)rrNrr-rrrr�_generate_cpy_extern_python_ctxs�


�z*Recompiler._generate_cpy_extern_python_ctxcCs�|j}|d�t|t�s$|�d�}n
|�d�z|�d�WntyVd|}Yn0|�d�D]|}|}td�tur�tt	|�}nt
|�dd�}|d	|���d
}|D]*}t|�dkr�||�d
}|d|f7}q�||�qbdS)
Nz=// # NB. this is not a string because of a size limit in MSVCzutf-8�asciis# -*- encoding: utf8 -*-
Tz//rr�z// r&�Lz%d,)
r�r[�bytes�encode�decode�UnicodeDecodeError�
splitlinesr��map�ordrJ�rstripr#)r�sr��liner0Zprinted_line�crrrr�s,

z)Recompiler._print_string_literal_in_arraycCsttt�|j|<dSr)r��OP_PRIMITIVEZ	PRIM_VOIDrU�rrNr�rrr�_emit_bytecode_VoidType=sz"Recompiler._emit_bytecode_VoidTypecCst|j}tt|�|j|<dSr)ZPRIMITIVE_TO_INDEXrr�rWrU)rrNr�r?rrr�_emit_bytecode_PrimitiveType@s
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)�rr�rWrU�rrNr�rTrrr�!_emit_bytecode_UnknownIntegerTypeDs�z,Recompiler._emit_bytecode_UnknownIntegerTypecCs$d|j|jf}tt|�|j|<dS)Nzd_cffi_prim_float(sizeof(%s) *
           (((%s)1) / 2) * 2 /* integer => 0, float => 1 */
         )r[r\rrr�_emit_bytecode_UnknownFloatTypeJs
�z*Recompiler._emit_bytecode_UnknownFloatTypecCs�tt|j|j�|j|<|d7}|jD]H}|j|}||krft|tj�rV|�	||�ntt
|�|j|<|d7}q&t|j�}|j
dur�|j
dkr�|dO}ntd|j
f��tt|�|j|<dS)NrrAr<zabi=%r)r�ZOP_FUNCTIONrQr�rUrWr[rr�rZr(r5r�r�r%ZOP_FUNCTION_END)rrNr�rbZ	realindexr.rrr�_emit_bytecode_RawFunctionTypePs






z)Recompiler._emit_bytecode_RawFunctionTypecCstt|j|j�|j|<dSr)r��
OP_POINTERrQr�rUrXrrr�_emit_bytecode_PointerTypecsz%Recompiler._emit_bytecode_PointerTypecCs"|��}tt|j|�|j|<dSr)rjr�r`rQrU)rrNr��rawrrr�_emit_bytecode_FunctionPtrTypeisz)Recompiler._emit_bytecode_FunctionPtrTypecCs||j|j}|jdur(tt|�|j|<nP|jdkrNtdt|��dd�f��n*tt	|�|j|<tdt|j��|j|d<dS)Nrzitype %s badly placed: the '...' array length can only be used on global arrays or on fields of structuresz/*...*/r)
rQr�rXr�Z
OP_OPEN_ARRAYrUrrTr�ZOP_ARRAY)rrNr�Z
item_indexrrr�_emit_bytecode_ArrayTypems

���z#Recompiler._emit_bytecode_ArrayTypecCs|j|}tt|�|j|<dSr)rYr�ZOP_STRUCT_UNIONrU)rrNr�Zstruct_indexrrr�_emit_bytecode_StructType{s
z$Recompiler._emit_bytecode_StructTypecCs|j|}tt|�|j|<dSr)rZr�r3rU)rrNr�Z
enum_indexrrr�_emit_bytecode_EnumType�s
z"Recompiler._emit_bytecode_EnumType)F)r&)N)Nr&N)N)Vrrrr�rrIrergrhrRr�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�rnrrr�rr�rZ_generate_cpy_union_collecttyperrZ_generate_cpy_union_declrZ_generate_cpy_union_ctxr r#r%r*r,r�r�rr"r$r7r8r9r:r�r;r>r?r@Z*_generate_cpy_dllexport_python_collecttypeZ._generate_cpy_extern_python_plus_c_collecttyperDrFrGrHrIZ"_generate_cpy_dllexport_python_ctxZ&_generate_cpy_extern_python_plus_c_ctxr�rYrZr]r^r_raZ_emit_bytecode_ConstPointerTypeZ_emit_bytecode_NamedPointerTypercrdreZ_emit_bytecode_UnionTyperfrrrrrB~s�
:
.
4-##"
V	�

	
�6
�rB)rcseZdZ�fdd�Z�ZS)�NativeIOcs(t|t�r|�d�}tt|��|�dS)NrJ)r[ZunicoderM�superrgr�)rrT�r`rrr��s

zNativeIO.write)rrrr��
__classcell__rrrirrg�srgc
Cs@|rtd|f�t|||dud�}|��|��t�}|�||�|��}zVt|d��*}|�t	|�d�|krvt
�Wd�n1s�0Y|r�td�WdSt
�y:d|t��f}	t|	d��}|�
|�Wd�n1s�0Yzt�|	|�Wn*t�y2t�|�t�|	|�Yn0Yd	S0dS)
Nz
generating %s)rEr�rz(already up-to-date)Fz%s.~%d�wT)�printrBrer�rgr��getvaluer��readr#�IOErrorr��getpidr��rename�OSError�unlink)
rCrDr�Ztarget_file�verboseZ
recompilerr��output�f1Ztmp_filerrr�_make_c_or_py_source�s6�"(
rwFcCst|||||�Sr�rw)rCrDr�Z
target_c_filertrrr�
make_c_source�s
�rycCst||d||�Srrx)rCrDZtarget_py_filertrrr�make_py_source�s
�rzcCsl|�d�}z&t�tjj|g|dd��R��WntyBYn0|d|7<tjj|g|�R�|fS)Nr�r�)rrr��makedirsr�r7rr)Z	outputdir�modname�	extension�partsrrr�_modname_to_file�s
&rcCs*t||�}|�|||f�t|||�|Sr)r_rV�setattr)�	patchlist�clsrZnew_meth�oldrrr�_patch_meth�s
r�cCs$t|�D]\}}}t|||�qdSr)�reversedr�)r�r�rZold_methrrr�_unpatch_meths�sr�cs\tjdkr(ddlm}t||ddd��tjdkrXddlm}�fd	d
�}t||d|��dS)Nrr)�MSVCCompilerZ_remove_visual_c_refcSs|Srr)rZ
manifest_filerrrrO�rPz&_patch_for_embedding.<locals>.<lambda>�darwin)�	CCompilercsBd|jvr,t|j�|_|j�d�}d|j|<�|g|�Ri|��S)Nz-bundlez-dynamiclib)Z	linker_sor�r�)rrW�kwdsrc�Zold_link_shared_objectrr�my_link_shared_object�s


z3_patch_for_embedding.<locals>.my_link_shared_objectZlink_shared_object)�sys�platformZdistutils.msvc9compilerr�r�Zdistutils.ccompilerr�)r�r�r�r�rr�r�_patch_for_embedding�s
�
�r�cslddlm}��d�rR�dd��tjdkr6�d7�ntjdkrJ�d7�n�d	7�t||d
�fdd��dS)
Nr)�	build_extz.*rrz.dllr�z.dylibz.soZget_ext_filenamecs�Srr)rZext_name��targetrrrO�rPz#_patch_for_target.<locals>.<lambda>)Zdistutils.command.build_extr��endswithr�r�r�)r�r�r�rr�r�_patch_for_target�s





�r�r�T�.cc	Ks�t|t�s|�d�}|jr$|�|�|du�rt|jdu}|rF|�|�|durzt|||�\}}
|rl|g|
}
tj	j
|
�}n|}|	dur�|r�d|}	nd}	tj||fi|��}t
|||||d�}|�rjg}t��}z�|r�t|�|	dkr�t||	�|�r"|dk�rd}nd}td|tj	�|�f�t�|�t�d|||
�}Wt�|�t|�nt�|�t|�0|S||fSn<|du�r�t||d	�\}}t||||d�}|�r�|Sd|fSdS)
NrJz%s.*r)rtr�zthe current directory isz setting the current directory toz%s %rz.py)r[rTrMZ_windows_unicodeZ_apply_windows_unicoder�Z_apply_embedding_fixrr�r�r7rZ
get_extensionry�getcwdr�r�rl�abspath�chdir�compiler�rz)rCrDr�ZtmpdirZcall_c_compilerZc_fileZsource_extensionZextradirZcompiler_verboser��debugr�Z	embeddingr~Z
ext_c_file�ext�updatedr��cwd�msgZoutputfilenamerprrr�	recompile�sn





�


�


�

�



�r�)F)F)r�TNr�NrNN)!r�r��ior&rr�errorrZcffi_opcoderFr�r�r��version_infor�r
rr,r8r@rB�StringIOrg�BytesIOrwryrzrr�r�r�r�r�rrrr�<module>sP�


�