From d20431e11290320f04d62e26b9a7cbcefaf5480c Mon Sep 17 00:00:00 2001
From: Boris Kolpackov <boris@codesynthesis.com>
Date: Wed, 27 Apr 2016 19:05:23 +0200
Subject: Add temp_directory(), temp_path()

---
 butl/path.cxx | 147 +++++++++++++++++++++++++++++++++++++++++++++++++++++++---
 1 file changed, 141 insertions(+), 6 deletions(-)

(limited to 'butl/path.cxx')

diff --git a/butl/path.cxx b/butl/path.cxx
index 3e82190..1f8f195 100644
--- a/butl/path.cxx
+++ b/butl/path.cxx
@@ -5,18 +5,29 @@
 #include <butl/path>
 
 #ifdef _WIN32
-#  include <stdlib.h> // _MAX_PATH
-#  include <direct.h> // _[w]getcwd(), _[w]chdir()
+#  include <stdlib.h>  // _MAX_PATH
+#  include <direct.h>  // _[w]getcwd(), _[w]chdir()
+#  ifndef WIN32_LEAN_AND_MEAN
+#    define WIN32_LEAN_AND_MEAN
+#  endif
+#  include <windows.h> // GetTempPath*()
+#  include <memory>    // unique_ptr
 #else
-#  include <errno.h>  // EINVAL
-#  include <stdlib.h> // mbstowcs(), wcstombs(), realpath()
-#  include <limits.h> // PATH_MAX
-#  include <unistd.h> // getcwd(), chdir()
+#  include <errno.h>     // EINVAL
+#  include <stdlib.h>    // mbstowcs(), wcstombs(), realpath(), getenv()
+#  include <limits.h>    // PATH_MAX
+#  include <unistd.h>    // getcwd(), chdir()
+#  include <string.h>    // strlen(), strcpy()
+#  include <sys/stat.h>  // stat(), S_IS*
+#  include <sys/types.h> // stat
 #endif
 
+#include <atomic>
 #include <cassert>
 #include <system_error>
 
+#include <butl/process>
+
 #ifndef _WIN32
 #  ifndef PATH_MAX
 #    define PATH_MAX 4096
@@ -69,6 +80,88 @@ namespace butl
 #endif
   }
 
+#ifdef _WIN32
+  struct msg_deleter
+  {
+    void operator() (char* p) const {LocalFree (p);}
+  };
+
+  static string
+  last_error ()
+  {
+    DWORD e (GetLastError ());
+    char* msg;
+    if (!FormatMessageA (
+          FORMAT_MESSAGE_ALLOCATE_BUFFER |
+          FORMAT_MESSAGE_FROM_SYSTEM |
+          FORMAT_MESSAGE_IGNORE_INSERTS |
+          FORMAT_MESSAGE_MAX_WIDTH_MASK,
+          0,
+          e,
+          MAKELANGID (LANG_NEUTRAL, SUBLANG_DEFAULT),
+          (char*)&msg,
+          0,
+          0))
+      return "unknown error code " + to_string (e);
+
+    unique_ptr<char, msg_deleter> m (msg);
+    return msg;
+  }
+#else
+  static const char*
+  temp_directory ()
+  {
+    const char* dir (nullptr);
+    const char* env[] = {"TMPDIR", "TMP", "TEMP", "TEMPDIR", nullptr};
+
+    for (auto e (env); dir == nullptr && *e != nullptr; ++e)
+      dir = getenv (*e);
+
+    if (dir == nullptr)
+      dir = "/tmp";
+
+    struct stat s;
+    if (stat (dir, &s) != 0)
+      throw system_error (errno, system_category ());
+
+    if (!S_ISDIR (s.st_mode))
+      throw system_error (ENOTDIR, system_category ());
+
+    return dir;
+  }
+#endif
+
+  template <>
+  path_traits<char>::string_type path_traits<char>::
+  temp_directory ()
+  {
+#ifdef _WIN32
+    char d[_MAX_PATH + 1];
+    DWORD r (GetTempPathA (_MAX_PATH + 1, d));
+
+    if (r == 0)
+    {
+      string e (last_error ());
+      throw system_error (ENOTDIR, system_category (), e);
+    }
+
+    return string_type (d);
+#else
+    return string_type (butl::temp_directory ());
+#endif
+  }
+
+  static atomic<size_t> temp_name_count;
+
+  template <>
+  path_traits<char>::string_type path_traits<char>::
+  temp_name (string_type const& prefix)
+  {
+    return prefix
+      + "-" + to_string (process::current_id ())
+      + "-" + to_string (temp_name_count++);
+  }
+
 #ifndef _WIN32
   template <>
   void path_traits<char>::
@@ -135,6 +228,48 @@ namespace butl
 #endif
   }
 
+  template <>
+  path_traits<wchar_t>::string_type path_traits<wchar_t>::
+  temp_directory ()
+  {
+#ifdef _WIN32
+    wchar_t d[_MAX_PATH + 1];
+    DWORD r (GetTempPathW (_MAX_PATH + 1, d));
+
+    if (r == 0)
+    {
+      string e (last_error ());
+      throw system_error (ENOTDIR, system_category (), e);
+    }
+#else
+    wchar_t d[PATH_MAX];
+
+    // The usage of mbstowcs() supposes the program's C-locale is set to the
+    // proper locale before the call (can be done with setlocale(LC_ALL, "...")
+    // call). Otherwise mbstowcs() fails with EILSEQ errno for non-ASCII
+    // directory paths.
+    //
+    size_t r (mbstowcs (d, butl::temp_directory (), PATH_MAX));
+
+    if (r == size_t (-1))
+      throw system_error (EINVAL, system_category ());
+
+    if (r == PATH_MAX)
+      throw system_error (ENOTSUP, system_category ());
+#endif
+
+    return string_type (d);
+  }
+
+  template <>
+  path_traits<wchar_t>::string_type path_traits<wchar_t>::
+  temp_name (string_type const& prefix)
+  {
+    return prefix +
+      L"-" + to_wstring (process::current_id ()) +
+      L"-" + to_wstring (temp_name_count++);
+  }
+
 #ifndef _WIN32
   template <>
   void path_traits<wchar_t>::
-- 
cgit v1.1