mirror of
				https://github.com/ossrs/srs.git
				synced 2025-03-09 15:49:59 +00:00 
			
		
		
		
	
		
			
				
	
	
		
			393 lines
		
	
	
	
		
			12 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
			
		
		
	
	
			393 lines
		
	
	
	
		
			12 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
| /*
 | |
|  * Copyright 1995-2018 The OpenSSL Project Authors. All Rights Reserved.
 | |
|  * Copyright 2005 Nokia. 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 <stdio.h>
 | |
| #include <stdlib.h>
 | |
| #include "ssl_local.h"
 | |
| #include <openssl/asn1t.h>
 | |
| #include <openssl/x509.h>
 | |
| 
 | |
| typedef struct {
 | |
|     uint32_t version;
 | |
|     int32_t ssl_version;
 | |
|     ASN1_OCTET_STRING *cipher;
 | |
|     ASN1_OCTET_STRING *comp_id;
 | |
|     ASN1_OCTET_STRING *master_key;
 | |
|     ASN1_OCTET_STRING *session_id;
 | |
|     ASN1_OCTET_STRING *key_arg;
 | |
|     int64_t time;
 | |
|     int64_t timeout;
 | |
|     X509 *peer;
 | |
|     ASN1_OCTET_STRING *session_id_context;
 | |
|     int32_t verify_result;
 | |
|     ASN1_OCTET_STRING *tlsext_hostname;
 | |
|     uint64_t tlsext_tick_lifetime_hint;
 | |
|     uint32_t tlsext_tick_age_add;
 | |
|     ASN1_OCTET_STRING *tlsext_tick;
 | |
| #ifndef OPENSSL_NO_PSK
 | |
|     ASN1_OCTET_STRING *psk_identity_hint;
 | |
|     ASN1_OCTET_STRING *psk_identity;
 | |
| #endif
 | |
| #ifndef OPENSSL_NO_SRP
 | |
|     ASN1_OCTET_STRING *srp_username;
 | |
| #endif
 | |
|     uint64_t flags;
 | |
|     uint32_t max_early_data;
 | |
|     ASN1_OCTET_STRING *alpn_selected;
 | |
|     uint32_t tlsext_max_fragment_len_mode;
 | |
|     ASN1_OCTET_STRING *ticket_appdata;
 | |
| } SSL_SESSION_ASN1;
 | |
| 
 | |
| ASN1_SEQUENCE(SSL_SESSION_ASN1) = {
 | |
|     ASN1_EMBED(SSL_SESSION_ASN1, version, UINT32),
 | |
|     ASN1_EMBED(SSL_SESSION_ASN1, ssl_version, INT32),
 | |
|     ASN1_SIMPLE(SSL_SESSION_ASN1, cipher, ASN1_OCTET_STRING),
 | |
|     ASN1_SIMPLE(SSL_SESSION_ASN1, session_id, ASN1_OCTET_STRING),
 | |
|     ASN1_SIMPLE(SSL_SESSION_ASN1, master_key, ASN1_OCTET_STRING),
 | |
|     ASN1_IMP_OPT(SSL_SESSION_ASN1, key_arg, ASN1_OCTET_STRING, 0),
 | |
|     ASN1_EXP_OPT_EMBED(SSL_SESSION_ASN1, time, ZINT64, 1),
 | |
|     ASN1_EXP_OPT_EMBED(SSL_SESSION_ASN1, timeout, ZINT64, 2),
 | |
|     ASN1_EXP_OPT(SSL_SESSION_ASN1, peer, X509, 3),
 | |
|     ASN1_EXP_OPT(SSL_SESSION_ASN1, session_id_context, ASN1_OCTET_STRING, 4),
 | |
|     ASN1_EXP_OPT_EMBED(SSL_SESSION_ASN1, verify_result, ZINT32, 5),
 | |
|     ASN1_EXP_OPT(SSL_SESSION_ASN1, tlsext_hostname, ASN1_OCTET_STRING, 6),
 | |
| #ifndef OPENSSL_NO_PSK
 | |
|     ASN1_EXP_OPT(SSL_SESSION_ASN1, psk_identity_hint, ASN1_OCTET_STRING, 7),
 | |
|     ASN1_EXP_OPT(SSL_SESSION_ASN1, psk_identity, ASN1_OCTET_STRING, 8),
 | |
| #endif
 | |
|     ASN1_EXP_OPT_EMBED(SSL_SESSION_ASN1, tlsext_tick_lifetime_hint, ZUINT64, 9),
 | |
|     ASN1_EXP_OPT(SSL_SESSION_ASN1, tlsext_tick, ASN1_OCTET_STRING, 10),
 | |
|     ASN1_EXP_OPT(SSL_SESSION_ASN1, comp_id, ASN1_OCTET_STRING, 11),
 | |
| #ifndef OPENSSL_NO_SRP
 | |
|     ASN1_EXP_OPT(SSL_SESSION_ASN1, srp_username, ASN1_OCTET_STRING, 12),
 | |
| #endif
 | |
|     ASN1_EXP_OPT_EMBED(SSL_SESSION_ASN1, flags, ZUINT64, 13),
 | |
|     ASN1_EXP_OPT_EMBED(SSL_SESSION_ASN1, tlsext_tick_age_add, ZUINT32, 14),
 | |
|     ASN1_EXP_OPT_EMBED(SSL_SESSION_ASN1, max_early_data, ZUINT32, 15),
 | |
|     ASN1_EXP_OPT(SSL_SESSION_ASN1, alpn_selected, ASN1_OCTET_STRING, 16),
 | |
|     ASN1_EXP_OPT_EMBED(SSL_SESSION_ASN1, tlsext_max_fragment_len_mode, ZUINT32, 17),
 | |
|     ASN1_EXP_OPT(SSL_SESSION_ASN1, ticket_appdata, ASN1_OCTET_STRING, 18)
 | |
| } static_ASN1_SEQUENCE_END(SSL_SESSION_ASN1)
 | |
| 
 | |
| IMPLEMENT_STATIC_ASN1_ENCODE_FUNCTIONS(SSL_SESSION_ASN1)
 | |
| 
 | |
| /* Utility functions for i2d_SSL_SESSION */
 | |
