1 files changed, 10 insertions, 22 deletions

diff --git a/server/sbus/sssd_dbus_connection.c b/server/sbus/sssd_dbus_connection.c
index 186df3563..d02dc6c85 100644
--- a/server/sbus/sssd_dbus_connection.c
+++ b/server/sbus/sssd_dbus_connection.c
@@ -483,7 +483,7 @@ DBusHandlerResult sbus_message_handler(DBusConnection *conn,
     const char *method;
     const char *path;
     const char *msg_interface;
-    struct sbus_message_ctx *reply = NULL;
+    DBusMessage *reply = NULL;
     int i, ret;
     int found;
 
@@ -503,22 +503,14 @@ DBusHandlerResult sbus_message_handler(DBusConnection *conn,
     if (strcmp(path, ctx->method_ctx->path) != 0)
         return DBUS_HANDLER_RESULT_NOT_YET_HANDLED;
 
-    reply = talloc_zero(ctx, struct sbus_message_ctx);
-    if (!reply) return DBUS_HANDLER_RESULT_NOT_YET_HANDLED;
-    reply->mh_ctx = ctx;
-    reply->reply_message = NULL;
-
     /* Validate the method interface */
     if (strcmp(msg_interface, ctx->method_ctx->interface) == 0) {
         found = 0;
         for (i = 0; ctx->method_ctx->methods[i].method != NULL; i++) {
             if (strcmp(method, ctx->method_ctx->methods[i].method) == 0) {
                 found = 1;
-                ret = ctx->method_ctx->methods[i].fn(message, reply);
-                if (ret != EOK) {
-                    talloc_free(reply);
-                    return DBUS_HANDLER_RESULT_NOT_YET_HANDLED;
-                }
+                ret = ctx->method_ctx->methods[i].fn(message, ctx, &reply);
+                if (ret != EOK) return DBUS_HANDLER_RESULT_NOT_YET_HANDLED;
                 break;
             }
         }
@@ -526,7 +518,7 @@ DBusHandlerResult sbus_message_handler(DBusConnection *conn,
         if (!found) {
             /* Reply DBUS_ERROR_UNKNOWN_METHOD */
             DEBUG(1, ("No matching method found for %s.\n", method));
-            reply->reply_message = dbus_message_new_error(message, DBUS_ERROR_UNKNOWN_METHOD, NULL);
+            reply = dbus_message_new_error(message, DBUS_ERROR_UNKNOWN_METHOD, NULL);
         }
     }
     else {
@@ -540,24 +532,20 @@ DBusHandlerResult sbus_message_handler(DBusConnection *conn,
                 /* If we have been asked for introspection data and we have
                  * an introspection function registered, user that.
                  */
-                ret = ctx->method_ctx->introspect_fn(message, reply);
-                if (ret != EOK) {
-                    talloc_free(reply);
-                    return DBUS_HANDLER_RESULT_NOT_YET_HANDLED;
-                }
+                ret = ctx->method_ctx->introspect_fn(message, ctx, &reply);
+                if (ret != EOK) return DBUS_HANDLER_RESULT_NOT_YET_HANDLED;
             }
         }
         else
             return DBUS_HANDLER_RESULT_NOT_YET_HANDLED;
     }
 
-    DEBUG(5, ("Method %s complete. Reply was %ssent.\n", method, reply->reply_message?"":"not "));
+    DEBUG(5, ("Method %s complete. Reply was %ssent.\n", method, reply?"":"not "));
 
-    if (reply->reply_message) {
-        dbus_connection_send(conn, reply->reply_message, NULL);
-        dbus_message_unref(reply->reply_message);
+    if (reply) {
+        dbus_connection_send(conn, reply, NULL);
+        dbus_message_unref(reply);
     }
-    talloc_free(reply);
 
     return DBUS_HANDLER_RESULT_HANDLED;
 }