main.c :modify for security

yuk7 · Nov 19, 2017 · b5f6c9d · b5f6c9d
1 parent 7f8bba4
commit b5f6c9d
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/main.c b/main.c
@@ -249,11 +249,11 @@ int main(int argc,char *argv[])
             }
         }
 
-        wchar_t rArgs[100] = L"";
+        wchar_t rArgs[300] = L"";
         for (int i=2;i<wargc;i++)
         {
-            wcscat(rArgs,L" ");
-            wcscat(rArgs,wargv[i]);
+            wcscat_s(rArgs,ARRAY_LENGTH(rArgs),L" ");
+            wcscat_s(rArgs,ARRAY_LENGTH(rArgs,wargv[i]);
         }
 
         wchar_t wcmd[300] = L"wsl.exe ";