a
   0hjs                   @   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/usr/lib/python3.9/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   |D ]J}|| }|d u rt j}|D ]*}tj||}tj|r"|    S q"qd S r   )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 r   r   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declwin32   ZPyInit_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 r   )	_cffi_dirr   )
FFILibraryr   r   __dir__T   s    z7VGenericEngine.load_library.<locals>.FFILibrary.__dir__N)__name__
__module____qualname__Z_cffi_generic_moduler
   Z	_cffi_ffir/   r1   r   r0   moduler   r   r   r0   P   s   r0   r   Zloadedlibrary)r
   _backendr   r   r   curdirr	   Zmodulefilenameload_library_loadtypes
ModuleType)r   flagsbackendr   r8   r   r5   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>^       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   |   D ]\}}|dd\}}zt| d||f }W n tyT   td| Y n0 z||| W q ty } zt||  W Y d }~qd }~0 0 q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namerC   kindrealnamemethoder   r   r   r(   c   s    
zVGenericEngine._generatec           
      K   s   |   D ]v\}}|dd\}}t| d||f }z||||fi | W q ty| }	 zt|	|  W Y d }	~	qd }	~	0 0 qd S )NrI   r   z
_%s_gen_%s)rH   rJ   rK   rM   r   rN   )
r   r6   rO   r   rP   rC   rQ   rR   rS   rT   r   r   r   r<   r   s    zVGenericEngine._loadc                 C   s   d S r   r   r   rC   rP   r   r   r   _generate_nothing|   s    z VGenericEngine._generate_nothingc                 K   s   d S r   r   r   rC   rP   r6   r   r   r   r   _loaded_noop   s    zVGenericEngine._loaded_noopc                    s  t |tjsJ |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)rA   typeargcontextr   r   rD      s   z>VGenericEngine._generate_gen_function_decl.<locals>.<listcomp>r   z *rz, Zvoid
_cffi_f_%srI   z	 %s%s(%s)zresult of %s{z*r = zreturn z  %s%s(%s);})
isinstancer   FunctionPtrTypeellipsis_generate_gen_constr"   lenargs	enumerateStructOrUnionappendzipresultinsertrZ   	void_typer   r   abiZVoidType)r   rC   rP   r+   ZnumargsZargnamesir[   ZindirectionZarglistZtpresultwrappernamero   ZfuncdeclZresult_coder   r]   r   _generate_gen_function_decl   sL    


z*VGenericEngine._generate_gen_function_declc                 C   sj  t |tjsJ |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 u 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   )rb   r   ri   )rA   typr   r   r   	<genexpr>   rE   z6VGenericEngine._loaded_gen_function.<locals>.<genexpr>z*'%s' is used as result type, but is opaquer   rl   r_   )rb   r   rc   rd   _load_constantanyrg   rl   ri   rh   PointerTyperj   Zfldtypes	TypeError_get_c_namerm   rn   tupler
   _get_cached_btypeload_function_make_struct_wrappersetattrr[   r/   )r   rC   rP   r6   r8   ZnewfunctionZindirectionsbase_tpZindirect_argsrp   rs   Zindirect_resultBFuncrq   r   r   r   _loaded_gen_function   sJ    




z#VGenericEngine._loaded_gen_functionc                    sP   | j j| j | dkr4| j  fdd}n fdd}||_|S )Nrl   c                     s"     }|g| R   |d S Nr   )new)rg   res)BTyper
   oldfuncr   r   newfunc   s    
z4VGenericEngine._make_struct_wrapper.<locals>.newfuncc                     s6   | d    |  f | d d   } |  S )Nr   )Znewp)rg   )r   r@   rp   r   r   r   r      s    .)r
   r9   r{   Z_cffi_base_type)r   r   rp   rC   r   r   r   )r   r@   r
   rp   r   r   r}      s    z#VGenericEngine._make_struct_wrapperc                 C   s    ||j ksJ | |d| d S NstructrP   _generate_struct_or_union_declrU   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   rC   rP   r6   r   r   r   _loading_gen_struct   s    z"VGenericEngine._loading_gen_structc                 K   s   |  | d S r   _loaded_struct_or_unionrW   r   r   r   _loaded_gen_struct   s    z!VGenericEngine._loaded_gen_structc                 C   s    ||j ksJ | |d| d S Nunionr   rU   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   rW   r   r   r   _loaded_gen_union  s    z VGenericEngine._loaded_gen_unionc                 C   s  |j d u 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	y } z|dt
