/*
 * 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 Oxide Computer Company
 */

/*
 * Test various aspects of wcslcat.
 */

#include <wchar.h>
#include <err.h>
#include <stdlib.h>
#include <stdbool.h>
#include <stdio.h>
#include <sys/sysmacros.h>

#define	WCSLCAT_BUFLEN	16

typedef struct wcslcat_test {
	const char *wt_desc;
	wchar_t *wt_src;
	size_t wt_rval;
	size_t wt_dstlen;
	wchar_t wt_buf[WCSLCAT_BUFLEN];
	wchar_t wt_res[WCSLCAT_BUFLEN];
} wcslcat_test_t;

static const wcslcat_test_t wcslcat_tests[] = { {
	.wt_desc = "Zero-sized Destination Buffer (1)",
	.wt_src = L"Hello, World!",
	.wt_rval = 13,
	.wt_dstlen = 0,
	.wt_buf = { L'-', L'-', L'-', L'-', L'-', L'-', L'-', L'-',
		    L'-', L'-', L'-', L'-', L'-', L'-', L'-', L'-' },
	.wt_res = { L'-', L'-', L'-', L'-', L'-', L'-', L'-', L'-',
		    L'-', L'-', L'-', L'-', L'-', L'-', L'-', L'-' }
}, {
	.wt_desc = "Zero-sized Destination Buffer (2)",
	.wt_src = L"光",
	.wt_rval = 1,
	.wt_dstlen = 0,
	.wt_buf = { L'-', L'-', L'-', L'-', L'-', L'-', L'-', L'-',
		    L'-', L'-', L'-', L'-', L'-', L'-', L'-', L'-' },
	.wt_res = { L'-', L'-', L'-', L'-', L'-', L'-', L'-', L'-',
		    L'-', L'-', L'-', L'-', L'-', L'-', L'-', L'-' }
}, {
	.wt_desc = "Append at start, empty src",
	.wt_src = L"",
	.wt_rval = 0,
	.wt_dstlen = WCSLCAT_BUFLEN,
	.wt_buf = { L'\0', L'-', L'-', L'-', L'-', L'-', L'-', L'-',
		    L'-', L'-', L'-', L'-', L'-', L'-', L'-', L'-' },
	.wt_res = { L'\0', L'-', L'-', L'-', L'-', L'-', L'-', L'-',
		    L'-', L'-', L'-', L'-', L'-', L'-', L'-', L'-' }
}, {
	.wt_desc = "Append at start, don't fill dest",
	.wt_src = L"It's a trap?!",
	.wt_rval = 13,
	.wt_dstlen = WCSLCAT_BUFLEN,
	.wt_buf = { L'\0', L'-', L'-', L'-', L'-', L'-', L'-', L'-',
		    L'-', L'-', L'-', L'-', L'-', L'-', L'-', L'-' },
	.wt_res = { L'I', L't', L'\'', L's', L' ', L'a', L' ', L't',
		    L'r', L'a', L'p', L'?', L'!', L'\0', L'-', L'-' }

}, {
	.wt_desc = "Append at start, truncate src",
	.wt_src = L"This little string went to the market",
	.wt_rval = 37,
	.wt_dstlen = WCSLCAT_BUFLEN,
	.wt_buf = { L'\0', L'-', L'-', L'-', L'-', L'-', L'-', L'-',
		    L'-', L'-', L'-', L'-', L'-', L'-', L'-', L'-' },
	.wt_res = { L'T', L'h', L'i', L's', L' ', L'l', L'i', L't',
		    L't', L'l', L'e', L' ', L's', L't', L'r', L'\0' }

}, {
	.wt_desc = "Full buffer (no NUL), empty src",
	.wt_src = L"",
	.wt_rval = WCSLCAT_BUFLEN,
	.wt_dstlen = WCSLCAT_BUFLEN,
	.wt_buf = { L'-', L'-', L'-', L'-', L'-', L'-', L'-', L'-',
		    L'-', L'-', L'-', L'-', L'-', L'-', L'-', L'-' },
	.wt_res = { L'-', L'-', L'-', L'-', L'-', L'-', L'-', L'-',
		    L'-', L'-', L'-', L'-', L'-', L'-', L'-', L'-' }
}, {
	.wt_desc = "Full buffer (no NUL), non-empty src (1)",
	.wt_src = L"光",
	.wt_rval = WCSLCAT_BUFLEN + 1,
	.wt_dstlen = WCSLCAT_BUFLEN,
	.wt_buf = { L'-', L'-', L'-', L'-', L'-', L'-', L'-', L'-',
		    L'-', L'-', L'-', L'-', L'-', L'-', L'-', L'-' },
	.wt_res = { L'-', L'-', L'-', L'-', L'-', L'-', L'-', L'-',
		    L'-', L'-', L'-', L'-', L'-', L'-', L'-', L'-' }
}, {
	.wt_desc = "Full buffer (no NUL), non-empty src (2)",
	.wt_src = L"Link? Zelda!",
	.wt_rval = WCSLCAT_BUFLEN + 12,
	.wt_dstlen = WCSLCAT_BUFLEN,
	.wt_buf = { L'-', L'-', L'-', L'-', L'-', L'-', L'-', L'-',
		    L'-', L'-', L'-', L'-', L'-', L'-', L'-', L'-' },
	.wt_res = { L'-', L'-', L'-', L'-', L'-', L'-', L'-', L'-',
		    L'-', L'-', L'-', L'-', L'-', L'-', L'-', L'-' }
}, {
	.wt_desc = "Full buffer (w/ NUL), empty src",
	.wt_src = L"",
	.wt_rval = WCSLCAT_BUFLEN - 1,
	.wt_dstlen = WCSLCAT_BUFLEN,
	.wt_buf = { L'-', L'-', L'-', L'-', L'-', L'-', L'-', L'-',
		    L'-', L'-', L'-', L'-', L'-', L'-', L'-', L'\0' },
	.wt_res = { L'-', L'-', L'-', L'-', L'-', L'-', L'-', L'-',
		    L'-', L'-', L'-', L'-', L'-', L'-', L'-', L'\0' }
}, {
	.wt_desc = "Full buffer (w/ NUL), non-empty src (1)",
	.wt_src = L"光",
	.wt_rval = WCSLCAT_BUFLEN,
	.wt_dstlen = WCSLCAT_BUFLEN,
	.wt_buf = { L'-', L'-', L'-', L'-', L'-', L'-', L'-', L'-',
		    L'-', L'-', L'-', L'-', L'-', L'-', L'-', L'\0' },
	.wt_res = { L'-', L'-', L'-', L'-', L'-', L'-', L'-', L'-',
		    L'-', L'-', L'-', L'-', L'-', L'-', L'-', L'\0' }
}, {
	.wt_desc = "Full buffer (w/ NUL), non-empty src (2)",
	.wt_src = L"Link? Zelda!",
	.wt_rval = WCSLCAT_BUFLEN + 11,
	.wt_dstlen = WCSLCAT_BUFLEN,
	.wt_buf = { L'-', L'-', L'-', L'-', L'-', L'-', L'-', L'-',
		    L'-', L'-', L'-', L'-', L'-', L'-', L'-', L'\0' },
	.wt_res = { L'-', L'-', L'-', L'-', L'-', L'-', L'-', L'-',
		    L'-', L'-', L'-', L'-', L'-', L'-', L'-', L'\0' }
}, {
	.wt_desc = "Partial buffer (1)",
	.wt_src = L"",
	.wt_rval = 5,
	.wt_dstlen = WCSLCAT_BUFLEN,
	.wt_buf = { L'H', L'e', L'l', L'l', L'o', L'\0', L'-', L'-',
		    L'-', L'-', L'-', L'-', L'-', L'-', L'-', L'-' },
	.wt_res = { L'H', L'e', L'l', L'l', L'o', L'\0', L'-', L'-',
		    L'-', L'-', L'-', L'-', L'-', L'-', L'-', L'-' },
}, {
	.wt_desc = "Partial buffer (2)",
	.wt_src = L", world!",
	.wt_rval = 13,
	.wt_dstlen = WCSLCAT_BUFLEN,
	.wt_buf = { L'H', L'e', L'l', L'l', L'o', L'\0', L'-', L'-',
		    L'-', L'-', L'-', L'-', L'-', L'-', L'-', L'-' },
	.wt_res = { L'H', L'e', L'l', L'l', L'o', L',', L' ', L'w',
		    L'o', L'r', L'l', L'd', L'!', L'\0', L'-', L'-' },
}, {
	.wt_desc = "Partial buffer truncation",
	.wt_src = L"7777 Aerith lives?",
	.wt_rval = 27,
	.wt_dstlen = WCSLCAT_BUFLEN,
	.wt_buf = { L'S', L'e', L'p', L'h', L'i', L'r', L'o', L't',
		    L'h', L'\0', L'-', L'-', L'-', L'-', L'-', L'-' },
	.wt_res = { L'S', L'e', L'p', L'h', L'i', L'r', L'o', L't',
		    L'h', L'7', L'7', L'7', L'7', L' ', L'A', L'\0' }
}  };

