summaryrefslogtreecommitdiff
path: root/libavformat/rtmpdh.c
diff options
context:
space:
mode:
authorMartin Storsjö <martin@martin.st>2015-05-28 10:11:27 +0300
committerMartin Storsjö <martin@martin.st>2015-05-29 09:42:32 +0300
commit0508faaa11bf7507ffdd655aee57c9dc5a8203f4 (patch)
tree0dbb44c924f107ee9094c4caa078d504a2930c23 /libavformat/rtmpdh.c
parent9f1b3050d9e31e9283d818f3640f3460ac8cfb5b (diff)
rtmpdh: Pass the actual buffer size of the output secret key
Signed-off-by: Martin Storsjö <martin@martin.st>
Diffstat (limited to 'libavformat/rtmpdh.c')
-rw-r--r--libavformat/rtmpdh.c5
1 files changed, 3 insertions, 2 deletions
diff --git a/libavformat/rtmpdh.c b/libavformat/rtmpdh.c
index 80cced9312..f328e68fbf 100644
--- a/libavformat/rtmpdh.c
+++ b/libavformat/rtmpdh.c
@@ -313,7 +313,8 @@ int ff_dh_write_public_key(FF_DH *dh, uint8_t *pub_key, int pub_key_len)
}
int ff_dh_compute_shared_secret_key(FF_DH *dh, const uint8_t *pub_key,
- int pub_key_len, uint8_t *secret_key)
+ int pub_key_len, uint8_t *secret_key,
+ int secret_key_len)
{
FFBigNum q1 = NULL, pub_key_bn = NULL;
int ret;
@@ -333,7 +334,7 @@ int ff_dh_compute_shared_secret_key(FF_DH *dh, const uint8_t *pub_key,
/* when the public key is valid we have to compute the shared secret key */
if ((ret = dh_is_valid_public_key(pub_key_bn, dh->p, q1)) < 0) {
goto fail;
- } else if ((ret = dh_compute_key(dh, pub_key_bn, pub_key_len,
+ } else if ((ret = dh_compute_key(dh, pub_key_bn, secret_key_len,
secret_key)) < 0) {
ret = AVERROR(EINVAL);
goto fail;