13
0
livetrax/libs/pbd/epa.cc

157 lines
4.3 KiB
C++
Raw Normal View History

/*
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
the Free Software Foundation; either version 2 of the License, or
(at your option) any later version.
This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.
You should have received a copy of the GNU General Public License
along with this program; if not, write to the Free Software
Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
*/
#include <glib.h>
#include <cstdlib>
#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 (bool arm, const std::string& envname)
: _armed (arm)
, _envname (envname)
{
if (_armed) {
save ();
}
}
EnvironmentalProtectionAgency::~EnvironmentalProtectionAgency()
{
if (_armed) {
restore ();
}
}
void
EnvironmentalProtectionAgency::arm ()
{
_armed = true;
}
void
EnvironmentalProtectionAgency::save ()
{
e.clear ();
if (!_envname.empty()) {
/* fetch environment from named environment variable, rather than "environ"
*/
const char* estr = g_getenv (_envname.c_str());
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));
}
}
}
void
EnvironmentalProtectionAgency::restore () const
{
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 ('=');
2015-10-05 10:17:49 -04:00
if (equal == string::npos) {
/* say what? an environ value without = ? */
continue;
}
2015-10-05 10:17:49 -04:00
string var_name = (*e).substr (0, equal);
g_unsetenv(var_name.c_str());
}
}