X-Git-Url: http://git.efficios.com/?a=blobdiff_plain;f=liblttng-ust%2Flttng-ust-abi.c;h=5063ec6c7c1b14d981ce8fed0ea4d222631cd6cf;hb=e9fe6aad66010614f987e917866837706d19053b;hp=aad64a6dd326dbd13b01332c07d0bde8326b837c;hpb=33f43caa08bc332900bc8671a156912a09e88aa2;p=deliverable%2Flttng-ust.git

diff --git a/liblttng-ust/lttng-ust-abi.c b/liblttng-ust/lttng-ust-abi.c
index aad64a6d..5063ec6c 100644
--- a/liblttng-ust/lttng-ust-abi.c
+++ b/liblttng-ust/lttng-ust-abi.c
@@ -346,10 +346,10 @@ long lttng_abi_tracer_version(int objd,
 }
 
 static
-int lttng_abi_event_notifier_send_fd(void *owner, int event_notifier_notif_fd)
+int lttng_abi_event_notifier_send_fd(void *owner, int *event_notifier_notif_fd)
 {
 	struct lttng_event_notifier_group *event_notifier_group;
-	int event_notifier_group_objd, ret, fd_flag, close_ret;
+	int event_notifier_group_objd, ret, fd_flag;
 
 	event_notifier_group = lttng_event_notifier_group_create();
 	if (!event_notifier_group)
@@ -358,11 +358,11 @@ int lttng_abi_event_notifier_send_fd(void *owner, int event_notifier_notif_fd)
 	/*
 	 * Set this file descriptor as NON-BLOCKING.
 	 */
-	fd_flag = fcntl(event_notifier_notif_fd, F_GETFL);
+	fd_flag = fcntl(*event_notifier_notif_fd, F_GETFL);
 
 	fd_flag |= O_NONBLOCK;
 
-	ret = fcntl(event_notifier_notif_fd, F_SETFL, fd_flag);
+	ret = fcntl(*event_notifier_notif_fd, F_SETFL, fd_flag);
 	if (ret) {
 		ret = -errno;
 		goto fd_error;
@@ -377,18 +377,15 @@ int lttng_abi_event_notifier_send_fd(void *owner, int event_notifier_notif_fd)
 
 	event_notifier_group->objd = event_notifier_group_objd;
 	event_notifier_group->owner = owner;
-	event_notifier_group->notification_fd = event_notifier_notif_fd;
+	event_notifier_group->notification_fd = *event_notifier_notif_fd;
+	/* Object descriptor takes ownership of notification fd. */
+	*event_notifier_notif_fd = -1;
 
 	return event_notifier_group_objd;
 
 objd_error:
 	lttng_event_notifier_group_destroy(event_notifier_group);
 fd_error:
-	close_ret = close(event_notifier_notif_fd);
-	if (close_ret) {
-		PERROR("close");
-	}
-
 	return ret;
 }
 
@@ -443,7 +440,7 @@ long lttng_cmd(int objd, unsigned int cmd, unsigned long arg,
 		return 0;
 	case LTTNG_UST_EVENT_NOTIFIER_GROUP_CREATE:
 		return lttng_abi_event_notifier_send_fd(owner,
-			uargs->event_notifier_handle.event_notifier_notif_fd);
+			&uargs->event_notifier_handle.event_notifier_notif_fd);
 	default:
 		return -EINVAL;
 	}
@@ -453,6 +450,7 @@ static const struct lttng_ust_objd_ops lttng_ops = {
 	.cmd = lttng_cmd,
 };
 
+static
 int lttng_abi_map_channel(int session_objd,
 		struct lttng_ust_channel *ust_chan,
 		union ust_args *uargs,
@@ -497,6 +495,10 @@ int lttng_abi_map_channel(int session_objd,
 		goto handle_error;
 	}
 
+	/* Ownership of chan_data and wakeup_fd taken by channel handle. */
+	uargs->channel.chan_data = NULL;
+	uargs->channel.wakeup_fd = -1;
+
 	chan = shmp(channel_handle, channel_handle->chan);
 	assert(chan);
 	chan->handle = channel_handle;
@@ -580,24 +582,9 @@ alloc_error:
 	channel_destroy(chan, channel_handle, 0);
 	return ret;
 
-	/*
-	 * error path before channel creation (owning chan_data and
-	 * wakeup_fd).
-	 */
 handle_error:
 active:
 invalid:
-	{
-		int close_ret;
-
-		lttng_ust_lock_fd_tracker();
-		close_ret = close(wakeup_fd);
-		lttng_ust_unlock_fd_tracker();
-		if (close_ret) {
-			PERROR("close");
-		}
-	}
-	free(chan_data);
 	return ret;
 }
 
@@ -728,7 +715,7 @@ long lttng_event_notifier_enabler_cmd(int objd, unsigned int cmd, unsigned long
 			(struct lttng_ust_bytecode_node *) arg);
 	case LTTNG_UST_EXCLUSION:
 		return lttng_event_notifier_enabler_attach_exclusion(event_notifier_enabler,
-			(struct lttng_ust_excluder_node *) arg);
+			(struct lttng_ust_excluder_node **) arg);
 	case LTTNG_UST_CAPTURE:
 		return lttng_event_notifier_enabler_attach_capture_bytecode(
 			event_notifier_enabler,
@@ -816,10 +803,10 @@ int lttng_ust_event_notifier_group_create_error_counter(int event_notifier_group
 		return -EINVAL;
 
 	switch (error_counter_conf->bitness) {
-	case LTTNG_UST_COUNTER_BITNESS_64BITS:
+	case LTTNG_UST_COUNTER_BITNESS_64:
 		counter_transport_name = "counter-per-cpu-64-modular";
 		break;
-	case LTTNG_UST_COUNTER_BITNESS_32BITS:
+	case LTTNG_UST_COUNTER_BITNESS_32:
 		counter_transport_name = "counter-per-cpu-32-modular";
 		break;
 	default:
@@ -1117,6 +1104,9 @@ int lttng_abi_map_stream(int channel_objd, struct lttng_ust_stream *info,
 		info->stream_nr, info->len);
 	if (ret)
 		goto error_add_stream;
+	/* Take ownership of shm_fd and wakeup_fd. */
+	uargs->stream.shm_fd = -1;
+	uargs->stream.wakeup_fd = -1;
 
 	return 0;
 
@@ -1309,7 +1299,7 @@ long lttng_event_enabler_cmd(int objd, unsigned int cmd, unsigned long arg,
 	case LTTNG_UST_EXCLUSION:
 	{
 		return lttng_event_enabler_attach_exclusion(enabler,
-				(struct lttng_ust_excluder_node *) arg);
+				(struct lttng_ust_excluder_node **) arg);
 	}
 	default:
 		return -EINVAL;