aboutsummaryrefslogtreecommitdiffstats
path: root/src/tests/test_xmss.cpp
blob: eb75daa385f95fb07c26e553a4ba66eaddf55eaa (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
/*
 * Extended Hash-Based Signatures Tests
 *
 * (C) 2014,2015 Jack Lloyd
 * (C) 2016 Matthias Gierlings
 **/

#include "tests.h"

#if defined(BOTAN_HAS_XMSS)
    #include <botan/xmss.h>
    #include "test_pubkey.h"
#endif

namespace Botan_Tests {

namespace {

#if defined(BOTAN_HAS_XMSS)

class XMSS_Signature_Tests : public PK_Signature_Generation_Test
   {
   public:
      XMSS_Signature_Tests() : PK_Signature_Generation_Test(
         "XMSS",
         "pubkey/xmss_sig.vec",
         {"Params", "Msg", "PrivateKey", "Signature"},
         {})
         {}

      std::string default_padding(const VarMap& vars) const override
         {
         return get_req_str(vars, "Params");
         }

      std::unique_ptr<Botan::Private_Key> load_private_key(const VarMap& vars) override
         {
         const std::vector<byte> raw_key = get_req_bin(vars, "PrivateKey");
         const Botan::secure_vector<byte> sec_key(raw_key.begin(), raw_key.end());

         std::unique_ptr<Botan::Private_Key> key(new Botan::XMSS_PrivateKey(sec_key));
         return key;
         }
   };

class XMSS_Signature_Verify_Tests : public PK_Signature_Verification_Test
   {
   public:
      XMSS_Signature_Verify_Tests() : PK_Signature_Verification_Test(
         "XMSS",
         "pubkey/xmss_verify.vec",
         {"Params", "Msg", "PublicKey", "Signature"},
         {})
         {}

      std::string default_padding(const VarMap& vars) const override
         {
         return get_req_str(vars, "Params");
         }

      std::unique_ptr<Botan::Public_Key> load_public_key(const VarMap& vars) override
         {
         const std::vector<byte> raw_key = get_req_bin(vars, "PublicKey");
         const Botan::secure_vector<byte> sec_key(raw_key.begin(), raw_key.end());

         std::unique_ptr<Botan::Public_Key> key(new Botan::XMSS_PublicKey(sec_key));
         return key;
         }
   };

class XMSS_Keygen_Tests : public PK_Key_Generation_Test
   {
   public:
      std::vector<std::string> keygen_params() const override
         {
         return { "XMSS_SHA2-256_W16_H10" };
         }
      std::string algo_name() const override { return "XMSS"; }
   };

BOTAN_REGISTER_TEST("xmss_sign", XMSS_Signature_Tests);
BOTAN_REGISTER_TEST("xmss_verify", XMSS_Signature_Verify_Tests);
BOTAN_REGISTER_TEST("xmss_keygen", XMSS_Keygen_Tests);

#endif

}

}