RepoMirrors/mediamtx
1
0
Fork 0
mirror of https://github.com/bluenviron/mediamtx synced 2025-01-31 11:41:59 +00:00
Code Issues Packages Projects Releases Wiki Activity

rtmp: fix RTMPE handshake error when a public key starts with zero (#2269)

Browse Source
This commit is contained in:
Alessandro Ros 2023-08-30 19:47:48 +02:00 committed by GitHub
parent f69be81e20
commit fc353ce66b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 12 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
internal/rtmp/handshake/dh.go
View File

@ -1,6 +1,7 @@
package handshake
import (
"bytes"
"crypto/rand"
"fmt"
"math/big"
@ -100,6 +101,10 @@ func dhGenerateKeyPair() ([]byte, []byte, error) {
y.Exp(&g, &x, &p)
pub := y.Bytes()
if len(pub) < dhKeyLength {
pub = append(bytes.Repeat([]byte{0}, dhKeyLength-len(pub)), pub...)
}
return priv, pub, nil
}
@ -114,5 +119,11 @@ func dhComputeSharedSecret(priv []byte, pub []byte) []byte {
p.SetBytes(p1024)
var z big.Int
z.Exp(&y, &x, &p)
return z.Bytes()
sec := z.Bytes()
if len(sec) < dhKeyLength {
sec = append(bytes.Repeat([]byte{0}, dhKeyLength-len(sec)), sec...)
}
return sec
}