XylotrechusZ
3
M��Z�� � @ s d dl Z d dlZd dlZddlmZmZ ddlmZ ddlT dZ dZ
dZ G d d
� d
�Z G d d � d �Z
G d
d� d�ZG dd� d�ZG dd� d�ZG dd� d�Zejd0kr�ejZnG dd� dej�Zdd� Zd1dd �Zd2d d �Zd d � Zd!d"� Zd#d$� Zd%d&� Z d'd(� Z d3d,d-�Z d.d/� Z dS )4� N� )� ffiplatform�model)�VerificationError)�*i& i' i( c @ s& e Zd Zd dd�Zdd� Zdd� ZdS )
�
GlobalExprr c C s" || _ || _|| _|| _|| _d S )N)�name�address�type_op�size� check_value)�selfr r r
r r � r �"/usr/lib64/python3.6/recompiler.py�__init__ s
zGlobalExpr.__init__c C s d| j | j| jj� | 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� | 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 s" || _ || _|| _|| _|| _d S )N)r � field_offset�
field_size�fbitsize�
field_type_op)r
r r r r r r r r r s
zFieldExpr.__init__c C s> dt | j� }d| j| jf d|| jf d|| jj� f S )N� z { "%s", %s,
z %s %s,
z %s %s },)�lenr r r r r )r
Zspacesr r r r # s zFieldExpr.as_c_exprc C s t �d S )N)�NotImplementedError)r
r r r r ) s zFieldExpr.as_python_exprc C sD | j jtkrd}n | j jtkr*t| j�}nt�d| j 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 @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�StructUnionExprc C s4 || _ || _|| _|| _|| _|| _|| _|| _d S )N)r �
type_index�flagsr � alignment�comment�first_field_index�c_fields) r
r r& r' r r( r) r* r+ r r r r 8 s zStructUnionExpr.__init__c C sP d| j | j| jf d| j| jf d| jt| j�f | jrFd| j nd d S )Nz { "%s", %d, %s,z
%s, %s, z%d, %d z /* %s */ r z},) r r& r' r r( r* r r+ r) )r
r r r r C s zStructUnionExpr.as_c_exprc C s> t | jt�}dd� | jD �}dt| j�t|�| jdj|�f S )Nc S s g | ] }|j � �qS r )r$ )�.0Zc_fieldr r r �
<listcomp>L s z2StructUnionExpr.as_python_expr.<locals>.<listcomp>z(b'%s%s%s',%s)�,)�evalr' ZG_FLAGSr+ r# r& r �join)r
r' Z fields_exprr r r r J s
z StructUnionExpr.as_python_exprN)r r r r r r r r r r r% 7 s r% c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�EnumExprc C s" || _ || _|| _|| _|| _d S )N)r r& r �signed�allenums)r
r r& r r2 r3 r r r r U s
zEnumExpr.__init__c C s d| j | j| j| j| jf S )Nz1 { "%s", %d, _cffi_prim_int(%s, %s),
"%s" },)r r& r r2 r3 )r
r r r r \ s zEnumExpr.as_c_exprc C sN dt dtd td
td td td
tdti| j| j f }dt
| j �t
|�| j | j
f S )Nr r � � � zb'%s%s%s\x00%s')r r )r r )r4 r )r4 r )r5 r )r5 r )r6 r )r6 r )Z
PRIM_UINT8Z PRIM_INT8Z PRIM_UINT16Z
PRIM_INT16Z PRIM_UINT32Z
PRIM_INT32Z PRIM_UINT64Z
PRIM_INT64r r2 r# r& r r3 )r
�
prim_indexr r r r a s
zEnumExpr.as_python_exprN)r r r r r r r r r r r1 T s r1 c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )� TypenameExprc C s || _ || _d S )N)r r& )r
r r& r r r r m s zTypenameExpr.__init__c C s d| j | jf S )Nz { "%s", %d },)r r& )r
r r r r q s zTypenameExpr.as_c_exprc C s dt | j�| jf S )Nzb'%s%s')r# r& r )r
r r r r t s zTypenameExpr.as_python_exprN)r r r r r r r r r r r8 l s r8 c @ s� e Zd ZdZd�dd�Zdd� Zdd� Zd d
� Zd d � Zd
ddddgZ dd� Z
d�dd�Z dd� Z dd� Z
dd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd+d,� Zd-d.� Zd/d0� Zd1d2� Zd3d4� Zd5d6� Zd7d8� Z d9d:� Z d;d<� Z d=d>� Z d?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� Z<dudv� Z=e= Z>Z?dwdx� Z@dydz� ZAd{d|� ZBd}d~� ZCdd�� ZDeD ZEZFd�d�� ZGd�d�� ZHd�d�� ZId�d�� ZJd�d�� ZKd�d�� ZLd�d�� ZMeMZNeMZOd�d�� ZPd�d�� ZQd�d�� ZReRZSd�d�� ZTdAS )��
Recompilerr Fc C s || _ || _|| _t| _d S )N)�ffi� module_name�target_is_python� VERSION_BASE�_version)r
r: r; r< r r r r ~ s zRecompiler.__init__c C s t | j|�| _d S )N)�maxr> )r
Zverr r r �
needs_version� s zRecompiler.needs_versionc C s� i | _ | jd� t| j td�}g | _xt|D ]l}|jr*t| j�| j |< | jj|� x8|jD ].}| j | d krzt| j�| j |< | jj|� qXW | jjd� q*W xZ|D ]R}|j r�| j | d kr�t| j�| j |< | jj|� |j r�|j
d k r�| jjd� q�W i | _ i | _ x@|D ]8}t
|tj��r&d | j |< nt
|tj��rd | j |< �qW x.tt| j dd� d��D ]\}}|| j |< �q\W x.tt| j dd� d��D ]\}}|| j |< �q�W x0|D ](}t| d|jj �}||| j | � �q�W x| jD ]}�q�W t| j�| _d S ) NZ collecttype)�keyZENDZLENc S s | j S )N)r )�tpr r r �<lambda>� s z/Recompiler.collect_type_table.<locals>.<lambda>c S s | j S )N)r )rB r r r rC � s Z_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)r
Z all_declsrB �tp1�i�methodr r r r �collect_type_table� sL
z Recompiler.collect_type_tablec C s� t |tj�s2t |t�r.x|D ]}| j|� q W d S || jkr�d | j|< t |tj�rb| j|j� � ntt |tj�r�|j d k r�|| j
j j kr�xN|j
� D ] \}}}}| j| j|||�� q�W n x |j� D ]\}}| j|� q�W d S )N)rN r ZBaseTypeByIdentityrT �_do_collect_typerD �FunctionPtrType�as_raw_functionrO �fldtypesr: �_parser�_included_declarations�
enumfields� _field_typeZ
_get_items)r
rB �xZname1rU �_r r r rY � s
zRecompiler._do_collect_typec
C s� | j jjj� }x�t|�D ]�\}\}}|jdd�\}}yt| d||f �}W n tk
rj td| ��Y nX y|| _ |||� W q t
k
r� } zt j | |� � W Y d d } ~ X qX qW d S )Nr r z_generate_cpy_%s_%sz"not implemented in recompile(): %r)
r: r] Z
_declarations�itemsrF �splitrR �AttributeErrorr �_current_quals� Exceptionr Zattach_exception_info)
r
� step_name�lstr rB �qualsZkindZrealnamerW �er r r rE � s zRecompiler._generate�global�field� struct_union�enum�typenamec C s� i | _ x| jD ]}g | j |< qW t� | _| jd� | j� x>| jD ]4}| j | }|dkrh|jdd� d� t|�| j |< qBW | j d }x| jj � D ]\}}q�W | j d }x| j
j � D ]\}}q�W d S )NZctxrm c S s | j S )N)r )�entryr r r rC � s z0Recompiler.collect_step_tables.<locals>.<lambda>)rA rn ro ) �_lsts� ALL_STEPS�set�_seen_struct_unionsrE �_add_missing_struct_unions�sortrT rL rc rM )r
rh ri rB rV r r r �collect_step_tables� s$
z Recompiler.collect_step_tablesr c C s | j j|d � d S )N�
)�_f�write)r
Zwhatr r r �_prnt s zRecompiler._prntc C s" | j r| j|� n | j||� d S )N)r< �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 tjjtjjt�|�d�}|j� }|j� |S )N�r)�open�os�pathr0 �dirname�__file__� readlines�close)r
�filename�g�linesr r r �_rel_readlines s zRecompiler._rel_readlinesc C sz || _ | j}| jjd k r |d� | jd�}|jd�}| jd�|||d �< |dj|�� | jjd�dI }| jjd k �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� t dd� | 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| �|
| < |
| d k�r|d | | f � x| D ]}
||
j� � �qBW |d � |� �qW | jj�r|d � xr| jjD ]f}y|jd d!� \}}W n( tk
�r� t d"| j|f ��Y nX |d k�r�t d#��|d$|f � �q�W |d%� |d � |� |d&� |d'� x8| jD ].} |
| d k�r@|d(| � n |d)| � �q W x.| 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| j f � |d<� |d� |d2� |dF|f � |d4� |dG| j| j f � |d<� |d� |� |d0� |dH� |d� d | _ d S )JNz#define _CFFI_USE_EMBEDDINGz_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_%sz#elsez*# define _CFFI_PYTHON_STARTUP_FUNC init%sz#endifz _embedding.hz#include "_cffi_errors.h"
z_cffi_errors.hz>/************************************************************/z static void *_cffi_types[] = {c S s g | ]\}}||f�qS r r )r, rB rV r r r r- G s 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[] = {r4 zQffi 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 */rm 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 >= 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);z init%s(void)z. _cffi_init("%s", 0x%x, &_cffi_type_context);z # pragma GCC visibility pop���) rz r| r: �
_embeddingr� �indexr0 r; rd � _print_string_literal_in_arrayr@ �VERSION_EMBEDDED�dictrD rc rQ rH � _get_c_namer rt �_seen_constantsrE rs rr r �_included_ffis�_assigned_sourcere r �
_num_externpyr> )r
r r� �prntr� rV Zbase_module_name�typeindex2typer r) Znumsrh ri rq �ffi_to_include�included_module_name�included_sourcer' r r r r~ s
z Recompiler.write_c_source_to_fc sb t |t�rd|f S t |ttf�rZ� fdd�|D �}t|�dkrJ|jd� ddj|�f S |j� S )Nzb'%s'c s g | ]}� j |��qS r )�_to_py)r, �item)r
r r r- � s z%Recompiler._to_py.<locals>.<listcomp>r r z(%s)r. )rN rG �listrT r rI r0 r )r
ra Zrepr )r
r r� � s
zRecompiler._to_pyc
C s� || _ | j}|d� |d� t| jjp(f �}xzt|�D ]n}| jj| }y|jd d� \}}W n& tk
r� td| j|f ��Y nX |d k r�td��|d||f � q6W |� |d| j f � |d| j
f � d | _
t | j �| _ d d
� | j D �}|d | j
d j|��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� d S )Nz# auto-generated filezimport _cffi_backendr4 zQffi 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 ffiz from %s import ffi as _ffi%dz ffi = _cffi_backend.FFI('%s',z _version = 0x%x,c S s g | ] }|j � �qS r )r )r, r r r r r- � s z3Recompiler.write_py_source_to_f.<locals>.<listcomp>z _types = %s,r c S s g | ]\}}||f�qS r r )r, rB rV r r r r- � s r rm z _%ss = %s,z _includes = (%s,),z, c S s g | ] }d | �qS )z_ffi%dr )r, rV r r r r- � s �))rz r| r r: r� �ranger� re r r; r> rT rH r� r0 r� rD rc rs rr ) r
r r� Z num_includesrV r� r� r� Z types_lstr� rh ri r r r r} � sB
z Recompiler.write_py_source_to_fc C s
| j | S )N)rD )r
�typer r r � _gettypenum s zRecompiler._gettypenumc C sp d}t |tj�r�|j� r�|j� r<|jdkr<d}d|j }nPt |tj�rZd|jd�f }n2|jd�}d||jjdd�f }|dkr�| j t
� d }n�t |tj �r�| j ||||� d S t |tj
�s�t |tj�r�| jd || j|�|f � | jd
| � d S t |tj��r&d|jd� }d| j|� }d}nt|��| jd||||f � | jd||jd�|f � | jd
| � d S )Nr �_BoolZ_cffi_to_c_intz, %sz(%s)_cffi_to_c_doublez(%s)_cffi_to_c_%sr rb �char16_t�char32_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� )rN r �BasePrimitiveType�is_complex_type�is_integer_typer �UnknownFloatType�
get_c_name�replacer@ �VERSION_CHAR16CHAR32� PointerType�"_convert_funcarg_to_c_ptr_or_arrayZStructOrUnionOrEnumr| r� rZ r ) r
rB �fromvar�tovar�errcodeZextraargZ converter�cnameZerrvaluer r r �_convert_funcarg_to_c s@
z Recompiler._convert_funcarg_to_cc C s t |tj�r|jd� d S )NzPy_ssize_t datasize)rN r r� �add)r
rB � localvarsr r r �_extra_local_variables3 s z!Recompiler._extra_local_variablesc C s� | j d� | j d| j|�||f � | j d� | j d� | j d| � | j d||jd�f � | j d|f � | j d || j|�|f � | j d| � | j d
� d S ) Nz1 datasize = _cffi_prepare_pointer_call_argument(z( _cffi_type(%d), %s, (char **)&%s);z if (datasize != 0) {z if (datasize < 0)z %s;z& %s = (%s)alloca((size_t)datasize);r z, memset((void *)%s, 0, (size_t)datasize);zL if (_cffi_convert_array_from_object((char *)%s, _cffi_type(%d), %s) < 0)z })r| r� r� )r
rB r� r� r� r r r r� 7 s
z-Recompiler._convert_funcarg_to_c_ptr_or_arrayc C sL t |tj�r�|j� r,|jdkr,d||jf S t |tj�rBd|f S |jdkr�|j� r�|jjdd�}|dkrv| jt � d ||f S d
|| j
|�f S n�t |tj tj f�r�d || j
|�f S t |tj
�r�d || j
tj |j��f S t |tj��r |jd k�rtd |j� |f ��d
|| j
|�f S t |tj��r@d
|| j
|�f S t|��d S )Nr� z_cffi_from_c_int(%s, %s)z_cffi_from_c_double(%s)z long doubler rb 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)))r� r� )rN r r� r� r r� r� r� r@ r� r� r� rZ � ArrayTyper� rO Zfldnames� TypeErrorr� rP r )r
rB �var�contextr� r r r �_convert_expr_from_cG s8
z Recompiler._convert_expr_from_cc C s | j |d|f �S )Nz
(*(%s *)0))� _global_type)r
rB r r r r �
_typedef_typej s zRecompiler._typedef_typec C s | j | j||�� d S )N)rY r� )r
rB r r r r �!_generate_cpy_typedef_collecttypem s z,Recompiler._generate_cpy_typedef_collecttypec C s d S )Nr )r
rB r r r r �_generate_cpy_typedef_declp s z%Recompiler._generate_cpy_typedef_declc C s$ | j | }| jd jt||�� d S )Nrp )rD rr rI r8 )r
rB r r& r r r � _typedef_ctxs s
zRecompiler._typedef_ctxc C sf | j ||�}| j||� t|dd �dkr<| j||jd d� n&t|tj�rb| j|j|jj|j|d� d S )N�originZ unknown_type)�
approxname)r� � named_ptr) r� r� rR � _struct_ctxr rN r ZNamedPointerType�totype)r
rB r r r r �_generate_cpy_typedef_ctxw s z$Recompiler._generate_cpy_typedef_ctxc C s* | j |j� � |jr&| j r&| j |� d S )N)rY r[ �ellipsisr< )r
rB r r r r �"_generate_cpy_function_collecttype� s z-Recompiler._generate_cpy_function_collecttypec C s� |j r| j||� d S | j}t|j�}|dkr4d}n|dkrBd}nd}g }g }d| }x:t|j�D ],\} }
|j|
jd| |�� |jd| � qbW d j|�} | p�d
} |j r�|j d } nd } d
| || f }
|d|j
j|
�f � |d� d j|�}d}t |j
t j
��rd }|d|||f � |d� |d� |d� |d||f � |d� d| }x4t|j�D ]&\} }
|
jd| |�}|d| � �q^W t� }x |jD ]}
| j|
|� �q�W x |D ]}|d|f � �q�W t |j
t j
��sd}d| }d|j
jd|� }||� nd }d }t|j�dk�r~tt|j��}x|D ]} |d| � �q.W |� |d|t|�t|�d jd d � |D ��f � |d � |� x6t|j�D ](\} }
| j|
d | d| d � |� �q�W |d!� |d"� d#d � tt|j��D �}d j|�}|d$|||f � |d%� |d&� |� |d'� |dk�r0|d(� |�rP|d)| j|j
d*d+� � n|d,� |d-� |d� |d.� d/d0� }d1}g }g }d| }x^t|j�D ]P\} }
d }||
��r�d2}d3}|
jd4|| f |�}|j|� |jd5|| f � �q�W |j
}||��r.d| }|jd6|�}|jd|� t j}d }d7}d3}|�r�d j|�} | �pFd
} d8| || f }
|d|j|
�f � |d� |�r�||� d j|�}|d$|||f � |�r�|d9� |d� n|d:||f � |d;� |� d S )<Nr Znoargr Zarg0rJ zargument of %sz x%dzx%dz, �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 = z result of %sz resultz PyObject *arg%d;z1 if (!PyArg_UnpackTuple(args, "%s", %d, %d, %s))c S s g | ] }d | �qS )z&arg%dr )r, rV r r r r- � s z:Recompiler._generate_cpy_function_decl.<locals>.<listcomp>z return NULL;zarg%dz return NULLz Py_BEGIN_ALLOW_THREADSz _cffi_restore_errno();c S s g | ] }d | �qS )zx%dr )r, rV r r r r- � s z { %s%s(%s); }z _cffi_save_errno();z Py_END_ALLOW_THREADSz (void)self; /* unused */z (void)noarg; /* unused */z return %s;�resultz result typez Py_INCREF(Py_None);z return Py_None;z#elsec S s t | tj�p t | tj�o | j� S )N)rN r rO �
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_%sz#endif)r� �_generate_cpy_constant_declr| r rJ rQ rI r� r0 �abir� rN r �VoidTypert r� r� r� r� �insertZ void_type)r
rB r r� �numargsZargname� argumentsZcall_argumentsr� rV r� �repr_argumentsr� �name_and_argumentsZ result_code�argr� r� Z result_decl�rngr� �
differenceZ indirectionZ tp_resultr r r �_generate_cpy_function_decl� s�
z&Recompiler._generate_cpy_function_declc C s� |j r | j r | j||� d S | j|j� }t|j�}| jrBt}n |dkrPt}n|dkr^t }nt
}| j d j t
|d| t||�d| d�� d S )Nr r rl z
_cffi_f_%sz
_cffi_d_%s)r )r� r< �_generate_cpy_constant_ctxrD r[ r rJ ZOP_DLOPEN_FUNCZOP_CPYTHON_BLTN_NZOP_CPYTHON_BLTN_OZOP_CPYTHON_BLTN_Vrr rI r �CffiOp)r
rB r r& r� Z meth_kindr r r �_generate_cpy_function_ctx# s
z%Recompiler._generate_cpy_function_ctxc C sT t |tj�rP|j}|dkr0|jd�}d||f }| j|d| |j�}tj||�}|S )Nz...r z _cffi_array_len(((%s)0)->%s)z%s[0])rN r r� rK r� r` r� )r
Z tp_structZ
field_nameZtp_field�
actual_lengthZptr_struct_name�tp_itemr r r r` 9 s
zRecompiler._field_typec C s0 | j |� | jr,x|j� D ]}| j|� qW d S )N)rY r< �anonymous_struct_fields�_struct_collecttype)r
rB �fldtyper r r r� E s
z Recompiler._struct_collecttypec C s< |j d krd S | j}d|f }|d� |d||f � |d� |d� |d� x�|j� D ]�\}}}} y||j� sv|dkr�|d|||f � wXx2t|tj�r�|jd ks�|jd kr�|j}|d
}q�W |d |j d d
| | d�|f � W qX t
k
�r }
z|dt |
� � W Y d d }
~
X qXX qXW |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 z> (void)((p->%s) | 0); /* check that '%s.%s' is an integer */z...z[0]z { %s = &p->%s; (void)tmp; }z*tmpzfield %r)rj z
/* %s */r� z(struct _cffi_align_%s { char x; %s y; };) r\ r| r_ r� rN r r� rK r� r� r rG ) r
rB r� r� r� Z
checkfuncnameZfnameZftyper �fqualrk r r r � _struct_declL s6
&zRecompiler._struct_declNc C s� | j | }d }g }t|tj�r(|jd� |jd kr@|jd� d}|| jjjkr�|d ksd|| jjjkr�|jd krpn |j s�t
|j � �r�n
|jd� |j r�|jd� n|jd� d}dj
|�p�d }g }|d k�rZ| j } t|j| ��}
�x
|
D �] \} } }
}| j|| | �} | j| d
|j| f � t}|
d k�r6t}d |
}nL|d k�sZt| tj��r`| jd k�r`d
}n"d|d k�rv|jd�n|j| f }|d k�s�|
d k�r�d
}n,|d k �r�d|j| f }nd|jd�| f }|jt| |||
t|| j | ��� q�W t| j d �}| j d j |� |d k�r$d}d }d}n4|d k �r@d|jf }d}nd|f }d|f }d }nd
}d }d }|}| j d jt |j|||||||�� | j j |� d S ) NZ
_CFFI_F_UNIONZ_CFFI_F_OPAQUEZopaqueZ_CFFI_F_CHECK_FIELDSZ_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 rm z
(size_t)-2r4 Zunnamedzsizeof(*(%s)0)z-1 /* unknown alignment */z
sizeof(%s)z"offsetof(struct _cffi_align_%s, y)r rn ���r� r� )!rD rN r Z UnionTyperI r\ r: r] r^ �partial�anyr� Zpackedr0 r< r� r_ r` �_check_not_opaquer r! r" r� rK r� r r� r rr �extendr% ru r� )r
rB r� r� r� r& Zreason_for_not_expandingr' r+ Z expand_anonymous_struct_unionr_ Zfldnamer� r r� r r �offsetr* Zalignr) r r r r� m s�
zRecompiler._struct_ctxc C s> xt |tj�r|j}qW t |tj�r:|jd kr:td| ��d S )Nz0%s is of an opaque type (not declared in cdef()))rN r r� r� rO r\ r� )r
rB �locationr r r r� � s
z Recompiler._check_not_opaquec C s� 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$W d S )
Nc S s | d S )Nr r )Ztp_orderr r r rC � s z7Recompiler._add_missing_struct_unions.<locals>.<lambda>)rA zDinternal inconsistency: %r is partial but was not seen at this point�$r Z_IO_FILE�FILEzinternal inconsistency: %r)
r� rL rc rw ru r� r r �
startswith�isdigitZ forcenamer� r� )r
ri rB �orderr� r r r rv � s
z%Recompiler._add_missing_struct_unionsc C s | j |� d S )N)r� )r
rB r r r r � _generate_cpy_struct_collecttype� s z+Recompiler._generate_cpy_struct_collecttypec C s2 |j d�}d|kr"||jdd�fS |d| fS d S )Nr r rb )r� r� )r
rB r� r r r �
_struct_names� s
zRecompiler._struct_namesc C s | j |f| j|��� d S )N)r� r� )r
rB r r r r �_generate_cpy_struct_decl� s z$Recompiler._generate_cpy_struct_declc C s | j |f| j|��� d S )N)r� r� )r
rB r r r r �_generate_cpy_struct_ctx� s z#Recompiler._generate_cpy_struct_ctxc C s( t |tj�r| j||� n
| j|� d S )N)rN r rP � _generate_cpy_enum_collecttyper� )r
rB r r r r �#_generate_cpy_anonymous_collecttype� s z.Recompiler._generate_cpy_anonymous_collecttypec C s. t |tj�r| j|� n| j||d| � d S )N�typedef_)rN r rP �_generate_cpy_enum_declr� )r
rB r r r r � _generate_cpy_anonymous_decl� s z'Recompiler._generate_cpy_anonymous_declc C s0 t |tj�r| j||� n| j||d| � d S )Nr� )rN r rP � _enum_ctxr� )r
rB r r r r �_generate_cpy_anonymous_ctx� s z&Recompiler._generate_cpy_anonymous_ctx�constc C s� ||f| j kr td||f ��| j j||f� | j}d||f }|r�|d| � |d� |d|f � |d||f � |d k r�|dkr�d|f }|d |f � |d
� |d � |d � n2|d
| � |d� |d|jd�|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 rB �categoryr r� �funcnamer r r �_generate_cpy_const s0
z Recompiler._generate_cpy_constc C s" |j � }| s| jr | j|� d S )N)r� r< rY )r
rB r r r r r �"_generate_cpy_constant_collecttype& s z-Recompiler._generate_cpy_constant_collecttypec C s |j � }| j|||� d S )N)r� r )r
rB r r r r r r� + s z&Recompiler._generate_cpy_constant_declc C s` | j r |j� r ttd�}n$| j r(t}nt}| j| }t||�}| jd jt |d| |�� d S )Nr rl z_cffi_const_%sr� )
r< r� r� �OP_CONSTANT_INTZOP_DLOPEN_CONSTZ OP_CONSTANTrD rr rI r )r
rB r r
Z
const_kindr& r r r r� / s
z%Recompiler._generate_cpy_constant_ctxc C s | j |� d S )N)rY )r
rB r r r r r� ? s z)Recompiler._generate_cpy_enum_collecttypec C s x|j D ]}| jd|� qW d S )NT)� enumeratorsr )r
rB r �
enumeratorr r r r� B s z"Recompiler._generate_cpy_enum_declc C s� | j | }ttd �}| jr"|j� x:t|j|j�D ](\}}| jd j t
|d| ||d�� q2W |d k r�d|kr�| j r�d| }d| }n4|j | j g �} | j j
| �}tt| j j| d ��dk �}d j|j�}
| jd
j t|j||||
�� d S )
Nr rl z_cffi_const_%s)r r� z
sizeof(%s)z
((%s)-1) <= 0r r. ro r� r� )rD r� �OP_ENUMr< Zcheck_not_partial�zipr Z
enumvaluesrr rI r Zbuild_baseinttyper: Zsizeof�int�castr0 r1 r ) r
rB r� r& r
r
Z enumvaluer r2 Zbasetpr3 r r r r F s"
zRecompiler._enum_ctxc C s | j ||j� � d S )N)r r� )r
rB r r r r �_generate_cpy_enum_ctx[ s z!Recompiler._generate_cpy_enum_ctxc C s d S )Nr )r
rB r r r r � _generate_cpy_macro_collecttypea s z*Recompiler._generate_cpy_macro_collecttypec C s& |dkrd }n|}| j d||d� d S )Nz...T)r )r )r
rB r r r r r �_generate_cpy_macro_decld s z#Recompiler._generate_cpy_macro_declc C sT |dkr"| j r td|f ��d }n|}ttd�}| jd jt|d| ||d�� d S )Nz...zGcannot use the syntax '...' in '#define %s ...' when using the ABI moder rl z_cffi_const_%s)r r� )r< r r� r rr rI r )r
rB r r r
r r r �_generate_cpy_macro_ctxk s
z"Recompiler._generate_cpy_macro_ctxc C sF t |tj�rB|j}|dkr$d|f }| j|jd| �}tj||�}|S )Nz...z_cffi_array_len(%s)z%s[0])rN r r� rK r� r� )r
rB Z global_namer� r� r r r r� | s
zRecompiler._global_typec C s | j | j||�� d S )N)rY r� )r
rB r r r r �"_generate_cpy_variable_collecttype� s z-Recompiler._generate_cpy_variable_collecttypec C s� | j }| j||�}t|tj�r4|jd kr4|j}d}nd}d|f }|d|j|| jd� � |d� |d||f � |d� |� d S ) Nr �&z*_cffi_var_%s(void)zstatic )rj r� z return %s(%s);r� ) r| r� rN r r� rK r� r� rf )r
rB r r� Z ampersandr� r r r �_generate_cpy_variable_decl� s
z&Recompiler._generate_cpy_variable_declc C sL | j ||�}| j| }| jr"t}nt}| jd jt|d| t||��� d S )Nrl z _cffi_var_%s) r� rD r< Z
OP_GLOBAL_VARZOP_GLOBAL_VAR_Frr rI r r� )r
rB r r& r r r r �_generate_cpy_variable_ctx� s
z%Recompiler._generate_cpy_variable_ctxc C s | j |� d S )N)rY )r
rB r r r r �'_generate_cpy_extern_python_collecttype� s z2Recompiler._generate_cpy_extern_python_collecttypec C s | j }t|jtj�rd}n d| }d|jjd|�f }|d| � |d| j||f � |� g }d| }x0t|j�D ]"\}} | jd| |�}
|j |
� qtW d j
|�} | p�d
} d || f } |j d kr�d
| } dd� }
t t
|j�d d�}|
|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 � |� | jd 7 _d S ) Nr� z result of %sz(int)sizeof(%s)r z3static struct _cffi_externpy_s _cffi_externpy__%s =z { "%s.%s", %s };zargument of %sz a%dz, r� z%s(%s)� __stdcallz_cffi_stdcall c S s t | tj�o| jdkS )Nz long double)rN r r� r )rB r r r �may_need_128_bits� s z9Recompiler._extern_python_decl.<locals>.may_need_128_bitsr6 � z!sizeof(%s) > %d ? sizeof(%s) : %dz%s%sr� z
char a[%s];z char *p = a;za%dr z *(%s)(p + %d) = %s;r z, _cffi_call_python(&_cffi_externpy__%s, p);z return *(%s)p;r� r )r| rN r� r r� r� r; rQ rJ rI r0 r� r? r rO r� r� )r
rB r Z
tag_and_spacer� Zsize_of_resultr� r� rV r� r� r� r� r Z size_of_ar r r �_extern_python_decl� sX
" z Recompiler._extern_python_declc C s | j ||d� d S )Nzstatic )r )r
rB r r r r � _generate_cpy_extern_python_decl� s z+Recompiler._generate_cpy_extern_python_declc C s | j ||d� d S )NzCFFI_DLLEXPORT )r )r
rB r r r r �#_generate_cpy_dllexport_python_decl� s z.Recompiler._generate_cpy_dllexport_python_declc C s | j ||d� d S )Nr )r )r
rB r r r r �'_generate_cpy_extern_python_plus_c_decl� s z2Recompiler._generate_cpy_extern_python_plus_c_declc C sR | j rtd��|jr td��| j| }tt|�}| jd jt |d| ||�� d S )Nz,cannot use 'extern "Python"' in the ABI modez$a vararg function is extern "Python"rl z&_cffi_externpy__%s)
r< r r� r rD r� ZOP_EXTERN_PYTHONrr rI r )r
rB r r& r
r r r � _generate_cpy_extern_python_ctx� s
z*Recompiler._generate_cpy_extern_python_ctxc C sz | j }|d� xf|jd�D ]X}|d| j� � d}x6|D ].}t|�dkrT||� d}|dt|�f 7 }q8W ||� qW d S )Nz=// # NB. this is not a string because of a size limit in MSVCTz// r �L z%d,)r| �
splitlines�rstripr �ord)r
�sr� �lineZ printed_line�cr r r r� s
z)Recompiler._print_string_literal_in_arrayc C s t tt�| j|< d S )N)r� � OP_PRIMITIVEZ PRIM_VOIDrH )r
rB r� r r r �_emit_bytecode_VoidType s z"Recompiler._emit_bytecode_VoidTypec C s t |j }tt|�| j|< d S )N)ZPRIMITIVE_TO_INDEXr r� r' rH )r
rB r� r7 r r r � _emit_bytecode_PrimitiveType s
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 r� r' rH )r
rB r� r$ r r r �!_emit_bytecode_UnknownIntegerType 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 */
))r r� r' rH )r
rB r� r$ r r r � _emit_bytecode_UnknownFloatType s z*Recompiler._emit_bytecode_UnknownFloatTypec C s� t t| j|j �| j|< |d7 }xR|jD ]H}| j| }||krht|tj�rX| j ||� nt t
|�| j|< |d7 }q(W t |j �}|j
d k r�|j
dkr�|dO }ntd|j
f ��t t|�| j|< d S )Nr r r4 zabi=%r)r� Z OP_FUNCTIONrD r� rH rJ rN r r� r) r! r
r� r� r ZOP_FUNCTION_END)r
rB r� rU Z realindexr' r r r � _emit_bytecode_RawFunctionType# s
z)Recompiler._emit_bytecode_RawFunctionTypec C s t t| j|j �| j|< d S )N)r� �
OP_POINTERrD r� rH )r
rB r� r r r �_emit_bytecode_PointerType6 s z%Recompiler._emit_bytecode_PointerTypec C s" |j � }tt| j| �| j|< d S )N)r[ r� r- rD rH )r
rB r� �rawr r r � _emit_bytecode_FunctionPtrType< s z)Recompiler._emit_bytecode_FunctionPtrTypec C s| | j |j }|jd kr(tt|�| j|< nP|jdkrNtdt|�jdd�f ��n*tt |�| j|< td t|j��| j|d < d S )Nz...zitype %s badly placed: the '...' array length can only be used on global arrays or on fields of structuresz/*...*/r )
rD r� rK r� Z
OP_OPEN_ARRAYrH r rG r� ZOP_ARRAY)r
rB r� Z
item_indexr r r �_emit_bytecode_ArrayType@ s
z#Recompiler._emit_bytecode_ArrayTypec C s | j | }tt|�| j|< d S )N)rL r� ZOP_STRUCT_UNIONrH )r
rB r� Z struct_indexr r r �_emit_bytecode_StructTypeN s
z$Recompiler._emit_bytecode_StructTypec C s | j | }tt|�| j|< d S )N)rM r� r rH )r
rB r� Z
enum_indexr r r �_emit_bytecode_EnumTypeS s
z"Recompiler._emit_bytecode_EnumType)F)r )N)Nr N)N)Ur r r r� r r@ rX rY rE rs rx r| r� r� r~ r� r} r� r� r� r� r� r� r� r� r� r� r� r� r� r` r� r� r� r� rv r� Z _generate_cpy_union_collecttyper� r� Z_generate_cpy_union_declr� Z_generate_cpy_union_ctxr� r� r r r r� r� r� r� r r r r r r� r r r r Z*_generate_cpy_dllexport_python_collecttypeZ._generate_cpy_extern_python_plus_c_collecttyper r r r r Z"_generate_cpy_dllexport_python_ctxZ&_generate_cpy_extern_python_plus_c_ctxr� r( r) r* r+ r, r. Z _emit_bytecode_ConstPointerTypeZ _emit_bytecode_NamedPointerTyper0 r1 r2 Z_emit_bytecode_UnionTyper3 r r r r r9 { s�
:
,
4-# !
Q
5
r9 � c s e Zd Z� fdd�Z� ZS )�NativeIOc s( t |t�r|jd�}tt| �j|� d S )N�ascii)rN Zunicode�encode�superr5 r{ )r
r$ )rS r r r{ \ s
zNativeIO.write)r r r r{ �
__classcell__r r )rS r r5 [ s r5 c
C s |rt d|f � t| ||d kd�}|j� |j� t� }|j||� |j� }y@t|d�� }|jt |�d �|krvt
�W d Q R X |r�t d� dS t
k
�r d|t j � f } t| d��}|j
|� W d Q R X yt j| |� W n, tk
�r t j|� t j| |� Y nX d S X d S )
Nz
generating %s)r< r� r z(already up-to-date)Fz%s.~%d�wT)�printr9 rX rx r5 r� �getvaluer� �readr �IOErrorr� �getpidr{ �rename�OSError�unlink)
r: r; r� Z target_file�verboseZ
recompilerr �output�f1Ztmp_filer r r �_make_c_or_py_sourcea s4
rF Fc C s t | ||||�S )N)rF )r: r; r� Z
target_c_filerC r r r �
make_c_source} s
rG c C s t | |d ||�S )N)rF )r: r; Ztarget_py_filerC r r r �make_py_source� s
rH c C sj |j d�}y$tjtjj| f|d d� �� � W n tk
rB Y nX |d |7 < tjj| f|�� |fS )Nr� r r� r� )rd r� �makedirsr� r0 rA )Z outputdir�modname� extension�partsr r r �_modname_to_file� s
$rM c C s* t ||�}| j|||f� t|||� |S )N)rR rI �setattr)� patchlist�clsr Znew_meth�oldr r r � _patch_meth� s
rR c C s( x"t | �D ]\}}}t|||� q
W d S )N)�reversedrN )rO rP r Zold_methr r r �_unpatch_meths� s rT c s\ t jdkr(ddlm} t| |ddd� � t jdkrXddlm} � fd d
�}t| |d |�� d S ) N�win32r )� MSVCCompilerZ_remove_visual_c_refc S s |S )Nr )r
Z
manifest_filer r r rC � s z&_patch_for_embedding.<locals>.<lambda>�darwin)� CCompilerc s< d| j kr,t| j �| _ | j jd�}d| j |< � | f|�|�S )Nz-bundlez -dynamiclib)Z linker_sor� r� )r
rJ �kwdsrV )�old_link_shared_objectr r �my_link_shared_object� s
z3_patch_for_embedding.<locals>.my_link_shared_objectZlink_shared_object)�sys�platformZdistutils.msvc9compilerrV rR Zdistutils.ccompilerrX )rO rV rX r[ r )rZ r �_patch_for_embedding� s
r^ c sl ddl m} � jd�rR� d d
� � tjdkr6� d7 � n tjdkrJ� d7 � n� d 7 � t| |d
� fd d �� d S )Nr )� build_extz.*r4 rU z.dllrW z.dylibz.soZget_ext_filenamec s � S )Nr )r
Zext_name)�targetr r rC � s z#_patch_for_target.<locals>.<lambda>r� )Zdistutils.command.build_extr_ �endswithr\ r] rR )rO r` r_ r )r` r �_patch_for_target� s
rb r� T�.cc K s� t |t�s|jd�}| jr$| j| � |d k �r^| jd k } | rF| j| � |d krzt|||�\}}
|rl|g|
}
tj j
|
� }n|}| d kr�| r�d| } nd} t j ||f| �}t
| ||||d�}|�rTg }tj� }zn| r�t|� | dkr�t|| � |�r |dk�rd}nd}td|tj j|�f � tj|� t jd|||
�}W d tj|� t|� X |S ||fS n<|d k�rxt||d �\}}t| |||d�}|�r�|S d |fS d S )
Nr6 z%s.*r )rC r� zthe current directory isz setting the current directory toz%s %rz.py)rN rG r7 Z_windows_unicodeZ_apply_windows_unicoder� Z_apply_embedding_fixrM r� r� r0 r Z
get_extensionrG �getcwdr^ rb r; �abspath�chdir�compilerT rH )r: r; r� �tmpdirZcall_c_compilerZc_fileZsource_extensionZextradirZcompiler_verboser` �debugrY Z embeddingrL Z
ext_c_fileZext�updatedrO �cwd�msg�outputfilenamerb r r r � recompile� sb
rn c O s� ddl m} dd l}|jdt|�� t| ||f|�|�}|j||�}xFt|j�D ]8} | j d�sPt
|j| �}
|
t
| | t � �k rPt | | |
� qPW dd� } | | _
x6t| �D ]*} | j d� r�t|j| � r�t | | t� q�W |jS )Nr )�udirrh rb c _ s t �d S )N)r )rJ rY r r r �typeof_disabled s z _verify.<locals>.typeof_disabled)Z testing.udirro �imp�
setdefaultrG rn Z load_dynamic�dirr: r� rR �objectrN Z_typeof�hasattr�NotImplemented�lib) r: r; r� rJ rY ro rq rm �moduler �attrrp r r r �_verify s
rz )r4 )F)F)r� TNrc Nr NN) r� r\ �ior r r �errorr Z cffi_opcoder= r� r� r r r% r1 r8 r9 � version_info�StringIOr5 �BytesIOrF rG rH rM rR rT r^ rb rn rz r r r r �<module> sH f
<