mirror of
https://github.com/Monibuca/plugin-gb28181.git
synced 2025-12-24 13:27:57 +08:00
410 lines
11 KiB
Go
410 lines
11 KiB
Go
package gb28181
|
||
|
||
import (
|
||
"errors"
|
||
"fmt"
|
||
"math/rand"
|
||
"net/http"
|
||
"strconv"
|
||
"strings"
|
||
"sync"
|
||
"time"
|
||
|
||
. "m7s.live/engine/v4"
|
||
"github.com/ghettovoice/gosip/sip"
|
||
"go.uber.org/zap"
|
||
"m7s.live/plugin/gb28181/v4/utils"
|
||
)
|
||
|
||
type ChannelEx struct {
|
||
device *Device
|
||
RecordPublisher *GBPublisher `json:"-"`
|
||
LivePublisher *GBPublisher
|
||
LiveSubSP string //实时子码流
|
||
Records []*Record
|
||
RecordStartTime string
|
||
RecordEndTime string
|
||
recordStartTime time.Time
|
||
recordEndTime time.Time
|
||
liveInviteLock sync.Mutex
|
||
tcpPortIndex uint16
|
||
GpsTime time.Time //gps时间
|
||
Longitude string //经度
|
||
Latitude string //纬度
|
||
}
|
||
|
||
// Channel 通道
|
||
type Channel struct {
|
||
DeviceID string
|
||
ParentID string
|
||
Name string
|
||
Manufacturer string
|
||
Model string
|
||
Owner string
|
||
CivilCode string
|
||
Address string
|
||
Parental int
|
||
SafetyWay int
|
||
RegisterWay int
|
||
Secrecy int
|
||
Status string
|
||
Children []*Channel `json:"-"`
|
||
*ChannelEx //自定义属性
|
||
}
|
||
|
||
func (c *Channel) Copy(v *Channel) {
|
||
if v == nil {
|
||
return
|
||
}
|
||
|
||
c.DeviceID = v.DeviceID
|
||
c.ParentID = v.ParentID
|
||
c.Name = v.Name
|
||
c.Manufacturer = v.Manufacturer
|
||
c.Model = v.Model
|
||
c.Owner = v.Owner
|
||
c.CivilCode = v.CivilCode
|
||
c.Address = v.Address
|
||
c.Parental = v.Parental
|
||
c.SafetyWay = v.SafetyWay
|
||
c.RegisterWay = v.RegisterWay
|
||
c.Secrecy = v.Secrecy
|
||
c.Status = v.Status
|
||
c.Status = v.Status
|
||
c.ChannelEx = v.ChannelEx
|
||
}
|
||
|
||
func (c *Channel) CreateRequst(Method sip.RequestMethod) (req sip.Request) {
|
||
d := c.device
|
||
d.sn++
|
||
|
||
callId := sip.CallID(utils.RandNumString(10))
|
||
userAgent := sip.UserAgentHeader("Monibuca")
|
||
cseq := sip.CSeq{
|
||
SeqNo: uint32(d.sn),
|
||
MethodName: Method,
|
||
}
|
||
port := sip.Port(conf.SipPort)
|
||
serverAddr := sip.Address{
|
||
//DisplayName: sip.String{Str: d.serverConfig.Serial},
|
||
Uri: &sip.SipUri{
|
||
FUser: sip.String{Str: conf.Serial},
|
||
FHost: d.sipIP,
|
||
FPort: &port,
|
||
},
|
||
Params: sip.NewParams().Add("tag", sip.String{Str: utils.RandNumString(9)}),
|
||
}
|
||
channelAddr := sip.Address{
|
||
//DisplayName: sip.String{Str: d.serverConfig.Serial},
|
||
Uri: &sip.SipUri{FUser: sip.String{Str: c.DeviceID}, FHost: conf.Realm},
|
||
}
|
||
req = sip.NewRequest(
|
||
"",
|
||
Method,
|
||
channelAddr.Uri,
|
||
"SIP/2.0",
|
||
[]sip.Header{
|
||
serverAddr.AsFromHeader(),
|
||
channelAddr.AsToHeader(),
|
||
&callId,
|
||
&userAgent,
|
||
&cseq,
|
||
serverAddr.AsContactHeader(),
|
||
},
|
||
"",
|
||
nil,
|
||
)
|
||
|
||
req.SetTransport(conf.SipNetwork)
|
||
req.SetDestination(d.NetAddr)
|
||
return req
|
||
}
|
||
func (channel *Channel) QueryRecord(startTime, endTime string) int {
|
||
d := channel.device
|
||
channel.RecordStartTime = startTime
|
||
channel.RecordEndTime = endTime
|
||
channel.recordStartTime, _ = time.Parse(TIME_LAYOUT, startTime)
|
||
channel.recordEndTime, _ = time.Parse(TIME_LAYOUT, endTime)
|
||
channel.Records = nil
|
||
request := d.CreateRequest(sip.MESSAGE)
|
||
contentType := sip.ContentType("Application/MANSCDP+xml")
|
||
request.AppendHeader(&contentType)
|
||
body := fmt.Sprintf(`<?xml version="1.0"?>
|
||
<Query>
|
||
<CmdType>RecordInfo</CmdType>
|
||
<SN>%d</SN>
|
||
<DeviceID>%s</DeviceID>
|
||
<StartTime>%s</StartTime>
|
||
<EndTime>%s</EndTime>
|
||
<Secrecy>0</Secrecy>
|
||
<Type>all</Type>
|
||
</Query>`, d.sn, channel.DeviceID, startTime, endTime)
|
||
request.SetBody(body, true)
|
||
resp, err := d.SipRequestForResponse(request)
|
||
if err != nil {
|
||
return http.StatusRequestTimeout
|
||
}
|
||
return int(resp.StatusCode())
|
||
}
|
||
func (channel *Channel) Control(PTZCmd string) int {
|
||
d := channel.device
|
||
request := d.CreateRequest(sip.MESSAGE)
|
||
contentType := sip.ContentType("Application/MANSCDP+xml")
|
||
request.AppendHeader(&contentType)
|
||
body := fmt.Sprintf(`<?xml version="1.0"?>
|
||
<Control>
|
||
<CmdType>DeviceControl</CmdType>
|
||
<SN>%d</SN>
|
||
<DeviceID>%s</DeviceID>
|
||
<PTZCmd>%s</PTZCmd>
|
||
</Control>`, d.sn, channel.DeviceID, PTZCmd)
|
||
request.SetBody(body, true)
|
||
resp, err := d.SipRequestForResponse(request)
|
||
if err != nil {
|
||
return http.StatusRequestTimeout
|
||
}
|
||
return int(resp.StatusCode())
|
||
}
|
||
|
||
type InviteOptions struct {
|
||
Start int
|
||
End int
|
||
dump string
|
||
ssrc string
|
||
SSRC uint32
|
||
MediaPort uint16
|
||
}
|
||
|
||
func (o InviteOptions) IsLive() bool {
|
||
return o.Start == 0 || o.End == 0
|
||
}
|
||
|
||
func (o InviteOptions) Record() bool {
|
||
return !o.IsLive()
|
||
}
|
||
|
||
func (o *InviteOptions) Validate(start, end string) error {
|
||
if start != "" {
|
||
sint, err1 := strconv.ParseInt(start, 10, 0)
|
||
if err1 != nil {
|
||
return err1
|
||
}
|
||
o.Start = int(sint)
|
||
}
|
||
if end != "" {
|
||
eint, err2 := strconv.ParseInt(end, 10, 0)
|
||
if err2 != nil {
|
||
return err2
|
||
}
|
||
o.End = int(eint)
|
||
}
|
||
if o.Start >= o.End {
|
||
return errors.New("start < end")
|
||
}
|
||
return nil
|
||
}
|
||
|
||
func (o InviteOptions) String() string {
|
||
return fmt.Sprintf("t=%d %d", o.Start, o.End)
|
||
}
|
||
|
||
func (o *InviteOptions) CreateSSRC() {
|
||
ssrc := make([]byte, 10)
|
||
if o.IsLive() {
|
||
ssrc[0] = '0'
|
||
} else {
|
||
ssrc[0] = '1'
|
||
}
|
||
copy(ssrc[1:6], conf.Serial[3:8])
|
||
randNum := 1000 + rand.Intn(8999)
|
||
copy(ssrc[6:], strconv.Itoa(randNum))
|
||
o.ssrc = string(ssrc)
|
||
_ssrc, _ := strconv.ParseInt(o.ssrc, 10, 0)
|
||
o.SSRC = uint32(_ssrc)
|
||
}
|
||
|
||
/*
|
||
f字段： f = v/编码格式/分辨率/帧率/码率类型/码率大小a/编码格式/码率大小/采样率
|
||
各项具体含义：
|
||
v：后续参数为视频的参数；各参数间以 “/”分割；
|
||
编码格式：十进制整数字符串表示
|
||
1 –MPEG-4 2 –H.264 3 – SVAC 4 –3GP
|
||
分辨率：十进制整数字符串表示
|
||
1 – QCIF 2 – CIF 3 – 4CIF 4 – D1 5 –720P 6 –1080P/I
|
||
帧率：十进制整数字符串表示 0～99
|
||
码率类型：十进制整数字符串表示
|
||
1 – 固定码率（CBR） 2 – 可变码率（VBR）
|
||
码率大小：十进制整数字符串表示 0～100000（如 1表示1kbps）
|
||
a：后续参数为音频的参数；各参数间以 “/”分割；
|
||
编码格式：十进制整数字符串表示
|
||
1 – G.711 2 – G.723.1 3 – G.729 4 – G.722.1
|
||
码率大小：十进制整数字符串
|
||
音频编码码率： 1 — 5.3 kbps （注：G.723.1中使用）
|
||
2 — 6.3 kbps （注：G.723.1中使用）
|
||
3 — 8 kbps （注：G.729中使用）
|
||
4 — 16 kbps （注：G.722.1中使用）
|
||
5 — 24 kbps （注：G.722.1中使用）
|
||
6 — 32 kbps （注：G.722.1中使用）
|
||
7 — 48 kbps （注：G.722.1中使用）
|
||
8 — 64 kbps（注：G.711中使用）
|
||
采样率：十进制整数字符串表示
|
||
1 — 8 kHz（注：G.711/ G.723.1/ G.729中使用）
|
||
2—14 kHz（注：G.722.1中使用）
|
||
3—16 kHz（注：G.722.1中使用）
|
||
4—32 kHz（注：G.722.1中使用）
|
||
注1：字符串说明
|
||
本节中使用的“十进制整数字符串”的含义为“0”～“4294967296” 之间的十进制数字字符串。
|
||
注2：参数分割标识
|
||
各参数间以“/”分割，参数间的分割符“/”不能省略；
|
||
若两个分割符 “/”间的某参数为空时（即两个分割符 “/”直接将相连时）表示无该参数值；
|
||
注3：f字段说明
|
||
使用f字段时，应保证视频和音频参数的结构完整性，即在任何时候，f字段的结构都应是完整的结构：
|
||
f = v/编码格式/分辨率/帧率/码率类型/码率大小a/编码格式/码率大小/采样率
|
||
若只有视频时，音频中的各参数项可以不填写，但应保持 “a///”的结构:
|
||
f = v/编码格式/分辨率/帧率/码率类型/码率大小a///
|
||
若只有音频时也类似处理，视频中的各参数项可以不填写，但应保持 “v/”的结构：
|
||
f = v/a/编码格式/码率大小/采样率
|
||
f字段中视、音频参数段之间不需空格分割。
|
||
可使用f字段中的分辨率参数标识同一设备不同分辨率的码流。
|
||
*/
|
||
func (channel *Channel) Invite(opt InviteOptions) (code int, err error) {
|
||
if opt.IsLive() {
|
||
if !channel.liveInviteLock.TryLock() {
|
||
return 304, nil
|
||
}
|
||
defer func() {
|
||
if code != 200 {
|
||
channel.liveInviteLock.Unlock()
|
||
}
|
||
}()
|
||
}
|
||
channel.Bye(opt.IsLive())
|
||
d := channel.device
|
||
streamPath := fmt.Sprintf("%s/%s", d.ID, channel.DeviceID)
|
||
s := "Play"
|
||
opt.CreateSSRC()
|
||
if opt.Record() {
|
||
s = "Playback"
|
||
streamPath = fmt.Sprintf("%s/%s/%d-%d", d.ID, channel.DeviceID, opt.Start, opt.End)
|
||
}
|
||
if opt.dump == "" {
|
||
opt.dump = conf.DumpPath
|
||
}
|
||
// size := 1
|
||
// fps := 15
|
||
// bitrate := 200
|
||
// fmt.Sprintf("f=v/2/%d/%d/1/%da///", size, fps, bitrate)
|
||
publisher := &GBPublisher{
|
||
InviteOptions: opt,
|
||
channel: channel,
|
||
}
|
||
protocol := ""
|
||
if conf.IsMediaNetworkTCP() {
|
||
protocol = "TCP/"
|
||
if conf.tcpPorts.Valid {
|
||
opt.MediaPort, err = publisher.ListenTCP()
|
||
if err != nil {
|
||
return 500, err
|
||
}
|
||
} else if opt.MediaPort == 0 {
|
||
opt.MediaPort = conf.MediaPort
|
||
}
|
||
} else {
|
||
if conf.udpPorts.Valid {
|
||
opt.MediaPort, err = publisher.ListenUDP()
|
||
if err != nil {
|
||
code = 500
|
||
return
|
||
}
|
||
} else if opt.MediaPort == 0 {
|
||
opt.MediaPort = conf.MediaPort
|
||
}
|
||
}
|
||
// if opt.MediaPort == 0 {
|
||
// opt.MediaPort = conf.MediaPort
|
||
// if conf.IsMediaNetworkTCP() {
|
||
// protocol = "TCP/"
|
||
// opt.MediaPort = conf.MediaPort + channel.tcpPortIndex
|
||
// if channel.tcpPortIndex++; channel.tcpPortIndex >= conf.MediaPortMax {
|
||
// channel.tcpPortIndex = 0
|
||
// }
|
||
// }
|
||
// }
|
||
sdpInfo := []string{
|
||
"v=0",
|
||
fmt.Sprintf("o=%s 0 0 IN IP4 %s", channel.DeviceID, d.mediaIP),
|
||
"s=" + s,
|
||
"u=" + channel.DeviceID + ":0",
|
||
"c=IN IP4 " + d.mediaIP,
|
||
opt.String(),
|
||
fmt.Sprintf("m=video %d %sRTP/AVP 96", opt.MediaPort, protocol),
|
||
"a=recvonly",
|
||
"a=rtpmap:96 PS/90000",
|
||
"y=" + opt.ssrc,
|
||
"",
|
||
}
|
||
if conf.IsMediaNetworkTCP() {
|
||
sdpInfo = append(sdpInfo, "a=setup:passive", "a=connection:new")
|
||
}
|
||
invite := channel.CreateRequst(sip.INVITE)
|
||
contentType := sip.ContentType("application/sdp")
|
||
invite.AppendHeader(&contentType)
|
||
|
||
invite.SetBody(strings.Join(sdpInfo, "\r\n"), true)
|
||
|
||
subject := sip.GenericHeader{
|
||
HeaderName: "Subject", Contents: fmt.Sprintf("%s:%s,%s:0", channel.DeviceID, opt.ssrc, conf.Serial),
|
||
}
|
||
invite.AppendHeader(&subject)
|
||
publisher.inviteRes, err = d.SipRequestForResponse(invite)
|
||
if err != nil {
|
||
return http.StatusRequestTimeout, err
|
||
}
|
||
code = int(publisher.inviteRes.StatusCode())
|
||
plugin.Info(fmt.Sprintf("Channel :%s invite response status code: %d", channel.DeviceID, code))
|
||
if code == 200 {
|
||
ds := strings.Split(publisher.inviteRes.Body(), "\r\n")
|
||
for _, l := range ds {
|
||
if ls := strings.Split(l, "="); len(ls) > 1 {
|
||
if ls[0] == "y" && len(ls[1]) > 0 {
|
||
if _ssrc, err := strconv.ParseInt(ls[1], 10, 0); err == nil {
|
||
opt.SSRC = uint32(_ssrc)
|
||
} else {
|
||
plugin.Error("read invite response y ", zap.Error(err))
|
||
}
|
||
break
|
||
}
|
||
}
|
||
}
|
||
if conf.UdpCacheSize > 0 && !conf.IsMediaNetworkTCP() {
|
||
publisher.udpCache = utils.NewPqRtp()
|
||
}
|
||
if err = plugin.Publish(streamPath, publisher); err != nil {
|
||
code = 403
|
||
return
|
||
}
|
||
ack := sip.NewAckRequest("", invite, publisher.inviteRes, "", nil)
|
||
srv.Send(ack)
|
||
} else if opt.IsLive() && conf.AutoInvite {
|
||
time.AfterFunc(time.Second*5, func() {
|
||
channel.Invite(InviteOptions{})
|
||
})
|
||
}
|
||
return
|
||
}
|
||
|
||
func (channel *Channel) Bye(live bool) int {
|
||
d := channel.device
|
||
streamPath := fmt.Sprintf("%s/%s", d.ID, channel.DeviceID)
|
||
if s := Streams.Get(streamPath); s != nil {
|
||
s.Close()
|
||
}
|
||
if live && channel.LivePublisher != nil {
|
||
return channel.LivePublisher.Bye()
|
||
}
|
||
if !live && channel.RecordPublisher != nil {
|
||
return channel.RecordPublisher.Bye()
|
||
}
|
||
return 404
|
||
}
|