Your IP : 216.73.216.60
.�W)� �
@ s� d d l Z d d l Z d d l Z d d l m Z m Z d d l Td Z d Z Gd d � d � Z Gd d
� d
� Z
Gd d � d � Z Gd
d � d � Z Gd d � d � Z
Gd d � d � Z e j d f k r� e j Z n Gd d � d e j � Z d d � Z d d d � Z d d d � Z d d � Z d d � Z d! d" � Z d# d$ � Z d% d& � Z d' d( d d) d d d d d* d+ � Z d, d- � Z d S).� N� )� ffiplatform�model)�*Z0x2601Z0x2701c @ s: e Z d Z d d d d � Z d d � Z d d � Z d S) �
GlobalExprr c C s1 | | _ | | _ | | _ | | _ | | _ d S)N)�name�address�type_op�size� check_value)�selfr r r r
r � r
�B/opt/alt/python35/lib64/python3.5/site-packages/cffi/recompiler.py�__init__
s
zGlobalExpr.__init__c C s& d | j | j | j j � | j f 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 | j f S)Nz
b'%s%s',%d)r �as_python_bytesr r )r r
r
r �as_python_expr s zGlobalExpr.as_python_exprN)�__name__�
__module__� __qualname__r r r r
r
r
r r s r c @ s@ e Z d Z d d � Z d d � Z d d � Z d d � Z d S)
� FieldExprc C s1 | | _ | | _ | | _ | | _ | | _ 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 sO d t | j � } d | j | j f d | | j f d | | j j � 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 sb | j j t k r d } n* | j j t k r? t | j � } n t � d | j j � | | j f 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 @ s4 e Z d Z d d � Z d d � Z d d � Z d S)�StructUnionExprc C sL | | _ | | _ | | _ | | _ | | _ | | _ | | _ | | _ 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 6 s zStructUnionExpr.__init__c C sf d | j | j | j f d | j | j f d | j t | j � f | j r] d | j n d 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 A s zStructUnionExpr.as_c_exprc C sW t | j t � } d d � | j D� } d t | j � t | � | j d j | � f S)Nc S s g | ] } | j � � q Sr
)r# )�.0Zc_fieldr
r
r �
<listcomp>J 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 H s
z StructUnionExpr.as_python_exprN)r r r r r r r
r
r
r r$ 5 s r$ c @ s4 e Z d Z d d � Z d d � Z d d � Z d S)�EnumExprc C s1 | | _ | | _ | | _ | | _ | | _ d S)N)r r% r
�signed�allenums)r r r% r
r1 r2 r
r
r r S s
zEnumExpr.__init__c C s& d | j | j | j | j | j f S)Nz1 { "%s", %d, _cffi_prim_int(%s, %s),
"%s" },)r r% r
r1 r2 )r r
r
r r Z s zEnumExpr.as_c_exprc C so d t d t d t d
t d t d t d
t d t i | j | j f } d t
| j � t
| � | j | j
f S)Nr r � � � zb'%s%s%s\x00%s')r r )r r )r3 r )r3 r )r4 r )r4 r )r5 r )r5 r )Z
PRIM_UINT8Z PRIM_INT8Z PRIM_UINT16Z
PRIM_INT16Z PRIM_UINT32Z
PRIM_INT32Z PRIM_UINT64Z
PRIM_INT64r
r1 r" r% r r2 )r �
prim_indexr
r
r r _ s zEnumExpr.as_python_exprN)r r r r r r r
r
r
r r0 R s r0 c @ s4 e Z d Z d d � Z d d � Z d d � Z d S)� TypenameExprc C s | | _ | | _ d S)N)r r% )r r r% r
r
r r k s zTypenameExpr.__init__c C s d | j | j f S)Nz { "%s", %d },)r r% )r r
r
r r o s zTypenameExpr.as_c_exprc C s d t | j � | j f S)Nzb'%s%s')r" r% r )r r
r
r r r s zTypenameExpr.as_python_exprN)r r r r r r r
r
r
r r7 j s r7 c @ s� e Z d Z d Z d d d � Z d d � Z d d � Z d d
� Z d d d
d d g Z d d � Z d d d � Z
d d � Z d d � Z d d � Z
d d � Z d d � Z d d � Z d! d" � Z d# d$ � Z d% d& � Z d' d( � Z d) d* � Z d+ d, � Z d- d. � Z d/ d0 � Z d1 d2 � Z d3 d4 � Z d5 d6 � Z d7 d8 � Z d9 d: � Z d; d< � Z d= d> d? � Z d@ dA � Z dB dC � Z! dD dE � Z" e" Z# dF dG � Z$ dH dI � Z% e% Z& dJ dK � Z' e' Z( dL dM � Z) dN dO � Z* dP dQ � Z+ d= dR d= dS dT � Z, dU dV � Z- dW dX � Z. dY dZ � Z/ d[ d\ � Z0 d= d] d^ � Z1 d_ d` � Z2 da db � Z3 dc dd � Z4 de df � Z5 dg dh � Z6 di dj � Z7 dk dl � Z8 dm dn � Z9 do dp � Z: dq dr � Z; e; Z< Z= ds dt � Z> du dv � Z? dw dx � Z@ dy dz � ZA d{ d| � ZB eB ZC ZD d} d~ � ZE d d� � ZF d� d� � ZG d� d� � ZH d� d� � ZI d� d� � ZJ d� d� � ZK eK ZL eK ZM d� d� � ZN d� d� � ZO d� d� � ZP eP ZQ d� d� � ZR d= S)��
Recompilerr Fc C s | | _ | | _ | | _ d S)N)�ffi� module_name�target_is_python)r r9 r: r; r
r
r r | s zRecompiler.__init__c C s i | _ | j d � t | j d t �} g | _ x� | D]� } | j r; | j | d k sc t � t | j � | j | <| j j | � x} | j D]r } t
| t j t j
t j t j t j f � s� t � | j | d k r� t | j � | j | <| j j | � q� W| j j d � q; Wxy | D]q } | j r$| j | d k r$t | j � | j | <| j j | � | j r$| j d k r$| j j d � q$Wd | j j � k s�t � i | _ i | _ xO | D]G } t
| t j � r�d | j | <q�t
| t j � r�d | j | <q�Wx< t t | j d d d � �� D] \ } } | | j | <q:Wx< t t | j d d d � �� D] \ } } | | j | <qyWx; | D]3 } t | d | j j � } | | | j | � q�Wx& | j D] } t
| t � s�t � q�Wt | j � | _ d S) NZ collecttype�keyZEND�LENc 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 )r> r
r
r r? � s Z_emit_bytecode_) �
_typesdict� _generate�sorted�str�
cffi_typesZis_raw_function�AssertionErrorr �append�args�
isinstancer �VoidType�BasePrimitiveType� PointerType�StructOrUnionOrEnum�FunctionPtrTypeZ
is_array_type�length�values�_struct_unions�_enums�
StructOrUnion�EnumType� enumerate�getattr� __class__r �CffiOp�tuple)r Z all_declsr> �tp1�i�methodr r
r
r �collect_type_table� sZ
z Recompiler.collect_type_tablec C s- t | t j � sC t | t � r? x | D] } | j | � q( Wd S| | j k r)d | j | <t | t j � r� | j | j � � n� t | t j � r� | j d k r)| | j
j j k r)xi | j
� D]. \ } } } } | j | j | | | � � q� Wn* x' | j � D] \ } } | j | � q Wd S)N)rH r ZBaseTypeByIdentityrX �_do_collect_typer@ rM �as_raw_functionrR �fldtypesr9 �_parser�_included_declarations�
enumfields� _field_typeZ
_get_items)r r> �xZname1rY �_r
r
r r] � s
#zRecompiler._do_collect_typec
C s� | j j j j � } x� t | � D]� \ } \ } } | j d d � \ } } y t | d | | f � } Wn% t k
r� t j d | � � Yn Xy | | _
| | | � Wq" t k
r� } z t j
| | � � WYd d } ~ Xq" Xq" Wd S)Nr r z_generate_cpy_%s_%sz"not implemented in recompile(): %r)r9 r` Z
_declarations�itemsrB �splitrU �AttributeErrorr �VerificationError�_current_quals� Exceptionr Zattach_exception_info)
r � step_name�lstr r> �quals�kindZrealnamer[ �er
r
r rA � s
zRecompiler._generate�global�field� struct_union�enum�typenamec C s� i | _ x | j D] } g | j | <q Wt � | _ | j d � | j � xS | j D]H } | j | } | d k r� | j d d d � � t | � | j | <qW W| j d } xQ | j j � D]@ \ } } | t
| � k s� t � | | j | j k s� t � q� Wt
| � t
| j � k s%t � | j d } xQ | j
j � D]@ \ } } | t
| � k sft � | | j | j k sBt � qBWt
| � t
| j
� k s�t � d S)N�ctxrr r< c S s | j S)N)r )�entryr
r
r r? � s z0Recompiler.collect_step_tables.<locals>.<lambda>rs rt )�_lsts� ALL_STEPS�set�_seen_struct_unionsrA �_add_missing_struct_unions�sortrX rP rf r rE r rQ )r rl rm r> rZ r
r
r �collect_step_tables� s*
!
z Recompiler.collect_step_tablesr c C s | j j | d � d S)N�
)�_f�write)r �whatr
r
r �_prnt s zRecompiler._prntc C sQ | j r+ | d k s t � | j | � n" | d k s= t � | j | | � d S)N)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 sD t t j j t j j t � | � d � } | j � } | j � | S)N�r)�open�os�pathr/ �dirname�__file__� readlines�close)r �filename�g�linesr
r
r �_rel_readlines
s *
zRecompiler._rel_readlinesc C s: | | _ | j } | j j d k r. | d � | j d � } | j d � } | j d � | | | d � <| d j | � � | j j d � dA } | j j d k r^| d | j f � | d | j | j j � f � | d
� | d | f � | d � | d
| f � | d � | d | f � | d � | j d � } | d j | � � t
} n t } | d � | � | | � | � | d � | � | d � t d d � | j
j � D� � } x` t | j � D]O \ } } d }
| | k rd | | j � }
| d | | j � |
f � q�W| j s9| d � | d � | � t � | _ | j d � i } x� | j D]y } | j | }
t |
� | | <| | d k rs| d | | f � x |
D] } | | j � � q�W| d � | � qsW| j j r�| d � x� | j j D]� } y | j d d � \ } } Wn. t k
rft j d | j | f � � Yn X| d k r�t j d � � | d! | f � qW| d" � | d � | � | d# � | d$ � x@ | j D]5 } | | d k r�| d% | � q�| d& | � q�Wx5 | j D]* } | d' k r| d( | | | f � qW| j j rZ| d) � n
| d* � | d+ t | j � f � d } | j r�| d O} | d, | � | d � | � | d
� | d- � | d. | f � | d/ � | j r | d0 � | d1 � | d2 � | d3 | � | d4 � | d5 � | d6 � | d7 � | d8 � | d9 | f � | d: � | d; | f � | d< � | d< � | d � | d- � | d= | f � | d/ � | d> | j | f � | d5 � | d � | d- � | d? | f � | d/ � | d@ | j | f � | d5 � | d � d S)BNz#define _CFFI_USE_EMBEDDINGz_cffi_include.hz#include "parse_c_type.h"
zparse_c_type.hr r �.z #define _CFFI_MODULE_NAME "%s"z%#define _CFFI_PYTHON_STARTUP_CODE %sz#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>/************************************************************/z static void *_cffi_types[] = {c S s" g | ] \ } } | | f � q Sr
r
)r+ r> rZ r
r
r r, @ 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[] = {r3 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 */rr z %d, /* num_%ss */z _cffi_includes,z NULL, /* no includes */z %d, /* num_types */z %d, /* flags */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 *)%s;z p[1] = &_cffi_type_context;�}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)z3 return _cffi_init("%s", %s, &_cffi_type_context);z init%s(void)z, _cffi_init("%s", %s, &_cffi_type_context);���) r� r� r9 �
_embeddingr� �indexr/ r: rg �_string_literal�VERSION_EMBEDDED�VERSION�dictr@ rf rT rD � _get_c_namer rz �_seen_constantsrA ry rx r �_included_ffis�_assigned_sourcerh r ri �
_num_externpy)r r� r� �prntr� rZ Zbase_module_name�version�typeindex2typer r( �numsrl rm rw �ffi_to_include�included_module_name�included_sourcer& r
r
r r� s�
" !
z Recompiler.write_c_source_to_fc s� t | t � r d | f St | t t f � r{ � f d d � | D� } t | � d k rg | j d � d d j | � f S| j � S)Nzb'%s'c s g | ] } � j | � � q Sr
)�_to_py)r+ �item)r r
r r, � s z%Recompiler._to_py.<locals>.<listcomp>r r z(%s)r- )rH rC �listrX r rF r/ r )r rd �repr
)r r r� � s
zRecompiler._to_pyc
C s | | _ | j } | d � | d � t | j j p8 f � } x� t | � D]� } | j j | } y | j d d � \ } } Wn. t k
r� t j d | j | f � � Yn X| d k r� t j d � � | d | | f � qK W| � | d | j
f � | d t f � t | j
� | _
d d
� | j
D� } | d | j d j | � � f � t d
d
� | j j � D� � } xY | j D]N }
| j |
} t | � d k r�|
d k r�| d |
| j | � f � q�W| d k r| d d j d d
� t | � D� � f � | d � d S)Nz# auto-generated filezimport _cffi_backendr3 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 = %s,c S s g | ] } | j � � q Sr
)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 � q Sr
r
)r+ r> rZ r
r
r r, � s r rr z _%ss = %s,z _includes = (%s,),z, c S s g | ] } d | � q S)z_ffi%dr
)r+ rZ r
r
r r, � s �))r� r� r r9 r� �ranger� rh r ri r: r� rX rD r� r/ r� r@ rf ry rx ) r r� r� Z num_includesrZ r� r� r� Z types_lstr� rl rm r
r
r r� � s@
#"
! 'z Recompiler.write_py_source_to_fc C s | j | S)N)r@ )r �typer
r
r � _gettypenum� s zRecompiler._gettypenumc C s� d } t | t j � r� | j � rI | j d k rI d } d | j } nS t | t j � rt d | j d � f } n( d | j d � | j j d d � f } d } n� t | t j � r� | j | | | | � d St | t j
� r| j d
| | j | � | f � | j d | � d St | t j
� r\d | j d � } d
| j | � } d } n t | � � | j d | | | | f � | j d | | j d � | f � | j d | � d S)Nr Z_BoolZ_cffi_to_c_intz, %sz(%s)_cffi_to_c_doublez(%s)_cffi_to_c_%sr re z-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()))rH r rJ �is_integer_typer �UnknownFloatType�
get_c_name�replacerK �"_convert_funcarg_to_c_ptr_or_arrayrL r� r� rM r )r r> �fromvar�tovar�errcodeZextraarg� converterZerrvaluer
r
r �_convert_funcarg_to_c� s8
z Recompiler._convert_funcarg_to_cc C s# t | t j � r | j d � d S)NzPy_ssize_t datasize)rH r rK �add)r r> � localvarsr
r
r �_extra_local_variables s z!Recompiler._extra_local_variablesc C s� | j d � | j d | j | � | | f � | j d � | j d � | j d | � | j d | | j d � 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 r> r� r� r� r
r
r r� s
z-Recompiler._convert_funcarg_to_c_ptr_or_arrayc C s� t | t j � r� | j � r/ d | | j f St | t j � rL d | f S| j d k rx d | j j d d � | f Sd | | j | � f Sn� t | t j t j f � r� d | | j | � f St | t j
� r� d | | j t j | j � � f St | t j � rM| j
d k r6t d | j � | f � � d
| | j | � f St | t j � rvd | | j | � f St | � � d S) Nz_cffi_from_c_int(%s, %s)z_cffi_from_c_double(%s)z long doublez_cffi_from_c_%s(%s)r re 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)))rH r rJ r� r r� r� r� rK rM � ArrayTyper� rR Zfldnames� TypeErrorr� rS r )r r> �var�contextr
r
r �_convert_expr_from_c. s2 z Recompiler._convert_expr_from_cc C s | j | � d S)N)r] )r r> r r
r
r �!_generate_cpy_typedef_collecttypeN s z,Recompiler._generate_cpy_typedef_collecttypec C s d S)Nr
)r r> r r
r
r �_generate_cpy_typedef_declQ s z%Recompiler._generate_cpy_typedef_declc C s. | j | } | j d j t | | � � d S)Nru )r@ rx rF r7 )r r> r r% r
r
r � _typedef_ctxT s
zRecompiler._typedef_ctxc C s� | j | | � t | d d � d k rD | j | | j d d �n: t | t j � r~ | j | j | j j d | j d | �d S)N�originZ unknown_type�
approxname� named_ptr)r� rU � _struct_ctxr rH r ZNamedPointerType�totype)r r> r r
r
r �_generate_cpy_typedef_ctxX s !z$Recompiler._generate_cpy_typedef_ctxc C s7 | j | j � � | j r3 | j r3 | j | � d S)N)r] r^ �ellipsisr; )r r> r r
r
r �"_generate_cpy_function_collecttypec s z-Recompiler._generate_cpy_function_collecttypec C s| | j s t � t | t j � s( t � | j rE | j | | � d S| j } t | j � } | d k rr d } n | d k r� d } n d } g } g } d | } xK t
| j � D]: \ } }
| j |
j d | | � � | j d | � q� Wd j
| � } | p d
} | j r%| j d } n d } d
| | | f }
| d | j j |
� f � | d � d j
| � } d } t | j t j � r�d } | d | | | f � | d � | d � | d � | d | | f � | d � d | } xA t
| j � D]0 \ } }
|
j d | | � } | d | � qWt � } x! | j D] }
| j |
| � qIWx | D] } | d | f � qjWt | j t j � s�d } d | } d | j j d | � } | | � n d } d } t | j � d k rrt t | j � � } x | D] } | d | � q
W| � | d | t | � t | � d j
d d � | D� � f � | d � | � xB t
| j � D]1 \ } }
| j |
d | d | d � | � q�W| d! � | d" � d# d � t t | j � � D� } d j
| � } | d$ | | | f � | d% � | d& � | � | d' � | d k rU| d( � | r~| d) | j | j d* d+ � � n | d, � | d- � | d � | d. � d/ } g } g } d | } x� t
| j � D]p \ } }
d } t |
t j � rd0 } d1 } |
j d2 | | f | � } | j | � | j d3 | | f � q�W| j } t | t j � r�d | } | j d4 | � } | j d | � t j } d } d5 } d1 } | rSd j
| � } | p�d
} d6 | | | f }
| d | j |
� f � | d � | r| | � d j
| � } | d$ | | | f � | rF| d7 � | d � n | d8 | | f � | d9 � | � d S):Nr Znoargr Zarg0rG 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 | � q S)z&arg%dr
)r+ rZ 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 | � q S)zx%dr
)r+ rZ 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#elseFr 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; rE rH r rM r� �_generate_cpy_constant_declr� r rG rT rF r� r/ �abir� rI rz r� r� r� r� rR �insertZ void_type)r r> r r� �numargsZargname� argumentsZcall_argumentsr� rZ r� �repr_argumentsr� �name_and_argumentsZ result_code�argr� r� Z result_decl�rng�
differenceZ indirectionZ tp_resultr
r
r �_generate_cpy_function_declh s�
!
"
z&Recompiler._generate_cpy_function_declc C s� | j r' | j r' | j | | � d S| j | j � } t | j � } | j r[ t } n0 | d k rp t } n | d k r� t } n t
} | j d j t
| d | t | | � d d | �� d S)Nr r rq z
_cffi_f_%sr
z
_cffi_d_%s)r� r; �_generate_cpy_constant_ctxr@ r^ r rG ZOP_DLOPEN_FUNCZOP_CPYTHON_BLTN_NZOP_CPYTHON_BLTN_OZOP_CPYTHON_BLTN_Vrx rF r rW )r r> r r% r� Z meth_kindr
r
r �_generate_cpy_function_ctx� s
z%Recompiler._generate_cpy_function_ctxc C sx t | t j � rt | j } | d k rF | j d � } d | | f } | j | d | | j � } t j | | � } | S)Nz...r z _cffi_array_len(((%s)0)->%s)z%s[0])rH r r� rN r� rc r� )r Z tp_struct�
field_nameZtp_field�
actual_lengthZptr_struct_name�tp_itemr
r
r rc s
zRecompiler._field_typec C s | j | � d S)N)r] )r r> r
r
r �_struct_collecttype s z Recompiler._struct_collecttypec C s� | j d k r d S| j } d | f } | d � | d | | f � | d � | d � | d � x | j � D]� \ } } } } y� | j � s� | d k r� | d | | | f � wr xG t | t j � r| j d k s� | j d k r| j } | d
} q� W| d | j d d
| d | �| f � Wqr t
j k
rl}
z | d t |
� � WYd d }
~
Xqr Xqr W| 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 %rrn z
/* %s */r� z(struct _cffi_align_%s { char x; %s y; };)
r_ r� rb r� rH r r� rN r� r� r ri rC ) r r> �cnamer� r� Z
checkfuncname�fnameZftyper �fqualrp r
r
r � _struct_decl# s6
+
zRecompiler._struct_declNc C s} | j | } d } g } t | t j � r8 | j d � | j d k rZ | j d � d } | | j j j k r� | d k s� | | j j j k r� | j d k r� n% | j s� | j
� r� n
| j d � | j r� | j d � n | j d � d } d j | � pd } g } | d k r t
| j � � } x`| D]X\ }
} } }
| j | |
| � } | j | d
| j |
f � t } | d k r�t } d | } nd | d k s�t | t j � r�| j d k r�d
} n. d | d k r�| j d � n | j |
f } | d k s| d k r!d
} n; | d k rCd | j |
f } n d | j d � |
f } | j t |
| | | t | | j | � � � q3Wt | j d � } | j d j | � | d k r�d } d } d } q7| d k r�d | j f } d } n d | f } d | f } d } n d
} d } d } | } | j d j t | 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 rr z
(size_t)-2r3 Zunnamedzsizeof(*(%s)0)z-1 /* unknown alignment */z
sizeof(%s)z"offsetof(struct _cffi_align_%s, y)r rs ���r� r� ) r@ rH r Z UnionTyperF r_ r9 r` ra �partialZhas_anonymous_struct_fieldsZpackedr/ r� rb rc �_check_not_opaquer r r! r� rN r� r rW r rx �extendr$ r{ r� )r r> r� r� r� r% Zreason_for_not_expandingr& r* rb ZfldnameZfldtyper r� r r
�offsetr) Zalignr( r
r
r r� D s�
zRecompiler._struct_ctxc C sW x t | t j � r! | j } q Wt | t j � rS | j d k rS t d | � � d S)Nz0%s is of an opaque type (not declared in cdef()))rH r r� r� rR r_ r� )r r> �locationr
r
r r� � s
!z Recompiler._check_not_opaquec C s t | j j � � } | j d d d � � x� | D]� \ } } | | j k r2 | j ri t d | f � � | j j d � r� | j d d � j � r� | j d d � } nJ | j d k r� | j
d k r� d } | j | d � n t d | f � � | j | d | � q2 Wd S)
Nr< c S s | d S)Nr r
)Ztp_orderr
r
r r? � s z7Recompiler._add_missing_struct_unions.<locals>.<lambda>zDinternal inconsistency: %r is partial but was not seen at this point�$r Z_IO_FILEZFILEzinternal inconsistency: %r)
r� rP rf r} r{ r� r r �
startswith�isdigitZ forcenamer� r� )r rm r> �orderr� r
r
r r| � s
+
z%Recompiler._add_missing_struct_unionsc C s | j | � d S)N)r� )r r> r r
r
r � _generate_cpy_struct_collecttype� s z+Recompiler._generate_cpy_struct_collecttypec C sC | j d � } d | k r1 | | j d d � f S| d | f Sd S)Nr r re )r� r� )r r> r� r
r
r �
_struct_names� s zRecompiler._struct_namesc C s | j | | j | � � d S)N)r� r� )r r> r r
r
r �_generate_cpy_struct_decl� s z$Recompiler._generate_cpy_struct_declc C s | j | | j | � � d S)N)r� r� )r r> r r
r
r �_generate_cpy_struct_ctx� s z#Recompiler._generate_cpy_struct_ctxc C s6 t | t j � r% | j | | � n
| j | � d S)N)rH r rS � _generate_cpy_enum_collecttyper� )r r> r r
r
r �#_generate_cpy_anonymous_collecttype� s z.Recompiler._generate_cpy_anonymous_collecttypec C s= t | t j � r" | j | � n | j | | d | � d S)N�typedef_)rH r rS �_generate_cpy_enum_declr� )r r> r r
r
r � _generate_cpy_anonymous_decl� s z'Recompiler._generate_cpy_anonymous_declc C s@ t | t j � r% | j | | � n | j | | d | � d S)Nr� )rH r rS � _enum_ctxr� )r r> r r
r
r �_generate_cpy_anonymous_ctx� s z&Recompiler._generate_cpy_anonymous_ctx�constc C sS | | f | j k r. t j d | | f � � | j j | | f � | j } d | | f } | r� | d | � | d � | d | f � | d | | f � | d k r� | d k r� d | f } | d | f � | d
� | d � | d � nQ | d k s t � | d
| � | d � | d | j 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 ri r� r� rE r� )r �is_intr r> �categoryr r� �funcnamer
r
r �_generate_cpy_const� s2
z Recompiler._generate_cpy_constc C s- | j � } | s | j r) | j | � d S)N)r� r; r] )r r> 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 r> r r r
r
r r� s z&Recompiler._generate_cpy_constant_declc C s� | j r( | j � r( t t d � } n4 | j r: t } n t } | j | } t | | � } | j d j t | d | | � � d S)Nr rq z_cffi_const_%sr� )
r; r� rW �OP_CONSTANT_INTZOP_DLOPEN_CONSTZ OP_CONSTANTr@ rx rF r )r r> r r Z
const_kindr% r
r
r r� s
z%Recompiler._generate_cpy_constant_ctxc C s | j | � d S)N)r] )r r> r r
r
r r� s z)Recompiler._generate_cpy_enum_collecttypec C s( x! | j D] } | j d | � q
Wd S)NT)� enumeratorsr )r r> r �
enumeratorr
r
r r� s z"Recompiler._generate_cpy_enum_declc C sE | j | } t t d � } | j r/ | j � xM t | j | j � D]6 \ } } | j d j t
| d | | d | �� qE W| d k r� d | k r� | j r� d | } d | } nN | j | j g � } | j j
| � } t t | j j | d � � d k � } d j | j � }
| j d
j t | j | | | |
� � d S)
Nr rq z_cffi_const_%sr r� z
sizeof(%s)z
((%s)-1) <= 0r r- rt r� r� )r@ rW �OP_ENUMr; Zcheck_not_partial�zipr Z
enumvaluesrx rF r Zbuild_baseinttyper9 �sizeof�int�castr/ r0 r ) r r> r� r% r r Z enumvaluer
r1 Zbasetpr2 r
r
r r� s"
"
"
'
zRecompiler._enum_ctxc C s | j | | j � � d S)N)r� r� )r r> r r
r
r �_generate_cpy_enum_ctx1 s z!Recompiler._generate_cpy_enum_ctxc C s d S)Nr
)r r> r r
r
r � _generate_cpy_macro_collecttype7 s z*Recompiler._generate_cpy_macro_collecttypec C s5 | d k r d } n | } | j d | d | �d S)Nz...Tr )r )r r> r r r
r
r �_generate_cpy_macro_decl: s z#Recompiler._generate_cpy_macro_declc C sw | d k r4 | j r+ t j d | f � � d } n | } t t d � } | j d j t | d | | d | �� d S)Nz...zGcannot use the syntax '...' in '#define %s ...' when using the ABI moder rq z_cffi_const_%sr r� )r; r ri rW r rx rF r )r r> r r r r
r
r �_generate_cpy_macro_ctxA s
z"Recompiler._generate_cpy_macro_ctxc C sc t | t j � r_ | j } | d k r4 d | f } | j | j d | � } t j | | � } | S)Nz...z_cffi_array_len(%s)z%s[0])rH r r� rN � _global_typer� )r r> Z global_namer� r� r
r
r r R s
zRecompiler._global_typec C s | j | j | | � � d S)N)r] r )r r> r r
r
r �"_generate_cpy_variable_collecttype[ s z-Recompiler._generate_cpy_variable_collecttypec C s� | j } | j | | � } t | t j � rN | j d k rN | j } d } n d } d | f } | d | j | d | j �� | d � | d | | f � | d � | � d S) Nr �&z*_cffi_var_%s(void)zstatic rn r� z return %s(%s);r� ) r� r rH r r� rN r� r� rj )r r> r r� Z ampersandr� r
r
r �_generate_cpy_variable_decl^ s !
z&Recompiler._generate_cpy_variable_declc C sh | j | | � } | j | } | j r1 t } n t } | j d j t | d | t | | � � � d S)Nrq z _cffi_var_%s) r r@ r; Z
OP_GLOBAL_VARZOP_GLOBAL_VAR_Frx rF r rW )r r> r r% r r
r
r �_generate_cpy_variable_ctxu s
z%Recompiler._generate_cpy_variable_ctxc C s) t | t j � s t � | j | � d S)N)rH r rM rE r] )r r> r r
r
r �'_generate_cpy_extern_python_collecttype� s z2Recompiler._generate_cpy_extern_python_collecttypec C s� | j } t | j t j � r' d } n& d | } d | j j d | � f } | d | � | d | | f � | � g } d | } x@ t | j � D]/ \ } } | j d | | � }
| j |
� q� Wd j | � } | p� d
} d | | f } | j
d k r
d
| } d d � }
t t | j � d d � } |
| j � rSt | d � } t | j t j
� r�d | j j d � | | j j d � | f } | d | | j j | � f � | d � | d | � | d � x� t | j � D]q \ } } d | }
t | t j
� s |
| � r5d |
}
t j | � } | d | j d � | d |
f � q�W| d | � t | j t j � s�| d | j j d � f � | d � | � | j d 7_ d S) Nr� z result of %sz(int)sizeof(%s)r z3static struct _cffi_externpy_s _cffi_externpy__%s =z { "%s", %s };zargument of %sz a%dz, r� z%s(%s)� __stdcallz_cffi_stdcall c S s t | t j � o | j d k S)Nz long double)rH r �
PrimitiveTyper )r> r
r
r �may_need_128_bits� s z9Recompiler._extern_python_decl.<locals>.may_need_128_bitsr5 � 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� rH r� r rI r� rT rG rF r/ r� �maxr rR rK r� )r r> r Z
tag_and_spacer� Zsize_of_resultr� r� rZ 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 r> 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 r> 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 r> r r
r
r �'_generate_cpy_extern_python_plus_c_decl� s z2Recompiler._generate_cpy_extern_python_plus_c_declc C st | j r t j d � � | j r- t d � � | j | } t t | � } | j d j t
| d | | | � � d S)Nz,cannot use 'extern "Python"' in the ABI modez$a vararg function is extern "Python"rq z&_cffi_externpy__%s) r; r ri r� r r@ rW ZOP_EXTERN_PYTHONrx rF r )r r> r r% r r
r
r � _generate_cpy_extern_python_ctx� s
z*Recompiler._generate_cpy_extern_python_ctxc so d d � � g } xM | j d � p* d g D]3 } | j d d j � f d d � | D� � � q+ Wd j | � S) Nc S sR | d k r d | Sd | k o+ d k n r4 | S| d k rD d Sd t | � S)Nz\"?�\r �r z\nz\%03o)�ord)�cr
r
r �
_char_repr� s z.Recompiler._string_literal.<locals>._char_reprTr z"%s"c s g | ] } � | � � q Sr
r
)r+ r$ )r% r
r r, � s z.Recompiler._string_literal.<locals>.<listcomp>z \
)�
splitlinesrF r/ )r �sr� �liner
)r% r r� � s
1zRecompiler._string_literalc C s t t t � | j | <d S)N)rW � OP_PRIMITIVEZ PRIM_VOIDrD )r r> r� r
r
r �_emit_bytecode_VoidType� s z"Recompiler._emit_bytecode_VoidTypec C s' t | j } t t | � | j | <d S)N)ZPRIMITIVE_TO_INDEXr rW r) rD )r r> r� r6 r
r
r � _emit_bytecode_PrimitiveType� s
z'Recompiler._emit_bytecode_PrimitiveTypec C s6 d | j | j | j f } t t | � | j | <d S)Nzl_cffi_prim_int(sizeof(%s), (
((%s)-1) | 0 /* check that %s is an integer type */
) <= 0))r rW r) rD )r r> r� r' r
r
r �!_emit_bytecode_UnknownIntegerType� s z,Recompiler._emit_bytecode_UnknownIntegerTypec C s0 d | j | j f } t t | � | j | <d S)Nzd_cffi_prim_float(sizeof(%s) *
(((%s)1) / 2) * 2 /* integer => 0, float => 1 */
))r rW r) rD )r r> r� r' r
r
r � _emit_bytecode_UnknownFloatType� s z*Recompiler._emit_bytecode_UnknownFloatTypec C s t t | j | j � | j | <| d 7} xo | j D]d } | j | } | | k r� t | t j � rx | j | | � n t t
| � | j | <| d 7} q4 Wt | j � } | j
d k r� | j
d k r� | d O} n t d | j
f � � t t | � | j | <d S)Nr r r3 zabi=%r)rW Z OP_FUNCTIONr@ r� rD rG rH r r r+ r r r� r� r ZOP_FUNCTION_END)r r> r� rY Z realindexr& r
r
r � _emit_bytecode_RawFunctionType� s
z)Recompiler._emit_bytecode_RawFunctionTypec C s$ t t | j | j � | j | <d S)N)rW �
OP_POINTERr@ r� rD )r r> r� r
r
r �_emit_bytecode_PointerType s z%Recompiler._emit_bytecode_PointerTypec C s- | j � } t t | j | � | j | <d S)N)r^ rW r/ r@ rD )r r> r� �rawr
r
r � _emit_bytecode_FunctionPtrType s z)Recompiler._emit_bytecode_FunctionPtrTypec C s� | j | j } | j d k r8 t t | � | j | <n� | j d k rr t j d t | � j d d � f � � nV | j | d d k s� t
� t t | � | j | <t d 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 r= ) r@ r� rN rW Z
OP_OPEN_ARRAYrD r ri rC r� rE ZOP_ARRAY)r r> r� Z
item_indexr
r
r �_emit_bytecode_ArrayType s " z#Recompiler._emit_bytecode_ArrayTypec C s' | j | } t t | � | j | <d S)N)rP rW ZOP_STRUCT_UNIONrD )r r> r� Z struct_indexr
r
r �_emit_bytecode_StructType# s
z$Recompiler._emit_bytecode_StructTypec C s' | j | } t t | � | j | <d S)N)rQ rW r rD )r r> r� Z
enum_indexr
r
r �_emit_bytecode_EnumType( s
z"Recompiler._emit_bytecode_EnumType)Sr r r r� r r\ r] rA ry r~ r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� rc 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� 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. r0 Z _emit_bytecode_ConstPointerTypeZ _emit_bytecode_NamedPointerTyper2 r3 r4 Z_emit_bytecode_UnionTyper5 r
r
r
r r8 y s� : �
3 ( � !P
5
r8 � c s" e Z d Z � f d d � Z � S)�NativeIOc s8 t | t � r | j d � } t t | � j | � d S)N�ascii)rH �unicode�encode�superr7 r� )r r' )rV r
r r� 1 s zNativeIO.write)r r r r� r
r
)rV r r7 0 s r7 c
C s` | r t d | f � t | | d | d k �} | j � | j � t � } | j | | � | j � } yV t | d � �, } | j t | � d � | k r� t
� Wd QRX| r� t d � d SWn� t
k
r[d | t j � f } t | d � � } | j
| � Wd QRXy t j | | � Wn/ t k
rRt j | � t j | | � Yn Xd SYn Xd S)
Nz
generating %sr; r� r z(already up-to-date)Fz%s.~%d�wT)�printr8 r\ r~ r7 r� �getvaluer� �readr �IOErrorr� �getpidr� �rename�OSError�unlink)
r9 r: r� Z target_file�verboseZ
recompilerr� �output�f1Ztmp_filer
r
r �_make_c_or_py_source6 s4
rH Fc C s( | d k s t � t | | | | | � S)N)rE rH )r9 r: r� Z
target_c_filerE r
r
r �
make_c_sourceR s rI c C s t | | d | | � S)N)rH )r9 r: Ztarget_py_filerE r
r
r �make_py_sourceW s rJ c C su | j d � } y* t j t j j | | d d � � � Wn t k
rM Yn X| d | 7<t j j | | � | f S)Nr� r r� r� )rg r� �makedirsr� r/ rC )Z outputdir�modname� extension�partsr
r
r �_modname_to_file[ s *
rO c C s9 t | | � } | j | | | f � t | | | � | S)N)rU rF �setattr)� patchlist�clsr Znew_meth�oldr
r
r � _patch_methi s rT c C s4 x- t | � D] \ } } } t | | | � q
Wd S)N)�reversedrP )rQ rR r Zold_methr
r
r �_unpatch_methso s rV c s� t j d k r8 d d l m } t | | d d d � � t j d k r~ d d l m } � f d d
� } t | | d | � � d S) N�win32r )� MSVCCompilerZ_remove_visual_c_refc S s | S)Nr
)r Z
manifest_filer
r
r r? x s z&_patch_for_embedding.<locals>.<lambda>�darwin)� CCompilerc sP d | j k r@ t | j � | _ | j j d � } d | j | <� | | | � S)Nz-bundlez -dynamiclib)� linker_sor� r� )r rG �kwdsrZ )�old_link_shared_objectr
r �my_link_shared_object} s
z3_patch_for_embedding.<locals>.my_link_shared_object�link_shared_object)�sys�platformZdistutils.msvc9compilerrX rT �distutils.ccompilerrZ )rQ rX rZ r^ r
)r] r �_patch_for_embeddings s
rc c s� d d l m } � j d � rq � d d
� � t j d k rK � d 7� n& t j d k rg � d 7� n
� d 7� t | | d
� f d d � � d S)Nr )� build_extz.*r3 rW z.dllrY z.dylibz.soZget_ext_filenamec s � S)Nr
)r Zext_name)�targetr
r r? � s z#_patch_for_target.<locals>.<lambda>r� )Zdistutils.command.build_extrd �endswithr` ra rT )rQ re rd r
)re r �_patch_for_target� s
rg r� Tz.cc K s� t | t � s | j d � } | j r4 | j | � | d k r�| j d k } | rb | j | � | d k r� t | | | � \ } }
| r� | g |
}
t j j
|
� } n | } | d k r� | r� d | } n d } t j | | | � } t
| | | | d | �} | r�g } t j � } zR | r4t | � | d k rMt | | � t j | � t j d | | |
� } Wd t j | � t | � X| S| | f SnP | d k r�t | | d � \ } } t | | | d | �} | r�| Sd | f Sd S)Nr8 z%s.*r rE r� z.py)rH rC r: Z_windows_unicodeZ_apply_windows_unicoder� Z_apply_embedding_fixrO r� r� r/ r Z
get_extensionrI �getcwdrc rg �chdir�compilerV rJ )r9 r: r� �tmpdirZcall_c_compilerZc_fileZsource_extensionZextradirZcompiler_verbosere �debugr\ Z embeddingrN Z
ext_c_file�ext�updatedrQ �cwd�outputfilenamere r
r
r � recompile� sX
rq c O sJ d d l m } d d l } | t j k s> t d | f � � | j d t | � � t | | | | | � } | j | | � } xc t
| j � D]R } | j d � s� t
| j | � }
|
t
| | t � � k r� t | | |
� q� Wd d � } | | _ xG t
| � D]9 } | j d � rt | j | � rt | | t � qW| j S)Nr )�udirzmodule name conflict: %rrk re c _ s
t � d S)N)r )rG r\ r
r
r �typeof_disabled� s z _verify.<locals>.typeof_disabled)Z testing.udirrr �impr` �modulesrE �
setdefaultrC rq � load_dynamic�dirr9 r� rU �objectrP Z_typeof�hasattr�NotImplemented�lib) r9 r: r� rG r\ rr rt rp �moduler �attrrs r
r
r �_verify� s$
#r ) r� r` �ior r r Z cffi_opcoder� r� r r r$ r0 r7 r8 � version_info�StringIOr7 �BytesIOrH rI rJ rO rT rV rc rg rq r r
r
r
r �<module> s: $
� � � � � 6