mirror of
				https://github.com/ossrs/srs.git
				synced 2025-03-09 15:49:59 +00:00 
			
		
		
		
	
		
			
				
	
	
		
			425 lines
		
	
	
	
		
			13 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
			
		
		
	
	
			425 lines
		
	
	
	
		
			13 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
| /*
 | |
|  * Copyright 2002-2019 The OpenSSL Project Authors. All Rights Reserved.
 | |
|  *
 | |
|  * Licensed under the OpenSSL license (the "License").  You may not use
 | |
|  * this file except in compliance with the License.  You can obtain a copy
 | |
|  * in the file LICENSE in the source distribution or at
 | |
|  * https://www.openssl.org/source/license.html
 | |
|  */
 | |
| 
 | |
| #include <string.h>
 | |
| #include <openssl/err.h>
 | |
| #include <openssl/obj_mac.h>
 | |
| #include <openssl/rand.h>
 | |
| #include "crypto/bn.h"
 | |
| #include "ec_local.h"
 | |
| 
 | |
| int ossl_ecdsa_sign(int type, const unsigned char *dgst, int dlen,
 | |
|                     unsigned char *sig, unsigned int *siglen,
 | |
|                     const BIGNUM *kinv, const BIGNUM *r, EC_KEY *eckey)
 | |
| {
 | |
|     ECDSA_SIG *s;
 | |
| 
 | |
|     s = ECDSA_do_sign_ex(dgst, dlen, kinv, r, eckey);
 | |
|     if (s == NULL) {
 | |
|         *siglen = 0;
 | |
|         return 0;
 | |
|     }
 | |
|     *siglen = i2d_ECDSA_SIG(s, &sig);
 | |
|     ECDSA_SIG_free(s);
 | |
|     return 1;
 | |
| }
 | |
| 
 | |
| static int ecdsa_sign_setup(EC_KEY *eckey, BN_CTX *ctx_in,
 | |
|                             BIGNUM **kinvp, BIGNUM **rp,
 | |
|                             const unsigned char *dgst, int dlen)
 | |
