B
    1d<h                 @   sH   d dl Z d dlZd dlZddlmZ ddlmZ G dd deZdZ	dS )    N   )model)VerificationErrorc               @   s  e Zd ZdZdZdd Zdd Zdd Zd	d
 ZdZddZ	dd Z
d[ddZdd Zdd Zdd Zdd Zdd ZeZeZeZdd Ze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d1d2 Zd3d4 Z d5d6 Z!d7d8 Z"d9d: Z#d\d=d>Z$d?d@ Z%eZ&d]dAdBZ'dCdD Z(dEdF Z)dGdH Z*dIdJ Z+d^dLdMZ,d_dNdOZ-dPdQ Z.dRdS Z/eZ0dTdU Z1dVdW Z2eZ3dXdY Z4d;S )`VGenericEnginegFc             C   s   || _ |j| _g | _i | _d S )N)verifierffiexport_symbols_struct_pending_verification)selfr    r   ]/work/yifan.wang/ringdown/master-ringdown-env/lib/python3.7/site-packages/cffi/vengine_gen.py__init__   s    zVGenericEngine.__init__c             C   s   | d| j d S )Nr	   )
setdefaultr	   )r   kwdsr   r   r   patch_extension_kwds   s    z#VGenericEngine.patch_extension_kwdsc             C   sT   xN|D ]F}|| }|d kr t j}x*|D ]"}tj||}tj|r&|S q&W qW d S )N)syspathosjoinisfile)r   module_namer   Zso_suffixesZ	so_suffixbasenamedirnamefilenamer   r   r   find_module   s    

zVGenericEngine.find_modulec             C   s   d S )Nr   )r   r   r   r   collect_types%   s    zVGenericEngine.collect_types c             C   s   | j |d  d S )N
)_fwrite)r   whatr   r   r   _prnt(   s    zVGenericEngine._prntc             C   s`   | j }|t || jj | d tjdkr\tjdkr>d}nd}| j }|d||f  d S )Ndeclwin32)   ZPyInit_initzvoid %s%s(void) { }
)	r"   cffimod_headerr   preamble	_generater   platformversion_infoZget_module_name)r   prntprefixmodnamer   r   r   write_source_to_f+   s    



z VGenericEngine.write_source_to_fr   c                sj   j j}tjtjjj}|||	d G  fdddt
j  d}j	d|d |S )NZloadingc                   s&   e Zd ZZjZg Z fddZdS )z/VGenericEngine.load_library.<locals>.FFILibraryc                s    j S )N)	_cffi_dir)r   )
FFILibraryr   r   __dir__T   s    z7VGenericEngine.load_library.<locals>.FFILibrary.__dir__N)__name__
__module____qualname__Z_cffi_generic_moduler   Z	_cffi_ffir0   r2   r   )r1   moduler   r   r   r1   P   s   r1   r   Zloaded)library)r   _backendr   r   r   curdirr   Zmodulefilenameload_library_loadtypes
ModuleType)r   flagsbackendr   r7   r   )r1   r6   r   r   r:   A   s    zVGenericEngine.load_libraryc             C   s$   dd | j jj D }|  |S )Nc             S   s   g | ]\}\}}||fqS r   r   ).0keytpZqualr   r   r   
<listcomp>^   s    z4VGenericEngine._get_declarations.<locals>.<listcomp>)r   _parserZ_declarationsitemssort)r   lstr   r   r   _get_declarations]   s    z VGenericEngine._get_declarationsc             C   s   x|   D ]\}}|dd\}}yt| d||f }W n  tk
rX   td| Y nX y||| W q
 tk
r } zt||  W d d }~X Y q
X q
W d S )N r   z_generate_gen_%s_%sznot implemented in verify(): %r)rH   splitgetattrAttributeErrorr   	Exceptionr   attach_exception_info)r   	step_namenamerB   kindrealnamemethoder   r   r   r)   c   s    zVGenericEngine._generatec       
      K   s   x||   D ]p\}}|dd\}}t| d||f }y||||f| W q
 tk
rx }	 zt|	|  W d d }	~	X Y q
X q
W d S )NrI   r   z
_%s_gen_%s)rH   rJ   rK   rM   r   rN   )
r   r6   rO   r   rP   rB   rQ   rR   rS   rT   r   r   r   r;   r   s    zVGenericEngine._loadc             C   s   d S )Nr   )r   rB   rP   r   r   r   _generate_nothing|   s    z VGenericEngine._generate_nothingc             K   s   d S )Nr   )r   rB   rP   r6   r   r   r   r   _loaded_noop   s    zVGenericEngine._loaded_noopc                s  t |tjst|jr(| d|| d S | j}t|j}g }x<t	|jD ].\}}d}t |tj
rdd}|d||f  qHW d|   fddt|j|D }	|j}
t |
tj
r|	d|
d	  tj}
d
|	pd}	d| }| j| |jr|jd }nd}d|||	f }d|  ||
|  |d t |jtj
rDd}nt |jtjsZd}nd}|d||d
|f  |d |  d S )NFr   *z%sx%dzargument of %sc                s    g | ]\}}| d |  qS )z %s)
get_c_name)r@   typearg)contextr   r   rC      s   z>VGenericEngine._generate_gen_function_decl.<locals>.<listcomp>r   z *rz, voidz
_cffi_f_%srI   z	 %s%s(%s)zresult of %s{z*r = zreturn z  %s%s(%s);})
isinstancer   FunctionPtrTypeAssertionErrorellipsis_generate_gen_constr"   lenargs	enumerateStructOrUnionappendzipresultinsertrX   	void_typer   r	   abiZVoidType)r   rB   rP   r,   ZnumargsargnamesirY   ZindirectionZarglistZtpresultwrappernamerm   ZfuncdeclZresult_coder   )r[   r   _generate_gen_function_decl   sJ    

z*VGenericEngine._generate_gen_function_declc             C   sr  t |tjst|jr*| d|||}n(g }|}tdd |jD sVt |jtj	rg }xDt
|jD ]6\}	}
t |
tj	rt|
}
||	|
f ||
 qfW |j}t |tj	r|jd krtd| f t|}|d| |dd|f tj}tt|||j}| j|}d| }|||}x"|D ]\}	}
| ||	|
|}q4W t||| t|j| d S )NFc             s   s   | ]}t |tjV  qd S )N)r_   r   rg   )r@   typr   r   r   	<genexpr>   s    z6VGenericEngine._loaded_gen_function.<locals>.<genexpr>z*'%s' is used as result type, but is opaquer   rj   z
_cffi_f_%s)r_   r   r`   ra   rb   _load_constantanyre   rj   rg   rf   PointerTyperh   Zfldtypes	TypeError_get_c_namerk   rl   tupler   _get_cached_btypeload_function_make_struct_wrappersetattrrY   r0   )r   rB   rP   r6   r7   ZnewfunctionZindirectionsbase_tpZindirect_argsro   rr   Zindirect_resultBFuncrp   r   r   r   _loaded_gen_function   s@    





