/*-
 * Copyright (c) 2026 Dag-Erling Smørgrav
 *
 * SPDX-License-Identifier: BSD-2-Clause
 */

#include <sys/param.h>
#include <sys/wait.h>

#include <libutil.h>
#include <paths.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>

#include <atf-c.h>

#define INVALID_PREFIX "/dev/null/"

#define X31 "/xxxxxxxxxxxxxxxxxxxxxxxxxxxxxx"
#define X32 "/xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx"
#define X64 X32 X32
#define X128 X64 X64
#define X256 X128 X128
#define X512 X256 X256
#define X1024 X512 X512
#define X1023 X512 X256 X128 X64 X32 X31

static struct {
	const char *in, *out;
} tests[] = {
	{ "", _PATH_LOCALBASE },
	{ "/", "/" },
	{ "//", "/" },
	{ "///", "/" },
	{ "foo", INVALID_PREFIX },
	{ "/foo", "/foo" },
	{ "//foo", "/foo" },
	{ "/foo/", "/foo" },
	{ "/foo//", "/foo" },
	{ "//foo//", "/foo" },
	{ "/foo/bar", "/foo/bar" },
	{ "/foo/bar/", "/foo/bar" },
	{ "/foo//bar", "/foo/bar" },
	{ "/foo//bar/", "/foo/bar" },
	{ "//foo/bar", "/foo/bar" },
	{ "//foo/bar/", "/foo/bar" },
	{ "//foo//bar", "/foo/bar" },
	{ "//foo//bar/", "/foo/bar" },
	{ _PATH_LOCALBASE, _PATH_LOCALBASE },
	{ X31, X31 },
	{ X32, X32 },
	{ X64, X64 },
	{ X128, X128 },
	{ X256, X256 },
	{ X512, X512 },
	{ X1023, X1023 },
	{ X1024, INVALID_PREFIX },
	{ 0 }
};
_Static_assert(sizeof(X1023) == MAXPATHLEN, "Unexpected MAXPATHLEN value");

ATF_TC(environment);
ATF_TC_HEAD(environment, tc)
{
	atf_tc_set_md_var(tc, "descr", "Tests getting the local base from the "
	    "environment and path normalization");
}

ATF_TC_BODY(environment, tc)
{
	pid_t pid;
	int i, wstatus;

	for (i = 0; tests[i].in != NULL; i++) {
		ATF_REQUIRE((pid = fork()) >= 0);
		if (pid == 0) {
			ATF_REQUIRE_EQ(setenv("LOCALBASE", tests[i].in, 1), 0);
			ATF_REQUIRE_STREQ(getlocalbase(), tests[i].out);
			_exit(0);
		}
		ATF_REQUIRE_EQ(waitpid(0, &wstatus, 0), pid);
		ATF_CHECK(WIFEXITED(wstatus));
		ATF_CHECK_EQ(WEXITSTATUS(wstatus), 0);
	}
}

ATF_TP_ADD_TCS(tp)
{
	ATF_TP_ADD_TC(tp, environment);
	return (atf_no_error());
}