dh_sys.c

NIST推荐的素域上的椭圆曲线

#ifdef PK_PACKET

int dh_encrypt(const unsigned char *in,  unsigned long len, 
                     unsigned char *out, unsigned long *outlen,
                     prng_state *prng, int wprng, int cipher, int hash, 
                     dh_key *key)
{
    unsigned char pub_expt[1536], dh_shared[1536], IV[MAXBLOCKSIZE], skey[MAXBLOCKSIZE];
    dh_key pubkey;
    unsigned long x, y, z, hashsize, blocksize, pubkeysize;
    int keysize, errno;
    symmetric_CTR ctr;

    _ARGCHK(in != NULL);
    _ARGCHK(out != NULL);
    _ARGCHK(outlen != NULL);
    _ARGCHK(key != NULL);

    /* check that wprng/cipher/hash are not invalid */
    if ((errno = prng_is_valid(wprng)) != CRYPT_OK) {
       return errno;
    }

    if ((errno = hash_is_valid(hash)) != CRYPT_OK) {
       return errno;
    }

    if ((errno = cipher_is_valid(cipher)) != CRYPT_OK) {
       return errno;
    }

    /* make a random key and export the public copy */
    if ((errno = dh_make_key(prng, wprng, dh_get_size(key), &pubkey)) != CRYPT_OK) {
       return errno;
    }

    pubkeysize = sizeof(pub_expt);
    if ((errno = dh_export(pub_expt, &pubkeysize, PK_PUBLIC, &pubkey)) != CRYPT_OK) {
       dh_free(&pubkey);
       return errno;
    }
    
    /* now check if the out buffer is big enough */
    if (*outlen < (10 + PACKET_SIZE + pubkeysize + cipher_descriptor[cipher].block_length + len)) {
       dh_free(&pubkey);
       return CRYPT_BUFFER_OVERFLOW;
    }

    /* make random key */
    blocksize = cipher_descriptor[cipher].block_length;
    hashsize  = hash_descriptor[hash].hashsize;
    keysize = hashsize;
    if ((errno = cipher_descriptor[cipher].keysize(&keysize)) != CRYPT_OK) {
       dh_free(&pubkey);
       return errno;
    }

    x = sizeof(dh_shared);
    if ((errno = dh_shared_secret(&pubkey, key, dh_shared, &x)) != CRYPT_OK) {
       dh_free(&pubkey);
       return errno;
    }
    dh_free(&pubkey);

    z = sizeof(skey);
    if ((errno = hash_memory(hash, dh_shared, x, skey, &z)) != CRYPT_OK) {
       return errno;
    }

    /* make up IV */
    if (prng_descriptor[wprng].read(IV, cipher_descriptor[cipher].block_length, prng) != 
        cipher_descriptor[cipher].block_length) {
       return CRYPT_ERROR_READPRNG;
    }

    /* setup CTR mode */
    if ((errno = ctr_start(cipher, IV, skey, keysize, 0, &ctr)) != CRYPT_OK) {
       return errno;
    }

    /* output header */
    y = PACKET_SIZE;
 
    /* size of cipher name and the name itself */
    out[y++] = cipher_descriptor[cipher].ID;

    /* size of hash name and the name itself */
    out[y++] = hash_descriptor[hash].ID;

    /* length of DH pubkey and the key itself */
    STORE32L(pubkeysize, out+y);
    y += 4;
    for (x = 0; x < pubkeysize; x++, y++) {
        out[y] = pub_expt[x];
    }

    /* cipher IV */
    for (x = 0; x < blocksize; x++, y++) {
        out[y] = IV[x];
    }

    /* length of ciphertext */
    STORE32L(len, out+y);
    y += 4;

    /* encrypt the message */
    if ((errno = ctr_encrypt(in, out+y, len, &ctr)) != CRYPT_OK) {
       return errno;
    }
    y += len;

    /* store header */
    packet_store_header(out, PACKET_SECT_DH, PACKET_SUB_ENCRYPTED, y);

#ifdef CLEAN_STACK
    /* clean up */
    zeromem(pub_expt, sizeof(pub_expt));
    zeromem(dh_shared, sizeof(dh_shared));
    zeromem(skey, sizeof(skey));
    zeromem(IV, sizeof(IV));
    zeromem(&ctr, sizeof(ctr));
#endif
    *outlen = y;
    return CRYPT_OK;
}

