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 #include <cassert>
31 #include <map>
32
33 #include "config.hpp"
34 #include "env.hpp"
35 #include "text.hpp"
36
37 static std::map< std::string, std::string > m_variables;
38
39 static struct var {
40 const char *name;
41 const char *default_value;
42 bool can_be_empty;
43 } vars[] = {
44 { "ATF_ARCH", ATF_ARCH, false, },
45 { "ATF_BUILD_CC", ATF_BUILD_CC, false, },
46 { "ATF_BUILD_CFLAGS", ATF_BUILD_CFLAGS, true, },
47 { "ATF_BUILD_CPP", ATF_BUILD_CPP, false, },
48 { "ATF_BUILD_CPPFLAGS", ATF_BUILD_CPPFLAGS, true, },
49 { "ATF_BUILD_CXX", ATF_BUILD_CXX, false, },
50 { "ATF_BUILD_CXXFLAGS", ATF_BUILD_CXXFLAGS, true, },
51 { "ATF_CONFDIR", ATF_CONFDIR, false, },
52 { "ATF_INCLUDEDIR", ATF_INCLUDEDIR, false, },
53 { "ATF_LIBDIR", ATF_LIBDIR, false, },
54 { "ATF_LIBEXECDIR", ATF_LIBEXECDIR, false, },
55 { "ATF_MACHINE", ATF_MACHINE, false, },
56 { "ATF_PKGDATADIR", ATF_PKGDATADIR, false, },
57 { "ATF_SHELL", ATF_SHELL, false, },
58 { "ATF_WORKDIR", ATF_WORKDIR, false, },
59 { NULL, NULL, false, },
60 };
61
62 //
63 // Adds all predefined standard build-time variables to the m_variables
64 // map, considering the values a user may have provided in the environment.
65 //
66 // Can only be called once during the program's lifetime.
67 //
68 static
69 void
init_variables(void)70 init_variables(void)
71 {
72 assert(m_variables.empty());
73
74 for (struct var* v = vars; v->name != NULL; v++) {
75 const std::string varname = tools::text::to_lower(v->name);
76
77 if (tools::env::has(v->name)) {
78 const std::string envval = tools::env::get(v->name);
79 if (envval.empty() && !v->can_be_empty)
80 m_variables[varname] = v->default_value;
81 else
82 m_variables[varname] = envval;
83 } else {
84 m_variables[varname] = v->default_value;
85 }
86 }
87
88 assert(!m_variables.empty());
89 }
90
91 const std::string&
get(const std::string & varname)92 tools::config::get(const std::string& varname)
93 {
94 if (m_variables.empty())
95 init_variables();
96
97 assert(has(varname));
98 return m_variables[varname];
99 }
100
101 const std::map< std::string, std::string >&
get_all(void)102 tools::config::get_all(void)
103 {
104 if (m_variables.empty())
105 init_variables();
106
107 return m_variables;
108 }
109
110 bool
has(const std::string & varname)111 tools::config::has(const std::string& varname)
112 {
113 if (m_variables.empty())
114 init_variables();
115
116 return m_variables.find(varname) != m_variables.end();
117 }
118
119 namespace tools {
120 namespace config {
121 //
122 // Auxiliary function for the t_config test program so that it can
123 // revert the configuration's global status to an empty state and
124 // do new tests from there on.
125 //
126 // Ideally this shouldn't be part of the production library... but
127 // this is so small that it does not matter.
128 //
129 void
__reinit(void)130 __reinit(void)
131 {
132 m_variables.clear();
133 }
134 } // namespace config
135 } // namespace tools
136