commit 00a680397ef0393536b0020d5869007882ce7af6
author Mark Salyzyn <salyzyn@google.com> Tue Nov 13 15:34:38 2018 -0800
committer Mark Salyzyn <salyzyn@google.com> Wed Nov 14 12:52:13 2018 -0800
tree 33de60bd12ca8a79862936e7923b3b8adef3cd64
parent 6f91d217978e81cf1b581da31b77269431e5cca1

base: Add sensitivity to TMPDIR environment for TemporaryDir and TemporaryFile

Test: compile
Bug: 119313545
Change-Id: I2917847f0e90df94d74f4bfc0b13fbc6a5f243c3

file.cpp

diff --git a/file.cpp b/file.cpp
index e4407d0..f362f94 100644
--- a/file.cpp
+++ b/file.cpp
@@ -72,7 +72,8 @@
 
 std::string GetSystemTempDir() {
 #if defined(__ANDROID__)
-  const char* tmpdir = "/data/local/tmp";
+  const auto* tmpdir = getenv("TMPDIR");
+  if (tmpdir == nullptr) tmpdir = "/data/local/tmp";
   if (access(tmpdir, R_OK | W_OK | X_OK) == 0) {
     return tmpdir;
   }
@@ -81,7 +82,7 @@
   return ".";
 #elif defined(_WIN32)
   char tmp_dir[MAX_PATH];
-  DWORD result = GetTempPathA(sizeof(tmp_dir), tmp_dir);
+  DWORD result = GetTempPathA(sizeof(tmp_dir), tmp_dir);  // checks TMP env
   CHECK_NE(result, 0ul) << "GetTempPathA failed, error: " << GetLastError();
   CHECK_LT(result, sizeof(tmp_dir)) << "path truncated to: " << result;
 
@@ -91,7 +92,9 @@
   tmp_dir[result - 1] = '\0';
   return tmp_dir;
 #else
-  return "/tmp";
+  const auto* tmpdir = getenv("TMPDIR");
+  if (tmpdir == nullptr) tmpdir = "/tmp";
+  return tmpdir;
 #endif
 }