z#VGenericEngine._loaded_gen_functionc                sP   | j j| j | dkr4| j  fdd}n fdd}||_|S )Nrj   c                 s      }|f|   |d S )Nr   )new)re   res)BTyper   oldfuncr   r   newfunc   s    
z4VGenericEngine._make_struct_wrapper.<locals>.newfuncc                 s6   | d    |  f | d d   } |  S )Nr   )Znewp)re   )r   r?   ro   r   r   r   r      s    .)r   r8   rz   Z_cffi_base_type)r   r   ro   rB   r~   r   r   )r   r?   r   ro   r   r   r|      s    z#VGenericEngine._make_struct_wrapperc             C   s    ||j kst| |d| d S )Nstruct)rP   ra   _generate_struct_or_union_decl)r   rB   rP   r   r   r   _generate_gen_struct_decl   s    z(VGenericEngine._generate_gen_struct_declc             C   s   |  |d|| d S )Nr   )_loading_struct_or_union)r   rB   rP   r6   r   r   r   _loading_gen_struct   s    z"VGenericEngine._loading_gen_structc             K   s   |  | d S )N)_loaded_struct_or_union)r   rB   rP   r6   r   r   r   r   _loaded_gen_struct   s    z!VGenericEngine._loaded_gen_structc             C   s    ||j kst| |d| d S )Nunion)rP   ra   r   )r   rB   rP   r   r   r   _generate_gen_union_decl   s    z'VGenericEngine._generate_gen_union_declc             C   s   |  |d|| d S )Nr   )r   )r   rB   rP   r6   r   r   r   _loading_gen_union   s    z!VGenericEngine._loading_gen_unionc             K   s   |  | d S )N)r   )r   rB   rP   r6   r   r   r   r   _loaded_gen_union  s    z VGenericEngine._loaded_gen_unionc             C   s  |j d krd S d||f }d||f }d||f  }| j}|d||f  |d |d |d x| D ]\}}	}
}t|	tjr|	 s|
dkr|d	|  qny$|d
|	jdd| |d|f  W qn t	k
r } z|dt
|  W d d }~X Y qnX qnW |d | j| |d|f  |d |d|  |d |d|  |d xt| D ]h\}}	}
}|
dkrvq\|d||f  t|	tjr|	jd kr|d|	   n|d||f  q\W |d |d |d |d |d|f  |d |  d S )Nz_cffi_check_%s_%sz_cffi_layout_%s_%sz%s %szstatic void %s(%s *p)r]   z8  /* only to generate compile-time warnings or errors */z
  (void)p;r   z  (void)((p->%s) << 1);z  { %s = &p->%s; (void)tmp; }z*tmpzfield %r)Zqualsz
  /* %s */r^   zintptr_t %s(intptr_t i)z,  struct _cffi_aligncheck { char x; %s y; };z  static intptr_t nums[] = {z    sizeof(%s),z)    offsetof(struct _cffi_aligncheck, y),z    offsetof(%s, %s),z    0,  /* %s */z    sizeof(((%s *)0)->%s),z    -1z  };z  return nums[i];z3  /* the next line is not executed, but compiled */z  %s(0);)fldnamesstripr"   
