qapi/qmp-output-visitor.c

   1 /*
   2  * Core Definitions for QAPI/QMP Command Registry
   3  *
   4  * Copyright IBM, Corp. 2011
   5  *
   6  * Authors:
   7  *  Anthony Liguori   <aliguori@us.ibm.com>
   8  *
   9  * This work is licensed under the terms of the GNU LGPL, version 2.1 or later.
  10  * See the COPYING.LIB file in the top-level directory.
  11  *
  12  */
  13
  14 #include "qmp-output-visitor.h"
  15 #include "qemu-queue.h"
  16 #include "qemu-common.h"
  17 #include "qemu-objects.h"
  18
  19 typedef struct QStackEntry
  20 {
  21     QObject *value;
  22     QTAILQ_ENTRY(QStackEntry) node;
  23 } QStackEntry;
  24
  25 typedef QTAILQ_HEAD(QStack, QStackEntry) QStack;
  26
  27 struct QmpOutputVisitor
  28 {
  29     Visitor visitor;
  30     QStack stack;
  31 };
  32
  33 #define qmp_output_add(qov, name, value) qmp_output_add_obj(qov, name, QOBJECT(value))
  34 #define qmp_output_push(qov, value) qmp_output_push_obj(qov, QOBJECT(value))
  35
  36 static QmpOutputVisitor *to_qov(Visitor *v)
  37 {
  38     return container_of(v, QmpOutputVisitor, visitor);
  39 }
  40
  41 static void qmp_output_push_obj(QmpOutputVisitor *qov, QObject *value)
  42 {
  43     QStackEntry *e = qemu_mallocz(sizeof(*e));
  44
  45     e->value = value;
  46     QTAILQ_INSERT_HEAD(&qov->stack, e, node);
  47 }
  48
  49 static QObject *qmp_output_pop(QmpOutputVisitor *qov)
  50 {
  51     QStackEntry *e = QTAILQ_FIRST(&qov->stack);
  52     QObject *value;
  53     QTAILQ_REMOVE(&qov->stack, e, node);
  54     value = e->value;
  55     qemu_free(e);
  56     return value;
  57 }
  58
  59 static QObject *qmp_output_first(QmpOutputVisitor *qov)
  60 {
  61     QStackEntry *e = QTAILQ_LAST(&qov->stack, QStack);
  62     return e->value;
  63 }
  64
  65 static QObject *qmp_output_last(QmpOutputVisitor *qov)
  66 {
  67     QStackEntry *e = QTAILQ_FIRST(&qov->stack);
  68     return e->value;
  69 }
  70
  71 static void qmp_output_add_obj(QmpOutputVisitor *qov, const char *name, QObject *value)
  72 {
  73     QObject *cur;
  74
  75     if (QTAILQ_EMPTY(&qov->stack)) {
  76         qmp_output_push_obj(qov, value);
  77         return;
  78     }
  79
  80     cur = qmp_output_last(qov);
  81
  82     switch (qobject_type(cur)) {
  83     case QTYPE_QDICT:
  84         qdict_put_obj(qobject_to_qdict(cur), name, value);
  85         break;
  86     case QTYPE_QLIST:
  87         qlist_append_obj(qobject_to_qlist(cur), value);
  88         break;
  89     default:
  90         qobject_decref(qmp_output_pop(qov));
  91         qmp_output_push_obj(qov, value);
  92         break;
  93     }
  94 }
  95
  96 static void qmp_output_start_struct(Visitor *v, void **obj, const char *kind, const char *name, size_t unused, Error **errp)
  97 {
  98     QmpOutputVisitor *qov = to_qov(v);
  99     QDict *dict = qdict_new();
 100
 101     qmp_output_add(qov, name, dict);
 102     qmp_output_push(qov, dict);
 103 }
 104
 105 static void qmp_output_end_struct(Visitor *v, Error **errp)
 106 {
 107     QmpOutputVisitor *qov = to_qov(v);
 108     qmp_output_pop(qov);
 109 }
 110
 111 static void qmp_output_start_list(Visitor *v, const char *name, Error **errp)
 112 {
 113     QmpOutputVisitor *qov = to_qov(v);
 114     QList *list = qlist_new();
 115
 116     qmp_output_add(qov, name, list);
 117     qmp_output_push(qov, list);
 118 }
 119
 120 static GenericList *qmp_output_next_list(Visitor *v, GenericList **list, Error **errp)
 121 {
 122     GenericList *retval = *list;
 123     *list = retval->next;
 124     return retval;
 125 }
 126
 127 static void qmp_output_end_list(Visitor *v, Error **errp)
 128 {
 129     QmpOutputVisitor *qov = to_qov(v);
 130     qmp_output_pop(qov);
 131 }
 132
 133 static void qmp_output_type_int(Visitor *v, int64_t *obj, const char *name, Error **errp)
 134 {
 135     QmpOutputVisitor *qov = to_qov(v);
 136     qmp_output_add(qov, name, qint_from_int(*obj));
 137 }
 138
 139 static void qmp_output_type_bool(Visitor *v, bool *obj, const char *name, Error **errp)
 140 {
 141     QmpOutputVisitor *qov = to_qov(v);
 142     qmp_output_add(qov, name, qbool_from_int(*obj));
 143 }
 144
 145 static void qmp_output_type_str(Visitor *v, char **obj, const char *name, Error **errp)
 146 {
 147     QmpOutputVisitor *qov = to_qov(v);
 148     if (*obj) {
 149         qmp_output_add(qov, name, qstring_from_str(*obj));
 150     } else {
 151         qmp_output_add(qov, name, qstring_from_str(""));
 152     }
 153 }
 154
 155 static void qmp_output_type_number(Visitor *v, double *obj, const char *name, Error **errp)
 156 {
 157     QmpOutputVisitor *qov = to_qov(v);
 158     qmp_output_add(qov, name, qfloat_from_double(*obj));
 159 }
 160
 161 static void qmp_output_type_enum(Visitor *v, int *obj, const char *kind, const char *name, Error **errp)
 162 {
 163     int64_t value = *obj;
 164     qmp_output_type_int(v, &value, name, errp);
 165 }
 166
 167 QObject *qmp_output_get_qobject(QmpOutputVisitor *qov)
 168 {
 169     QObject *obj = qmp_output_first(qov);
 170     if (obj) {
 171         qobject_incref(obj);
 172     }
 173     return obj;
 174 }
 175
 176 Visitor *qmp_output_get_visitor(QmpOutputVisitor *v)
 177 {
 178     return &v->visitor;
 179 }
 180
 181 void qmp_output_visitor_cleanup(QmpOutputVisitor *v)
 182 {
 183     QStackEntry *e, *tmp;
 184
 185     QTAILQ_FOREACH_SAFE(e, &v->stack, node, tmp) {
 186         QTAILQ_REMOVE(&v->stack, e, node);
 187         if (e->value) {
 188             qobject_decref(e->value);
 189         }
 190         qemu_free(e);
 191     }
 192
 193     qemu_free(v);
 194 }
 195
 196 QmpOutputVisitor *qmp_output_visitor_new(void)
 197 {
 198     QmpOutputVisitor *v;
 199
 200     v = qemu_mallocz(sizeof(*v));
 201
 202     v->visitor.start_struct = qmp_output_start_struct;
 203     v->visitor.end_struct = qmp_output_end_struct;
 204     v->visitor.start_list = qmp_output_start_list;
 205     v->visitor.next_list = qmp_output_next_list;
 206     v->visitor.end_list = qmp_output_end_list;
 207     v->visitor.type_enum = qmp_output_type_enum;
 208     v->visitor.type_int = qmp_output_type_int;
 209     v->visitor.type_bool = qmp_output_type_bool;
 210     v->visitor.type_str = qmp_output_type_str;
 211     v->visitor.type_number = qmp_output_type_number;
 212
 213     QTAILQ_INIT(&v->stack);
 214
 215     return v;
 216 }