Files
mautrix-telegram/pkg/gotd/mtproxy/faketls/server_hello_test.go
T
2025-06-27 20:03:37 -07:00

58 lines
1.4 KiB
Go

package faketls
import (
"bytes"
"encoding/hex"
"os"
"path/filepath"
"strings"
"testing"
"github.com/stretchr/testify/require"
)
func Test_readServerHello(t *testing.T) {
p := filepath.Join("testdata", "server_hello")
entries, err := os.ReadDir(p)
if err != nil {
t.Fatal(err)
}
clientRandom := map[string][32]uint8{
"alexbers.hex": {
0xa1, 0x32, 0xe3, 0x91, 0x60, 0x83, 0xb3, 0x14, 0xc1, 0xb9, 0x74, 0xd0, 0x57, 0x85, 0xe8, 0xee,
0x70, 0x45, 0x6e, 0x5f, 0x86, 0x6d, 0x96, 0x57, 0xd5, 0x0a, 0x5c, 0x08, 0xea, 0x38, 0x31, 0x8e,
},
"mtgv2.hex": {
0xca, 0x36, 0xbb, 0xa8, 0x33, 0x80, 0x9a, 0x33, 0xaa, 0x62, 0x7e, 0xbb, 0x5a, 0x32, 0xa1, 0x01,
0x02, 0xd1, 0xa6, 0x1e, 0x1e, 0x6c, 0x58, 0xa4, 0x61, 0xd9, 0x34, 0x57, 0x4d, 0x2e, 0x2e, 0xa3,
},
}
secret := []uint8{
0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x01,
}
for _, entry := range entries {
if entry.IsDir() {
t.Error("Unexpected directory in testdata")
continue
}
fileName := entry.Name()
t.Run(fileName, func(t *testing.T) {
a := require.New(t)
data, err := os.ReadFile(filepath.Join(p, fileName))
a.NoError(err)
decode, err := hex.DecodeString(strings.TrimSpace(string(data)))
a.NoError(err)
r := bytes.NewReader(decode)
a.NoError(readServerHello(r, clientRandom[entry.Name()], secret))
a.Zero(r.Len(), "should read all data")
})
}
}