Add packet padding container for packet length shaping: refactor arg names

This commit is contained in:
Shelikhoo 2025-01-28 14:19:20 +00:00
parent 1689279e95
commit fbcb9bc863
No known key found for this signature in database
GPG key ID: 4C9764E9FE80A3DC
3 changed files with 26 additions and 26 deletions

View file

@ -9,18 +9,18 @@ func New() PacketPaddingContainer {
type packetPaddingContainer struct {
}
func (c packetPaddingContainer) Pack(data_OWNERSHIP_RELINQUISHED []byte, padding int) []byte {
data := append(data_OWNERSHIP_RELINQUISHED, make([]byte, padding)...)
func (c packetPaddingContainer) Pack(data_OWNERSHIP_RELINQUISHED []byte, paddingLength int) []byte {
data := append(data_OWNERSHIP_RELINQUISHED, make([]byte, paddingLength)...)
dataLength := len(data_OWNERSHIP_RELINQUISHED)
data = binary.BigEndian.AppendUint16(data, uint16(dataLength))
return data
}
func (c packetPaddingContainer) Pad(padding int) []byte {
if assertPaddingLengthIsNotNegative := padding < 0; assertPaddingLengthIsNotNegative {
func (c packetPaddingContainer) Pad(paddingLength int) []byte {
if assertPaddingLengthIsNotNegative := paddingLength < 0; assertPaddingLengthIsNotNegative {
return nil
}
switch padding {
switch paddingLength {
case 0:
return []byte{}
case 1:
@ -28,7 +28,7 @@ func (c packetPaddingContainer) Pad(padding int) []byte {
case 2:
return []byte{0, 0}
default:
return append(make([]byte, padding-2), byte(padding>>8), byte(padding))
return append(make([]byte, paddingLength-2), byte(paddingLength>>8), byte(paddingLength))
}
}