mirror of
https://github.com/renorris/openfsd
synced 2026-03-22 22:47:06 +08:00
63 lines
1.4 KiB
Go
63 lines
1.4 KiB
Go
package protocol
|
|
|
|
import (
|
|
"fmt"
|
|
"slices"
|
|
"strings"
|
|
)
|
|
|
|
type ClientQueryResponsePDU struct {
|
|
From string `validate:"required,alphanum,max=16"`
|
|
To string `validate:"required,max=16"`
|
|
QueryType string `validate:"required,ascii,min=2,max=7"`
|
|
Payload string `validate:""`
|
|
}
|
|
|
|
func (p *ClientQueryResponsePDU) Serialize() string {
|
|
if p.Payload == "" {
|
|
return fmt.Sprintf("$CR%s:%s:%s%s",
|
|
p.From, p.To, p.QueryType, PacketDelimiter)
|
|
} else {
|
|
return fmt.Sprintf("$CR%s:%s:%s:%s%s",
|
|
p.From, p.To, p.QueryType, p.Payload, PacketDelimiter)
|
|
}
|
|
}
|
|
|
|
func (p *ClientQueryResponsePDU) Parse(packet string) error {
|
|
packet = strings.TrimSuffix(packet, PacketDelimiter)
|
|
packet = strings.TrimPrefix(packet, "$CR")
|
|
|
|
// Extract fields
|
|
var fields []string
|
|
if fields = strings.SplitN(packet, Delimiter, 4); len(fields) < 3 {
|
|
return NewGenericFSDError(SyntaxError, "",
|
|
"invalid parameter count")
|
|
}
|
|
|
|
pdu := ClientQueryResponsePDU{
|
|
From: fields[0],
|
|
To: fields[1],
|
|
QueryType: fields[2],
|
|
}
|
|
|
|
if len(fields) == 4 {
|
|
pdu.Payload = fields[3]
|
|
}
|
|
|
|
if err := V.Struct(&pdu); err != nil {
|
|
if validatorErr := getFSDErrorFromValidatorErrors(err); err != nil {
|
|
return validatorErr
|
|
}
|
|
return err
|
|
}
|
|
|
|
if _, exists := slices.BinarySearch(supportedClientQueryTypes, pdu.QueryType); !exists {
|
|
return NewGenericFSDError(SyntaxError, fields[2], "invalid query type")
|
|
}
|
|
|
|
// Copy new pdu into receiver
|
|
*p = pdu
|
|
|
|
return nil
|
|
}
|