3
M��Z0T � @ s� d dl Z d dlZddlmZ ddlmZmZmZ dZdZ dZ
dd� Z G d d
� d
e �Z
G d d � d e
�ZG d
d� de�Ze� ZG dd� de�ZG dd� de�ZG dd� de�ZG dd� de�ZG dd� de�ZG dd� de�ZG dd � d e�ZG d d � d e�Zee�Zd d � Zee�ZG d!d"� d"e�Z G d#d$� d$e�Z e ed%�d�Z G d&d'� d'e
�Z G d(d)� d)e �Z G d*d+� d+e �Z!G d,d-� d-e �Z"G d.d/� d/e �Z#d� N� )�
allocate_lock)� CDefError�VerificationError�VerificationMissing� � c C s@ | t @ rd|j� }| t@ r(d|j� }| t@ r)r )r r r r �__repr__A s zBaseTypeByIdentity.__repr__c s � fdd�� j D �S )Nc s g | ]}|t � |�f�qS r )�getattr)�.0�name)r r r �
E s z1BaseTypeByIdentity._get_items..)�_attrs_)r r )r r �
_get_itemsD s z BaseTypeByIdentity._get_itemsN)r r r )F) �__name__�
__module__� __qualname__�
is_array_type�is_raw_functionr! r r" r# r, r- r3 r r r r r s
r c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�BaseTypec C s | j |j ko| j� |j� kS )N)� __class__r3 )r �otherr r r �__eq__J s zBaseType.__eq__c C s
| |k S )Nr )r r; r r r �__ne__N s zBaseType.__ne__c C s t | jt| j� �f�S )N)�hashr: �tupler3 )r r r r �__hash__Q s zBaseType.__hash__N)r4 r5 r6 r< r= r@ r r r r r9 H s r9 c @ s e Zd Zf Zdd� Zdd� ZdS )�VoidTypec C s
d| _ d S )Nzvoid&)r )r r r r �__init__X s zVoidType.__init__c C s t | |d�S )NZ
new_void_type)� global_cache)r r( r) r r r r& [ s zVoidType.build_backend_typeN)r4 r5 r6 r2 rB r& r r r r rA U s rA c @ s e Zd Zdd� ZdS )�BasePrimitiveTypec C s dS )NFr )r r r r �is_complex_typeb s z!BasePrimitiveType.is_complex_typeN)r4 r5 r6 rE r r r r rD a s rD c 4 @ s� e Zd ZdZdddddddddddddddddddddddddddddddddddddddddddddddddddd�3Zdd� Zd d
� Zd d � Zd
d� Zdd� Z dd� Z
dS )�
PrimitiveTyper0 �c�i�f�j)3�charZshort�int�longz long longz signed charz
unsigned charzunsigned shortz unsigned intz
unsigned longzunsigned long long�floatZdoublez long doublezfloat _Complexzdouble _ComplexZ_BoolZwchar_tZchar16_tZchar32_tZint8_tZuint8_tZint16_tZuint16_tZint32_tZuint32_tZint64_tZuint64_tZ int_least8_tZ
uint_least8_tZ
int_least16_tZuint_least16_tZ
int_least32_tZuint_least32_tZ
int_least64_tZuint_least64_tZ int_fast8_tZ uint_fast8_tZ int_fast16_tZ
uint_fast16_tZ int_fast32_tZ
uint_fast32_tZ int_fast64_tZ
uint_fast64_tZintptr_tZ uintptr_tZintmax_tZ uintmax_tZ ptrdiff_tZsize_tZssize_tc C s || _ |d | _d S )Nr )r0 r )r r0 r r r rB � s zPrimitiveType.__init__c C s | j | j dkS )NrG )�ALL_PRIMITIVE_TYPESr0 )r r r r � is_char_type� s zPrimitiveType.is_char_typec C s | j | j dkS )NrH )rO r0 )r r r r r# � s z PrimitiveType.is_integer_typec C s | j | j dkS )NrI )rO r0 )r r r r �
is_float_type� s zPrimitiveType.is_float_typec C s | j | j dkS )NrJ )rO r0 )r r r r rE � s z PrimitiveType.is_complex_typec C s t | |d| j�S )NZnew_primitive_type)rC r0 )r r( r) r r r r& � s z PrimitiveType.build_backend_typeN)r0 ) r4 r5 r6 r2 rO rB rP r# rQ rE r& r r r r rF f st rF c @ s( e Zd Zd Zdd� Zdd� Zdd� ZdS )
�UnknownIntegerTyper0 c C s || _ |d | _d S )Nr )r0 r )r r0 r r r rB � s zUnknownIntegerType.__init__c C s dS )NTr )r r r r r# � s z"UnknownIntegerType.is_integer_typec C s t d| j ��d S )Nz4integer type '%s' can only be used after compilation)�NotImplementedErrorr0 )r r( r) r r r r& � s z%UnknownIntegerType.build_backend_typeN)r0 )r4 r5 r6 r2 rB r# r& r r r r rR � s rR c @ s e Zd ZdZdd� Zdd� ZdS )�UnknownFloatTyper0 c C s || _ |d | _d S )Nr )r0 r )r r0 r r r rB � s zUnknownFloatType.__init__c C s t d| j ��d S )Nz2float type '%s' can only be used after compilation)rS r0 )r r( r) r r r r& � s z#UnknownFloatType.build_backend_typeN)r0 )r4 r5 r6 r2 rB r& r r r r rT � s rT c @ s e Zd ZdZd dd�ZdS )
�BaseFunctionType�argsr �ellipsis�abiNc C s� || _ || _|| _|| _dd� | j D �}| jr8|jd� |p@dg}| jdj|�f }|d k r||d d� | d |dd � }| jjjd|�| _d S ) Nc S s g | ] }|j � �qS r )r )r/ �argr r r r1 � s z-BaseFunctionType.__init__..z...�voidz, r r r ) rV r rW rX �append�
_base_pattern�joinr r )r rV r rW rX Zreprargsr r r r rB � s
zBaseFunctionType.__init__)rV r rW rX )N)r4 r5 r6 r2 rB r r r r rU � s rU c @ s$ e Zd ZdZdZdd� Zdd� ZdS )�RawFunctionTypez(&)(%s)Tc C s t d| f ��d S )NzPcannot render the type %r: it is a function type, not a pointer-to-function type)r )r r( r) r r r r& � s z"RawFunctionType.build_backend_typec C s t | j| j| j| j�S )N)�FunctionPtrTyperV r rW rX )r r r r �as_function_pointer� s z#RawFunctionType.as_function_pointerN)r4 r5 r6 r\ r8 r& r` r r r r r^ � s r^ c @ s e Zd ZdZdd� Zdd� ZdS )r_ z(*&)(%s)c C s� | j j||�}g }x | jD ]}|j|j||�� qW f }| jdkrl| jsly|jjf}W n tk
rj Y nX t | |dt
|�|| jf|�� S )NZ __stdcallZnew_function_type) r r, rV r[ rX rW �_backendZ FFI_STDCALL�AttributeErrorrC r? )r r( r) r rV �tpZabi_argsr r r r& � s
z"FunctionPtrType.build_backend_typec C s t | j| j| j| j�S )N)r^ rV r rW rX )r r r r �as_raw_function s z FunctionPtrType.as_raw_functionN)r4 r5 r6 r\ r& rd r r r r r_ � s r_ c @ s" e Zd Zd Zd
dd�Zdd� ZdS ) � PointerType�totyper
r c C s> || _ || _t|d�}|jr*d|j� f }|jjd|�| _d S )Nz *&z(%s)r )rf r
r r7 r
r r )r rf r
Zextrar r r rB s
zPointerType.__init__c C s | j j||dd�}t| |d|�S )NT)r* �new_pointer_type)rf r, rC )r r( r) ZBItemr r r r& s z PointerType.build_backend_typeN)rf r
)r )r4 r5 r6 r2 rB r& r r r r re s
re c C s
t | t�S )N)re r )rf r r r �ConstPointerType s rh c @ s e Zd ZdZddd�ZdS ) �NamedPointerTyperf r0 r c C s" t j| ||� || _|d | _d S )Nr )re rB r0 r )r rf r0 r
r r r rB s zNamedPointerType.__init__N)rf r0 )r )r4 r5 r6 r2 rB r r r r ri s ri c @ s, e Zd Zd ZdZdd� Zdd� Zdd � Zd
S ) � ArrayType�item�lengthTc C sF || _ || _|d krd}n|dkr(d}nd| }| j jjd|�| _d S )Nz&[]z...z
&[/*...*/]z&[%s]r )rk rl r r )r rk rl Zbracketsr r r rB ) s zArrayType.__init__c C s t | j|�S )N)rj rk )r Z newlengthr r r �resolve_length6 s zArrayType.resolve_lengthc C sJ | j dkrtd| f ��| jj||� t| j�j||�}t| |d|| j �S )Nz...z)cannot render the type %r: unknown lengthZnew_array_type)rl r rk r, re rC )r r( r) ZBPtrItemr r r r& 9 s
z ArrayType.build_backend_typeN)rk rl )r4 r5 r6 r2 r7 rB rm r& r r r r rj % s
rj rK c @ s, e Zd Zd ZdZdd� Zdd� Zdd� ZdS )
�StructOrUnionOrEnumr0 Nc C s$ | j pd| j| jf }|d | _d S )Nz%s %sr )� forcename�kindr0 r )r r0 r r r �build_c_name_with_markerH s z,StructOrUnionOrEnum.build_c_name_with_markerc C s || _ | j� d S )N)ro rq )r ro r r r �force_the_nameL s z"StructOrUnionOrEnum.force_the_namec C s | j d d� S )Nr �����)r )r r r r �get_official_nameP s z%StructOrUnionOrEnum.get_official_name)r0 )r4 r5 r6 r2 ro rq rr rt r r r r rn D s
rn c @ sj e Zd ZdZdZdZdZddd�Zdd� Zdd d
�Z d d � Z
dd
d�Z dd� Z dd� Z
dd� Zdd� ZdS )�
StructOrUnionNr Fc C s* || _ || _|| _|| _|| _| j� d S )N)r0 �fldnames�fldtypes�
fldbitsize�fldqualsrq )r r0 rv rw rx ry r r r rB [ s zStructOrUnion.__init__c c sB | j d k r>x2t| j| j �D ] \}}|dkrt|t�r|V qW d S )Nr )rw �ziprv �
isinstanceru )r r0 �typer r r �anonymous_struct_fieldsc s
z%StructOrUnion.anonymous_struct_fieldsTc c s� | j }|d kr dt| j� }xbt| j| j| j|�D ]J\}}}}|dkrnt|t�rn|rnx&|j� D ]
}|V q^W q2||||fV q2W d S )Nr r )r ) ry �lenrv rz rw rx r{ ru �
enumfields)r Z expand_anonymous_struct_unionry r0 r| �bitsizer
r r r r r i s zStructOrUnion.enumfieldsc C s~ g }g }g }g }x@| j � D ]4\}}}}|j|� |j|� |j|� |j|� qW t|�| _t|�| _t|�| _t|�| _d S )N)r r[ r? rv rw rx ry ) r �names�typesZbitsizesry r0 r| r� r
r r r �
force_flattenw s
zStructOrUnion.force_flattenc C s$ t j| |||�}|s | j||� |S )N)rn r, �finish_backend_type)r r( r) r* r+ r r r r, � s
z StructOrUnion.get_cached_btypec s | j r$| j dkr td| jf ��d S � j| }d| _ | jd krB�n�| jd kr�� �fdd�| jD �}tt| j|| j ��}d}| j
r�d}� j j ||| d d
|� �nfg }| j\}}} }
�x&t
t| j��D �]} || } | j| }
t|
t�o�|
jdk�rn|
jj� ��}t| � j|��\}}|dk�r<| jd | j| j| �p4d
f � |
j|�}
| jd | � |
f | j| d d � | _|
j� ��}t|
t��r�|
jd k�r�n8� j|�}|| k�r�| jd | j| j| �p�d
|| f � |j|� q�W tt| j|| j |��}� j j ||| | |
� d| _ d S )Nr z(recursive structure declaration for '%s'r c s g | ]}|j � ���qS r )r, )r/ rc )r( r) r r r1 � s z5StructOrUnion.finish_backend_type..r � z...z field '%s.%s' has a bogus size?z{}z=field '%s.%s' is declared as %d bytes, but is really %d bytesrs rs )� completedrS r0 r$ rw � fixedlayout�listrz rv rx �packedra Zcomplete_struct_or_union�ranger~ r{ rj rl rk r, �divmod�sizeof�_verification_errorrm r[ )r r( r) r+ rw ZlstZsflagsZfieldofsZ fieldsizeZ totalsizeZtotalalignmentrH ZfsizeZftypeZ BItemTypeZnlenZnrestZ
BFieldTypeZ bitemsizer )r( r) r r� � s`
z!StructOrUnion.finish_backend_typec C s t |��d S )N)r )r �msgr r r r� � s z!StructOrUnion._verification_errorc C s | j r | jd kr t| j� ��d S )N)�partialr� r r )r r r r �check_not_partial� s z StructOrUnion.check_not_partialc C s. | j � |j| � t| |d| j | j� | d�S )Nz new_%s_type)�key)r� r[ rC rp rt )r r( r) r r r r& � s
z StructOrUnion.build_backend_type)N)T)F)r4 r5 r6 r� r� r� r� rB r} r r� r, r� r� r� r& r r r r ru U s
<ru c @ s e Zd ZdZdS )�
StructType�structN)r4 r5 r6 rp r r r r r� � s r� c @ s e Zd ZdZdS )� UnionType�unionN)r4 r5 r6 rp r r r r r� � s r� c @ sB e Zd ZdZdZdZddd�Zdd� Zdd � Zd
d � Z d d
� Z
dS )�EnumType�enumFNc C s$ || _ || _|| _|| _| j� d S )N)r0 � enumerators�
enumvalues� baseinttyperq )r r0 r� r� r� r r r rB � s
zEnumType.__init__c C s4 t j| |� | jd kr0| j� }d|jdd� | _d S )Nr r �_)rn rr ro rt r )r ro r0 r r r rr � s
zEnumType.force_the_namec C s | j r| j rt| j� ��d S )N)r� �partial_resolvedr r )r r r r r� � s zEnumType.check_not_partialc
C s4 | j � | j||�}t| |d| j� | j| j|| d�S )NZ
new_enum_type)r� )r� �build_baseinttyperC rt r� r� )r r( r) Z
base_btyper r r r& � s zEnumType.build_backend_typec
C sH | j d k r| j j||�S | jr4t| j�}t| j�}nDdd l}y tj� W n tk
r\ Y nX |j d| j
� � d }}|dk r�d}t d�}t d�}nd}t d�}t d�}|j||�} |j||�}
|j | �} |j |
�} |d
d| d > ko�|dd| | > k �r| S |d d| d > k�r4|dd| | > k �r4|
S t
d | j
� ��d S ) Nr zU%r has no values explicitly defined; guessing that it is equivalent to 'unsigned int'r rL rM z unsigned intz
unsigned longr� z=%s values don't all fit into either 'long' or 'unsigned long'rs rs )r� r, r� �min�max�warningsZ__warningregistry__�clear� NameError�warnr rF r� r )
r r( r) Zsmallest_valueZ
largest_valuer� ZsignZ
candidate1Z
candidate2Zbtype1Zbtype2Zsize1Zsize2r r r r� s@
zEnumType.build_baseinttype)N) r4 r5 r6 rp r� r� rB rr r� r& r� r r r r r� � s
r� c C s2 |d krd| }t |d d d �}|j| � d|_|S )Nz$%s� unknown_type)r� rr �origin)r0 �
structnamerc r r r r� * s
r� c C s( |d krd| }t |d d d �}t|| �S )Nz$$%s)r� ri )r0 r� rc r r r �unknown_ptr_type2 s r� c
C sF t | tj�rtS t�( tt| �d�s2tj� t| �_ t| �j S Q R X d S )N� __typecache)
r{ r� �
ModuleType�_typecache_cffi_backend� global_lock�hasattrr| �weakref�WeakValueDictionaryr� )Zbackendr r r �
get_typecache< s r� c
O s� |j d||f�}y
|j| S tk
r. Y nX yt|j|�|� }W n4 tk
rx } ztd|| |f ��W Y d d }~X nX |j}t�( |j|�} | d kr�|||< |S | S W d Q R X d S )Nr� z
%s: %r: %s)�popZ
_typecacher% r. ra rS r� �get)
Zsrctyper( �funcnamerV �kwdsr� �res�e�cacheZres1r r r rC G s
$
rC c C s t d| d|�S )N�?rg )rC )r( r+ r r r �
pointer_cache^ s r� c C s@ | j r sN
/
L G