U
    ڲg;i                     @   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   4/tmp/pip-unpacked-wheel-lbi5ej3i/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   |D ]J}|| }|d krt j}|D ]*}tj||}tj|r"|    S q"qd S r   )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 r   r   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 r   )	_cffi_dirr   )
FFILibraryr   r   __dir__T   s    z7VGenericEngine.load_library.<locals>.FFILibrary.__dir__N)__name__
__module____qualname__Z_cffi_generic_moduler
   Z	_cffi_ffir4   r6   r   r5   moduler   r   r   r5   P   s   r5   r    Zloadedlibrary)r
   _backendr   r   r   curdirr	   Zmodulefilenameload_library_loadtypes
ModuleType)r   flagsbackendr   r=   r   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   |   D ]\}}|dd\}}zt| d||f }W n  tk
rV   td| Y nX z||| W q tk
r } zt||  W 5 d }~X Y qX qd S )N r   z_generate_gen_%s_%sznot implemented in verify(): %r)rN   splitgetattrAttributeErrorr   	Exceptionr   attach_exception_info)r   	step_namenamerH   kindrealnamemethoder   r   r   r-   c   s    
zVGenericEngine._generatec           
      K   s~   |   D ]p\}}|dd\}}t| d||f }z||||f| W q tk
rv }	 zt|	|  W 5 d }	~	X Y qX qd S )NrO   r   z
_%s_gen_%s)rN   rP   rQ   rS   r   rT   )
r   r;   rU   r   rV   rH   rW   rX   rY   rZ   r   r   r   rA   r   s    zVGenericEngine._loadc                 C   s   d S r   r   r   rH   rV   r   r   r   _generate_nothing|   s    z VGenericEngine._generate_nothingc                 K   s   d S r   r   r   rH   rV   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 }t	|jD ].\}}d}t |tj
rbd}|d||f  qFd|   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
r@d}nt |jtjsVd}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)rF   typeargcontextr   r   rI      s   z>VGenericEngine._generate_gen_function_decl.<locals>.<listcomp>r   z *rz, void
_cffi_f_%srO   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insertr`   	void_typer   r   abiZVoidType)r   rH   rV   r0   Znumargsargnamesira   ZindirectionZarglistZtpresultwrappernamerw   ZfuncdeclZresult_coder   rc   r   _generate_gen_function_decl   sL    


z*VGenericEngine._generate_gen_function_declc                 C   sj  t |tjst|jr*| d|||}n g }|}tdd |jD sVt |jtj	r
g }t
|jD ]6\}	}
t |
tj	rt|
}
||	|
f ||
 qd|j}t |tj	r|jd krtd| f t|}|d| |dd|f tj}tt|||j}| j|}d| }|||}|D ]\}	}
| ||	|
|}q.t||| t|j| d S )NFc                 s   s   | ]}t |tjV  qd S r   )ri   r   rq   )rF   typr   r   r   	<genexpr>   s     z6VGenericEngine._loaded_gen_function.<locals>.<genexpr>z*'%s' is used as result type, but is opaquer   rt   rf   )ri   r   rj   rk   rl   _load_constantanyro   rt   rq   rp   PointerTyperr   Zfldtypes	TypeError_get_c_nameru   rv   tupler
   _get_cached_btypeload_function_make_struct_wrappersetattrra   r4   )r   rH   rV   r;   r=   ZnewfunctionZindirectionsbase_tpZindirect_argsry   r|   Zindirect_resultBFuncrz   r   r   r   _loaded_gen_function   sL    



 
z#VGenericEngine._loaded_gen_functionc                    sP   | j j| j | dkr4| j  fdd}n fdd}||_|S )Nrt   c                     s      }|f|   |d S Nr   )new)ro   res)BTyper
   oldfuncr   r   newfunc   s    
z4VGenericEngine._make_struct_wrapper.<locals>.newfuncc                     s6   | d    |  f | d d   } |  S )Nr   )Znewp)ro   )r   rE   ry   r   r   r   r      s    .)r
   r>   r   Z_cffi_base_type)r   r   ry   rH   r   r   r   )r   rE   r
   ry   r   r   r      s    z#VGenericEngine._make_struct_wrapperc                 C   s    ||j kst| |d| d S NstructrV   rk   _generate_struct_or_union_declr[   r   r   r   _generate_gen_struct_decl   s    z(VGenericEngine._generate_gen_struct_declc                 C   s   |  |d|| d S r   _loading_struct_or_unionr   rH   rV   r;   r   r   r   _loading_gen_struct   s    z"VGenericEngine._loading_gen_structc                 K   s   |  | d S r   _loaded_struct_or_unionr]   r   r   r   _loaded_gen_struct   s    z!VGenericEngine._loaded_gen_structc                 C   s    ||j kst| |d| d S Nunionr   r[   r   r   r   _generate_gen_union_decl   s    z'VGenericEngine._generate_gen_union_declc                 C   s   |  |d|| d S r   r   r   r   r   r   _loading_gen_union   s    z!VGenericEngine._loading_gen_unionc                 K   s   |  | d S r   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 | D ]\}}	}
}t|	tjr|	 s|
dkr|d	|  qlz$|d
|	jdd| |d|f  W ql t	k
r } z|dt
|  W 5 d }~X Y qlX ql|d | j| |d|f  |d |d|  |d |d|  |d | D ]h\}}	}
}|
dkrpqV|d||f  t|	tjr|	jd kr|d|	   n|d||f  qV|d |d |d |d |d|f  |d |  d S )Nz_cffi_check_%s_%s_cffi_layout_%s_%s%s %szstatic void %s(%s *p)rg   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 */rh   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%   
enumfieldsri   r   PrimitiveTypeis_integer_typer`   r   strr   rr   	ArrayTypelengthr   )r   rH   r1   rV   Zcheckfuncnamelayoutfuncnamecnamer0   fnameftypefbitsizefqualrZ   r   r   r   r     s\    
$
z-VGenericEngine._generate_struct_or_union_declc                 C   s  |j d krd S d||f }| jdd }|||}g }d}	||	}
|
dk rPqd||
 |	d7 }	q>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 )Nr   zintptr_t(*)(intptr_t)r   r      r)   r   )r   r
   _typeof_lockedr   rr   ri   r   rq   partialZforce_flattenrn   rk   Zfixedlayoutr   r   )r   rH   r1   rV   r;   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}| D ]v\}}	}
}|
dkrq||| |||d|f  ||d  dkr||	}|||d  ||d	|f  |d7 }q|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
   r   r   popsizeofZalignofr   Zoffsetofrn   rk   )r   rH   r   r
   ZBStructr   r   ry   r   r   r   r   ZBFieldr   r   r   r   S  s0    