int dh_decrypt(const unsigned char *in,  unsigned long len, 
                     unsigned char *out, unsigned long *outlen, 
                     dh_key *key)
{
   unsigned char shared_secret[1536], skey[MAXBLOCKSIZE];
   unsigned long x, y, z, res, hashsize, blocksize;
   int hash, cipher, keysize, errno;
   dh_key pubkey;
   symmetric_CTR ctr;

   _ARGCHK(in != NULL);
   _ARGCHK(out != NULL);
   _ARGCHK(outlen != NULL);
   _ARGCHK(key != NULL);

   /* right key type? */
   if (key->type != PK_PRIVATE) {
      return CRYPT_PK_NOT_PRIVATE;
   }

   /* is header correct? */
   if ((errno = packet_valid_header((unsigned char *)in, PACKET_SECT_DH, PACKET_SUB_ENCRYPTED)) != CRYPT_OK) {
      return errno;
   }

   /* now lets get the cipher name */
   y = PACKET_SIZE;
   cipher = find_cipher_id(in[y++]);
   if (cipher == -1) {
      return CRYPT_INVALID_CIPHER;
   }

   /* now lets get the hash name */
   hash = find_hash_id(in[y++]);
   if (hash == -1) {
      return CRYPT_INVALID_HASH;
   }

   /* common values */
   blocksize = cipher_descriptor[cipher].block_length;
   hashsize  = hash_descriptor[hash].hashsize;
   keysize = hashsize;
   if ((errno = cipher_descriptor[cipher].keysize(&keysize)) != CRYPT_OK) {
      return errno;
   }

   /* get public key */
   LOAD32L(x, in+y);
   y += 4;
   if ((errno = dh_import(in+y, &pubkey)) != CRYPT_OK) {
      return errno;
   }
   y += x;

   /* make shared key */
   x = sizeof(shared_secret);
   if ((errno = dh_shared_secret(key, &pubkey, shared_secret, &x)) != CRYPT_OK) {
      dh_free(&pubkey);
      return errno;
   }
   dh_free(&pubkey);

   z = sizeof(skey);
   if ((errno = hash_memory(hash, shared_secret, x, skey, &z)) != CRYPT_OK) {
      return errno;
   }

   /* setup CTR mode */
   if ((errno = ctr_start(cipher, in+y, skey, keysize, 0, &ctr)) != CRYPT_OK) {
      res = errno;
      goto done;
   }

   /* skip over the IV */
   y += blocksize;

   /* get length */
   LOAD32L(len,in+y);
   y += 4;

   /* buffer overflow? */
   if (len > *outlen) {
      res = CRYPT_BUFFER_OVERFLOW;
      goto done;
   }

   /* decrypt message */
   if ((errno = ctr_decrypt(in+y, out, len, &ctr)) != CRYPT_OK) {
      res = errno;
      goto done;
   }
   *outlen = len;
   
   res = CRYPT_OK;
done:
#ifdef CLEAN_STACK
   zeromem(shared_secret, sizeof(shared_secret));
   zeromem(skey, sizeof(skey));
   zeromem(&ctr, sizeof(ctr));
#endif
   return res;
}