| 
 | |
| /* Initialise OCTET STRING from buffer and length */
 | |
| 
 | |
| static void ssl_session_oinit(ASN1_OCTET_STRING **dest, ASN1_OCTET_STRING *os,
 | |
|                               unsigned char *data, size_t len)
 | |
| {
 | |
|     os->data = data;
 | |
|     os->length = (int)len;
 | |
|     os->flags = 0;
 | |
|     *dest = os;
 | |
| }
 | |
| 
 | |
| /* Initialise OCTET STRING from string */
 | |
| static void ssl_session_sinit(ASN1_OCTET_STRING **dest, ASN1_OCTET_STRING *os,
 | |
|                               char *data)
 | |
| {
 | |
|     if (data != NULL)
 | |
|         ssl_session_oinit(dest, os, (unsigned char *)data, strlen(data));
 | |
|     else
 | |
|         *dest = NULL;
 | |
| }
 | |
| 
 | |
| int i2d_SSL_SESSION(SSL_SESSION *in, unsigned char **pp)
 | |
| {
 | |
| 
 | |
|     SSL_SESSION_ASN1 as;
 | |
| 
 | |
|     ASN1_OCTET_STRING cipher;
 | |
|     unsigned char cipher_data[2];
 | |
|     ASN1_OCTET_STRING master_key, session_id, sid_ctx;
 | |
| 
 | |
| #ifndef OPENSSL_NO_COMP
 | |
|     ASN1_OCTET_STRING comp_id;
 | |
|     unsigned char comp_id_data;
 | |
| #endif
 | |
|     ASN1_OCTET_STRING tlsext_hostname, tlsext_tick;
 | |
| #ifndef OPENSSL_NO_SRP
 | |
|     ASN1_OCTET_STRING srp_username;
 | |
| #endif
 | |
| #ifndef OPENSSL_NO_PSK
 | |
|     ASN1_OCTET_STRING psk_identity, psk_identity_hint;
 | |
| #endif
 | |
|     ASN1_OCTET_STRING alpn_selected;
 | |
|     ASN1_OCTET_STRING ticket_appdata;
 | |
| 
 | |
|     long l;
 | |
| 
 | |
|     if ((in == NULL) || ((in->cipher == NULL) && (in->cipher_id == 0)))
 | |
|         return 0;
 | |
| 
 | |
|     memset(&as, 0, sizeof(as));
 | |
| 
 | |
|     as.version = SSL_SESSION_ASN1_VERSION;
 | |
|     as.ssl_version = in->ssl_version;
 | |
| 
 | |
|     if (in->cipher == NULL)
 | |
|         l = in->cipher_id;
 | |
|     else
 | |
|         l = in->cipher->id;
 | |
|     cipher_data[0] = ((unsigned char)(l >> 8L)) & 0xff;
 | |
|     cipher_data[1] = ((unsigned char)(l)) & 0xff;
 | |
| 
 | |
|     ssl_session_oinit(&as.cipher, &cipher, cipher_data, 2);
 | |
| 
 | |
| #ifndef OPENSSL_NO_COMP
 | |
|     if (in->compress_meth) {
 | |
|         comp_id_data = (unsigned char)in->compress_meth;
 | |
|         ssl_session_oinit(&as.comp_id, &comp_id, &comp_id_data, 1);
 | |
|     }
 | |
| #endif
 | |
| 
 | |
|     ssl_session_oinit(&as.master_key, &master_key,
 | |
|                       in->master_key, in->master_key_length);
 | |
| 
 | |
|     ssl_session_oinit(&as.session_id, &session_id,
 | |
|                       in->session_id, in->session_id_length);
 | |
| 
 | |
|     ssl_session_oinit(&as.session_id_context, &sid_ctx,
 | |
|                       in->sid_ctx, in->sid_ctx_length);
 | |
| 
 | |
|     as.time = in->time;
 | |
|     as.timeout = in->timeout;
 | |
|     as.verify_result = in->verify_result;
 | |
| 
 | |
|     as.peer = in->peer;
 | |
| 
 | |
|     ssl_session_sinit(&as.tlsext_hostname, &tlsext_hostname,
 | |
|                       in->ext.hostname);
 | |
|     if (in->ext.tick) {
 | |
|         ssl_session_oinit(&as.tlsext_tick, &tlsext_tick,
 | |
|                           in->ext.tick, in->ext.ticklen);
 | |
|     }
 | |
|     if (in->ext.tick_lifetime_hint > 0)
 | |
|         as.tlsext_tick_lifetime_hint = in->ext.tick_lifetime_hint;
 | |
|     as.tlsext_tick_age_add = in->ext.tick_age_add;
 | |
| #ifndef OPENSSL_NO_PSK
 | |
|     ssl_session_sinit(&as.psk_identity_hint, &psk_identity_hint,
 | |
|                       in->psk_identity_hint);
 | |
|     ssl_session_sinit(&as.psk_identity, &psk_identity, in->psk_identity);
 | |
| #endif                          /* OPENSSL_NO_PSK */
 | |
| #ifndef OPENSSL_NO_SRP
 | |
|     ssl_session_sinit(&as.srp_username, &srp_username, in->srp_username);
 | |
| #endif                          /* OPENSSL_NO_SRP */
 | |
| 
 | |
|     as.flags = in->flags;
 | |
|     as.max_early_data = in->ext.max_early_data;
 | |
| 
 | |
|     if (in->ext.alpn_selected == NULL)
 | |
|         as.alpn_selected = NULL;
 | |
|     else
 | |
|         ssl_session_oinit(&as.alpn_selected, &alpn_selected,
 | |
|                           in->ext.alpn_selected, in->ext.alpn_selected_len);
 | |
| 
 | |
|     as.tlsext_max_fragment_len_mode = in->ext.max_fragment_len_mode;
 | |
| 
 | |
|     if (in->ticket_appdata == NULL)
 | |
|         as.ticket_appdata = NULL;
 | |
|     else
 | |
|         ssl_session_oinit(&as.ticket_appdata, &ticket_appdata,
 | |
|                           in->ticket_appdata, in->ticket_appdata_len);
 | |
| 
 | |
|     return i2d_SSL_SESSION_ASN1(&as, pp);
 | |
| 
 | |
| }
 | |
