Use PBD::ffs for portability

[ardour.git] / libs / pbd / epa.cc

diff --git a/libs/pbd/epa.cc b/libs/pbd/epa.cc
index 71387b4c23ab37b34c05f4c4595fde71297f83a4..d8a3cd5a65d0f99678218d01efc6fb620ac87b08 100644 (file)
--- a/libs/pbd/epa.cc
+++ b/libs/pbd/epa.cc
@@ -17,10 +17,12 @@
 
 */
 
+#include <glib.h>
+
 #include <cstdlib>
-#include <iostream>
 
 #include "pbd/epa.h"
+#include "pbd/strsplit.h"
 
 extern char** environ;
 
@@ -29,14 +31,26 @@ using namespace std;
 
 EnvironmentalProtectionAgency* EnvironmentalProtectionAgency::_global_epa = 0;
 
-EnvironmentalProtectionAgency::EnvironmentalProtectionAgency ()
+EnvironmentalProtectionAgency::EnvironmentalProtectionAgency (bool arm, const std::string& envname)
+        : _armed (arm)
+        , _envname (envname)
 {
-        save ();
+        if (_armed) {
+                save ();
+        }
 }
 
 EnvironmentalProtectionAgency::~EnvironmentalProtectionAgency()
 {
-        restore ();
+        if (_armed) {
+                restore ();
+        }
+}
+
+void
+EnvironmentalProtectionAgency::arm ()
+{
+        _armed = true;
 }
 
 void
@@ -44,29 +58,87 @@ EnvironmentalProtectionAgency::save ()
 {
         e.clear ();
 
-        for (size_t i = 0; environ[i]; ++i) {
+        if (!_envname.empty()) {
+                
+                /* fetch environment from named environment variable, rather than "environ"
+                 */
 
-                string estring = environ[i];
-                string::size_type equal = estring.find_first_of ('=');
+                const char* estr = g_getenv (_envname.c_str());
 
-                if (equal == string::npos) {
-                        /* say what? an environ value without = ? */
-                        continue;
+                if (!estr) {
+                        return;
+                }
+                
+                /* parse line by line, and save into "e" 
+                 */
+
+                vector<string> lines;
+                split (estr, lines, '\n');
+
+                for (vector<string>::iterator i = lines.begin(); i != lines.end(); ++i) {
+
+                        string estring = *i;
+                        string::size_type equal = estring.find_first_of ('=');
+                        
+                        if (equal == string::npos) {
+                                /* say what? an environ value without = ? */
+                                continue;
+                        }
+                        
+                        string before = estring.substr (0, equal);
+                        string after = estring.substr (equal+1);
+                        
+                        e.insert (pair<string,string>(before,after));
+                }
+                
+        } else {
+
+                /* fetch environment from "environ"
+                 */
+
+                for (size_t i = 0; environ[i]; ++i) {
+                        
+                        string estring = environ[i];
+                        string::size_type equal = estring.find_first_of ('=');
+                        
+                        if (equal == string::npos) {
+                                /* say what? an environ value without = ? */
+                                continue;
+                        }
+                        
+                        string before = estring.substr (0, equal);
+                        string after = estring.substr (equal+1);
+                        
+                        e.insert (pair<string,string>(before,after));
                 }
-
-                string before = estring.substr (0, equal);
-                string after = estring.substr (equal+1);
-
-                cerr << "Save [" << before << "] = " << after << endl;
-
-                e.insert (pair<string,string>(before,after));
         }
 }                         
 void
-EnvironmentalProtectionAgency::restore ()
+EnvironmentalProtectionAgency::restore () const
 {
-        for (map<string,string>::iterator i = e.begin(); i != e.end(); ++i) {
-                cerr << "Restore [" << i->first << "] = " << i->second << endl;
-                setenv (i->first.c_str(), i->second.c_str(), 1);
+               clear ();
+
+        for (map<string,string>::const_iterator i = e.begin(); i != e.end(); ++i) {
+                g_setenv (i->first.c_str(), i->second.c_str(), 1);
         }
-}                         
+} 
+
+void
+EnvironmentalProtectionAgency::clear () const
+{
+        char** the_environ = environ;
+
+        for (size_t i = 0; the_environ[i]; ++i) {
+                       
+                string estring = the_environ[i];
+                string::size_type equal = estring.find_first_of ('=');
+                       
+                if (equal == string::npos) {
+                        /* say what? an environ value without = ? */
+                        continue;
+                }
+                       
+                string before = estring.substr (0, equal);
+                g_unsetenv(before.c_str());
+        }
+}