B
    ]4h                 @   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   /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   Z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
)Z_fwrite)r   Z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 )NZdeclZwin32)   ZPyInit_Zinitzvoid %s%s(void) { }
)	r   cffimod_headerr   Z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_ffir)   r+   r   )r*   moduler   r   r   r*   P   s   r*   r   Zloaded)library)r   _backendr   r   r   curdirr   Zmodulefilenameload_library_loadtypes
ModuleType)r   flagsbackendr   r0   r   )r*   r/   r   r   r3   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   Z_parserZ_declarationsitemssort)r   Z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)r?   splitgetattrAttributeErrorr   	Exceptionr   attach_exception_info)r   	step_namenamer;   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 )Nr@   r   z
_%s_gen_%s)r?   rA   rB   rD   r   rE   )
r   r/   rF   r   rG   r;   rH   rI   rJ   rK   r   r   r   r4   r   s    zVGenericEngine._loadc             C   s   d S )Nr   )r   r;   rG   r   r   r   _generate_nothing|   s    z VGenericEngine._generate_nothingc             K   s   d S )Nr   )r   r;   rG   r/   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)r9   typearg)contextr   r   r<      s   z>VGenericEngine._generate_gen_function_decl.<locals>.<listcomp>r   z *rz, Zvoidz
_cffi_f_%sr@   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insertrO   	void_typer   r	   abiZVoidType)r   r;   rG   r%   ZnumargsZargnamesirP   ZindirectionZarglistZtpresultwrappernamerc   ZfuncdeclZresult_coder   )rR   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)rU   r   r]   )r9   typr   r   r   	<genexpr>   s    z6VGenericEngine._loaded_gen_function.<locals>.<genexpr>z*'%s' is used as result type, but is opaquer   r`   z
_cffi_f_%s)rU   r   rV   rW   rX   _load_constantanyr[   r`   r]   r\   PointerTyper^   Zfldtypes	TypeError_get_c_namera   rb   tupler   _get_cached_btypeload_function_make_struct_wrappersetattrrP   r)   )r   r;   rG   r/   r0   ZnewfunctionZindirectionsbase_tpZindirect_argsrd   rg   Zindirect_resultBFuncre   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 )Nr`   c                 s      }|f|   |d S )Nr   )new)r[   Zres)BTyper   oldfuncr   r   newfunc   s    
z4VGenericEngine._make_struct_wrapper.<locals>.newfuncc                 s6   | d    |  f | d d   } |  S )Nr   )Znewp)r[   )rw   r8   rd   rx   r   r   ry      s    .)r   r1   ro   Z_cffi_base_type)r   rx   rd   r;   rs   ry   r   )rw   r8   r   rd   rx   r   rq      s    z#VGenericEngine._make_struct_wrapperc             C   s    ||j kst| |d| d S )Nstruct)rG   rW   _generate_struct_or_union_decl)r   r;   rG   r   r   r   _generate_gen_struct_decl   s    z(VGenericEngine._generate_gen_struct_declc             C   s   |  |d|| d S )Nrz   )_loading_struct_or_union)r   r;   rG   r/   r   r   r   _loading_gen_struct   s    z"VGenericEngine._loading_gen_structc             K   s   |  | d S )N)_loaded_struct_or_union)r   r;   rG   r/   r   r   r   r   _loaded_gen_struct   s    z!VGenericEngine._loaded_gen_structc             C   s    ||j kst| |d| d S )Nunion)rG   rW   r{   )r   r;   rG   r   r   r   _generate_gen_union_decl   s    z'VGenericEngine._generate_gen_union_declc             C   s   |  |d|| d S )Nr   )r}   )r   r;   rG   r/   r   r   r   _loading_gen_union   s    z!VGenericEngine._loading_gen_unionc             K   s   |  | d S )N)r   )r   r;   rG   r/   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)rS   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 */rT   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   
enumfieldsrU   r   PrimitiveTypeis_integer_typerO   r   strr	   r^   	ArrayTypelengthrm   )r   r;   r&   rG   Zcheckfuncnamelayoutfuncnamecnamer%   fnameftypefbitsizefqualrK   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_lockedrp   r^   rU   r   r]   partialZforce_flattenrZ   rW   Zfixedlayoutr   r
   )r   r;   r&   rG   r/   r   rt   functionlayoutZ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   ro   r
   popsizeofZalignofr   ZoffsetofrZ   rW   )r   r;   r   r   ZBStructr   r   rd   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   )rU   r   EnumType_generate_gen_enum_declr{   )r   r;   rG   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   )rU   r   r   _loading_gen_enumr}   )r   r;   rG   r/   r   r   r   _loading_gen_anonymous{  s    z%VGenericEngine._loading_gen_anonymousc             K   s.   t |tjr | j|||f| n
| | d S )N)rU   r   r   _loaded_gen_enumr   )r   r;   rG   r/   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)rS   z  return 0;rT   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	   r^   rW   _check_int_constant_valuerU   r   r]   rO   )
r   is_intrG   r;   categorycheck_valuer%   funcnameZ	ampersandZextrar   r   r   rY     s@    

z"VGenericEngine._generate_gen_constc             C   s&   t |tjo| }| ||| d S )N)rU   r   r   r   rY   )r   r;   rG   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)rN   )rW   _load_known_int_constantr   r   rp   rv   intr   rU   r   r]   rO   )r   r   r;   rG   r/   r   r   valuerw   rt   r   pnegativeZ	BLongLongZfntypeextrar   r   r   ri     s2    zVGenericEngine._load_constantc             C   sD   t |tjo| }| ||||}t||| t|j| d S )N)	rU   r   r   r   ri   rr   rP   r)   r^   )r   r;   rG   r/   r0   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   rG   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   rp   rv   stringr   r$   r   r   )r   r/   r   rw   rt   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&   rG   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)rS   z  return 0;rT   )
r   enumeratorsrY   r   r	   r^   r   r_   
enumvaluesr   )r   r;   rG   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)ri   )r9   r   )r/   r   r;   r   r   r<     s   z4VGenericEngine._loading_gen_enum.<locals>.<listcomp>T)r   r   rn   r   Zpartial_resolvedr   r   )r   r;   rG   r/   r&   r   r   r   )r/   r   r;   r   r     s    

z VGenericEngine._loading_gen_enumc             C   s<   x6t |j|jD ]$\}}t||| t|j| qW d S )N)r_   r   r   rr   rP   r)   r^   )r   r;   rG   r/   r0   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   )rY   )r   r;   rG   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   )ri   rr   rP   r)   r^   )r   r;   rG   r/   r0   r   r   r   r   r   _loaded_gen_macro)  s    z VGenericEngine._loaded_gen_macroc             C   s   t |tjrx|jdkr\| j}d|f }| j| |d|  |d |d|f  |d t|j}| 	d|| nt|}| j	d||dd	 d S )
Nz...z_cffi_sizeof_%szsize_t %s(void)rS   z  return sizeof(%s);rT   Fr   )r   )
rU   r   r   r   r   r	   r^   rk   itemrY   )r   r;   rG   r%   r   tp_ptrr   r   r   _generate_gen_variable_decl6  s    


z*VGenericEngine._generate_gen_variable_declc                s^  t |tjr|jdkrd|f }| jdd }|||}| }| j|j}	t	|| j
|	\}
}|dkrt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...z_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   )r0   )ptrr   r   getterg  s    z3VGenericEngine._loaded_gen_variable.<locals>.getterc                s   | d< d S )Nr   r   )r0   r   )r   r   r   setteri  s    z3VGenericEngine._loaded_gen_variable.<locals>.setter)rU   r   r   r   r   r   rp   ro   r   divmodr   r   Zresolve_lengthrk   ri   castrr   rP   r)   r^   rO   property)r   r;   rG   r/   r0   r   rt   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   )5r,   r-   r.   Z
_class_keyZ_gen_python_moduler   r   r   r   r   r(   r3   r?   r"   r4   rL   rM   Z_generate_gen_typedef_declZ_loading_gen_typedefZ_loaded_gen_typedefrf   Z_loading_gen_functionru   rq   r|   r~   r   r   r   r   r{   r}   r   r   r   r   rY   r   Z_loading_gen_constantri   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   r5   r   r   r   r   objectr   r!   r   r   r   r   <module>   s        