관리-도구
편집 파일: recompiler.cpython-39.pyc
a <� `8� � @ s0 d dl Z d dlZd dlZddlmZmZ ddlmZ ddlT dZ dZ dZ ej d kpdej d k pdej d kZG d d � d �ZG dd� d�ZG dd� d�ZG dd� d�ZG dd� d�ZG dd� d�Zej dkr�ejZnG dd� dej�Zdd � Zd1d d �Zd2d d!�Zd"d#� Zd$d%� Z d&d'� Z d(d)� Z d*d+� Z d3d/d0�Z dS )4� N� )� ffiplatform�model)�VerificationError)�*i& i' i( �win32)� r )r � 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 �5/usr/lib64/python3.9/site-packages/cffi/recompiler.py�__init__ s 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 s" || _ || _|| _|| _|| _d S r )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|| 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 �d S r )�NotImplementedErrorr r r r r , s zFieldExpr.as_python_exprc C sD | j jtkrd}n | j jtkr*t| 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 @ s$ 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 r- r. r r/ r0 r1 r2 r r r r ; 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/ r1 r# r2 r0 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 s g | ] }|� � �qS r )r+ )�.0Zc_fieldr r r � <listcomp>O s �z2StructUnionExpr.as_python_expr.<locals>.<listcomp>z(b'%s%s%s',%s)�,)�evalr. ZG_FLAGSr2 r* r- r �join)r r. Z fields_exprr r r r M s ��z StructUnionExpr.as_python_exprNr r r r r r, : s r, c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�EnumExprc C s" || _ || _|| _|| _|| _d S r �r r- r �signed�allenums)r r r- r r: r; r r r r X s zEnumExpr.__init__c C s d| j | j| j| j| jf S )Nz1 { "%s", %d, _cffi_prim_int(%s, %s), "%s" },r9 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 )r< r )� r )r= r )� r )r> 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* r- r r; )r � prim_indexr r r r d s � � �zEnumExpr.as_python_exprNr r r r r r8 W s r8 c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )� TypenameExprc C s || _ || _d S r �r r- )r r r- r r r r p s zTypenameExpr.__init__c C s d| j | jf S )Nz { "%s", %d },rA r r r r r t s zTypenameExpr.as_c_exprc C s dt | j�| jf S )Nzb'%s%s')r* r- r r r r r r w s zTypenameExpr.as_python_exprNr r r r r r@ o s r@ 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.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 rC rD rE r r r r � s zRecompiler.__init__c C s t | j|�| _d S r )�maxrG )r Zverr r r � needs_version� s zRecompiler.needs_versionc C s. i | _ | �d� t| j td�}g | _|D ]�}|jr(| j | d u sDJ �t| j�| j |< | j�|� |jD ]P}t |t j t j t j t jt jf�s�J �| j | d u r�t| j�| j |< | j�|� qf| j�d� q(|D ]P}|js�| j | d u r�t| j�| j |< | j�|� |jr�|jd ur�| j�d� q�d | j �� v�s0J �i | _i | _|D ]8}t |t j��r^d | j|< nt |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 ��qt | j�| _d S ) NZ collecttype��keyZEND�LENc S s | j S r �r ��tpr r r �<lambda>� � z/Recompiler.collect_type_table.<locals>.<lambda>c S s | j S r rM rN r r r rP � rQ 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_declsrO �tp1�i�methodr' r r r �collect_type_table� s` � �� z Recompiler.collect_type_tablec C s | j }|�|�S r )rE � enumfields)r rO Z expand_anonymous_struct_unionr r r � _enum_fields� s zRecompiler._enum_fieldsc C s� t |tj�s.t |t�r*|D ]}| �|� qd S || jvr�d | j|< t |tj�r^| �|�� � nnt |tj�r�|j d ur�|| j j j vr�| � |�D ] \}}}}| �| �|||�� q�n |�� D ]\}}| �|� q�d S r )rY r ZBaseTypeByIdentityri �_do_collect_typerR r^ �as_raw_functionrc �fldtypesrC �_parser�_included_declarationsro � _field_typeZ _get_items)r rO �xZname1rj �_r r r rp � s" �zRecompiler._do_collect_typec C s� | j jj�� }t|�D ]�\}\}}|�dd�\}}zt| d||f �}W n tyf td| ��Y n0 z|| _ |||� W q t y� } zt � | |� � W Y d } ~ qd } ~ 0 0 qd S )Nr" r z_generate_cpy_%s_%sz"not implemented in recompile(): %r) rC rs Z _declarations�itemsrT �splitrf �AttributeErrorr �_current_quals� Exceptionr Zattach_exception_info) r � step_name�lstr rO �quals�kindZrealnamerl �er r r rS � s" � � zRecompiler._generate)�global�field� struct_union�enum�typenamec C s( i | _ | jD ]}g | j |< q t� | _| �d� | �� | jD ]4}| j | }|dkrb|jdd� d� t|�| j |< q<| j d }| j� � D ],\}}|t |�k s�J �|| j |j ks�J �q�t |�t | j�ks�J �| j d }| j � � D ],\}}|t |�k s�J �|| j |j ks�J �q�t |�t | j �k�s$J �d S )N�ctxr� c S s | j S r rM )�entryr r r rP rQ z0Recompiler.collect_step_tables.<locals>.<lambda>rJ r� r� ) �_lsts� ALL_STEPS�set�_seen_struct_unionsrS �_add_missing_struct_unions�sortri ra rx r# r rb )r r} r~ rO rk 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 s zRecompiler._prntc C s: | j r |d u sJ �| �|� n|d us*J �| �||� d S r )rE �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�pathr7 �dirname�__file__� readlines�close)r �filename�g�linesr r r �_rel_readlines! s zRecompiler._rel_readlinesc C sl || _ | j}| jjd ur |d� ts,|d� | �d�}|�d�}| �d�|||d �< |d�|�� | j� d�d }| jjd u�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|jd d#� \}}W n& t �y� t d$| j|f ��Y n0 |d u �r�t d%��|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| 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 s g | ]\}}||f�qS r r �r3 rO rk r r r r4 W rQ 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[] = {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� �indexr7 rD ry � _print_string_literal_in_arrayrI �VERSION_EMBEDDED�dictrR rx re rV � _get_c_namer r� �_seen_constantsrS r� r� r# �_included_ffis�_assigned_sourcerz r � _num_externpyrG )r r� r� �prntr� rk Zbase_module_name�typeindex2typer' r0 Znumsr} r~ r� �ffi_to_include�included_module_name�included_sourcer. r r r r� ' s2 ��� � � ��� � ��z Recompiler.write_c_source_to_fc sb t |t�rd|f S t |ttf�rZ� fdd�|D �}t|�dkrJ|�d� dd�|�f S |�� S )Nzb'%s'c s g | ]}� � |��qS r )�_to_py)r3 �itemr r r r4 � rQ z%Recompiler._to_py.<locals>.<listcomp>r r& z(%s)r5 )rY rU �listri r# rW r7 r )r rv Zrepr r r r� � s zRecompiler._to_pyc C s� || _ | j}|d� |d� t| jjp(d�}t|�D ]l}| jj| }z|jd d� \}}W n$ ty~ td| j|f ��Y n0 |d ur�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� d S )Nz# auto-generated filezimport _cffi_backendr r< 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 s g | ] }|� � �qS r )r )r3 r' r r r r4 � rQ z3Recompiler.write_py_source_to_f.<locals>.<listcomp>z _types = %s,r& c S s g | ]\}}||f�qS r r r� r r r r4 rQ r r� z _%ss = %s,z _includes = (%s,),�, c S s g | ] }d | �qS )z_ffi%dr �r3 rk r r r r4 rQ �))r� r� r# rC r� �ranger� rz r rD rG ri rV r� r7 r� rR rx r� r� ) r r� r� Z num_includesrk r� r� r� Z types_lstr� r} r~ r r r r� � sN � ��� � �z Recompiler.write_py_source_to_fc C s | j | S r )rR )r �typer r r � _gettypenum s zRecompiler._gettypenumc C sn d}t |tj�r�|�� s�|�� r:|jdkr:d}d|j }nPt |tj�rXd|�d�f }n2|�d�}d||j�dd�f }|d v r�| � t � d }n�t |tj �r�| � ||||� d S t |tj �s�t |tj�r�| �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" rw �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()))rY r r[ �is_complex_type�is_integer_typer �UnknownFloatType� get_c_name�replacerI �VERSION_CHAR16CHAR32r\ �"_convert_funcarg_to_c_ptr_or_arrayr] r� r� r^ r% ) r rO �fromvar�tovar�errcodeZextraargZ converter�cnameZerrvaluer r r �_convert_funcarg_to_c sJ � � �� �z Recompiler._convert_funcarg_to_cc C s. t |tj�r*|�d� |�d� |�d� 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);)rY r r\ �add)r rO � 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 rO r� r� r� r r r r� J s � ��� z-Recompiler._convert_funcarg_to_c_ptr_or_arrayc C sJ t |tj�r�|�� r,|jdkr,d||jf S t |tj�rBd|f S |jdkr�|�� s�|j�dd�}|dv rt| �t � d||f S d || � |�f S n�t |tj tj f�r�d || � |�f S t |tj �r�d || � t� |j��f S t |tj��r |jd u �r td |�� |f ��d || � |�f S t |tj��r>d || � |�f S t|��d S ) Nr� z_cffi_from_c_int(%s, %s)z_cffi_from_c_double(%s)� long doubler" rw 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)))rY r r[ r� r r� r� r� rI r� r� r\ r^ � ArrayTyper� rc Zfldnames� TypeErrorr� rd r% )r rO �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 rO r r r r � _typedef_type{ s zRecompiler._typedef_typec C s | � | �||�� d S r )rp r� r� r r r �!_generate_cpy_typedef_collecttype~ s z,Recompiler._generate_cpy_typedef_collecttypec C s d S r r r� r r r �_generate_cpy_typedef_decl� s z%Recompiler._generate_cpy_typedef_declc C s$ | j | }| jd �t||�� d S )Nr� )rR r� rW r@ )r rO r r- r r r � _typedef_ctx� s zRecompiler._typedef_ctxc C sf | � ||�}| �||� 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� rf � _struct_ctxr rY r ZNamedPointerType�totyper� r r r �_generate_cpy_typedef_ctx� s �z$Recompiler._generate_cpy_typedef_ctxc C s( | � |�� � |jr$| js$| � |� d S r )rp rq �ellipsisrE r� r r r �"_generate_cpy_function_collecttype� s z-Recompiler._generate_cpy_function_collecttypec C s | j r J �t|tj�sJ �|jr0| �||� d S | j}t|j�}|dkrNd}n|dkr\d}nd}g }g }d| }t |j�D ],\} } |� | � d| |�� |� d| � qzd � |�} | p�d } |j r�|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| � �qrt� }t� }|jD ]} | �| ||� �q�t|�D ]}|d|f � �q�t|jtj��s d}d| }d|j� d|� }||� |d� nd }d }t|j�dk�r�tt|j��}|D ]} |d| � �qJ|� |d |t|�t|�d � d d � |D ��f � |d � |� t |j�D ](\} } | �| d | d| d!� |� �q�|d"� |d#� d$d � tt|j��D �}d � |�}|d%|||f � |d&� |d'� |� |d(� |dk�rF|d)� |�r�|d*| �|jd+d,� � |D ]}|d-| � �qh|d.� n(|D ]}|d-| � �q�|d/� |d0� |d� |d1� d2d3� }d4}g }g }d| }t |j�D ]P\} } d }|| ��rd5}d6}| � d7|| f |�}|� |� |� d8|| f � �q�|j}||��rxd| }|� d9|�}|�d|� tj}d }d:}d6}|�r�d � |�} | �p�d } d;| || f } |d|� | �f � |d� |�r�||� d � |�}|d%|||f � |�r�|d<� |d� n|d=||f � |d>� |� d S )?Nr Znoargr Zarg0rX �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 s g | ] }d | �qS )z&arg%dr r� r r r r4 � rQ z:Recompiler._generate_cpy_function_decl.<locals>.<listcomp>r� zarg%dz return NULLz Py_BEGIN_ALLOW_THREADSz _cffi_restore_errno();c S s g | ] }d | �qS )r� r r� r r r r4 � rQ 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�p t | tj�o | �� S r )rY r rc � 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� )rE rY r r^ r� �_generate_cpy_constant_declr� r# rX re rW r� r7 �abir� rZ r� r� rT r� r� r� �insertZ void_type)r rO r r� �numargs�argname� argumentsZcall_argumentsr� rk 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 |dkrNt}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� rE �_generate_cpy_constant_ctxrR rq r# rX ZOP_DLOPEN_FUNCZOP_CPYTHON_BLTN_NZOP_CPYTHON_BLTN_OZOP_CPYTHON_BLTN_Vr� rW r rh )r rO 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|�d�}d||f }| �|d| |j�}t�||�}|S )N�...r z _cffi_array_len(((%s)0)->%s)�%s[0])rY r r� r_ r� ru r� )r Z tp_struct� field_nameZtp_field� actual_lengthZptr_struct_name�tp_itemr r r ru Q s � � zRecompiler._field_typec C s, | � |� | jr(|�� D ]}| �|� qd S r )rp rE �anonymous_struct_fields�_struct_collecttype)r rO �fldtyper r r r ] s z Recompiler._struct_collecttypec C sB |j d u rd S | j}d|f }|d� |d||f � |d� |d� |d� | �|�D ]�\}}}} z�|�� sv|dkr�|dkr�|d |||f � W qXt|tj�r�|jd u s�|jd kr�|j}|d }q�|d |j d d| | d�|f � W qX t �y } z |dt | � � W Y d } ~ qXd } ~ 0 0 qX|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; };) rr r� ro r� rY r r� r_ r� r� r rU ) r rO r� r� r� Z checkfuncname�fnameZftyper �fqualr� r r r � _struct_decld sB � �� � (zRecompiler._struct_declNc C s� | j | }d }g }t|tj�r(|�d� |jd u r@|�d� d}|| jjjvr�|d u sd|| jjjvr�|jd u rpn |j s�t |� � �r�n |�d� |j r�|j dkr�t d||j f ��|�d� n|�d� d }d �|�p�d }g }|d u �rht| �|��} | D �] \} } } } | �|| | �} | �| d |j| f � t}| d k�rFt}d| }nL|d u �sjt| tj��rp| jd u �rpd}n"d|d u �r�|�d�n|j| f }|d u �s�| d k�r�d}n,|d u�r�d|j| f }nd|�d�| f }|�t| ||| t|| j | ��� q�t| j d �}| j d � |� |d u �r2d}d}d}n4|d u�rNd|jf }d}nd|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� )!rR rY r Z UnionTyperW rr rC rs rt �partial�anyr Zpackedr% r7 r� ro ru �_check_not_opaquer r( r) r� r_ r� r rh r# r� �extendr, r� r� )r rO r� r� r� r- Zreason_for_not_expandingr. r2 rn Zfldnamer r r r' r �offsetr1 Zalignr0 r r r r� � s� � � �� � ��� � �� ��zRecompiler._struct_ctxc C s: t |tj�r|j}q t |tj�r6|jd u r6td| ��d S )Nz0%s is of an opaque type (not declared in cdef()))rY r r� r� rc rr r� )r rO �locationr r r r � s �z Recompiler._check_not_opaquec C s� t | j�� �}|jdd� d� |D ]�\}}|| jvr"|jrHtd|f ��|j�d�rv|jdd � � � rv|jdd � }n4|jdkr�|j dkr�d}| � |d� ntd |f ��| � |d |� q"d S ) Nc S s | d S )Nr r )Ztp_orderr r r rP � rQ z7Recompiler._add_missing_struct_unions.<locals>.<lambda>rJ zDinternal inconsistency: %r is partial but was not seen at this point�$r Z_IO_FILE�FILEzinternal inconsistency: %r) r� ra rx r� r� r r% r � startswith�isdigitZ forcenamer� r� )r r~ rO �orderr� r r r r� � s" � �z%Recompiler._add_missing_struct_unionsc C s | � |� d S r )r r� r r r � _generate_cpy_struct_collecttype� s z+Recompiler._generate_cpy_struct_collecttypec C s2 |� d�}d|v r"||�dd�fS |d| fS d S )Nr& r" rw )r� r� )r rO r� r r r � _struct_names� s zRecompiler._struct_namesc C s | j |g| �|��R � d S r )r r r� r r r �_generate_cpy_struct_decl s z$Recompiler._generate_cpy_struct_declc C s | j |g| �|��R � d S r )r� r r� r r r �_generate_cpy_struct_ctx s z#Recompiler._generate_cpy_struct_ctxc C s( t |tj�r| �||� n | �|� d S r )rY r rd � _generate_cpy_enum_collecttyper r� r r r �#_generate_cpy_anonymous_collecttype s z.Recompiler._generate_cpy_anonymous_collecttypec C s. t |tj�r| �|� n| �||d| � d S �NZtypedef_)rY r rd �_generate_cpy_enum_declr r� r r r � _generate_cpy_anonymous_decl s z'Recompiler._generate_cpy_anonymous_declc C s0 t |tj�r| �||� n| �||d| � d S r$ )rY r rd � _enum_ctxr� r� r r r �_generate_cpy_anonymous_ctx s z&Recompiler._generate_cpy_anonymous_ctx�constc C s ||f| j v r td||f ��| j �||f� | j}d||f }|r�|d| � |d� |d|f � |d||f � |d ur�|dkr�d|f }|d |f � |d � |d � |d � n>|d u s�J �|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 rO �categoryr r� �funcnamer r r �_generate_cpy_const& s6 � � z Recompiler._generate_cpy_constc C s |� � }|r| jr | �|� d S r )r� rE rp �r rO r r* r r r �"_generate_cpy_constant_collecttypeD s z-Recompiler._generate_cpy_constant_collecttypec C s |� � }| �|||� d S r )r� r- r. r r r r� I s z&Recompiler._generate_cpy_constant_declc C s^ | j s|�� rttd�}n$| j r&t}nt}| j| }t||�}| jd �t |d| |�� d S )Nr� r� �_cffi_const_%s) rE r� rh �OP_CONSTANT_INTZOP_DLOPEN_CONSTZ OP_CONSTANTrR r� rW r )r rO r r Z const_kindr- r r r r M s �z%Recompiler._generate_cpy_constant_ctxc C s | � |� d S r )rp r� r r r r" ] s z)Recompiler._generate_cpy_enum_collecttypec C s |j D ]}| �d|� qd S )NT)� enumeratorsr- )r rO 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�� q0|d ur�d|vr�| js�d| }d| }n4|� | j g �} | j � | �}tt| j �| d��dk �}d �|j�} | jd � t|j|||| �� d S ) Nr� r� r0 �r r r z ((%s)-1) <= 0r r5 r� )rR rh �OP_ENUMrE Zcheck_not_partial�zipr2 Z enumvaluesr� rW r Zbuild_baseinttyperC Zsizeof�int�castr7 r8 r ) r rO r� r- r r3 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 s z!Recompiler._generate_cpy_enum_ctxc C s d S r r r� r r r � _generate_cpy_macro_collecttype s z*Recompiler._generate_cpy_macro_collecttypec C s& |dkrd }n|}| j d||d� d S )Nr Tr4 )r- )r rO r r r r r �_generate_cpy_macro_decl� s z#Recompiler._generate_cpy_macro_declc C sT |dkr"| j r td|f ��d }n|}ttd�}| jd �t|d| ||d�� d S )Nr zGcannot use the syntax '...' in '#define %s ...' when using the ABI moder� r� r0 r4 )rE r rh r1 r� rW r )r rO r r r r r r �_generate_cpy_macro_ctx� s �� ��z"Recompiler._generate_cpy_macro_ctxc C sF t |tj�rB|j}|dkr$d|f }| �|jd| �}t�||�}|S )Nr z_cffi_array_len(%s)r )rY r r� r_ r� r� )r rO Z global_namer r r r r r� � s zRecompiler._global_typec C s | � | �||�� d S r )rp r� r� r r r �"_generate_cpy_variable_collecttype� s z-Recompiler._generate_cpy_variable_collecttypec C s� | j }| �||�}t|tj�r4|jd u r4|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� rY r r� r_ r� r� r{ )r rO r r� Z ampersandr� r r r �_generate_cpy_variable_decl� s z&Recompiler._generate_cpy_variable_declc C sL | � ||�}| j| }| jr"t}nt}| jd �t|d| t||��� d S )Nr� z _cffi_var_%s) r� rR rE Z OP_GLOBAL_VARZOP_GLOBAL_VAR_Fr� rW r rh )r rO r r- r' r r r �_generate_cpy_variable_ctx� s �z%Recompiler._generate_cpy_variable_ctxc C s t |tj�sJ �| �|� d S r )rY r r^ rp 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|j�d|�f }|d| � |d| j||f � |� g }d| }t|j�D ]"\}} | �d| |�} |� | � qrd � |�} | p�d } d || f } |j d kr�d | } 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 ]X\}} d| } t| tj��s�| | ��r�d| } t�| �} |d| �d�|d | f � �qd|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� )rY r r� r rN r r r �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;r z, _cffi_call_python(&_cffi_externpy__%s, p);z return *(%s)p;r� r )r� rY r� r rZ r� rD re rX rW r7 r� rH r# rc r\ r� )r rO r Z tag_and_spacer� Zsize_of_resultr� r� rk r� r� r� r� rD Z size_of_ar r r �_extern_python_decl� sb � � � � z Recompiler._extern_python_declc C s | � ||d� d S )Nr? �rF r� r r r � _generate_cpy_extern_python_decl s z+Recompiler._generate_cpy_extern_python_declc C s | � ||d� d S )NzCFFI_DLLEXPORT rG r� r r r �#_generate_cpy_dllexport_python_decl s z.Recompiler._generate_cpy_dllexport_python_declc C s | � ||d� d S )Nr& rG 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 �t |d| ||�� d S )Nz,cannot use 'extern "Python"' in the ABI modez$a vararg function is extern "Python"r� z&_cffi_externpy__%s) rE r r� r% rR rh ZOP_EXTERN_PYTHONr� rW r )r rO r r- 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 tyV d| }Y n0 |�d�D ]|}|}td�tu r�tt |�}nt |�dd� }|d | � � � d }|D ]*}t |�d kr�||� d }|d |f 7 }q�||� qbd 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� rY �bytes�encode�decode�UnicodeDecodeError� splitlinesr� �map�ordrL �rstripr# )r �sr� �liner0 Z printed_line�cr r r r� s, z)Recompiler._print_string_literal_in_arrayc C s t tt�| j|< d S r )rh � OP_PRIMITIVEZ PRIM_VOIDrV �r rO r� r r r �_emit_bytecode_VoidType= s z"Recompiler._emit_bytecode_VoidTypec C s t |j }tt|�| j|< d S r )ZPRIMITIVE_TO_INDEXr rh rY rV )r rO r� r? 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 rh rY rV �r rO r� rV 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 */ )r] r^ r r r � _emit_bytecode_UnknownFloatTypeJ s �z*Recompiler._emit_bytecode_UnknownFloatTypec C s� t t| j|j �| j|< |d7 }|jD ]H}| j| }||krft|tj�rV| � ||� nt t |�| j|< |d7 }q&t |j �}|j d ur�|j dkr�|dO }ntd|j f ��t t|�| j|< d S )Nr rC r< zabi=%r)rh Z OP_FUNCTIONrR r� rV rX rY r r� r\ r( r7 r� r� r% ZOP_FUNCTION_END)r rO r� rj Z realindexr. r r r � _emit_bytecode_RawFunctionTypeP s z)Recompiler._emit_bytecode_RawFunctionTypec C s t t| j|j �| j|< d S r )rh � OP_POINTERrR r� rV rZ r r r �_emit_bytecode_PointerTypec s z%Recompiler._emit_bytecode_PointerTypec C s" |� � }tt| j| �| j|< d S r )rq rh rb rR rV )r rO r� �rawr r r � _emit_bytecode_FunctionPtrTypei s z)Recompiler._emit_bytecode_FunctionPtrTypec C s� | j |j }|jd u r(tt|�| j|< nf|jdkrNtdt|��dd�f ��n@| j|d dksdJ �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 rL ) rR r� r_ rh Z OP_OPEN_ARRAYrV r rU r� ZOP_ARRAY)r rO r� Z item_indexr r r �_emit_bytecode_ArrayTypem s ���z#Recompiler._emit_bytecode_ArrayTypec C s | j | }tt|�| j|< d S r )ra rh ZOP_STRUCT_UNIONrV )r rO r� Z struct_indexr r r �_emit_bytecode_StructType{ s z$Recompiler._emit_bytecode_StructTypec C s | j | }tt|�| j|< d S r )rb rh r5 rV )r rO r� Z enum_indexr r r �_emit_bytecode_EnumType� s z"Recompiler._emit_bytecode_EnumType)F)r& )N)Nr) N)N)Vr r r r� r rI rm ro rp rS r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r ru r r r� r r� 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' r9 r: r; r< r� r= r@ rA rB Z*_generate_cpy_dllexport_python_collecttypeZ._generate_cpy_extern_python_plus_c_collecttyperF rH rI rJ rK Z"_generate_cpy_dllexport_python_ctxZ&_generate_cpy_extern_python_plus_c_ctxr� r[ r\ r_ r` ra rc Z _emit_bytecode_ConstPointerTypeZ _emit_bytecode_NamedPointerTypere rf rg Z_emit_bytecode_UnionTyperh r r r r rB ~ s� : . 4-# # " V � �6 � rB )r c s e Zd Z� fdd�Z� ZS )�NativeIOc s( t |t�r|�d�}tt| ��|� d S )NrL )rY ZunicoderO �superri r� )r rV �rg r r r� � s zNativeIO.write)r r r r� � __classcell__r r rk r ri � s ri c C s@ |rt d|f � t| ||d u d�}|�� |�� t� }|�||� |�� }zVt|d��*}|�t |�d �|krvt �W d � n1 s�0 Y |r�t d� W dS t �y: d|t � � f } t| d��}|� |� W d � n1 s�0 Y zt �| |� W n* t�y2 t �|� t �| |� Y n0 Y d S 0 d S ) Nz generating %s)rE r� r z(already up-to-date)Fz%s.~%d�wT)�printrB rm r� ri r� �getvaluer� �readr# �IOErrorr� �getpidr� �rename�OSError�unlink) rC rD r� Z target_file�verboseZ recompilerr� �output�f1Ztmp_filer r r �_make_c_or_py_source� s6 � " ( ry Fc C s |d us J �t | ||||�S r �ry )rC rD r� Z target_c_filerv r r r � make_c_source� s �r{ c C s t | |d ||�S r rz )rC rD Ztarget_py_filerv r r r �make_py_source� s �r| c C sl |� d�}z&t�tjj| g|d d� �R � � W n tyB Y n0 |d |7 < tjj| g|�R � |fS )Nr� r� )ry r� �makedirsr� r7 rt )Z outputdir�modname� extension�partsr r r �_modname_to_file� s & r� c C s* t ||�}| �|||f� t|||� |S r )rf rW �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 jdkr(ddlm} t| |ddd� � t jdkrXddlm} � fd d �}t| |d |�� d S ) Nr r )� MSVCCompilerZ_remove_visual_c_refc S s |S r r )r Z manifest_filer r r rP � rQ z&_patch_for_embedding.<locals>.<lambda>�darwin)� CCompilerc sB d| j v r,t| j �| _ | j �d�}d| j |< � | g|�R i |��S )Nz-bundlez -dynamiclib)Z linker_sor� r� )r rX �kwdsrk �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�rR� d d� � tjdkr6� d7 � n tjdkrJ� 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 rP � rQ 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 u�rt| jd u} | rF| �| � |d u rzt|||�\}} |rl|g| } tj j | � }n|}| d u r�| r�d| } nd} t j ||fi | ��}t | ||||d�}|�rjg }t�� }z�| r�t|� | dkr�t|| � |�r"|dk�rd}nd}td|tj �|�f � t�|� t �d||| �}W t�|� t|� nt�|� t|� 0 |S ||fS n<|d u �r�t||d �\}}t| |||d�}|�r�|S d |fS d S ) NrL z%s.*r )rv r� zthe current directory isz setting the current directory toz%s %rz.py)rY rU rO Z_windows_unicodeZ_apply_windows_unicoder� Z_apply_embedding_fixr� r� r� r7 r Z get_extensionr{ �getcwdr� r� rn �abspath�chdir�compiler� r| )rC rD r� ZtmpdirZcall_c_compilerZc_fileZsource_extensionZextradirZcompiler_verboser� �debugr� Z embeddingr� Z ext_c_file�ext�updatedr� �cwd�msgZoutputfilenamerw r r r � recompile� sn � � � � �r� )F)F)r� TNr� Nr NN)!r� r� �ior& r r �errorr Z cffi_opcoderF r� r� r� � version_infor� r r r, r8 r@ rB �StringIOri �BytesIOry r{ r| r� r� r� r� r� r� r r r r �<module> sP � �