| {
 | |
|     BN_CTX *ctx = NULL;
 | |
|     BIGNUM *k = NULL, *r = NULL, *X = NULL;
 | |
|     const BIGNUM *order;
 | |
|     EC_POINT *tmp_point = NULL;
 | |
|     const EC_GROUP *group;
 | |
|     int ret = 0;
 | |
|     int order_bits;
 | |
|     const BIGNUM *priv_key;
 | |
| 
 | |
|     if (eckey == NULL || (group = EC_KEY_get0_group(eckey)) == NULL) {
 | |
|         ECerr(EC_F_ECDSA_SIGN_SETUP, ERR_R_PASSED_NULL_PARAMETER);
 | |
|         return 0;
 | |
|     }
 | |
|     if ((priv_key = EC_KEY_get0_private_key(eckey)) == NULL) {
 | |
|         ECerr(EC_F_ECDSA_SIGN_SETUP, EC_R_MISSING_PRIVATE_KEY);
 | |
|         return 0;
 | |
|     }
 | |
| 
 | |
|     if (!EC_KEY_can_sign(eckey)) {
 | |
|         ECerr(EC_F_ECDSA_SIGN_SETUP, EC_R_CURVE_DOES_NOT_SUPPORT_SIGNING);
 | |
|         return 0;
 | |
|     }
 | |
| 
 | |
|     if ((ctx = ctx_in) == NULL) {
 | |
|         if ((ctx = BN_CTX_new()) == NULL) {
 | |
|             ECerr(EC_F_ECDSA_SIGN_SETUP, ERR_R_MALLOC_FAILURE);
 | |
|             return 0;
 | |
|         }
 | |
|     }
 | |
| 
 | |
|     k = BN_new();               /* this value is later returned in *kinvp */
 | |
|     r = BN_new();               /* this value is later returned in *rp */
 | |
|     X = BN_new();
 | |
|     if (k == NULL || r == NULL || X == NULL) {
 | |
|         ECerr(EC_F_ECDSA_SIGN_SETUP, ERR_R_MALLOC_FAILURE);
 | |
|         goto err;
 | |
|     }
 | |
|     if ((tmp_point = EC_POINT_new(group)) == NULL) {
 | |
|         ECerr(EC_F_ECDSA_SIGN_SETUP, ERR_R_EC_LIB);
 | |
|         goto err;
 | |
|     }
 | |
|     order = EC_GROUP_get0_order(group);
 | |
| 
 | |
|     /* Preallocate space */
 | |
|     order_bits = BN_num_bits(order);
 | |
|     if (!BN_set_bit(k, order_bits)
 | |
|         || !BN_set_bit(r, order_bits)
 | |
|         || !BN_set_bit(X, order_bits))
 | |
|         goto err;
 | |
| 
 | |
|     do {
 | |
|         /* get random k */
 | |
|         do {
 | |
|             if (dgst != NULL) {
 | |
|                 if (!BN_generate_dsa_nonce(k, order, priv_key,
 | |
|                                            dgst, dlen, ctx)) {
 | |
|                     ECerr(EC_F_ECDSA_SIGN_SETUP,
 | |
|                           EC_R_RANDOM_NUMBER_GENERATION_FAILED);
 | |
|                     goto err;
 | |
|                 }
 | |
|             } else {
 | |
|                 if (!BN_priv_rand_range(k, order)) {
 | |
|                     ECerr(EC_F_ECDSA_SIGN_SETUP,
 | |
|                           EC_R_RANDOM_NUMBER_GENERATION_FAILED);
 | |
|                     goto err;
 | |
|                 }
 | |
|             }
 | |
|         } while (BN_is_zero(k));
 | |
| 
 | |
|         /* compute r the x-coordinate of generator * k */
 | |
|         if (!EC_POINT_mul(group, tmp_point, k, NULL, NULL, ctx)) {
 | |
|             ECerr(EC_F_ECDSA_SIGN_SETUP, ERR_R_EC_LIB);
 | |
|             goto err;
 | |
|         }
 | |
| 
 | |
|         if (!EC_POINT_get_affine_coordinates(group, tmp_point, X, NULL, ctx)) {
 | |
|             ECerr(EC_F_ECDSA_SIGN_SETUP, ERR_R_EC_LIB);
 | |
|             goto err;
 | |
|         }
 | |
| 
 | |
|         if (!BN_nnmod(r, X, order, ctx)) {
 | |
|             ECerr(EC_F_ECDSA_SIGN_SETUP, ERR_R_BN_LIB);
 | |
|             goto err;
 | |
|         }
 | |
|     } while (BN_is_zero(r));
 | |
| 
 | |
|     /* compute the inverse of k */
 | |
|     if (!ec_group_do_inverse_ord(group, k, k, ctx)) {
 | |
|         ECerr(EC_F_ECDSA_SIGN_SETUP, ERR_R_BN_LIB);
 | |
|         goto err;
 | |
|     }
 | |
| 
 | |
|     /* clear old values if necessary */
 | |
|     BN_clear_free(*rp);
 | |
|     BN_clear_free(*kinvp);
 | |
|     /* save the pre-computed values  */
 | |
|     *rp = r;
 | |
|     *kinvp = k;
 | |
|     ret = 1;
 | |
|  err:
 | |
|     if (!ret) {
 | |
|         BN_clear_free(k);
 | |
|         BN_clear_free(r);
 | |
|     }
 | |
|     if (ctx != ctx_in)
 | |
|         BN_CTX_free(ctx);
 | |
|     EC_POINT_free(tmp_point);
 | |
|     BN_clear_free(X);
 | |
|     return ret;
 | |
| }
 | |
| 
 | |
| int ossl_ecdsa_sign_setup(EC_KEY *eckey, BN_CTX *ctx_in, BIGNUM **kinvp,
 | |
|                           BIGNUM **rp)
 | |
| {
 | |
|     return ecdsa_sign_setup(eckey, ctx_in, kinvp, rp, NULL, 0);
 | |
| }
 | |
| 
 | |
| ECDSA_SIG *ossl_ecdsa_sign_sig(const unsigned char *dgst, int dgst_len,
 | |
|                                const BIGNUM *in_kinv, const BIGNUM *in_r,
 | |
|                                EC_KEY *eckey)
 | |
