remove length property and make it calculated

This commit is contained in:
Joel Wetzell
2026-05-26 17:54:35 -05:00
parent 9546018fb7
commit 572fa39c16
2 changed files with 69 additions and 66 deletions
+30 -34
View File
@@ -19,70 +19,66 @@ type ArtCommand struct {
ProtVerHi uint8 ProtVerHi uint8
ProtVerLo uint8 ProtVerLo uint8
EstaMan uint16 EstaMan uint16
Length uint16
Data []uint8 Data []uint8
} }
func (ap *ArtCommand) GetOpCode() uint16 { func (ac *ArtCommand) GetOpCode() uint16 {
return ap.OpCode return ac.OpCode
} }
func (ap *ArtCommand) GetProtVer() uint16 { func (ac *ArtCommand) GetProtVer() uint16 {
return uint16(ap.ProtVerHi)<<8 + uint16(ap.ProtVerLo) return uint16(ac.ProtVerHi)<<8 + uint16(ac.ProtVerLo)
} }
func (ap *ArtCommand) GetID() [8]uint8 { func (ac *ArtCommand) GetID() [8]uint8 {
return ap.ID return ac.ID
} }
func (ap *ArtCommand) SetData(data string) error { func (ac *ArtCommand) Length() uint16 {
if len(data) > 512 { return uint16(len(ac.Data))
return errors.New("data length must be less than or equal to 512 bytes")
}
ap.Data = make([]uint8, len(data))
copy(ap.Data, data)
ap.Length = uint16(len(data))
return nil
} }
func (ap *ArtCommand) UnmarshalBinary(data []byte) error { func (ac *ArtCommand) UnmarshalBinary(data []byte) error {
if len(data) < 16 { if len(data) < 16 {
return errors.New("ArtCommand packet must be at least 16 bytes long") return errors.New("ArtCommand packet must be at least 16 bytes long")
} }
copy(ap.ID[:], data[0:8]) copy(ac.ID[:], data[0:8])
if !slices.Equal(ArtNetID[:], ap.ID[:]) { if !slices.Equal(ArtNetID[:], ac.ID[:]) {
return errors.New("ID does not match Art-Net ID") return errors.New("ID does not match Art-Net ID")
} }
ap.OpCode = binary.LittleEndian.Uint16(data[8:10]) ac.OpCode = binary.LittleEndian.Uint16(data[8:10])
ap.ProtVerHi = data[10] ac.ProtVerHi = data[10]
ap.ProtVerLo = data[11] ac.ProtVerLo = data[11]
offset := 12 offset := 12
ap.EstaMan = binary.LittleEndian.Uint16(data[offset : offset+2]) ac.EstaMan = binary.LittleEndian.Uint16(data[offset : offset+2])
ap.Length = binary.LittleEndian.Uint16(data[offset+2 : offset+4])
dataLength := int(ap.Length) dataLength := int(binary.LittleEndian.Uint16(data[offset+2 : offset+4]))
if len(data[offset+4:]) < dataLength { if len(data[offset+4:]) < dataLength {
return errors.New("[]byte length not long enough to contain data length specified in packet") return errors.New("[]byte length not long enough to contain data length specified in packet")
} }
ap.Data = make([]uint8, dataLength) ac.Data = make([]uint8, dataLength)
copy(ap.Data, data[offset+4:offset+4+dataLength]) copy(ac.Data, data[offset+4:offset+4+dataLength])
return nil return nil
} }
func (ap *ArtCommand) MarshalBinary() ([]byte, error) { func (ac *ArtCommand) MarshalBinary() ([]byte, error) {
data := make([]byte, 8+8) data := make([]byte, 8+8)
copy(data[0:8], ap.ID[:]) copy(data[0:8], ac.ID[:])
binary.LittleEndian.PutUint16(data[8:10], ap.OpCode) binary.LittleEndian.PutUint16(data[8:10], ac.OpCode)
data[10] = ap.ProtVerHi data[10] = ac.ProtVerHi
data[11] = ap.ProtVerLo data[11] = ac.ProtVerLo
binary.LittleEndian.PutUint16(data[12:14], ap.EstaMan) binary.LittleEndian.PutUint16(data[12:14], ac.EstaMan)
binary.LittleEndian.PutUint16(data[14:16], ap.Length) dataLength := uint16(len(ac.Data))
data = append(data, ap.Data...) if dataLength > 512 {
return nil, errors.New("data length must be less than or equal to 512 bytes")
}
binary.LittleEndian.PutUint16(data[14:16], dataLength)
data = append(data, ac.Data...)
return data, nil return data, nil
} }
+39 -32
View File
@@ -25,64 +25,71 @@ type ArtDiagData struct {
DiagPriority DiagPriority DiagPriority DiagPriority
LogicalPort uint8 LogicalPort uint8
filler3 uint8 filler3 uint8
Length uint16
Data []uint8 Data []uint8
} }
func (ap *ArtDiagData) GetOpCode() uint16 { func (add *ArtDiagData) GetOpCode() uint16 {
return ap.OpCode return add.OpCode
} }
func (ap *ArtDiagData) GetProtVer() uint16 { func (add *ArtDiagData) GetProtVer() uint16 {
return uint16(ap.ProtVerHi)<<8 + uint16(ap.ProtVerLo) return uint16(add.ProtVerHi)<<8 + uint16(add.ProtVerLo)
} }
func (ap *ArtDiagData) GetID() [8]uint8 { func (add *ArtDiagData) GetID() [8]uint8 {
return ap.ID return add.ID
} }
func (ap *ArtDiagData) UnmarshalBinary(data []byte) error { func (add *ArtDiagData) Length() uint16 {
return uint16(len(add.Data))
}
func (add *ArtDiagData) UnmarshalBinary(data []byte) error {
if len(data) < 18 { if len(data) < 18 {
return errors.New("ArtDiagData packet must be at least 18 bytes long") return errors.New("ArtDiagData packet must be at least 18 bytes long")
} }
copy(ap.ID[:], data[0:8]) copy(add.ID[:], data[0:8])
if !slices.Equal(ArtNetID[:], ap.ID[:]) { if !slices.Equal(ArtNetID[:], add.ID[:]) {
return errors.New("ID does not match Art-Net ID") return errors.New("ID does not match Art-Net ID")
} }
ap.OpCode = binary.LittleEndian.Uint16(data[8:10]) add.OpCode = binary.LittleEndian.Uint16(data[8:10])
ap.ProtVerHi = data[10] add.ProtVerHi = data[10]
ap.ProtVerLo = data[11] add.ProtVerLo = data[11]
offset := 12 offset := 12
ap.filler1 = data[offset] add.filler1 = data[offset]
ap.DiagPriority = DiagPriority(data[offset+1]) add.DiagPriority = DiagPriority(data[offset+1])
ap.LogicalPort = data[offset+2] add.LogicalPort = data[offset+2]
ap.filler3 = data[offset+3] add.filler3 = data[offset+3]
ap.Length = binary.LittleEndian.Uint16(data[offset+4 : offset+6]) dataLength := binary.LittleEndian.Uint16(data[offset+4 : offset+6])
if len(data[offset+6:]) < int(ap.Length) { if len(data[offset+6:]) < int(dataLength) {
return errors.New("[]byte length not long enough to contain data length specified in packet") return errors.New("[]byte length not long enough to contain data length specified in packet")
} }
ap.Data = make([]uint8, ap.Length) add.Data = make([]uint8, dataLength)
copy(ap.Data, data[offset+6:offset+6+int(ap.Length)]) copy(add.Data, data[offset+6:offset+6+int(dataLength)])
return nil return nil
} }
func (ap *ArtDiagData) MarshalBinary() ([]byte, error) { func (add *ArtDiagData) MarshalBinary() ([]byte, error) {
data := make([]byte, 8+10) data := make([]byte, 8+10)
copy(data[0:8], ap.ID[:]) copy(data[0:8], add.ID[:])
binary.LittleEndian.PutUint16(data[8:10], ap.OpCode) binary.LittleEndian.PutUint16(data[8:10], add.OpCode)
data[10] = ap.ProtVerHi data[10] = add.ProtVerHi
data[11] = ap.ProtVerLo data[11] = add.ProtVerLo
data[12] = ap.filler1 data[12] = add.filler1
data[13] = uint8(ap.DiagPriority) data[13] = uint8(add.DiagPriority)
data[14] = ap.LogicalPort data[14] = add.LogicalPort
data[15] = ap.filler3 data[15] = add.filler3
binary.LittleEndian.PutUint16(data[16:18], ap.Length) dataLength := uint16(len(add.Data))
data = append(data, ap.Data...) if dataLength > 512 {
return nil, errors.New("data length must be less than or equal to 512 bytes")
}
binary.LittleEndian.PutUint16(data[16:18], dataLength)
data = append(data, add.Data...)
return data, nil return data, nil
} }