enumfieldsr_   r   PrimitiveTypeis_integer_typerX   r   strr	   rh   	ArrayTypelengthrx   )r   rB   r-   rP   Zcheckfuncnamelayoutfuncnamecnamer,   fnameftypefbitsizefqualrT   r   r   r   r     sT    
&
z-VGenericEngine._generate_struct_or_union_declc             C   s  |j d krd S d||f }| jdd }|||}g }d}	x(||	}
|
dk rRP ||
 |	d7 }	q@W t|tjr|jr|d }|d }|dd d }|dd d }|	  t
|t
|  krt
|j ksn t||||f|_nd||f  }||f| j|< d S )Nz_cffi_layout_%s_%szintptr_t(*)(intptr_t)r   r      r%   z%s %s)r   r   _typeof_lockedr{   rh   r_   r   rg   partialZforce_flattenrd   ra   Zfixedlayoutr   r
   )r   rB   r-   rP   r6   r   r   functionlayoutnumxZ	totalsizeZtotalalignmentZfieldofsZ	fieldsizer   r   r   r   r   7  s.    
 
(z'VGenericEngine._loading_struct_or_unionc             C   s  |j d krd S | j| || jkrdd }| j}||}| j|\}}||d ||d ||d ||d d}x| D ]v\}}	}
}|
dkrq||| |||d|f  ||d  dkr||	}|||d  ||d	|f  |d7 }qW |t	|kst
d S )
Nc             S   s   | |krt d||| f d S )Nz'%s (we have %d, but C compiler says %d))r   )Z	realvalueZexpectedvaluemsgr   r   r   checkZ  s    z5VGenericEngine._loaded_struct_or_union.<locals>.checkr   zwrong total sizer   zwrong total alignmentr   zwrong offset for field %rzwrong size for field %r)r   r   rz   r
   popsizeofZalignofr   Zoffsetofrd   ra   )r   rB   r   r   ZBStructr   r   ro   r   r   r   r   ZBFieldr   r   r   r   S  s,    


z&VGenericEngine._loaded_struct_or_unionc             C   s.   t |tjr| ||d n| |d| d S )Nr   )r_   r   EnumType_generate_gen_enum_declr   )r   rB   rP   r   r   r   _generate_gen_anonymous_declu  s    z+VGenericEngine._generate_gen_anonymous_declc             C   s2   t |tjr| |||d n| |d|| d S )Nr   )r_   r   r   _loading_gen_enumr   )r   rB   rP   r6   r   r   r   _loading_gen_anonymous{  s    z%VGenericEngine._loading_gen_anonymousc             K   s.   t |tjr | j|||f| n
| | d S )N)r_   r   r   _loaded_gen_enumr   )r   rB   rP   r6   r   r   r   r   _loaded_gen_anonymous  s    z$VGenericEngine._loaded_gen_anonymousNconstc       
      C   sF  | j }d||f }| j| |d k	rl|s.t|dks:t|d|  |d | || |d |d n|r|dks|t|d|  |d |d|f  |d	|f  |d n|d k	st|d kst|d
krd}nd}d}	|dkrt|tjrd}	d}||d|	|f | |d |d||f  |d |  d S )Nz_cffi_%s_%sr   zint %s(char *out_error)r]   z  return 0;r^   zint %s(long long *out_value)z  *out_value = (long long)(%s);z  return (%s) <= 0;var&r   zconst *z %s%s(void)z  return (%s%s);)	r"   r	   rh   ra   _check_int_constant_valuer_   r   rg   rX   )
r   is_intrP   rB   categorycheck_valuer,   funcnameZ	ampersandextrar   r   r   rc     s@    

