File: //opt/saltstack/salt/lib/python3.10/site-packages/cffi/__pycache__/recompiler.cpython-310.pyc
o
�N�g8� � @ s6 d dl Z d dlZd dlZddlmZmZ ddlmZ ddlT dZ dZ
dZ ej d kp2ej
d
k p2ej
d kZG d d
� d
�ZG dd� d�ZG dd� d�ZG dd� d�ZG dd� d�ZG dd� d�Zej
dkrfejZn G dd� dej�Zdd � Zd1d d �Zd1d d!�Zd"d#� Zd$d%� Z d&d'� Z d(d)� Z d*d+� Z - d2d/d0�Z dS )3� N� )� ffiplatform�model)�VerificationError)�*i& i' i( �win32)� r )r � c @ s& e Zd Zd dd�Zdd� Zdd� ZdS )
�
GlobalExprr c C �" || _ || _|| _|| _|| _d S �N)�name�address�type_op�size� check_value)�selfr
r r r r � r �C/opt/saltstack/salt/lib/python3.10/site-packages/cffi/recompiler.py�__init__ �
zGlobalExpr.__init__c C s d| j | j| j�� | jf S )Nz' { "%s", (void *)%s, %s, (void *)%s },)r
r r � as_c_exprr �r r r r r s �zGlobalExpr.as_c_exprc C s d| j �� | j| jf S )Nz
b'%s%s',%d)r �as_python_bytesr
r r r r r �as_python_expr s �zGlobalExpr.as_python_exprN)r r ��__name__�
__module__� __qualname__r r r r r r r r
s
r
c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
� FieldExprc C r r )r
� field_offset�
field_size�fbitsize�
field_type_op)r r
r r! r" r# r r r r r zFieldExpr.__init__c C s> dt | j� }d| j| jf d|| jf d|| j�� f S )N� z { "%s", %s,
z %s %s,
z %s %s },)�lenr
r r! r# r )r �spacesr r r r & s ��zFieldExpr.as_c_exprc C s t �r )�NotImplementedErrorr r r r r , � zFieldExpr.as_python_exprc C sD | j jtkr d}n| j jtkrt| j�}nt�d| j �� || jf S )N� z b'%s%s%s') r# �op�OP_NOOP� OP_BITFIELD�format_four_bytesr" r' r r
)r Z size_exprr r r �as_field_python_expr/ s
�z FieldExpr.as_field_python_exprN)r r r r r r r. r r r r r s
r c @ �$ e Zd Zdd� Zdd� Zdd� ZdS )�StructUnionExprc C s4 || _ || _|| _|| _|| _|| _|| _|| _d S r )r
�
type_index�flagsr � alignment�comment�first_field_index�c_fields) r r
r1 r2 r r3 r4 r5 r6 r r r r ; s
zStructUnionExpr.__init__c C sV d| j | j| jf d| j| jf d| jt| j�f | jr&d| j d S d d S )Nz { "%s", %d, %s,z
%s, %s, z%d, %d z /* %s */ r) z},) r
r1 r2 r r3 r5 r% r6 r4 r r r r r F s ������zStructUnionExpr.as_c_exprc C s> t | jt�}dd� | jD �}dt| j�t|�| jd�|�f S )Nc S � g | ]}|� � �qS r )r. )�.0Zc_fieldr r r �
<listcomp>O s �z2StructUnionExpr.as_python_expr.<locals>.<listcomp>z(b'%s%s%s',%s)�,)�evalr2 ZG_FLAGSr6 r- r1 r
�join)r r2 Z fields_exprr r r r M s ��z StructUnionExpr.as_python_exprNr r r r r r0 : s r0 c @ r/ )�EnumExprc C r r �r
r1 r �signed�allenums)r r
r1 r r? r@ r r r r X r zEnumExpr.__init__c C s d| j | j| j| j| jf S )Nz1 { "%s", %d, _cffi_prim_int(%s, %s),
"%s" },r> r r r r r _ s
��zEnumExpr.as_c_exprc C s@ t tttttttd�| j| j f }dt
| j �t
|�| j | j
f S )N))r r )r r )� r )rA r )� r )rB r )� r )rC r zb'%s%s%s\x00%s')Z
PRIM_UINT8Z PRIM_INT8Z PRIM_UINT16Z
PRIM_INT16Z PRIM_UINT32Z
PRIM_INT32Z PRIM_UINT64Z
PRIM_INT64r r? r- r1 r
r@ )r �
prim_indexr r r r d s �
�
�zEnumExpr.as_python_exprNr r r r r r= W s r= c @ r/ )� TypenameExprc C s || _ || _d S r �r
r1 )r r
r1 r r r r p s
zTypenameExpr.__init__c C s d| j | jf S )Nz { "%s", %d },rF r r r r r t s zTypenameExpr.as_c_exprc C s dt | j�| jf S )Nzb'%s%s')r- r1 r
r r r r r w � zTypenameExpr.as_python_exprNr r r r r rE o s rE c @ s� e Zd ZdZd�dd�Zdd� Zdd� Zd d
� Zd d � Zd
d� Z g d�Z
dd� Z d�dd�Z dd� Z
dd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd+d,� Zd-d.� Zd/d0� Zd1d2� Zd3d4� Z d5d6� Z d7d8� Z d9d:� Z d;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. T ?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~� ZEeE ZFZGdd�� ZHd�d�� ZId�d�� ZJd�d�� ZKd�d�� ZLd�d�� ZMd�d�� ZNeNZOeNZPd�d�� ZQd�d�� ZRd�d�� ZSeSZTd�d�� ZUd?S )��
Recompilerr Fc C s || _ || _|| _t| _d S r )�ffi� module_name�target_is_python� VERSION_BASE�_version)r rI rJ rK r r r r � s
zRecompiler.__init__c C s t | j|�| _d S r )�maxrM )r �verr r r �
needs_version� � zRecompiler.needs_versionc C s i | _ | �d� t| j td�}g | _|D ]N}|jrb| j | d u s"J �t| j�| j |< | j�|� |jD ](}t |t
j t
j t
j
t
jt
jf�sFJ �| j | d u rUt| j�| j |< | j�|� q3| j�d� q|D ](}|js�| j | d u r�t| j�| j |< | j�|� |jr�|jd ur�| j�d� qed | j �� vs�J �i | _i | _|D ]}t |t
j�r�d | j|< q�t |t
j�r�d | j|< q�tt| jdd� d��D ] \}}|| j|< q�tt| jdd� d��D ] \}}|| j|< q�|D ]}t| d|jj �}||| j | � q�| jD ]
}t |t��sJ �q�t | j�| _d S ) NZ collecttype��keyZEND�LENc S � | j S r �r
��tpr r r �<lambda>� � z/Recompiler.collect_type_table.<locals>.<lambda>c S rU r rV rW r r r rY � rZ Z_emit_bytecode_) �
_typesdict� _generate�sorted�str�
cffi_typesZis_raw_functionr% �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_declsrX �tp1�i�methodr* r r r �collect_type_table� sf
� � �
�� �
z Recompiler.collect_type_tablec C s | j }|�|�S r )rK �
enumfields)r rX Z expand_anonymous_struct_unionr r r � _enum_fields� s
zRecompiler._enum_fieldsc C s� t |tj�st |t�r|D ]}| �|� q
d S || jvrld | j|< t |tj�r0| �|�� � d S t |tj�r^|j d urX|| j
j j vrZ| �
|�D ]\}}}}| �| �|||�� qGd S d S d S |�� D ] \}}| �|� qbd S d S r )rb r ZBaseTypeByIdentityrr �_do_collect_typer[ rg �as_raw_functionrl �fldtypesrI �_parser�_included_declarationsrx � _field_typeZ
_get_items)r rX �xZname1rs �_r r r ry � s*
� �
zRecompiler._do_collect_typec
C s� | j jj�� }t|�D ]D\}\}}|�dd�\}}z t| d||f �}W n
ty1 td| ��w z
|| _ |||� W q t
yO } zt � | |� � d } ~ ww d S )Nr$ r z_generate_cpy_%s_%sz"not implemented in recompile(): %r)
rI r| Z
_declarations�itemsr] �splitro �AttributeErrorr �_current_quals� Exceptionr Zattach_exception_info)
r � step_name�lstr
rX �quals�kindZrealnameru �er r r r\ � s* � �� ���zRecompiler._generate)�global�field� struct_union�enum�typenamec C s& i | _ | jD ]}g | j |< qt� | _| �d� | �� | jD ]}| j | }|dkr1|jdd� d� t|�| j |< q | j d }| j� � D ]\}}|t
|�k sOJ �|| j |j ksYJ �qCt
|�t
| j�kseJ �| j d }| j � � D ]\}}|t
|�k s{J �|| j |j ks�J �qot
|�t
| j �ks�J �d S )N�ctxr� c S rU r rV )�entryr r r rY rZ z0Recompiler.collect_step_tables.<locals>.<lambda>rR r� r� )
�_lsts� ALL_STEPS�set�_seen_struct_unionsr\ �_add_missing_struct_unions�sortrr rj r� r% r
rk )r r� r� rX rt r r r �collect_step_tables� s*
z Recompiler.collect_step_tablesr) c C s | j �|d � d S )N�
)�_f�write)r Zwhatr r r �_prnt rG zRecompiler._prntc C s< | j r|d u s J �| �|� d S |d usJ �| �||� d S r )rK �write_py_source_to_f�write_c_source_to_f)r �f�preambler r r �write_source_to_f s
z Recompiler.write_source_to_fc C s0 t tj�tj�t�|�d�}|�� }|�� |S )N�r)�open�os�pathr< �dirname�__file__� readlines�close)r �filename�g�linesr r r �_rel_readlines! s zRecompiler._rel_readlinesc C sb || _ | j}| jjd ur|d� ts|d� | �d�}|�d�}| �d�|||d �< |d�|�� | j� d�d }| jjd ur�|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�| js�|d � |d � |� t� | _| �d � i }
| jD ]0} | j| } t| �|
| < |
| d k�r0|d!| | f � | D ] }
||
�� � �q |d � |� �q| jj�r{|d"� | jjD ]0}z |jd d#� \}}W n t �y] t d$| j|f ��w |d u �rgt d%��|d&|f � �q?|d'� |d � |� |d(� |d)� | jD ]} |
| d k�r�|d*| � �q�|d+| � �q�| jD ]} | d,k�r�|d-|
| | f � �q�| jj�r�|d.� n|d/� |d0t| j�f � d }| j �r�|dO }|d1| � |d � |� |d2� |d3� |d� |� |d
� |d4� |d5|f � |d6� | j �r|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| j f � |d>� |d� |d4� |dH|f � |d6� |dI| j| j f � |d>� |d� |� |d2� |dJ� |d� d | _ d S )KNz#define _CFFI_USE_EMBEDDINGz #define _CFFI_NO_LIMITED_APIz_cffi_include.hz#include "parse_c_type.h"
zparse_c_type.hr r) �.���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>/************************************************************/z static void *_cffi_types[] = {c S � g | ]\}}||f�qS r r �r8 rX rt r r r r9 W � z2Recompiler.write_c_source_to_f.<locals>.<listcomp>z // z/* %2d */ %s,%sz 0z};�declr z.static const struct _cffi_%s_s _cffi_%ss[] = {z.static const char * const _cffi_includes[] = {rA �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� rI �
_embedding�USE_LIMITED_APIr� �indexr< rJ r� � _print_string_literal_in_arrayrP �VERSION_EMBEDDED�dictr[ r� rn r_ � _get_c_namer r� �_seen_constantsr\ r� r� r% �_included_ffis�_assigned_sourcer� r �
_num_externpyrM )r r� r� �prntr� rt Zbase_module_name�typeindex2typer* r4 Znumsr� r� r� �ffi_to_include�included_module_name�included_sourcer2 r r r r� ' s8
���
��
�����
�
�
��
z Recompiler.write_c_source_to_fc sb t |t�r
d|f S t |ttf�r-� fdd�|D �}t|�dkr%|�d� dd�|�f S |�� S )Nzb'%s'c s g | ]}� � |��qS r )�_to_py)r8 �itemr r r r9 � s z%Recompiler._to_py.<locals>.<listcomp>r r) z(%s)r: )rb r^ �listrr r% r` r< r )r r Zrepr r r r� � s
zRecompiler._to_pyc C s� || _ | j}|d� |d� t| jjpd�}t|�D ]4}| jj| }z |jd d� \}}W n ty= td| j|f ��w |d urFtd��|d||f � q|� |d| j f � |d | j
f � d | _
t | j �| _ d
d � | j D �}|d | �
d
�|��f � tdd � | j�� D ��} | jD ] }
| j|
} t| �dkr�|
dkr�|d|
| �
| �f � q�|dkr�|dd�dd � t|�D ��f � |d� d S )Nz# auto-generated filezimport _cffi_backendr rA r� zMnot implemented yet: ffi.include() of a C-based ffi inside a Python-based ffiz from %s import ffi as _ffi%dz ffi = _cffi_backend.FFI('%s',z _version = 0x%x,c S r7 r )r )r8 r* r r r r9 � � z3Recompiler.write_py_source_to_f.<locals>.<listcomp>z _types = %s,r) c S r� r r r� r r r r9 r� r r� z _%ss = %s,z _includes = (%s,),�, c S � g | ]}d | �qS )z_ffi%dr �r8 rt r r r r9 r� �))r� r� r% rI r� �ranger� r� r rJ rM rr r_ r� r< r� r[ r� r� r� ) r r� r� Z num_includesrt r� r� r� Z types_lstr� r� r� r r r r� � sR
� �����
�� z Recompiler.write_py_source_to_fc C s
| j | S r )r[ )r �typer r r � _gettypenum s
zRecompiler._gettypenumc C sl d}t |tj�rH|�� sH|�� r |jdkr d}d|j }n(t |tj�r,d|�d�f }n|�d�}d||j�dd�f }|d v rE| � t
� d
}nMt |tj �rX| � ||||� d S t |tj
�sdt |tj�rz| �d || �|�|f � | �d | � d S t |tj�r�d
|�d� }d| �|� }d}nt|��| �d||||f � | �d||�d�|f � | �d | � d S )Nr) �_BoolZ_cffi_to_c_intz, %sz(%s)_cffi_to_c_doublez(%s)_cffi_to_c_%sr$ r� �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()))rb r rd �is_complex_type�is_integer_typer
�UnknownFloatType�
get_c_name�replacerP �VERSION_CHAR16CHAR32re �"_convert_funcarg_to_c_ptr_or_arrayrf r� r� rg r' ) r rX �fromvar�tovar�errcodeZextraarg� converter�cnameZerrvaluer r r �_convert_funcarg_to_c sJ
�
�
�� �z Recompiler._convert_funcarg_to_cc C s2 t |tj�r|�d� |�d� |�d� d S d S )NzPy_ssize_t datasizez-struct _cffi_freeme_s *large_args_free = NULLzIif (large_args_free != NULL) _cffi_free_array_arguments(large_args_free);)rb r re �add)r rX � localvars� freelinesr r r �_extra_local_variablesC s
�z!Recompiler._extra_local_variablesc C s� | � d� | � d| �|�||f � | � d� | � d||�d�f � | � d| �|�||f � | � d� | � d| � | � d � d S )
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� )r rX r� r� r� r r r r� J s
�
���
z-Recompiler._convert_funcarg_to_c_ptr_or_arrayc C s> t |tj�rI|�� r|jdkrd||jf S t |tj�r!d|f S |jdkr@|�� s@|j�dd�}|dv r:| �t � d||f S d || �
|�f S t |tj tj f�r[d
|| �
|�f S t |tj
�rnd
|| �
t� |j��f S t |tj�r�|jd u r�td |�� |f ��d || �
|�f S t |tj�r�d || �
|�f S t|��)
Nr� z_cffi_from_c_int(%s, %s)z_cffi_from_c_double(%s)� long doubler$ r� r� 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)))rb r rd r� r
r� r� r� rP r� r� re rg � ArrayTyper� rl Zfldnames� TypeErrorr� rm r' )r rX �var�contextr� r r r �_convert_expr_from_cX sD
�
� �
�
�
�z Recompiler._convert_expr_from_cc C s | � |d|f �S )Nz
(*(%s *)0))� _global_type�r rX r
r r r �
_typedef_type{ rQ zRecompiler._typedef_typec C � | � | �||�� d S r )ry r� r� r r r �!_generate_cpy_typedef_collecttype~ � z,Recompiler._generate_cpy_typedef_collecttypec C � d S r r r� r r r �_generate_cpy_typedef_decl� r( z%Recompiler._generate_cpy_typedef_declc C s$ | j | }| jd �t||�� d S )Nr� )r[ r� r` rE )r rX r
r1 r r r � _typedef_ctx� s
zRecompiler._typedef_ctxc C sl | � ||�}| �||� t|dd �dkr | j||jd d� d S t|tj�r4| j|j|jj|j|d� d S d S )N�originZ unknown_type)�
approxname)r� � named_ptr) r� r� ro � _struct_ctxr
rb r ZNamedPointerType�totyper� r r r �_generate_cpy_typedef_ctx� s
��z$Recompiler._generate_cpy_typedef_ctxc C s0 | � |�� � |jr| js| � |� d S d S d S r )ry rz �ellipsisrK r� r r r �"_generate_cpy_function_collecttype� s �z-Recompiler._generate_cpy_function_collecttypec C s | j rJ �t|tj�s
J �|jr| �||� d S | j}t|j�}|dkr'd}n |dkr.d}nd}g }g }d| }t |j�D ]\} }
|�
|
� d| |�� |�
d| � q=d � |�} | p\d
} |j
rf|j
d } nd } d
| || f }
|d|j� |
�f � |d� d � |�}d}t|jtj�r�d }|d|||f � |d� |d� |d� |d||f � |d� d| }t |j�D ]\} }
|
� d| |�}|d| � q�t� }t� }|jD ] }
| �|
||� q�t|�D ] }|d|f � q�t|jtj��s d}d| }d|j� d|� }||� |d� nd }d }t|j�dk�rHtt|j��}|D ] } |d| � �q!|� |d |t|�t|�d � d d � |D ��f � |d � |� t |j�D ]\} }
| �|
d | d| d!� |� �qP|d"� |d#� d$d � tt|j��D �}d � |�}|d%|||f � |d&� |d'� |� |d(� |dk�r�|d)� |�r�|d*| �|jd+d,� � |D ] }|d-| � �q�|d.� n|D ] }|d-| � �q�|d/� |d0� |d� |d1� d2d3� }d4}g }g }d| }t |j�D ](\} }
d }||
��r�d5}d6}|
� d7|| f |�}|�
|� |�
d8|| f � �q�|j}||��r8d| }|� d9|�}|�d|� tj}d }d:}d6}|�r{d � |�} | �pDd
} d;| || f }
|d|� |
�f � |d� |�ra||� d � |�}|d%|||f � |�rv|d<� |d� n|d=||f � |d>� |� d S )?Nr Znoargr Zarg0ra �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))c S r� )z&arg%dr r� r r r r9 � r� z:Recompiler._generate_cpy_function_decl.<locals>.<listcomp>r� zarg%dz return NULLz Py_BEGIN_ALLOW_THREADSz _cffi_restore_errno();c S r� )r r r� r r r r9 � r� z { %s%s(%s); }z _cffi_save_errno();z Py_END_ALLOW_THREADSz (void)self; /* unused */z (void)noarg; /* unused */z pyresult = %s;�resultz result typez z return pyresult;z Py_INCREF(Py_None);z return Py_None;r� c S s t | tj�pt | tj�o| �� S r )rb r rl �
PrimitiveTyper� )r� r r r �need_indirection s �z@Recompiler._generate_cpy_function_decl.<locals>.need_indirectionFr Tz %sx%dz%sx%dz *resultz
*result = z%s_cffi_f_%s(%s)z return result;z # define _cffi_f_%s _cffi_d_%sr� )rK rb r rg r� �_generate_cpy_constant_declr� r% ra rn r` r� r< �abir rc r� r� r] r� r� r� �insertZ void_type)r rX r
r� �numargs�argname� argumentsZcall_argumentsr� rt r� �repr_argumentsr �name_and_argumentsZ result_code�argr� r� r� Z result_decl�rngZfreeliner �
differenceZ indirectionZ tp_resultr r r �_generate_cpy_function_decl� s
��
�
�
z&Recompiler._generate_cpy_function_declc C s� |j r| js| �||� d S | j|�� }t|j�}| jr t}n|dkr't}n |dkr.t }nt
}| j d � t
|d| t||�d| d�� d S )Nr r r� z
_cffi_f_%sz
_cffi_d_%s)r )r� rK �_generate_cpy_constant_ctxr[ rz r% ra ZOP_DLOPEN_FUNCZOP_CPYTHON_BLTN_NZOP_CPYTHON_BLTN_OZOP_CPYTHON_BLTN_Vr� r` r
rq )r rX r
r1 r Z meth_kindr r r �_generate_cpy_function_ctx; s$
��z%Recompiler._generate_cpy_function_ctxc C sT t |tj�r(|j}|dkr|�d�}d||f }| �|d| |j�}t�||�}|S )N�...r z _cffi_array_len(((%s)0)->%s)�%s[0])rb r r� rh r� r~ r� )r Z tp_struct�
field_nameZtp_field�
actual_lengthZptr_struct_name�tp_itemr r r r~ Q s
� � zRecompiler._field_typec C s0 | � |� | jr|�� D ] }| �|� q d S d S r )ry rK �anonymous_struct_fields�_struct_collecttype)r rX �fldtyper r r r ] s
�z Recompiler._struct_collecttypec C s\ |j d u rd S | j}d|f }|d� |d||f � |d� |d� |d� | �|�D ]p\}}}} zP|�� s;|dkrJ|dkrH|d |||f � W q,t|tj�rq|jd u sZ|jd
krq|j}|d }t|tj�rq|jd u sZ|jd
ksZ|d |j d
d| | d�|f � W q, t
y� }
z|dt |
� � W Y d }
~
q,d }
~
ww |d� |d||f � |� d S )Nz_cffi_checkfld_%sZ_CFFI_UNUSED_FNzstatic void %s(%s *p)r� z8 /* only to generate compile-time warnings or errors */z
(void)p;r r) z> (void)((p->%s) | 0); /* check that '%s.%s' is an integer */r z[0]z { %s = &p->%s; (void)tmp; }z*tmpzfield %r�r� z
/* %s */r� z(struct _cffi_align_%s { char x; %s y; };) r{ r� rx r� rb r r� rh r� r� r r^ ) r rX r� r� r� Z
checkfuncname�fnameZftyper" �fqualr� r r r � _struct_decld sD
� � � ��
zRecompiler._struct_declNc C s� | j | }d }g }t|tj�r|�d� |jd u r |�d� d}|| jjjvr^|d u s2|| jjjvr^|jd u r8n|j sAt
|� � �rBn|�d� |j r]|j dkrXt
d||j f ��|�d� n|�d� d }d
�|�pkd }g }|d u �r+t| �|��} | D ]x\}
} } }
| �||
| �} | �| d |j|
f � t}| d
kr�t}d| }n"|d u s�t| tj�r�| jd u r�d}nd|d u r�|�d�n|j|
f }|d u s�| d
kr�d}n|d ur�d|j|
f }n d|�d�|
f }|�t|
||| t|| j | ��� q|t| j d �}| j d � |� |d u �rd}d}d}n#|d u�r d|jf }d}n
d|f }d |f }d }nd}d }d }|}| j d �t |j|||||||�� | j � |� d S ) NZ
_CFFI_F_UNIONZ_CFFI_F_OPAQUEZopaqueZ_CFFI_F_CHECK_FIELDSr z�%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'r z
%d /* bits */z
(size_t)-1zsizeof(((%s)0)->%s)r z"((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� )!r[ rb r � UnionTyper` r{ rI r| r} �partial�anyr �packedr' r< r� rx r~ �_check_not_opaquer
r+ r, r� rh r� r rq r% r� �extendr0 r� r� )r rX r� r� r� r1 Zreason_for_not_expandingr2 r6 rw Zfldnamer r" r r* r �offsetr5 Zalignr4 r r r r� � s�
��
�
�
�
��
��
��zRecompiler._struct_ctxc C sH t |tj�r|j}t |tj�st |tj�r |jd u r"td| ��d S d S )Nz0%s is of an opaque type (not declared in cdef()))rb r r� r� rl r{ r� )r rX �locationr r r r( � s ���z Recompiler._check_not_opaquec C s� t | j�� �}|jdd� d� |D ]K\}}|| jvr\|jr$td|f ��|j�d�r;|jdd � � � r;|jdd � }n|jdkrN|j
dkrNd}| � |d� ntd |f ��| � |d |� qd S )
Nc S s | d S )Nr r )Ztp_orderr r r rY � s z7Recompiler._add_missing_struct_unions.<locals>.<lambda>rR zDinternal inconsistency: %r is partial but was not seen at this point�$r Z_IO_FILE�FILEzinternal inconsistency: %r)
r� rj r� r� r� r% r' r
�
startswith�isdigitZ forcenamer� r� )r r� rX �orderr� r r r r� � s&
� ���z%Recompiler._add_missing_struct_unionsc C � | � |� d S r )r r� r r r � _generate_cpy_struct_collecttype� � z+Recompiler._generate_cpy_struct_collecttypec C s. |� d�}d|v r||�dd�fS |d| fS )Nr) r$ r� )r� r� )r rX r� r r r �
_struct_names� s
zRecompiler._struct_namesc C � | j |g| �|��R � d S r )r r4 r� r r r �_generate_cpy_struct_decl � z$Recompiler._generate_cpy_struct_declc C r5 r )r� r4 r� r r r �_generate_cpy_struct_ctx r7 z#Recompiler._generate_cpy_struct_ctxc C s* t |tj�r| �||� d S | �|� d S r )rb r rm � _generate_cpy_enum_collecttyper r� r r r �#_generate_cpy_anonymous_collecttype s z.Recompiler._generate_cpy_anonymous_collecttypec C s0 t |tj�r
| �|� d S | �||d| � d S �NZtypedef_)rb r rm �_generate_cpy_enum_declr r� r r r � _generate_cpy_anonymous_decl s z'Recompiler._generate_cpy_anonymous_declc C s2 t |tj�r| �||� d S | �||d| � d S r; )rb r rm � _enum_ctxr� r� r r r �_generate_cpy_anonymous_ctx s z&Recompiler._generate_cpy_anonymous_ctx�constc C s ||f| j v rtd||f ��| j �||f� | j}d||f }|r\|d| � |d� |d|f � |d||f � |d urS|dkrHd|f }|d |f � |d
� |d � |d � n |d u sbJ �|d
| � |d� |d|�d�|f � |d � |� d S )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 */r z%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� r r� r� r� )r �is_intr
rX �categoryr r� �funcnamer r r �_generate_cpy_const& s6
� �
z Recompiler._generate_cpy_constc C s$ |� � }|r | jr| �|� d S d S r )r� rK ry �r rX r
rA r r r �"_generate_cpy_constant_collecttypeD s
�z-Recompiler._generate_cpy_constant_collecttypec C s |� � }| �|||� d S r )r� rD rE r r r r I s z&Recompiler._generate_cpy_constant_declc C s^ | j s
|�� r
ttd�}n| j rt}nt}| j| }t||�}| jd �t |d| |�� d S )Nr� r� �_cffi_const_%s)
rK r� rq �OP_CONSTANT_INTZOP_DLOPEN_CONSTZ OP_CONSTANTr[ r� r` r
)r rX r
r Z
const_kindr1 r r r r M s
�z%Recompiler._generate_cpy_constant_ctxc C r1 r )ry r� r r r r9 ] r3 z)Recompiler._generate_cpy_enum_collecttypec C s |j D ]}| �d|� qd S )NT)� enumeratorsrD )r rX r
�
enumeratorr r r r< ` s
�z"Recompiler._generate_cpy_enum_declc C s� | j | }ttd�}| jr|�� t|j|j�D ]\}}| jd � t
|d| ||d�� q|d urAd|vrA| jsAd| }d| }n|� | j g �} | j �
| �}tt| j �| d��dk �}d �|j�}
| jd
� t|j||||
�� d S ) Nr� r� rG �r r, r# z
((%s)-1) <= 0r r: r� )r[ rq �OP_ENUMrK Zcheck_not_partial�ziprI Z
enumvaluesr� r` r
Zbuild_baseinttyperI �sizeof�int�castr< r= r
) r rX r� r1 r rJ Z enumvaluer r? Zbasetpr@ r r r r> d s(
��
�zRecompiler._enum_ctxc C s | � ||�� � d S r )r> r� r� r r r �_generate_cpy_enum_ctxy rG z!Recompiler._generate_cpy_enum_ctxc C r� r r r� r r r � _generate_cpy_macro_collecttype r( z*Recompiler._generate_cpy_macro_collecttypec C s& |dkrd }n|}| j d||d� d S )Nr TrK )rD )r rX r
r r r r �_generate_cpy_macro_decl� s z#Recompiler._generate_cpy_macro_declc C sT |dkr| j rtd|f ��d }n|}ttd�}| jd �t|d| ||d�� d S )Nr zGcannot use the syntax '...' in '#define %s ...' when using the ABI moder� r� rG rK )rK r rq rH r� r` r
)r rX r
r r r r r �_generate_cpy_macro_ctx� s ��
��z"Recompiler._generate_cpy_macro_ctxc C sF t |tj�r!|j}|dkrd|f }| �|jd| �}t�||�}|S )Nr z_cffi_array_len(%s)r )rb r r� rh r� r� )r rX Z global_namer r r r r r� � s
zRecompiler._global_typec C r� r )ry r� r� r r r �"_generate_cpy_variable_collecttype� r� z-Recompiler._generate_cpy_variable_collecttypec C s� | j }| �||�}t|tj�r|jd u r|j}d}nd}d|f }|d|j|| jd� � |d� |d||f � |d� |� d S ) Nr) �&z*_cffi_var_%s(void)�static r r� z return %s(%s);r� ) r� r� rb r r� rh r� r� r� )r rX r
r� Z ampersandr� r r r �_generate_cpy_variable_decl� s
z&Recompiler._generate_cpy_variable_declc C sL | � ||�}| j| }| jrt}nt}| jd �t|d| t||��� d S )Nr� z _cffi_var_%s) r� r[ rK Z
OP_GLOBAL_VARZOP_GLOBAL_VAR_Fr� r` r
rq )r rX r
r1 r* r r r �_generate_cpy_variable_ctx� s
�z%Recompiler._generate_cpy_variable_ctxc C s t |tj�sJ �| �|� d S r )rb r rg ry r� r r r �'_generate_cpy_extern_python_collecttype� s z2Recompiler._generate_cpy_extern_python_collecttypec C s | j }t|jtj�r
d}nd| }d|j�d|�f }|d| � |d| j||f � |� g }d| }t|j�D ]\}} | �d| |�}
|� |
� q9d �
|�} | pSd
} d || f } |j d krcd
| } dd� }
t t
|j�d d�}|
|j�r{t |d�}t|jtj�r�d|j�d�||j�d�|f }|d||j�| �f � |d� |d| � |d� t|j�D ])\}} d| }
t| tj�s�|
| �r�d|
}
t�| �} |d| �d�|d |
f � q�|d| � t|jtj�s�|d |j�d�f � |d � |� | jd 7 _d S ) 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 c S s t | tj�o
| jdkS )Nr� )rb r r r
rW r r r �may_need_128_bits� s �z9Recompiler._extern_python_decl.<locals>.may_need_128_bitsrC � z!sizeof(%s) > %d ? sizeof(%s) : %dz%s%sr� z
char a[%s];z char *p = a;za%drV z *(%s)(p + %d) = %s;r z, _cffi_call_python(&_cffi_externpy__%s, p);z return *(%s)p;r� r )r� rb r r rc r� rJ rn ra r` r< r rN r% rl re r� )r rX r
Z
tag_and_spacer� Zsize_of_resultr� r rt r� r r r
r\ Z size_of_ar r r �_extern_python_decl� sb � �
� �
z Recompiler._extern_python_declc C � | � ||d� d S )NrW �r^ r� r r r � _generate_cpy_extern_python_decl rQ z+Recompiler._generate_cpy_extern_python_declc C r_ )NzCFFI_DLLEXPORT r` r� r r r �#_generate_cpy_dllexport_python_decl
rQ z.Recompiler._generate_cpy_dllexport_python_declc C r_ )Nr) r` r� r r r �'_generate_cpy_extern_python_plus_c_decl
rQ z2Recompiler._generate_cpy_extern_python_plus_c_declc C sR | j rtd��|jrtd��| j| }tt|�}| jd �t |d| ||�� d S )Nz,cannot use 'extern "Python"' in the ABI modez$a vararg function is extern "Python"r� z&_cffi_externpy__%s)
rK r r� r' r[ rq ZOP_EXTERN_PYTHONr� r` r
)r rX r
r1 r r r r � _generate_cpy_extern_python_ctx s �
�z*Recompiler._generate_cpy_extern_python_ctxc C s� | j }|d� t|t�s|�d�}n|�d� z|�d� W n
ty+ d| }Y nw |�d�D ]>}|}td�tu rAtt |�}nt
|�dd� }|d | � � � d
}|D ]}t |�d krc||� d
}|d |f 7 }qU||� q1d S )
Nz=// # NB. this is not a string because of a size limit in MSVCzutf-8�asciis # -*- encoding: utf8 -*-
Tz//r r� z// r) �L z%d,)
r� rb �bytes�encode�decode�UnicodeDecodeError�
splitlinesr� �map�ordre �rstripr% )r �sr� �liner4 Z printed_line�cr r r r� s0
�
�z)Recompiler._print_string_literal_in_arrayc C s t tt�| j|< d S r )rq � OP_PRIMITIVEZ PRIM_VOIDr_ �r rX r� r r r �_emit_bytecode_VoidType= rG z"Recompiler._emit_bytecode_VoidTypec C s t |j }tt|�| j|< d S r )ZPRIMITIVE_TO_INDEXr
rq rr r_ )r rX r� rD r r r � _emit_bytecode_PrimitiveType@ �
z'Recompiler._emit_bytecode_PrimitiveTypec C s( d|j |j |j f }tt|�| j|< d S )Nzl_cffi_prim_int(sizeof(%s), (
((%s)-1) | 0 /* check that %s is an integer type */
) <= 0)�r
rq rr r_ �r rX r� ro r r r �!_emit_bytecode_UnknownIntegerTypeD s �z,Recompiler._emit_bytecode_UnknownIntegerTypec C s$ d|j |j f }tt|�| j|< d S )Nzd_cffi_prim_float(sizeof(%s) *
(((%s)1) / 2) * 2 /* integer => 0, float => 1 */
)rw rx r r r � _emit_bytecode_UnknownFloatTypeJ s
�z*Recompiler._emit_bytecode_UnknownFloatTypec C s� t t| j|j �| j|< |d7 }|jD ]$}| j| }||kr3t|tj�r+| � ||� nt t
|�| j|< |d7 }qt |j �}|j
d urT|j
dkrL|dO }ntd|j
f ��t t|�| j|< d S )Nr r[ rA zabi=%r)rq Z OP_FUNCTIONr[ r r_ ra rb r r ru r+ rO r� r r' ZOP_FUNCTION_END)r rX r� rs Z realindexr2 r r r � _emit_bytecode_RawFunctionTypeP s
z)Recompiler._emit_bytecode_RawFunctionTypec C s t t| j|j �| j|< d S r )rq �
OP_POINTERr[ r� r_ rs r r r �_emit_bytecode_PointerTypec r7 z%Recompiler._emit_bytecode_PointerTypec C s" |� � }tt| j| �| j|< d S r )rz rq r| r[ r_ )r rX r� �rawr r r � _emit_bytecode_FunctionPtrTypei s z)Recompiler._emit_bytecode_FunctionPtrTypec C s� | j |j }|jd u rtt|�| j|< d S |jdkr'tdt|��dd�f ��| j|d dks2J �tt |�| j|< td t|j��| j|d < d S )Nr zitype %s badly placed: the '...' array length can only be used on global arrays or on fields of structuresz/*...*/r rT )
r[ r� rh rq Z
OP_OPEN_ARRAYr_ r r^ r� ZOP_ARRAY)r rX r� Z
item_indexr r r �_emit_bytecode_ArrayTypem s
��� z#Recompiler._emit_bytecode_ArrayTypec C � | j | }tt|�| j|< d S r )rj rq ZOP_STRUCT_UNIONr_ )r rX r� Z struct_indexr r r �_emit_bytecode_StructType{ rv z$Recompiler._emit_bytecode_StructTypec C r� r )rk rq rL r_ )r rX r� Z
enum_indexr r r �_emit_bytecode_EnumType� rv z"Recompiler._emit_bytecode_EnumType�F)r) r )Nr@ N)Vr r r r� r rP rv rx ry r\ r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r~ r r r� r( r� r2 Z _generate_cpy_union_collecttyper4 r6 Z_generate_cpy_union_declr8 Z_generate_cpy_union_ctxr: r= r? rD rF r r r9 r< r> rQ rR rS rT r� rU rX rY rZ Z*_generate_cpy_dllexport_python_collecttypeZ._generate_cpy_extern_python_plus_c_collecttyper^ ra rb rc rd Z"_generate_cpy_dllexport_python_ctxZ&_generate_cpy_extern_python_plus_c_ctxr� rt ru ry rz r{ r} Z _emit_bytecode_ConstPointerTypeZ _emit_bytecode_NamedPointerTyper r� r� Z_emit_bytecode_UnionTyper� r r r r rH ~ s�
:
.
4-# #
"V
�
�6
� rH )r c s e Zd Z� fdd�Z� ZS )�NativeIOc s( t |t�r
|�d�}tt| ��|� d S )Nre )rb Zunicoderh �superr� r� )r ro �rp r r r� � s
zNativeIO.write)r r r r� �
__classcell__r r r� r r� � s r� c
C s: |r t d|f � t| ||d u d�}|�� |�� t� }|�||� |�� }z+t|d��}|�t |�d �|kr;t
�W d � n1 sEw Y |rPt d� W dS t
y� d|t � � f } t| d��
}|�
|� W d � n1 svw Y z
t �| |� W Y d S ty� t �|� t �| |� Y Y d S w w )
Nz
generating %s)rK r� r z(already up-to-date)Fz%s.~%d�wT)�printrH rv r� r� r� �getvaluer� �readr% �IOErrorr� �getpidr� �rename�OSError�unlink)
rI rJ r� Z target_file�verboseZ
recompilerr� �output�f1Ztmp_filer r r �_make_c_or_py_source� sB � � � � �
��r� Fc C s |d usJ �t | ||||�S r �r� )rI rJ r� Z
target_c_filer� r r r �
make_c_source� s
�r� c C s t | |d ||�S r r� )rI rJ Ztarget_py_filer� r r r �make_py_source� s
�r� c C sl |� d�}zt�tjj| g|d d� �R � � W n ty! Y nw |d |7 < tjj| g|�R � |fS )Nr� r� )r� r� �makedirsr� r< r� )Z outputdir�modname� extension�partsr r r �_modname_to_file� s
& �r� c C s* t ||�}| �|||f� t|||� |S r )ro r` �setattr)� patchlist�clsr
Znew_meth�oldr r r � _patch_meth� s
r� c C s$ t | �D ] \}}}t|||� qd S r )�reversedr� )r� r� r
Zold_methr r r �_unpatch_meths� s �r� c s` t jdkrddlm} t| |ddd� � t jdkr.ddlm} � fd d
�}t| |d |�� d S d S ) Nr r )� MSVCCompilerZ_remove_visual_c_refc S s |S r r )r Z
manifest_filer r r rY � � z&_patch_for_embedding.<locals>.<lambda>�darwin)� CCompilerc sB d| j v rt| j �| _ | j �d�}d| j |< � | g|�R i |��S )Nz-bundlez -dynamiclib)Z linker_sor� r� )r ra �kwdsrt �Zold_link_shared_objectr r �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� r r� r �_patch_for_embedding� s
�
��r� c sl ddl m} � �d�r)� d d� � tjdkr� d7 � ntjdkr%� d7 � n� d 7 � t| |d
� fd d �� d S )
Nr )� build_extz.*r" r z.dllr� z.dylibz.soZget_ext_filenamec s � S r r )r Zext_name��targetr r rY � r� z#_patch_for_target.<locals>.<lambda>)Zdistutils.command.build_extr� �endswithr� r� r� )r� r� r� r r� r �_patch_for_target� s
�r� r� T�.cc K s� t |t�s
|�d�}| jr| �| � |d ur�| jd u} | r"| �| � |d u r<t|||�\}}
|r5|g|
}
tj j
|
� }n|}| d u rK| rId| } nd} t j ||fi | ��}t
| ||||d�}|r�g }t�� }z?| rmt|� | dkrvt|| � |r�|dkrd}nd}td|tj �|�f � t�|� t �d|||
�}W t�|� t|� |S t�|� t|� w ||fS |d u r�t||d �\}}t| |||d�}|r�|S d |fS )
Nre z%s.*r )r� r� zthe current directory isz setting the current directory toz%s %rz.py)rb r^ rh Z_windows_unicodeZ_apply_windows_unicoder� Z_apply_embedding_fixr� r� r� r< r Z
get_extensionr� �getcwdr� r� r� �abspath�chdir�compiler� r� )rI rJ r� ZtmpdirZcall_c_compilerZc_fileZsource_extensionZextradirZcompiler_verboser� �debugr� Z embeddingr� Z
ext_c_file�ext�updatedr� �cwd�msgZoutputfilenamer� r r r � recompile� sn
�
�
�
�
�r� r� )r� TNr� Nr NN)!r� r� �ior) r r �errorr Z cffi_opcoderL r� r� r� � version_infor� r
r r0 r= rE rH �StringIOr� �BytesIOr� r� r� r� r� r� r� r� r� r r r r �<module> sR �
