From: hyc <hyc@400ebc74-4327-4243-bc38-086b20814532>
Date: Sun, 30 May 2010 21:14:28 +0000 (+0000)
Subject: Fix RC4 teardown for GnuTLS/gcrypt
X-Git-Tag: v2.4~66
X-Git-Url: https://granicus.if.org/sourcecode?a=commitdiff_plain;h=4464675526d8076c6a77bc49fefae9771a481317;p=rtmpdump

Fix RC4 teardown for GnuTLS/gcrypt


git-svn-id: svn://svn.mplayerhq.hu/rtmpdump/trunk@498 400ebc74-4327-4243-bc38-086b20814532
---

diff --git a/librtmp/handshake.h b/librtmp/handshake.h
index c246286..e17c2be 100644
--- a/librtmp/handshake.h
+++ b/librtmp/handshake.h
@@ -35,10 +35,11 @@
 #define HMAC_finish(ctx, dig, dlen)	dlen = SHA256_DIGEST_LENGTH; sha2_hmac_finish(&ctx, dig)
 
 typedef arc4_context *	RC4_handle;
-#define RC4_setup(h)	*h = malloc(sizeof(arc4_context))
+#define RC4_alloc(h)	*h = malloc(sizeof(arc4_context))
 #define RC4_setkey(h,l,k)	arc4_setup(h,k,l)
 #define RC4_encrypt(h,l,d)	arc4_crypt(h,l,(unsigned char *)d,(unsigned char *)d)
 #define RC4_encrypt2(h,l,s,d)	arc4_crypt(h,l,(unsigned char *)s,(unsigned char *)d)
+#define RC4_free(h)	free(h)
 
 #elif defined(USE_GNUTLS)
 #include <gcrypt.h>
@@ -51,10 +52,11 @@ typedef arc4_context *	RC4_handle;
 #define HMAC_finish(ctx, dig, dlen)	dlen = SHA256_DIGEST_LENGTH; memcpy(dig, gcry_md_read(ctx, 0), dlen); gcry_md_close(ctx)
 
 typedef gcry_cipher_hd_t	RC4_handle;
-#define	RC4_setup(h)	gcry_cipher_open(h, GCRY_CIPHER_ARCFOUR, GCRY_CIPHER_MODE_STREAM, 0)
+#define	RC4_alloc(h)	gcry_cipher_open(h, GCRY_CIPHER_ARCFOUR, GCRY_CIPHER_MODE_STREAM, 0)
 #define RC4_setkey(h,l,k)	gcry_cipher_setkey(h,k,l)
 #define RC4_encrypt(h,l,d)	gcry_cipher_encrypt(h,(void *)d,l,NULL,0)
 #define RC4_encrypt2(h,l,s,d)	gcry_cipher_encrypt(h,(void *)d,l,(void *)s,l)
+#define RC4_free(h)	gcry_cipher_close(h)
 
 #else	/* USE_OPENSSL */
 #include <openssl/sha.h>
@@ -68,10 +70,11 @@ typedef gcry_cipher_hd_t	RC4_handle;
 #define HMAC_finish(ctx, dig, dlen)	HMAC_Final(&ctx, dig, &dlen); HMAC_CTX_cleanup(&ctx)
 
 typedef RC4_KEY *	RC4_handle;
-#define RC4_setup(h)	*h = malloc(sizeof(RC4_KEY))
+#define RC4_alloc(h)	*h = malloc(sizeof(RC4_KEY))
 #define RC4_setkey(h,l,k)	RC4_set_key(h,l,k)
 #define RC4_encrypt(h,l,d)	RC4(h,l,(uint8_t *)d,(uint8_t *)d)
 #define RC4_encrypt2(h,l,s,d)	RC4(h,l,(uint8_t *)s,(uint8_t *)d)
+#define RC4_free(h)	free(h)
 #endif
 
 #define FP10
@@ -112,8 +115,8 @@ static void InitRC4Encryption
   unsigned int digestLen = 0;
   HMAC_CTX ctx;
 
-  RC4_setup(rc4keyIn);
-  RC4_setup(rc4keyOut);
+  RC4_alloc(rc4keyIn);
+  RC4_alloc(rc4keyOut);
 
   HMAC_setup(ctx, secretKey, 128);
   HMAC_crunch(ctx, pubKeyIn, 128);
diff --git a/librtmp/rtmp.c b/librtmp/rtmp.c
index 1089636..2bf9868 100644
--- a/librtmp/rtmp.c
+++ b/librtmp/rtmp.c
@@ -3384,12 +3384,12 @@ RTMP_Close(RTMP *r)
     }
   if (r->Link.rc4keyIn)
     {
-      free(r->Link.rc4keyIn);
+      RC4_free(r->Link.rc4keyIn);
       r->Link.rc4keyIn = NULL;
     }
   if (r->Link.rc4keyOut)
     {
-      free(r->Link.rc4keyOut);
+      RC4_free(r->Link.rc4keyOut);
       r->Link.rc4keyOut = NULL;
     }
 #endif