z"VGenericEngine._generate_gen_constc             C   s&   t |tjo| }| ||| d S )N)r_   r   r   r   rc   )r   rB   rP   r   r   r   r   _generate_gen_constant_decl  s    z*VGenericEngine._generate_gen_constant_declc             C   s  d| }|d k	r*|st | || |}n|r| jdd }| jdd }	||	|}
| j|}|
|}t|d }|dk r|s| jdd }|dd| j| > 7 }nd|d kst d}t|t	j
rd	| }| j|||d }	||	|}
|
 }t|t	j
r|d }|S )
Nz_cffi_const_%sz
long long*r   zint(*)(long long*)z	long longr      z	(*)(void)rW   )ra   _load_known_int_constantr   r   r{   r   intr   r_   r   rg   rX   )r   r   rB   rP   r6   r   r   valuer   r   r   pnegativeZ	BLongLongZfntypeextrar   r   r   rt     s2    zVGenericEngine._load_constantc             C   sD   t |tjo| }| ||||}t||| t|j| d S )N)	r_   r   r   r   rt   r}   rY   r0   rh   )r   rB   rP   r6   r7   r   r   r   r   r   _loaded_gen_constant  s    z#VGenericEngine._loaded_gen_constantc             C   s   | j }|dkr"|d|||f  n|d|||f  |d |d|  |d|  |d |d|  |d	 |d
|d d |f  |d |d d S )Nr   z&  if ((%s) > 0 || (long)(%s) != %dL) {z1  if ((%s) <= 0 || (unsigned long)(%s) != %dUL) {z    char buf[64];z    if ((%s) <= 0)z)        sprintf(buf, "%%ld", (long)(%s));z    elsez2        sprintf(buf, "%%lu", (unsigned long)(%s));z:    sprintf(out_error, "%s has the real value %s, not %s",z            "%s", buf, "%d");d   z    return -1;z  })r"   )r   rP   r   r,   r   r   r   r     s     z(VGenericEngine._check_int_constant_valuec             C   sr   | j dd }| j dd }|||}| j |d}||dk rn| j |}tjdkrft|d}t|d S )Nzchar[]r   zint(*)(char*)   )r%   zutf-8)	r   r   r{   r   stringr   r+   r   r   )r   r6   r   r   r   r   r   errorr   r   r   r     s    

z'VGenericEngine._load_known_int_constantc             C   s   | dd}d||f S )N$Z___D_z_cffi_e_%s_%s)replace)r   r-   rP   r   r   r   _enum_funcname  s    zVGenericEngine._enum_funcnameenumc             C   s   |j r&x|jD ]}| d| qW d S | ||}| j| | j}|d|  |d x&t|j|jD ]\}}| 	|| qhW |d |d |  d S )NTzint %s(char *out_error)r]   z  return 0;r^   )
r   enumeratorsrc   r   r	   rh   r"   ri   
enumvaluesr   )r   rB   rP   r-   
enumeratorr   r,   	enumvaluer   r   r   r     s    z&VGenericEngine._generate_gen_enum_declc                sL   j r0 fddjD }t|_d_n||} | d S )Nc                s   g | ]} d | qS )T)rt   )r@   r   )r6   r   rB   r   r   rC     s   z4VGenericEngine._loading_gen_enum.<locals>.<listcomp>T)r   r   ry   r   Zpartial_resolvedr   r   )r   rB   rP   r6   r-   r   r   r   )r6   r   rB   r   r     s    

z VGenericEngine._loading_gen_enumc             C   s<   x6t |j|jD ]$\}}t||| t|j| qW d S )N)ri   r   r   r}   rY   r0   rh   )r   rB   rP   r6   r7   r   r   r   r   r   r     s    zVGenericEngine._loaded_gen_enumc             C   s&   |dkrd }n|}| j d||d d S )Nz...T)r   )rc   )r   rB   rP   r   r   r   r   _generate_gen_macro_decl   s    z'VGenericEngine._generate_gen_macro_declc             C   sF   |dkrd }n|}| j d||||d}t||| t|j| d S )Nz...T)r   )rt   r}   rY   r0   rh   )r   rB   rP   r6   r7   r   r   r   r   r   _loaded_gen_macro)  s    z VGenericEngine._loaded_gen_macroc             C   s   t |tjrv| rZ| j}d|f }| j| |d|  |d |d|f  |d t|j}| 	d|| nt|}| j	d||dd d S )	Nz_cffi_sizeof_%szsize_t %s(void)r]   z  return sizeof(%s);r^   Fr   )r   )
r_   r   r   length_is_unknownr"   r	   rh   rv   itemrc   )r   rB   rP   r,   r   tp_ptrr   r   r   _generate_gen_variable_decl6  s    

