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 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 krt| j| j |< | jj| |j r|j
d k r| jjd qW i | _ i | _ x@|D ]8}t
|tjr&d | j |< nt
|tjrd | j |< qW x.tt| j dd dD ]\}}|| j |< q\W x.tt| j dd dD ]\}}|| j |< qW x0|D ](}t| d|jj }||| j | qW x| jD ]}qW t| j| _d S ) NZ collecttype)keyZENDZLENc S s | j S )N)r )tpr r r s z/Recompiler.collect_type_table..c S s | j S )N)r )rB r r r rC s Z_emit_bytecode_)
_typesdict _generatesortedstr
cffi_typesZis_raw_functionr appendargsZ
is_array_typelength_struct_unions_enums
isinstancer
StructOrUnionEnumType enumerategetattr __class__r tuple)r
Z all_declsrB tp1imethodr r r r collect_type_table sL
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)rN r ZBaseTypeByIdentityrT _do_collect_typerD FunctionPtrTypeas_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 }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: r] Z
_declarationsitemsrF splitrR AttributeErrorr _current_quals Exceptionr Zattach_exception_info)
r
step_namelstr rB qualsZkindZrealnamerW er r r rE 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 ]\}}qW | j d }x| j
j D ]\}}qW d S )NZctxrm c S s | j S )N)r )entryr r r rC s z0Recompiler.collect_step_tables..)rA rn ro ) _lsts ALL_STEPSset_seen_struct_unionsrE _add_missing_struct_unionssortrT 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
)_fwrite)r
Zwhatr r r _prnt s zRecompiler._prntc C s" | j r| j| n | j|| d S )N)r< 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, rB rV 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 */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_EMBEDDEDdictrD 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_nametypeindex2typer r) Znumsrh ri rq 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. )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 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, 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 |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, 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 | tjp t | tjo | j S )N)rN r rO
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 _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_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 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 |tjrP|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_nametp_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|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)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|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 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^ partialanyr 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 |tjr|j}qW t |tjr:|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 |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 rC 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 rL rc rw ru r r r
startswithisdigitZ 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"||jddfS |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 |tjr| 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 |tjr| j| n| j||d| d S )Ntypedef_)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 |tjr| 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_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 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|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 rl z_cffi_const_%s)r r z
sizeof(%s)z
((%s)-1) <= 0r r. ro r r )rD r OP_ENUMr< Zcheck_not_partialzipr Z
enumvaluesrr rI r Zbuild_baseinttyper: Zsizeofintcastr0 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 |tjrB|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|tjr4|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|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)rN r r r )rB 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| 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|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_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|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_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|jddf 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 fddZ ZS )NativeIOc s( t |tr|jd}tt| j| d S )Nascii)rN Zunicodeencodesuperr5 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 |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 rX rx r5 r getvaluer readr IOErrorr getpidr{ renameOSErrorunlink)
r: r; r Z target_fileverboseZ
recompilerr outputf1Ztmp_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 outputdirmodname extensionpartsr r r _modname_to_file s
$rM c C s* t ||}| j|||f t||| |S )N)rR rI setattr) patchlistclsr Znew_metholdr 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 ) Nwin32r ) MSVCCompilerZ_remove_visual_c_refc S s |S )Nr )r
Z
manifest_filer r r rC 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
rJ kwdsrV )old_link_shared_objectr r my_link_shared_object s
z3_patch_for_embedding..my_link_shared_objectZlink_shared_object)sysplatformZdistutils.msvc9compilerrV rR Zdistutils.ccompilerrX )rO rV rX r[ r )rZ r _patch_for_embedding s
r^ 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 rU z.dllrW z.dylibz.soZget_ext_filenamec s S )Nr )r
Zext_name)targetr r rC s z#_patch_for_target..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 |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 )
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; abspathchdircompilerT rH )r: r; r tmpdirZcall_c_compilerZc_fileZsource_extensionZextradirZcompiler_verboser` debugrY Z embeddingrL Z
ext_c_fileZextupdatedrO cwdmsgoutputfilenamerb r r r recompile sb
rn c O s ddl m} dd l}|jdt| t| ||f||}|j||}xFt|jD ]8} | j dsPt
|j| }
|
t
| | t k rPt | | |
qPW dd } | | _
x6t| D ]*} | j d rt|j| rt | | t qW |jS )Nr )udirrh rb c _ s t d S )N)r )rJ rY r r r typeof_disabled s z _verify..typeof_disabled)Z testing.udirro imp
setdefaultrG rn Z load_dynamicdirr: r rR objectrN Z_typeofhasattrNotImplementedlib) 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_infoStringIOr5 BytesIOrF rG rH rM rR rT r^ rb rn rz r r r r sH f
<