/*
 * Copyright (c) 2000-2024 Apple Inc. All rights reserved.
 *
 * @APPLE_OSREFERENCE_LICENSE_HEADER_START@
 *
 * This file contains Original Code and/or Modifications of Original Code
 * as defined in and that are subject to the Apple Public Source License
 * Version 2.0 (the 'License'). You may not use this file except in
 * compliance with the License. The rights granted to you under the License
 * may not be used to create, or enable the creation or redistribution of,
 * unlawful or unlicensed copies of an Apple operating system, or to
 * circumvent, violate, or enable the circumvention or violation of, any
 * terms of an Apple operating system software license agreement.
 *
 * Please obtain a copy of the License at
 * http://www.opensource.apple.com/apsl/ and read it before using this file.
 *
 * The Original Code and all software distributed under the License are
 * distributed on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER
 * EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES,
 * INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE, QUIET ENJOYMENT OR NON-INFRINGEMENT.
 * Please see the License for the specific language governing rights and
 * limitations under the License.
 *
 * @APPLE_OSREFERENCE_LICENSE_HEADER_END@
 */

#include "unit_test_utils.h"
#include "dt_proxy.h"
#include "sys/queue.h"
#include <string.h>

extern int backtrace(void **array, int size);
extern char **backtrace_symbols(void *const *array, int size);

#ifdef __BUILDING_WITH_SANITIZER__
extern void __sanitizer_symbolize_pc(void *pc, const char *fmt, char *out_buf, size_t out_buf_size);
#endif

extern int kernel_sysctlbyname(const char *, void *, size_t *, void *, size_t);

int64_t
run_sysctl_test(const char *t, int64_t value, int argc, char* const* argv)
{
	char name[1024];
	int64_t result = 0;
	size_t s = sizeof(value);
	int rc;

	snprintf(name, sizeof(name), "debug.test.%s", t);

	bool run_real = (argc > 0 && strcmp(argv[0], "real_sysctl") == 0);
	if (!run_real) {
		rc = kernel_sysctlbyname(name, &result, &s, &value, s);
	} else {
		rc = sysctlbyname(name, &result, &s, &value, s);
	}
	PT_QUIET; PT_ASSERT_POSIX_ZERO(rc, "sysctlbyname()");
	return result;
}

void *
checked_alloc_align(size_t size, size_t align)
{
	void *ptr = NULL;
	if (align < sizeof(void *)) {
		ptr = calloc(1, size);
		PT_QUIET; PT_ASSERT_NOTNULL(ptr, "failed alloc");
	} else {
		ptr = aligned_alloc(align, size);
		PT_QUIET; PT_ASSERT_NOTNULL(ptr, "failed alloc");
		memset(ptr, 0, size);
	}
	return ptr;
}

struct backtrace_array *
collect_current_backtrace(void)
{
	struct backtrace_array *bt = malloc(sizeof(struct backtrace_array));
	bt->nptrs = backtrace(bt->buffer, 100);
	return bt;
}

void
print_collected_backtrace(struct backtrace_array *bt)
{
#ifdef __BUILDING_WITH_SANITIZER__
	// If compiled with any sanitizer, use __sanitizer_symbolize_pc as it gives much more info compared to backtrace_symbols
	char description[1024];
	for (int idx = 0; idx < bt->nptrs; idx++) {
		__sanitizer_symbolize_pc(bt->buffer[idx], "%p %F %L", description,
		    sizeof(description));
		raw_printf("%d\t%s\n", idx, description);
	}
#else
	char** strings = backtrace_symbols(bt->buffer, bt->nptrs);
	PT_QUIET; PT_ASSERT_NOTNULL(strings, "backtrace_symbols");
	for (int idx = 0; idx < bt->nptrs; idx++) {
		raw_printf("%s\n", strings[idx]);
	}
	free(strings);
#endif
	raw_printf("\n");
}

void
print_current_backtrace(void)
{
	struct backtrace_array bt;
	bt.nptrs = backtrace(bt.buffer, 100);
	print_collected_backtrace(&bt);
}