/*
- Copyright (C) 2010 Paul Davis
+ Copyright (C) 2010 Paul Davis
This program is free software; you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
*/
+#include <glib.h>
+
#include <cstdlib>
-#include <iostream>
#include "pbd/epa.h"
+#include "pbd/strsplit.h"
+#ifdef COMPILER_MSVC
+#define environ _environ
+_CRTIMP extern char ** _environ;
+#else
extern char** environ;
+#endif
using namespace PBD;
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
{
e.clear ();
- for (size_t i = 0; environ[i]; ++i) {
+ if (!_envname.empty()) {
- string estring = environ[i];
- string::size_type equal = estring.find_first_of ('=');
+ /* fetch environment from named environment variable, rather than "environ"
+ */
- if (equal == string::npos) {
- /* say what? an environ value without = ? */
- continue;
+ const char* estr = g_getenv (_envname.c_str());
+
+ if (!estr) {
+ return;
}
- string before = estring.substr (0, equal);
- string after = estring.substr (equal+1);
+ /* 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) {
- cerr << "Save [" << before << "] = " << after << endl;
+ string estring = environ[i];
+ string::size_type equal = estring.find_first_of ('=');
- e.insert (pair<string,string>(before,after));
+ 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));
+ }
}
-}
+}
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
+{
+ /* Copy the environment before using (g_)unsetenv() because on some
+ platforms (maybe all?) this directly modifies the environ array,
+ cause complications for iterating through it.
+ */
+
+ vector<string> ecopy;
+
+ for (size_t i = 0; environ[i]; ++i) {
+ ecopy.push_back (environ[i]);
+ }
+
+ for (vector<string>::const_iterator e = ecopy.begin(); e != ecopy.end(); ++e) {
+ string::size_type equal = (*e).find_first_of ('=');
+
+ if (equal == string::npos) {
+ /* say what? an environ value without = ? */
+ continue;
+ }
+
+ string var_name = (*e).substr (0, equal);
+ g_unsetenv(var_name.c_str());
+ }
+}