From 55ace2599b96a235baf2a547e0a3ebd284a35fa7 Mon Sep 17 00:00:00 2001
From: NtQuery <NtQuery>
Date: Fri, 14 Mar 2014 13:49:21 +0100
Subject: [PATCH] char white list

---
 TitanScriptGui/TitanScriptGui.cpp | 18 +++++++++++++++++-
 1 file changed, 17 insertions(+), 1 deletion(-)

diff --git a/TitanScriptGui/TitanScriptGui.cpp b/TitanScriptGui/TitanScriptGui.cpp
index 56ea25c..cc03d5f 100644
--- a/TitanScriptGui/TitanScriptGui.cpp
+++ b/TitanScriptGui/TitanScriptGui.cpp
@@ -255,11 +255,27 @@ static void AddLogMessageW(const wchar_t* szLogMessage, eLogType Type)
     }
 }
 
+static bool IsValidChar(char s)
+{
+    if (isalnum(s))
+    {
+        return true;
+    }
+    else if (s == '$' || s == '-' || s == '>')
+    {
+        return true;
+    }
+    else
+    {
+        return false;
+    }
+}
+
 static void AddLogMessage(const char* szLogMessage, eLogType Type)
 {
     if (strlen(szLogMessage) > 0)
     {
-        if (isalnum(szLogMessage[0]))
+        if (IsValidChar(szLogMessage[0]))
         {
             TCHAR * buf = (TCHAR *)calloc(strlen(szLogMessage) + 1, sizeof(TCHAR));
             if (buf)