From 7eb5d220fd9c0f9200709dd81f8d65368635eacb Mon Sep 17 00:00:00 2001
From: Bananymous <oskari.alaranta@bananymous.com>
Date: Wed, 14 Feb 2024 15:01:27 +0200
Subject: [PATCH] Userspace: Implement getopt for testing libc getopt()

---
 userspace/CMakeLists.txt        |  1 +
 userspace/getopt/CMakeLists.txt | 16 ++++++++++++
 userspace/getopt/main.cpp       | 44 +++++++++++++++++++++++++++++++++
 3 files changed, 61 insertions(+)
 create mode 100644 userspace/getopt/CMakeLists.txt
 create mode 100644 userspace/getopt/main.cpp

diff --git a/userspace/CMakeLists.txt b/userspace/CMakeLists.txt
index cb4a6542..facadf46 100644
--- a/userspace/CMakeLists.txt
+++ b/userspace/CMakeLists.txt
@@ -10,6 +10,7 @@ set(USERSPACE_PROJECTS
 	dd
 	dhcp-client
 	echo
+	getopt
 	id
 	image
 	init
diff --git a/userspace/getopt/CMakeLists.txt b/userspace/getopt/CMakeLists.txt
new file mode 100644
index 00000000..83509b77
--- /dev/null
+++ b/userspace/getopt/CMakeLists.txt
@@ -0,0 +1,16 @@
+cmake_minimum_required(VERSION 3.26)
+
+project(getopt CXX)
+
+set(SOURCES
+	main.cpp
+)
+
+add_executable(getopt ${SOURCES})
+target_compile_options(getopt PUBLIC -O2 -g)
+target_link_libraries(getopt PUBLIC libc ban)
+
+add_custom_target(getopt-install
+	COMMAND ${CMAKE_COMMAND} -E copy ${CMAKE_CURRENT_BINARY_DIR}/getopt ${BANAN_BIN}/
+	DEPENDS getopt
+)
diff --git a/userspace/getopt/main.cpp b/userspace/getopt/main.cpp
new file mode 100644
index 00000000..a5f379a0
--- /dev/null
+++ b/userspace/getopt/main.cpp
@@ -0,0 +1,44 @@
+#include <BAN/String.h>
+#include <BAN/Vector.h>
+#include <stdio.h>
+#include <unistd.h>
+
+int main(int argc, char** argv)
+{
+	if (argc < 2)
+	{
+		fprintf(stderr, "usage: %s OPTSTRING [PARAMETERS]...", argv[0]);
+		return 1;
+	}
+
+	BAN::Vector<char*> argv_copy(argc - 1);
+	argv_copy[0] = argv[0];
+	for (int i = 2; i < argc; i++)
+		argv_copy[i - 1] = argv[i];
+
+	int opt;
+	BAN::String parsed;
+	while ((opt = getopt(argc - 1, argv_copy.data(), argv[1])) != -1)
+	{
+		if (opt == ':' || opt == '?')
+			continue;
+
+		MUST(parsed.append(" -"));
+		MUST(parsed.push_back(opt));
+
+		if (optarg)
+		{
+			MUST(parsed.push_back(' '));
+			MUST(parsed.append(optarg));
+		}
+
+		optarg = nullptr;
+	}
+
+	printf("%s --", parsed.data());
+	for (; optind < argc - 1; optind++)
+		printf(" %s", argv_copy[optind]);
+	printf("\n");
+
+	return 0;
+}