src/test/auditd_dispatch_test.c - master - githubmirror/audit-userspace

src/test/auditd_dispatch_test.c
155 строк3 KB
Steve Grubb
Fix inconsistent licenses
20 июл 2026, 22:59
20 июл 2026, 22:59674a75b
/*
 * auditd_dispatch_test.c - dispatcher protocol length tests
 * Copyright 2026 Red Hat Inc.
 * All Rights Reserved.
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2 of the License, or
 * (at your option) any later version.
 */

#include "config.h"
#include <assert.h>
#include <stdlib.h>
#include <string.h>
#include "auditd-dispatch.h"
#include "libdisp.h"

static event_t *queued_event;

/*
 * libdisp_active - make the dispatcher available to the unit under test
 *
 * Returns: Non-zero.
 */
int libdisp_active(void)
{
	return 1;
}

/*
 * libdisp_enqueue - capture a dispatched event for assertions
 * @e: event transferred by the dispatcher
 *
 * Returns: Success.
 */
int libdisp_enqueue(event_t *e)
{
	queued_event = e;
	return 0;
}

/*
 * libdisp_init - provide the unused dispatcher setup symbol for linking
 * @config: dispatcher configuration
 *
 * Returns: Success.
 */
int libdisp_init(const struct daemon_conf *config)
{
	(void)config;
	return 0;
}

/*
 * libdisp_shutdown - provide the unused dispatcher shutdown symbol for linking
 *
 * Returns: None.
 */
void libdisp_shutdown(void)
{
}

/*
 * libdisp_reconfigure - provide the unused reconfigure symbol for linking
 * @config: dispatcher configuration
 *
 * Returns: None.
 */
void libdisp_reconfigure(const struct daemon_conf *config)
{
	(void)config;
}

/*
 * free_queued_event - release the event captured by the dispatcher stub
 *
 * Returns: None.
 */
static void free_queued_event(void)
{
	free(queued_event);
	queued_event = NULL;
}

/*
 * test_netlink_payload_length - verify netlink headers are not copied as data
 *
 * Returns: None.
 */
static void test_netlink_payload_length(void)
{
	struct audit_reply rep;

	memset(&rep, 0, sizeof(rep));
	memset(rep.msg.data, 'a', sizeof(rep.msg.data));
	rep.type = AUDIT_SYSCALL;
	rep.nlh = &rep.msg.nlh;
	rep.msg.nlh.nlmsg_len = NLMSG_LENGTH(sizeof(rep.msg.data));

	assert(dispatch_event(&rep, AUDISP_PROTOCOL_VER) == 0);
	assert(queued_event != NULL);
	assert(queued_event->hdr.size == sizeof(queued_event->data));
	assert(memcmp(queued_event->data, rep.msg.data,
		      sizeof(queued_event->data)) == 0);
	free_queued_event();
}

/*
 * test_invalid_netlink_length - reject malformed embedded netlink lengths
 *
 * Returns: None.
 */
static void test_invalid_netlink_length(void)
{
	struct audit_reply rep;

	memset(&rep, 0, sizeof(rep));
	rep.nlh = &rep.msg.nlh;
	rep.msg.nlh.nlmsg_len = NLMSG_HDRLEN - 1;
	assert(dispatch_event(&rep, AUDISP_PROTOCOL_VER) == -1);
	assert(queued_event == NULL);

	rep.msg.nlh.nlmsg_len = NLMSG_LENGTH(sizeof(rep.msg.data)) + 1;
	assert(dispatch_event(&rep, AUDISP_PROTOCOL_VER) == -1);
	assert(queued_event == NULL);
}

/*
 * test_synthetic_payload_length - retain auditd's local V1 length convention
 *
 * Returns: None.
 */
static void test_synthetic_payload_length(void)
{
	struct audit_reply rep;

	memset(&rep, 0, sizeof(rep));
	rep.msg.nlh.nlmsg_len = 2;
	memcpy(rep.msg.data, "ok", 2);

	assert(dispatch_event(&rep, AUDISP_PROTOCOL_VER) == 0);
	assert(queued_event != NULL);
	assert(queued_event->hdr.size == 2);
	assert(memcmp(queued_event->data, "ok", 2) == 0);
	free_queued_event();
}

int main(void)
{
	test_netlink_payload_length();
	test_invalid_netlink_length();
	test_synthetic_payload_length();
	return 0;
}