z*VGenericEngine._generate_gen_variable_declc                s\  t |tjr| rd|f }| jdd }|||}| }| j|j}	t	|| j
|	\}
}|dkr~td||jf ||
}t|j}| d|||}|jd k	r| j|}| j||}t||| t|j| d S d| }| j|d|d }|||}|   fdd	} fd
d}tt||t|| t|j| d S )Nz_cffi_sizeof_%szsize_t(*)(void)r   z/bad size: %r does not seem to be an array of %sFz_cffi_var_%sz
*(*)(void)c                s    d S )Nr   r   )r7   )ptrr   r   getterg  s    z3VGenericEngine._loaded_gen_variable.<locals>.getterc                s   | d< d S )Nr   r   )r7   r   )r   r   r   setteri  s    z3VGenericEngine._loaded_gen_variable.<locals>.setter)r_   r   r   r   r   r   r{   rz   r   divmodr   r   Zresolve_lengthrv   rt   r   castr}   rY   r0   rh   rX   property)r   rB   rP   r6   r7   r   r   r   sizeZ	BItemTyper   restr   r   ZBArrayr   r   r   )r   r   _loaded_gen_variableH  s:    


z#VGenericEngine._loaded_gen_variable)r   )r   )Nr   N)N)r   )r   )5r3   r4   r5   Z
_class_keyZ_gen_python_moduler   r   r   r   r"   r/   r:   rH   r)   r;   rU   rV   Z_generate_gen_typedef_declZ_loading_gen_typedefZ_loaded_gen_typedefrq   Z_loading_gen_functionr   r|   r   r   r   r   r   r   r   r   r   r   r   r   rc   r   Z_loading_gen_constantrt   r   r   r   r   r   r   r   r   Z_loading_gen_macror   r   Z_loading_gen_variabler   r   r   r   r   r      sf   



-$1"	 
$
	


r   ad  
#include <stdio.h>
#include <stddef.h>
#include <stdarg.h>
#include <errno.h>
#include <sys/types.h>   /* XXX for ssize_t on some platforms */

/* this block of #ifs should be kept exactly identical between
   c/_cffi_backend.c, cffi/vengine_cpy.py, cffi/vengine_gen.py
   and cffi/_cffi_include.h */
#if defined(_MSC_VER)
# include <malloc.h>   /* for alloca() */
# if _MSC_VER < 1600   /* MSVC < 2010 */
   typedef __int8 int8_t;
   typedef __int16 int16_t;
   typedef __int32 int32_t;
   typedef __int64 int64_t;
   typedef unsigned __int8 uint8_t;
   typedef unsigned __int16 uint16_t;
   typedef unsigned __int32 uint32_t;
   typedef unsigned __int64 uint64_t;
   typedef __int8 int_least8_t;
   typedef __int16 int_least16_t;
   typedef __int32 int_least32_t;
   typedef __int64 int_least64_t;
   typedef unsigned __int8 uint_least8_t;
   typedef unsigned __int16 uint_least16_t;
   typedef unsigned __int32 uint_least32_t;
   typedef unsigned __int64 uint_least64_t;
   typedef __int8 int_fast8_t;
   typedef __int16 int_fast16_t;
   typedef __int32 int_fast32_t;
   typedef __int64 int_fast64_t;
   typedef unsigned __int8 uint_fast8_t;
   typedef unsigned __int16 uint_fast16_t;
   typedef unsigned __int32 uint_fast32_t;
   typedef unsigned __int64 uint_fast64_t;
   typedef __int64 intmax_t;
   typedef unsigned __int64 uintmax_t;
# else
#  include <stdint.h>
# endif
# if _MSC_VER < 1800   /* MSVC < 2013 */
#  ifndef __cplusplus
    typedef unsigned char _Bool;
#  endif
# endif
#else
# include <stdint.h>
# if (defined (__SVR4) && defined (__sun)) || defined(_AIX) || defined(__hpux)
#  include <alloca.h>
# endif
#endif
)
r   r   r<   r   r   r   r   objectr   r'   r   r   r   r   <module>   s        