apps/engine
1
0
Fork 0
mirror of https://github.com/Monibuca/engine.git synced 2025-10-06 00:56:58 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

修复内置鉴权，音频轨道因IDR锁环并发问题，增加Internal配置，mp4重放能力

Browse Source
This commit is contained in:
langhuihui
2023-04-29 21:57:40 +08:00
parent 2eea5c3706
commit da92a2238f
12 changed files with 50 additions and 35 deletions
Download Patch File Download Diff File Expand all files Collapse all files

30
io.go
View File

@@ -129,6 +129,20 @@ var (
OnAuthPub func(p *util.Promise[IPublisher]) error
)
func (io *IO) auth(key string, secret string, expire string) bool {
if unixTime, err := strconv.ParseInt(expire, 16, 64); err != nil || time.Now().Unix() > unixTime {
return false
}
trueSecret := md5.Sum([]byte(key + io.Stream.Path + expire))
for i := 0; i < 16; i++ {
hex, err := strconv.ParseInt(secret[i<<1:(i<<1)+2], 16, 16)
if trueSecret[i] != byte(hex) || err != nil {
return false
}
}
return true
}
// receive 用于接收发布或者订阅
func (io *IO) receive(streamPath string, specific IIO) error {
streamPath = strings.Trim(streamPath, "/")
@@ -206,13 +220,7 @@ func (io *IO) receive(streamPath string, specific IIO) error {
return err
}
} else if conf.Key != "" {
secret := io.Args.Get(conf.SecretArgName)
t := io.Args.Get(conf.ExpireArgName)
if unixTime, err := strconv.ParseInt(t, 16, 64); err != nil || time.Now().Unix() > unixTime {
return ErrAuth
}
trueSecret := md5.Sum([]byte(conf.Key + s.StreamName + t))
if string(trueSecret[:]) != secret {
if !io.auth(conf.Key, io.Args.Get(conf.SecretArgName), io.Args.Get(conf.ExpireArgName)) {
return ErrAuth
}
}
@@ -247,13 +255,7 @@ func (io *IO) receive(streamPath string, specific IIO) error {
return err
}
} else if conf := specific.(ISubscriber).GetSubscriber().Config; conf.Key != "" {
secret := io.Args.Get(conf.SecretArgName)
t := io.Args.Get(conf.ExpireArgName)
if unixTime, err := strconv.ParseInt(t, 16, 64); err != nil || time.Now().Unix() > unixTime {
return ErrAuth
}
trueSecret := md5.Sum([]byte(conf.Key + s.StreamName + t))
if string(trueSecret[:]) != secret {
if !io.auth(conf.Key, io.Args.Get(conf.SecretArgName), io.Args.Get(conf.ExpireArgName)) {
return ErrAuth
}
}