X-Git-Url: http://git.lttng.org./?a=blobdiff_plain;f=liblttng-ust%2Fevent-notifier-notification.c;h=559c7fe88e21a62d42382d3aa657a595aa680aa0;hb=7ee76145ae6a7efc03968571b9e4a22d46e92929;hp=6fce4357fe85fe5d1511495b675449b5e10797a6;hpb=d37ecb3fc622dee6f80f84c21f38d32eef407262;p=lttng-ust.git

diff --git a/liblttng-ust/event-notifier-notification.c b/liblttng-ust/event-notifier-notification.c
index 6fce4357..559c7fe8 100644
--- a/liblttng-ust/event-notifier-notification.c
+++ b/liblttng-ust/event-notifier-notification.c
@@ -1,34 +1,20 @@
 /*
- * event-notifier-notification.c
+ * SPDX-License-Identifier: LGPL-2.1-only
  *
  * Copyright (C) 2020 Francis Deslauriers <francis.deslauriers@efficios.com>
- *
- * This library is free software; you can redistribute it and/or
- * modify it under the terms of the GNU Lesser General Public
- * License as published by the Free Software Foundation; only
- * version 2.1 of the License.
- *
- * This library is distributed in the hope that it will be useful,
- * but WITHOUT ANY WARRANTY; without even the implied warranty of
- * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
- * Lesser General Public License for more details.
- *
- * You should have received a copy of the GNU Lesser General Public
- * License along with this library; if not, write to the Free Software
- * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  */
 
 #define _LGPL_SOURCE
 
 #include <assert.h>
-#include <byteswap.h>
 #include <errno.h>
-#include <lttng/ust-events.h>
+#include <lttng/ust-endian.h>
 #include <usterr-signal-safe.h>
 
+#include "ust-events-internal.h"
 #include "../libmsgpack/msgpack.h"
 #include "lttng-bytecode.h"
-#include "share.h"
+#include "ust-share.h"
 
 /*
  * We want this write to be atomic AND non-blocking, meaning that we
@@ -188,7 +174,7 @@ void capture_sequence(struct lttng_msgpack_writer *writer,
 	case atype_integer:
 		integer_type = &nested_type->u.integer;
 		break;
-	case atype_enum:
+	case atype_enum_nestable:
 		/* Treat enumeration as an integer. */
 		integer_type = &nested_type->u.enum_nestable.container_type->u.integer;
 		break;
@@ -280,18 +266,47 @@ void notification_append_empty_capture(
 	lttng_msgpack_write_nil(&notif->writer);
 }
 
+static void record_error(struct lttng_event_notifier *event_notifier)
+{
+	struct lttng_event_notifier_group *event_notifier_group =
+			event_notifier->group;
+	struct lttng_counter *error_counter;
+	size_t dimension_index[1];
+	int ret;
+
+	error_counter = CMM_LOAD_SHARED(event_notifier_group->error_counter);
+	/*
+	 * load-acquire paired with store-release orders creation of the
+	 * error counter and setting error_counter_len before the
+	 * error_counter is used.
+	 * Currently a full memory barrier is used, which could be
+	 * turned into acquire-release barriers.
+	 */
+	cmm_smp_mb();
+	/* This group may not have an error counter attached to it. */
+	if (!error_counter)
+		return;
+
+	dimension_index[0] = event_notifier->error_counter_index;
+	ret = event_notifier_group->error_counter->ops->counter_add(
+			error_counter->counter, dimension_index, 1);
+	if (ret)
+		WARN_ON_ONCE(1);
+}
+
 static
-void notification_send(struct lttng_event_notifier_notification *notif)
+void notification_send(struct lttng_event_notifier_notification *notif,
+		struct lttng_event_notifier *event_notifier)
 {
 	ssize_t ret;
 	size_t content_len;
 	int iovec_count = 1;
-	struct lttng_ust_event_notifier_notification ust_notif;
+	struct lttng_ust_event_notifier_notification ust_notif = {0};
 	struct iovec iov[2];
 
 	assert(notif);
 
-	ust_notif.token = notif->event_notifier_token;
+	ust_notif.token = event_notifier->user_token;
 
 	/*
 	 * Prepare sending the notification from multiple buffers using an
@@ -326,10 +341,11 @@ void notification_send(struct lttng_event_notifier_notification *notif)
 	ust_notif.capture_buf_size = content_len;
 
 	/* Send all the buffers. */
-	ret = patient_writev(notif->notification_fd, iov, iovec_count);
+	ret = ust_patient_writev(notif->notification_fd, iov, iovec_count);
 	if (ret == -1) {
 		if (errno == EAGAIN) {
-			DBG("Cannot send event notifier notification without blocking: %s",
+			record_error(event_notifier);
+			DBG("Cannot send event_notifier notification without blocking: %s",
 				strerror(errno));
 		} else {
 			DBG("Error to sending event notifier notification: %s",
@@ -339,7 +355,8 @@ void notification_send(struct lttng_event_notifier_notification *notif)
 	}
 }
 
-void lttng_event_notifier_notification_send(struct lttng_event_notifier *event_notifier,
+void lttng_event_notifier_notification_send(
+		struct lttng_event_notifier *event_notifier,
 		const char *stack_data)
 {
 	/*
@@ -375,5 +392,5 @@ void lttng_event_notifier_notification_send(struct lttng_event_notifier *event_n
 	 * Send the notification (including the capture buffer) to the
 	 * sessiond.
 	 */
-	notification_send(&notif);
+	notification_send(&notif, event_notifier);
 }