#include <stdlib.h>
#include <unistd.h>
#include <stdint.h>
#include <stdio.h>
#include <sys/wait.h>
#include "aux.h"

#include <libinit/init.h>
#define print(m, ...) init_log(m, ##__VA_ARGS__)

void test_args(int argc, char *argv[], char *envp[])
{
	print("%p %p %p\n", (void *)(uint64_t)&argc, (void *)&argv, (void *)&envp);
	print("I have %d arguments\n", argc);
	for (int i = 0; i < argc; i++)
		print("argv[%d] = (%p) %s\n", i, argv[i], argv[i]);

	int envc = 0;
	while (envp[envc] != NULL)
		envc++;

	print("I have %d environment variables\n", envc);
	for (int i = 0; i < envc; i++)
		print("envp[%d] = (%p) %s\n", i, envp[i], envp[i]);

	Elf64_auxv_t *auxv;
	char **e = envp;

	while (*e++ != NULL)
		;

	for (auxv = (Elf64_auxv_t *)e; auxv->a_type != AT_NULL; auxv++)
		print("auxv: %ld %#lx\n", auxv->a_type, auxv->a_un.a_val);
}

int main(int argc, char *argv[], char *envp[])
{
	// test_args(argc, argv, envp);
	FILE *test = fopen("/Test.txt", "r");
	if (test == NULL)
	{
		print("Failed to open file\n");
		return 1;
	}

	char buf[1024];
	int read = fread(buf, 1024, 1, test);
	print("/Test.txt (%ld): %s\n", read, buf);
	fclose(test);

	pid_t pid;
	int status;

	pid = fork();

	if (pid == 0) // Child process
	{
		print("Creating shell process\n");
		char *args[] = {"/bin/sh", NULL};
		execv(args[0], args);
		exit(EXIT_FAILURE);
	}
	else if (pid > 0)
	{
		wait(&status);
		if (WIFEXITED(status))
		{
			print("Child process exited with code: %d\n", WEXITSTATUS(status));
			return WEXITSTATUS(status);
		}
		else
		{
			print("Execution failed.\n");
			exit(EXIT_FAILURE);
		}
	}
	else
	{
		print("\eFF0000Failed to create the process.\n");
		exit(EXIT_FAILURE);
	}

	return 0;
}