| 
 | |
| /* Utility functions for d2i_SSL_SESSION */
 | |
| 
 | |
| /* OPENSSL_strndup an OCTET STRING */
 | |
| 
 | |
| static int ssl_session_strndup(char **pdst, ASN1_OCTET_STRING *src)
 | |
| {
 | |
|     OPENSSL_free(*pdst);
 | |
|     *pdst = NULL;
 | |
|     if (src == NULL)
 | |
|         return 1;
 | |
|     *pdst = OPENSSL_strndup((char *)src->data, src->length);
 | |
|     if (*pdst == NULL)
 | |
|         return 0;
 | |
|     return 1;
 | |
| }
 | |
| 
 | |
| /* Copy an OCTET STRING, return error if it exceeds maximum length */
 | |
| 
 | |
| static int ssl_session_memcpy(unsigned char *dst, size_t *pdstlen,
 | |
|                               ASN1_OCTET_STRING *src, size_t maxlen)
 | |
| {
 | |
|     if (src == NULL) {
 | |
|         *pdstlen = 0;
 | |
|         return 1;
 | |
|     }
 | |
|     if (src->length < 0 || src->length > (int)maxlen)
 | |
|         return 0;
 | |
|     memcpy(dst, src->data, src->length);
 | |
|     *pdstlen = src->length;
 | |
|     return 1;
 | |
| }
 | |
