X-Git-Url: https://git.lttng.org/?p=lttng-tools.git;a=blobdiff_plain;f=src%2Fcommon%2Fpayload-view.c;h=247ecec0ae46da86d29ea6b5587a43aa460e0d03;hp=fdfef6150e4a00d98b6cba9e23c9ea8471d8c28f;hb=3e6e0df2f8f9f23d252c2508b6d741916dfcc4b3;hpb=9e620ea7b016fc3fd4e08b5d094ffa40b36b50f6

diff --git a/src/common/payload-view.c b/src/common/payload-view.c
index fdfef6150..247ecec0a 100644
--- a/src/common/payload-view.c
+++ b/src/common/payload-view.c
@@ -5,21 +5,28 @@
  *
  */
 
+#include <common/dynamic-array.h>
 #include <common/buffer-view.h>
 #include "payload-view.h"
 #include "payload.h"
 #include <stddef.h>
 
+LTTNG_HIDDEN
+bool lttng_payload_view_is_valid(const struct lttng_payload_view *view)
+{
+	return view && lttng_buffer_view_is_valid(&view->buffer);
+}
+
 LTTNG_HIDDEN
 struct lttng_payload_view lttng_payload_view_from_payload(
 		const struct lttng_payload *payload, size_t offset,
 		ptrdiff_t len)
 {
-	return (struct lttng_payload_view) {
+	return payload ? (struct lttng_payload_view) {
 		.buffer = lttng_buffer_view_from_dynamic_buffer(
 			&payload->buffer, offset, len),
-		._fds = payload->_fds,
-	};
+		._fd_handles = payload->_fd_handles,
+		} : (struct lttng_payload_view) {};
 }
 
 LTTNG_HIDDEN
@@ -27,12 +34,13 @@ struct lttng_payload_view lttng_payload_view_from_view(
 		struct lttng_payload_view *view, size_t offset,
 		ptrdiff_t len)
 {
-	return (struct lttng_payload_view) {
+	return view ? (struct lttng_payload_view) {
 		.buffer = lttng_buffer_view_from_view(
-			&view->buffer, offset, len),
-		._fds = view->_fds,
-		._iterator.p_fds_position = &view->_iterator.fds_position,
-	};
+				&view->buffer, offset, len),
+		._fd_handles = view->_fd_handles,
+		._iterator.p_fd_handles_position = view->_iterator.p_fd_handles_position ?:
+				&view->_iterator.fd_handles_position,
+		} : (struct lttng_payload_view) {};
 }
 
 LTTNG_HIDDEN
@@ -40,36 +48,82 @@ struct lttng_payload_view lttng_payload_view_from_dynamic_buffer(
 		const struct lttng_dynamic_buffer *buffer, size_t offset,
 		ptrdiff_t len)
 {
-	return (struct lttng_payload_view) {
+	return buffer ? (struct lttng_payload_view) {
 		.buffer = lttng_buffer_view_from_dynamic_buffer(
 			buffer, offset, len)
+		} : (struct lttng_payload_view) {};
+}
+
+LTTNG_HIDDEN
+struct lttng_payload_view lttng_payload_view_from_buffer_view(
+		const struct lttng_buffer_view *view, size_t offset,
+		ptrdiff_t len)
+{
+	return view ? (struct lttng_payload_view) {
+		.buffer = lttng_buffer_view_from_view(
+			view, offset, len)
+		} : (struct lttng_payload_view) {};
+}
+
+LTTNG_HIDDEN
+struct lttng_payload_view lttng_payload_view_init_from_buffer(
+	const char *src, size_t offset, ptrdiff_t len)
+{
+	return (struct lttng_payload_view) {
+		.buffer = lttng_buffer_view_init(
+			src, offset, len)
 	};
 }
 
 LTTNG_HIDDEN
-int lttng_payload_view_pop_fd(struct lttng_payload_view *view)
+int lttng_payload_view_get_fd_handle_count(
+		const struct lttng_payload_view *payload_view)
 {
-	int ret = 0;
-	size_t fd_count;
-	size_t *pos;
+	int ret;
+	size_t position;
 
-	if (!view) {
+	if (!payload_view) {
 		ret = -1;
 		goto end;
 	}
 
-	fd_count = lttng_dynamic_array_get_count(&view->_fds);
-	pos = view->_iterator.p_fds_position ? view->_iterator.p_fds_position :
-		&view->_iterator.fds_position;
+	ret = lttng_dynamic_pointer_array_get_count(&payload_view->_fd_handles);
+	if (ret < 0) {
+		goto end;
+	}
+
+	position = payload_view->_iterator.p_fd_handles_position ?
+			*payload_view->_iterator.p_fd_handles_position :
+			payload_view->_iterator.fd_handles_position;
+	ret = ret - (int) position;
+end:
+	return ret;
+}
 
-	if (*pos >= fd_count) {
-		ret = -1;
+LTTNG_HIDDEN
+struct fd_handle *lttng_payload_view_pop_fd_handle(
+		struct lttng_payload_view *view)
+{
+	struct fd_handle *handle = NULL;
+	size_t fd_handle_count;
+	size_t *pos;
+
+	if (!view) {
 		goto end;
 	}
 
-	ret = *((int *) lttng_dynamic_array_get_element(
-			&view->_fds, *pos));
+	fd_handle_count = lttng_payload_view_get_fd_handle_count(view);
+	if (fd_handle_count == 0) {
+		goto end;
+	}
+
+	pos = view->_iterator.p_fd_handles_position ?
+			view->_iterator.p_fd_handles_position :
+			&view->_iterator.fd_handles_position;
+	handle = lttng_dynamic_pointer_array_get_pointer(&view->_fd_handles,
+			*pos);
 	(*pos)++;
+	fd_handle_get(handle);
 end:
-	return ret;
+	return handle;
 }