static SECStatus swap_in_key_value(PRArenaPool *arena, mp_int *mpval, SECItem *buffer) { int len; mp_err err = MP_OKAY; memset(buffer->data, 0, buffer->len); len = mp_unsigned_octet_size(mpval); if (len <= 0) return SECFailure; if ((unsigned int)len <= buffer->len) { /* The new value is no longer than the old buffer, so use it */ err = mp_to_unsigned_octets(mpval, buffer->data, len); if (err >= 0) err = MP_OKAY; buffer->len = len; } else if (arena) { /* The new value is longer, but working within an arena */ (void)SECITEM_AllocItem(arena, buffer, len); err = mp_to_unsigned_octets(mpval, buffer->data, len); if (err >= 0) err = MP_OKAY; } else { /* The new value is longer, no arena, can't handle this key */ return SECFailure; } return (err == MP_OKAY) ? SECSuccess : SECFailure; }
SECStatus KEA_Derive(SECItem *prime, SECItem *public1, SECItem *public2, SECItem *private1, SECItem *private2, SECItem *derivedSecret) { mp_int p, Y, R, r, x, t, u, w; mp_err err; unsigned char *secret = NULL; unsigned int len = 0, offset; if (!prime || !public1 || !public2 || !private1 || !private2 || !derivedSecret) { PORT_SetError(SEC_ERROR_INVALID_ARGS); return SECFailure; } memset(derivedSecret, 0, sizeof *derivedSecret); MP_DIGITS(&p) = 0; MP_DIGITS(&Y) = 0; MP_DIGITS(&R) = 0; MP_DIGITS(&r) = 0; MP_DIGITS(&x) = 0; MP_DIGITS(&t) = 0; MP_DIGITS(&u) = 0; MP_DIGITS(&w) = 0; CHECK_MPI_OK( mp_init(&p) ); CHECK_MPI_OK( mp_init(&Y) ); CHECK_MPI_OK( mp_init(&R) ); CHECK_MPI_OK( mp_init(&r) ); CHECK_MPI_OK( mp_init(&x) ); CHECK_MPI_OK( mp_init(&t) ); CHECK_MPI_OK( mp_init(&u) ); CHECK_MPI_OK( mp_init(&w) ); SECITEM_TO_MPINT(*prime, &p); SECITEM_TO_MPINT(*public1, &Y); SECITEM_TO_MPINT(*public2, &R); SECITEM_TO_MPINT(*private1, &r); SECITEM_TO_MPINT(*private2, &x); /* t = DH(Y, r, p) = Y ** r mod p */ CHECK_MPI_OK( mp_exptmod(&Y, &r, &p, &t) ); /* u = DH(R, x, p) = R ** x mod p */ CHECK_MPI_OK( mp_exptmod(&R, &x, &p, &u) ); /* w = (t + u) mod p */ CHECK_MPI_OK( mp_addmod(&t, &u, &p, &w) ); /* allocate a buffer for the full derived secret */ len = mp_unsigned_octet_size(&w); secret = PORT_Alloc(len); if (secret == NULL) { err = MP_MEM; goto cleanup; } /* grab the secret */ err = mp_to_unsigned_octets(&w, secret, len); if (err > 0) err = MP_OKAY; /* allocate output buffer */ if (SECITEM_AllocItem(NULL, derivedSecret, KEA_DERIVED_SECRET_LEN) == NULL) { err = MP_MEM; goto cleanup; } memset(derivedSecret->data, 0, derivedSecret->len); /* copy in the 128 lsb of the secret */ if (len >= KEA_DERIVED_SECRET_LEN) { memcpy(derivedSecret->data, secret + (len - KEA_DERIVED_SECRET_LEN), KEA_DERIVED_SECRET_LEN); } else { offset = KEA_DERIVED_SECRET_LEN - len; memcpy(derivedSecret->data + offset, secret, len); } cleanup: mp_clear(&p); mp_clear(&Y); mp_clear(&R); mp_clear(&r); mp_clear(&x); mp_clear(&t); mp_clear(&u); mp_clear(&w); if (secret) PORT_ZFree(secret, len); if (err) { MP_TO_SEC_ERROR(err); if (derivedSecret->data) PORT_ZFree(derivedSecret->data, derivedSecret->len); return SECFailure; } return SECSuccess; }
SECStatus DH_Derive(SECItem *publicValue, SECItem *prime, SECItem *privateValue, SECItem *derivedSecret, unsigned int outBytes) { mp_int p, Xa, Yb, ZZ, psub1; mp_err err = MP_OKAY; unsigned int len = 0; unsigned int nb; unsigned char *secret = NULL; if (!publicValue || !prime || !privateValue || !derivedSecret) { PORT_SetError(SEC_ERROR_INVALID_ARGS); return SECFailure; } memset(derivedSecret, 0, sizeof *derivedSecret); MP_DIGITS(&p) = 0; MP_DIGITS(&Xa) = 0; MP_DIGITS(&Yb) = 0; MP_DIGITS(&ZZ) = 0; MP_DIGITS(&psub1) = 0; CHECK_MPI_OK( mp_init(&p) ); CHECK_MPI_OK( mp_init(&Xa) ); CHECK_MPI_OK( mp_init(&Yb) ); CHECK_MPI_OK( mp_init(&ZZ) ); CHECK_MPI_OK( mp_init(&psub1) ); SECITEM_TO_MPINT(*publicValue, &Yb); SECITEM_TO_MPINT(*privateValue, &Xa); SECITEM_TO_MPINT(*prime, &p); CHECK_MPI_OK( mp_sub_d(&p, 1, &psub1) ); /* We assume that the modulus, p, is a safe prime. That is, p = 2q+1 where * q is also a prime. Thus the orders of the subgroups are factors of 2q: * namely 1, 2, q and 2q. * * We check that the peer's public value isn't zero (which isn't in the * group), one (subgroup of order one) or p-1 (subgroup of order 2). We * also check that the public value is less than p, to avoid being fooled * by values like p+1 or 2*p-1. * * Thus we must be operating in the subgroup of size q or 2q. */ if (mp_cmp_d(&Yb, 1) <= 0 || mp_cmp(&Yb, &psub1) >= 0) { err = MP_BADARG; goto cleanup; } /* ZZ = (Yb)**Xa mod p */ CHECK_MPI_OK( mp_exptmod(&Yb, &Xa, &p, &ZZ) ); /* number of bytes in the derived secret */ len = mp_unsigned_octet_size(&ZZ); if (len <= 0) { err = MP_BADARG; goto cleanup; } /* * We check to make sure that ZZ is not equal to 1 or -1 mod p. * This helps guard against small subgroup attacks, since an attacker * using a subgroup of size N will produce 1 or -1 with probability 1/N. * When the protocol is executed within a properly large subgroup, the * probability of this result will be negligibly small. For example, * with a strong prime of the form 2p+1, the probability will be 1/p. * * We return MP_BADARG because this is probably the result of a bad * public value or a bad prime having been provided. */ if (mp_cmp_d(&ZZ, 1) == 0 || mp_cmp(&ZZ, &psub1) == 0) { err = MP_BADARG; goto cleanup; } /* allocate a buffer which can hold the entire derived secret. */ secret = PORT_Alloc(len); if (secret == NULL) { err = MP_MEM; goto cleanup; } /* grab the derived secret */ err = mp_to_unsigned_octets(&ZZ, secret, len); if (err >= 0) err = MP_OKAY; /* ** if outBytes is 0 take all of the bytes from the derived secret. ** if outBytes is not 0 take exactly outBytes from the derived secret, zero ** pad at the beginning if necessary, and truncate beginning bytes ** if necessary. */ if (outBytes > 0) nb = outBytes; else nb = len; if (SECITEM_AllocItem(NULL, derivedSecret, nb) == NULL) { err = MP_MEM; goto cleanup; } if (len < nb) { unsigned int offset = nb - len; memset(derivedSecret->data, 0, offset); memcpy(derivedSecret->data + offset, secret, len); } else { memcpy(derivedSecret->data, secret + len - nb, nb); } cleanup: mp_clear(&p); mp_clear(&Xa); mp_clear(&Yb); mp_clear(&ZZ); mp_clear(&psub1); if (secret) { /* free the buffer allocated for the full secret. */ PORT_ZFree(secret, len); } if (err) { MP_TO_SEC_ERROR(err); if (derivedSecret->data) PORT_ZFree(derivedSecret->data, derivedSecret->len); return SECFailure; } return SECSuccess; }
SECStatus DH_Derive(SECItem *publicValue, SECItem *prime, SECItem *privateValue, SECItem *derivedSecret, unsigned int outBytes) { mp_int p, Xa, Yb, ZZ; mp_err err = MP_OKAY; int len = 0; unsigned int nb; unsigned char *secret = NULL; if (!publicValue || !prime || !privateValue || !derivedSecret) { PORT_SetError(SEC_ERROR_INVALID_ARGS); return SECFailure; } memset(derivedSecret, 0, sizeof *derivedSecret); MP_DIGITS(&p) = 0; MP_DIGITS(&Xa) = 0; MP_DIGITS(&Yb) = 0; MP_DIGITS(&ZZ) = 0; CHECK_MPI_OK( mp_init(&p) ); CHECK_MPI_OK( mp_init(&Xa) ); CHECK_MPI_OK( mp_init(&Yb) ); CHECK_MPI_OK( mp_init(&ZZ) ); SECITEM_TO_MPINT(*publicValue, &Yb); SECITEM_TO_MPINT(*privateValue, &Xa); SECITEM_TO_MPINT(*prime, &p); /* ZZ = (Yb)**Xa mod p */ CHECK_MPI_OK( mp_exptmod(&Yb, &Xa, &p, &ZZ) ); /* number of bytes in the derived secret */ len = mp_unsigned_octet_size(&ZZ); if (len <= 0) { err = MP_BADARG; goto cleanup; } /* allocate a buffer which can hold the entire derived secret. */ secret = PORT_Alloc(len); /* grab the derived secret */ err = mp_to_unsigned_octets(&ZZ, secret, len); if (err >= 0) err = MP_OKAY; /* ** if outBytes is 0 take all of the bytes from the derived secret. ** if outBytes is not 0 take exactly outBytes from the derived secret, zero ** pad at the beginning if necessary, and truncate beginning bytes ** if necessary. */ if (outBytes > 0) nb = outBytes; else nb = len; SECITEM_AllocItem(NULL, derivedSecret, nb); if (len < nb) { unsigned int offset = nb - len; memset(derivedSecret->data, 0, offset); memcpy(derivedSecret->data + offset, secret, len); } else { memcpy(derivedSecret->data, secret + len - nb, nb); } cleanup: mp_clear(&p); mp_clear(&Xa); mp_clear(&Yb); mp_clear(&ZZ); if (secret) { /* free the buffer allocated for the full secret. */ PORT_ZFree(secret, len); } if (err) { MP_TO_SEC_ERROR(err); if (derivedSecret->data) PORT_ZFree(derivedSecret->data, derivedSecret->len); return SECFailure; } return SECSuccess; }
SECStatus DH_Derive(SECItem *publicValue, SECItem *prime, SECItem *privateValue, SECItem *derivedSecret, unsigned int maxOutBytes) { mp_int p, Xa, Yb, ZZ; mp_err err = MP_OKAY; unsigned int len = 0, nb; unsigned char *secret = NULL; if (!publicValue || !prime || !privateValue || !derivedSecret) { PORT_SetError(SEC_ERROR_INVALID_ARGS); return SECFailure; } memset(derivedSecret, 0, sizeof *derivedSecret); MP_DIGITS(&p) = 0; MP_DIGITS(&Xa) = 0; MP_DIGITS(&Yb) = 0; MP_DIGITS(&ZZ) = 0; CHECK_MPI_OK( mp_init(&p) ); CHECK_MPI_OK( mp_init(&Xa) ); CHECK_MPI_OK( mp_init(&Yb) ); CHECK_MPI_OK( mp_init(&ZZ) ); SECITEM_TO_MPINT(*publicValue, &Yb); SECITEM_TO_MPINT(*privateValue, &Xa); SECITEM_TO_MPINT(*prime, &p); /* ZZ = (Yb)**Xa mod p */ CHECK_MPI_OK( mp_exptmod(&Yb, &Xa, &p, &ZZ) ); /* number of bytes in the derived secret */ len = mp_unsigned_octet_size(&ZZ); /* allocate a buffer which can hold the entire derived secret. */ secret = PORT_Alloc(len); /* grab the derived secret */ err = mp_to_unsigned_octets(&ZZ, secret, len); if (err >= 0) err = MP_OKAY; /* Take minimum of bytes requested and bytes in derived secret, ** if maxOutBytes is 0 take all of the bytes from the derived secret. */ if (maxOutBytes > 0) nb = PR_MIN(len, maxOutBytes); else nb = len; SECITEM_AllocItem(NULL, derivedSecret, nb); memcpy(derivedSecret->data, secret, nb); cleanup: mp_clear(&p); mp_clear(&Xa); mp_clear(&Yb); mp_clear(&ZZ); if (secret) { /* free the buffer allocated for the full secret. */ PORT_ZFree(secret, len); } if (err) { MP_TO_SEC_ERROR(err); if (derivedSecret->data) PORT_ZFree(derivedSecret->data, derivedSecret->len); return SECFailure; } return SECSuccess; }