vbox的更動 20101 路徑 trunk/src/VBox/Runtime/r3/path.cpp

時間撮記:

2009-5-27 下午04:43:08 (15 年 以前)

作者:

vboxsync

訊息:

IPRT: Rewrote RTPathTemp and added a testcase for it.

檔案:

• trunk/src/VBox/Runtime/r3/path.cpp (修改) (2 筆差異)

trunk/src/VBox/Runtime/r3/path.cpp

@@ -964 +964 @@
 }
 
+
 /**
  * Gets the temporary directory path.
  *
  * @returns iprt status code.
+ *
  * @param   pszPath     Buffer where to store the path.
  * @param   cchPath     Buffer size in bytes.
@@ -973 +975 @@
 RTDECL(int) RTPathTemp(char *pszPath, size_t cchPath)
 {
-    int rc;
-    const char *pszTmpEnv = RTEnvGet("TMP");
-    if (!pszTmpEnv)
-        pszTmpEnv = RTEnvGet("TEMP");
-    char *pszTmpDir;
-    /* Make a copy in any case. */
-    if (pszTmpEnv)
-        pszTmpDir = RTStrDup(pszTmpEnv);
-    else
-        pszTmpDir = RTStrDup("/tmp");
-
-    size_t cchTmpDir = strlen(pszTmpDir);
-    if (cchTmpDir < cchPath)
-        memcpy(pszPath, pszTmpDir, cchTmpDir + 1);
-    else
-        rc = VERR_BUFFER_OVERFLOW;
-    RTStrFree(pszTmpDir);
-    return rc;
+    /*
+     * Try get it from the environment first.
+     */
+    static const char * const s_apszVars[] =
+    {
+        "IPRT_TMPDIR"
+#if defined(RT_OS_WINDOWS)
+        , "TMP", "TEMP", "USERPROFILE"
+#elif defined(RT_OS_OS2)
+        , "TMP", "TEMP", "TMPDIR"
+#else
+        , "TMPDIR"
+#endif
+    };
+    for (size_t iVar = 0; iVar < RT_ELEMENTS(s_apszVars); iVar++)
+    {
+        int rc = RTEnvGetEx(RTENV_DEFAULT, s_apszVars[iVar], pszPath, cchPath, NULL);
+        if (rc != VERR_ENV_VAR_NOT_FOUND)
+            return rc;
+    }
+
+    /*
+     * Here we should use some sane system default, instead we just use
+     * the typical unix temp dir for now.
+     */
+    /** @todo Windows should default to the windows directory, see GetTempPath.
+     * Some unixes has path.h and _PATH_TMP. There is also a question about
+     * whether /var/tmp wouldn't be a better place...  */
+    if (cchPath < sizeof("/tmp") )
+        return VERR_BUFFER_OVERFLOW;
+    memcpy(pszPath, "/tmp", sizeof("/tmp"));
+    return VINF_SUCCESS;
 }