| {
 | |
|     int ok = 0, i;
 | |
|     BIGNUM *kinv = NULL, *s, *m = NULL;
 | |
|     const BIGNUM *order, *ckinv;
 | |
|     BN_CTX *ctx = NULL;
 | |
|     const EC_GROUP *group;
 | |
|     ECDSA_SIG *ret;
 | |
|     const BIGNUM *priv_key;
 | |
| 
 | |
|     group = EC_KEY_get0_group(eckey);
 | |
|     priv_key = EC_KEY_get0_private_key(eckey);
 | |
| 
 | |
|     if (group == NULL) {
 | |
|         ECerr(EC_F_OSSL_ECDSA_SIGN_SIG, ERR_R_PASSED_NULL_PARAMETER);
 | |
|         return NULL;
 | |
|     }
 | |
|     if (priv_key == NULL) {
 | |
|         ECerr(EC_F_OSSL_ECDSA_SIGN_SIG, EC_R_MISSING_PRIVATE_KEY);
 | |
|         return NULL;
 | |
|     }
 | |
| 
 | |
|     if (!EC_KEY_can_sign(eckey)) {
 | |
|         ECerr(EC_F_OSSL_ECDSA_SIGN_SIG, EC_R_CURVE_DOES_NOT_SUPPORT_SIGNING);
 | |
|         return NULL;
 | |
|     }
 | |
| 
 | |
|     ret = ECDSA_SIG_new();
 | |
|     if (ret == NULL) {
 | |
|         ECerr(EC_F_OSSL_ECDSA_SIGN_SIG, ERR_R_MALLOC_FAILURE);
 | |
|         return NULL;
 | |
|     }
 | |
|     ret->r = BN_new();
 | |
|     ret->s = BN_new();
 | |
|     if (ret->r == NULL || ret->s == NULL) {
 | |
|         ECerr(EC_F_OSSL_ECDSA_SIGN_SIG, ERR_R_MALLOC_FAILURE);
 | |
|         goto err;
 | |
|     }
 | |
|     s = ret->s;
 | |
| 
 | |
|     if ((ctx = BN_CTX_new()) == NULL
 | |
|         || (m = BN_new()) == NULL) {
 | |
|         ECerr(EC_F_OSSL_ECDSA_SIGN_SIG, ERR_R_MALLOC_FAILURE);
 | |
|         goto err;
 | |
|     }
 | |
| 
 | |
|     order = EC_GROUP_get0_order(group);
 | |
|     i = BN_num_bits(order);
 | |
|     /*
 | |
|      * Need to truncate digest if it is too long: first truncate whole bytes.
 | |
|      */
 | |
|     if (8 * dgst_len > i)
 | |
|         dgst_len = (i + 7) / 8;
 | |
|     if (!BN_bin2bn(dgst, dgst_len, m)) {
 | |
|         ECerr(EC_F_OSSL_ECDSA_SIGN_SIG, ERR_R_BN_LIB);
 | |
|         goto err;
 | |
|     }
 | |
|     /* If still too long, truncate remaining bits with a shift */
 | |
|     if ((8 * dgst_len > i) && !BN_rshift(m, m, 8 - (i & 0x7))) {
 | |
|         ECerr(EC_F_OSSL_ECDSA_SIGN_SIG, ERR_R_BN_LIB);
 | |
|         goto err;
 | |
|     }
 | |
|     do {
 | |
|         if (in_kinv == NULL || in_r == NULL) {
 | |
|             if (!ecdsa_sign_setup(eckey, ctx, &kinv, &ret->r, dgst, dgst_len)) {
 | |
|                 ECerr(EC_F_OSSL_ECDSA_SIGN_SIG, ERR_R_ECDSA_LIB);
 | |
|                 goto err;
 | |
|             }
 | |
|             ckinv = kinv;
 | |
|         } else {
 | |
|             ckinv = in_kinv;
 | |
|             if (BN_copy(ret->r, in_r) == NULL) {
 | |
|                 ECerr(EC_F_OSSL_ECDSA_SIGN_SIG, ERR_R_MALLOC_FAILURE);
 | |
|                 goto err;
 | |
|             }
 | |
|         }
 | |
| 
 | |
|         /*
 | |
|          * With only one multiplicant being in Montgomery domain
 | |
|          * multiplication yields real result without post-conversion.
 | |
|          * Also note that all operations but last are performed with
 | |
|          * zero-padded vectors. Last operation, BN_mod_mul_montgomery
 | |
|          * below, returns user-visible value with removed zero padding.
 | |
|          */
 | |
|         if (!bn_to_mont_fixed_top(s, ret->r, group->mont_data, ctx)
 | |
|             || !bn_mul_mont_fixed_top(s, s, priv_key, group->mont_data, ctx)) {
 | |
|             ECerr(EC_F_OSSL_ECDSA_SIGN_SIG, ERR_R_BN_LIB);
 | |
|             goto err;
 | |
|         }
 | |
|         if (!bn_mod_add_fixed_top(s, s, m, order)) {
 | |
|             ECerr(EC_F_OSSL_ECDSA_SIGN_SIG, ERR_R_BN_LIB);
 | |
|             goto err;
 | |
|         }
 | |
|         /*
 | |
|          * |s| can still be larger than modulus, because |m| can be. In
 | |
|          * such case we count on Montgomery reduction to tie it up.
 | |
|          */
 | |
|         if (!bn_to_mont_fixed_top(s, s, group->mont_data, ctx)
 | |
|             || !BN_mod_mul_montgomery(s, s, ckinv, group->mont_data, ctx)) {
 | |
|             ECerr(EC_F_OSSL_ECDSA_SIGN_SIG, ERR_R_BN_LIB);
 | |
|             goto err;
 | |
|         }
 | |
| 
 | |
|         if (BN_is_zero(s)) {
 | |
|             /*
 | |
|              * if kinv and r have been supplied by the caller, don't
 | |
|              * generate new kinv and r values
 | |
|              */
 | |
|             if (in_kinv != NULL && in_r != NULL) {
 | |
|                 ECerr(EC_F_OSSL_ECDSA_SIGN_SIG, EC_R_NEED_NEW_SETUP_VALUES);
 | |
|                 goto err;
 | |
|             }
 | |
|         } else {
 | |
|             /* s != 0 => we have a valid signature */
 | |
|             break;
 | |
|         }
 | |
|     } while (1);
 | |
| 
 | |
|     ok = 1;
 | |
|  err:
 | |
|     if (!ok) {
 | |
|         ECDSA_SIG_free(ret);
 | |
|         ret = NULL;
 | |
|     }
 | |
|     BN_CTX_free(ctx);
 | |
|     BN_clear_free(m);
 | |
|     BN_clear_free(kinv);
 | |
|     return ret;
 | |
| }
 | |
