71 lines
2.0 KiB
Go
Raw Normal View History

2022-03-10 10:44:48 +01:00
package sctp
import (
"encoding/binary"
"encoding/hex"
"errors"
"fmt"
)
type paramHeader struct {
typ paramType
len int
raw []byte
}
const (
paramHeaderLength = 4
)
2024-05-15 19:15:00 -04:00
// Parameter header parse errors
2022-03-10 10:44:48 +01:00
var (
2024-05-15 19:15:00 -04:00
ErrParamHeaderTooShort = errors.New("param header too short")
ErrParamHeaderSelfReportedLengthShorter = errors.New("param self reported length is shorter than header length")
ErrParamHeaderSelfReportedLengthLonger = errors.New("param self reported length is longer than header length")
ErrParamHeaderParseFailed = errors.New("failed to parse param type")
2022-03-10 10:44:48 +01:00
)
func (p *paramHeader) marshal() ([]byte, error) {
paramLengthPlusHeader := paramHeaderLength + len(p.raw)
rawParam := make([]byte, paramLengthPlusHeader)
binary.BigEndian.PutUint16(rawParam[0:], uint16(p.typ))
binary.BigEndian.PutUint16(rawParam[2:], uint16(paramLengthPlusHeader))
copy(rawParam[paramHeaderLength:], p.raw)
return rawParam, nil
}
func (p *paramHeader) unmarshal(raw []byte) error {
if len(raw) < paramHeaderLength {
2024-05-15 19:15:00 -04:00
return ErrParamHeaderTooShort
2022-03-10 10:44:48 +01:00
}
paramLengthPlusHeader := binary.BigEndian.Uint16(raw[2:])
if int(paramLengthPlusHeader) < paramHeaderLength {
2024-05-15 19:15:00 -04:00
return fmt.Errorf("%w: param self reported length (%d) shorter than header length (%d)", ErrParamHeaderSelfReportedLengthShorter, int(paramLengthPlusHeader), paramHeaderLength)
2022-03-10 10:44:48 +01:00
}
if len(raw) < int(paramLengthPlusHeader) {
2024-05-15 19:15:00 -04:00
return fmt.Errorf("%w: param length (%d) shorter than its self reported length (%d)", ErrParamHeaderSelfReportedLengthLonger, len(raw), int(paramLengthPlusHeader))
2022-03-10 10:44:48 +01:00
}
typ, err := parseParamType(raw[0:])
if err != nil {
2024-05-15 19:15:00 -04:00
return fmt.Errorf("%w: %v", ErrParamHeaderParseFailed, err) //nolint:errorlint
2022-03-10 10:44:48 +01:00
}
p.typ = typ
p.raw = raw[paramHeaderLength:paramLengthPlusHeader]
p.len = int(paramLengthPlusHeader)
return nil
}
func (p *paramHeader) length() int {
return p.len
}
// String makes paramHeader printable
func (p paramHeader) String() string {
return fmt.Sprintf("%s (%d): %s", p.typ, p.len, hex.Dump(p.raw))
}