aboutsummaryrefslogtreecommitdiffstats
path: root/src/cli/asn1.cpp
blob: 32cec2c253baf8c46cef4636aebf16c75f66720e (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
/*
* (C) 2014,2015 Jack Lloyd
*
* Botan is released under the Simplified BSD License (see license.txt)
*/

#include "cli.h"

#if defined(BOTAN_HAS_ASN1)

#include <botan/asn1_print.h>
#include <botan/data_src.h>

#if defined(BOTAN_HAS_PEM_CODEC)
  #include <botan/pem.h>
#endif

namespace Botan_CLI {

class ASN1_Printer final : public Command
   {
   public:
      ASN1_Printer() : Command("asn1print --skip-context-specific --print-limit=4096 --bin-limit=2048 --max-depth=64 --pem file") {}

      std::string group() const override
         {
         return "codec";
         }

      std::string description() const override
         {
         return "Decode and print file with ASN.1 Basic Encoding Rules (BER)";
         }

      bool first_n(const std::vector<uint8_t>& data, size_t n, uint8_t b)
         {
         if(data.size() < n)
            return false;

         for(size_t i = 0; i != n; ++i)
            if(data[i] != b)
               return false;

         return true;
         }

      void go() override
         {
         const std::string input = get_arg("file");
         const size_t print_limit = get_arg_sz("print-limit");
         const size_t bin_limit = get_arg_sz("bin-limit");
         const bool print_context_specific = flag_set("skip-context-specific") == false;
         const size_t max_depth = get_arg_sz("max-depth");

         const size_t value_column = 60;
         const size_t initial_level = 0;

         std::vector<uint8_t> file_contents = slurp_file(input);
         std::vector<uint8_t> data;

         if(flag_set("pem") ||
            (input.size() > 4 && input.substr(input.size() - 4) == ".pem") ||
            (file_contents.size() > 20 && first_n(file_contents, 5, '-')))
            {
#if defined(BOTAN_HAS_PEM_CODEC)
            std::string pem_label;
            Botan::DataSource_Memory src(file_contents);
            data = unlock(Botan::PEM_Code::decode(src, pem_label));
#else
            throw CLI_Error_Unsupported("PEM decoding not available in this build");
#endif
            }
         else
            {
            data.swap(file_contents);
            }

         Botan::ASN1_Pretty_Printer printer(print_limit, bin_limit, print_context_specific,
                                            initial_level, value_column, max_depth);

         printer.print_to_stream(output(), data.data(), data.size());
         }
   };

BOTAN_REGISTER_COMMAND("asn1print", ASN1_Printer);

}

#endif // BOTAN_HAS_ASN1 && BOTAN_HAS_PEM_CODEC