int dh_sign(const unsigned char *in,  unsigned long inlen,
                  unsigned char *out, unsigned long *outlen,
                  int hash, prng_state *prng, int wprng,
                  dh_key *key)
{
   mp_int a, b, k, m, g, p, p1, tmp;
   unsigned char buf[1536], md[MAXBLOCKSIZE];
   unsigned long x, y, z;
   int res, errno;

    _ARGCHK(in != NULL);
    _ARGCHK(out != NULL);
    _ARGCHK(outlen != NULL);
    _ARGCHK(key != NULL);

   /* check parameters */
   if (key->type != PK_PRIVATE) {
      return CRYPT_PK_NOT_PRIVATE;
   }

   if ((errno = prng_is_valid(wprng)) != CRYPT_OK) {
      return errno;
   }

   if ((errno = hash_is_valid(hash)) != CRYPT_OK) {
      return errno;
   }

   /* hash the message */
   z = sizeof(md) - 1;
   md[0] = 0;
   if ((errno = hash_memory(hash, in, inlen, md+1, &z)) != CRYPT_OK) {
      return errno;
   }

   /* make up a random value k,
    * since the order of the group is prime
    * we need not check if gcd(k, r) is 1 
    */
   buf[0] = 0;
   if (prng_descriptor[wprng].read(buf+1, sets[key->idx].size-1, prng) != (unsigned long)(sets[key->idx].size-1)) {
      return CRYPT_ERROR_READPRNG;
   }

   /* init bignums */
   if (mp_init_multi(&a, &b, &k, &m, &p, &g, &p1, &tmp, NULL) != MP_OKAY) { 
      return CRYPT_MEM;
   }

   /* load k and m */
   if (mp_read_raw(&m, md,  1+hash_descriptor[hash].hashsize) != MP_OKAY)     { goto error; }
   if (mp_read_raw(&k, buf, sets[key->idx].size) != MP_OKAY)                  { goto error; }

   /* load g, p and p1 */
   if (mp_read_radix(&g, sets[key->idx].base, 10) != MP_OKAY)                 { goto error; }
   if (mp_read_radix(&p, sets[key->idx].prime, 10) != MP_OKAY)                { goto error; }
   if (mp_sub_d(&p, 1, &p1) != MP_OKAY)                         { goto error; } /* p1 = p-1 */
   if (mp_div_2(&p1, &p1) != MP_OKAY)                           { goto error; } /* p1 = (p-1)/2 */

   /* now get a = g^k mod p */
   if (mp_exptmod(&g, &k, &p, &a) != MP_OKAY)                   { goto error; } /* a = g^k mod p */

   /* now find M = xa + kb mod p1 or just b = (M - xa)/k mod p1 */
   if (mp_invmod(&k, &p1, &k) != MP_OKAY)                       { goto error; } /* k = 1/k mod p1 */
   if (mp_mulmod(&a, &key->x, &p1, &tmp) != MP_OKAY)            { goto error; } /* tmp = xa */
   if (mp_submod(&m, &tmp, &p1, &tmp) != MP_OKAY)               { goto error; } /* tmp = M - xa */
   if (mp_mulmod(&k, &tmp, &p1, &b) != MP_OKAY)                 { goto error; } /* b = (M - xa)/k */

   /* store header  */
   y = PACKET_SIZE;

   /* store length and name of hash */
   buf[y++] = hash_descriptor[hash].ID;                         /* store hash ID */
 
   /* now store them both (a,b) */
   x = mp_raw_size(&a);                                         /* get raw size of a */
   STORE32L(x, buf+y);  y += 4;                                 /* store size of a */
   mp_toraw(&a, buf+y); y += x;                                 /* store a itself */

   x = mp_raw_size(&b);                                         /* get raw size of b */
   STORE32L(x, buf+y);  y += 4;                                 /* store size of b */
   mp_toraw(&b, buf+y); y += x;                                 /* store b itself */

   /* check if size too big */
   if (*outlen < y) { goto error; }

   /* store header */
   packet_store_header(buf, PACKET_SECT_DH, PACKET_SUB_SIGNED, y);

   /* store it */
   memcpy(out, buf, y);
   *outlen = y;

#ifdef CLEAN_STACK
   zeromem(md, sizeof(md));
   zeromem(buf, sizeof(buf));
#endif

   res = CRYPT_OK;
   goto done;
error:
   res = CRYPT_MEM;
done:
   mp_clear_multi(&tmp, &p1, &g, &p, &m, &k, &b, &a, NULL);
   return res;
}

int dh_verify(const unsigned char *sig, const unsigned char *msg, 
                    unsigned long inlen, int *stat, 
                    dh_key *key)
{
   mp_int a, b, p, g, m, tmp;
   unsigned char md[MAXBLOCKSIZE];
   unsigned long x, y, z;
   int hash, res, errno;

   _ARGCHK(sig != NULL);
   _ARGCHK(msg != NULL);
   _ARGCHK(stat != NULL);
   _ARGCHK(key != NULL);

   /* default to invalid */
   *stat = 0;

   /* header ok? */
   if ((errno = packet_valid_header((unsigned char *)sig, PACKET_SECT_DH, PACKET_SUB_SIGNED)) != CRYPT_OK) {
      return errno;
   }

   /* get hash out of packet */
   y = PACKET_SIZE;
   hash = find_hash_id(sig[y++]);

   if (hash == -1) {
      return CRYPT_INVALID_HASH;
   }

   /* hash the message */
   md[0] = 0;
   z = sizeof(md) - 1;
   if ((errno = hash_memory(hash, msg, inlen, md+1, &z)) != CRYPT_OK) {
      return errno;
   }

   /* init all bignums */
   if (mp_init_multi(&a, &p, &b, &g, &m, &tmp, NULL) != MP_OKAY) { 
      return CRYPT_MEM;
   }

   /* load a and b */
   LOAD32L(x, sig+y);
   y += 4;
   if (mp_read_raw(&a, (unsigned char *)sig+y, x) != MP_OKAY)        { goto error; }
   y += x;

   LOAD32L(x, sig+y);
   y += 4;
   if (mp_read_raw(&b, (unsigned char *)sig+y, x) != MP_OKAY)        { goto error; }
   y += x;