Refactor and simplify package structure and interfaces.
Reorganize code by removing unused files, restructuring package organization, and updating import references to new paths. This simplifies handling of smart and protocol-related operations, improves maintainability, and eliminates redundancy.
This commit is contained in:
168
pkg/models/certificate.go
Normal file
168
pkg/models/certificate.go
Normal file
@ -0,0 +1,168 @@
|
||||
package models
|
||||
|
||||
type NetConnectionType uint
|
||||
|
||||
const (
|
||||
NetConnectionWired NetConnectionType = iota
|
||||
NetConnectionWireless
|
||||
)
|
||||
|
||||
type TimeShiftSupportFlag uint8
|
||||
|
||||
const (
|
||||
TimeShiftNotSupported TimeShiftSupportFlag = iota
|
||||
TimeShiftSupported
|
||||
)
|
||||
|
||||
type FileSystemVersionNumber uint8
|
||||
|
||||
const (
|
||||
FileSystemVersion4 FileSystemVersionNumber = iota
|
||||
FileSystemVersion5
|
||||
)
|
||||
|
||||
type CertificateConnectRequest struct {
|
||||
Net NetConnectionType `json:"NET"`
|
||||
SerialNumber string `json:"DSNO"`
|
||||
DeviceName string `json:"DEVNAME"`
|
||||
ChannelsNumber uint `json:"CHANNEL"`
|
||||
LicensePlate string `json:"CARNUM"`
|
||||
DeviceNumber string `json:"AUTONO"`
|
||||
VehicleNumber string `json:"AUTOCAR"`
|
||||
TimeShiftSupport TimeShiftSupportFlag `json:"TSE"`
|
||||
FileSystemVersion FileSystemVersionNumber `json:"FSV"`
|
||||
ICCID string `json:"ICCID"`
|
||||
EvidenceSupport string `json:"EV"`
|
||||
}
|
||||
|
||||
type CertificateConnectClientRequest struct {
|
||||
UK string `json:"UK"`
|
||||
}
|
||||
|
||||
type CertificateConnectResponse struct {
|
||||
ErrorCode uint `json:"ERRORCODE"`
|
||||
ErrorCause string `json:"ERRORCAUSE"`
|
||||
CommandMask uint `json:"MASKCMD"`
|
||||
}
|
||||
|
||||
type CertificateConnectClientResponse struct {
|
||||
S0 string `json:"S0"`
|
||||
}
|
||||
|
||||
type CertificateVerificationRequest struct {
|
||||
S0 string `json:"S0"`
|
||||
}
|
||||
|
||||
type CertificateVerificationResponse struct {
|
||||
ErrorCode uint `json:"ERRORCODE"`
|
||||
ErrorCause string `json:"ERRORCAUSE"`
|
||||
ReturnFlag bool `json:"RETURN"`
|
||||
}
|
||||
|
||||
type CertificateLoginRequest struct {
|
||||
ClientID uint `json:"CID,omitempty"`
|
||||
MAC string `json:"MAC,omitempty"`
|
||||
User string `json:"USER"`
|
||||
Password string `json:"PASSWD"`
|
||||
PlayDevID uint `json:"PLAYDEVID"`
|
||||
}
|
||||
|
||||
type CertificateLoginResponse struct {
|
||||
SerialNumber string `json:"DSNO"`
|
||||
DeviceName string `json:"DEVNAME"`
|
||||
ChannelsNumber uint `json:"CHANNEL"`
|
||||
UID string `json:"UID"`
|
||||
AlarmInputNumber uint `json:"ALARMIN"`
|
||||
AlarmOutputNumber uint `json:"ALARMOUT"`
|
||||
DeviceType string `json:"TYPE"`
|
||||
DeviceClass DeviceType `json:"DEVCLASS"`
|
||||
CurrentVersion string `json:"PRO"`
|
||||
LicensePlate string `json:"CARNUM"`
|
||||
UserLever UserAccessLevel `json:"LEVEL"`
|
||||
CompanyName string `json:"CPN"`
|
||||
CustomerName string `json:"CNAME"`
|
||||
AudioChannelsNumber uint `json:"ACHN"`
|
||||
}
|
||||
|
||||
type CertificateCreateStreamRequest struct {
|
||||
StreamName string `json:"STREAMNAME"`
|
||||
}
|
||||
|
||||
type CertificateCreateStreamResponse struct {
|
||||
ErrorCode uint `json:"ERRORCODE"`
|
||||
ErrorCause string `json:"ERRORCAUSE"`
|
||||
}
|
||||
|
||||
type CommandMaskParameters uint
|
||||
|
||||
const (
|
||||
CommandMaskAlarm = 1 << iota
|
||||
CommandMaskScanningGun
|
||||
CommandMaskPassengerFlow
|
||||
CommandMaskFaceContrast
|
||||
CommandMaskCard
|
||||
CommandMaskShutdownReport
|
||||
CommandMaskGPSReport
|
||||
CommandMaskAll = 0b1111111
|
||||
)
|
||||
|
||||
type DeviceType uint
|
||||
|
||||
const (
|
||||
DeviceTypeDVR DeviceType = 1 + iota
|
||||
DeviceTypeIPC
|
||||
DeviceTypeNVR
|
||||
DeviceTypeMIPC
|
||||
DeviceTypeMDVR
|
||||
)
|
||||
|
||||
type UserAccessLevel uint
|
||||
|
||||
const (
|
||||
UserAccessLevelSuperAdmin UserAccessLevel = iota
|
||||
UserAccessLevelAdministrator
|
||||
UserAccessLeverUser
|
||||
)
|
||||
|
||||
/*
|
||||
func (e *Package) RequestConnect(session string, serial string, numOfCams int) {
|
||||
e.Payload = map[string]any{
|
||||
"MODULE": "CERTIFICATE",
|
||||
"OPERATION": "CONNECT",
|
||||
"PARAMETER": map[string]any{
|
||||
"DSNO": serial,
|
||||
"CHANNEL": numOfCams,
|
||||
},
|
||||
"SESSION": session,
|
||||
}
|
||||
}
|
||||
|
||||
// video server util
|
||||
func (e *Package) ResponseConnect(Sid string, streamName string) {
|
||||
e.Payload = map[string]any{
|
||||
"MODULE": "CERTIFICATE",
|
||||
"OPERATION": "CREATESTREAM",
|
||||
"RESPONSE": map[string]any{
|
||||
"ERRORCODE": 0,
|
||||
"STREAMNAME": streamName,
|
||||
},
|
||||
"SESSION": Sid,
|
||||
}
|
||||
}
|
||||
|
||||
// main server util
|
||||
func (e *Package) ResponseCertificateConnect(Sid string) {
|
||||
e.Payload = map[string]any{
|
||||
"MODULE": "CERTIFICATE",
|
||||
"OPERATION": "CONNECT",
|
||||
"RESPONSE": map[string]any{
|
||||
"ERRORCAUSE": "",
|
||||
"ERRORCODE": 0,
|
||||
"MASKCMD": 5,
|
||||
"PRO": "1.0.5",
|
||||
},
|
||||
"SESSION": Sid,
|
||||
}
|
||||
}
|
||||
|
||||
*/
|
||||
83
pkg/models/configmodel.go
Normal file
83
pkg/models/configmodel.go
Normal file
@ -0,0 +1,83 @@
|
||||
package models
|
||||
|
||||
import (
|
||||
"gitea.unprism.ru/KRBL/n9m/v2/pkg/parameters"
|
||||
)
|
||||
|
||||
/*
|
||||
|
||||
// request reqistration parameters (directly to register)
|
||||
func (e *Package) RequestParameters(params map[string]any, serial int, session string) {
|
||||
e.Payload = map[string]any{
|
||||
"MODULE": "CONFIGMODEL",
|
||||
"OPERATION": "GET",
|
||||
"PARAMETER": map[string]any{
|
||||
"MDVR": params["MDVR"],
|
||||
"SERIAL": serial,
|
||||
},
|
||||
"SESSION": session,
|
||||
}
|
||||
} // end of 'RequestParameters' function
|
||||
|
||||
// set reigeter parameters (directly to register)
|
||||
func (e *Package) SetParameters(params map[string]any, serial int, session string) {
|
||||
e.Payload = map[string]any{
|
||||
"MODULE": "CONFIGMODEL",
|
||||
"OPERATION": "SET",
|
||||
"PARAMETER": map[string]any{
|
||||
"MDVR": params["MDVR"],
|
||||
"SERIAL": serial,
|
||||
},
|
||||
"SESSION": session,
|
||||
}
|
||||
// log.Println(e.Payload)
|
||||
} // end of 'SetParameters' function
|
||||
*/
|
||||
|
||||
type ConfigModelSetRequest struct {
|
||||
MDVR parameters.Setting `json:"MDVR"`
|
||||
}
|
||||
|
||||
type ConfigModelGetRequest struct {
|
||||
MDVR interface{} `json:"MDVR"`
|
||||
}
|
||||
|
||||
type ConfigModelSetResponse struct {
|
||||
MDVR parameters.Setting `json:"MDVR"`
|
||||
}
|
||||
|
||||
func InitialConfig() ConfigModelSetRequest {
|
||||
return ConfigModelSetRequest{
|
||||
MDVR: parameters.Setting{
|
||||
KEYS: parameters.KEYS{
|
||||
GV: 1, // GPS version
|
||||
},
|
||||
PGDSM: parameters.PGDSM{
|
||||
PGPS: parameters.PGPS{
|
||||
EN: 1, // Real-time position monitoring
|
||||
MODE: 0b10, // Enable timer
|
||||
TM: 10, // Time interval
|
||||
},
|
||||
},
|
||||
SUBSTRNET: parameters.SUBSTRNET{
|
||||
SM: 1,
|
||||
},
|
||||
},
|
||||
}
|
||||
}
|
||||
|
||||
/*
|
||||
func (e *Package) ResponseConfigModelSet(Sid string) {
|
||||
e.Payload = map[string]any{
|
||||
"MODULE": "CONFIGMODUL", // it's not error
|
||||
"OPERATION": "SET",
|
||||
"RESPONSE": map[string]any{
|
||||
"ERRORCODE": 0,
|
||||
"ERRORCAUSE": "None",
|
||||
"ERRORDESCRIPTION": "None",
|
||||
},
|
||||
"SESSION": Sid,
|
||||
}
|
||||
}
|
||||
|
||||
*/
|
||||
215
pkg/models/devemm.go
Normal file
215
pkg/models/devemm.go
Normal file
@ -0,0 +1,215 @@
|
||||
package models
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"gitea.unprism.ru/KRBL/n9m/v2/pkg/protocol"
|
||||
"strconv"
|
||||
"time"
|
||||
)
|
||||
|
||||
/*
|
||||
|
||||
// main server util
|
||||
func (e *Package) RequestGeolocation(serial int, Sid string) {
|
||||
e.Payload = map[string]any{
|
||||
"MODULE": "DEVEMM",
|
||||
"OPERATION": "GETPOS",
|
||||
"PARAMETER": map[string]any{
|
||||
"SERIAL": serial,
|
||||
},
|
||||
"SESSION": Sid,
|
||||
}
|
||||
}
|
||||
|
||||
func (e *Package) ResponseGeolocation(errorCode int, errorCause string, serial int, longitude float32, latitude float32, altitude float32, speed int, course int, time string) {
|
||||
e.Payload = map[string]any{
|
||||
"MODULE": "DEVEMM",
|
||||
"OPERATION": "GETPOS",
|
||||
"RESPONSE": map[string]any{
|
||||
"ERRORCODE": errorCode,
|
||||
"ERRORCAUSE": errorCause,
|
||||
"SERIAL": serial,
|
||||
"P": map[string]any{
|
||||
"V": errorCode == 0,
|
||||
"J": fmt.Sprintf("%4.6v", longitude),
|
||||
"W": fmt.Sprintf("%4.6v", latitude),
|
||||
"H": fmt.Sprintf("%4.6v", altitude),
|
||||
"S": speed, // unit - 0.01 km/h
|
||||
"C": course, // direction (angle from north)
|
||||
"T": time, // yyyymmddhhmmss
|
||||
},
|
||||
},
|
||||
}
|
||||
}
|
||||
|
||||
*/
|
||||
|
||||
// 3.4.5.28
|
||||
type SpiParameters struct {
|
||||
DriveFlag uint `json:"T"`
|
||||
DataMask uint `json:"M"`
|
||||
Position protocol.GPSData `json:"P"`
|
||||
DeviceStatus SpiDeviceStatus `json:"S"`
|
||||
}
|
||||
|
||||
// 3.4.5.28.1
|
||||
type SpiDeviceStatus struct {
|
||||
Status3G uint `json:"G3"`
|
||||
Status3GStrength uint `json:"G3S"`
|
||||
Status4G uint `json:"G4"`
|
||||
Status4GStrength uint `json:"G4S"`
|
||||
WIFIStatus uint `json:"W"`
|
||||
WIFIStrength uint `json:"WS"`
|
||||
Voltage float64 `json:"V"`
|
||||
DeviceTemperature float64 `json:"DT"`
|
||||
IndoorTemperature float64 `json:"TC"`
|
||||
Speed float64 `json:"S"`
|
||||
KeyIgnitionState uint `json:"SW"`
|
||||
RecordStatus []uint `json:"RE"`
|
||||
Time time.Time `json:"T"`
|
||||
StorageDeviceNumber uint `json:"STC"`
|
||||
StorageDeviceInfo []StorageDeviceInfo `json:"SINFO"`
|
||||
VideoLossStatus []uint `json:"VS"`
|
||||
Humidity float64 `json:"H"`
|
||||
TotalMileage float64 `json:"TM"`
|
||||
HardDriveHeating uint `json:"HTR"`
|
||||
}
|
||||
|
||||
func (g *SpiDeviceStatus) MarshalJSON() ([]byte, error) {
|
||||
var alias struct {
|
||||
Status3G uint `json:"G3"`
|
||||
Status3GStrength uint `json:"G3S"`
|
||||
Status4G uint `json:"G4"`
|
||||
Status4GStrength uint `json:"G4S"`
|
||||
WIFIStatus uint `json:"W"`
|
||||
WIFIStrength uint `json:"WS"`
|
||||
Voltage uint `json:"V"`
|
||||
DeviceTemperature uint `json:"DT"`
|
||||
IndoorTemperature uint `json:"TC"`
|
||||
Speed uint `json:"S"`
|
||||
SpeedUnits uint `json:"SU"`
|
||||
KeyIgnitionState uint `json:"SW"`
|
||||
RecordStatus []uint `json:"RE"`
|
||||
Time string `json:"T"`
|
||||
StorageDeviceNumber uint `json:"STC"`
|
||||
StorageDeviceInfo []StorageDeviceInfo `json:"SINFO"`
|
||||
VideoLossStatus []uint `json:"VS"`
|
||||
Humidity uint `json:"H"`
|
||||
TotalMileage string `json:"TM"`
|
||||
HardDriveHeating uint `json:"HTR"`
|
||||
}
|
||||
|
||||
convert := func(v float64) uint {
|
||||
if v < 0 {
|
||||
return uint(-v * 100)
|
||||
} else {
|
||||
return uint((v + 100) * 100)
|
||||
}
|
||||
}
|
||||
|
||||
alias.Status3G = g.Status3G
|
||||
alias.Status3GStrength = g.Status3GStrength
|
||||
alias.Status4G = g.Status4G
|
||||
alias.Status4GStrength = g.Status4GStrength
|
||||
alias.WIFIStatus = g.WIFIStatus
|
||||
alias.WIFIStrength = g.WIFIStrength
|
||||
alias.Voltage = uint(g.Voltage * 100)
|
||||
alias.DeviceTemperature = convert(g.DeviceTemperature)
|
||||
alias.IndoorTemperature = convert(g.IndoorTemperature)
|
||||
alias.Speed = uint(g.Speed * 100)
|
||||
alias.SpeedUnits = 0
|
||||
alias.KeyIgnitionState = g.KeyIgnitionState
|
||||
alias.RecordStatus = g.RecordStatus
|
||||
alias.Time = g.Time.Format("20060102150405")
|
||||
alias.StorageDeviceNumber = g.StorageDeviceNumber
|
||||
alias.StorageDeviceInfo = g.StorageDeviceInfo
|
||||
alias.VideoLossStatus = g.VideoLossStatus
|
||||
alias.Humidity = uint(g.Humidity * 10000)
|
||||
alias.TotalMileage = fmt.Sprintf("%.6f", g.TotalMileage)
|
||||
alias.HardDriveHeating = g.HardDriveHeating
|
||||
|
||||
return json.Marshal(alias)
|
||||
}
|
||||
|
||||
func (g *SpiDeviceStatus) UnmarshalJSON(data []byte) (err error) {
|
||||
var alias struct {
|
||||
Status3G uint `json:"G3"`
|
||||
Status3GStrength uint `json:"G3S"`
|
||||
Status4G uint `json:"G4"`
|
||||
Status4GStrength uint `json:"G4S"`
|
||||
WIFIStatus uint `json:"W"`
|
||||
WIFIStrength uint `json:"WS"`
|
||||
Voltage uint `json:"V"`
|
||||
DeviceTemperature uint `json:"TD"`
|
||||
IndoorTemperature uint `json:"TC"`
|
||||
Speed uint `json:"S"`
|
||||
SpeedUnits uint `json:"SU"`
|
||||
KeyIgnitionState uint `json:"SW"`
|
||||
RecordStatus []uint `json:"RE"`
|
||||
Time string `json:"T"`
|
||||
StorageDeviceNumber uint `json:"STC"`
|
||||
StorageDeviceInfo []StorageDeviceInfo `json:"SINFO"`
|
||||
VideoLossStatus []uint `json:"VS"`
|
||||
Humidity uint `json:"H"`
|
||||
TotalMileage string `json:"TM"`
|
||||
HardDriveHeating uint `json:"HTR"`
|
||||
}
|
||||
|
||||
if err = json.Unmarshal(data, &alias); err != nil {
|
||||
return
|
||||
}
|
||||
|
||||
convert := func(v uint) float64 {
|
||||
if v < 10000 {
|
||||
return -float64(v) / 100
|
||||
} else {
|
||||
return float64(v-10000) / 100
|
||||
}
|
||||
}
|
||||
|
||||
g.Status3G = alias.Status3G
|
||||
g.Status3GStrength = alias.Status3GStrength
|
||||
g.Status4G = alias.Status4G
|
||||
g.Status4GStrength = alias.Status4GStrength
|
||||
g.WIFIStatus = alias.WIFIStatus
|
||||
g.WIFIStrength = alias.WIFIStrength
|
||||
g.Voltage = float64(alias.Voltage) / 100
|
||||
g.DeviceTemperature = convert(alias.DeviceTemperature)
|
||||
g.IndoorTemperature = convert(alias.IndoorTemperature)
|
||||
|
||||
g.Speed = float64(alias.Speed) / 100.0
|
||||
switch alias.SpeedUnits {
|
||||
case 0:
|
||||
break
|
||||
case 1:
|
||||
g.Speed *= 1.609
|
||||
default:
|
||||
return fmt.Errorf("Strange speed units")
|
||||
}
|
||||
|
||||
g.KeyIgnitionState = alias.KeyIgnitionState
|
||||
g.RecordStatus = alias.RecordStatus
|
||||
g.Time, _ = time.Parse("20060102150405", alias.Time)
|
||||
g.StorageDeviceNumber = alias.StorageDeviceNumber
|
||||
g.StorageDeviceInfo = alias.StorageDeviceInfo
|
||||
g.VideoLossStatus = alias.VideoLossStatus
|
||||
g.Humidity = float64(alias.Humidity) / 10000
|
||||
|
||||
if g.TotalMileage, err = strconv.ParseFloat(alias.TotalMileage, 64); err != nil {
|
||||
return fmt.Errorf("invalid longitude: %w", err)
|
||||
}
|
||||
|
||||
g.HardDriveHeating = alias.HardDriveHeating
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
// 3.4.5.28.3
|
||||
type StorageDeviceInfo struct {
|
||||
Type uint `json:"T"`
|
||||
MediaTime uint `json:"O"`
|
||||
Status uint `json:"S"`
|
||||
Capacity uint `json:"TS"`
|
||||
FreeCapacity uint `json:"LS"`
|
||||
}
|
||||
90
pkg/models/evem.go
Normal file
90
pkg/models/evem.go
Normal file
@ -0,0 +1,90 @@
|
||||
package models
|
||||
|
||||
import (
|
||||
"gitea.unprism.ru/KRBL/n9m/v2/pkg/protocol"
|
||||
)
|
||||
|
||||
// 3.4.1.3
|
||||
type EventModelGetAlarmStatusInfoResponse struct {
|
||||
ErrorCode uint `json:"ERRORCODE"`
|
||||
ErrorCause string `json:"ERRORCAUSE"`
|
||||
StorageErrors []StorageErrorStatus `json:"ST"`
|
||||
AnotherStorageErrors []AnotherStorageErrorStatus `json:"VS"`
|
||||
VideoLossErrors []VideoLossErrorStatus `json:"VL"`
|
||||
GPSError GPSErrorStatus `json:"GFA"`
|
||||
GPSAntennaError GPSAntennaErrorStatus `json:"GPSS"`
|
||||
}
|
||||
|
||||
// 3.4.1.4.21
|
||||
type StorageErrorStatus struct {
|
||||
CameraCovered uint `json:"ISA"`
|
||||
ChannelBind uint `json:"LHC"`
|
||||
}
|
||||
|
||||
// 3.4.1.4.4
|
||||
type AnotherStorageErrorStatus struct {
|
||||
}
|
||||
|
||||
// 3.4.1.4.5
|
||||
type VideoLossErrorStatus struct {
|
||||
}
|
||||
|
||||
// 3.4.1.4.44
|
||||
type GPSErrorStatus struct {
|
||||
}
|
||||
|
||||
// 3.4.1.4.46
|
||||
type GPSAntennaErrorStatus struct {
|
||||
}
|
||||
|
||||
// 3.4.1.5
|
||||
// Alarm upload
|
||||
type SendAlarmInfoParameters struct {
|
||||
AlarmType protocol.AlarmType `json:"ALARMTYPE"`
|
||||
CommandType uint `json:"CMDTYPE"`
|
||||
AlarmUID uint `json:"ALARMUID"`
|
||||
NumberOfRestarts uint `json:"RUN"`
|
||||
AlarmLevel protocol.AlarmLevel `json:"ALARMAS"`
|
||||
AlarmCount uint `json:"ALARMCOUNT"`
|
||||
TriggerType protocol.TriggerType `json:"TRIGGERTYPE"`
|
||||
ContinueTime uint `json:"CONTINUETIME"`
|
||||
CurrentTime uint `json:"CURRENTTIME"`
|
||||
Language protocol.Language `json:"L"`
|
||||
GPSData protocol.GPSData `json:"P"`
|
||||
RealTimeUpload uint `json:"REAL"`
|
||||
InstructionSerial uint `json:"CMDNO"`
|
||||
}
|
||||
|
||||
// 3.4.1.5
|
||||
// Alarm upload
|
||||
type SendAlarmInfoResponse struct {
|
||||
ErrorCode uint `json:"ERRORCODE"`
|
||||
AlarmType protocol.AlarmType `json:"ALARMTYPE"`
|
||||
ErrorCause string `json:"ERRORCAUSE"`
|
||||
CommandType uint `json:"CMDTYPE"`
|
||||
AlarmUID uint `json:"ALARMUID"`
|
||||
NumberOfRestarts uint `json:"RUN"`
|
||||
InstructionSerial uint `json:"CMDNO"`
|
||||
}
|
||||
|
||||
// 3.4.1.5.1
|
||||
type SendAlarmInfoCameraParameters struct {
|
||||
SendAlarmInfoParameters
|
||||
|
||||
Channel uint `json:"CHANNEL"`
|
||||
ChannelMask uint `json:"CHANNELMASK"`
|
||||
LCH []uint `json:"LCH"`
|
||||
Push uint `json:"PUSH"`
|
||||
AlarmName string `json:"ALARMNAME"`
|
||||
AlarmAbbreviation string `json:"SER"`
|
||||
}
|
||||
|
||||
type EventModelGetAlarmingResponse struct {
|
||||
CameraCoveredChannelMask uint `json:"VS_CH"`
|
||||
CameraCoveredAlarmMask uint `json:"VS_AT"`
|
||||
CameraCoveredStatusMask uint `json:"VS_AS"`
|
||||
|
||||
VideoLossChannelMask uint `json:"VL_CH"`
|
||||
VideoLossAlarmMask uint `json:"VL_AT"`
|
||||
VideoLossStatusMask uint `json:"VL_AS"`
|
||||
}
|
||||
160
pkg/models/mediastreammodel.go
Normal file
160
pkg/models/mediastreammodel.go
Normal file
@ -0,0 +1,160 @@
|
||||
package models
|
||||
|
||||
import "gitea.unprism.ru/KRBL/n9m/v2/pkg/protocol"
|
||||
|
||||
type StreamType uint
|
||||
|
||||
const (
|
||||
StreamTypeSub StreamType = iota
|
||||
StreamTypeMain
|
||||
StreamTypeMobile
|
||||
)
|
||||
|
||||
type MediaStreamModelRequestLiveVideoRequest struct {
|
||||
SSRC uint `json:"SSRC,omitempty"`
|
||||
StreamName string `json:"STREAMNAME"`
|
||||
StreamType StreamType `json:"STREAMTYPE"`
|
||||
Channel uint `json:"CHANNEL"`
|
||||
AudioValid uint `json:"AUDIOVALID"`
|
||||
Destination string `json:"IPANDPORT,omitempty"`
|
||||
FrameCount uint `json:"FRAMECOUNT,omitempty"`
|
||||
FrameMode uint `json:"FRAMEMODE"`
|
||||
}
|
||||
|
||||
type MediaStreamModelRequestLiveVideoResponse struct {
|
||||
SSRC uint `json:"SSRC"`
|
||||
StreamName string `json:"STREAMNAME"`
|
||||
StreamType StreamType `json:"STREAMTYPE"`
|
||||
ErrorCode uint `json:"ERRORCODE"`
|
||||
ErrorCause string `json:"ERRORCAUSE"`
|
||||
}
|
||||
|
||||
type MediaStreamCommand uint
|
||||
|
||||
const (
|
||||
MediaStreamCommandStop MediaStreamCommand = iota
|
||||
MediaStreamCommandResume
|
||||
MediaStreamCommandPause
|
||||
MediaStreamCommandSwitchVideoStream
|
||||
MediaStreamAudioManagement
|
||||
MediaStreamFrameRate
|
||||
MediaStreamSendingMode
|
||||
)
|
||||
|
||||
type MediaStreamModelControlStreamRequest struct {
|
||||
PayloadType protocol.PayloadType `json:"PT"`
|
||||
SSRC uint16 `json:"SSRC"`
|
||||
StreamName string `json:"STREAMNAME"`
|
||||
Command MediaStreamCommand `json:"CMD"`
|
||||
|
||||
StreamType StreamType `json:"STREAMTYPE,omitempty"`
|
||||
AudioValid uint `json:"AUDIOVALID,omitempty"`
|
||||
FrameMode uint `json:"FRAMEMODE,omitempty"`
|
||||
}
|
||||
|
||||
/*
|
||||
var ip string = os.Getenv("SERVER_IP")
|
||||
|
||||
func (e *Package) MediaRequestDownloadVideo(token int, serial string, session string, camNo int, date string, begin_time string, end_time string, recordID string, serverId int) {
|
||||
e.Payload = map[string]any{
|
||||
"MODULE": "MEDIASTREAMMODEL",
|
||||
"OPERATION": "REQUESTDOWNLOADVIDEO",
|
||||
"PARAMETER": map[string]any{
|
||||
"PT": 3,
|
||||
"SSRC": 1,
|
||||
"STREAMNAME": "DOWNLOAD" + "_" + serial + "_" + fmt.Sprint(camNo) + "_" + fmt.Sprint(serverId),
|
||||
"STREAMTYPE": 1, // main stream
|
||||
"RECORDID": recordID,
|
||||
"CHANNEL": 1 << (camNo - 1),
|
||||
"STARTTIME": date + begin_time,
|
||||
"ENDTIME": date + end_time,
|
||||
"OFFSETFLAG": 1,
|
||||
"OFFSET": 0,
|
||||
"IPANDPORT": ip + ":12092",
|
||||
"SERIAL": token,
|
||||
"DT": 1, // high speed download
|
||||
},
|
||||
"SESSION": session,
|
||||
}
|
||||
}
|
||||
|
||||
// main server util
|
||||
func (e *Package) MediaRequestAliveVideo(token int, camNo int64, Sid string, serial string, quality int64) {
|
||||
|
||||
channel := 0
|
||||
|
||||
if camNo == 1 {
|
||||
channel = 1
|
||||
} else {
|
||||
channel = 1 << (camNo - 1)
|
||||
}
|
||||
e.Payload = map[string]any{
|
||||
"MODULE": "MEDIASTREAMMODEL",
|
||||
"OPERATION": "REQUESTALIVEVIDEO",
|
||||
"PARAMETER": map[string]any{
|
||||
"AUDIOVALID": 1,
|
||||
"CHANNEL": channel,
|
||||
"FRAMEMODE": 0,
|
||||
"IPANDPORT": ip + ":12092",
|
||||
"STREAMNAME": "LIVE" + "_" + serial + "_" + fmt.Sprint(camNo),
|
||||
"STREAMTYPE": quality,
|
||||
"SERIAL": token,
|
||||
},
|
||||
"SESSION": Sid,
|
||||
}
|
||||
}
|
||||
|
||||
// main server util
|
||||
func (e *Package) MediaRequestRemotePlayback(token int, serial string, session string, camNo int, date string, begin_time string, end_time string, serverId int) {
|
||||
if end_time == "" {
|
||||
e.Payload = map[string]any{
|
||||
"MODULE": "MEDIASTREAMMODEL",
|
||||
"OPERATION": "REQUESTREMOTEPLAYBACK",
|
||||
"PARAMETER": map[string]any{
|
||||
"STREAMNAME": "PLAYBACK" + "_" + fmt.Sprint(serial) + "_" + fmt.Sprint(camNo) + "_" + fmt.Sprint(serverId),
|
||||
"STREAMTYPE": 1, // main stream
|
||||
"VIDEOTYPE": 2, // common files
|
||||
"CHANNEL": 1 << (camNo - 1),
|
||||
"STARTTIME": date + begin_time,
|
||||
"IPANDPORT": ip + ":12092",
|
||||
"SERIAL": token,
|
||||
"PBST": 0,
|
||||
},
|
||||
"SESSION": session,
|
||||
}
|
||||
} else {
|
||||
e.Payload = map[string]any{
|
||||
"MODULE": "MEDIASTREAMMODEL",
|
||||
"OPERATION": "REQUESTREMOTEPLAYBACK",
|
||||
"PARAMETER": map[string]any{
|
||||
"STREAMNAME": "PLAYBACK" + "_" + fmt.Sprint(serial) + "_" + fmt.Sprint(camNo) + "_" + fmt.Sprint(serverId),
|
||||
"STREAMTYPE": 1, // main stream
|
||||
"VIDEOTYPE": 2, // common files
|
||||
"CHANNEL": 1 << (camNo - 1),
|
||||
"STARTTIME": date + begin_time,
|
||||
"ENDTIME": date + end_time,
|
||||
"IPANDPORT": ip + ":12092",
|
||||
"SERIAL": token,
|
||||
"PBST": 0,
|
||||
},
|
||||
"SESSION": session,
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// main server util
|
||||
func (e *Package) ControlRemotePlayback(token int, serial string, session string, camNo int, date string, begin_time string, end_time string, serverId int) {
|
||||
e.Payload = map[string]any{
|
||||
"MODULE": "MEDIASTREAMMODEL",
|
||||
"OPERATION": "CONTROLREMOTEPLAYBACK",
|
||||
"PARAMETER": map[string]any{
|
||||
"STREAMNAME": fmt.Sprint(serial) + "_" + fmt.Sprint(camNo) + "_" + fmt.Sprint(serverId),
|
||||
"SERIAL": token,
|
||||
"PALYBACKCMD": 5, // main stream
|
||||
"CHANNEL": 268435455, // common files
|
||||
},
|
||||
"SESSION": session,
|
||||
}
|
||||
}
|
||||
|
||||
*/
|
||||
84
pkg/models/storm.go
Normal file
84
pkg/models/storm.go
Normal file
@ -0,0 +1,84 @@
|
||||
package models
|
||||
|
||||
/*
|
||||
|
||||
func (e *Package) ResponseCalendar(errorCode int, errorCause string, serial int, dates []string) {
|
||||
e.Payload = map[string]any{
|
||||
"MODULE": "STORM",
|
||||
"OPERATION": "GETCALENDAR",
|
||||
"RESPONSE": map[string]any{
|
||||
"ERRORCODE": errorCode,
|
||||
"ERRORCAUSE": errorCause,
|
||||
"SERIAL": serial,
|
||||
"COUNT": len(dates),
|
||||
"CALENDER": dates,
|
||||
// no CHCALENDER[COUNT]
|
||||
// no T[COUNT]
|
||||
},
|
||||
}
|
||||
}
|
||||
|
||||
func (e *Package) RequestCalendar(queryTime string, serial int, session string, camNo int64) {
|
||||
channel := 1 << (camNo - 1)
|
||||
e.Payload = map[string]any{
|
||||
"MODULE": "STORM",
|
||||
"OPERATION": "GETCALENDAR",
|
||||
"PARAMETER": map[string]any{
|
||||
"CALENDARTYPE": 1, // Month data
|
||||
"STREAMTYPE": 1, // Main Stream
|
||||
"FILETYPE": 0b111111, // get file type
|
||||
"PICMTYPE": 0b10, // fixed timing pictures (fixed framerate)
|
||||
"APT0": 0xFFFFFF, // get every alarm
|
||||
"APT1": 0xFFFF, // get every alarm
|
||||
"AUDIOTYPE": 0b111, // normal recording, passenger complaints, alarm recording
|
||||
"CHANNEL": channel, // request all channels
|
||||
"QUERYTIME": queryTime, // year + month = xxxxxx
|
||||
"SERIAL": serial,
|
||||
"NEWSTREAMTYPE": 0b111, // master stream (bit1)
|
||||
"RFSTORAGE": 0, // 0 - hdd, 1 - sd
|
||||
},
|
||||
"SESSION": session,
|
||||
}
|
||||
}
|
||||
|
||||
// filenames without fileextension
|
||||
func (e *Package) ResponseFileList(errorCode int, errorCause string, serial int, filenames []string, fileextensions []int, ids []string) {
|
||||
e.Payload = map[string]any{
|
||||
"MODULE": "STORM",
|
||||
"OPERATION": "QUERYFILELIST",
|
||||
"RESPONSE": map[string]any{
|
||||
"ERRORCODE": errorCode,
|
||||
"ERRORCAUSE": errorCause,
|
||||
"SERIAL": serial,
|
||||
"SENDFILECOUNT": len(filenames),
|
||||
"RECORD": filenames,
|
||||
"FILETYPE": fileextensions,
|
||||
"RECORDID": ids,
|
||||
},
|
||||
}
|
||||
}
|
||||
|
||||
func (e *Package) RequestFileList(queryTime string, serial int, session string, camNo int64) {
|
||||
channel := 1 << (camNo - 1)
|
||||
e.Payload = map[string]any{
|
||||
"MODULE": "STORM",
|
||||
"OPERATION": "QUERYFILELIST",
|
||||
"PARAMETER": map[string]any{
|
||||
"STREAMTYPE": 1, // Main Stream
|
||||
"FILETYPE": 0b111111, // get all filetypes
|
||||
"PICMTYPE": 0b10, // fixed timing pictures (fixed framerate)
|
||||
"APT0": 0xFFFFFF, // get every alarm
|
||||
"APT1": 0xFFFF, // get every alarm
|
||||
"AUDIOTYPE": 0b111, // normal recording, passenger complaints, alarm recording
|
||||
"CHANNEL": channel, // request all channels
|
||||
"STARTTIME": queryTime + "000000",
|
||||
"ENDTIME": queryTime + "235959",
|
||||
"SERIAL": serial,
|
||||
"NEWSTREAMTYPE": 0b10, // master stream (bit1)
|
||||
"RFSTORAGE": 0, // 0 - hdd, 1 - sd
|
||||
},
|
||||
"SESSION": session,
|
||||
}
|
||||
}
|
||||
|
||||
*/
|
||||
335
pkg/parameters/params.go
Normal file
335
pkg/parameters/params.go
Normal file
@ -0,0 +1,335 @@
|
||||
package parameters
|
||||
|
||||
// 7.2
|
||||
type RIP struct {
|
||||
DEVID string // ID номер устройства
|
||||
BN string // бортовой номер
|
||||
BID string // гос. номер
|
||||
LN string // Название линии
|
||||
DID string // номер водительского удостоверения
|
||||
DNAME string // ФИО водителя
|
||||
}
|
||||
|
||||
type VS struct {
|
||||
VIN string // VIN номер
|
||||
}
|
||||
|
||||
type TIMEP struct {
|
||||
DATEM int // формат даты
|
||||
TIMEM int // формат времени
|
||||
TIMEZ string // часовой пояс
|
||||
}
|
||||
|
||||
type ATP struct {
|
||||
GE int // Синхронизация времени при помощи ГНСС
|
||||
NE int // Синхронизация времени при помощи NTP
|
||||
NS string // NTP сервер для синхронизации
|
||||
}
|
||||
|
||||
type SSP struct {
|
||||
UPT int // вкл/выкл
|
||||
DDS int // задержка перед выключением (при выборе зажигания)
|
||||
UH int // час включения
|
||||
UM int // минута включения
|
||||
US int // минута включения
|
||||
DH int // час выключения
|
||||
DM int // минута выключения
|
||||
DS int // секунда выключения
|
||||
CPM int // время выключения света
|
||||
}
|
||||
|
||||
type SWUS struct {
|
||||
IOWP int // Включение через IO
|
||||
GSWP int // включение через G-сенсор
|
||||
TX string // порог по X для G-сенсорa
|
||||
TY string // порог по Y для G-сенсорa
|
||||
TZ string // порог по Z для G-сенсорa
|
||||
RTWP int // удаленное включение
|
||||
PHWP int // включение через телефон
|
||||
PTWP int // включение через SMS
|
||||
}
|
||||
|
||||
type UMP struct {
|
||||
UIF []UIF // список юзеров
|
||||
}
|
||||
|
||||
type UIF struct {
|
||||
UN string // username
|
||||
PW string // password
|
||||
UR int // level
|
||||
}
|
||||
|
||||
type SUBSTRNET struct {
|
||||
NEC []NEC
|
||||
RLE int
|
||||
SAE int
|
||||
SM int // Качество дополнительных потоков
|
||||
TOTALBW int // Максимальная скорость интернета
|
||||
}
|
||||
|
||||
type NEC struct {
|
||||
AEN int // звук (вкл/выкл)
|
||||
VEN int // видео (вкл/выкл)
|
||||
BR int // битрейт
|
||||
ECT int // кодировка
|
||||
FR int // фреймрейт
|
||||
FT int // тип кадров
|
||||
QLT int // качество трансляции
|
||||
RST int // разрешение
|
||||
}
|
||||
|
||||
type DOSD struct {
|
||||
CHN []string // названия каналов
|
||||
}
|
||||
|
||||
type AR struct {
|
||||
HDAE int // Двойная запись на жесткий диск
|
||||
HID int // выбрать место для записи резервного потока
|
||||
RM int // режим записи резервного потока
|
||||
VEC []VEC // параметры суб. потоков
|
||||
}
|
||||
|
||||
type VEC struct {
|
||||
VEN int // видео (вкл/выкл)
|
||||
RST int // разрешение видеопотока
|
||||
ECT int // кодировка
|
||||
FR int // фреймрейт
|
||||
QLT int // качество записи
|
||||
AEN int // звук (вкл/выкл)
|
||||
ALT int // качество видео при тревоге
|
||||
}
|
||||
|
||||
type EOSD struct {
|
||||
COSD []COSD
|
||||
DE int // показывать ли ID регистратора
|
||||
DTE int
|
||||
DX int
|
||||
DY int
|
||||
GE int // показывать ли GPS
|
||||
GX int
|
||||
GY int
|
||||
NE int // показывать ли название канала
|
||||
NX int
|
||||
NY int
|
||||
SE int // показывать ли скорость
|
||||
SX int
|
||||
SY int
|
||||
TE int // показывать ли время
|
||||
TX int
|
||||
TY int
|
||||
VE int // показывать ли номер устройства
|
||||
VX int
|
||||
VY int
|
||||
WME int
|
||||
}
|
||||
|
||||
type COSD struct {
|
||||
EN int
|
||||
F int
|
||||
ID int
|
||||
L int
|
||||
T string
|
||||
TP int
|
||||
X int
|
||||
Y int
|
||||
}
|
||||
|
||||
type IOP struct {
|
||||
EN int
|
||||
AS int
|
||||
EL int
|
||||
SDT int
|
||||
APR APR
|
||||
}
|
||||
|
||||
type SAP struct {
|
||||
EN int
|
||||
AS int
|
||||
WP int
|
||||
SV int
|
||||
AT int
|
||||
APR APR
|
||||
UN int
|
||||
}
|
||||
|
||||
type APR struct {
|
||||
AR ARR
|
||||
SS SS
|
||||
ET int
|
||||
}
|
||||
|
||||
type ARR struct {
|
||||
CH int
|
||||
D int
|
||||
}
|
||||
|
||||
type SS struct {
|
||||
EN int
|
||||
}
|
||||
|
||||
type UAP struct {
|
||||
EN int
|
||||
AS int
|
||||
VT int
|
||||
SDT int
|
||||
}
|
||||
|
||||
type PVLAS struct {
|
||||
EN int
|
||||
AS int
|
||||
SDT int
|
||||
CH int
|
||||
}
|
||||
|
||||
type PMDAS struct {
|
||||
EN int
|
||||
AS int
|
||||
SDT int
|
||||
CH int
|
||||
}
|
||||
|
||||
type DSM struct {
|
||||
DSMA int
|
||||
DSMFE int
|
||||
RWFE int
|
||||
}
|
||||
|
||||
type GSP struct {
|
||||
LANT int // Язык системы
|
||||
GM int // Система геопозиционирования
|
||||
}
|
||||
|
||||
type ETHERNET struct {
|
||||
IPMODE int // Тип IP
|
||||
PIP PIP // Параметры IP
|
||||
DNSMODE int // Автоматическое получение DNS
|
||||
DNS DNS // Параметры DNS
|
||||
KEYS KEYS
|
||||
}
|
||||
|
||||
type PIP struct {
|
||||
IPADDR string // IP адрес
|
||||
SUBMASK string // Маска подсети
|
||||
GATEWAY string // Шлюз сети
|
||||
}
|
||||
|
||||
type DNS struct {
|
||||
PDNS string // Основной DNS
|
||||
ADNS string // Дополнительный DNS
|
||||
}
|
||||
|
||||
type KEYS struct {
|
||||
MAC string `json:"MAC,omitempty"` // MAC-адрес
|
||||
GV uint `json:"GV,omitempty"` // GPS version
|
||||
}
|
||||
|
||||
type WIFI struct {
|
||||
ENABLE int // Статус WIFI
|
||||
ESSID string // Название точки доступа
|
||||
ECRYPTTYPE int // Тип защиты WIFI
|
||||
PWD string // Пароль от WIFI
|
||||
IPMODE int // Вид IP
|
||||
PIP PIP // Параметры IP
|
||||
}
|
||||
|
||||
type M3G struct {
|
||||
M3M M3M // Управление активацией
|
||||
MP MP // Параметры первого модуля связи
|
||||
M4G MP // Параметры второго модуля связи
|
||||
}
|
||||
|
||||
type M3M struct {
|
||||
AT int // Режим работы модуля связи
|
||||
TN1 string // Номер активации 1
|
||||
TN2 string // Номер активации 2
|
||||
TN3 string // Номер активации 3
|
||||
}
|
||||
|
||||
type MP struct {
|
||||
NM int // Выбор соединения
|
||||
APN string // APN
|
||||
UN string // Логин
|
||||
PW string // Пароль
|
||||
}
|
||||
|
||||
type MCMS struct {
|
||||
M int // Битовая маска включенных серверов
|
||||
SP []SP // Параметры сервера
|
||||
}
|
||||
|
||||
type SP struct {
|
||||
EN int // Статус сервера
|
||||
CP int // Протокол подключения
|
||||
NWT int // Способ подключения
|
||||
CS string // Адрес сервера
|
||||
MS string // Адрес медиасервера
|
||||
CPORT int // TCP-порт сервера
|
||||
MPORT int // TCP-порт медиасервера
|
||||
CUPORT int // UDP-порт сервера
|
||||
MUPORT int // UDP-порт медиасервера
|
||||
}
|
||||
|
||||
// 7.30
|
||||
type PGDSM struct {
|
||||
PGPS PGPS `json:"PGPS,omitempty"`
|
||||
PDSM PDSM `json:"PDSM,omitempty"`
|
||||
}
|
||||
|
||||
// 7.30.1
|
||||
// GPS (position) status parameter
|
||||
type PGPS struct {
|
||||
EN uint // Enable
|
||||
MODE uint
|
||||
SEP uint `json:"SEP,omitempty"`
|
||||
TM uint `json:"TM,omitempty"`
|
||||
NUM uint `json:"NUM,omitempty"`
|
||||
}
|
||||
|
||||
// 7.30.2
|
||||
type PDSM struct {
|
||||
}
|
||||
|
||||
// 7.39.3.9
|
||||
// Platform basic information
|
||||
type PSI struct {
|
||||
CG CG `json:"CG,omitempty"`
|
||||
}
|
||||
|
||||
// 7.39.3.8
|
||||
// Total record information for call
|
||||
type CG struct {
|
||||
ECL uint // The maximum calling time
|
||||
ECML uint // The maximum calling time per month
|
||||
AS uint // Terminal phonecall answering strategy
|
||||
AT uint // Answer automatically when exceed time
|
||||
}
|
||||
|
||||
type Setting struct {
|
||||
KEYS KEYS `json:"KEYS,omitempty"`
|
||||
RIP RIP `json:"RIP,omitempty"`
|
||||
TIMEP TIMEP `json:"TIMEP,omitempty"`
|
||||
ATP ATP `json:"ATP,omitempty"`
|
||||
SSP SSP `json:"SSP,omitempty"`
|
||||
GSP GSP `json:"GSP,omitempty"`
|
||||
UMP UMP `json:"UMP,omitempty"`
|
||||
ETHERNET ETHERNET `json:"ETHERNET,omitempty"`
|
||||
WIFI WIFI `json:"WIFI,omitempty"`
|
||||
M3G M3G `json:"M3G,omitempty"`
|
||||
MCMS MCMS `json:"MCMS,omitempty"`
|
||||
SUBSTRNET SUBSTRNET `json:"SUBSTRNET,omitempty"`
|
||||
DOSD DOSD `json:"DOSD,omitempty"`
|
||||
AR AR `json:"AR,omitempty"`
|
||||
MAIN []VEC `json:"MAIN,omitempty"`
|
||||
EOSD []EOSD `json:"EOSD,omitempty"`
|
||||
IOP []IOP `json:"IOP,omitempty"`
|
||||
SAP SAP `json:"SAP,omitempty"`
|
||||
UAP UAP `json:"UAP,omitempty"`
|
||||
PGDSM PGDSM `json:"PGDSM,omitempty"`
|
||||
PVLAS PVLAS `json:"PVLAS,omitempty"`
|
||||
PMDAS PMDAS `json:"PMDAS,omitempty"`
|
||||
VS VS `json:"VS,omitempty"`
|
||||
PSI PSI `json:"PSI,omitempty"`
|
||||
|
||||
SWUS SWUS `json:"SWUS,omitempty"`
|
||||
// DSM DSM `json:"DSM,omitempty"`
|
||||
}
|
||||
57
pkg/protocol/alarms.go
Normal file
57
pkg/protocol/alarms.go
Normal file
@ -0,0 +1,57 @@
|
||||
package protocol
|
||||
|
||||
type AlarmType uint
|
||||
|
||||
const (
|
||||
AlarmTypeVideoLoss AlarmType = iota
|
||||
AlarmTypeCameraCovered
|
||||
AlarmTypeMotionDetection
|
||||
AlarmTypeStorageAbnormal
|
||||
AlarmTypeUserDefined
|
||||
AlarmTypeSentriesInspection
|
||||
AlarmTypeViolation
|
||||
AlarmTypeEmergency
|
||||
AlarmTypeSpeedAlarm
|
||||
AlarmTypeLowVoltage
|
||||
AlarmTypeOutOfFence = iota + 7
|
||||
AlarmTypeAccAlarm
|
||||
AlarmTypePeripheralsDropped
|
||||
AlarmTypeStopAnnouncement
|
||||
AlarmTypeGpsAntenna
|
||||
AlarmTypeDayNightSwitch
|
||||
AlarmTypeProhibitDriving
|
||||
AlarmTypeSerialAlarm = iota + 15
|
||||
AlarmTypeFatigueAlarm
|
||||
AlarmTypeTakeOutParking
|
||||
AlarmTypeGestureAlarm
|
||||
AlarmTypeGreenDriving
|
||||
AlarmTypeIllegalIgnition
|
||||
AlarmTypeIllegalShutdown
|
||||
AlarmTypeCustomExternal
|
||||
AlarmTypeThinkingLKJ
|
||||
AlarmTypeTAX3
|
||||
AlarmTypeOilAlarm
|
||||
AlarmTypeBusLineOccupation
|
||||
AlarmTypeForgottenAlarm
|
||||
AlarmTypeSpecialCustomerFault
|
||||
AlarmTypeTemperatureAbnormal
|
||||
AlarmTypeTemperatureChangeAbnormal
|
||||
AlarmTypeSmokeAlarm
|
||||
AlarmTypeGBox
|
||||
AlarmTypeLicensePlateRecognition
|
||||
AlarmTypeAnotherSpeedAlarm
|
||||
AlarmTypeWirelessSignalAbnormal
|
||||
AlarmTypeArming
|
||||
AlarmTypePhoneCall
|
||||
AlarmTypeGPSFault
|
||||
AlarmTypeDSMFault
|
||||
AlarmTypeFireBox
|
||||
)
|
||||
|
||||
type AlarmLevel uint
|
||||
|
||||
const (
|
||||
AlarmLevelImportant AlarmLevel = iota
|
||||
AlarmLevelGeneral
|
||||
AlarmLevelEmergency
|
||||
)
|
||||
192
pkg/protocol/io.go
Normal file
192
pkg/protocol/io.go
Normal file
@ -0,0 +1,192 @@
|
||||
package protocol
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"encoding/binary"
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"github.com/icza/bitio"
|
||||
"log"
|
||||
"time"
|
||||
)
|
||||
|
||||
// Read package
|
||||
func (e *Package) ReadPackage() bool {
|
||||
if len(e.Accum) < 12 {
|
||||
return false
|
||||
}
|
||||
|
||||
r := bitio.NewReader(bytes.NewBuffer(e.Accum))
|
||||
e.Version = uint8(r.TryReadBits(2))
|
||||
e.EncryptionFlag = r.TryReadBool()
|
||||
e.CompressFlag = r.TryReadBool()
|
||||
e.CSRCCount = uint8(r.TryReadBits(4))
|
||||
e.PayloadType = PayloadType(r.TryReadBits(8))
|
||||
e.SSRC = SpecialPayloadType((r.TryReadBits(8) | (r.TryReadBits(8) << 8)))
|
||||
|
||||
if e.EncryptionFlag && e.CompressFlag {
|
||||
// TODO: get snippet, that use this code
|
||||
r.TryReadBits(8 * 4)
|
||||
e.payloadLen = r.TryReadBits(8)
|
||||
r.TryReadBits(3 * 8)
|
||||
|
||||
if uint64(len(e.Accum)) < e.payloadLen+12 {
|
||||
return false
|
||||
}
|
||||
} else {
|
||||
e.payloadLen = r.TryReadBits(32)
|
||||
|
||||
// WTF: e.CC is useless
|
||||
for i := uint64(0); i < 1; i++ {
|
||||
e.CSRC[i] = r.TryReadBits(32)
|
||||
}
|
||||
}
|
||||
|
||||
if e.payloadLen > 1e6 {
|
||||
log.Printf("%v\n", e)
|
||||
log.Panicln("CORRUPTED PACKAGE")
|
||||
}
|
||||
|
||||
numOfBytes := 0
|
||||
|
||||
if e.payloadLen != 0 {
|
||||
e.RawPayload = make([]byte, e.payloadLen)
|
||||
numOfBytes = r.TryRead(e.RawPayload)
|
||||
} else {
|
||||
e.RawPayload = []byte{}
|
||||
}
|
||||
|
||||
if numOfBytes != int(e.payloadLen) {
|
||||
return false
|
||||
}
|
||||
|
||||
e.Accum = e.Accum[12+e.payloadLen:]
|
||||
|
||||
switch e.PayloadType {
|
||||
case PayloadTypeData:
|
||||
if err := json.Unmarshal(e.RawPayload, &e.Payload); err != nil {
|
||||
log.Printf("Error parsing JSON payload: %v", err)
|
||||
return false
|
||||
}
|
||||
case PayloadTypeSpecial:
|
||||
switch e.SSRC {
|
||||
case SpecialPayloadTypeGPS:
|
||||
e.GPS.GPSStatus = e.RawPayload[0]
|
||||
e.GPS.Expand = e.RawPayload[1]
|
||||
e.GPS.Real = e.RawPayload[2]
|
||||
|
||||
e.GPS.Longitude = float64(binary.BigEndian.Uint32(e.RawPayload[4:8])) / 1e6
|
||||
e.GPS.Latitude = float64(binary.BigEndian.Uint32(e.RawPayload[8:12])) / 1e6
|
||||
e.GPS.Speed = float64(binary.BigEndian.Uint32(e.RawPayload[12:16])) / 100
|
||||
e.GPS.Direction = float64(binary.BigEndian.Uint32(e.RawPayload[16:20])) / 100
|
||||
e.GPS.Altitude = int32(binary.BigEndian.Uint32(e.RawPayload[20:24]))
|
||||
|
||||
var err error
|
||||
if e.GPS.Time, err = time.Parse("20060102150405", string(e.RawPayload[24:38])); err != nil {
|
||||
log.Printf("Error parsing time: %v", err)
|
||||
}
|
||||
default:
|
||||
fmt.Println("N9M parser warning: unknown special payload type", e.SSRC)
|
||||
}
|
||||
// default:
|
||||
// fmt.Println("N9M parser warning: unknown payload type", e.PayloadType)
|
||||
}
|
||||
|
||||
if r.TryError != nil {
|
||||
log.Printf("TryError encountered: %v", r.TryError)
|
||||
return false
|
||||
}
|
||||
|
||||
return true
|
||||
}
|
||||
|
||||
func (e *Package) PackPayload() (err error) {
|
||||
e.RawPayload, err = json.Marshal(e.Payload)
|
||||
e.payloadLen = uint64(len(e.RawPayload))
|
||||
|
||||
if e.payloadLen != 0 {
|
||||
e.RawPayload = append(e.RawPayload, 0)
|
||||
e.payloadLen++
|
||||
}
|
||||
|
||||
return
|
||||
}
|
||||
|
||||
func (e *Package) PackPackage() []byte {
|
||||
var err error
|
||||
|
||||
if err = e.PackPayload(); err != nil {
|
||||
log.Printf("Error while packing payload: %v", err)
|
||||
return []byte{}
|
||||
}
|
||||
|
||||
return e.PackRawPackage()
|
||||
}
|
||||
|
||||
func (e *Package) PackRawPackage() []byte {
|
||||
var err error
|
||||
|
||||
b := &bytes.Buffer{}
|
||||
w := bitio.NewWriter(b)
|
||||
|
||||
w.TryWriteBits(uint64(e.Version), 2)
|
||||
|
||||
w.TryWriteBool(e.EncryptionFlag)
|
||||
w.TryWriteBool(e.CompressFlag)
|
||||
|
||||
w.TryWriteBits(uint64(e.CSRCCount), 4)
|
||||
w.TryWriteBits(uint64(e.PayloadType), 8)
|
||||
w.TryWriteBits(uint64(e.SSRC), 16)
|
||||
|
||||
w.TryWriteBits(e.payloadLen, 32)
|
||||
|
||||
// WTF: e.CC is useless
|
||||
for i := uint64(0); i < 1; i++ {
|
||||
w.TryWriteBits(e.CSRC[i], 32)
|
||||
}
|
||||
|
||||
if e.payloadLen != 0 {
|
||||
w.TryWrite(e.RawPayload)
|
||||
}
|
||||
|
||||
if err = w.Close(); err != nil {
|
||||
log.Printf("Error while closing writer: %v", err)
|
||||
return []byte{}
|
||||
}
|
||||
|
||||
return b.Bytes()
|
||||
}
|
||||
|
||||
func (e *Package) GetParametersAs(parameters any) error {
|
||||
marshal, err := json.Marshal(e.Payload.Parameter)
|
||||
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
return json.Unmarshal(marshal, parameters)
|
||||
}
|
||||
|
||||
func (e *Package) SetParameters(parameters any) {
|
||||
e.Payload.Response = nil
|
||||
e.Payload.Parameter = parameters
|
||||
}
|
||||
|
||||
func (e *Package) GetResponseAs(response any) error {
|
||||
marshal, err := json.Marshal(e.Payload.Response)
|
||||
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
return json.Unmarshal(marshal, response)
|
||||
}
|
||||
|
||||
func (e *Package) SetResponse(response any) {
|
||||
e.Payload.Parameter = nil
|
||||
e.Payload.Response = response
|
||||
}
|
||||
|
||||
func (e *Package) AddToAccum(data []byte) {
|
||||
e.Accum = append(e.Accum, data...)
|
||||
}
|
||||
22
pkg/protocol/languages.go
Normal file
22
pkg/protocol/languages.go
Normal file
@ -0,0 +1,22 @@
|
||||
package protocol
|
||||
|
||||
type Language uint
|
||||
|
||||
const (
|
||||
LanguageSimplifiedChinese Language = iota
|
||||
LanguageEnglish
|
||||
LanguageKorean
|
||||
LanguageItalian
|
||||
LanguageGerman
|
||||
LanguageThai
|
||||
LanguageTurkey
|
||||
LanguagePortugal
|
||||
LanguageSpain
|
||||
LanguageRomania
|
||||
LanguageGreece
|
||||
LanguageFrench
|
||||
LanguageRussian
|
||||
LanguageDutch
|
||||
LanguageHebrew
|
||||
LanguageChineseTraditional
|
||||
)
|
||||
139
pkg/protocol/scheme.go
Normal file
139
pkg/protocol/scheme.go
Normal file
@ -0,0 +1,139 @@
|
||||
package protocol
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"strconv"
|
||||
"time"
|
||||
)
|
||||
|
||||
type PayloadType uint8
|
||||
|
||||
const (
|
||||
PayloadTypeData PayloadType = 0
|
||||
PayloadTypeLive PayloadType = 2
|
||||
PayloadTypeDownload PayloadType = 3
|
||||
PayloadTypePlayback PayloadType = 4
|
||||
PayloadTypeCapturedPhotos PayloadType = 6
|
||||
PayloadTypeParameterImport PayloadType = 10
|
||||
PayloadTypeParameterExport PayloadType = 11
|
||||
PayloadTypeTransmissionSubStream PayloadType = 15
|
||||
PayloadTypeRecordingSubStream PayloadType = 16
|
||||
PayloadTypeBlackBox PayloadType = 17
|
||||
PayloadTypeSpecial PayloadType = 22
|
||||
PayloadTypeMaintainData PayloadType = 30
|
||||
)
|
||||
|
||||
type SpecialPayloadType uint16
|
||||
|
||||
const (
|
||||
SpecialPayloadTypeHeartbeat SpecialPayloadType = iota
|
||||
SpecialPayloadTypeHeartbeatWithoutBody
|
||||
SpecialPayloadTypeGPS
|
||||
SpecialPayloadTypeMileage
|
||||
SpecialPayloadTypeEnvironmentalQuality
|
||||
SpecialPayloadTypeDrivingPosture
|
||||
SpecialPayloadTypeScanningGun
|
||||
SpecialPayloadTypeOil
|
||||
SpecialPayloadTypeGDS
|
||||
SpecialPayloadTypeGPSToBWS
|
||||
SpecialPayloadTypeCANBOX
|
||||
SpecialPayloadTypeGSenor
|
||||
SpecialPayloadTypeAckGPS
|
||||
)
|
||||
|
||||
type Message struct {
|
||||
Module string `json:"MODULE"`
|
||||
Session string `json:"SESSION"`
|
||||
Operation string `json:"OPERATION"`
|
||||
Parameter interface{} `json:"PARAMETER,omitempty"`
|
||||
Response interface{} `json:"RESPONSE,omitempty"`
|
||||
}
|
||||
|
||||
// 3.4.5.27.1
|
||||
type GPSData struct {
|
||||
GPSStatus uint8
|
||||
Expand uint8
|
||||
Real uint8
|
||||
Longitude float64
|
||||
Latitude float64
|
||||
Speed float64
|
||||
Direction float64
|
||||
Altitude int32
|
||||
Time time.Time
|
||||
}
|
||||
|
||||
func (g *GPSData) MarshalJSON() ([]byte, error) {
|
||||
var alias struct {
|
||||
GPSStatus uint8 `json:"V"`
|
||||
Longitude string `json:"J"`
|
||||
Latitude string `json:"W"`
|
||||
Speed uint `json:"S"`
|
||||
Direction uint `json:"C"`
|
||||
Altitude int32 `json:"H"`
|
||||
Time string `json:"T"`
|
||||
}
|
||||
|
||||
alias.GPSStatus = g.GPSStatus
|
||||
alias.Longitude = fmt.Sprintf("%.6f", g.Longitude)
|
||||
alias.Latitude = fmt.Sprintf("%.6f", g.Latitude)
|
||||
alias.Speed = uint(g.Speed * 100)
|
||||
alias.Direction = uint(g.Direction * 100)
|
||||
alias.Altitude = g.Altitude
|
||||
alias.Time = g.Time.Format("20060102150405")
|
||||
|
||||
return json.Marshal(alias)
|
||||
}
|
||||
|
||||
func (g *GPSData) UnmarshalJSON(data []byte) (err error) {
|
||||
var alias struct {
|
||||
GPSStatus uint8 `json:"V"`
|
||||
Longitude string `json:"J"`
|
||||
Latitude string `json:"W"`
|
||||
Speed uint `json:"S"`
|
||||
Direction uint `json:"C"`
|
||||
Altitude int32 `json:"H"`
|
||||
Time string `json:"T"`
|
||||
}
|
||||
|
||||
if err = json.Unmarshal(data, &alias); err != nil {
|
||||
return
|
||||
}
|
||||
|
||||
g.GPSStatus = alias.GPSStatus
|
||||
|
||||
if g.Longitude, err = strconv.ParseFloat(alias.Longitude, 64); err != nil {
|
||||
return fmt.Errorf("invalid longitude: %w", err)
|
||||
}
|
||||
|
||||
if g.Latitude, err = strconv.ParseFloat(alias.Latitude, 64); err != nil {
|
||||
return fmt.Errorf("invalid latitude: %w", err)
|
||||
}
|
||||
|
||||
g.Speed = float64(alias.Speed) / 100.0
|
||||
g.Direction = float64(alias.Direction) / 100.0
|
||||
g.Altitude = alias.Altitude
|
||||
g.Time, _ = time.Parse("20060102150405", alias.Time)
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
type Package struct {
|
||||
Version uint8
|
||||
EncryptionFlag bool
|
||||
CompressFlag bool
|
||||
CSRCCount uint8
|
||||
|
||||
PayloadType PayloadType
|
||||
SSRC SpecialPayloadType
|
||||
Reserved uint64
|
||||
CSRC [16]uint64
|
||||
|
||||
GPS GPSData
|
||||
|
||||
payloadLen uint64
|
||||
Payload Message
|
||||
RawPayload []byte
|
||||
|
||||
Accum []byte
|
||||
}
|
||||
8
pkg/protocol/trigger.go
Normal file
8
pkg/protocol/trigger.go
Normal file
@ -0,0 +1,8 @@
|
||||
package protocol
|
||||
|
||||
type TriggerType uint
|
||||
|
||||
const (
|
||||
TriggerTypeManual TriggerType = iota
|
||||
TriggerTypeAutomatic
|
||||
)
|
||||
167
pkg/smart/channel.go
Normal file
167
pkg/smart/channel.go
Normal file
@ -0,0 +1,167 @@
|
||||
package smart
|
||||
|
||||
import (
|
||||
"errors"
|
||||
"gitea.unprism.ru/KRBL/n9m/v2/pkg/models"
|
||||
"gitea.unprism.ru/KRBL/n9m/v2/pkg/protocol"
|
||||
"gitea.unprism.ru/KRBL/n9m/v2/pkg/utils"
|
||||
"net"
|
||||
"sync"
|
||||
)
|
||||
|
||||
const (
|
||||
charset = "ABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789"
|
||||
length = 6
|
||||
)
|
||||
|
||||
func NewSmartChannelPackage(conn net.Conn, mainSmartPackage *SmartPackage) (*SmartChannelPackage, error) {
|
||||
pack := NewSmartPackage(conn)
|
||||
pack.pack.Payload.Session = mainSmartPackage.pack.Payload.Session
|
||||
|
||||
channelName, err := registerChannelHandles(pack)
|
||||
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
smartChannelPackage := &SmartChannelPackage{
|
||||
pack: pack,
|
||||
mainPack: mainSmartPackage,
|
||||
|
||||
channelName: channelName,
|
||||
|
||||
mutex: sync.RWMutex{},
|
||||
ssrc: make(map[uint16][]func([]byte) error),
|
||||
}
|
||||
|
||||
pack.AddPayloadHandler(protocol.PayloadTypeLive, smartChannelPackage.handleLiveVideo)
|
||||
|
||||
return smartChannelPackage, nil
|
||||
}
|
||||
|
||||
func (channelPack *SmartChannelPackage) Run() error {
|
||||
for {
|
||||
if err := channelPack.pack.Handle(); err != nil {
|
||||
return err
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func (channelPack *SmartChannelPackage) GetChannelName() string {
|
||||
return channelPack.channelName
|
||||
}
|
||||
|
||||
func (channelPack *SmartChannelPackage) AddLiveSource(ssrc uint16, source func([]byte) error) {
|
||||
channelPack.mutex.Lock()
|
||||
defer channelPack.mutex.Unlock()
|
||||
|
||||
channelPack.ssrc[ssrc] = append(channelPack.ssrc[ssrc], source)
|
||||
}
|
||||
|
||||
func (channelPack *SmartChannelPackage) handleLiveVideo(sPack *SmartPackage, pack protocol.Package) error {
|
||||
channelPack.mutex.RLock()
|
||||
|
||||
sources, ok := channelPack.ssrc[uint16(pack.SSRC)]
|
||||
|
||||
if !ok || len(sources) == 0 {
|
||||
channelPack.mutex.RUnlock()
|
||||
|
||||
var request = models.MediaStreamModelControlStreamRequest{
|
||||
PayloadType: protocol.PayloadTypeLive,
|
||||
SSRC: uint16(pack.SSRC),
|
||||
StreamName: channelPack.channelName,
|
||||
Command: models.MediaStreamCommandStop,
|
||||
}
|
||||
|
||||
pack.PayloadType = protocol.PayloadTypeData
|
||||
pack.Payload.Module = "MEDIASTREAMMODEL"
|
||||
pack.Payload.Operation = "CONTROLSTREAM"
|
||||
pack.SetParameters(request)
|
||||
|
||||
if _, err := channelPack.mainPack.Write(pack.PackPackage()); err != nil {
|
||||
panic(err)
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
var errorMask = map[int]bool{}
|
||||
|
||||
for i, source := range sources {
|
||||
if err := source(pack.RawPayload); err != nil {
|
||||
errorMask[i] = true
|
||||
}
|
||||
}
|
||||
|
||||
channelPack.mutex.RUnlock()
|
||||
|
||||
if len(errorMask) > 0 {
|
||||
newSources := make([]func([]byte) error, 0, len(sources))
|
||||
|
||||
for i, source := range sources {
|
||||
if !errorMask[i] {
|
||||
newSources = append(newSources, source)
|
||||
}
|
||||
}
|
||||
|
||||
channelPack.mutex.Lock()
|
||||
channelPack.ssrc[uint16(pack.SSRC)] = newSources
|
||||
channelPack.mutex.Unlock()
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func registerChannelHandles(pack *SmartPackage) (string, error) {
|
||||
res := make(chan error, 1)
|
||||
|
||||
pack.AddJSONHandler("CERTIFICATE", "CREATESTREAM", func(smartPackage *SmartPackage, p protocol.Package) error {
|
||||
var params models.CertificateCreateStreamResponse
|
||||
|
||||
if err := p.GetResponseAs(¶ms); err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
if params.ErrorCode != 0 {
|
||||
res <- errors.New(params.ErrorCause)
|
||||
} else {
|
||||
res <- nil
|
||||
}
|
||||
|
||||
return nil
|
||||
})
|
||||
|
||||
channelName := utils.RandomString(length, charset)
|
||||
|
||||
var request = models.CertificateCreateStreamRequest{
|
||||
StreamName: channelName,
|
||||
}
|
||||
|
||||
pack.pack.Payload.Module = "CERTIFICATE"
|
||||
pack.pack.Payload.Operation = "CREATESTREAM"
|
||||
pack.pack.SetParameters(request)
|
||||
|
||||
pack.Write(pack.pack.PackPackage())
|
||||
|
||||
for {
|
||||
if err := pack.Handle(); err != nil {
|
||||
return "", err
|
||||
}
|
||||
|
||||
select {
|
||||
case err := <-res:
|
||||
close(res)
|
||||
|
||||
if err != nil {
|
||||
return "", err
|
||||
}
|
||||
|
||||
pack.AddJSONHandler("CERTIFICATE", "CREATESTREAM", func(smartPackage *SmartPackage, p protocol.Package) error {
|
||||
return errors.New("stream already exists")
|
||||
})
|
||||
|
||||
return channelName, nil
|
||||
default:
|
||||
}
|
||||
}
|
||||
}
|
||||
33
pkg/smart/scheme.go
Normal file
33
pkg/smart/scheme.go
Normal file
@ -0,0 +1,33 @@
|
||||
package smart
|
||||
|
||||
import (
|
||||
"gitea.unprism.ru/KRBL/n9m/v2/pkg/models"
|
||||
"gitea.unprism.ru/KRBL/n9m/v2/pkg/protocol"
|
||||
"net"
|
||||
"sync"
|
||||
)
|
||||
|
||||
type ProcessFunc func(*SmartPackage, protocol.Package) error
|
||||
type AlarmProcessFunc func(*SmartPackage, protocol.Package, models.SendAlarmInfoResponse) error
|
||||
|
||||
type SmartPackage struct {
|
||||
pack protocol.Package
|
||||
|
||||
conn net.Conn
|
||||
buff []byte
|
||||
|
||||
payloadProcess map[protocol.PayloadType]ProcessFunc
|
||||
jsonProcess map[string]ProcessFunc
|
||||
alarmProcess map[protocol.AlarmType]AlarmProcessFunc
|
||||
Storage map[string]interface{}
|
||||
}
|
||||
|
||||
type SmartChannelPackage struct {
|
||||
pack *SmartPackage
|
||||
mainPack *SmartPackage
|
||||
|
||||
channelName string
|
||||
|
||||
mutex sync.RWMutex
|
||||
ssrc map[uint16][]func([]byte) error
|
||||
}
|
||||
144
pkg/smart/smart.go
Normal file
144
pkg/smart/smart.go
Normal file
@ -0,0 +1,144 @@
|
||||
package smart
|
||||
|
||||
import (
|
||||
"errors"
|
||||
"fmt"
|
||||
"gitea.unprism.ru/KRBL/n9m/v2/pkg/models"
|
||||
"gitea.unprism.ru/KRBL/n9m/v2/pkg/protocol"
|
||||
"net"
|
||||
)
|
||||
|
||||
type notFoundError struct {
|
||||
message string
|
||||
}
|
||||
|
||||
func (e *notFoundError) Error() string {
|
||||
return fmt.Sprintf("not found %s", e.message)
|
||||
}
|
||||
|
||||
func NewSmartPackage(conn net.Conn) *SmartPackage {
|
||||
return &SmartPackage{
|
||||
pack: protocol.Package{},
|
||||
conn: conn,
|
||||
buff: make([]byte, 1024),
|
||||
payloadProcess: make(map[protocol.PayloadType]ProcessFunc),
|
||||
jsonProcess: make(map[string]ProcessFunc),
|
||||
alarmProcess: make(map[protocol.AlarmType]AlarmProcessFunc),
|
||||
Storage: make(map[string]interface{}),
|
||||
}
|
||||
}
|
||||
|
||||
func (pack *SmartPackage) AddPayloadHandler(payloadType protocol.PayloadType, processFunc ProcessFunc) {
|
||||
pack.payloadProcess[payloadType] = processFunc
|
||||
}
|
||||
|
||||
func (pack *SmartPackage) AddJSONHandler(module, operation string, processFunc ProcessFunc) {
|
||||
pack.jsonProcess[fmt.Sprintf("%s:%s", module, operation)] = processFunc
|
||||
}
|
||||
|
||||
func (pack *SmartPackage) AddAlarmHandler(alarmType protocol.AlarmType, processFunc AlarmProcessFunc) {
|
||||
pack.alarmProcess[alarmType] = processFunc
|
||||
}
|
||||
|
||||
func (pack *SmartPackage) handleAlarm() (err error) {
|
||||
if !(pack.pack.PayloadType == protocol.PayloadTypeData && pack.pack.Payload.Module == "EVEM" && pack.pack.Payload.Operation == "SENDALARMINFO") {
|
||||
return fmt.Errorf("invalid payload type or operation for alarm handling")
|
||||
}
|
||||
|
||||
var params models.SendAlarmInfoParameters
|
||||
if err = pack.pack.GetParametersAs(¶ms); err != nil {
|
||||
return fmt.Errorf("invalid payload")
|
||||
}
|
||||
|
||||
var processFunc AlarmProcessFunc
|
||||
var ok bool
|
||||
if processFunc, ok = pack.alarmProcess[params.AlarmType]; !ok {
|
||||
return ¬FoundError{
|
||||
message: fmt.Sprintf("alarm %d", params.AlarmType),
|
||||
}
|
||||
}
|
||||
|
||||
var response models.SendAlarmInfoResponse
|
||||
|
||||
response.ErrorCode = 0
|
||||
response.AlarmType = params.AlarmType
|
||||
response.CommandType = params.CommandType
|
||||
response.AlarmUID = params.AlarmUID
|
||||
response.NumberOfRestarts = params.NumberOfRestarts
|
||||
response.InstructionSerial = params.InstructionSerial
|
||||
|
||||
return processFunc(pack, pack.pack, response)
|
||||
}
|
||||
|
||||
func (pack *SmartPackage) handleJson() (err error) {
|
||||
if pack.pack.PayloadType != protocol.PayloadTypeData {
|
||||
return fmt.Errorf("invalid json payload type")
|
||||
}
|
||||
|
||||
var nfErr *notFoundError
|
||||
if err = pack.handleAlarm(); err == nil || errors.As(err, &nfErr) {
|
||||
return
|
||||
}
|
||||
|
||||
var processFunc ProcessFunc
|
||||
var ok bool
|
||||
var key = fmt.Sprintf("%s:%s", pack.pack.Payload.Module, pack.pack.Payload.Operation)
|
||||
if processFunc, ok = pack.jsonProcess[key]; !ok {
|
||||
return ¬FoundError{
|
||||
message: fmt.Sprintf("json %s", key),
|
||||
}
|
||||
}
|
||||
|
||||
return processFunc(pack, pack.pack)
|
||||
}
|
||||
|
||||
func (pack *SmartPackage) handle() (err error) {
|
||||
var nfErr *notFoundError
|
||||
if err = pack.handleJson(); err == nil || errors.As(err, &nfErr) {
|
||||
return
|
||||
}
|
||||
|
||||
var processFunc ProcessFunc
|
||||
var ok bool
|
||||
if processFunc, ok = pack.payloadProcess[pack.pack.PayloadType]; !ok {
|
||||
return ¬FoundError{
|
||||
message: fmt.Sprintf("payload type %d", pack.pack.PayloadType),
|
||||
}
|
||||
}
|
||||
|
||||
return processFunc(pack, pack.pack)
|
||||
}
|
||||
|
||||
func (pack *SmartPackage) handleLoop() (err error) {
|
||||
for pack.pack.ReadPackage() {
|
||||
if err = pack.handle(); err != nil {
|
||||
return err
|
||||
}
|
||||
}
|
||||
|
||||
return
|
||||
}
|
||||
|
||||
func (pack *SmartPackage) Handle() (err error) {
|
||||
if err = pack.handleLoop(); err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
var rn int
|
||||
|
||||
if rn, err = pack.conn.Read(pack.buff); err != nil {
|
||||
return
|
||||
}
|
||||
|
||||
pack.pack.AddToAccum(pack.buff[:rn])
|
||||
|
||||
return pack.handleLoop()
|
||||
}
|
||||
|
||||
func (pack *SmartPackage) GetPackage() protocol.Package {
|
||||
return pack.pack
|
||||
}
|
||||
|
||||
func (pack *SmartPackage) Write(data []byte) (int, error) {
|
||||
return pack.conn.Write(data)
|
||||
}
|
||||
15
pkg/utils/crypto.go
Normal file
15
pkg/utils/crypto.go
Normal file
@ -0,0 +1,15 @@
|
||||
package utils
|
||||
|
||||
import (
|
||||
"crypto/hmac"
|
||||
"crypto/md5"
|
||||
"encoding/hex"
|
||||
)
|
||||
|
||||
func GenerateVerifyKey(key string) string {
|
||||
mac := hmac.New(md5.New, []byte(key))
|
||||
|
||||
mac.Write([]byte(key))
|
||||
|
||||
return hex.EncodeToString(mac.Sum(nil))
|
||||
}
|
||||
18
pkg/utils/random.go
Normal file
18
pkg/utils/random.go
Normal file
@ -0,0 +1,18 @@
|
||||
package utils
|
||||
|
||||
import (
|
||||
"math/rand"
|
||||
"time"
|
||||
)
|
||||
|
||||
// seededRand is a source of random numbers seeded with the current time.
|
||||
var seededRand *rand.Rand = rand.New(rand.NewSource(time.Now().UnixNano()))
|
||||
|
||||
// randomString generates a random string of the specified length using the given charset.
|
||||
func RandomString(length int, charset string) string {
|
||||
b := make([]byte, length)
|
||||
for i := range b {
|
||||
b[i] = charset[seededRand.Intn(len(charset))]
|
||||
}
|
||||
return string(b)
|
||||
}
|
||||
Reference in New Issue
Block a user