A signaling server for GB28181

2024-04-17 14:31:33 +08:00
parent 8774b234b4
commit 0b7126b12b
50 changed files with 11136 additions and 1 deletions
@@ -0,0 +1,153 @@
+package service
+
+import (
+	"bytes"
+	"encoding/xml"
+	"net/http"
+	"strconv"
+
+	"github.com/emiago/sipgo/sip"
+	"github.com/ossrs/go-oryx-lib/logger"
+	"github.com/ossrs/srs-sip/pkg/service/stack"
+	"golang.org/x/net/html/charset"
+)
+
+const GB28181_ID_LENGTH = 20
+
+type VideoChannelStatus struct {
+	ID        string
+	ParentID  string
+	MediaHost string
+	MediaPort int
+	Ssrc      string
+	Status    string
+}
+
+func (s *UAS) onRegister(req *sip.Request, tx sip.ServerTransaction) {
+	id := req.From().Address.User
+	if len(id) != GB28181_ID_LENGTH {
+		logger.E(s.ctx, "invalid device ID")
+		return
+	}
+
+	isUnregister := false
+	if exps := req.GetHeaders("Expires"); len(exps) > 0 {
+		exp := exps[0]
+		expSec, err := strconv.ParseInt(exp.Value(), 10, 32)
+		if err != nil {
+			logger.Ef(s.ctx, "parse expires header error: %s", err.Error())
+			return
+		}
+		if expSec == 0 {
+			isUnregister = true
+		}
+	} else {
+		logger.E(s.ctx, "empty expires header")
+		return
+	}
+
+	if isUnregister {
+		DM.RemoveDevice(id)
+		logger.Wf(s.ctx, "Device %s unregistered", id)
+		return
+	} else {
+		if d, ok := DM.GetDevice(id); !ok {
+			DM.AddDevice(id, &DeviceInfo{
+				DeviceID:    id,
+				SourceAddr:  req.Source(),
+				NetworkType: req.Transport(),
+			})
+			s.respondRegister(req, http.StatusOK, "OK", tx)
+			logger.Tf(s.ctx, "%s Register success, source:%s, req: %s", id, req.Source(), req.String())
+
+			go s.Catalog(id)
+		} else {
+			if d.SourceAddr != req.Source() {
+				logger.Ef(s.ctx, "Device %s[%s] already registered, %s is NOT allowed.", id, d.SourceAddr, req.Source())
+				// TODO: 国标没有明确定义重复ID注册的处理方式，这里暂时返回冲突
+				s.respondRegister(req, http.StatusConflict, "Conflict Device ID", tx)
+			} else {
+				// TODO: 刷新DM里面的设备信息
+				s.respondRegister(req, http.StatusOK, "OK", tx)
+			}
+		}
+	}
+}
+
+func (s *UAS) respondRegister(req *sip.Request, code sip.StatusCode, reason string, tx sip.ServerTransaction) {
+	res := stack.NewRegisterResponse(req, code, reason)
+	_ = tx.Respond(res)
+
+}
+
+func (s *UAS) onMessage(req *sip.Request, tx sip.ServerTransaction) {
+	id := req.From().Address.User
+	if len(id) != 20 {
+		logger.Ef(s.ctx, "invalid device ID %s", req.String())
+	}
+
+	//logger.Tf(s.ctx, "Received MESSAGE: %s", req.String())
+
+	temp := &struct {
+		XMLName      xml.Name
+		CmdType      string
+		SN           int // 请求序列号，一般用于对应 request 和 response
+		DeviceID     string
+		DeviceName   string
+		Manufacturer string
+		Model        string
+		Channel      string
+		DeviceList   []ChannelInfo `xml:"DeviceList>Item"`
+		// RecordList   []*Record     `xml:"RecordList>Item"`
+		// SumNum       int
+	}{}
+	decoder := xml.NewDecoder(bytes.NewReader([]byte(req.Body())))
+	decoder.CharsetReader = charset.NewReaderLabel
+	if err := decoder.Decode(temp); err != nil {
+		logger.Ef(s.ctx, "decode message error: %s\n message:%s", err.Error(), req.Body())
+	}
+	var body string
+	switch temp.CmdType {
+	case "Keepalive":
+		logger.T(s.ctx, "Keepalive")
+		if _, ok := DM.GetDevice(temp.DeviceID); !ok {
+			// unregister device
+			tx.Respond(sip.NewResponseFromRequest(req, http.StatusBadRequest, "", nil))
+			return
+		}
+	case "SensorCatalog": // 兼容宇视，非国标
+	case "Catalog":
+		logger.T(s.ctx, "Catalog")
+		DM.UpdateChannels(temp.DeviceID, temp.DeviceList...)
+		//go s.AutoInvite(temp.DeviceID, temp.DeviceList...)
+	case "Alarm":
+		logger.T(s.ctx, "Alarm")
+	default:
+		logger.Wf(s.ctx, "Not supported CmdType: %s", temp.CmdType)
+		response := sip.NewResponseFromRequest(req, http.StatusBadRequest, "", nil)
+		tx.Respond(response)
+		return
+	}
+	tx.Respond(sip.NewResponseFromRequest(req, http.StatusOK, "OK", []byte(body)))
+}
+
+func (s *UAS) onNotify(req *sip.Request, tx sip.ServerTransaction) {
+	logger.T(s.ctx, "Received NOTIFY request")
+	tx.Respond(sip.NewResponseFromRequest(req, http.StatusOK, "OK", nil))
+}
+
+func (s *UAS) AddVideoChannelStatue(channelID string, status VideoChannelStatus) {
+	s.channelsStatue.Store(channelID, status)
+}
+
+func (s *UAS) GetVideoChannelStatue(channelID string) (VideoChannelStatus, bool) {
+	v, ok := s.channelsStatue.Load(channelID)
+	if !ok {
+		return VideoChannelStatus{}, false
+	}
+	return v.(VideoChannelStatus), true
+}
+
+func (s *UAS) RemoveVideoChannelStatue(channelID string) {
+	s.channelsStatue.Delete(channelID)
+}