| 
 | |
| SSL_SESSION *d2i_SSL_SESSION(SSL_SESSION **a, const unsigned char **pp,
 | |
|                              long length)
 | |
| {
 | |
|     long id;
 | |
|     size_t tmpl;
 | |
|     const unsigned char *p = *pp;
 | |
|     SSL_SESSION_ASN1 *as = NULL;
 | |
|     SSL_SESSION *ret = NULL;
 | |
| 
 | |
|     as = d2i_SSL_SESSION_ASN1(NULL, &p, length);
 | |
|     /* ASN.1 code returns suitable error */
 | |
|     if (as == NULL)
 | |
|         goto err;
 | |
| 
 | |
|     if (!a || !*a) {
 | |
|         ret = SSL_SESSION_new();
 | |
|         if (ret == NULL)
 | |
|             goto err;
 | |
|     } else {
 | |
|         ret = *a;
 | |
|     }
 | |
| 
 | |
|     if (as->version != SSL_SESSION_ASN1_VERSION) {
 | |
|         SSLerr(SSL_F_D2I_SSL_SESSION, SSL_R_UNKNOWN_SSL_VERSION);
 | |
|         goto err;
 | |
|     }
 | |
| 
 | |
|     if ((as->ssl_version >> 8) != SSL3_VERSION_MAJOR
 | |
|         && (as->ssl_version >> 8) != DTLS1_VERSION_MAJOR
 | |
|         && as->ssl_version != DTLS1_BAD_VER) {
 | |
|         SSLerr(SSL_F_D2I_SSL_SESSION, SSL_R_UNSUPPORTED_SSL_VERSION);
 | |
|         goto err;
 | |
|     }
 | |
| 
 | |
|     ret->ssl_version = (int)as->ssl_version;
 | |
| 
 | |
|     if (as->cipher->length != 2) {
 | |
|         SSLerr(SSL_F_D2I_SSL_SESSION, SSL_R_CIPHER_CODE_WRONG_LENGTH);
 | |
|         goto err;
 | |
|     }
 | |
| 
 | |
|     id = 0x03000000L | ((unsigned long)as->cipher->data[0] << 8L)
 | |
|                      | (unsigned long)as->cipher->data[1];
 | |
| 
 | |
|     ret->cipher_id = id;
 | |
|     ret->cipher = ssl3_get_cipher_by_id(id);
 | |
|     if (ret->cipher == NULL)
 | |
|         goto err;
 | |
| 
 | |
|     if (!ssl_session_memcpy(ret->session_id, &ret->session_id_length,
 | |
|                             as->session_id, SSL3_MAX_SSL_SESSION_ID_LENGTH))
 | |
|         goto err;
 | |
| 
 | |
|     if (!ssl_session_memcpy(ret->master_key, &tmpl,
 | |
|                             as->master_key, TLS13_MAX_RESUMPTION_PSK_LENGTH))
 | |
|         goto err;
 | |
| 
 | |
|     ret->master_key_length = tmpl;
 | |
| 
 | |
|     if (as->time != 0)
 | |
|         ret->time = (long)as->time;
 | |
|     else
 | |
|         ret->time = (long)time(NULL);
 | |
| 
 | |
|     if (as->timeout != 0)
 | |
|         ret->timeout = (long)as->timeout;
 | |
|     else
 | |
|         ret->timeout = 3;
 | |
| 
 | |
|     X509_free(ret->peer);
 | |
|     ret->peer = as->peer;
 | |
|     as->peer = NULL;
 | |
| 
 | |
|     if (!ssl_session_memcpy(ret->sid_ctx, &ret->sid_ctx_length,
 | |
|                             as->session_id_context, SSL_MAX_SID_CTX_LENGTH))
 | |
|         goto err;
 | |
| 
 | |
|     /* NB: this defaults to zero which is X509_V_OK */
 | |
|     ret->verify_result = as->verify_result;
 | |
| 
 | |
|     if (!ssl_session_strndup(&ret->ext.hostname, as->tlsext_hostname))
 | |
|         goto err;
 | |
| 
 | |
| #ifndef OPENSSL_NO_PSK
 | |
|     if (!ssl_session_strndup(&ret->psk_identity_hint, as->psk_identity_hint))
 | |
|         goto err;
 | |
|     if (!ssl_session_strndup(&ret->psk_identity, as->psk_identity))
 | |
|         goto err;
 | |
| #endif
 | |
| 
 | |
|     ret->ext.tick_lifetime_hint = (unsigned long)as->tlsext_tick_lifetime_hint;
 | |
|     ret->ext.tick_age_add = as->tlsext_tick_age_add;
 | |
|     OPENSSL_free(ret->ext.tick);
 | |
|     if (as->tlsext_tick != NULL) {
 | |
|         ret->ext.tick = as->tlsext_tick->data;
 | |
|         ret->ext.ticklen = as->tlsext_tick->length;
 | |
|         as->tlsext_tick->data = NULL;
 | |
|     } else {
 | |
|         ret->ext.tick = NULL;
 | |
|     }
 | |
| #ifndef OPENSSL_NO_COMP
 | |
|     if (as->comp_id) {
 | |
|         if (as->comp_id->length != 1) {
 | |
|             SSLerr(SSL_F_D2I_SSL_SESSION, SSL_R_BAD_LENGTH);
 | |
|             goto err;
 | |
|         }
 | |
|         ret->compress_meth = as->comp_id->data[0];
 | |
|     } else {
 | |
|         ret->compress_meth = 0;
 | |
|     }
 | |
| #endif
 | |
| 
 | |
| #ifndef OPENSSL_NO_SRP
 | |
|     if (!ssl_session_strndup(&ret->srp_username, as->srp_username))
 | |
|         goto err;
 | |
| #endif                          /* OPENSSL_NO_SRP */
 | |
|     /* Flags defaults to zero which is fine */
 | |
|     ret->flags = (int32_t)as->flags;
 | |
|     ret->ext.max_early_data = as->max_early_data;
 | |
| 
 | |
|     OPENSSL_free(ret->ext.alpn_selected);
 | |
|     if (as->alpn_selected != NULL) {
 | |
|         ret->ext.alpn_selected = as->alpn_selected->data;
 | |
|         ret->ext.alpn_selected_len = as->alpn_selected->length;
 | |
|         as->alpn_selected->data = NULL;
 | |
|     } else {
 | |
|         ret->ext.alpn_selected = NULL;
 | |
|         ret->ext.alpn_selected_len = 0;
 | |
|     }
 | |
| 
 | |
|     ret->ext.max_fragment_len_mode = as->tlsext_max_fragment_len_mode;
 | |
| 
 | |
|     OPENSSL_free(ret->ticket_appdata);
 | |
|     if (as->ticket_appdata != NULL) {
 | |
|         ret->ticket_appdata = as->ticket_appdata->data;
 | |
|         ret->ticket_appdata_len = as->ticket_appdata->length;
 | |
|         as->ticket_appdata->data = NULL;
 | |
|     } else {
 | |
|         ret->ticket_appdata = NULL;
 | |
|         ret->ticket_appdata_len = 0;
 | |
|     }
 | |
| 
 | |
|     M_ASN1_free_of(as, SSL_SESSION_ASN1);
 | |
| 
 | |
|     if ((a != NULL) && (*a == NULL))
 | |
|         *a = ret;
 | |
|     *pp = p;
 | |
|     return ret;
 | |
| 
 | |
|  err:
 | |
|     M_ASN1_free_of(as, SSL_SESSION_ASN1);
 | |
|     if ((a == NULL) || (*a != ret))
 | |
|         SSL_SESSION_free(ret);
 | |
|     return NULL;
 | |
| }
 |