library/cpp/digest/argonish/internal/proxies/macro/proxy_macros.h


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
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194

#pragma once

//
// Created by Evgeny Sidorov on 12/04/17.
//
/**
 * ZEROUPPER macro is only used for AVX2 instruction set to clear up the upper half of YMM registers
 * It's done to avoid performance penalty when CPU switches to non-AVX2 code (according to Agner)
 * and the post at https://software.intel.com/en-us/articles/intel-avx-state-transitions-migrating-sse-code-to-avx
 */

#define ARGON2_PROXY_CLASS_DECL(IS)                                                                                   \
    class TArgon2Proxy##IS final: public IArgon2Base {                                                                \
    public:                                                                                                           \
        TArgon2Proxy##IS(EArgon2Type atype, ui32 tcost, ui32 mcost, ui32 threads,                                     \
                         const ui8* key = nullptr, ui32 keylen = 0);                                                  \
        virtual ~TArgon2Proxy##IS();                                                                                  \
                                                                                                                      \
        virtual void Hash(const ui8* pwd, ui32 pwdlen, const ui8* salt, ui32 saltlen,                                 \
                          ui8* out, ui32 outlen, const ui8* aad = nullptr, ui32 aadlen = 0) const override;           \
        virtual bool Verify(const ui8* pwd, ui32 pwdlen, const ui8* salt, ui32 saltlen,                               \
                            const ui8* hash, ui32 hashlen, const ui8* aad = nullptr, ui32 aadlen = 0) const override; \
        virtual void HashWithCustomMemory(ui8* memory, size_t mlen, const ui8* pwd, ui32 pwdlen,                      \
                                          const ui8* salt, ui32 saltlen, ui8* out, ui32 outlen,                       \
                                          const ui8* aad = nullptr, ui32 aadlen = 0) const override;                  \
        virtual bool VerifyWithCustomMemory(ui8* memory, size_t mlen, const ui8* pwd, ui32 pwdlen,                    \
                                            const ui8* salt, ui32 saltlen, const ui8* hash, ui32 hashlen,             \
                                            const ui8* aad = nullptr, ui32 aadlen = 0) const override;                \
        virtual size_t GetMemorySize() const override;                                                                \
                                                                                                                      \
    protected:                                                                                                        \
        THolder<IArgon2Base> argon2;                                                                                  \
    };

#define ARGON2_INSTANCE_DECL(IS_val, mcost_val, threads_val)                                     \
    if (mcost == mcost_val && threads == threads_val) {                                          \
        argon2 = MakeHolder<TArgon2##IS_val<mcost_val, threads_val>>(atype, tcost, key, keylen); \
        return;                                                                                  \
    }

