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 <cstdlib> 31 #include <iostream> 32 #include <map> 33 #include <string> 34 35 extern "C" { 36 #include "atf-c/defs.h" 37 } 38 39 #include "atf-c++/config.hpp" 40 41 #include "atf-c++/detail/application.hpp" 42 #include "atf-c++/detail/sanity.hpp" 43 44 class atf_config : public atf::application::app { 45 static const char* m_description; 46 47 bool m_tflag; 48 49 void process_option(int, const char*); 50 std::string specific_args(void) const; 51 options_set specific_options(void) const; 52 53 std::string format_var(const std::string&, const std::string&); 54 55 public: 56 atf_config(void); 57 58 int main(void); 59 }; 60 61 const char* atf_config::m_description = 62 "atf-config is a tool that queries the value of several " 63 "installation-specific configuration values of the atf. " 64 "It can be used by external tools to discover where specific " 65 "internal atf files are installed."; 66 67 atf_config::atf_config(void) : 68 app(m_description, "atf-config(1)", "atf(7)"), 69 m_tflag(false) 70 { 71 } 72 73 void 74 atf_config::process_option(int ch, const char* arg ATF_DEFS_ATTRIBUTE_UNUSED) 75 { 76 switch (ch) { 77 case 't': 78 m_tflag = true; 79 break; 80 81 default: 82 UNREACHABLE; 83 } 84 } 85 86 std::string 87 atf_config::specific_args(void) 88 const 89 { 90 return "[var1 [.. varN]]"; 91 } 92 93 atf_config::options_set 94 atf_config::specific_options(void) 95 const 96 { 97 using atf::application::option; 98 options_set opts; 99 opts.insert(option('t', "", "Terse output: show values only")); 100 return opts; 101 } 102 103 std::string 104 atf_config::format_var(const std::string& name, const std::string& val) 105 { 106 std::string str; 107 108 if (m_tflag) 109 str = val; 110 else 111 str = name + " : " + val; 112 113 return str; 114 } 115 116 int 117 atf_config::main(void) 118 { 119 if (m_argc < 1) { 120 std::map< std::string, std::string > cv = atf::config::get_all(); 121 122 for (std::map< std::string, std::string >::const_iterator iter = 123 cv.begin(); iter != cv.end(); iter++) 124 std::cout << format_var((*iter).first, (*iter).second) << "\n"; 125 } else { 126 for (int i = 0; i < m_argc; i++) { 127 if (!atf::config::has(m_argv[i])) 128 throw std::runtime_error(std::string("Unknown variable `") + 129 m_argv[i] + "'"); 130 } 131 132 for (int i = 0; i < m_argc; i++) { 133 std::cout << format_var(m_argv[i], atf::config::get(m_argv[i])) 134 << "\n"; 135 } 136 } 137 138 return EXIT_SUCCESS; 139 } 140 141 int 142 main(int argc, char* const* argv) 143 { 144 return atf_config().run(argc, argv); 145 } 146