crypto/poly1305/poly1305_pmeth.c

   1 /*
   2  * Copyright 2007-2017 The OpenSSL Project Authors. All Rights Reserved.
   3  *
   4  * Licensed under the OpenSSL license (the "License").  You may not use
   5  * this file except in compliance with the License.  You can obtain a copy
   6  * in the file LICENSE in the source distribution or at
   7  * https://www.openssl.org/source/license.html
   8  */
   9
  10 #include <stdio.h>
  11 #include "internal/cryptlib.h"
  12 #include <openssl/x509.h>
  13 #include <openssl/x509v3.h>
  14 #include <openssl/evp.h>
  15 #include "internal/poly1305.h"
  16 #include "poly1305_local.h"
  17 #include "internal/evp_int.h"
  18
  19 /* POLY1305 pkey context structure */
  20
  21 typedef struct {
  22     ASN1_OCTET_STRING ktmp;     /* Temp storage for key */
  23     POLY1305 ctx;
  24 } POLY1305_PKEY_CTX;
  25
  26 static int pkey_poly1305_init(EVP_PKEY_CTX *ctx)
  27 {
  28     POLY1305_PKEY_CTX *pctx;
  29
  30     pctx = OPENSSL_zalloc(sizeof(*pctx));
  31     if (pctx == NULL)
  32         return 0;
  33     pctx->ktmp.type = V_ASN1_OCTET_STRING;
  34
  35     EVP_PKEY_CTX_set_data(ctx, pctx);
  36     EVP_PKEY_CTX_set0_keygen_info(ctx, NULL, 0);
  37     return 1;
  38 }
  39
  40 static void pkey_poly1305_cleanup(EVP_PKEY_CTX *ctx)
  41 {
  42     POLY1305_PKEY_CTX *pctx = EVP_PKEY_CTX_get_data(ctx);
  43
  44     if (pctx != NULL) {
  45         OPENSSL_clear_free(pctx->ktmp.data, pctx->ktmp.length);
  46         OPENSSL_clear_free(pctx, sizeof(*pctx));
  47         EVP_PKEY_CTX_set_data(ctx, NULL);
  48     }
  49 }
  50
  51 static int pkey_poly1305_copy(EVP_PKEY_CTX *dst, EVP_PKEY_CTX *src)
  52 {
  53     POLY1305_PKEY_CTX *sctx, *dctx;
  54
  55     /* allocate memory for dst->data and a new POLY1305_CTX in dst->data->ctx */
  56     if (!pkey_poly1305_init(dst))
  57         return 0;
  58     sctx = EVP_PKEY_CTX_get_data(src);
  59     dctx = EVP_PKEY_CTX_get_data(dst);
  60     if (ASN1_STRING_get0_data(&sctx->ktmp) != NULL &&
  61         !ASN1_STRING_copy(&dctx->ktmp, &sctx->ktmp)) {
  62         /* cleanup and free the POLY1305_PKEY_CTX in dst->data */
  63         pkey_poly1305_cleanup(dst);
  64         return 0;
  65     }
  66     memcpy(&dctx->ctx, &sctx->ctx, sizeof(POLY1305));
  67     return 1;
  68 }
  69
  70 static int pkey_poly1305_keygen(EVP_PKEY_CTX *ctx, EVP_PKEY *pkey)
  71 {
  72     ASN1_OCTET_STRING *key;
  73     POLY1305_PKEY_CTX *pctx = EVP_PKEY_CTX_get_data(ctx);
  74
  75     if (ASN1_STRING_get0_data(&pctx->ktmp) == NULL)
  76         return 0;
  77     key = ASN1_OCTET_STRING_dup(&pctx->ktmp);
  78     if (key == NULL)
  79         return 0;
  80     return EVP_PKEY_assign_POLY1305(pkey, key);
  81 }
  82
  83 static int int_update(EVP_MD_CTX *ctx, const void *data, size_t count)
  84 {
  85     POLY1305_PKEY_CTX *pctx = EVP_PKEY_CTX_get_data(EVP_MD_CTX_pkey_ctx(ctx));
  86
  87     Poly1305_Update(&pctx->ctx, data, count);
  88     return 1;
  89 }
  90
  91 static int poly1305_signctx_init(EVP_PKEY_CTX *ctx, EVP_MD_CTX *mctx)
  92 {
  93     POLY1305_PKEY_CTX *pctx = ctx->data;
  94     ASN1_OCTET_STRING *key = (ASN1_OCTET_STRING *)ctx->pkey->pkey.ptr;
  95
  96     if (key->length != POLY1305_KEY_SIZE)
  97         return 0;
  98     EVP_MD_CTX_set_flags(mctx, EVP_MD_CTX_FLAG_NO_INIT);
  99     EVP_MD_CTX_set_update_fn(mctx, int_update);
 100     Poly1305_Init(&pctx->ctx, key->data);
 101     return 1;
 102 }
 103 static int poly1305_signctx(EVP_PKEY_CTX *ctx, unsigned char *sig, size_t *siglen,
 104                             EVP_MD_CTX *mctx)
 105 {
 106     POLY1305_PKEY_CTX *pctx = ctx->data;
 107
 108     *siglen = POLY1305_DIGEST_SIZE;
 109     if (sig != NULL)
 110         Poly1305_Final(&pctx->ctx, sig);
 111     return 1;
 112 }
 113
 114 static int pkey_poly1305_ctrl(EVP_PKEY_CTX *ctx, int type, int p1, void *p2)
 115 {
 116     POLY1305_PKEY_CTX *pctx = EVP_PKEY_CTX_get_data(ctx);
 117     const unsigned char *key;
 118     size_t len;
 119
 120     switch (type) {
 121
 122     case EVP_PKEY_CTRL_MD:
 123         /* ignore */
 124         break;
 125
 126     case EVP_PKEY_CTRL_SET_MAC_KEY:
 127     case EVP_PKEY_CTRL_DIGESTINIT:
 128         if (type == EVP_PKEY_CTRL_SET_MAC_KEY) {
 129             /* user explicitly setting the key */
 130             key = p2;
 131             len = p1;
 132         } else {
 133             /* user indirectly setting the key via EVP_DigestSignInit */
 134             key = EVP_PKEY_get0_poly1305(EVP_PKEY_CTX_get0_pkey(ctx), &len);
 135         }
 136         if (key == NULL || len != POLY1305_KEY_SIZE ||
 137             !ASN1_OCTET_STRING_set(&pctx->ktmp, key, len))
 138             return 0;
 139         Poly1305_Init(&pctx->ctx, ASN1_STRING_get0_data(&pctx->ktmp));
 140         break;
 141
 142     default:
 143         return -2;
 144
 145     }
 146     return 1;
 147 }
 148
 149 static int pkey_poly1305_ctrl_str(EVP_PKEY_CTX *ctx,
 150                                   const char *type, const char *value)
 151 {
 152     if (value == NULL)
 153         return 0;
 154     if (strcmp(type, "key") == 0)
 155         return EVP_PKEY_CTX_str2ctrl(ctx, EVP_PKEY_CTRL_SET_MAC_KEY, value);
 156     if (strcmp(type, "hexkey") == 0)
 157         return EVP_PKEY_CTX_hex2ctrl(ctx, EVP_PKEY_CTRL_SET_MAC_KEY, value);
 158     return -2;
 159 }
 160
 161 const EVP_PKEY_METHOD poly1305_pkey_meth = {
 162     EVP_PKEY_POLY1305,
 163     EVP_PKEY_FLAG_SIGCTX_CUSTOM, /* we don't deal with a separate MD */
 164     pkey_poly1305_init,
 165     pkey_poly1305_copy,
 166     pkey_poly1305_cleanup,
 167
 168     0, 0,
 169
 170     0,
 171     pkey_poly1305_keygen,
 172
 173     0, 0,
 174
 175     0, 0,
 176
 177     0, 0,
 178
 179     poly1305_signctx_init,
 180     poly1305_signctx,
 181
 182     0, 0,
 183
 184     0, 0,
 185
 186     0, 0,
 187
 188     0, 0,
 189
 190     pkey_poly1305_ctrl,
 191     pkey_poly1305_ctrl_str
 192 };