#define ARGON2_PROXY_CLASS_IMPL(IS)                                                                                           \
    TArgon2Proxy##IS::TArgon2Proxy##IS(EArgon2Type atype, ui32 tcost, ui32 mcost, ui32 threads,                               \
                                       const ui8* key, ui32 keylen) {                                                         \
        if ((key == nullptr && keylen > 0) || keylen > ARGON2_SECRET_MAX_LENGTH)                                              \
            ythrow yexception() << "key is null or keylen equals 0 or key is too long";                                       \
                                                                                                                              \
        ARGON2_INSTANCE_DECL(IS, 1, 1)                                                                                        \
        ARGON2_INSTANCE_DECL(IS, 8, 1)                                                                                        \
        ARGON2_INSTANCE_DECL(IS, 16, 1)                                                                                       \
        ARGON2_INSTANCE_DECL(IS, 32, 1)                                                                                       \
        ARGON2_INSTANCE_DECL(IS, 64, 1)                                                                                       \
        ARGON2_INSTANCE_DECL(IS, 128, 1)                                                                                       \
        ARGON2_INSTANCE_DECL(IS, 256, 1)                                                                                       \
        ARGON2_INSTANCE_DECL(IS, 512, 1)                                                                                      \
        ARGON2_INSTANCE_DECL(IS, 1024, 1)                                                                                     \
        ARGON2_INSTANCE_DECL(IS, 2048, 1)                                                                                     \
        ARGON2_INSTANCE_DECL(IS, 4096, 1)                                                                                     \
        ARGON2_INSTANCE_DECL(IS, 8192, 1)                                                                                     \
        ARGON2_INSTANCE_DECL(IS, 16384, 1)                                                                                    \
        ARGON2_INSTANCE_DECL(IS, 32768, 1)                                                                                    \
        ARGON2_INSTANCE_DECL(IS, 65536, 1)                                                                                    \
        ARGON2_INSTANCE_DECL(IS, 131072, 1)                                                                                   \
        ARGON2_INSTANCE_DECL(IS, 262144, 1)                                                                                   \
        ARGON2_INSTANCE_DECL(IS, 524288, 1)                                                                                   \
        ARGON2_INSTANCE_DECL(IS, 1048576, 1)                                                                                  \
        ARGON2_INSTANCE_DECL(IS, 1, 2)                                                                                        \
        ARGON2_INSTANCE_DECL(IS, 32, 2)                                                                                       \
        ARGON2_INSTANCE_DECL(IS, 64, 2)                                                                                       \
        ARGON2_INSTANCE_DECL(IS, 512, 2)                                                                                      \
        ARGON2_INSTANCE_DECL(IS, 1024, 2)                                                                                     \
        ARGON2_INSTANCE_DECL(IS, 2048, 2)                                                                                     \
        ARGON2_INSTANCE_DECL(IS, 4096, 2)                                                                                     \
        ARGON2_INSTANCE_DECL(IS, 8192, 2)                                                                                     \
        ARGON2_INSTANCE_DECL(IS, 16384, 2)                                                                                    \
        ARGON2_INSTANCE_DECL(IS, 32768, 2)                                                                                    \
        ARGON2_INSTANCE_DECL(IS, 65536, 2)                                                                                    \
        ARGON2_INSTANCE_DECL(IS, 131072, 2)                                                                                   \
        ARGON2_INSTANCE_DECL(IS, 262144, 2)                                                                                   \
        ARGON2_INSTANCE_DECL(IS, 524288, 2)                                                                                   \
        ARGON2_INSTANCE_DECL(IS, 1048576, 2)                                                                                  \
        ARGON2_INSTANCE_DECL(IS, 1, 4)                                                                                        \
        ARGON2_INSTANCE_DECL(IS, 32, 4)                                                                                       \
        ARGON2_INSTANCE_DECL(IS, 64, 4)                                                                                       \
        ARGON2_INSTANCE_DECL(IS, 512, 4)                                                                                      \
        ARGON2_INSTANCE_DECL(IS, 1024, 4)                                                                                     \
        ARGON2_INSTANCE_DECL(IS, 2048, 4)                                                                                     \
        ARGON2_INSTANCE_DECL(IS, 4096, 4)                                                                                     \
        ARGON2_INSTANCE_DECL(IS, 8192, 4)                                                                                     \
        ARGON2_INSTANCE_DECL(IS, 16384, 4)                                                                                    \
        ARGON2_INSTANCE_DECL(IS, 32768, 4)                                                                                    \
        ARGON2_INSTANCE_DECL(IS, 65536, 4)                                                                                    \
        ARGON2_INSTANCE_DECL(IS, 131072, 4)                                                                                   \
        ARGON2_INSTANCE_DECL(IS, 262144, 4)                                                                                   \
        ARGON2_INSTANCE_DECL(IS, 524288, 4)                                                                                   \
        ARGON2_INSTANCE_DECL(IS, 1048576, 4)                                                                                  \
                                                                                                                              \
        ythrow yexception() << "These parameters are not supported. Please add the corresponding ARGON2_INSTANCE_DECL macro"; \
    }                                                                                                                         \
                                                                                                                              \
    TArgon2Proxy##IS::~TArgon2Proxy##IS() {                                                                                   \
    }                                                                                                                         \
                                                                                                                              \
    void TArgon2Proxy##IS::Hash(const ui8* pwd, ui32 pwdlen, const ui8* salt, ui32 saltlen,                                   \
                                ui8* out, ui32 outlen, const ui8* aad, ui32 aadlen) const {                                   \
        if (saltlen < ARGON2_SALT_MIN_LEN)                                                                                    \
            ythrow yexception() << "salt is too short";                                                                       \
        if (outlen < ARGON2_MIN_OUTLEN)                                                                                       \
            ythrow yexception() << "output length is too short";                                                              \
                                                                                                                              \
        argon2->Hash(pwd, pwdlen, salt, saltlen, out, outlen, aad, aadlen);                                                   \
        ZEROUPPER                                                                                                             \
    }                                                                                                                         \
                                                                                                                              \
    bool TArgon2Proxy##IS::Verify(const ui8* pwd, ui32 pwdlen, const ui8* salt, ui32 saltlen,                                 \
                                  const ui8* hash, ui32 hashlen, const ui8* aad, ui32 aadlen) const {                         \
        if (saltlen < ARGON2_SALT_MIN_LEN)                                                                                    \
            ythrow yexception() << "salt is too short";                                                                       \
        if (hashlen < ARGON2_MIN_OUTLEN)                                                                                      \
            ythrow yexception() << "hash length is too short";                                                                \
                                                                                                                              \
        return argon2->Verify(pwd, pwdlen, salt, saltlen, hash, hashlen, aad, aadlen);                                        \
        ZEROUPPER                                                                                                             \
    }                                                                                                                         \
                                                                                                                              \
    void TArgon2Proxy##IS::HashWithCustomMemory(ui8* memory, size_t mlen, const ui8* pwd, ui32 pwdlen,                        \
                                                const ui8* salt, ui32 saltlen, ui8* out, ui32 outlen,                         \
                                                const ui8* aad, ui32 aadlen) const {                                          \
        if (saltlen < ARGON2_SALT_MIN_LEN)                                                                                    \
            ythrow yexception() << "salt is too short";                                                                       \
        if (outlen < ARGON2_MIN_OUTLEN)                                                                                       \
            ythrow yexception() << "output length is too short";                                                              \
                                                                                                                              \
        argon2->HashWithCustomMemory(memory, mlen, pwd, pwdlen, salt, saltlen, out, outlen, aad, aadlen);                     \
        ZEROUPPER                                                                                                             \
    }                                                                                                                         \
                                                                                                                              \
    bool TArgon2Proxy##IS::VerifyWithCustomMemory(ui8* memory, size_t mlen, const ui8* pwd, ui32 pwdlen,                      \
                                                  const ui8* salt, ui32 saltlen, const ui8* hash, ui32 hashlen,               \
                                                  const ui8* aad, ui32 aadlen) const {                                        \
        if (saltlen < ARGON2_SALT_MIN_LEN)                                                                                    \
            ythrow yexception() << "salt is too short";                                                                       \
        if (hashlen < ARGON2_MIN_OUTLEN)                                                                                      \
            ythrow yexception() << "hash length is too short";                                                                \
                                                                                                                              \
        return argon2->VerifyWithCustomMemory(memory, mlen, pwd, pwdlen, salt, saltlen, hash, hashlen, aad, aadlen);          \
        ZEROUPPER                                                                                                             \
    }                                                                                                                         \
                                                                                                                              \
    size_t TArgon2Proxy##IS::GetMemorySize() const {                                                                          \
        return argon2->GetMemorySize();                                                                                       \
    }

