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
30extern "C" {
31#include <errno.h>
32#include <stdlib.h>
33#include <string.h>
34}
35
36#include <cassert>
37
38#include "env.hpp"
39#include "exceptions.hpp"
40
41namespace impl = tools::env;
42#define IMPL_NAME "tools::env"
43
44// ------------------------------------------------------------------------
45// Free functions.
46// ------------------------------------------------------------------------
47
48std::string
49impl::get(const std::string& name)
50{
51    const char *n =name.c_str();
52    const char* val = getenv(n);
53    if (val != NULL)
54	return val;
55    if (strcmp(n, "PATH") == 0)
56	return "/bin:/usr/bin:/sbin:/usr/sbin";
57
58    throw tools::system_error(IMPL_NAME "::set",
59			    "Cannot get environment variable '" + name +
60			    "'", errno);
61}
62
63bool
64impl::has(const std::string& name)
65{
66    return getenv(name.c_str()) != NULL;
67}
68
69void
70impl::set(const std::string& name, const std::string& val)
71{
72    if (setenv(name.c_str(), val.c_str(), 1) == -1)
73        throw tools::system_error(IMPL_NAME "::set",
74                                "Cannot set environment variable '" + name +
75                                "' to '" + val + "'",
76                                errno);
77}
78
79void
80impl::unset(const std::string& name)
81{
82    unsetenv(name.c_str());
83}
84