63 lines
1.2 KiB
Go
63 lines
1.2 KiB
Go
package packets
|
|
|
|
import (
|
|
"bufio"
|
|
"errors"
|
|
"io"
|
|
|
|
"badat.dev/maeqtt/v2/mqtt/properties"
|
|
"badat.dev/maeqtt/v2/mqtt/types"
|
|
)
|
|
|
|
type PublishPacket struct {
|
|
Dup bool
|
|
Retain bool
|
|
QOSLevel byte
|
|
TopicName string
|
|
Payload []byte
|
|
PacketId *uint16
|
|
Properties properties.PublishPacketProperties
|
|
}
|
|
|
|
func (p PublishPacket) Visit(v PacketVisitor) {
|
|
v.VisitPublish(p)
|
|
}
|
|
|
|
func parsePublishPacket(control controlPacket) (PublishPacket, error) {
|
|
var err error
|
|
r := bufio.NewReader(control.reader)
|
|
packet := PublishPacket{}
|
|
|
|
if control.packetType != PacketTypePublish {
|
|
return packet, errors.New("Wrong packet type for parseConnectPacket")
|
|
}
|
|
|
|
packet.Retain = control.flags&1 == 1
|
|
packet.QOSLevel = byte((control.flags >> 1) & 0b11)
|
|
packet.Dup = (control.flags>>3)&1 == 0
|
|
|
|
packet.TopicName, err = types.DecodeUTF8String(r)
|
|
if err != nil {
|
|
return packet, err
|
|
}
|
|
|
|
if packet.QOSLevel != 0 {
|
|
packId, err := types.DecodeUint16(r)
|
|
if err != nil {
|
|
return packet, err
|
|
}
|
|
packet.PacketId = &packId
|
|
}
|
|
err = properties.ParseProperties(r, packet.Properties.ArrayOf())
|
|
if err != nil {
|
|
return packet, err
|
|
}
|
|
|
|
packet.Payload, err = io.ReadAll(r)
|
|
if err != nil {
|
|
return packet, err
|
|
}
|
|
|
|
return packet, nil
|
|
}
|