Prechádzať zdrojové kódy

qapi-commands.py: Don't call the output marshal on error

Today we generate something like this:

    int qmp_marshal_input_query_foo(...)

        ...

        retval = qmp_query_foo(errp);
        qmp_marshal_output_query_foo(retval, ret, errp);

        ...

However, if qmp_query_foo() fails 'retval' will probably be NULL,
which can cause a segfault as not all visitors check if 'retval'
is valid.

This commit fixes that by changing the code generator to only
call the output marshal if qmp_query_foo() succeeds, like this:

    retval = qmp_query_foo(errp);
    if (!error_is_set(errp)) {
        qmp_marshal_output_query_foo(retval, ret, errp);
    }

Signed-off-by: Luiz Capitulino <lcapitulino@redhat.com>
Luiz Capitulino 14 rokov pred
rodič
commit
694a099a54
1 zmenil súbory, kde vykonal 3 pridanie a 1 odobranie
  1. 3 1
      scripts/qapi-commands.py

+ 3 - 1
scripts/qapi-commands.py

@@ -62,7 +62,9 @@ def gen_sync_call(name, args, ret_type, indent=0):
                 name=c_var(name), args=arglist, retval=retval).rstrip()
                 name=c_var(name), args=arglist, retval=retval).rstrip()
     if ret_type:
     if ret_type:
         ret += "\n" + mcgen(''''
         ret += "\n" + mcgen(''''
-%(marshal_output_call)s
+if (!error_is_set(errp)) {
+    %(marshal_output_call)s
+}
 ''',
 ''',
                             marshal_output_call=gen_marshal_output_call(name, ret_type)).rstrip()
                             marshal_output_call=gen_marshal_output_call(name, ret_type)).rstrip()
     pop_indent(indent)
     pop_indent(indent)