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