z&VGenericEngine._loaded_struct_or_unionc                 C   s.   t |tjr| ||d n| |d| d S Nr    )ri   r   EnumType_generate_gen_enum_declr   r[   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 r   )ri   r   r   _loading_gen_enumr   r   r   r   r   _loading_gen_anonymous{  s    z%VGenericEngine._loading_gen_anonymousc                 K   s.   t |tjr | j|||f| n
| | d S r   )ri   r   r   _loaded_gen_enumr   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   int %s(char *out_error)rg     return 0;rh   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   rr   rk   _check_int_constant_valueri   r   rq   r`   )
r   is_intrV   rH   categorycheck_valuer0   funcnameZ	ampersandextrar   r   r   rm     s@    

z"VGenericEngine._generate_gen_constc                 C   s&   t |tjo| }| ||| d S r   )ri   r   r   r   rm   )r   rH   rV   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)r_   )rk   _load_known_int_constantr
   r   r   r   intr   ri   r   rq   r`   )r   r   rH   rV   r;   r   r   valuer   r   r   pnegativeZ	BLongLongZfntypeextrar   r   r   r~     s2    zVGenericEngine._load_constantc                 C   sD   t |tjo| }| ||||}t||| t|j| d S r   )	ri   r   r   r   r~   r   ra   r4   rr   )r   rH   rV   r;   r=   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   rV   r   r0   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   r;   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   r1   rV   r   r   r   _enum_funcname  s    zVGenericEngine._enum_funcnameenumc                 C   s   |j r"|jD ]}| d| qd S | ||}| j| | j}|d|  |d t|j|jD ]\}}| 	|| qb|d |d |  d S )NTr   rg   r   rh   )
r   enumeratorsrm   r   r   rr   r%   rs   
enumvaluesr   )r   rH   rV   r1   
enumeratorr   r0   	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)r~   )rF   r   r;   r   rH   r   r   rI     s   z4VGenericEngine._loading_gen_enum.<locals>.<listcomp>T)r   r   r   r   Zpartial_resolvedr   r   )r   rH   rV   r;   r1   r   r   r   r   r   r     s    
z VGenericEngine._loading_gen_enumc                 C   s8   t |j|jD ]$\}}t||| t|j| qd S r   )rs   r   r   r   ra   r4   rr   )r   rH   rV   r;   r=   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   )rm   )r   rH   rV   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 r   )r~   r   ra   r4   rr   )r   rH   rV   r;   r=   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 )	N_cffi_sizeof_%szsize_t %s(void)rg   z  return sizeof(%s);rh   Fr   )r   )
ri   r   r   length_is_unknownr%   r   rr   r   itemrm   )r   rH   rV   r0   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 )Nr   z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 r   r   r<   ptrr   r   getterg  s    z3VGenericEngine._loaded_gen_variable.<locals>.getterc                    s   | d< d S r   r   )r=   r   r   r   r   setteri  s    z3VGenericEngine._loaded_gen_variable.<locals>.setter)ri   r   r   r   r
   r   r   r   r   divmodr   r   Zresolve_lengthr   r~   r   castr   ra   r4   rr   r`   property)r   rH   rV   r;   r=   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   )5r7   r8   r9   Z
_class_keyZ_gen_python_moduler   r   r   r   r%   r3   r@   rN   r-   rA   r\   r^   Z_generate_gen_typedef_declZ_loading_gen_typedefZ_loaded_gen_typedefr{   Z_loading_gen_functionr   r   r   r   r   r   r   r   r   r   r   r   r   r   rm   r   Z_loading_gen_constantr~   r   r   r   r   r   r   r   r   Z_loading_gen_macror   r   Z_loading_gen_variabler   r   r   r   r   r      sh   



-$1"	  
%
	


r   ac  
#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
# define _cffi_float_complex_t   _Fcomplex    /* include <complex.h> for it */
# define _cffi_double_complex_t  _Dcomplex    /* include <complex.h> for it */
#else
# include <stdint.h>
# if (defined (__SVR4) && defined (__sun)) || defined(_AIX) || defined(__hpux)
#  include <alloca.h>
# endif
# define _cffi_float_complex_t   float _Complex
# define _cffi_double_complex_t  double _Complex
#endif
)
r   r   rB   r    r   r   r   objectr   r+   r   r   r   r   <module>   s       g