static bool
wcslcat_test_one(const wcslcat_test_t *test)
{
	wchar_t buf[WCSLCAT_BUFLEN];
	size_t wcret;
	bool ret = true;

	(void) wmemcpy(buf, test->wt_buf, ARRAY_SIZE(test->wt_buf));
	wcret = wcslcat(buf, test->wt_src, test->wt_dstlen);

	if (wcret != test->wt_rval) {
		warnx("TEST FAILED: %s: wcslcat() returned %zu, expected %zu",
		    test->wt_desc, wcret, test->wt_rval);
		ret = false;
	}

	if (wmemcmp(buf, test->wt_res, ARRAY_SIZE(buf)) != 0) {
		warnx("TEST FAILED: %s: resulting buffer mismatch: found vs. "
		    "expected", test->wt_desc);
		for (size_t i = 0; i < ARRAY_SIZE(buf); i++) {
			(void) printf("\t[%zu] = [0x%x] vs [0x%x]\n", i, buf[i],
			    test->wt_res[i]);
		}
		ret = false;
	}

	if (ret) {
		(void) printf("TEST PASSED: %s\n", test->wt_desc);
	}

	return (ret);
}

int
main(void)
{
	int ret = EXIT_SUCCESS;

	for (size_t i = 0; i < ARRAY_SIZE(wcslcat_tests); i++) {
		if (!wcslcat_test_one(&wcslcat_tests[i]))
			ret = EXIT_FAILURE;
	}

	if (ret == EXIT_SUCCESS) {
		(void) printf("All tests passed successfully!\n");
	}

	return (ret);
}