|
|
// Copyright 2021, Chef. All rights reserved.
|
|
|
// https://github.com/q191201771/lal
|
|
|
//
|
|
|
// Use of this source code is governed by a MIT-style license
|
|
|
// that can be found in the License file.
|
|
|
//
|
|
|
// Author: Chef (191201771@qq.com)
|
|
|
|
|
|
package sdp
|
|
|
|
|
|
import (
|
|
|
"encoding/base64"
|
|
|
"encoding/hex"
|
|
|
"fmt"
|
|
|
"strings"
|
|
|
|
|
|
"github.com/q191201771/naza/pkg/nazaerrors"
|
|
|
|
|
|
"github.com/q191201771/lal/pkg/aac"
|
|
|
|
|
|
"github.com/q191201771/lal/pkg/base"
|
|
|
)
|
|
|
|
|
|
func Pack(vps, sps, pps, asc []byte) (ctx LogicContext, err error) {
|
|
|
// 判断音频、视频是否存在，以及视频是H264还是H265
|
|
|
var hasAudio, hasVideo, isHevc bool
|
|
|
if sps != nil && pps != nil {
|
|
|
hasVideo = true
|
|
|
if vps != nil {
|
|
|
isHevc = true
|
|
|
}
|
|
|
}
|
|
|
if asc != nil {
|
|
|
hasAudio = true
|
|
|
}
|
|
|
|
|
|
if !hasAudio && !hasVideo {
|
|
|
err = nazaerrors.Wrap(base.ErrSdp)
|
|
|
return
|
|
|
}
|
|
|
|
|
|
// 判断AAC的采样率
|
|
|
var samplingFrequency int
|
|
|
if asc != nil {
|
|
|
var ascCtx *aac.AscContext
|
|
|
ascCtx, err = aac.NewAscContext(asc)
|
|
|
if err != nil {
|
|
|
return
|
|
|
}
|
|
|
samplingFrequency, err = ascCtx.GetSamplingFrequency()
|
|
|
if err != nil {
|
|
|
return
|
|
|
}
|
|
|
}
|
|
|
|
|
|
sdpStr := fmt.Sprintf(`v=0
|
|
|
o=- 0 0 IN IP4 127.0.0.1
|
|
|
s=No Name
|
|
|
c=IN IP4 127.0.0.1
|
|
|
t=0 0
|
|
|
a=tool:%s
|
|
|
`, base.LalPackSdp)
|
|
|
|
|
|
streamid := 0
|
|
|
|
|
|
if hasVideo {
|
|
|
if isHevc {
|
|
|
tmpl := `m=video 0 RTP/AVP 98
|
|
|
a=rtpmap:98 H265/90000
|
|
|
a=fmtp:98 profile-id=1;sprop-sps=%s;sprop-pps=%s;sprop-vps=%s
|
|
|
a=control:streamid=%d
|
|
|
`
|
|
|
sdpStr += fmt.Sprintf(tmpl, base64.StdEncoding.EncodeToString(sps), base64.StdEncoding.EncodeToString(pps), base64.StdEncoding.EncodeToString(vps), streamid)
|
|
|
} else {
|
|
|
tmpl := `m=video 0 RTP/AVP 96
|
|
|
a=rtpmap:96 H264/90000
|
|
|
a=fmtp:96 packetization-mode=1; sprop-parameter-sets=%s,%s; profile-level-id=640016
|
|
|
a=control:streamid=%d
|
|
|
`
|
|
|
sdpStr += fmt.Sprintf(tmpl, base64.StdEncoding.EncodeToString(sps), base64.StdEncoding.EncodeToString(pps), streamid)
|
|
|
}
|
|
|
|
|
|
streamid++
|
|
|
}
|
|
|
|
|
|
if hasAudio {
|
|
|
tmpl := `m=audio 0 RTP/AVP 97
|
|
|
b=AS:128
|
|
|
a=rtpmap:97 MPEG4-GENERIC/%d/2
|
|
|
a=fmtp:97 profile-level-id=1;mode=AAC-hbr;sizelength=13;indexlength=3;indexdeltalength=3; config=%s
|
|
|
a=control:streamid=%d
|
|
|
`
|
|
|
sdpStr += fmt.Sprintf(tmpl, samplingFrequency, hex.EncodeToString(asc), streamid)
|
|
|
}
|
|
|
|
|
|
raw := []byte(strings.ReplaceAll(sdpStr, "\n", "\r\n"))
|
|
|
ctx, err = ParseSdp2LogicContext(raw)
|
|
|
return
|
|
|
}
|