mirror of
https://github.com/Monibuca/plugin-gb28181.git
synced 2026-04-23 00:17:13 +08:00
langhuihui
297 lines
8.9 KiB
Go
297 lines
8.9 KiB
Go
package gb28181
|
|
|
|
import (
|
|
"bytes"
|
|
"crypto/md5"
|
|
"encoding/xml"
|
|
"fmt"
|
|
|
|
"github.com/logrusorgru/aurora"
|
|
"go.uber.org/zap"
|
|
"m7s.live/plugin/gb28181/v4/utils"
|
|
|
|
"github.com/ghettovoice/gosip/sip"
|
|
|
|
"net/http"
|
|
"time"
|
|
|
|
"golang.org/x/net/html/charset"
|
|
)
|
|
|
|
type Authorization struct {
|
|
*sip.Authorization
|
|
}
|
|
|
|
func (a *Authorization) Verify(username, passwd, realm, nonce string) bool {
|
|
|
|
//1、将 username,realm,password 依次组合获取 1 个字符串，并用算法加密的到密文 r1
|
|
s1 := fmt.Sprintf("%s:%s:%s", username, realm, passwd)
|
|
r1 := a.getDigest(s1)
|
|
//2、将 method，即REGISTER ,uri 依次组合获取 1 个字符串，并对这个字符串使用算法 加密得到密文 r2
|
|
s2 := fmt.Sprintf("REGISTER:%s", a.Uri())
|
|
r2 := a.getDigest(s2)
|
|
|
|
if r1 == "" || r2 == "" {
|
|
plugin.Error("Authorization algorithm wrong")
|
|
return false
|
|
}
|
|
//3、将密文 1，nonce 和密文 2 依次组合获取 1 个字符串，并对这个字符串使用算法加密，获得密文 r3，即Response
|
|
s3 := fmt.Sprintf("%s:%s:%s", r1, nonce, r2)
|
|
r3 := a.getDigest(s3)
|
|
|
|
//4、计算服务端和客户端上报的是否相等
|
|
return r3 == a.Response()
|
|
}
|
|
|
|
func (a *Authorization) getDigest(raw string) string {
|
|
switch a.Algorithm() {
|
|
case "MD5":
|
|
return fmt.Sprintf("%x", md5.Sum([]byte(raw)))
|
|
default: //如果没有算法，默认使用MD5
|
|
return fmt.Sprintf("%x", md5.Sum([]byte(raw)))
|
|
}
|
|
}
|
|
|
|
func (c *GB28181Config) OnRegister(req sip.Request, tx sip.ServerTransaction) {
|
|
from, _ := req.From()
|
|
|
|
id := from.Address.User().String()
|
|
plugin.Sugar().Infof("OnRegister: %s, %s, from: %s", req.Destination(), id, req.Source())
|
|
if len(id) != 20 {
|
|
plugin.Sugar().Infof("Wrong GB-28181 id: %s", id)
|
|
return
|
|
}
|
|
passAuth := false
|
|
// 不需要密码情况
|
|
if c.Username == "" && c.Password == "" {
|
|
passAuth = true
|
|
} else {
|
|
// 需要密码情况 设备第一次上报，返回401和加密算法
|
|
if hdrs := req.GetHeaders("Authorization"); len(hdrs) > 0 {
|
|
authenticateHeader := hdrs[0].(*sip.GenericHeader)
|
|
auth := &Authorization{sip.AuthFromValue(authenticateHeader.Contents)}
|
|
|
|
// 有些摄像头没有配置用户名的地方，用户名就是摄像头自己的国标id
|
|
var username string
|
|
if auth.Username() == id {
|
|
username = id
|
|
} else {
|
|
username = c.Username
|
|
}
|
|
|
|
if dc, ok := DeviceRegisterCount.LoadOrStore(id, 1); ok && dc.(int) > MaxRegisterCount {
|
|
response := sip.NewResponseFromRequest("", req, http.StatusForbidden, "Forbidden", "")
|
|
tx.Respond(response)
|
|
return
|
|
} else {
|
|
// 设备第二次上报，校验
|
|
_nonce, loaded := DeviceNonce.Load(id)
|
|
if loaded && auth.Verify(username, c.Password, c.Realm, _nonce.(string)) {
|
|
passAuth = true
|
|
} else {
|
|
DeviceRegisterCount.Store(id, dc.(int)+1)
|
|
}
|
|
}
|
|
}
|
|
}
|
|
if passAuth {
|
|
var d *Device
|
|
if v, ok := Devices.Load(id); ok {
|
|
d = v.(*Device)
|
|
c.RecoverDevice(d, req)
|
|
} else {
|
|
d = c.StoreDevice(id, req)
|
|
}
|
|
DeviceNonce.Delete(id)
|
|
DeviceRegisterCount.Delete(id)
|
|
resp := sip.NewResponseFromRequest("", req, http.StatusOK, "OK", "")
|
|
to, _ := resp.To()
|
|
resp.ReplaceHeaders("To", []sip.Header{&sip.ToHeader{Address: to.Address, Params: sip.NewParams().Add("tag", sip.String{Str: utils.RandNumString(9)})}})
|
|
resp.RemoveHeader("Allow")
|
|
expires := sip.Expires(3600)
|
|
resp.AppendHeader(&expires)
|
|
resp.AppendHeader(&sip.GenericHeader{
|
|
HeaderName: "Date",
|
|
Contents: time.Now().Format(TIME_LAYOUT),
|
|
})
|
|
_ = tx.Respond(resp)
|
|
//订阅设备更新
|
|
go d.syncChannels()
|
|
} else {
|
|
response := sip.NewResponseFromRequest("", req, http.StatusUnauthorized, "Unauthorized", "")
|
|
_nonce, _ := DeviceNonce.LoadOrStore(id, utils.RandNumString(32))
|
|
auth := fmt.Sprintf(
|
|
`Digest realm="%s",algorithm=%s,nonce="%s"`,
|
|
c.Realm,
|
|
"MD5",
|
|
_nonce.(string),
|
|
)
|
|
response.AppendHeader(&sip.GenericHeader{
|
|
HeaderName: "WWW-Authenticate",
|
|
Contents: auth,
|
|
})
|
|
_ = tx.Respond(response)
|
|
}
|
|
}
|
|
|
|
// syncChannels
|
|
// 同步设备信息、下属通道信息，包括主动查询通道信息，订阅通道变化情况
|
|
func (d *Device) syncChannels() {
|
|
if time.Since(d.lastSyncTime) > 2*conf.HeartbeatInterval {
|
|
d.lastSyncTime = time.Now()
|
|
d.Catalog()
|
|
d.Subscribe()
|
|
d.QueryDeviceInfo()
|
|
}
|
|
}
|
|
|
|
func (c *GB28181Config) OnMessage(req sip.Request, tx sip.ServerTransaction) {
|
|
from, _ := req.From()
|
|
id := from.Address.User().String()
|
|
plugin.Sugar().Debugf("SIP<-OnMessage from %s : %s", req.Source(), req.String())
|
|
if v, ok := Devices.Load(id); ok {
|
|
d := v.(*Device)
|
|
switch d.Status {
|
|
case "RECOVER":
|
|
c.RecoverDevice(d, req)
|
|
go d.syncChannels()
|
|
//return
|
|
case string(sip.REGISTER):
|
|
d.Status = "ONLINE"
|
|
}
|
|
d.UpdateTime = time.Now()
|
|
temp := &struct {
|
|
XMLName xml.Name
|
|
CmdType string
|
|
DeviceID string
|
|
DeviceName string
|
|
Manufacturer string
|
|
Model string
|
|
Channel string
|
|
DeviceList []*Channel `xml:"DeviceList>Item"`
|
|
RecordList []*Record `xml:"RecordList>Item"`
|
|
}{}
|
|
decoder := xml.NewDecoder(bytes.NewReader([]byte(req.Body())))
|
|
decoder.CharsetReader = charset.NewReaderLabel
|
|
err := decoder.Decode(temp)
|
|
if err != nil {
|
|
err = utils.DecodeGbk(temp, []byte(req.Body()))
|
|
if err != nil {
|
|
plugin.Error("decode catelog err", zap.Error(err))
|
|
}
|
|
}
|
|
var body string
|
|
switch temp.CmdType {
|
|
case "Keepalive":
|
|
d.LastKeepaliveAt = time.Now()
|
|
//callID !="" 说明是订阅的事件类型信息
|
|
if d.lastSyncTime.IsZero() {
|
|
go d.syncChannels()
|
|
} else {
|
|
d.channelMap.Range(func(key, value interface{}) bool {
|
|
channel := value.(*Channel)
|
|
channel.TryAutoInvite(&InviteOptions{})
|
|
return true
|
|
})
|
|
}
|
|
//为什么要查找子码流?
|
|
//d.CheckSubStream()
|
|
//在KeepLive 进行位置订阅的处理，如果开启了自动订阅位置，则去订阅位置
|
|
if c.Position.AutosubPosition && time.Since(d.GpsTime) > c.Position.Interval*2 {
|
|
d.MobilePositionSubscribe(d.ID, c.Position.Expires, c.Position.Interval)
|
|
plugin.Sugar().Debugf("位置自动订阅，设备[%s]成功\n", d.ID)
|
|
}
|
|
case "Catalog":
|
|
d.UpdateChannels(temp.DeviceList)
|
|
case "RecordInfo":
|
|
d.UpdateRecord(temp.DeviceID, temp.RecordList)
|
|
case "DeviceInfo":
|
|
// 主设备信息
|
|
d.Name = temp.DeviceName
|
|
d.Manufacturer = temp.Manufacturer
|
|
d.Model = temp.Model
|
|
case "Alarm":
|
|
d.Status = "Alarmed"
|
|
body = BuildAlarmResponseXML(d.ID)
|
|
default:
|
|
plugin.Sugar().Warnf("DeviceID:", aurora.Red(d.ID), " Not supported CmdType : "+temp.CmdType+" body:\n", req.Body)
|
|
response := sip.NewResponseFromRequest("", req, http.StatusBadRequest, "", "")
|
|
tx.Respond(response)
|
|
return
|
|
}
|
|
|
|
tx.Respond(sip.NewResponseFromRequest("", req, http.StatusOK, "OK", body))
|
|
}
|
|
}
|
|
func (c *GB28181Config) OnBye(req sip.Request, tx sip.ServerTransaction) {
|
|
tx.Respond(sip.NewResponseFromRequest("", req, http.StatusOK, "OK", ""))
|
|
}
|
|
|
|
// OnNotify 订阅通知处理
|
|
func (c *GB28181Config) OnNotify(req sip.Request, tx sip.ServerTransaction) {
|
|
from, _ := req.From()
|
|
id := from.Address.User().String()
|
|
if v, ok := Devices.Load(id); ok {
|
|
d := v.(*Device)
|
|
d.UpdateTime = time.Now()
|
|
temp := &struct {
|
|
XMLName xml.Name
|
|
CmdType string
|
|
DeviceID string
|
|
Time string //位置订阅-GPS时间
|
|
Longitude string //位置订阅-经度
|
|
Latitude string //位置订阅-维度
|
|
// Speed string //位置订阅-速度(km/h)(可选)
|
|
// Direction string //位置订阅-方向(取值为当前摄像头方向与正北方的顺时针夹角,取值范围0°~360°,单位:°)(可选)
|
|
// Altitude string //位置订阅-海拔高度,单位:m(可选)
|
|
DeviceList []*notifyMessage `xml:"DeviceList>Item"` //目录订阅
|
|
}{}
|
|
decoder := xml.NewDecoder(bytes.NewReader([]byte(req.Body())))
|
|
decoder.CharsetReader = charset.NewReaderLabel
|
|
err := decoder.Decode(temp)
|
|
if err != nil {
|
|
err = utils.DecodeGbk(temp, []byte(req.Body()))
|
|
if err != nil {
|
|
plugin.Error("decode catelog err", zap.Error(err))
|
|
}
|
|
}
|
|
var body string
|
|
switch temp.CmdType {
|
|
case "Catalog":
|
|
//目录状态
|
|
d.UpdateChannelStatus(temp.DeviceList)
|
|
case "MobilePosition":
|
|
//更新channel的坐标
|
|
d.UpdateChannelPosition(temp.DeviceID, temp.Time, temp.Longitude, temp.Latitude)
|
|
// case "Alarm":
|
|
// //报警事件通知 TODO
|
|
default:
|
|
plugin.Sugar().Warnf("DeviceID:", aurora.Red(d.ID), " Not supported CmdType : "+temp.CmdType+" body:", req.Body)
|
|
response := sip.NewResponseFromRequest("", req, http.StatusBadRequest, "", "")
|
|
tx.Respond(response)
|
|
return
|
|
}
|
|
|
|
tx.Respond(sip.NewResponseFromRequest("", req, http.StatusOK, "OK", body))
|
|
}
|
|
}
|
|
|
|
type notifyMessage struct {
|
|
DeviceID string
|
|
ParentID string
|
|
Name string
|
|
Manufacturer string
|
|
Model string
|
|
Owner string
|
|
CivilCode string
|
|
Address string
|
|
Port int
|
|
Parental int
|
|
SafetyWay int
|
|
RegisterWay int
|
|
Secrecy int
|
|
Status string
|
|
//状态改变事件 ON:上线,OFF:离线,VLOST:视频丢失,DEFECT:故障,ADD:增加,DEL:删除,UPDATE:更新(必选)
|
|
Event string
|
|
}
|