From 7868d181a3001c6df37c84345fd545e24baf1306 Mon Sep 17 00:00:00 2001 From: Simon McVittie Date: Thu, 22 Jun 2017 18:13:10 +0100 Subject: [PATCH] bus/containers: Each connection to a container holds a reference Signed-off-by: Simon McVittie --- bus/containers.c | 34 ++++++++++++++++++++++++++++++++-- 1 file changed, 32 insertions(+), 2 deletions(-) diff --git a/bus/containers.c b/bus/containers.c index 4a908e32..d1f22f7c 100644 --- a/bus/containers.c +++ b/bus/containers.c @@ -71,12 +71,23 @@ struct BusContainers dbus_uint64_t next_container_id; }; +/* Data slot on DBusConnection, holding BusContainerInstance */ +static dbus_int32_t contained_data_slot = -1; + BusContainers * bus_containers_new (void) { /* We allocate the hash table lazily, expecting that the common case will * be a connection where this feature is never used */ - BusContainers *self = dbus_new0 (BusContainers, 1); + BusContainers *self = NULL; + + /* One reference per BusContainers, unless we ran out of memory the first + * time we tried to allocate it, in which case it will be -1 when we + * free the BusContainers */ + if (!dbus_connection_allocate_data_slot (&contained_data_slot)) + goto oom; + + self = dbus_new0 (BusContainers, 1); if (self == NULL) goto oom; @@ -122,7 +133,15 @@ bus_containers_new (void) oom: if (self != NULL) - bus_containers_unref (self); + { + /* This will free the data slot too */ + bus_containers_unref (self); + } + else + { + if (contained_data_slot != -1) + dbus_connection_free_data_slot (&contained_data_slot); + } return NULL; } @@ -148,6 +167,9 @@ bus_containers_unref (BusContainers *self) _dbus_string_free (&self->address_template); dbus_free (self); + + if (contained_data_slot != -1) + dbus_connection_free_data_slot (&contained_data_slot); } } @@ -306,6 +328,14 @@ new_connection_cb (DBusServer *server, { BusContainerInstance *instance = data; + if (!dbus_connection_set_data (new_connection, contained_data_slot, + bus_container_instance_ref (instance), + (DBusFreeFunction) bus_container_instance_unref)) + { + bus_container_instance_unref (instance); + return; + } + if (!bus_context_add_incoming_connection (instance->context, new_connection)) return; -- 2.13.2