| 
 | |
| /*-
 | |
|  * returns
 | |
|  *      1: correct signature
 | |
|  *      0: incorrect signature
 | |
|  *     -1: error
 | |
|  */
 | |
| int ossl_ecdsa_verify(int type, const unsigned char *dgst, int dgst_len,
 | |
|                       const unsigned char *sigbuf, int sig_len, EC_KEY *eckey)
 | |
| {
 | |
|     ECDSA_SIG *s;
 | |
|     const unsigned char *p = sigbuf;
 | |
|     unsigned char *der = NULL;
 | |
|     int derlen = -1;
 | |
|     int ret = -1;
 | |
| 
 | |
|     s = ECDSA_SIG_new();
 | |
|     if (s == NULL)
 | |
|         return ret;
 | |
|     if (d2i_ECDSA_SIG(&s, &p, sig_len) == NULL)
 | |
|         goto err;
 | |
|     /* Ensure signature uses DER and doesn't have trailing garbage */
 | |
|     derlen = i2d_ECDSA_SIG(s, &der);
 | |
|     if (derlen != sig_len || memcmp(sigbuf, der, derlen) != 0)
 | |
|         goto err;
 | |
|     ret = ECDSA_do_verify(dgst, dgst_len, s, eckey);
 | |
|  err:
 | |
|     OPENSSL_free(der);
 | |
|     ECDSA_SIG_free(s);
 | |
|     return ret;
 | |
| }
 | |
| 
 | |
| int ossl_ecdsa_verify_sig(const unsigned char *dgst, int dgst_len,
 | |
|                           const ECDSA_SIG *sig, EC_KEY *eckey)
 | |
