3
M��ZR� � @ s� d dl mZ d dlmZmZ d dlmZmZ yd dl mZ W n e
k
rX ddl Z Y nX ddl Z ddl Z ddl
Z
y(e
jd#k r�ddlZnddlZej� ZW n e
k
r� dZY nX dZe jd e je jB �Ze jd
e je jB �Ze jd �Ze jd �Ze jd
�Ze jd�Z da e jde j �Z e jd�Z e jd�Z!e jd�Z"e jd�Z#e jd�Z$e jd�Z%e jd�Z&dd� Z'dd� Z(dd � Z)d d � Z*d d � Z+G d!d"� d"e,�Z-dS )$� )�model)� COMMON_TYPES�resolve_common_type)�FFIError� CDefError)�
_pycparser� N� zz /\*.*?\*/|//([^\n\\]|\\.)*?$z?^\s*#\s*define\s+([A-Za-z_][A-Za-z_0-9]*)\b((?:[^\n\\]|\\.)*?)$z =\s*\.\.\.\s*[,}]|\.\.\.\s*\}z__dotdotdot\d+__$z\[\s*\.\.\.\s*\]z\w+|\Sz-?0?x?[0-9a-f]+[lu]*$z\b(__stdcall|WINAPI)\bz[(]\s*(__stdcall|WINAPI)\bz \b__cdecl\bz9\bextern\s*"(Python|Python\s*\+\s*C|C\s*\+\s*Python)"\s*.z'[*]\s*((const|volatile|restrict)\b\s*)+z3(\b(int|long|short|signed|unsigned|char)\s*)+\.\.\.z\b(double|float)\s*\.\.\.c C s t d krtj� a t S )N)�
_parser_cache� pycparserZCParser� r r � /usr/lib64/python3.6/cparser.py� _get_parser* s r c C s g }x�t j| �}|sP |j| d |j� � � |jd� d}|j|j� � |j� }| jd|�rn|jd� |d7 }d}|}x`|t| �k r�| | }|dkr�|d7 }n.|dkr�|dkr�P |d8 }n|dkr�|dkr�P |d7 }qxW | ||� | | |d � } qW |j| � dj|�S )N�(�)�*r r z,;=� ) �_r_star_const_space�search�append�start�group�end�
startswith�len�join)�csource�parts�match�closing�endpos�level�i�cr r r
� _workaround_for_old_pycparser0 s<
r$ c C s0 g }�xt j| �}|sP |j� d }|j| d |j� � � d|jd�krT|jd� n
|jd� | | dkr�| jd|�}|dk r�td��| jd|d |�dkr�td ��|j| |d |� � | |d d � } nB| jd
|�}|dk r�td ��|j| ||d � � | |d d � } |jd � qW |j| � d
j |�S )Nr �Cz(void __cffi_extern_python_plus_c_start; z!void __cffi_extern_python_start; �{�}r z!'extern "Python" {': no '}' foundz7cannot use { } inside a block 'extern "Python" { ... }'�;z 'extern "Python": no ';' foundz void __cffi_extern_python_stop;r )
�_r_extern_pythonr r r r r �findr �NotImplementedErrorr )r r r r r Z semicolonr r r
�_preprocess_extern_pythonZ s4
r, c C s\ t jd| �} i }x6tj| �D ](}|j� \}}|jdd�j� }|||< q W tjd| �} tjdk rft | �} t
jd| �} t jd| �} t jd| �} t
| �} tjd| �} ttj| ��}x�tt|��D ]v\}}|j� }| | dk�r| jd ||j� �}d
| d |� || |d d � f } q�d | d |� || |d d � f } q�W tjd
| �} tjd| �} | jd d�|fS )N� z\
r z2.14z volatile volatile const(z volatile volatile const z[__dotdotdotarray__]�=z...z%s,__dotdotdot%d__ %sr z%s __dotdotdot%d__ %sz __dotdotdotint__ z __dotdotdotfloat__ z __dotdotdot__ )�
_r_comment�sub� _r_define�finditer�groups�replace�stripr � __version__r$ � _r_stdcall2� _r_stdcall1�_r_cdeclr, �_r_partial_array�list�_r_partial_enum� enumerate�reversedr r* r �_r_int_dotdotdot�_r_float_dotdotdot) r �macrosr Z macronameZ
macrovalueZmatchesZnumber�pZp2r r r
� _preprocess� s4
rC c C s t t�}|jd� |jd� |jd� |jd� |jd� t � }d}d}d}x�tj| �D ]�}||kr�|dkr�|r�|j|� |j|� d}nj|dkr�d }d}nX|dkr�|d
7 }nF|dkr�|d
8 }n4|dkr�|r�|dkr�|j|� |j|� n
|j|� |}qXW |S ) Nr( �,r r �typedefFr r Tr )�setr �add�_r_words�findall�discard)r Zlook_for_wordsZ
words_usedZ
is_typedefZparenZ
previous_wordZwordr r r
�_common_type_names� s>
rK c @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd8d
d �Zd d
� Zdd� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zd9d d �Zd d � Zd:d"d#�Zd;d$d%�Zdd.d/�Zd0d1� Zd2d3� Zd4d5� Zd6d7� Z d!S )?�Parserc C s: i | _ t� | _d| _tj� | _i | _i | _g | _ d | _
d S )Nr ) �
_declarationsrF �_included_declarations�_anonymous_counter�weakref�WeakKeyDictionary�_structnode2type�_options�_int_constants� _recomplete�_uses_new_feature)�selfr r r
�__init__� s
zParser.__init__c C s4 t |�\}}t|�}g }x"ztypedef int %s;z@typedef int __dotdotdotint__, __dotdotdotfloat__, __dotdotdot__;z# 1 "%s"�
)rC rK �sortedrM r r rJ �CDEF_SOURCE_STRINGr �lock�acquirer �parser Zc_parserZ
ParseError�convert_pycparser_error�release) rW r rA Zctn� typenames�name� csourcelines�typenameZ fullcsource�ast�er r r
�_parse� s4
"
z
Parser._parsec C sf d }t |�}tjdtf |�}|rbt|jd�d�}|j� }d| koPt|�kn rb||d }|S )Nz %s:(\d+):r �
)�str�rer r\ �intr �
splitlinesr )rW rg r �line�msgr Zlinenumrd r r r
�_convert_pycparser_error s z Parser._convert_pycparser_errorc C s@ | j ||�}t|�}|r*d|j� |f }n
d|f }t|��d S )Nzcannot parse "%s"
%szparse error
%s)rp rj r5 r )rW rg r rn ro r r r
r` + s
z Parser.convert_pycparser_errorFc C s0 | j }z |||d�| _ | j|� W d || _ X d S )N)�override�packed� dllexport)rS �_internal_parse)rW r rq rr rs Z prev_optionsr r r
r_ 5 s
z Parser.parsec C s | j |�\}}}| j|� t|j�}x|D ]}|jdkr*P q*W d }�yBd| _�x4|D �]*}|}t|tjj �rx| j
|� qTt|tjj ��r^|js�t d|��d}t|j
j
tjj�r�|j
j
jd jd�r�| j|�}nxt|j
tjj��r.t|j
j
tjj��r.t|j
j
j
tjj��r.|j
j
j
jd jd��r.| j|�}n| j|j
|jdd�\}}| jd |j ||d
� qT|jjd k�rnqTt d |jj |��qTW W n� t k
�r� } z&t| j�dk�r�| j|f | _� W Y d d } ~ X nN tk
�r } z0| j | |�}
|
�r| jd d
|
f| _� W Y d d } ~ X nX d S )N�
__dotdotdot__�__cffi_extern_python_stopz!typedef does not declare any namer r Z __dotdotdotT)rc �partial_length_okztypedef )�qualsZPragmazBunexpected <%s>: this construct is valid C but not valid in cdef()z
*** Err: %s�����ry ) rh �_process_macros�iter�extrc �_inside_extern_python�
isinstancer �c_astZDecl� _parse_declZTypedefr �type�IdentifierType�namesr �_get_unknown_type�PtrDecl�TypeDecl�_get_unknown_ptr_type�_get_type_and_quals�_declare� __class__�__name__r �argsr rp ) rW r rf rA �iterator�declZ current_declrx Zrealtyperg ro r r r
rt ? sV
zParser._internal_parsec C s8 || j kr*| j | |kr d S td|f ��|| j |< d S )Nz%multiple declarations of constant: %s)rT r )rW �key�valr r r
�_add_constantsu s
zParser._add_constantsc C s� |j � jd�}|jd�}|r(|dd � }|jd�rV|dkrV|jd� rVd|dd � }t|d�}|rj| }| j||� | jd| |� d S ) NZul�-r �0�0xZ0or zmacro )�lower�rstripr rl r� r� )rW rc Zint_str�negZpyvaluer r r
�_add_integer_constant} s
z Parser._add_integer_constantc C sh xb|j � D ]V\}}|j� }tj|�r2| j||� q
|dkrL| jd| |� q
td||||f ��q
W d S )Nz...zmacro z�only supports one of the following syntax:
#define %s ... (literally dot-dot-dot)
#define %s NUMBER (with NUMBER an integer constant, decimal/hex/octal)
got:
#define %s %s)�itemsr5 �_r_int_literalr r� r� r )rW rA r� �valuer r r
rz � s
zParser._process_macrosc C sX | j ||�}| jjd�r d}n$| jdkr.d}n| jdkr>d}nd}| j||j |� d S )Nrs zdllexport_python Z__cffi_extern_python_startzextern_python Z!__cffi_extern_python_plus_c_startzextern_python_plus_c z function )�_get_type_pointerrS �getr} r� rc )rW �tprx r� �tagr r r
�_declare_function� s
zParser._declare_functionc C s� |j }t|tjj�r:| j||jd�\}}| j|||� �n�t|tjj�rV| j d|� nHt|tjj
�rr| j d|� n,t|tjj �r�| j d|� n|js�t d|��|j�r�| j|dd�\}}|j
r�| j|||� �n$|j� o�t|d�o�t|jd �o�tj|jj��r| j|j|jj� n�|j� �rxt|jtjj��rx|jjd
k�rxt|jjd ��rxtj|jjj��rx| j|jd
|jjj � n||tjk�r�|jjd ��r�|j| _nX| jd k�r�t d
��|tj @ �r�|j �r�| j d|j ||d� n| j d|j ||d� d S )N)rc �struct�union�enumz'construct does not declare any variableT)rw �initr� r� Z__cffi_extern_python_rv z| s z/Parser._parse_function_type..rx r r� r� Z __stdcallry ry ry �����)r; �getattrr� r= r� r r r~ r� r r r� r� r� �popr r� r� rx r� �tuple) rW r� �funcnamer� r" �argZellipsisr� �resultrx �abir )rW r
r� g s6
zParser._parse_function_typec C s6 t |tj�rtj|j|�S t |tj�r.|j� S |S d S )N)r~ r r� r� �itemr� r� )rW r� rx r r r
r� � s
zParser._as_func_argc
C s, y
| j | S tk
r Y nX |}|j}|d krb|d k rDd| }n| jd7 _d| j }d }n"|}d||f }| jj|d�\}} |d k�r|dkr�tj|d d d �}nP|dkr�tj|d d d �}n6|dkr�|dkr�t d ��| j
||j �}nt d
|f ��|d k �r2| j
||� n"|dk�r2|j d k �r2td | ��|j�sD|j|� |j�rjd |jk�rj| j
d
|j |� || j |< |dk�r�|S |jd k�r�|S |jd k �r�t d| ��g }
g } g } g }
x�|jD ]�}t|jtjj��r�dj|jj�dk�r�| j||� �q�|jd k�rd}n | j |j�}d| _ | j |jdd�\}}| j �rH| j||� t|tj��rj|j �rj| j||� |
j |j�pxd� | j |� | j |� |
j |� �q�W t!|
�|_t!| �|_"t!| �|_#t!|
�|_$| dgt%| � k�r t|tj��r |j �r td|f ��| j&jd�|_'|j(�r(d|_(| j)j |� |S )Nz$%sr z$%dz%s %sr� r� r� ru z!Enums cannot be declared with ...z kind = %rz^enum %s: the '{}' declaration should appear on the first time the enum is mentioned, not laterr� z anonymous %sz"duplicate declaration of struct %sr FT)rw z#%s: using both bitfields and '...;'rr r )NNry ry )*rR �KeyErrorrc rO rM r� r Z
StructTypeZ UnionTyper �_build_enum_type�values�AssertionErrorr� r+ r� Zforce_the_name�decls�fldnamesr~ r� r r r� r r� �
_make_partial�bitsizer� �_partial_lengthr� �partialr r� �fldtypes�
fldbitsize�fldqualsr rS rr Z completedrU )rW �kindr� rc r� Z
force_name�
explicit_namer� r� �_r� r� r� r� r� r� Zfqualr r r
r� � s�
z"Parser._get_struct_union_enum_typec C sB t |tj�std|f ��|j� r8| r8td|f ��d|_d S )Nz%s cannot be partialz %s is partial but has no C nameT)r~ r Z
StructOrUnionr Z
has_c_namer+ r� )rW r� r� r r r
r� s
zParser._make_partialc C s� t |tjj�r�|j}|jd�rF|jd�s2|jd�r sN
*27(