From 56095b29f4beab8f05721ea356e061400e7816b7 Mon Sep 17 00:00:00 2001
From: Julian Brost <julian.brost@icinga.com>
Date: Thu, 28 Jan 2021 16:25:12 +0100
Subject: [PATCH] Add tests for Utility::EscapeCreateProcessArg

---
 test/CMakeLists.txt   |  1 +
 test/base-utility.cpp | 35 +++++++++++++++++++++++++++++++++++
 2 files changed, 36 insertions(+)

diff --git a/test/CMakeLists.txt b/test/CMakeLists.txt
index b973a4361..8649a4865 100644
--- a/test/CMakeLists.txt
+++ b/test/CMakeLists.txt
@@ -111,6 +111,7 @@ add_boost_test(base
     base_utility/comparepasswords_works
     base_utility/comparepasswords_issafe
     base_utility/validateutf8
+    base_utility/EscapeCreateProcessArg
     base_value/scalar
     base_value/convert
     base_value/format
diff --git a/test/base-utility.cpp b/test/base-utility.cpp
index a73c32008..43fba3b30 100644
--- a/test/base-utility.cpp
+++ b/test/base-utility.cpp
@@ -4,6 +4,12 @@
 #include <chrono>
 #include <BoostTestTargetConfig.h>
 
+#ifdef _WIN32
+# include <windows.h>
+# include <shellapi.h>
+# include <atlstr.h>
+#endif /* _WIN32 */
+
 using namespace icinga;
 
 BOOST_AUTO_TEST_SUITE(base_utility)
@@ -58,4 +64,33 @@ BOOST_AUTO_TEST_CASE(validateutf8)
 	BOOST_CHECK(Utility::ValidateUTF8("\xC3\xA4") == "\xC3\xA4");
 }
 
+BOOST_AUTO_TEST_CASE(EscapeCreateProcessArg)
+{
+#ifdef _WIN32
+	std::vector<std::string> testdata = {
+		R"(foobar)",
+		R"(foo bar)",
+		R"(foo"bar)",
+		R"("foo bar")",
+		R"(" \" \\" \\\" \\\\")",
+		R"( !"#$$%&'()*+,-./09:;<=>?@AZ[\]^_`az{|}~ " \" \\" \\\" \\\\")",
+		"'foo\nbar'",
+	};
+
+	for (const auto& t : testdata) {
+		// Prepend some fake exec name as the first argument is handled differently.
+		std::string escaped = "some.exe " + Utility::EscapeCreateProcessArg(t);
+		int argc;
+		std::shared_ptr<LPWSTR> argv(CommandLineToArgvW(CA2W(escaped.c_str()), &argc), LocalFree);
+		BOOST_CHECK_MESSAGE(argv != nullptr, "CommandLineToArgvW() should not return nullptr for " << t);
+		BOOST_CHECK_MESSAGE(argc == 2, "CommandLineToArgvW() should find 2 arguments for " << t);
+		if (argc >= 2) {
+			std::string unescaped = CW2A(argv.get()[1]);
+			BOOST_CHECK_MESSAGE(unescaped == t,
+				"CommandLineToArgvW() should return original value for " << t << " (got: " << unescaped << ")");
+		}
+	}
+#endif /* _WIN32 */
+}
+
 BOOST_AUTO_TEST_SUITE_END()