Rework encoding to use go binary

This commit is contained in:
PhatPhuckDave
2024-07-22 21:56:09 +02:00
parent 0ac3f31c3f
commit 8ebca73181
2 changed files with 22 additions and 33 deletions

View File

@@ -1,10 +1,13 @@
package pdu
import "encoding/binary"
type (
PDU interface {
EncodeInto(*[]byte)
Encode() []byte
Decode([]byte)
EncodeInto(*[]uint8)
Encode() []uint8
Decode([]uint8)
Size() uint32
}
PDU_HEADER struct {
@@ -19,24 +22,24 @@ type (
}
)
func (p *PDU_HEADER) Encode() []byte {
buf := make([]byte, 16)
EncodeUint32(buf[0:4], p.command_length)
EncodeUint32(buf[4:8], p.command_id)
EncodeUint32(buf[8:12], p.command_status)
EncodeUint32(buf[12:16], p.sequence_number)
func (p *PDU_HEADER) Encode() []uint8 {
buf := make([]uint8, 16)
binary.BigEndian.PutUint32(buf[0:4], p.command_length)
binary.BigEndian.PutUint32(buf[4:8], p.command_id)
binary.BigEndian.PutUint32(buf[8:12], p.command_status)
binary.BigEndian.PutUint32(buf[12:16], p.sequence_number)
return buf
}
func (p *PDU_HEADER) EncodeInto(buf *[]byte) {
func (p *PDU_HEADER) EncodeInto(buf *[]uint8) {
bufVal := *buf
EncodeUint32(bufVal[0:4], p.command_length)
EncodeUint32(bufVal[4:8], p.command_id)
EncodeUint32(bufVal[8:12], p.command_status)
EncodeUint32(bufVal[12:16], p.sequence_number)
binary.BigEndian.PutUint32(bufVal[0:4], p.command_length)
binary.BigEndian.PutUint32(bufVal[4:8], p.command_id)
binary.BigEndian.PutUint32(bufVal[8:12], p.command_status)
binary.BigEndian.PutUint32(bufVal[12:16], p.sequence_number)
}
func (p *PDU_HEADER) Decode(data []byte) {
p.command_length = DecodeUint32(data[0:4])
p.command_id = DecodeUint32(data[4:8])
p.command_status = DecodeUint32(data[8:12])
p.sequence_number = DecodeUint32(data[12:16])
func (p *PDU_HEADER) Decode(data []uint8) {
p.command_length = binary.BigEndian.Uint32(data[0:4])
p.command_id = binary.BigEndian.Uint32(data[4:8])
p.command_status = binary.BigEndian.Uint32(data[8:12])
p.sequence_number = binary.BigEndian.Uint32(data[12:16])
}