Fix: lttng: leak of userspace probe path on listing

[lttng-tools.git] / src / common / payload-view.c

/*
 * Copyright (C) 2020 Jérémie Galarneau <jeremie.galarneau@efficios.com>
 *
 * SPDX-License-Identifier: LGPL-2.1-only
 *
 */

#include <common/dynamic-array.h>
#include <common/buffer-view.h>
#include "payload-view.h"
#include "payload.h"
#include <stddef.h>

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) {
		.buffer = lttng_buffer_view_from_dynamic_buffer(
			&payload->buffer, offset, len),
		._fds = payload->_fds,
	};
}

LTTNG_HIDDEN
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) {
		.buffer = lttng_buffer_view_from_view(
			&view->buffer, offset, len),
		._fds = view->_fds,
		._iterator.p_fds_position = &view->_iterator.fds_position,
	};
}

LTTNG_HIDDEN
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) {
		.buffer = lttng_buffer_view_from_dynamic_buffer(
			buffer, offset, len)
	};
}

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 (struct lttng_payload_view) {
		.buffer = lttng_buffer_view_from_view(
			view, offset, len)
	};
}

LTTNG_HIDDEN
int lttng_payload_view_get_fd_count(struct lttng_payload_view *payload_view)
{
	int ret;
	size_t position;

	if (!payload_view) {
		ret = -1;
		goto end;
	}

	ret = lttng_dynamic_array_get_count(&payload_view->_fds);
	if (ret < 0) {
		goto end;
	}

	position = payload_view->_iterator.p_fds_position ?
			*payload_view->_iterator.p_fds_position :
			payload_view->_iterator.fds_position;
	ret = ret - (int) position;
end:
	return ret;
}

LTTNG_HIDDEN
int lttng_payload_view_pop_fd(struct lttng_payload_view *view)
{
	int ret = 0;
	size_t fd_count;
	size_t *pos;

	if (!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;

	if (*pos >= fd_count) {
		ret = -1;
		goto end;
	}

	ret = *((int *) lttng_dynamic_array_get_element(
			&view->_fds, *pos));
	(*pos)++;
end:
	return ret;
}