1
2
3
4
5
6
7
8
9
10
11
12
13
14#include "qemu/osdep.h"
15#include "qapi/error.h"
16#include "qapi/qmp/types.h"
17#include "qapi/qmp/dispatch.h"
18#include "qapi/qmp/json-parser.h"
19#include "qapi/qmp/qjson.h"
20#include "qapi-types.h"
21#include "qapi/qmp/qerror.h"
22
23static QDict *qmp_dispatch_check_obj(const QObject *request, Error **errp)
24{
25 const QDictEntry *ent;
26 const char *arg_name;
27 const QObject *arg_obj;
28 bool has_exec_key = false;
29 QDict *dict = NULL;
30
31 dict = qobject_to_qdict(request);
32 if (!dict) {
33 error_setg(errp, "QMP input must be a JSON object");
34 return NULL;
35 }
36
37 for (ent = qdict_first(dict); ent;
38 ent = qdict_next(dict, ent)) {
39 arg_name = qdict_entry_key(ent);
40 arg_obj = qdict_entry_value(ent);
41
42 if (!strcmp(arg_name, "execute")) {
43 if (qobject_type(arg_obj) != QTYPE_QSTRING) {
44 error_setg(errp,
45 "QMP input member 'execute' must be a string");
46 return NULL;
47 }
48 has_exec_key = true;
49 } else if (!strcmp(arg_name, "arguments")) {
50 if (qobject_type(arg_obj) != QTYPE_QDICT) {
51 error_setg(errp,
52 "QMP input member 'arguments' must be an object");
53 return NULL;
54 }
55 } else {
56 error_setg(errp, "QMP input member '%s' is unexpected",
57 arg_name);
58 return NULL;
59 }
60 }
61
62 if (!has_exec_key) {
63 error_setg(errp, "QMP input lacks member 'execute'");
64 return NULL;
65 }
66
67 return dict;
68}
69
70static QObject *do_qmp_dispatch(QmpCommandList *cmds, QObject *request,
71 Error **errp)
72{
73 Error *local_err = NULL;
74 const char *command;
75 QDict *args, *dict;
76 QmpCommand *cmd;
77 QObject *ret = NULL;
78
79 dict = qmp_dispatch_check_obj(request, errp);
80 if (!dict) {
81 return NULL;
82 }
83
84 command = qdict_get_str(dict, "execute");
85 cmd = qmp_find_command(cmds, command);
86 if (cmd == NULL) {
87 error_set(errp, ERROR_CLASS_COMMAND_NOT_FOUND,
88 "The command %s has not been found", command);
89 return NULL;
90 }
91 if (!cmd->enabled) {
92 error_setg(errp, "The command %s has been disabled for this instance",
93 command);
94 return NULL;
95 }
96
97 if (!qdict_haskey(dict, "arguments")) {
98 args = qdict_new();
99 } else {
100 args = qdict_get_qdict(dict, "arguments");
101 QINCREF(args);
102 }
103
104 cmd->fn(args, &ret, &local_err);
105 if (local_err) {
106 error_propagate(errp, local_err);
107 } else if (cmd->options & QCO_NO_SUCCESS_RESP) {
108 g_assert(!ret);
109 } else if (!ret) {
110 ret = QOBJECT(qdict_new());
111 }
112
113 QDECREF(args);
114
115 return ret;
116}
117
118QObject *qmp_build_error_object(Error *err)
119{
120 return qobject_from_jsonf("{ 'class': %s, 'desc': %s }",
121 QapiErrorClass_str(error_get_class(err)),
122 error_get_pretty(err));
123}
124
125QObject *qmp_dispatch(QmpCommandList *cmds, QObject *request)
126{
127 Error *err = NULL;
128 QObject *ret;
129 QDict *rsp;
130
131 ret = do_qmp_dispatch(cmds, request, &err);
132
133 rsp = qdict_new();
134 if (err) {
135 qdict_put_obj(rsp, "error", qmp_build_error_object(err));
136 error_free(err);
137 } else if (ret) {
138 qdict_put_obj(rsp, "return", ret);
139 } else {
140 QDECREF(rsp);
141 return NULL;
142 }
143
144 return QOBJECT(rsp);
145}
146