#define BLAKE2B_PROXY_CLASS_DECL(IS)                                                    \
    class TBlake2BProxy##IS final: public IBlake2Base {                                 \
    public:                                                                             \
        TBlake2BProxy##IS(size_t outlen, const void* key = nullptr, size_t keylen = 0); \
        virtual void Update(ui32 in) override;                                          \
        virtual void Update(const void* pin, size_t inlen) override;                    \
        virtual void Final(void* out, size_t outlen) override;                          \
                                                                                        \
    protected:                                                                          \
        THolder<IBlake2Base> blake2;                                                    \
    };

#define BLAKE2B_PROXY_CLASS_IMPL(IS)                                                      \
    TBlake2BProxy##IS::TBlake2BProxy##IS(size_t outlen, const void* key, size_t keylen) { \
        if (!outlen || outlen > BLAKE2B_OUTBYTES)                                         \
            ythrow yexception() << "outlen equals 0 or too long";                         \
                                                                                          \
        if (key == nullptr) {                                                             \
            blake2 = MakeHolder<TBlake2B<EInstructionSet::IS>>(outlen);                   \
            return;                                                                       \
        }                                                                                 \
                                                                                          \
        if (!key || !keylen || keylen > BLAKE2B_KEYBYTES)                                 \
            ythrow yexception() << "key is null or too long";                             \
                                                                                          \
        blake2 = MakeHolder<TBlake2B<EInstructionSet::IS>>(outlen, key, keylen);          \
    }                                                                                     \
                                                                                          \
    void TBlake2BProxy##IS::Update(ui32 in) {                                             \
        blake2->Update(in);                                                               \
        ZEROUPPER                                                                         \
    }                                                                                     \
                                                                                          \
    void TBlake2BProxy##IS::Update(const void* pin, size_t inlen) {                       \
        blake2->Update(pin, inlen);                                                       \
        ZEROUPPER                                                                         \
    }                                                                                     \
                                                                                          \
    void TBlake2BProxy##IS::Final(void* out, size_t outlen) {                             \
        blake2->Final(out, outlen);                                                       \
        ZEROUPPER                                                                         \
    }