external/srs
1
0
Fork 0
mirror of https://github.com/ossrs/srs.git synced 2025-03-09 15:49:59 +00:00
Code Issues Releases Wiki Activity

For #774, refine code style

Browse source
This commit is contained in:
winlin 2017-02-25 12:10:41 +08:00
parent 01d5e4da17
commit ac0e27b936
1 changed files with 45 additions and 39 deletions
Download patch file Download diff file Expand all files Collapse all files

84
trunk/src/protocol/srs_rtmp_handshake.cpp
View file

@ -45,63 +45,69 @@ using namespace _srs_internal;
#include <openssl/dh.h> #include <openssl/dh.h>
#if OPENSSL_VERSION_NUMBER < 0x10100000L #if OPENSSL_VERSION_NUMBER < 0x10100000L
static HMAC_CTX *HMAC_CTX_new(void) static HMAC_CTX *HMAC_CTX_new(void)
{ {
HMAC_CTX *ctx = (HMAC_CTX *)malloc(sizeof(*ctx)); HMAC_CTX *ctx = (HMAC_CTX *)malloc(sizeof(*ctx));
if (ctx != NULL) if (ctx != NULL) {
HMAC_CTX_init(ctx); HMAC_CTX_init(ctx);
return ctx; }
return ctx;
} }
static void HMAC_CTX_free(HMAC_CTX *ctx) static void HMAC_CTX_free(HMAC_CTX *ctx)
{ {
if (ctx != NULL) { if (ctx != NULL) {
HMAC_CTX_cleanup(ctx); HMAC_CTX_cleanup(ctx);
free(ctx); free(ctx);
} }
} }
static void DH_get0_key(const DH *dh, const BIGNUM **pub_key, const BIGNUM **priv_key) static void DH_get0_key(const DH *dh, const BIGNUM **pub_key, const BIGNUM **priv_key)
{ {
if (pub_key != NULL) if (pub_key != NULL) {
*pub_key = dh->pub_key; *pub_key = dh->pub_key;
if (priv_key != NULL) }
*priv_key = dh->priv_key; if (priv_key != NULL) {
*priv_key = dh->priv_key;
}
} }
static int DH_set0_pqg(DH *dh, BIGNUM *p, BIGNUM *q, BIGNUM *g) static int DH_set0_pqg(DH *dh, BIGNUM *p, BIGNUM *q, BIGNUM *g)
{ {
/* If the fields p and g in d are NULL, the corresponding input /* If the fields p and g in d are NULL, the corresponding input
* parameters MUST be non-NULL. q may remain NULL. * parameters MUST be non-NULL. q may remain NULL.
*/ */
if ((dh->p == NULL && p == NULL) if ((dh->p == NULL && p == NULL)
|| (dh->g == NULL && g == NULL)) || (dh->g == NULL && g == NULL))
return 0; return 0;
if (p != NULL) { if (p != NULL) {
BN_free(dh->p); BN_free(dh->p);
dh->p = p; dh->p = p;
} }
if (q != NULL) { if (q != NULL) {
BN_free(dh->q); BN_free(dh->q);
dh->q = q; dh->q = q;
} }
if (g != NULL) { if (g != NULL) {
BN_free(dh->g); BN_free(dh->g);
dh->g = g; dh->g = g;
} }
if (q != NULL) { if (q != NULL) {
dh->length = BN_num_bits(q); dh->length = BN_num_bits(q);
} }
return 1; return 1;
} }
static int DH_set_length(DH *dh, long length) static int DH_set_length(DH *dh, long length)
{ {
dh->length = length; dh->length = length;
return 1; return 1;
} }
#endif #endif
namespace _srs_internal namespace _srs_internal