Build fix: poll compatibility mode: zmalloc prototype changed

[lttng-tools.git] / src / common / notification.cpp

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

#include <lttng/notification/notification-internal.hpp>
#include <lttng/condition/condition-internal.hpp>
#include <lttng/condition/evaluation-internal.hpp>
#include <lttng/condition/condition.h>
#include <lttng/condition/evaluation.h>
#include <lttng/trigger/trigger-internal.hpp>
#include <common/payload.hpp>
#include <common/payload-view.hpp>

struct lttng_notification *lttng_notification_create(
		struct lttng_trigger *trigger,
		struct lttng_evaluation *evaluation)
{
	struct lttng_notification *notification = NULL;

	if (!trigger || !evaluation) {
		goto end;
	}

	notification = zmalloc<lttng_notification>();
	if (!notification) {
		goto end;
	}

	notification->trigger = trigger;
	notification->evaluation = evaluation;
end:
	return notification;
}

int lttng_notification_serialize(const struct lttng_notification *notification,
		struct lttng_payload *payload)
{
	int ret;
	size_t header_offset, size_before_payload;
	struct lttng_notification_comm notification_comm = { 0 };
	struct lttng_notification_comm *header;

	header_offset = payload->buffer.size;
	ret = lttng_dynamic_buffer_append(&payload->buffer, &notification_comm,
			sizeof(notification_comm));
	if (ret) {
		goto end;
	}

	size_before_payload = payload->buffer.size;
	ret = lttng_trigger_serialize(notification->trigger,
			payload);
	if (ret) {
		goto end;
	}

	ret = lttng_evaluation_serialize(notification->evaluation, payload);
	if (ret) {
		goto end;
	}

	/* Update payload size. */
	header = (typeof(header)) (payload->buffer.data + header_offset);
	header->length = (uint32_t) (payload->buffer.size - size_before_payload);
end:
	return ret;

}

ssize_t lttng_notification_create_from_payload(
		struct lttng_payload_view *src_view,
		struct lttng_notification **notification)
{
	ssize_t ret, notification_size = 0, trigger_size, evaluation_size;
	struct lttng_trigger *trigger = NULL;
	struct lttng_evaluation *evaluation = NULL;
	const struct lttng_notification_comm *notification_comm;
	const struct lttng_payload_view notification_comm_view =
			lttng_payload_view_from_view(
					src_view, 0, sizeof(*notification_comm));

	if (!src_view || !notification) {
		ret = -1;
		goto error;
	}

	if (!lttng_payload_view_is_valid(&notification_comm_view)) {
		/* Payload not large enough to contain the header. */
		ret = -1;
		goto error;
	}

	notification_comm = (typeof(notification_comm)) notification_comm_view.buffer.data;
	notification_size += sizeof(*notification_comm);
	{
		/* struct lttng_condition */
		struct lttng_payload_view condition_view =
				lttng_payload_view_from_view(src_view,
						notification_size, -1);

		trigger_size = lttng_trigger_create_from_payload(
				&condition_view, &trigger);
	}

	if (trigger_size < 0) {
		ret = trigger_size;
		goto error;
	}

	notification_size += trigger_size;

	{
		/* struct lttng_evaluation */
		struct lttng_payload_view evaluation_view =
				lttng_payload_view_from_view(src_view,
						notification_size, -1);

		evaluation_size = lttng_evaluation_create_from_payload(
				lttng_trigger_get_const_condition(trigger),
				&evaluation_view, &evaluation);
	}

	if (evaluation_size < 0) {
		ret = evaluation_size;
		goto error;
	}

	notification_size += evaluation_size;

	/* Unexpected size of inner-elements; the buffer is corrupted. */
	if ((ssize_t) notification_comm->length !=
			trigger_size + evaluation_size) {
		ret = -1;
		goto error;
	}

	*notification = lttng_notification_create(trigger, evaluation);
	if (!*notification) {
		ret = -1;
		goto error;
	}

	ret = notification_size;
	return ret;

error:
	lttng_trigger_destroy(trigger);
	lttng_evaluation_destroy(evaluation);
	return ret;
}

void lttng_notification_destroy(struct lttng_notification *notification)
{
	if (!notification) {
		return;
	}

	lttng_trigger_destroy(notification->trigger);
	lttng_evaluation_destroy(notification->evaluation);
	free(notification);
}

const struct lttng_condition *lttng_notification_get_condition(
		struct lttng_notification *notification)
{
	return notification ? lttng_trigger_get_const_condition(notification->trigger) : NULL;
}

const struct lttng_evaluation *lttng_notification_get_evaluation(
		struct lttng_notification *notification)
{
	return notification ? notification->evaluation : NULL;
}

const struct lttng_trigger *lttng_notification_get_trigger(
		struct lttng_notification *notification)
{
	return notification ? notification->trigger : NULL;
}