3
MZ @ 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 dZG dd dZG dd dZG dd dZejd0krejZnG dd dejZdd Zd1dd Zd2d d Zd d Zd!d" Zd#d$ Zd%d& Z d'd( Z d3d,d-Z d.d/ Z dS )4 N ) ffiplatformmodel)VerificationError)*i& i' i( c @ s& e Zd Zd ddZdd Zdd ZdS )
GlobalExprr c C s" || _ || _|| _|| _|| _d S )N)nameaddresstype_opsize 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_sizefbitsize
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}ntd| j j || jf S )N z b'%s%s%s') r opOP_NOOP OP_BITFIELDformat_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_indexflagsr alignmentcommentfirst_field_indexc_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| jf | 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| jt|| jdj|f S )Nc S s g | ] }|j qS r )r$ ).0Zc_fieldr r r
L s z2StructUnionExpr.as_python_expr..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 signedallenums)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 ZdZdddZdd Zdd Zd d
Zd d Zd
ddddgZ dd Z
dddZ 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 ddBdCZ!dDdE Z"dFdG Z#dHdI Z$e$Z%dJdK Z&dLdM Z'e'Z(dNdO Z)e)Z*dPdQ Z+dRdS Z,dTdU Z-ddWdXZ.dYdZ Z/d[d\ Z0d]d^ Z1d_d` Z2ddadbZ3dcdd Z4dedf Z5dgdh Z6didj Z7dkdl Z8dmdn Z9dodp Z:dqdr Z;dsdt ZZ?dwdx Z@dydz ZAd{d| ZBd}d~ ZCdd ZDeD ZEZFdd ZGdd ZHdd ZIdd ZJdd ZKdd ZLdd ZMeMZNeMZOdd ZPdd ZQdd ZReRZSdd ZTdAS )
Recompilerr Fc C s || _ || _|| _t| _d S )N)ffi module_nametarget_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 sP i | _ | jd t| j td}g | _x|D ]}|jr*| j | d ksFtt| j| j |< | jj| xZ|j D ]P}t
|t j t j
t jt jt jfst| j | d krt| j| j |< | jj| qjW | jjd q*W xZ|D ]R}|j o| j | d krt| j| j |< | jj| |jr|jd k r| jjd qW d | j j ks>ti | _i | _x@|D ]8}t
|t jrnd | j|< nt
|t jrPd | j|< qPW x.tt| jdd dD ]\}}|| j|< qW x.tt| jdd dD ]\}}|| j|< qW x0|D ](}t| d|jj }||| j | qW x | jD ]}t
|t s&tq&W t | j| _d S ) NZ collecttype)keyZENDLENc S s | j S )N)r )tpr r r s z/Recompiler.collect_type_table..c S s | j S )N)r )rC r r r rD s Z_emit_bytecode_)
_typesdict _generatesortedstr
cffi_typesZis_raw_functionAssertionErrorr appendargs
isinstancer VoidTypeBasePrimitiveType PointerTypeStructOrUnionOrEnumFunctionPtrTypeZ
is_array_typelengthvalues_struct_unions_enums
StructOrUnionEnumType enumerategetattr __class__r CffiOptuple)r
Z all_declsrC tp1imethodr r r r collect_type_table sZ
z Recompiler.collect_type_tablec C s t |tjs2t |tr.x|D ]}| j| q W d S || jkrd | j|< t |tjrb| j|j ntt |tjr|j d k r|| j
j j krxN|j
D ] \}}}}| j| j||| qW n x |j D ]\}}| j| qW d S )N)rM r ZBaseTypeByIdentityr] _do_collect_typerE rR as_raw_functionrW fldtypesr: _parser_included_declarations
enumfields _field_typeZ
_get_items)r
rC xZname1r^ _r r r rb s
zRecompiler._do_collect_typec
C s | j jjj }xt|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: re Z
_declarationsitemsrG splitrZ AttributeErrorr _current_quals Exceptionr Zattach_exception_info)
r
step_namelstr rC qualsZkindZrealnamer` er r r rF s zRecompiler._generateglobalfield struct_unionenumtypenamec 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 ],\}}|t
|k st || j |j kst qW t
|t
| jkst | j d }x<| j
j D ].\}}|t
|k st || j |j kst qW t
|t
| j
ks6t d S )NZctxru c S s | j S )N)r )entryr r r rD s z0Recompiler.collect_step_tables..)rA rv rw )_lsts ALL_STEPSset_seen_struct_unionsrF _add_missing_struct_unionssortr] rU rk r rJ r rV )r
rp rq rC r_ r r r collect_step_tables s*
z Recompiler.collect_step_tablesr c C s | j j|d d S )N
)_fwrite)r
Zwhatr r r _prnt s zRecompiler._prntc C s: | j r |d kst| j| n|d k s*t| j|| d S )N)r< rJ write_py_source_to_fwrite_c_source_to_f)r
fpreambler r r write_source_to_f s
z Recompiler.write_source_to_fc C s0 t tjjtjjt|d}|j }|j |S )Nr)openospathr0 dirname__file__ readlinesclose)r
filenameglinesr r r _rel_readlines s zRecompiler._rel_readlinesc C sz || _ | j}| jjd k r |d | jd}|jd}| jd|||d < |dj| | jjddI }| 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| jD ]>\}}d} ||krd|| j } |d||j | f qW | js|d |d | t | _| jd i }
xn| jD ]d} | j| } t| |
| < |
| d kr|d | | f x| D ]}
||
j qBW |d | qW | jjr|d xr| jjD ]f}y|jd d! \}}W n( tk
r t d"| j|f Y nX |d krt d#|d$|f qW |d% |d | |d& |d' x8| jD ].} |
| d kr@|d(| n |d)| q W x.| jD ]$} | d*krZ|d+|
| | f qZW | jjr|d, n|d- |d.t| jf 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 | ]\}}||fqS r r )r, rC r_ r r r r- G s z2Recompiler.write_c_source_to_f..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 */ru 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) r r r:
_embeddingr indexr0 r; rl _print_string_literal_in_arrayr@ VERSION_EMBEDDEDdictrE rk rY rI _get_c_namer r| _seen_constantsrF r{ rz r _included_ffis_assigned_sourcerm r
_num_externpyr> )r
r r prntr r_ Zbase_module_nametypeindex2typer r) Znumsrp rq ry ffi_to_includeincluded_module_nameincluded_sourcer' r r r r s
z Recompiler.write_c_source_to_fc sb t |trd|f S t |ttfrZ 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..r r z(%s)r. )rM rH listr] r rK r0 r )r
ri 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 rtd|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| dkr*|
dkr*|d|
| j
| f q*W |dkr|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..z _types = %s,r c S s g | ]\}}||fqS r r )r, rC r_ r r r r- s r ru z _%ss = %s,z _includes = (%s,),z, c S s g | ] }d | qS )z_ffi%dr )r, r_ r r r r- s ))r r r r: r ranger rm r r; r> r] rI r r0 r rE rk r{ rz ) r
r r Z num_includesr_ r r r Z types_lstr rp rq r r r r sB
z Recompiler.write_py_source_to_fc C s
| j | S )N)rE )r
typer r r _gettypenum s zRecompiler._gettypenumc C sp d}t |tjr|j r|j r<|jdkr.z return NULL;zarg%dz return NULLz Py_BEGIN_ALLOW_THREADSz _cffi_restore_errno();c S s g | ] }d | qS )zx%dr )r, r_ 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 | tjp t | tjo | j S )N)rM r rW
PrimitiveTyper )r r r r need_indirection s z@Recompiler._generate_cpy_function_decl..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< rJ rM r rR r _generate_cpy_constant_declr r rL rY rK r r0 abir rN r| r r r r insertZ void_type)r
rC r r numargsZargname argumentsZcall_argumentsr r_ r repr_argumentsr name_and_argumentsZ result_codeargr r Z result_declrngr
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 rt z
_cffi_f_%sz
_cffi_d_%s)r )r r< _generate_cpy_constant_ctxrE rc r rL ZOP_DLOPEN_FUNCZOP_CPYTHON_BLTN_NZOP_CPYTHON_BLTN_OZOP_CPYTHON_BLTN_Vrz rK r r\ )r
rC r r& r Z meth_kindr r r _generate_cpy_function_ctx# s
z%Recompiler._generate_cpy_function_ctxc C sT t |tjrP|j}|dkr0|jd}d||f }| j|d| |j}tj||}|S )Nz...r z _cffi_array_len(((%s)0)->%s)z%s[0])rM r r rS r rh r )r
Z tp_structZ
field_nameZtp_field
actual_lengthZptr_struct_nametp_itemr r r rh 9 s
zRecompiler._field_typec C s0 | j | | jr,x|j D ]}| j| qW d S )N)rb r< anonymous_struct_fields_struct_collecttype)r
rC 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|tjr|jd ks|jd kr|j}|d
}qW |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)rr z
/* %s */r z(struct _cffi_align_%s { char x; %s y; };) rd r rg r rM r r rS r r r rH ) r
rC r r r Z
checkfuncnameZfnameZftyper fqualrs r r r _struct_declL s6
&zRecompiler._struct_declNc C s | j | }d }g }t|tjr(|jd |jd kr@|jd d}|| jjjkr|d ksd|| jjjkr|jd krpn |j st
|j rn
|jd |j r|jd n|jd d}dj
|pd }g }|d krZ| j } t|j| }
x
|
D ] \} } }
}| j|| | } | j| d
|j| f t}|
d kr6t}d |
}nL|d ksZt| tjr`| jd kr`d
}n"d|d krv|jdn|j| f }|d ks|
d krd
}n,|d k rd|j| f }nd|jd| f }|jt| |||
t|| j | qW t| j d }| j d j | |d kr$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 ru z
(size_t)-2r4 Zunnamedzsizeof(*(%s)0)z-1 /* unknown alignment */z
sizeof(%s)z"offsetof(struct _cffi_align_%s, y)r rv r r )!rE rM r Z UnionTyperK rd r: re rf partialanyr Zpackedr0 r< r rg rh _check_not_opaquer r! r" r rS r r r\ r rz extendr% r} r )r
rC r r r r& Zreason_for_not_expandingr' r+ Z expand_anonymous_struct_unionrg Zfldnamer r r r r offsetr* Zalignr) r r r r m s
zRecompiler._struct_ctxc C s> xt |tjr|j}qW t |tjr:|jd kr:td| d S )Nz0%s is of an opaque type (not declared in cdef()))rM r r r rW rd r )r
rC locationr r r r s
z Recompiler._check_not_opaquec C s t | jj }|jdd d x|D ]\}}|| jkr$|jrJtd|f |jjdrx|jdd j rx|jdd }n4|jdkr|j
dkrd}| j |d ntd |f | j |d | q$W d S )
Nc S s | d S )Nr r )Ztp_orderr r r rD s z7Recompiler._add_missing_struct_unions..)rA zDinternal inconsistency: %r is partial but was not seen at this point$r Z_IO_FILEFILEzinternal inconsistency: %r)
r rU rk r r} r r r
startswithisdigitZ forcenamer r )r
rq rC orderr r r r r~ s
z%Recompiler._add_missing_struct_unionsc C s | j | d S )N)r )r
rC r r r r _generate_cpy_struct_collecttype s z+Recompiler._generate_cpy_struct_collecttypec C s2 |j d}d|kr"||jddfS |d| fS d S )Nr r rj )r r )r
rC r r r r
_struct_names s
zRecompiler._struct_namesc C s | j |f| j| d S )N)r r )r
rC 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
rC r r r r _generate_cpy_struct_ctx s z#Recompiler._generate_cpy_struct_ctxc C s( t |tjr| j|| n
| j| d S )N)rM r rX _generate_cpy_enum_collecttyper )r
rC r r r r #_generate_cpy_anonymous_collecttype s z.Recompiler._generate_cpy_anonymous_collecttypec C s. t |tjr| j| n| j||d| d S )Ntypedef_)rM r rX _generate_cpy_enum_declr )r
rC r r r r _generate_cpy_anonymous_decl s z'Recompiler._generate_cpy_anonymous_declc C s0 t |tjr| j|| n| j||d| d S )Nr )rM r rX _enum_ctxr )r
rC r r r r _generate_cpy_anonymous_ctx s z&Recompiler._generate_cpy_anonymous_ctxconstc C s ||f| j kr td||f | j j||f | j}d||f }|r|d| |d |d|f |d||f |d k r|dkrd|f }|d |f |d
|d |d n>|d kst|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 rJ r )r
is_intr rC categoryr r funcnamer r r _generate_cpy_const s2
z Recompiler._generate_cpy_constc C s" |j }| s| jr | j| d S )N)r r< rb )r
rC 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
rC 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 rt z_cffi_const_%sr )
r< r r\ OP_CONSTANT_INTZOP_DLOPEN_CONSTZ OP_CONSTANTrE rz rK r )r
rC r r
Z
const_kindr& r r r r / s
z%Recompiler._generate_cpy_constant_ctxc C s | j | d S )N)rb )r
rC r r r r r ? s z)Recompiler._generate_cpy_enum_collecttypec C s x|j D ]}| jd| qW d S )NT) enumeratorsr
)r
rC r
enumeratorr r r r B s z"Recompiler._generate_cpy_enum_declc C s | j | }ttd }| jr"|j x:t|j|jD ](\}}| jd j t
|d| ||d q2W |d k rd|kr| j rd| }d| }n4|j | j g } | j j
| }tt| j j| d dk }d j|j}
| jd
j t|j||||
d S )
Nr rt z_cffi_const_%s)r r z
sizeof(%s)z
((%s)-1) <= 0r r. rw r r )rE r\ OP_ENUMr< Zcheck_not_partialzipr
Z
enumvaluesrz rK r Zbuild_baseinttyper: Zsizeofintcastr0 r1 r ) r
rC 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
rC r r r r _generate_cpy_enum_ctx[ s z!Recompiler._generate_cpy_enum_ctxc C s d S )Nr )r
rC 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
rC 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 rt z_cffi_const_%s)r r )r< r r\ r rz rK r )r
rC r r r
r r r _generate_cpy_macro_ctxk s
z"Recompiler._generate_cpy_macro_ctxc C sF t |tjrB|j}|dkr$d|f }| j|jd| }tj||}|S )Nz...z_cffi_array_len(%s)z%s[0])rM r r rS r r )r
rC Z global_namer r r r r r | s
zRecompiler._global_typec C s | j | j|| d S )N)rb r )r
rC r r r r "_generate_cpy_variable_collecttype s z-Recompiler._generate_cpy_variable_collecttypec C s | j }| j||}t|tjr4|jd kr4|j}d}nd}d|f }|d|j|| jd |d |d||f |d | d S ) Nr &z*_cffi_var_%s(void)zstatic )rr r z return %s(%s);r ) r r rM r r rS r r rn )r
rC 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 )Nrt z _cffi_var_%s) r rE r< Z
OP_GLOBAL_VARZOP_GLOBAL_VAR_Frz rK r r\ )r
rC r r& r r r r _generate_cpy_variable_ctx s
z%Recompiler._generate_cpy_variable_ctxc C s t |tjst| j| d S )N)rM r rR rJ rb )r
rC r r r r '_generate_cpy_extern_python_collecttype s z2Recompiler._generate_cpy_extern_python_collecttypec C s | j }t|jtjrd}n d| }d|jjd|f }|d| |d| j||f | g }d| }x0t|jD ]"\}} | jd| |}
|j |
qtW d j
|} | pd
} d || f } |j d krd
| } dd }
t t
|jd d}|
|jrt |d}t|jtjr*d|jjd||jjd|f }|d||jj| f |d |d| |d xft|jD ]X\}} d| }
t| tjs|
| rd|
}
tj| } |d| jd|d |
f qjW |d| t|jtjs|d |jjdf |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 | tjo| jdkS )Nz long double)rM r r r )rC r r r may_need_128_bits s z9Recompiler._extern_python_decl..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 rM r r rN r r; rY rL rK r0 r r? r rW rP r )r
rC r Z
tag_and_spacer Zsize_of_resultr r r_ 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
rC 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
rC 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
rC 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"rt z&_cffi_externpy__%s)
r< r r r rE r\ ZOP_EXTERN_PYTHONrz rK r )r
rC r r& r
r r r _generate_cpy_extern_python_ctx s
z*Recompiler._generate_cpy_extern_python_ctxc C sz | j }|d xf|jdD ]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
splitlinesrstripr ord)r
sr lineZ printed_linecr r r r s
z)Recompiler._print_string_literal_in_arrayc C s t tt| j|< d S )N)r\ OP_PRIMITIVEZ PRIM_VOIDrI )r
rC 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+ rI )r
rC 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+ rI )r
rC 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+ rI )r
rC 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|tjrX| 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_FUNCTIONrE r rI rL rM r r r- r! r r r r ZOP_FUNCTION_END)r
rC r r^ 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_POINTERrE r rI )r
rC r r r r _emit_bytecode_PointerType6 s z%Recompiler._emit_bytecode_PointerTypec C s" |j }tt| j| | j|< d S )N)rc r\ r1 rE rI )r
rC r rawr r r _emit_bytecode_FunctionPtrType< s z)Recompiler._emit_bytecode_FunctionPtrTypec C s | j |j }|jd kr(tt|| j|< nf|jdkrNtdt|jddf n@| j|d dksdt 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 rB ) rE r rS r\ Z
OP_OPEN_ARRAYrI r rH r rJ ZOP_ARRAY)r
rC r Z
item_indexr r r _emit_bytecode_ArrayType@ s
z#Recompiler._emit_bytecode_ArrayTypec C s | j | }tt|| j|< d S )N)rU r\ ZOP_STRUCT_UNIONrI )r
rC r Z struct_indexr r r _emit_bytecode_StructTypeN s
z$Recompiler._emit_bytecode_StructTypec C s | j | }tt|| j|< d S )N)rV r\ r rI )r
rC 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@ ra rb rF r{ r r r r r r r r r r r r r r r r r r r r rh 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/ r0 r2 Z _emit_bytecode_ConstPointerTypeZ _emit_bytecode_NamedPointerTyper4 r5 r6 Z_emit_bytecode_UnionTyper7 r r r r r9 { s
:
,
4-# !
Q
5
r9 c s e Zd Z fddZ ZS )NativeIOc s( t |tr|jd}tt| j| d S )Nascii)rM Zunicodeencodesuperr9 r )r
r( )r[ r r r \ s
zNativeIO.write)r r r r
__classcell__r r )r[ r r9 [ s r9 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 |rt 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.~%dwT)printr9 ra r r9 r getvaluer readr IOErrorr getpidr renameOSErrorunlink)
r: r; r Z target_fileverboseZ
recompilerr outputf1Ztmp_filer r r _make_c_or_py_sourcea s4
rJ Fc C s |d k s t t| ||||S )N)rJ rJ )r: r; r Z
target_c_filerG r r r
make_c_source} s
rK c C s t | |d ||S )N)rJ )r: r; Ztarget_py_filerG r r r make_py_source s
rL 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 )rl r makedirsr r0 rE )Z outputdirmodname extensionpartsr r r _modname_to_file s
$rQ c C s* t ||}| j|||f t||| |S )N)rZ rK setattr) patchlistclsr Znew_metholdr r r _patch_meth s
rV c C s( x"t | D ]\}}}t||| q
W d S )N)reversedrR )rS rT r Zold_methr r r _unpatch_meths s rX c s\ t jdkr(ddlm} t| |ddd t jdkrXddlm} fd d
}t| |d | d S ) Nwin32r ) MSVCCompilerZ_remove_visual_c_refc S s |S )Nr )r
Z
manifest_filer r r rD s z&_patch_for_embedding..darwin) CCompilerc s< d| j kr,t| j | _ | j jd}d| j |< | f||S )Nz-bundlez -dynamiclib)Z linker_sor r )r
rL kwdsr_ )old_link_shared_objectr r my_link_shared_object s
z3_patch_for_embedding..my_link_shared_objectZlink_shared_object)sysplatformZdistutils.msvc9compilerrZ rV Zdistutils.ccompilerr\ )rS rZ r\ r_ r )r^ r _patch_for_embedding s
rb c sl ddl m} jdrR d d
tjdkr6 d7 n tjdkrJ d7 n d 7 t| |d
fd d d S )Nr ) build_extz.*r4 rY z.dllr[ z.dylibz.soZget_ext_filenamec s S )Nr )r
Zext_name)targetr r rD s z#_patch_for_target..r )Zdistutils.command.build_extrc endswithr` ra rV )rS rd rc r )rd r _patch_for_target s
rf r T.cc K s t |ts|jd}| jr$| j| |d k r^| jd k } | rF| j| |d krzt|||\}}
|rl|g|
}
tj j
|
}n|}| d kr| rd| } nd} t j ||f| }t
| ||||d}|rTg }tj }zn| rt| | dkrt|| |r |dkrd}nd}td|tj j|f tj| t jd|||
}W d tj| t| X |S ||fS n<|d krxt||d \}}t| |||d}|r|S d |fS d S )
Nr: z%s.*r )rG r zthe current directory isz setting the current directory toz%s %rz.py)rM rH r; Z_windows_unicodeZ_apply_windows_unicoder Z_apply_embedding_fixrQ r r r0 r Z
get_extensionrK getcwdrb rf r? abspathchdircompilerX rL )r: r; r tmpdirZcall_c_compilerZc_fileZsource_extensionZextradirZcompiler_verboserd debugr] Z embeddingrP Z
ext_c_fileZextupdatedrS cwdmsgoutputfilenamerj r r r recompile sb
rr c O s ddl m} dd l}|tjks,td|f |jdt| t| ||f||}|j ||}xFt
|j D ]8} | j dsht
|j | }
|
t
| | t k rht| | |
qhW dd } | | _x6t
| D ]*} | j d rt|j | rt| | t qW |jS )Nr )udirzmodule name conflict: %rrl rj c _ s t d S )N)r )rL r] r r r typeof_disabled s z _verify..typeof_disabled)Z testing.udirrs impr` modulesrJ
setdefaultrH rr Z load_dynamicdirr: r rZ objectrR Z_typeofhasattrNotImplementedlib) r: r; r rL r] rs ru rq moduler attrrt r r r _verify s$
r )r8 )F)F)r TNrg Nr NN) r r` ior r r errorr Z cffi_opcoder= r r r r r% r1 r8 r9 version_infoStringIOr9 BytesIOrJ rK rL rQ rV rX rb rf rr r r r r r sH f
<