Add packet padding container for packet length shaping

This commit is contained in:
Shelikhoo 2025-01-28 14:03:59 +00:00
parent c1ac2aa577
commit 9e1cc35878
No known key found for this signature in database
GPG key ID: 4C9764E9FE80A3DC
3 changed files with 194 additions and 0 deletions

View file

@ -0,0 +1,113 @@
package packetPaddingContainer_test
import (
"testing"
. "github.com/smartystreets/goconvey/convey"
"gitlab.torproject.org/tpo/anti-censorship/pluggable-transports/snowflake/v2/common/packetPaddingContainer"
)
func TestPacketPaddingContainer(t *testing.T) {
Convey("Given a PacketPaddingContainer", t, func() {
container := packetPaddingContainer.New()
Convey("When packing data with padding", func() {
data := []byte("testdata")
padding := 4
packedData := container.Pack(data, padding)
Convey("The packed data should have the correct length", func() {
expectedLength := len(data) + padding + 2
So(len(packedData), ShouldEqual, expectedLength)
})
Convey("When unpacking the packed data", func() {
unpackedData, unpackedPadding := container.Unpack(packedData)
Convey("The unpacked data should match the original data", func() {
So(string(unpackedData), ShouldEqual, string(data))
})
Convey("The unpacked padding should match the original padding", func() {
So(unpackedPadding, ShouldEqual, padding)
})
})
})
Convey("When packing empty data with padding", func() {
data := []byte("")
padding := 4
packedData := container.Pack(data, padding)
Convey("The packed data should have the correct length", func() {
expectedLength := len(data) + padding + 2
So(len(packedData), ShouldEqual, expectedLength)
})
Convey("When unpacking the packed data", func() {
unpackedData, unpackedPadding := container.Unpack(packedData)
Convey("The unpacked data should match the original data", func() {
So(string(unpackedData), ShouldEqual, string(data))
})
Convey("The unpacked padding should match the original padding", func() {
So(unpackedPadding, ShouldEqual, padding)
})
})
})
Convey("When packing data with zero padding", func() {
data := []byte("testdata")
padding := 0
packedData := container.Pack(data, padding)
Convey("The packed data should have the correct length", func() {
expectedLength := len(data) + padding + 2
So(len(packedData), ShouldEqual, expectedLength)
})
Convey("When unpacking the packed data", func() {
unpackedData, unpackedPadding := container.Unpack(packedData)
Convey("The unpacked data should match the original data", func() {
So(string(unpackedData), ShouldEqual, string(data))
})
Convey("The unpacked padding should match the original padding", func() {
So(unpackedPadding, ShouldEqual, padding)
})
})
})
Convey("When padding data", func() {
Convey("With a positive padding length", func() {
padLength := 3
padData := container.Pad(padLength)
Convey("The padded data should have the correct length", func() {
So(len(padData), ShouldEqual, padLength)
})
})
Convey("With a zero padding length", func() {
padLength := 0
padData := container.Pad(padLength)
Convey("The padded data should be empty", func() {
So(len(padData), ShouldEqual, 0)
})
})
Convey("With a negative padding length", func() {
padLength := -1
padData := container.Pad(padLength)
Convey("The padded data should be nil", func() {
So(padData, ShouldBeNil)
})
})
})
})
}