/*-
 * Copyright (c) 2025 Klara, Inc.
 *
 * SPDX-License-Identifier: BSD-2-Clause
 */

#include <errno.h>
#include <stdlib.h>
#include <string.h>

#include <atf-c.h>

ATF_TC_WITHOUT_HEAD(getenv_r_ok);
ATF_TC_BODY(getenv_r_ok, tc)
{
	const char *ident = atf_tc_get_ident(tc);
	char buf[256];

	ATF_REQUIRE_EQ(0, setenv("ATF_TC_IDENT", ident, 1));
	ATF_REQUIRE_EQ(0, getenv_r("ATF_TC_IDENT", buf, sizeof(buf)));
	ATF_REQUIRE_STREQ(ident, buf);
}

ATF_TC_WITHOUT_HEAD(getenv_r_einval);
ATF_TC_BODY(getenv_r_einval, tc)
{
	char buf[256];

	errno = 0;
	ATF_REQUIRE_EQ(-1, getenv_r(NULL, buf, sizeof(buf)));
	ATF_REQUIRE_EQ(EINVAL, errno);
	errno = 0;
	ATF_REQUIRE_EQ(-1, getenv_r("", buf, sizeof(buf)));
	ATF_REQUIRE_EQ(EINVAL, errno);
	errno = 0;
	ATF_REQUIRE_EQ(-1, getenv_r("A=B", buf, sizeof(buf)));
	ATF_REQUIRE_EQ(EINVAL, errno);
}

ATF_TC_WITHOUT_HEAD(getenv_r_enoent);
ATF_TC_BODY(getenv_r_enoent, tc)
{
	char buf[256];

	errno = 0;
	ATF_REQUIRE_EQ(-1, getenv_r("no such variable", buf, sizeof(buf)));
	ATF_REQUIRE_EQ(ENOENT, errno);
}

ATF_TC_WITHOUT_HEAD(getenv_r_erange);
ATF_TC_BODY(getenv_r_erange, tc)
{
	const char *ident = atf_tc_get_ident(tc);
	char buf[256];

	ATF_REQUIRE_EQ(0, setenv("ATF_TC_IDENT", ident, 1));
	errno = 0;
	ATF_REQUIRE_EQ(-1, getenv_r("ATF_TC_IDENT", buf, strlen(ident)));
	ATF_REQUIRE_EQ(ERANGE, errno);
}

ATF_TP_ADD_TCS(tp)
{
	ATF_TP_ADD_TC(tp, getenv_r_ok);
	ATF_TP_ADD_TC(tp, getenv_r_einval);
	ATF_TP_ADD_TC(tp, getenv_r_enoent);
	ATF_TP_ADD_TC(tp, getenv_r_erange);
	return (atf_no_error());
}