| {
 | |
|     int ret = -1, i;
 | |
|     BN_CTX *ctx;
 | |
|     const BIGNUM *order;
 | |
|     BIGNUM *u1, *u2, *m, *X;
 | |
|     EC_POINT *point = NULL;
 | |
|     const EC_GROUP *group;
 | |
|     const EC_POINT *pub_key;
 | |
| 
 | |
|     /* check input values */
 | |
|     if (eckey == NULL || (group = EC_KEY_get0_group(eckey)) == NULL ||
 | |
|         (pub_key = EC_KEY_get0_public_key(eckey)) == NULL || sig == NULL) {
 | |
|         ECerr(EC_F_OSSL_ECDSA_VERIFY_SIG, EC_R_MISSING_PARAMETERS);
 | |
|         return -1;
 | |
|     }
 | |
| 
 | |
|     if (!EC_KEY_can_sign(eckey)) {
 | |
|         ECerr(EC_F_OSSL_ECDSA_VERIFY_SIG, EC_R_CURVE_DOES_NOT_SUPPORT_SIGNING);
 | |
|         return -1;
 | |
|     }
 | |
| 
 | |
|     ctx = BN_CTX_new();
 | |
|     if (ctx == NULL) {
 | |
|         ECerr(EC_F_OSSL_ECDSA_VERIFY_SIG, ERR_R_MALLOC_FAILURE);
 | |
|         return -1;
 | |
|     }
 | |
|     BN_CTX_start(ctx);
 | |
|     u1 = BN_CTX_get(ctx);
 | |
|     u2 = BN_CTX_get(ctx);
 | |
|     m = BN_CTX_get(ctx);
 | |
|     X = BN_CTX_get(ctx);
 | |
|     if (X == NULL) {
 | |
|         ECerr(EC_F_OSSL_ECDSA_VERIFY_SIG, ERR_R_BN_LIB);
 | |
|         goto err;
 | |
|     }
 | |
| 
 | |
|     order = EC_GROUP_get0_order(group);
 | |
|     if (order == NULL) {
 | |
|         ECerr(EC_F_OSSL_ECDSA_VERIFY_SIG, ERR_R_EC_LIB);
 | |
|         goto err;
 | |
|     }
 | |
| 
 | |
|     if (BN_is_zero(sig->r) || BN_is_negative(sig->r) ||
 | |
|         BN_ucmp(sig->r, order) >= 0 || BN_is_zero(sig->s) ||
 | |
|         BN_is_negative(sig->s) || BN_ucmp(sig->s, order) >= 0) {
 | |
|         ECerr(EC_F_OSSL_ECDSA_VERIFY_SIG, EC_R_BAD_SIGNATURE);
 | |
|         ret = 0;                /* signature is invalid */
 | |
|         goto err;
 | |
|     }
 | |
|     /* calculate tmp1 = inv(S) mod order */
 | |
|     if (!ec_group_do_inverse_ord(group, u2, sig->s, ctx)) {
 | |
|         ECerr(EC_F_OSSL_ECDSA_VERIFY_SIG, ERR_R_BN_LIB);
 | |
|         goto err;
 | |
|     }
 | |
|     /* digest -> m */
 | |
|     i = BN_num_bits(order);
 | |
|     /*
 | |
|      * Need to truncate digest if it is too long: first truncate whole bytes.
 | |
|      */
 | |
|     if (8 * dgst_len > i)
 | |
|         dgst_len = (i + 7) / 8;
 | |
|     if (!BN_bin2bn(dgst, dgst_len, m)) {
 | |
|         ECerr(EC_F_OSSL_ECDSA_VERIFY_SIG, ERR_R_BN_LIB);
 | |
|         goto err;
 | |
|     }
 | |
|     /* If still too long truncate remaining bits with a shift */
 | |
|     if ((8 * dgst_len > i) && !BN_rshift(m, m, 8 - (i & 0x7))) {
 | |
|         ECerr(EC_F_OSSL_ECDSA_VERIFY_SIG, ERR_R_BN_LIB);
 | |
|         goto err;
 | |
|     }
 | |
|     /* u1 = m * tmp mod order */
 | |
|     if (!BN_mod_mul(u1, m, u2, order, ctx)) {
 | |
|         ECerr(EC_F_OSSL_ECDSA_VERIFY_SIG, ERR_R_BN_LIB);
 | |
|         goto err;
 | |
|     }
 | |
|     /* u2 = r * w mod q */
 | |
|     if (!BN_mod_mul(u2, sig->r, u2, order, ctx)) {
 | |
|         ECerr(EC_F_OSSL_ECDSA_VERIFY_SIG, ERR_R_BN_LIB);
 | |
|         goto err;
 | |
|     }
 | |
| 
 | |
|     if ((point = EC_POINT_new(group)) == NULL) {
 | |
|         ECerr(EC_F_OSSL_ECDSA_VERIFY_SIG, ERR_R_MALLOC_FAILURE);
 | |
|         goto err;
 | |
|     }
 | |
|     if (!EC_POINT_mul(group, point, u1, pub_key, u2, ctx)) {
 | |
|         ECerr(EC_F_OSSL_ECDSA_VERIFY_SIG, ERR_R_EC_LIB);
 | |
|         goto err;
 | |
|     }
 | |
| 
 | |
|     if (!EC_POINT_get_affine_coordinates(group, point, X, NULL, ctx)) {
 | |
|         ECerr(EC_F_OSSL_ECDSA_VERIFY_SIG, ERR_R_EC_LIB);
 | |
|         goto err;
 | |
|     }
 | |
| 
 | |
|     if (!BN_nnmod(u1, X, order, ctx)) {
 | |
|         ECerr(EC_F_OSSL_ECDSA_VERIFY_SIG, ERR_R_BN_LIB);
 | |
|         goto err;
 | |
|     }
 | |
|     /*  if the signature is correct u1 is equal to sig->r */
 | |
|     ret = (BN_ucmp(u1, sig->r) == 0);
 | |
|  err:
 | |
|     BN_CTX_end(ctx);
 | |
|     BN_CTX_free(ctx);
 | |
|     EC_POINT_free(point);
 | |
|     return ret;
 | |
| }
 |