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