|  W Y d }~qld }~0 0 ql|d | j| |d|f  |d |d|  |d |d|  |d | D ]h\}}	}
}|
dkrrqX|d||f  t|	tjr|	jd u r|d|	   n|d||f  qX|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)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 */ra   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"   
enumfieldsrb   r   PrimitiveTypeis_integer_typerZ   r   strr   rj   	ArrayTypelengthry   )r   rC   r,   rP   Zcheckfuncnamelayoutfuncnamecnamer+   fnameftypefbitsizefqualrT   r   r   r   r     s\    
(
z-VGenericEngine._generate_struct_or_union_declc                 C   s  |j d u 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 J ||||f|_nd||f  }||f| j|< d S )Nr   zintptr_t(*)(intptr_t)r   r      r%   r   )r   r
   _typeof_lockedr|   rj   rb   r   ri   partialZforce_flattenrf   Zfixedlayoutr   r   )r   rC   r,   rP   r6   r   r   functionlayoutZnumxZ	totalsizeZtotalalignmentZfieldofsZ	fieldsizer   r   r   r   r   7  s*    



(z'VGenericEngine._loading_struct_or_unionc                 C   s  |j d u rd S | j| || jv 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J 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offsetofrf   )r   rC   r   r
   ZBStructr   r   rp   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   )rb   r   EnumType_generate_gen_enum_declr   rU   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   )rb   r   r   _loading_gen_enumr   r   r   r   r   _loading_gen_anonymous{  s    z%VGenericEngine._loading_gen_anonymousc                 K   s2   t |tjr$| j|||fi | n
| | d S r   )rb   r   r   _loaded_gen_enumr   rW   r   r   r   _loaded_gen_anonymous  s    z$VGenericEngine._loaded_gen_anonymousNconstc           
      C   sF  | j }d||f }| j| |d url|s.J |dks:J |d|  |d | || |d |d n|r|dks|J |d|  |d |d|f  |d	|f  |d n|d usJ |d u sJ |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)r`     return 0;ra   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   rj   _check_int_constant_valuerb   r   ri   rZ   )
r   is_intrP   rC   categorycheck_valuer+   funcnameZ	ampersandZextrar   r   r   re     s@    

z"VGenericEngine._generate_gen_constc                 C   s&   t |tjo| }| ||| d S r   )rb   r   r   r   re   )r   rC   rP   r   r   r   r   _generate_gen_constant_decl  s    z*VGenericEngine._generate_gen_constant_declc                 C   s  d| }|d ur*|sJ |  || |}n|r| jdd }| jdd }	||	|}
| j|}|
|}t|d }|dk r|s| jdd }|dd| j| > 7 }nd|d u sJ 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)rY   )_load_known_int_constantr
   r   r|   r   intr   rb   r   ri   rZ   )r   r   rC   rP   r6   r   r   valuer   r   r   pnegativeZ	BLongLongZfntypeextrar   r   r   ru     s2    zVGenericEngine._load_constantc                 C   sD   t |tjo| }| ||||}t||| t|j| d S r   )	rb   r   r   r   ru   r~   r[   r/   rj   )r   rC   rP   r6   r8   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"|jD ]}| d| qd S | ||}| j| | j}|d|  |d t|j|jD ]\}}| 	|| qb|d |d |  d S )NTr   r`   r   ra   )
r   enumeratorsre   r   r   rj   r"   rk   
enumvaluesr   )r   rC   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)ru   )rA   r   r6   r   rC   r   r   rD     s   z4VGenericEngine._loading_gen_enum.<locals>.<listcomp>T)r   r   rz   r   Zpartial_resolvedr   r   )r   rC   rP   r6   r,   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   )rk   r   r   r~   r[   r/   rj   )r   rC   rP   r6   r8   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   )re   )r   rC   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 r   )ru   r~   r[   r/   rj   )r   rC   rP   r6   r8   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)r`   z  return sizeof(%s);ra   Fr   )r   )
rb   r   r   length_is_unknownr"   r   rj   rw   itemre   )r   rC   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 u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   r7   Zptrr   r   getterg  s    z3VGenericEngine._loaded_gen_variable.<locals>.getterc                    s   | d< d S r   r   )r8   r   r   r   r   setteri  s    z3VGenericEngine._loaded_gen_variable.<locals>.setter)rb   r   r   r   r
   r   r|   r{   r   divmodr   r   Zresolve_lengthrw   ru   r   castr~   r[   r/   rj   rZ   property)r   rC   rP   r6   r8   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   )5r2   r3   r4   Z
_class_keyZ_gen_python_moduler   r   r   r   r"   r.   r;   rH   r(   r<   rV   rX   Z_generate_gen_typedef_declZ_loading_gen_typedefZ_loaded_gen_typedefrr   Z_loading_gen_functionr   r}   r   r   r   r   r   r   r   r   r   r   r   r   re   r   Z_loading_gen_constantru   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   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       g