crypto/rsa/rsa_oaep.c

   1 /* crypto/rsa/rsa_oaep.c */
   2 /* Written by Ulf Moeller. This software is distributed on an "AS IS"
   3    basis, WITHOUT WARRANTY OF ANY KIND, either express or implied. */
   4
   5 /* EME_OAEP as defined in RFC 2437 (PKCS #1 v2.0) */
   6
   7 #include <stdio.h>
   8 #include "cryptlib.h"
   9 #include "bn.h"
  10 #include "rsa.h"
  11 #include "sha.h"
  12 #include "rand.h"
  13
  14 int MGF1(unsigned char *mask, long len, unsigned char *seed, long seedlen);
  15
  16 int RSA_padding_add_PKCS1_OAEP(to, tlen, from, flen, param, plen)
  17      unsigned char *to;
  18      int tlen;
  19      unsigned char *from;
  20      int flen;
  21      unsigned char *param;
  22      int plen;
  23     {
  24     int i, emlen = tlen - 1;
  25     unsigned char *db, *seed;
  26     unsigned char *dbmask, seedmask[SHA_DIGEST_LENGTH];
  27
  28     if (flen > emlen - 2 * SHA_DIGEST_LENGTH - 1)
  29         {
  30         RSAerr(RSA_F_RSA_PADDING_ADD_PKCS1_OAEP,
  31                RSA_R_DATA_TOO_LARGE_FOR_KEY_SIZE);
  32         return (0);
  33         }
  34
  35     if (emlen < 2 * SHA_DIGEST_LENGTH + 1)
  36         {
  37         RSAerr(RSA_F_RSA_PADDING_ADD_PKCS1_OAEP, RSA_R_KEY_SIZE_TOO_SMALL);
  38         return (0);
  39         }
  40
  41     dbmask = Malloc(emlen - SHA_DIGEST_LENGTH);
  42     if (dbmask == NULL)
  43         {
  44         RSAerr(RSA_F_RSA_PADDING_ADD_PKCS1_OAEP, ERR_R_MALLOC_FAILURE);
  45         return (0);
  46         }
  47
  48     to[0] = 0;
  49     seed = to + 1;
  50     db = to + SHA_DIGEST_LENGTH + 1;
  51
  52     SHA1(param, plen, db);
  53     memset(db + SHA_DIGEST_LENGTH, 0,
  54            emlen - flen - 2 * SHA_DIGEST_LENGTH - 1);
  55     db[emlen - flen - SHA_DIGEST_LENGTH - 1] = 0x01;
  56     memcpy(db + emlen - flen - SHA_DIGEST_LENGTH, from, (unsigned int) flen);
  57     RAND_bytes(seed, SHA_DIGEST_LENGTH);
  58 #ifdef PKCS_TESTVECT
  59     memcpy(seed,
  60            "\xaa\xfd\x12\xf6\x59\xca\xe6\x34\x89\xb4\x79\xe5\x07\x6d\xde\xc2\xf0\x6c\xb5\x8f",
  61            20);
  62 #endif
  63
  64     MGF1(dbmask, emlen - SHA_DIGEST_LENGTH, seed, SHA_DIGEST_LENGTH);
  65     for (i = 0; i < emlen - SHA_DIGEST_LENGTH; i++)
  66         db[i] ^= dbmask[i];
  67
  68     MGF1(seedmask, SHA_DIGEST_LENGTH, db, emlen - SHA_DIGEST_LENGTH);
  69     for (i = 0; i < SHA_DIGEST_LENGTH; i++)
  70         seed[i] ^= seedmask[i];
  71
  72     Free(dbmask);
  73     return (1);
  74     }
  75
  76 int RSA_padding_check_PKCS1_OAEP(to, tlen, from, flen, num, param, plen)
  77      unsigned char *to;
  78      int tlen;
  79      unsigned char *from;
  80      int flen;
  81      int num;
  82      unsigned char *param;
  83      int plen;
  84     {
  85     int i, dblen, mlen = -1;
  86     unsigned char *maskeddb;
  87     unsigned char *db, seed[SHA_DIGEST_LENGTH], phash[SHA_DIGEST_LENGTH];
  88
  89     if (flen < 2 * SHA_DIGEST_LENGTH + 1)
  90         {
  91         RSAerr(RSA_F_RSA_PADDING_CHECK_PKCS1_OAEP, RSA_R_OAEP_DECODING_ERROR);
  92         return (-1);
  93         }
  94
  95     dblen = flen - SHA_DIGEST_LENGTH;
  96     db = Malloc(dblen);
  97     if (db == NULL)
  98         {
  99         RSAerr(RSA_F_RSA_PADDING_ADD_PKCS1_OAEP, ERR_R_MALLOC_FAILURE);
 100         return (-1);
 101         }
 102
 103     maskeddb = from + SHA_DIGEST_LENGTH;
 104
 105     MGF1(seed, SHA_DIGEST_LENGTH, maskeddb, dblen);
 106     for (i = 0; i < SHA_DIGEST_LENGTH; i++)
 107         seed[i] ^= from[i];
 108
 109     MGF1(db, dblen, seed, SHA_DIGEST_LENGTH);
 110     for (i = 0; i < dblen; i++)
 111         db[i] ^= maskeddb[i];
 112
 113     SHA1(param, plen, phash);
 114
 115     if (memcmp(db, phash, SHA_DIGEST_LENGTH) != 0)
 116         RSAerr(RSA_F_RSA_PADDING_CHECK_PKCS1_OAEP, RSA_R_OAEP_DECODING_ERROR);
 117     else
 118         {
 119         for (i = SHA_DIGEST_LENGTH; i < dblen; i++)
 120             if (db[i] != 0x00)
 121                 break;
 122         if (db[i] != 0x01 || i++ >= dblen)
 123             RSAerr(RSA_F_RSA_PADDING_CHECK_PKCS1_OAEP,
 124                    RSA_R_OAEP_DECODING_ERROR);
 125         else
 126             {
 127             mlen = dblen - i;
 128             if (tlen < mlen)
 129                 {
 130                 RSAerr(RSA_F_RSA_PADDING_ADD_PKCS1_OAEP, RSA_R_DATA_TOO_LARGE);
 131                 mlen = -1;
 132                 }
 133             else
 134                 memcpy(to, db + i, mlen);
 135             }
 136         }
 137     Free(db);
 138     return (mlen);
 139     }
 140
 141 int MGF1(unsigned char *mask, long len, unsigned char *seed, long seedlen)
 142     {
 143     long i, outlen = 0;
 144     unsigned char cnt[4];
 145     SHA_CTX c;
 146     unsigned char md[SHA_DIGEST_LENGTH];
 147
 148     for (i = 0; outlen < len; i++)
 149         {
 150         cnt[0] = (i >> 24) & 255, cnt[1] = (i >> 16) & 255,
 151           cnt[2] = (i >> 8) & 255, cnt[3] = i & 255;
 152         SHA1_Init(&c);
 153         SHA1_Update(&c, seed, seedlen);
 154         SHA1_Update(&c, cnt, 4);
 155         if (outlen + SHA_DIGEST_LENGTH <= len)
 156             {
 157             SHA1_Final(mask + outlen, &c);
 158             outlen += SHA_DIGEST_LENGTH;
 159             }
 160         else
 161             {
 162             SHA1_Final(md, &c);
 163             memcpy(mask + outlen, md, len - outlen);
 164             outlen = len;
 165             }
 166         }
 167     return (0);
 168     }