/*
 * This file and its contents are supplied under the terms of the
 * Common Development and Distribution License ("CDDL"), version 1.0.
 * You may only use this file in accordance with the terms of version
 * 1.0 of the CDDL.
 *
 * A full copy of the text of the CDDL should have accompanied this
 * source.  A copy of the CDDL is also available via the Internet at
 * http://www.illumos.org/license/CDDL.
 */

/*
 * Copyright 2025 Hans Rosenfeld
 */

/*
 * Basic dprintf test. Print something into a pipe, and verify that we can
 * read it back.
 */

#include <sys/types.h>
#include <sys/wait.h>
#include <sys/fork.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <err.h>

static const char hello[] = "Hello ";
static const char world[] = "World!";

static void
check_len(const char *ident, ssize_t len, ssize_t exp_len)
{
	if (len == -1)
		err(EXIT_FAILURE, "%s", ident);
	if (len != exp_len)
		errx(EXIT_FAILURE, "unexpected length from %s: %zd "
		    "(expected %zd)", ident, len, exp_len);
}

static void
check_buf(const char *buf, const char *exp, int len)
{
	if (strncmp(buf, exp, len) != 0)
		errx(EXIT_FAILURE, "unexpected buffer contents:\n"
		    "%s\nexpected:\n%s", buf, exp);
}

int
main(int argc, char **argv)
{
	int ret = -1;
	char buf[40] = { 0 };
	int fd[2];
	ssize_t len;
	pid_t pid;

	if (pipe(fd) == -1)
		err(EXIT_FAILURE, "pipe(fd)");

	pid = forkx(FORK_NOSIGCHLD | FORK_WAITPID);

	switch (pid) {
	case -1:
		err(EXIT_FAILURE, "fork()");

	case 0:
		(void) close(fd[0]);

		len = dprintf(fd[1], "%s", hello);
		check_len("dprintf(hello)", len, strlen(hello));

		len = dprintf(fd[1], "%s\n", world);
		check_len("dprintf(world)", len, strlen(world) + 1);

		len = dprintf(fd[1], "%sagain, %s\n", hello, world);
		check_len("dprintf(hello, world)", len,
		    strlen(hello) + strlen(world) + 8);

		return (0);

	default:
		(void) close(fd[1]);

		if (waitpid(pid, &ret, 0) != pid)
			err(EXIT_FAILURE, "waitpid()");

		if (ret != 0)
			errx(EXIT_FAILURE, "dprintf tests failed");

		len = read(fd[0], buf, sizeof (buf));
		check_len("read()", len,
		    2 * (strlen(hello) + strlen(world) + 1) + 7);
		check_buf(buf, "Hello World!\nHello again, World!\n", len);
	}

	(void) printf("dprintf tests passed\n");
	return (0);
}