audisp/plugins/zos-remote/test-zos-remote-queue.c - master - githubmirror/audit-userspace

audisp/plugins/zos-remote/test-zos-remote-queue.c

100 строк2 KB

Steve Grubb

zos-remote: wake submission worker for reload and shutdown

14 июл 2026, 23:02

14 июл 2026, 23:02957a7ed

/*
 * test-zos-remote-queue.c - ownership tests for z/OS remote queue drops
 * Copyright 2026 Red Hat Inc.
 * All Rights Reserved.
 *
 * This software may be freely redistributed and/or modified under the
 * terms of the GNU General Public License as published by the Free
 * Software Foundation; either version 2, or (at your option) any
 * later version.
 */

#include "config.h"
#include <assert.h>
#include <stdint.h>
#include <stdarg.h>
#include "zos-remote-log.h"

static BerElement *freed[2];
static unsigned int freed_count;

static void test_log(const char *fmt, ...)
{
	va_list ap;

	(void)fmt;
	va_start(ap, fmt);
	va_end(ap);
}

static void test_ber_free(BerElement *ber, int freebuf)
{
	(void)freebuf;
	if (ber == NULL)
		return;
	assert(freed_count < 2);
	freed[freed_count++] = ber;
}

#define log_err test_log
#define ber_free test_ber_free
#include "zos-remote-queue.c"
#undef ber_free
#undef log_err

static void test_full_queue_frees_dropped_event(void)
{
	BerElement *first = (BerElement *)(uintptr_t)1;
	BerElement *second = (BerElement *)(uintptr_t)2;

	freed_count = 0;
	assert(init_queue(1) == 0);
	enqueue(first);
	enqueue(second);
	assert(freed_count == 1);
	assert(freed[0] == second);
	destroy_queue();
	assert(freed_count == 2);
	assert(freed[1] == first);
}

struct dequeue_wait {
	const volatile sig_atomic_t *stop;
	const volatile sig_atomic_t *hup;
	BerElement *result;
};

static void *wait_for_dequeue(void *arg)
{
	struct dequeue_wait *wait = arg;

	wait->result = dequeue(wait->stop, wait->hup);
	return NULL;
}

static void test_nudge_wakes_reload_waiter(void)
{
	pthread_t thread;
	volatile sig_atomic_t stop = 0;
	volatile sig_atomic_t hup = 0;
	struct dequeue_wait wait = {
		.stop = &stop,
		.hup = &hup,
		.result = (BerElement *)(uintptr_t)1,
	};

	assert(init_queue(1) == 0);
	assert(pthread_create(&thread, NULL, wait_for_dequeue, &wait) == 0);
	hup = 1;
	nudge_queue();
	assert(pthread_join(thread, NULL) == 0);
	assert(wait.result == NULL);
	destroy_queue();
}

int main(void)
{
	test_full_queue_frees_dropped_event();
	test_nudge_wakes_reload_waiter();
	return 0;
}