xref: /freebsd/contrib/kyua/integration/helpers/dump_env.cpp (revision e6bfd18d21b225af6a0ed67ceeaf1293b7b9eba5)
1 // Copyright 2015 The Kyua Authors.
2 // All rights reserved.
3 //
4 // Redistribution and use in source and binary forms, with or without
5 // modification, are permitted provided that the following conditions are
6 // met:
7 //
8 // * Redistributions of source code must retain the above copyright
9 //   notice, this list of conditions and the following disclaimer.
10 // * Redistributions in binary form must reproduce the above copyright
11 //   notice, this list of conditions and the following disclaimer in the
12 //   documentation and/or other materials provided with the distribution.
13 // * Neither the name of Google Inc. nor the names of its contributors
14 //   may be used to endorse or promote products derived from this software
15 //   without specific prior written permission.
16 //
17 // THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
18 // "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
19 // LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
20 // A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
21 // OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
22 // SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
23 // LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
24 // DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
25 // THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
26 // (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
27 // OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
28 
29 // Dumps all environment variables.
30 //
31 // This helper program allows comparing the printed environment variables
32 // to what 'kyua report --verbose' may output.  It does so by sorting the
33 // variables and allowing the caller to customize how the output looks
34 // like (indentation for each line and for continuation lines).
35 
36 #include <cstdlib>
37 #include <iostream>
38 
39 #include "utils/env.hpp"
40 #include "utils/text/operations.ipp"
41 
42 namespace text = utils::text;
43 
44 
45 int
46 main(const int argc, const char* const* const argv)
47 {
48     if (argc != 3) {
49         std::cerr << "Usage: dump_env <prefix> <continuation-prefix>\n";
50         return EXIT_FAILURE;
51     }
52     const char* prefix = argv[1];
53     const char* continuation_prefix = argv[2];
54 
55     const std::map< std::string, std::string > env = utils::getallenv();
56     for (std::map< std::string, std::string >::const_iterator
57              iter = env.begin(); iter != env.end(); ++iter) {
58         const std::string& name = (*iter).first;
59         const std::vector< std::string > value = text::split(
60             (*iter).second, '\n');
61 
62         if (value.empty()) {
63             std::cout << prefix << name << "=\n";
64         } else {
65             std::cout << prefix << name << '=' << value[0] << '\n';
66             for (std::vector< std::string >::size_type i = 1;
67                  i < value.size(); ++i) {
68                 std::cout << continuation_prefix << value[i] << '\n';
69             }
70         }
71     }
72 
73     return EXIT_SUCCESS;
74 }
75