a
    Lg4.                     @   s^  d Z ddlmZmZmZmZ ddlmZmZ ddl	m
Z
mZmZmZmZ ddlmZmZmZmZmZ ddlmZ eee eee eedd	d
Zeee eee eedddZeedddZeeedddZeeedddZeedddZeee eee eeedddZeee eeeeedddZ G dd  d eZ!eeeed!d"d#d$Z"d!S )%aY  
QAPI command marshaller generator

Copyright IBM, Corp. 2011
Copyright (C) 2014-2018 Red Hat, Inc.

Authors:
 Anthony Liguori <aliguori@us.ibm.com>
 Michael Roth <mdroth@linux.vnet.ibm.com>
 Markus Armbruster <armbru@redhat.com>

This work is licensed under the terms of the GNU GPL, version 2.
See the COPYING file in the top-level directory.
    )DictListOptionalSet   )c_namemcgen)QAPIGenCQAPISchemaModularCVisitorbuild_paramsgen_special_features	ifcontext)
QAPISchemaQAPISchemaFeatureQAPISchemaIfCondQAPISchemaObjectTypeQAPISchemaType)QAPISourceInfo)namearg_typeboxedret_type	coroutinereturnc              	   C   s2   t d|r| pd|rdndt| t||ddS )Nz8
%(c_type)s %(coroutine_fn)sqmp_%(c_name)s(%(params)s);
Zvoidcoroutine_fn  zError **errp)c_typecoroutine_fnr   params)r   r   r   r   )r   r   r   r   r    r   @/home2/Puru_Virtio_Blk/virtio-demo/qemu/scripts/qapi/commands.pygen_command_decl)   s    

r!   )r   r   r   r   gen_tracingr   c           
      C   s,  d}d}|r|sJ d}nT|rn|j r(J |jD ]>}|j r@J | rZ|dt|j 7 }|dt|j 7 }q.d}|rzd}t| } |  }	|r|td|	| d7 }|td| ||d	7 }|td
7 }|r|td| d7 }|td7 }|r|td| d7 }|r(|r|td|	| d7 }n|td| d7 }|S )Nr   z&arg, zarg.has_%s, zarg.%s, z	retval = z

    if (trace_event_get_state_backends(TRACE_QMP_ENTER_%(upper)s)) {
        g_autoptr(GString) req_json = qobject_to_json(QOBJECT(args));

        trace_qmp_enter_%(name)s(req_json->str);
    }
)upperr   z)

    %(lhs)sqmp_%(name)s(%(args)s&err);
)r   argslhsz
    if (err) {
z@
        trace_qmp_exit_%(name)s(error_get_pretty(err), false);
r   z=
        error_propagate(errp, err);
        goto out;
    }
z8

    qmp_marshal_output_%(c_name)s(retval, ret, errp);
r   z

    if (trace_event_get_state_backends(TRACE_QMP_EXIT_%(upper)s)) {
        g_autoptr(GString) ret_json = qobject_to_json(*ret);

        trace_qmp_exit_%(name)s(ret_json->str, true);
    }
z+

    trace_qmp_exit_%(name)s("{}", true);
)	ZbranchesmembersifcondZ
is_presentZneed_hasr   r   r#   r   )
r   r   r   r   r"   retZargstrZmembr%   r#   r   r   r    gen_call7   sV    




r+   )r   r   c                 C   s   t d|  |  dS )Na  

static void qmp_marshal_output_%(c_name)s(%(c_type)s ret_in,
                                QObject **ret_out, Error **errp)
{
    Visitor *v;

    v = qobject_output_visitor_new_qmp(ret_out);
    if (visit_type_%(c_name)s(v, "unused", &ret_in, errp)) {
        visit_complete(v, ret_out);
    }
    visit_free(v);
    v = qapi_dealloc_visitor_new();
    visit_type_%(c_name)s(v, "unused", &ret_in, NULL);
    visit_free(v);
}
)r   r   )r   r   r   )r   r   r   r    gen_marshal_output   s    r,   )r   r   r   c                 C   s   d|r
dndt | dd S )Nz7void %(coroutine_fn)sqmp_marshal_%(c_name)s(%(params)s)r   r   z(QDict *args, QObject **ret, Error **errp)r   r   r   r'   r   r   r   r   r    build_marshal_proto   s
    
r.   c                 C   s   t dt| |dS )Nz
%(proto)s;
proto)r   r.   r-   r   r   r    gen_marshal_decl   s    r1   r   r   c                 C   s   t dt| dS )Nzl
qmp_enter_%(name)s(const char *json) "%%s"
qmp_exit_%(name)s(const char *result, bool succeeded) "%%s %%d"
r&   )r   r   r&   r   r   r    	gen_trace   s    r3   )r   r   r   r   r"   r   r   c           	      C   s   |p|o|   }|r*|d us"J | }tdt| |d}|rT|td| d7 }|rh|td|d7 }|td7 }|r|td|d	7 }n|td
7 }|td7 }|t| ||||7 }|td7 }|td7 }|r|td|d	7 }|td7 }|td7 }|S )NzJ

%(proto)s
{
    Error *err = NULL;
    bool ok = false;
    Visitor *v;
r/   z
    %(c_type)s retval;
)r   z
    %(c_name)s arg = {0};
r'   z

    v = qobject_input_visitor_new_qmp(QOBJECT(args));
    if (!visit_start_struct(v, NULL, NULL, 0, errp)) {
        goto out;
    }
