1/*
2 * Automated Testing Framework (atf)
3 *
4 * Copyright (c) 2007, 2008 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
34#include <errno.h>
35#include <stdlib.h>
36
37#include "atf-c/error.h"
38
39#include "env.h"
40#include "sanity.h"
41#include "text.h"
42
43const char *
44atf_env_get(const char *name)
45{
46    const char* val = getenv(name);
47    PRE(val != NULL);
48    return val;
49}
50
51bool
52atf_env_has(const char *name)
53{
54    return getenv(name) != NULL;
55}
56
57atf_error_t
58atf_env_set(const char *name, const char *val)
59{
60    atf_error_t err;
61
62#if defined(HAVE_SETENV)
63    if (setenv(name, val, 1) == -1)
64        err = atf_libc_error(errno, "Cannot set environment variable "
65                             "'%s' to '%s'", name, val);
66    else
67        err = atf_no_error();
68#elif defined(HAVE_PUTENV)
69    char *buf;
70
71    err = atf_text_format(&buf, "%s=%s", name, val);
72    if (!atf_is_error(err)) {
73        if (putenv(buf) == -1)
74            err = atf_libc_error(errno, "Cannot set environment variable "
75                                 "'%s' to '%s'", name, val);
76        free(buf);
77    }
78#else
79#   error "Don't know how to set an environment variable."
80#endif
81
82    return err;
83}
84
85atf_error_t
86atf_env_unset(const char *name)
87{
88    atf_error_t err;
89
90#if defined(HAVE_UNSETENV)
91    unsetenv(name);
92    err = atf_no_error();
93#elif defined(HAVE_PUTENV)
94    char *buf;
95
96    err = atf_text_format(&buf, "%s=", name);
97    if (!atf_is_error(err)) {
98        if (putenv(buf) == -1)
99            err = atf_libc_error(errno, "Cannot unset environment variable"
100                                 " '%s'", name);
101        free(buf);
102    }
103#else
104#   error "Don't know how to unset an environment variable."
105#endif
106
107    return err;
108}
109