env.cpp revision 1.1 
1//
2// Automated Testing Framework (atf)
3//
4// Copyright (c) 2007 The NetBSD Foundation, Inc.
5// All rights reserved.
6//
7// Redistribution and use in source and binary forms, with or without
8// modification, are permitted provided that the following conditions
9// are met:
10// 1. Redistributions of source code must retain the above copyright
11//    notice, this list of conditions and the following disclaimer.
12// 2. Redistributions in binary form must reproduce the above copyright
13//    notice, this list of conditions and the following disclaimer in the
14//    documentation and/or other materials provided with the distribution.
15//
16// THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND
17// CONTRIBUTORS ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES,
18// INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF
19// MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
20// IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS BE LIABLE FOR ANY
21// DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
22// DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE
23// GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
24// INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER
25// IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR
26// OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN
27// IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
28//
29
30#if defined(HAVE_CONFIG_H)
31#include "bconfig.h"
32#endif
33
34extern "C" {
35#include <errno.h>
36#include <stdlib.h>
37#include <string.h>
38}
39
40#include <cassert>
41
42#include "env.hpp"
43#include "exceptions.hpp"
44
45namespace impl = tools::env;
46#define IMPL_NAME "tools::env"
47
48// ------------------------------------------------------------------------
49// Free functions.
50// ------------------------------------------------------------------------
51
52std::string
53impl::get(const std::string& name)
54{
55    const char* val = getenv(name.c_str());
56    assert(val != NULL);
57    return val;
58}
59
60bool
61impl::has(const std::string& name)
62{
63    return getenv(name.c_str()) != NULL;
64}
65
66void
67impl::set(const std::string& name, const std::string& val)
68{
69#if defined(HAVE_SETENV)
70    if (setenv(name.c_str(), val.c_str(), 1) == -1)
71        throw tools::system_error(IMPL_NAME "::set",
72                                "Cannot set environment variable '" + name +
73                                "' to '" + val + "'",
74                                errno);
75#elif defined(HAVE_PUTENV)
76    const std::string buf = name + "=" + val;
77    if (putenv(strdup(buf.c_str())) == -1)
78        throw tools::system_error(IMPL_NAME "::set",
79                                "Cannot set environment variable '" + name +
80                                "' to '" + val + "'",
81                                errno);
82#else
83#   error "Don't know how to set an environment variable."
84#endif
85}
86
87void
88impl::unset(const std::string& name)
89{
90#if defined(HAVE_UNSETENV)
91    unsetenv(name.c_str());
92#elif defined(HAVE_PUTENV)
93    const std::string buf = name + "=";
94
95    if (putenv(strdup(buf.c_str())) == -1)
96        throw tools::system_error(IMPL_NAME "::unset",
97                                "Cannot unset environment variable '" +
98                                name + "'", errno);
99#else
100#   error "Don't know how to unset an environment variable."
101#endif
102}
103