ENH: move PutEnv to SystemTools

2 files changed, 31 insertions, 0 deletions

diff --git a/Source/kwsys/SystemTools.cxx b/Source/kwsys/SystemTools.cxx
index 2d90de2..45c4969 100644
--- a/Source/kwsys/SystemTools.cxx
+++ b/Source/kwsys/SystemTools.cxx
@@ -402,6 +402,33 @@ bool SystemTools::GetEnv(const char* key, kwsys_stl::string& result)
     }
 }
 
+class kwsysDeletingCharVector : public kwsys_stl::vector<char*>
+{
+public:
+  ~kwsysDeletingCharVector()
+    {
+      for(kwsys_stl::vector<char*>::iterator i = this->begin();
+          i != this->end(); ++i)
+        {
+        delete []*i;
+        }
+    }
+};
+
+        
+bool SystemTools::PutEnv(const char* value)
+{ 
+  static kwsysDeletingCharVector localEnvironment;
+  char* envVar = new char[strlen(value)+1];
+  strcpy(envVar, value);
+  int ret = putenv(envVar);
+  // save the pointer in the static vector so that it can
+  // be deleted on exit
+  localEnvironment.push_back(envVar);
+  return ret == 0;
+}
+
+
 const char* SystemTools::GetExecutableExtension()
 {
 #if defined(_WIN32) || defined(__CYGWIN__)
diff --git a/Source/kwsys/SystemTools.hxx.in b/Source/kwsys/SystemTools.hxx.in
index 31a876d..8cba810 100644
--- a/Source/kwsys/SystemTools.hxx.in
+++ b/Source/kwsys/SystemTools.hxx.in
@@ -754,6 +754,10 @@ public:
   static const char* GetEnv(const char* key);
   static bool GetEnv(const char* key, kwsys_stl::string& result);
 
+  /** Put a string into the environment
+      of the form var=value */
+  static bool PutEnv(const char* value);
+
   /**
    * Get current working directory CWD
    */