fix potential non-zero termination of a string buffer

Author: kroening

src/util/tempdir.cpp

@@ -15,7 +15,7 @@ Author: CM Wintersteiger
 #endif
 
 #include <cstdlib>
-#include <cstring>
+#include <vector>
 
 #if defined(__linux__) || \
     defined(__FreeBSD_kernel__) || \
@@ -34,26 +34,30 @@ std::string get_temporary_directory(const std::string &name_template)
   std::string result;
 
   #ifdef _WIN32
-    DWORD dwBufSize = MAX_PATH;
-    char lpPathBuffer[MAX_PATH];
+    DWORD dwBufSize = MAX_PATH+1;
+    char lpPathBuffer[MAX_PATH+1];
     DWORD dwRetVal = GetTempPathA(dwBufSize, lpPathBuffer);
 
     if(dwRetVal > dwBufSize || (dwRetVal == 0))
       throw "GetTempPath failed"; // NOLINT(readability/throw)
 
-    char t[MAX_PATH];
+    // GetTempFileNameA produces <path>\<pre><uuuu>.TMP
+    // where <pre> = "TLO"
+    // Thus, we must make the buffer 1+3+4+1+3=12 characters longer.
 
-    strncpy(t, name_template.c_str(), MAX_PATH);
+    std::size_t size=strlen(lpPathBuffer)+1;
+    std::vector<char> t(lpPathBuffer, lpPathBuffer+size);
+    t.resize(size+12);
 
-    UINT uRetVal=GetTempFileNameA(lpPathBuffer, "TLO", 0, t);
+    UINT uRetVal=GetTempFileNameA(lpPathBuffer, "TLO", 0, t.data());
     if(uRetVal == 0)
       throw "GetTempFileName failed"; // NOLINT(readability/throw)
 
-    unlink(t);
-    if(_mkdir(t)!=0)
+    unlink(t.data());
+    if(_mkdir(t.data())!=0)
       throw "_mkdir failed";
 
-    result=std::string(t);
+    result=std::string(t.data());
 
   #else
     std::string prefixed_name_template="/tmp/";
@@ -64,9 +68,9 @@ std::string get_temporary_directory(const std::string &name_template)
       prefixed_name_template+='/';
     prefixed_name_template+=name_template;
 
-    char t[1000];
-    strncpy(t, prefixed_name_template.c_str(), 1000);
-    const char *td = mkdtemp(t);
+    std::vector<char> t(prefixed_name_template.begin(), prefixed_name_template.end());
+    t.push_back('\0'); // add the zero
+    const char *td = mkdtemp(t.data());
     if(!td)
       throw "mkdtemp failed";
     result=std::string(td);