From 44add70b7ca0c549f442981501b71439d4372aaf Mon Sep 17 00:00:00 2001
From: LoRd_MuldeR <mulder2@gmx.de>
Date: Sun, 20 Sep 2015 14:54:37 +0200
Subject: [PATCH] Expose the crack_command_line() function.

---
 include/MUtils/OSSupport.h |  1 +
 src/OSSupport_Win32.cpp    | 36 +++++++++++++++++++++++++++++-------
 2 files changed, 30 insertions(+), 7 deletions(-)

diff --git a/include/MUtils/OSSupport.h b/include/MUtils/OSSupport.h
index 5499929..623e4e4 100644
--- a/include/MUtils/OSSupport.h
+++ b/include/MUtils/OSSupport.h
@@ -102,6 +102,7 @@ namespace MUtils
 
 		//CLI Arguments
 		typedef QMap<QString,QString> ArgumentMap;
+		MUTILS_API const QStringList crack_command_line(const QString &command_line = QString());
 		MUTILS_API const ArgumentMap &arguments(void);
 
 		//Copy file
diff --git a/src/OSSupport_Win32.cpp b/src/OSSupport_Win32.cpp
index 7012ad6..1f1ddcd 100644
--- a/src/OSSupport_Win32.cpp
+++ b/src/OSSupport_Win32.cpp
@@ -77,6 +77,28 @@ void MUtils::OS::system_message_err(const wchar_t *const title, const wchar_t *c
 static QReadWriteLock                          g_arguments_lock;
 static QScopedPointer<MUtils::OS::ArgumentMap> g_arguments_list;
 
+const QStringList MUtils::OS::crack_command_line(const QString &command_line)
+{
+	int nArgs = 0;
+	LPWSTR *szArglist = CommandLineToArgvW(command_line.isNull() ? GetCommandLineW() : MUTILS_WCHR(command_line), &nArgs);
+
+	QStringList command_line_tokens;
+	if(NULL != szArglist)
+	{
+		for(int i = 0; i < nArgs; i++)
+		{
+			const QString argStr = MUTILS_QSTR(szArglist[i]).trimmed();
+			if(!argStr.isEmpty())
+			{
+				command_line_tokens << argStr;
+			}
+		}
+		LocalFree(szArglist);
+	}
+
+	return command_line_tokens;
+}
+
 const MUtils::OS::ArgumentMap &MUtils::OS::arguments(void)
 {
 	QReadLocker readLock(&g_arguments_lock);
@@ -98,15 +120,16 @@ const MUtils::OS::ArgumentMap &MUtils::OS::arguments(void)
 
 	g_arguments_list.reset(new ArgumentMap());
 	int nArgs = 0;
-	LPWSTR *szArglist = CommandLineToArgvW(GetCommandLineW(), &nArgs);
+	const QStringList argList = crack_command_line();
 
-	if(NULL != szArglist)
+	if(argList.count() > 1)
 	{
-		const QChar separator = QLatin1Char('=');
 		const QString argPrefix = QLatin1String("--");
-		for(int i = 0; i < nArgs; i++)
+		const QChar separator = QLatin1Char('=');
+
+		for(int i = 1; i < nArgs; i++)
 		{
-			const QString argStr = MUTILS_QSTR(szArglist[i]).trimmed();
+			const QString &argStr = argList[i];
 			if(argStr.startsWith(argPrefix))
 			{
 				const QString argData = argStr.mid(2).trimmed();
@@ -126,9 +149,8 @@ const MUtils::OS::ArgumentMap &MUtils::OS::arguments(void)
 				}
 			}
 		}
-		LocalFree(szArglist);
 	}
-	else
+	else if(argList.empty())
 	{
 		qWarning("CommandLineToArgvW() has failed !!!");
 	}
-- 
2.11.0