zm
    if (visit_type_%(c_arg_type)s_members(v, &arg, errp)) {
        ok = visit_check_struct(v, errp);
    }
)Z
c_arg_typez'
    ok = visit_check_struct(v, errp);
zG
    visit_end_struct(v, NULL);
    if (!ok) {
        goto out;
    }
z

out:
    visit_free(v);
zU
    v = qapi_dealloc_visitor_new();
    visit_start_struct(v, NULL, NULL, 0, NULL);
z7
    visit_type_%(c_arg_type)s_members(v, &arg, NULL);
z3
    visit_end_struct(v, NULL);
    visit_free(v);

}
)Zis_emptyr   r   r.   r   r+   )	r   r   r   r   r"   r   Z	have_argsZarg_type_c_namer*   r   r   r    gen_marshal   s@    

r5   )r   featuressuccess_response	allow_ooballow_preconfigr   r   c                 C   sd   g }|s|dg7 }|r |dg7 }|r.|dg7 }|r<|dg7 }t d| t| d|pTdt|d}|S )	NZQCO_NO_SUCCESS_RESPZQCO_ALLOW_OOBZQCO_ALLOW_PRECONFIGZQCO_COROUTINEzs
    qmp_register_command(cmds, "%(name)s",
                         qmp_marshal_%(c_name)s, %(opts)s, %(feats)s);
z | r   )r   r   ZoptsZfeats)r   r   joinr   )r   r6   r7   r8   r9   r   optionsr*   r   r   r    gen_register_command  s    



r<   c                       s   e Zd Zeed fddZeddddZeddd	d
ZddddZ	ee
e eee e
e e
e eeeeeeddddZ  ZS )QAPISchemaGenCommandVisitor)prefixr"   c                    s(   t  j|ddd t|d i | _|| _d S )Nqapi-commandsz# * Schema-defined QAPI/QMP commands)r"   )super__init____doc___visited_ret_types_gen_tracing)selfr>   r"   	__class__r   r    rA   2  s    z$QAPISchemaGenCommandVisitor.__init__Nr2   c                 C   s   t  | j| j< | d|}| d|}| d|}| jtd||d | jrr|dkrr| jtdt|ddd	 | jtd
|d d S )Nr?   z
qapi-typesz
qapi-visitz
#include "qemu/osdep.h"
#include "qapi/compat-policy.h"
#include "qapi/visitor.h"
#include "qapi/qmp/qdict.h"
#include "qapi/dealloc-visitor.h"
#include "qapi/error.h"
#include "%(visit)s.h"
#include "%(commands)s.h"
)commandsvisitzJ
#include "qapi/qmp/qjson.h"
#include "trace/trace-%(nm)s_trace_events.h"
FZprotect)Znmz
#include "%(types)s.h"

)types)	setrC   _gencZ_module_basenameaddr   rD   r   _genh)rE   r   rH   rK   rI   r   r   r    _begin_user_module:  s    




z.QAPISchemaGenCommandVisitor._begin_user_module)schemar   c              	   C   sP   |  dd | jtdt| jddd | jtd| jt| jddd d S )	N./initz * QAPI Commands initializationzZ
#include "qapi/qmp/dispatch.h"

void %(c_prefix)sqmp_init_marshal(QmpCommandList *cmds);
FrJ   )c_prefixz
#include "qemu/osdep.h"
#include "%(prefix)sqapi-commands.h"
#include "%(prefix)sqapi-init-commands.h"

void %(c_prefix)sqmp_init_marshal(QmpCommandList *cmds)
{
    QTAILQ_INIT(cmds);

)r>   rS   )Z_add_modulerO   rN   r   r   Z_prefixrM   )rE   rQ   r   r   r    visit_beginZ  s    


z'QAPISchemaGenCommandVisitor.visit_begin)r   c                 C   s>   |  d  | jtd W d    n1 s00    Y  d S )NrR   r4   )_temp_modulerM   rN   r   )rE   r   r   r    	visit_endo  s    z%QAPISchemaGenCommandVisitor.visit_end)r   infor)   r6   r   r   genr7   r   r8   r9   r   r   c                 C   sx  |sd S |rp|| j | j vrp| j | j | t|j| j| j  | jt| W d    n1 sf0    Y  t|| j| jl | jt|||	|| | jt|| | jt	|||	|| j
| | j
r| jt| W d    n1 s0    Y  | d\ t|| j| j* | jt||||
|| W d    n1 sJ0    Y  W d    n1 sj0    Y  d S )NrR   )rC   rM   rN   r   r)   rO   r,   r!   r1   r5   rD   Z_gen_trace_eventsr3   rU   r<   )rE   r   rW   r)   r6   r   r   rX   r7   r   r8   r9   r   r   r   r    visit_commandu  s0    ..z)QAPISchemaGenCommandVisitor.visit_command)__name__
__module____qualname__strboolrA   rP   r   rT   rV   r   r   r   r   r   r   r   rY   __classcell__r   r   rF   r    r=   1  s"    r=   N)rQ   
output_dirr>   r"   r   c                 C   s"   t ||}| | || d S )N)r=   rI   write)rQ   r`   r>   r"   Zvisr   r   r    gen_commands  s    

rb   )#rB   typingr   r   r   r   commonr   r   rX   r	   r
   r   r   r   rQ   r   r   r   r   r   sourcer   r]   r^   r!   r+   r,   r.   r1   r3   r5   r<   r=   rb   r   r   r   r    <module>   sZ   W	Zl