]> www.wagner.pp.ru Git - openssl-gost/engine.git/blob - gost_crypt.c
update magma cipher ctr_acpkm mode encrypting
[openssl-gost/engine.git] / gost_crypt.c
1 /**********************************************************************
2  *             gost_crypt.c - Initialize all ciphers                  *
3  *                                                                    *
4  *             Copyright (c) 2005-2006 Cryptocom LTD                  *
5  *             Copyright (c) 2020 Chikunov Vitaly <vt@altlinux.org>   *
6  *         This file is distributed under the same license as OpenSSL *
7  *                                                                    *
8  *       OpenSSL interface to GOST 28147-89 cipher functions          *
9  *          Requires OpenSSL 0.9.9 for compilation                    *
10  **********************************************************************/
11 #include <string.h>
12 #include "gost89.h"
13 #include <openssl/err.h>
14 #include <openssl/rand.h>
15 #include "e_gost_err.h"
16 #include "gost_lcl.h"
17 #include "gost_gost2015.h"
18
19 #if !defined(CCGOST_DEBUG) && !defined(DEBUG)
20 # ifndef NDEBUG
21 #  define NDEBUG
22 # endif
23 #endif
24 #include <assert.h>
25
26 static int gost_cipher_init(EVP_CIPHER_CTX *ctx, const unsigned char *key,
27                             const unsigned char *iv, int enc);
28 static int gost_cipher_init_cbc(EVP_CIPHER_CTX *ctx, const unsigned char *key,
29                                 const unsigned char *iv, int enc);
30 static int gost_cipher_init_cpa(EVP_CIPHER_CTX *ctx, const unsigned char *key,
31                                 const unsigned char *iv, int enc);
32 static int gost_cipher_init_cp_12(EVP_CIPHER_CTX *ctx,
33                                   const unsigned char *key,
34                                   const unsigned char *iv, int enc);
35 /* Handles block of data in CFB mode */
36 static int gost_cipher_do_cfb(EVP_CIPHER_CTX *ctx, unsigned char *out,
37                               const unsigned char *in, size_t inl);
38 /* Handles block of data in CBC mode */
39 static int gost_cipher_do_cbc(EVP_CIPHER_CTX *ctx, unsigned char *out,
40                               const unsigned char *in, size_t inl);
41 /* Handles block of data in CNT mode */
42 static int gost_cipher_do_cnt(EVP_CIPHER_CTX *ctx, unsigned char *out,
43                               const unsigned char *in, size_t inl);
44 /* Cleanup function */
45 static int gost_cipher_cleanup(EVP_CIPHER_CTX *);
46 /* set/get cipher parameters */
47 static int gost89_set_asn1_parameters(EVP_CIPHER_CTX *ctx, ASN1_TYPE *params);
48 static int gost89_get_asn1_parameters(EVP_CIPHER_CTX *ctx, ASN1_TYPE *params);
49 /* Control function */
50 static int gost_cipher_ctl(EVP_CIPHER_CTX *ctx, int type, int arg, void *ptr);
51
52 static int magma_cipher_init(EVP_CIPHER_CTX *ctx, const unsigned char *key,
53                              const unsigned char *iv, int enc);
54 static int magma_cipher_init_ctr_acpkm_omac(EVP_CIPHER_CTX *ctx, const unsigned char *key,
55                              const unsigned char *iv, int enc);
56 /* Handles block of data in CBC mode */
57 static int magma_cipher_do_cbc(EVP_CIPHER_CTX *ctx, unsigned char *out,
58                                const unsigned char *in, size_t inl);
59 static int magma_cipher_do_ctr(EVP_CIPHER_CTX *ctx, unsigned char *out,
60                                const unsigned char *in, size_t inl);
61
62 static int magma_cipher_do_ctr_acpkm_omac(EVP_CIPHER_CTX *ctx, unsigned char *out,
63                                const unsigned char *in, size_t inl);
64
65 /* set/get cipher parameters */
66 static int magma_set_asn1_parameters(EVP_CIPHER_CTX *ctx, ASN1_TYPE *params);
67 static int magma_get_asn1_parameters(EVP_CIPHER_CTX *ctx, ASN1_TYPE *params);
68 /* Control function */
69 static int magma_cipher_ctl(EVP_CIPHER_CTX *ctx, int type, int arg, void *ptr);
70 static int magma_cipher_ctl_acpkm_omac(EVP_CIPHER_CTX *ctx, int type, int arg, void *ptr);
71
72 /*
73  * Single level template accessor.
74  * Note: that you cannot template 0 value.
75  */
76 #define TPL(st,field) ( \
77     ((st)->field) ?: TPL_VAL(st,field) \
78 )
79
80 #define TPL_VAL(st,field) ( \
81     ((st)->template ? (st)->template->field : 0) \
82 )
83
84 EVP_CIPHER *GOST_init_cipher(GOST_cipher *c)
85 {
86     if (c->cipher)
87         return c->cipher;
88
89     /* Some sanity checking. */
90     int flags = c->flags | TPL_VAL(c, flags);
91     int block_size = TPL(c, block_size);
92     switch (flags & EVP_CIPH_MODE) {
93     case EVP_CIPH_CTR_MODE:
94     case EVP_CIPH_CFB_MODE:
95     case EVP_CIPH_OFB_MODE:
96         OPENSSL_assert(block_size == 1);
97         OPENSSL_assert(flags & EVP_CIPH_NO_PADDING);
98         break;
99     default:
100         OPENSSL_assert(block_size != 1);
101         OPENSSL_assert(!(flags & EVP_CIPH_NO_PADDING));
102     }
103
104     if (TPL(c, iv_len))
105         OPENSSL_assert(flags & EVP_CIPH_CUSTOM_IV);
106     else
107         OPENSSL_assert(!(flags & EVP_CIPH_CUSTOM_IV));
108
109     EVP_CIPHER *cipher;
110     if (!(cipher = EVP_CIPHER_meth_new(c->nid, block_size, TPL(c, key_len)))
111         || !EVP_CIPHER_meth_set_iv_length(cipher, TPL(c, iv_len))
112         || !EVP_CIPHER_meth_set_flags(cipher, flags)
113         || !EVP_CIPHER_meth_set_init(cipher, TPL(c, init))
114         || !EVP_CIPHER_meth_set_do_cipher(cipher, TPL(c, do_cipher))
115         || !EVP_CIPHER_meth_set_cleanup(cipher, TPL(c, cleanup))
116         || !EVP_CIPHER_meth_set_impl_ctx_size(cipher, TPL(c, ctx_size))
117         || !EVP_CIPHER_meth_set_set_asn1_params(cipher, TPL(c, set_asn1_parameters))
118         || !EVP_CIPHER_meth_set_get_asn1_params(cipher, TPL(c, get_asn1_parameters))
119         || !EVP_CIPHER_meth_set_ctrl(cipher, TPL(c, ctrl))) {
120         EVP_CIPHER_meth_free(cipher);
121         cipher = NULL;
122     }
123     c->cipher = cipher;
124     return c->cipher;
125 }
126
127 void GOST_deinit_cipher(GOST_cipher *c)
128 {
129     if (c->cipher) {
130         EVP_CIPHER_meth_free(c->cipher);
131         c->cipher = NULL;
132     }
133 }
134
135 static GOST_cipher gost_template_cipher = {
136     .block_size = 8,
137     .key_len = 32,
138     .iv_len = 8,
139     .flags = EVP_CIPH_CUSTOM_IV |
140         EVP_CIPH_RAND_KEY |
141         EVP_CIPH_ALWAYS_CALL_INIT,
142     .cleanup = gost_cipher_cleanup,
143     .ctx_size = sizeof(struct ossl_gost_cipher_ctx),
144     .set_asn1_parameters = gost89_set_asn1_parameters,
145     .get_asn1_parameters = gost89_get_asn1_parameters,
146     .ctrl = gost_cipher_ctl,
147 };
148
149 GOST_cipher Gost28147_89_cipher = {
150     .nid = NID_id_Gost28147_89,
151     .template = &gost_template_cipher,
152     .block_size = 1,
153     .flags = EVP_CIPH_CFB_MODE |
154         EVP_CIPH_NO_PADDING,
155     .init = gost_cipher_init,
156     .do_cipher = gost_cipher_do_cfb,
157 };
158
159 GOST_cipher Gost28147_89_cbc_cipher = {
160     .nid = NID_gost89_cbc,
161     .template = &gost_template_cipher,
162     .flags = EVP_CIPH_CBC_MODE,
163     .init = gost_cipher_init_cbc,
164     .do_cipher = gost_cipher_do_cbc,
165 };
166
167 GOST_cipher Gost28147_89_cnt_cipher = {
168     .nid = NID_gost89_cnt,
169     .template = &gost_template_cipher,
170     .block_size = 1,
171     .flags = EVP_CIPH_OFB_MODE |
172         EVP_CIPH_NO_PADDING,
173     .init = gost_cipher_init_cpa,
174     .do_cipher = gost_cipher_do_cnt,
175 };
176
177 GOST_cipher Gost28147_89_cnt_12_cipher = {
178     .nid = NID_gost89_cnt_12,
179     .template = &gost_template_cipher,
180     .block_size = 1,
181     .flags = EVP_CIPH_OFB_MODE |
182         EVP_CIPH_NO_PADDING,
183     .init = gost_cipher_init_cp_12,
184     .do_cipher = gost_cipher_do_cnt,
185 };
186
187 static GOST_cipher magma_template_cipher = {
188     .block_size = 8,
189     .key_len = 32,
190     .iv_len = 8,
191     .flags = EVP_CIPH_CUSTOM_IV |
192         EVP_CIPH_RAND_KEY |
193         EVP_CIPH_ALWAYS_CALL_INIT,
194     .cleanup = gost_cipher_cleanup,
195     .ctx_size = sizeof(struct ossl_gost_cipher_ctx),
196     .set_asn1_parameters = magma_set_asn1_parameters,
197     .get_asn1_parameters = magma_get_asn1_parameters,
198     .do_cipher = magma_cipher_do_ctr,
199     .ctrl = magma_cipher_ctl,
200 };
201
202 GOST_cipher magma_ctr_cipher = {
203     .nid = NID_magma_ctr,
204     .template = &magma_template_cipher,
205     .block_size = 1,
206     .iv_len = 4,
207     .flags = EVP_CIPH_CTR_MODE |
208         EVP_CIPH_NO_PADDING,
209     .init = magma_cipher_init,
210 };
211
212 GOST_cipher magma_ctr_acpkm_cipher = {
213     .nid = NID_magma_ctr_acpkm,
214     .template = &magma_template_cipher,
215     .block_size = 1,
216     .iv_len = 4,
217     .flags = EVP_CIPH_CTR_MODE |
218         EVP_CIPH_NO_PADDING,
219     .init = magma_cipher_init,
220 };
221
222 GOST_cipher magma_ctr_acpkm_omac_cipher = {
223     .nid = NID_magma_ctr_acpkm_omac,
224     .template = &magma_template_cipher,
225     .block_size = 1,
226     .iv_len = 4,
227     .flags = EVP_CIPH_CTR_MODE |
228         EVP_CIPH_NO_PADDING |
229         EVP_CIPH_CUSTOM_COPY |
230         EVP_CIPH_FLAG_CUSTOM_CIPHER |
231         EVP_CIPH_FLAG_CIPHER_WITH_MAC,
232     .init = magma_cipher_init_ctr_acpkm_omac,
233     .do_cipher = magma_cipher_do_ctr_acpkm_omac,
234     .ctrl = magma_cipher_ctl_acpkm_omac,
235 };
236
237 GOST_cipher magma_cbc_cipher = {
238     .nid = NID_magma_cbc,
239     .template = &gost_template_cipher,
240     .flags = EVP_CIPH_CBC_MODE,
241     .init = magma_cipher_init,
242     .do_cipher = magma_cipher_do_cbc,
243 };
244
245 /* Implementation of GOST 28147-89 in MAC (imitovstavka) mode */
246 /* Init functions which set specific parameters */
247 static int gost_imit_init_cpa(EVP_MD_CTX *ctx);
248 static int gost_imit_init_cp_12(EVP_MD_CTX *ctx);
249 /* process block of data */
250 static int gost_imit_update(EVP_MD_CTX *ctx, const void *data, size_t count);
251 /* Return computed value */
252 static int gost_imit_final(EVP_MD_CTX *ctx, unsigned char *md);
253 /* Copies context */
254 static int gost_imit_copy(EVP_MD_CTX *to, const EVP_MD_CTX *from);
255 static int gost_imit_cleanup(EVP_MD_CTX *ctx);
256 /* Control function, knows how to set MAC key.*/
257 static int gost_imit_ctrl(EVP_MD_CTX *ctx, int type, int arg, void *ptr);
258
259 GOST_digest Gost28147_89_MAC_digest = {
260     .nid = NID_id_Gost28147_89_MAC,
261     .result_size = 4,
262     .input_blocksize = 8,
263     .app_datasize = sizeof(struct ossl_gost_imit_ctx),
264     .flags = EVP_MD_FLAG_XOF,
265     .init = gost_imit_init_cpa,
266     .update = gost_imit_update,
267     .final = gost_imit_final,
268     .copy = gost_imit_copy,
269     .cleanup = gost_imit_cleanup,
270     .ctrl = gost_imit_ctrl,
271 };
272
273 GOST_digest Gost28147_89_mac_12_digest = {
274     .nid = NID_gost_mac_12,
275     .result_size = 4,
276     .input_blocksize = 8,
277     .app_datasize = sizeof(struct ossl_gost_imit_ctx),
278     .flags = EVP_MD_FLAG_XOF,
279     .init = gost_imit_init_cp_12,
280     .update = gost_imit_update,
281     .final = gost_imit_final,
282     .copy = gost_imit_copy,
283     .cleanup = gost_imit_cleanup,
284     .ctrl = gost_imit_ctrl,
285 };
286
287 /*
288  * Correspondence between gost parameter OIDs and substitution blocks
289  * NID field is filed by register_gost_NID function in engine.c
290  * upon engine initialization
291  */
292
293 static struct gost_cipher_info gost_cipher_list[] = {
294     /*- NID *//*
295      * Subst block
296      *//*
297      * Key meshing
298      */
299     /*
300      * {NID_id_GostR3411_94_CryptoProParamSet,&GostR3411_94_CryptoProParamSet,0},
301      */
302     {NID_id_Gost28147_89_CryptoPro_A_ParamSet, &Gost28147_CryptoProParamSetA,
303      1},
304     {NID_id_Gost28147_89_CryptoPro_B_ParamSet, &Gost28147_CryptoProParamSetB,
305      1},
306     {NID_id_Gost28147_89_CryptoPro_C_ParamSet, &Gost28147_CryptoProParamSetC,
307      1},
308     {NID_id_Gost28147_89_CryptoPro_D_ParamSet, &Gost28147_CryptoProParamSetD,
309      1},
310     {NID_id_tc26_gost_28147_param_Z, &Gost28147_TC26ParamSetZ, 1},
311     {NID_id_Gost28147_89_TestParamSet, &Gost28147_TestParamSet, 1},
312     {NID_undef, NULL, 0}
313 };
314
315 /*
316  * get encryption parameters from crypto network settings FIXME For now we
317  * use environment var CRYPT_PARAMS as place to store these settings.
318  * Actually, it is better to use engine control command, read from
319  * configuration file to set them
320  */
321 const struct gost_cipher_info *get_encryption_params(ASN1_OBJECT *obj)
322 {
323     int nid;
324     struct gost_cipher_info *param;
325     if (!obj) {
326         const char *params = get_gost_engine_param(GOST_PARAM_CRYPT_PARAMS);
327         if (!params || !strlen(params)) {
328             int i;
329             for (i = 0; gost_cipher_list[i].nid != NID_undef; i++)
330                 if (gost_cipher_list[i].nid == NID_id_tc26_gost_28147_param_Z)
331                     return &gost_cipher_list[i];
332             return &gost_cipher_list[0];
333         }
334
335         nid = OBJ_txt2nid(params);
336         if (nid == NID_undef) {
337             GOSTerr(GOST_F_GET_ENCRYPTION_PARAMS,
338                     GOST_R_INVALID_CIPHER_PARAM_OID);
339             ERR_add_error_data(3, "Unsupported CRYPT_PARAMS='",
340                 params, "' specified in environment or in config");
341             return NULL;
342         }
343     } else {
344         nid = OBJ_obj2nid(obj);
345     }
346     for (param = gost_cipher_list; param->sblock != NULL && param->nid != nid;
347          param++) ;
348     if (!param->sblock) {
349         GOSTerr(GOST_F_GET_ENCRYPTION_PARAMS, GOST_R_INVALID_CIPHER_PARAMS);
350         return NULL;
351     }
352     return param;
353 }
354
355 /* Sets cipher param from paramset NID. */
356 static int gost_cipher_set_param(struct ossl_gost_cipher_ctx *c, int nid)
357 {
358     const struct gost_cipher_info *param;
359     param = get_encryption_params((nid == NID_undef ? NULL : OBJ_nid2obj(nid)));
360     if (!param)
361         return 0;
362
363     c->paramNID = param->nid;
364     c->key_meshing = param->key_meshing;
365     c->count = 0;
366     gost_init(&(c->cctx), param->sblock);
367     return 1;
368 }
369
370 /* Initializes EVP_CIPHER_CTX by paramset NID */
371 static int gost_cipher_init_param(EVP_CIPHER_CTX *ctx,
372                                   const unsigned char *key,
373                                   const unsigned char *iv, int enc,
374                                   int paramNID, int mode)
375 {
376     struct ossl_gost_cipher_ctx *c = EVP_CIPHER_CTX_get_cipher_data(ctx);
377     if (EVP_CIPHER_CTX_get_app_data(ctx) == NULL) {
378         if (!gost_cipher_set_param(c, paramNID))
379             return 0;
380         EVP_CIPHER_CTX_set_app_data(ctx, EVP_CIPHER_CTX_get_cipher_data(ctx));
381     }
382     if (key)
383         gost_key(&(c->cctx), key);
384     if (iv) {
385         memcpy((unsigned char *)EVP_CIPHER_CTX_original_iv(ctx), iv,
386                EVP_CIPHER_CTX_iv_length(ctx));
387     }
388     memcpy(EVP_CIPHER_CTX_iv_noconst(ctx),
389            EVP_CIPHER_CTX_original_iv(ctx), EVP_CIPHER_CTX_iv_length(ctx));
390     return 1;
391 }
392
393 static int gost_cipher_init_cnt(EVP_CIPHER_CTX *ctx,
394                                 const unsigned char *key,
395                                 const unsigned char *iv,
396                                 gost_subst_block * block)
397 {
398     struct ossl_gost_cipher_ctx *c = EVP_CIPHER_CTX_get_cipher_data(ctx);
399     gost_init(&(c->cctx), block);
400     c->key_meshing = 1;
401     c->count = 0;
402     if (key)
403         gost_key(&(c->cctx), key);
404     if (iv) {
405         memcpy((unsigned char *)EVP_CIPHER_CTX_original_iv(ctx), iv,
406                EVP_CIPHER_CTX_iv_length(ctx));
407     }
408     memcpy(EVP_CIPHER_CTX_iv_noconst(ctx),
409            EVP_CIPHER_CTX_original_iv(ctx), EVP_CIPHER_CTX_iv_length(ctx));
410     return 1;
411 }
412
413 static int gost_cipher_init_cpa(EVP_CIPHER_CTX *ctx, const unsigned char *key,
414                                 const unsigned char *iv, int enc)
415 {
416     return gost_cipher_init_cnt(ctx, key, iv, &Gost28147_CryptoProParamSetA);
417 }
418
419 static int gost_cipher_init_cp_12(EVP_CIPHER_CTX *ctx,
420                                   const unsigned char *key,
421                                   const unsigned char *iv, int enc)
422 {
423     return gost_cipher_init_cnt(ctx, key, iv, &Gost28147_TC26ParamSetZ);
424 }
425
426 /* Initializes EVP_CIPHER_CTX with default values */
427 static int gost_cipher_init(EVP_CIPHER_CTX *ctx, const unsigned char *key,
428                      const unsigned char *iv, int enc)
429 {
430     return gost_cipher_init_param(ctx, key, iv, enc, NID_undef,
431                                   EVP_CIPH_CFB_MODE);
432 }
433
434 /* Initializes EVP_CIPHER_CTX with default values */
435 static int gost_cipher_init_cbc(EVP_CIPHER_CTX *ctx, const unsigned char *key,
436                          const unsigned char *iv, int enc)
437 {
438     return gost_cipher_init_param(ctx, key, iv, enc, NID_undef,
439                                   EVP_CIPH_CBC_MODE);
440 }
441
442 /* Initializes EVP_CIPHER_CTX with default values */
443 static int magma_cipher_init(EVP_CIPHER_CTX *ctx, const unsigned char *key,
444                       const unsigned char *iv, int enc)
445 {
446     struct ossl_gost_cipher_ctx *c = EVP_CIPHER_CTX_get_cipher_data(ctx);
447     /* FIXME this is just initializtion check */
448     if (EVP_CIPHER_CTX_get_app_data(ctx) == NULL) {
449         if (!gost_cipher_set_param(c, NID_id_tc26_gost_28147_param_Z))
450             return 0;
451         EVP_CIPHER_CTX_set_app_data(ctx, EVP_CIPHER_CTX_get_cipher_data(ctx));
452
453         if (enc) {
454             if (init_zero_kdf_seed(c->kdf_seed) == 0)
455                 return -1;
456         }
457     }
458
459     if (key)
460         magma_key(&(c->cctx), key);
461     if (iv) {
462         memcpy((unsigned char *)EVP_CIPHER_CTX_original_iv(ctx), iv,
463                EVP_CIPHER_CTX_iv_length(ctx));
464     }
465     memcpy(EVP_CIPHER_CTX_iv_noconst(ctx),
466            EVP_CIPHER_CTX_original_iv(ctx), EVP_CIPHER_CTX_iv_length(ctx));
467
468     if (EVP_CIPHER_CTX_nid(ctx) == NID_magma_ctr_acpkm
469      || EVP_CIPHER_CTX_nid(ctx) == NID_magma_ctr_acpkm_omac) {
470        c->key_meshing = 1024;
471     } else {
472        c->key_meshing = 0;
473     }
474
475     return 1;
476 }
477
478 /* Initializes EVP_CIPHER_CTX with default values */
479 static int magma_cipher_init_ctr_acpkm_omac(EVP_CIPHER_CTX *ctx, const unsigned char *key,
480                       const unsigned char *iv, int enc)
481 {
482         if (key) {
483     struct ossl_gost_cipher_ctx *c = EVP_CIPHER_CTX_get_cipher_data(ctx);
484                 unsigned char cipher_key[32];
485                 c->omac_ctx = EVP_MD_CTX_new();
486
487                 if (c->omac_ctx == NULL) {
488                     GOSTerr(GOST_F_MAGMA_CIPHER_INIT_CTR_ACPKM_OMAC, ERR_R_MALLOC_FAILURE);
489                                 return 0;
490                 }
491
492                 if (gost2015_acpkm_omac_init(NID_magma_mac, enc, key,
493                                  c->omac_ctx, cipher_key, c->kdf_seed) != 1) {
494                     EVP_MD_CTX_free(c->omac_ctx);
495                                 c->omac_ctx = NULL;
496                     return 0;
497                 }
498
499                 return magma_cipher_init(ctx, cipher_key, iv, enc);
500         }
501
502         return magma_cipher_init(ctx, key, iv, enc);
503 }
504
505 /*
506  * Wrapper around gostcrypt function from gost89.c which perform key meshing
507  * when nesseccary
508  */
509 static void gost_crypt_mesh(void *ctx, unsigned char *iv, unsigned char *buf)
510 {
511     struct ossl_gost_cipher_ctx *c = ctx;
512     assert(c->count % 8 == 0 && c->count <= 1024);
513     if (c->key_meshing && c->count == 1024) {
514         cryptopro_key_meshing(&(c->cctx), iv);
515     }
516     gostcrypt(&(c->cctx), iv, buf);
517     c->count = c->count % 1024 + 8;
518 }
519
520 static void gost_cnt_next(void *ctx, unsigned char *iv, unsigned char *buf)
521 {
522     struct ossl_gost_cipher_ctx *c = ctx;
523     word32 g, go;
524     unsigned char buf1[8];
525     assert(c->count % 8 == 0 && c->count <= 1024);
526     if (c->key_meshing && c->count == 1024) {
527         cryptopro_key_meshing(&(c->cctx), iv);
528     }
529     if (c->count == 0) {
530         gostcrypt(&(c->cctx), iv, buf1);
531     } else {
532         memcpy(buf1, iv, 8);
533     }
534     g = buf1[0] | (buf1[1] << 8) | (buf1[2] << 16) | ((word32) buf1[3] << 24);
535     g += 0x01010101;
536     buf1[0] = (unsigned char)(g & 0xff);
537     buf1[1] = (unsigned char)((g >> 8) & 0xff);
538     buf1[2] = (unsigned char)((g >> 16) & 0xff);
539     buf1[3] = (unsigned char)((g >> 24) & 0xff);
540     g = buf1[4] | (buf1[5] << 8) | (buf1[6] << 16) | ((word32) buf1[7] << 24);
541     go = g;
542     g += 0x01010104;
543     if (go > g)                 /* overflow */
544         g++;
545     buf1[4] = (unsigned char)(g & 0xff);
546     buf1[5] = (unsigned char)((g >> 8) & 0xff);
547     buf1[6] = (unsigned char)((g >> 16) & 0xff);
548     buf1[7] = (unsigned char)((g >> 24) & 0xff);
549     memcpy(iv, buf1, 8);
550     gostcrypt(&(c->cctx), buf1, buf);
551     c->count = c->count % 1024 + 8;
552 }
553
554 /* GOST encryption in CBC mode */
555 static int gost_cipher_do_cbc(EVP_CIPHER_CTX *ctx, unsigned char *out,
556                        const unsigned char *in, size_t inl)
557 {
558     unsigned char b[8];
559     const unsigned char *in_ptr = in;
560     unsigned char *out_ptr = out;
561     int i;
562     struct ossl_gost_cipher_ctx *c = EVP_CIPHER_CTX_get_cipher_data(ctx);
563     unsigned char *iv = EVP_CIPHER_CTX_iv_noconst(ctx);
564     if (EVP_CIPHER_CTX_encrypting(ctx)) {
565         while (inl > 0) {
566
567             for (i = 0; i < 8; i++) {
568                 b[i] = iv[i] ^ in_ptr[i];
569             }
570             gostcrypt(&(c->cctx), b, out_ptr);
571             memcpy(iv, out_ptr, 8);
572             out_ptr += 8;
573             in_ptr += 8;
574             inl -= 8;
575         }
576     } else {
577         while (inl > 0) {
578             unsigned char tmpiv[8];
579             gostdecrypt(&(c->cctx), in_ptr, b);
580             memcpy(tmpiv, in_ptr, 8);
581             for (i = 0; i < 8; i++) {
582                 out_ptr[i] = iv[i] ^ b[i];
583             }
584             memcpy(iv, tmpiv, 8);
585             out_ptr += 8;
586             in_ptr += 8;
587             inl -= 8;
588         }
589     }
590     return 1;
591 }
592
593 /* MAGMA encryption in CBC mode */
594 static int magma_cipher_do_cbc(EVP_CIPHER_CTX *ctx, unsigned char *out,
595                         const unsigned char *in, size_t inl)
596 {
597     unsigned char b[8];
598     unsigned char d[8];
599     const unsigned char *in_ptr = in;
600     unsigned char *out_ptr = out;
601     int i;
602     struct ossl_gost_cipher_ctx *c = EVP_CIPHER_CTX_get_cipher_data(ctx);
603     unsigned char *iv = EVP_CIPHER_CTX_iv_noconst(ctx);
604     if (EVP_CIPHER_CTX_encrypting(ctx)) {
605         while (inl > 0) {
606
607             for (i = 0; i < 8; i++) {
608                 b[7 - i] = iv[i] ^ in_ptr[i];
609             }
610             gostcrypt(&(c->cctx), b, d);
611
612             for (i = 0; i < 8; i++) {
613                 out_ptr[7 - i] = d[i];
614             }
615             memcpy(iv, out_ptr, 8);
616             out_ptr += 8;
617             in_ptr += 8;
618             inl -= 8;
619         }
620     } else {
621         while (inl > 0) {
622             for (i = 0; i < 8; i++) {
623                 d[7 - i] = in_ptr[i];
624             }
625             gostdecrypt(&(c->cctx), d, b);
626             memcpy(d, in_ptr, 8);
627             for (i = 0; i < 8; i++) {
628                 out_ptr[i] = iv[i] ^ b[7 - i];
629             }
630             memcpy(iv, d, 8);
631             out_ptr += 8;
632             in_ptr += 8;
633             inl -= 8;
634         }
635     }
636     return 1;
637 }
638
639 /* increment counter (64-bit int) by 1 */
640 static void ctr64_inc(unsigned char *counter)
641 {
642     inc_counter(counter, 8);
643 }
644
645 #define MAGMA_BLOCK_SIZE 8
646 #define MAGMA_BLOCK_MASK (MAGMA_BLOCK_SIZE - 1)
647 static inline void apply_acpkm_magma(struct ossl_gost_cipher_ctx *
648                                            ctx, unsigned int *num)
649 {
650     if (!ctx->key_meshing || (*num < ctx->key_meshing))
651         return;
652     acpkm_magma_key_meshing(&ctx->cctx);
653     *num &= MAGMA_BLOCK_MASK;
654 }
655
656 /* MAGMA encryption in CTR mode */
657 static int magma_cipher_do_ctr(EVP_CIPHER_CTX *ctx, unsigned char *out,
658                                const unsigned char *in, size_t inl)
659 {
660     const unsigned char *in_ptr = in;
661     unsigned char *out_ptr = out;
662     size_t j;
663     struct ossl_gost_cipher_ctx *c = EVP_CIPHER_CTX_get_cipher_data(ctx);
664     unsigned char *buf = EVP_CIPHER_CTX_buf_noconst(ctx);
665     unsigned char *iv = EVP_CIPHER_CTX_iv_noconst(ctx);
666     unsigned int num = EVP_CIPHER_CTX_num(ctx);
667     size_t blocks, i, lasted = inl;
668     unsigned char b[8];
669 /* Process partial blocks */
670     while ((num & MAGMA_BLOCK_MASK) && lasted) {
671         *out_ptr++ = *in_ptr++ ^ buf[7 - (num & MAGMA_BLOCK_MASK)];
672         --lasted;
673         num++;
674     }
675     blocks = lasted / MAGMA_BLOCK_SIZE;
676
677 /* Process full blocks */
678     for (i = 0; i < blocks; i++) {
679         apply_acpkm_magma(c, &num);
680         for (j = 0; j < 8; j++) {
681             b[7 - j] = iv[j];
682         }
683         gostcrypt(&(c->cctx), b, buf);
684         for (j = 0; j < 8; j++) {
685             out_ptr[j] = buf[7 - j] ^ in_ptr[j];
686         }
687         ctr64_inc(iv);
688         c->count += MAGMA_BLOCK_SIZE;
689         in_ptr += MAGMA_BLOCK_SIZE;
690         out_ptr += MAGMA_BLOCK_SIZE;
691         num += MAGMA_BLOCK_SIZE;
692         lasted -= MAGMA_BLOCK_SIZE;
693     }
694
695 /* Process the rest of plaintext */
696     if (lasted > 0) {
697         apply_acpkm_magma(c, &num);
698         for (j = 0; j < 8; j++) {
699             b[7 - j] = iv[j];
700         }
701         gostcrypt(&(c->cctx), b, buf);
702
703         for (i = 0; i < lasted; i++)
704             out_ptr[i] = buf[7 - i] ^ in_ptr[i];
705         ctr64_inc(iv);
706         c->count += j;
707         num += lasted;
708     }
709     EVP_CIPHER_CTX_set_num(ctx, num);
710
711     return inl;
712 }
713
714 /* MAGMA encryption in CTR mode */
715 static int magma_cipher_do_ctr_acpkm_omac(EVP_CIPHER_CTX *ctx, unsigned char *out,
716                                const unsigned char *in, size_t inl)
717 {
718   struct ossl_gost_cipher_ctx *c = EVP_CIPHER_CTX_get_cipher_data(ctx);
719
720         if (in == NULL && inl == 0) /* Final call */
721                 return gost2015_final_call(ctx, c->omac_ctx, MAGMA_MAC_MAX_SIZE, c->tag, magma_cipher_do_ctr);
722
723   if (in == NULL)
724       return -1;
725
726         /* As in and out can be the same pointer, process unencrypted here */
727         if (EVP_CIPHER_CTX_encrypting(ctx))
728                 EVP_DigestSignUpdate(c->omac_ctx, in, inl);
729
730   if (magma_cipher_do_ctr(ctx, out, in, inl) != inl)
731       return -1;
732
733         /* As in and out can be the same pointer, process decrypted here */
734         if (!EVP_CIPHER_CTX_encrypting(ctx))
735                 EVP_DigestSignUpdate(c->omac_ctx, out, inl);
736
737         return inl;
738 }
739 /* GOST encryption in CFB mode */
740 static int gost_cipher_do_cfb(EVP_CIPHER_CTX *ctx, unsigned char *out,
741                        const unsigned char *in, size_t inl)
742 {
743     const unsigned char *in_ptr = in;
744     unsigned char *out_ptr = out;
745     size_t i = 0;
746     size_t j = 0;
747     unsigned char *buf = EVP_CIPHER_CTX_buf_noconst(ctx);
748     unsigned char *iv = EVP_CIPHER_CTX_iv_noconst(ctx);
749 /* process partial block if any */
750     if (EVP_CIPHER_CTX_num(ctx)) {
751         for (j = EVP_CIPHER_CTX_num(ctx), i = 0; j < 8 && i < inl;
752              j++, i++, in_ptr++, out_ptr++) {
753             if (!EVP_CIPHER_CTX_encrypting(ctx))
754                 buf[j + 8] = *in_ptr;
755             *out_ptr = buf[j] ^ (*in_ptr);
756             if (EVP_CIPHER_CTX_encrypting(ctx))
757                 buf[j + 8] = *out_ptr;
758         }
759         if (j == 8) {
760             memcpy(iv, buf + 8, 8);
761             EVP_CIPHER_CTX_set_num(ctx, 0);
762         } else {
763             EVP_CIPHER_CTX_set_num(ctx, j);
764             return 1;
765         }
766     }
767
768     for (; i + 8 < inl; i += 8, in_ptr += 8, out_ptr += 8) {
769         /*
770          * block cipher current iv
771          */
772         gost_crypt_mesh(EVP_CIPHER_CTX_get_cipher_data(ctx), iv, buf);
773         /*
774          * xor next block of input text with it and output it
775          */
776         /*
777          * output this block
778          */
779         if (!EVP_CIPHER_CTX_encrypting(ctx))
780             memcpy(iv, in_ptr, 8);
781         for (j = 0; j < 8; j++) {
782             out_ptr[j] = buf[j] ^ in_ptr[j];
783         }
784         /* Encrypt */
785         /* Next iv is next block of cipher text */
786         if (EVP_CIPHER_CTX_encrypting(ctx))
787             memcpy(iv, out_ptr, 8);
788     }
789 /* Process rest of buffer */
790     if (i < inl) {
791         gost_crypt_mesh(EVP_CIPHER_CTX_get_cipher_data(ctx), iv, buf);
792         if (!EVP_CIPHER_CTX_encrypting(ctx))
793             memcpy(buf + 8, in_ptr, inl - i);
794         for (j = 0; i < inl; j++, i++) {
795             out_ptr[j] = buf[j] ^ in_ptr[j];
796         }
797         EVP_CIPHER_CTX_set_num(ctx, j);
798         if (EVP_CIPHER_CTX_encrypting(ctx))
799             memcpy(buf + 8, out_ptr, j);
800     } else {
801         EVP_CIPHER_CTX_set_num(ctx, 0);
802     }
803     return 1;
804 }
805
806 static int gost_cipher_do_cnt(EVP_CIPHER_CTX *ctx, unsigned char *out,
807                               const unsigned char *in, size_t inl)
808 {
809     const unsigned char *in_ptr = in;
810     unsigned char *out_ptr = out;
811     size_t i = 0;
812     size_t j;
813     unsigned char *buf = EVP_CIPHER_CTX_buf_noconst(ctx);
814     unsigned char *iv = EVP_CIPHER_CTX_iv_noconst(ctx);
815 /* process partial block if any */
816     if (EVP_CIPHER_CTX_num(ctx)) {
817         for (j = EVP_CIPHER_CTX_num(ctx), i = 0; j < 8 && i < inl;
818              j++, i++, in_ptr++, out_ptr++) {
819             *out_ptr = buf[j] ^ (*in_ptr);
820         }
821         if (j == 8) {
822             EVP_CIPHER_CTX_set_num(ctx, 0);
823         } else {
824             EVP_CIPHER_CTX_set_num(ctx, j);
825             return 1;
826         }
827     }
828
829     for (; i + 8 < inl; i += 8, in_ptr += 8, out_ptr += 8) {
830         /*
831          * block cipher current iv
832          */
833         /* Encrypt */
834         gost_cnt_next(EVP_CIPHER_CTX_get_cipher_data(ctx), iv, buf);
835         /*
836          * xor next block of input text with it and output it
837          */
838         /*
839          * output this block
840          */
841         for (j = 0; j < 8; j++) {
842             out_ptr[j] = buf[j] ^ in_ptr[j];
843         }
844     }
845 /* Process rest of buffer */
846     if (i < inl) {
847         gost_cnt_next(EVP_CIPHER_CTX_get_cipher_data(ctx), iv, buf);
848         for (j = 0; i < inl; j++, i++) {
849             out_ptr[j] = buf[j] ^ in_ptr[j];
850         }
851         EVP_CIPHER_CTX_set_num(ctx, j);
852     } else {
853         EVP_CIPHER_CTX_set_num(ctx, 0);
854     }
855     return 1;
856 }
857
858 /* Cleaning up of EVP_CIPHER_CTX */
859 static int gost_cipher_cleanup(EVP_CIPHER_CTX *ctx)
860 {
861     struct ossl_gost_cipher_ctx *c = EVP_CIPHER_CTX_get_cipher_data(ctx);
862                 EVP_MD_CTX_free(c->omac_ctx);
863     gost_destroy(&(c->cctx));
864     EVP_CIPHER_CTX_set_app_data(ctx, NULL);
865     return 1;
866 }
867
868 /* Control function for gost cipher */
869 static int gost_cipher_ctl(EVP_CIPHER_CTX *ctx, int type, int arg, void *ptr)
870 {
871     switch (type) {
872     case EVP_CTRL_RAND_KEY:
873         {
874             if (RAND_priv_bytes
875                 ((unsigned char *)ptr, EVP_CIPHER_CTX_key_length(ctx)) <= 0) {
876                 GOSTerr(GOST_F_GOST_CIPHER_CTL, GOST_R_RNG_ERROR);
877                 return -1;
878             }
879             break;
880         }
881     case EVP_CTRL_PBE_PRF_NID:
882         if (ptr) {
883             const char *params = get_gost_engine_param(GOST_PARAM_PBE_PARAMS);
884             int nid = NID_id_tc26_hmac_gost_3411_2012_512;
885
886             if (params) {
887                 if (!strcmp("md_gost12_256", params))
888                     nid = NID_id_tc26_hmac_gost_3411_2012_256;
889                 else if (!strcmp("md_gost12_512", params))
890                     nid = NID_id_tc26_hmac_gost_3411_2012_512;
891                 else if (!strcmp("md_gost94", params))
892                     nid = NID_id_HMACGostR3411_94;
893             }
894             *((int *)ptr) = nid;
895             return 1;
896         } else {
897             return 0;
898         }
899
900     case EVP_CTRL_SET_SBOX:
901         if (ptr) {
902             struct ossl_gost_cipher_ctx *c =
903                 EVP_CIPHER_CTX_get_cipher_data(ctx);
904             int nid;
905             int cur_meshing;
906             int ret;
907
908             if (c == NULL) {
909                 return -1;
910             }
911
912             if (c->count != 0) {
913                 return -1;
914             }
915
916             nid = OBJ_txt2nid(ptr);
917             if (nid == NID_undef) {
918                 return 0;
919             }
920
921             cur_meshing = c->key_meshing;
922             ret = gost_cipher_set_param(c, nid);
923             c->key_meshing = cur_meshing;
924             return ret;
925         } else {
926             return 0;
927         }
928     case EVP_CTRL_KEY_MESH:
929         {
930             struct ossl_gost_cipher_ctx *c =
931                 EVP_CIPHER_CTX_get_cipher_data(ctx);
932
933             if (c == NULL) {
934                 return -1;
935             }
936
937             if (c->count != 0) {
938                 return -1;
939             }
940
941             c->key_meshing = arg;
942             return 1;
943         }
944     default:
945         GOSTerr(GOST_F_GOST_CIPHER_CTL, GOST_R_UNSUPPORTED_CIPHER_CTL_COMMAND);
946         return -1;
947     }
948     return 1;
949 }
950
951 /* Control function for gost cipher */
952 static int magma_cipher_ctl(EVP_CIPHER_CTX *ctx, int type, int arg, void *ptr)
953 {
954     switch (type) {
955     case EVP_CTRL_RAND_KEY:
956             if (RAND_priv_bytes
957                 ((unsigned char *)ptr, EVP_CIPHER_CTX_key_length(ctx)) <= 0) {
958                 GOSTerr(GOST_F_GOST_CIPHER_CTL, GOST_R_RNG_ERROR);
959                 return -1;
960             }
961             break;
962     case EVP_CTRL_KEY_MESH:
963         {
964             struct ossl_gost_cipher_ctx *c =
965                 EVP_CIPHER_CTX_get_cipher_data(ctx);
966
967             if (c == NULL) {
968                 return -1;
969             }
970
971             if (c->count != 0) {
972                 return -1;
973             }
974
975             c->key_meshing = arg;
976             return 1;
977         }
978     default:
979         GOSTerr(GOST_F_MAGMA_CIPHER_CTL, GOST_R_UNSUPPORTED_CIPHER_CTL_COMMAND);
980         return -1;
981     }
982     return 1;
983 }
984
985 static int magma_cipher_ctl_acpkm_omac(EVP_CIPHER_CTX *ctx, int type, int arg, void *ptr)
986 {
987         switch (type)
988         {
989                 case EVP_CTRL_PROCESS_UNPROTECTED:
990                 {
991                         struct ossl_gost_cipher_ctx *c = EVP_CIPHER_CTX_get_cipher_data(ctx);
992                         STACK_OF(X509_ATTRIBUTE) *x = ptr;
993       return gost2015_process_unprotected_attributes(x, arg, MAGMA_MAC_MAX_SIZE, c->tag);
994                 }
995     case EVP_CTRL_COPY: {
996                         EVP_CIPHER_CTX *out = ptr;
997       struct ossl_gost_cipher_ctx *in_cctx  = EVP_CIPHER_CTX_get_cipher_data(ctx);
998       struct ossl_gost_cipher_ctx *out_cctx = EVP_CIPHER_CTX_get_cipher_data(out);
999
1000                         if (in_cctx->omac_ctx == out_cctx->omac_ctx) {
1001                                 out_cctx->omac_ctx = EVP_MD_CTX_new();
1002                                 if (out_cctx->omac_ctx == NULL) {
1003                                         GOSTerr(GOST_F_MAGMA_CIPHER_CTL_ACPKM_OMAC, ERR_R_MALLOC_FAILURE);
1004                                         return -1;
1005                                 }
1006                         }
1007                         return EVP_MD_CTX_copy(out_cctx->omac_ctx, in_cctx->omac_ctx);
1008                 }
1009                 default:
1010                         return magma_cipher_ctl(ctx, type, arg, ptr);
1011                         break;
1012         }
1013 }
1014
1015 /* Set cipher parameters from ASN1 structure */
1016 static int gost89_set_asn1_parameters(EVP_CIPHER_CTX *ctx, ASN1_TYPE *params)
1017 {
1018     int len = 0;
1019     unsigned char *buf = NULL;
1020     unsigned char *p = NULL;
1021     struct ossl_gost_cipher_ctx *c = EVP_CIPHER_CTX_get_cipher_data(ctx);
1022     GOST_CIPHER_PARAMS *gcp = GOST_CIPHER_PARAMS_new();
1023     ASN1_OCTET_STRING *os = NULL;
1024     if (!gcp) {
1025         GOSTerr(GOST_F_GOST89_SET_ASN1_PARAMETERS, ERR_R_MALLOC_FAILURE);
1026         return 0;
1027     }
1028     if (!ASN1_OCTET_STRING_set
1029         (gcp->iv, EVP_CIPHER_CTX_iv(ctx), EVP_CIPHER_CTX_iv_length(ctx))) {
1030         GOST_CIPHER_PARAMS_free(gcp);
1031         GOSTerr(GOST_F_GOST89_SET_ASN1_PARAMETERS, ERR_R_MALLOC_FAILURE);
1032         return 0;
1033     }
1034     ASN1_OBJECT_free(gcp->enc_param_set);
1035     gcp->enc_param_set = OBJ_nid2obj(c->paramNID);
1036
1037     len = i2d_GOST_CIPHER_PARAMS(gcp, NULL);
1038     p = buf = OPENSSL_malloc(len);
1039     if (!buf) {
1040         GOST_CIPHER_PARAMS_free(gcp);
1041         GOSTerr(GOST_F_GOST89_SET_ASN1_PARAMETERS, ERR_R_MALLOC_FAILURE);
1042         return 0;
1043     }
1044     i2d_GOST_CIPHER_PARAMS(gcp, &p);
1045     GOST_CIPHER_PARAMS_free(gcp);
1046
1047     os = ASN1_OCTET_STRING_new();
1048
1049     if (!os || !ASN1_OCTET_STRING_set(os, buf, len)) {
1050         OPENSSL_free(buf);
1051         GOSTerr(GOST_F_GOST89_SET_ASN1_PARAMETERS, ERR_R_MALLOC_FAILURE);
1052         return 0;
1053     }
1054     OPENSSL_free(buf);
1055
1056     ASN1_TYPE_set(params, V_ASN1_SEQUENCE, os);
1057     return 1;
1058 }
1059
1060 /* Store parameters into ASN1 structure */
1061 static int gost89_get_asn1_parameters(EVP_CIPHER_CTX *ctx, ASN1_TYPE *params)
1062 {
1063     int len;
1064     GOST_CIPHER_PARAMS *gcp = NULL;
1065     unsigned char *p;
1066     struct ossl_gost_cipher_ctx *c = EVP_CIPHER_CTX_get_cipher_data(ctx);
1067     int nid;
1068
1069     if (ASN1_TYPE_get(params) != V_ASN1_SEQUENCE) {
1070         return -1;
1071     }
1072
1073     p = params->value.sequence->data;
1074
1075     gcp = d2i_GOST_CIPHER_PARAMS(NULL, (const unsigned char **)&p,
1076                                  params->value.sequence->length);
1077
1078     len = gcp->iv->length;
1079     if (len != EVP_CIPHER_CTX_iv_length(ctx)) {
1080         GOST_CIPHER_PARAMS_free(gcp);
1081         GOSTerr(GOST_F_GOST89_GET_ASN1_PARAMETERS, GOST_R_INVALID_IV_LENGTH);
1082         return -1;
1083     }
1084
1085     nid = OBJ_obj2nid(gcp->enc_param_set);
1086     if (nid == NID_undef) {
1087         GOST_CIPHER_PARAMS_free(gcp);
1088         GOSTerr(GOST_F_GOST89_GET_ASN1_PARAMETERS,
1089                 GOST_R_INVALID_CIPHER_PARAM_OID);
1090         return -1;
1091     }
1092
1093     if (!gost_cipher_set_param(c, nid)) {
1094         GOST_CIPHER_PARAMS_free(gcp);
1095         return -1;
1096     }
1097     /*XXX missing non-const accessor */
1098     memcpy((unsigned char *)EVP_CIPHER_CTX_original_iv(ctx), gcp->iv->data,
1099            EVP_CIPHER_CTX_iv_length(ctx));
1100
1101     GOST_CIPHER_PARAMS_free(gcp);
1102
1103     return 1;
1104 }
1105
1106 #define MAGMA_UKM_LEN 12
1107 static int magma_set_asn1_parameters (EVP_CIPHER_CTX *ctx, ASN1_TYPE *params)
1108 {
1109   struct ossl_gost_cipher_ctx *c = EVP_CIPHER_CTX_get_cipher_data(ctx);
1110         c->key_meshing = 8192;
1111
1112         return gost2015_set_asn1_params(params, EVP_CIPHER_CTX_original_iv(ctx), 4,
1113                 c->kdf_seed);
1114 }
1115
1116 static int magma_get_asn1_parameters(EVP_CIPHER_CTX *ctx, ASN1_TYPE *params)
1117 {
1118   struct ossl_gost_cipher_ctx *c = EVP_CIPHER_CTX_get_cipher_data(ctx);
1119         unsigned char iv[16];
1120
1121         c->key_meshing = 8192;
1122
1123         if (gost2015_get_asn1_params(params, MAGMA_UKM_LEN, iv, 4, c->kdf_seed) < 0)
1124             return -1;
1125
1126         memcpy(EVP_CIPHER_CTX_iv_noconst(ctx), iv, sizeof(iv));
1127         memcpy((unsigned char *)EVP_CIPHER_CTX_original_iv(ctx), iv, sizeof(iv));
1128         /* Key meshing 8 kb*/
1129         c->key_meshing = 8192;
1130
1131         return 1;
1132 }
1133
1134 static int gost_imit_init(EVP_MD_CTX *ctx, gost_subst_block * block)
1135 {
1136     struct ossl_gost_imit_ctx *c = EVP_MD_CTX_md_data(ctx);
1137     memset(c->buffer, 0, sizeof(c->buffer));
1138     memset(c->partial_block, 0, sizeof(c->partial_block));
1139     c->count = 0;
1140     c->bytes_left = 0;
1141     c->key_meshing = 1;
1142     c->dgst_size = 4;
1143     gost_init(&(c->cctx), block);
1144     return 1;
1145 }
1146
1147 static int gost_imit_init_cpa(EVP_MD_CTX *ctx)
1148 {
1149     return gost_imit_init(ctx, &Gost28147_CryptoProParamSetA);
1150 }
1151
1152 static int gost_imit_init_cp_12(EVP_MD_CTX *ctx)
1153 {
1154     return gost_imit_init(ctx, &Gost28147_TC26ParamSetZ);
1155 }
1156
1157 static void mac_block_mesh(struct ossl_gost_imit_ctx *c,
1158                            const unsigned char *data)
1159 {
1160     /*
1161      * We are using NULL for iv because CryptoPro doesn't interpret
1162      * internal state of MAC algorithm as iv during keymeshing (but does
1163      * initialize internal state from iv in key transport
1164      */
1165     assert(c->count % 8 == 0 && c->count <= 1024);
1166     if (c->key_meshing && c->count == 1024) {
1167         cryptopro_key_meshing(&(c->cctx), NULL);
1168     }
1169     mac_block(&(c->cctx), c->buffer, data);
1170     c->count = c->count % 1024 + 8;
1171 }
1172
1173 static int gost_imit_update(EVP_MD_CTX *ctx, const void *data, size_t count)
1174 {
1175     struct ossl_gost_imit_ctx *c = EVP_MD_CTX_md_data(ctx);
1176     const unsigned char *p = data;
1177     size_t bytes = count;
1178     if (!(c->key_set)) {
1179         GOSTerr(GOST_F_GOST_IMIT_UPDATE, GOST_R_MAC_KEY_NOT_SET);
1180         return 0;
1181     }
1182     if (c->bytes_left) {
1183         size_t i;
1184         for (i = c->bytes_left; i < 8 && bytes > 0; bytes--, i++, p++) {
1185             c->partial_block[i] = *p;
1186         }
1187         if (i == 8) {
1188             mac_block_mesh(c, c->partial_block);
1189         } else {
1190             c->bytes_left = i;
1191             return 1;
1192         }
1193     }
1194     while (bytes > 8) {
1195         mac_block_mesh(c, p);
1196         p += 8;
1197         bytes -= 8;
1198     }
1199     if (bytes > 0) {
1200         memcpy(c->partial_block, p, bytes);
1201     }
1202     c->bytes_left = bytes;
1203     return 1;
1204 }
1205
1206 static int gost_imit_final(EVP_MD_CTX *ctx, unsigned char *md)
1207 {
1208     struct ossl_gost_imit_ctx *c = EVP_MD_CTX_md_data(ctx);
1209     if (!c->key_set) {
1210         GOSTerr(GOST_F_GOST_IMIT_FINAL, GOST_R_MAC_KEY_NOT_SET);
1211         return 0;
1212     }
1213     if (c->count == 0 && c->bytes_left) {
1214         unsigned char buffer[8];
1215         memset(buffer, 0, 8);
1216         gost_imit_update(ctx, buffer, 8);
1217     }
1218     if (c->bytes_left) {
1219         int i;
1220         for (i = c->bytes_left; i < 8; i++) {
1221             c->partial_block[i] = 0;
1222         }
1223         mac_block_mesh(c, c->partial_block);
1224     }
1225     get_mac(c->buffer, 8 * c->dgst_size, md);
1226     return 1;
1227 }
1228
1229 static int gost_imit_ctrl(EVP_MD_CTX *ctx, int type, int arg, void *ptr)
1230 {
1231     switch (type) {
1232     case EVP_MD_CTRL_KEY_LEN:
1233         *((unsigned int *)(ptr)) = 32;
1234         return 1;
1235     case EVP_MD_CTRL_SET_KEY:
1236         {
1237             struct ossl_gost_imit_ctx *gost_imit_ctx = EVP_MD_CTX_md_data(ctx);
1238
1239             if (EVP_MD_meth_get_init(EVP_MD_CTX_md(ctx)) (ctx) <= 0) {
1240                 GOSTerr(GOST_F_GOST_IMIT_CTRL, GOST_R_MAC_KEY_NOT_SET);
1241                 return 0;
1242             }
1243             EVP_MD_CTX_set_flags(ctx, EVP_MD_CTX_FLAG_NO_INIT);
1244
1245             if (arg == 0) {
1246                 struct gost_mac_key *key = (struct gost_mac_key *)ptr;
1247                 if (key->mac_param_nid != NID_undef) {
1248                     const struct gost_cipher_info *param =
1249                         get_encryption_params(OBJ_nid2obj(key->mac_param_nid));
1250                     if (param == NULL) {
1251                         GOSTerr(GOST_F_GOST_IMIT_CTRL,
1252                                 GOST_R_INVALID_MAC_PARAMS);
1253                         return 0;
1254                     }
1255                     gost_init(&(gost_imit_ctx->cctx), param->sblock);
1256                 }
1257                 gost_key(&(gost_imit_ctx->cctx), key->key);
1258                 gost_imit_ctx->key_set = 1;
1259
1260                 return 1;
1261             } else if (arg == 32) {
1262                 gost_key(&(gost_imit_ctx->cctx), ptr);
1263                 gost_imit_ctx->key_set = 1;
1264                 return 1;
1265             }
1266             GOSTerr(GOST_F_GOST_IMIT_CTRL, GOST_R_INVALID_MAC_KEY_SIZE);
1267             return 0;
1268         }
1269     case EVP_MD_CTRL_XOF_LEN:
1270         {
1271             struct ossl_gost_imit_ctx *c = EVP_MD_CTX_md_data(ctx);
1272             if (arg < 1 || arg > 8) {
1273                 GOSTerr(GOST_F_GOST_IMIT_CTRL, GOST_R_INVALID_MAC_SIZE);
1274                 return 0;
1275             }
1276             c->dgst_size = arg;
1277             return 1;
1278         }
1279
1280     default:
1281         return 0;
1282     }
1283 }
1284
1285 static int gost_imit_copy(EVP_MD_CTX *to, const EVP_MD_CTX *from)
1286 {
1287     if (EVP_MD_CTX_md_data(to) && EVP_MD_CTX_md_data(from)) {
1288         memcpy(EVP_MD_CTX_md_data(to), EVP_MD_CTX_md_data(from),
1289                sizeof(struct ossl_gost_imit_ctx));
1290     }
1291     return 1;
1292 }
1293
1294 /* Clean up imit ctx */
1295 static int gost_imit_cleanup(EVP_MD_CTX *ctx)
1296 {
1297     memset(EVP_MD_CTX_md_data(ctx), 0, sizeof(struct ossl_gost_imit_ctx));
1298     return 1;
1299 }
1300 /* vim: set expandtab cinoptions=\:0,l1,t0,g0,(0 sw=4 : */