mirror of
https://github.com/gortc/stun.git
synced 2024-05-03 14:44:55 +08:00
190 lines
5.0 KiB
Go
190 lines
5.0 KiB
Go
package stun
|
|
|
|
import (
|
|
"net"
|
|
"testing"
|
|
)
|
|
|
|
func TestRFC5769(t *testing.T) {
|
|
// Test Vectors for Session Traversal Utilities for NAT (STUN)
|
|
// see https://tools.ietf.org/html/rfc5769
|
|
t.Run("Request", func(t *testing.T) {
|
|
m := &Message{
|
|
Raw: []byte("\x00\x01\x00\x58" +
|
|
"\x21\x12\xa4\x42" +
|
|
"\xb7\xe7\xa7\x01\xbc\x34\xd6\x86\xfa\x87\xdf\xae" +
|
|
"\x80\x22\x00\x10" +
|
|
"STUN test client" +
|
|
"\x00\x24\x00\x04" +
|
|
"\x6e\x00\x01\xff" +
|
|
"\x80\x29\x00\x08" +
|
|
"\x93\x2f\xf9\xb1\x51\x26\x3b\x36" +
|
|
"\x00\x06\x00\x09" +
|
|
"\x65\x76\x74\x6a\x3a\x68\x36\x76\x59\x20\x20\x20" +
|
|
"\x00\x08\x00\x14" +
|
|
"\x9a\xea\xa7\x0c\xbf\xd8\xcb\x56\x78\x1e\xf2\xb5" +
|
|
"\xb2\xd3\xf2\x49\xc1\xb5\x71\xa2" +
|
|
"\x80\x28\x00\x04" +
|
|
"\xe5\x7a\x3b\xcf",
|
|
),
|
|
}
|
|
if err := m.Decode(); err != nil {
|
|
t.Error(err)
|
|
}
|
|
software := new(Software)
|
|
if err := software.GetFrom(m); err != nil {
|
|
t.Error(err)
|
|
}
|
|
if software.String() != "STUN test client" {
|
|
t.Error("bad software: ", software)
|
|
}
|
|
if err := Fingerprint.Check(m); err != nil {
|
|
t.Error("check failed: ", err)
|
|
}
|
|
t.Run("Long-Term credentials", func(t *testing.T) {
|
|
m := &Message{
|
|
Raw: []byte("\x00\x01\x00\x60" +
|
|
"\x21\x12\xa4\x42" +
|
|
"\x78\xad\x34\x33\xc6\xad\x72\xc0\x29\xda\x41\x2e" +
|
|
"\x00\x06\x00\x12" +
|
|
"\xe3\x83\x9e\xe3\x83\x88\xe3\x83\xaa\xe3\x83\x83" +
|
|
"\xe3\x82\xaf\xe3\x82\xb9\x00\x00" +
|
|
"\x00\x15\x00\x1c" +
|
|
"\x66\x2f\x2f\x34\x39\x39\x6b\x39\x35\x34\x64\x36" +
|
|
"\x4f\x4c\x33\x34\x6f\x4c\x39\x46\x53\x54\x76\x79" +
|
|
"\x36\x34\x73\x41" +
|
|
"\x00\x14\x00\x0b" +
|
|
"\x65\x78\x61\x6d\x70\x6c\x65\x2e\x6f\x72\x67\x00" +
|
|
"\x00\x08\x00\x14" +
|
|
"\xf6\x70\x24\x65\x6d\xd6\x4a\x3e\x02\xb8\xe0\x71" +
|
|
"\x2e\x85\xc9\xa2\x8c\xa8\x96\x66",
|
|
),
|
|
}
|
|
if err := m.Decode(); err != nil {
|
|
t.Error(err)
|
|
}
|
|
u := new(Username)
|
|
if err := u.GetFrom(m); err != nil {
|
|
t.Error(err)
|
|
}
|
|
expectedUsername := "\u30DE\u30C8\u30EA\u30C3\u30AF\u30B9"
|
|
if u.String() != expectedUsername {
|
|
t.Errorf("username: %q (got) != %q (exp)", u, expectedUsername)
|
|
}
|
|
n := new(Nonce)
|
|
if err := n.GetFrom(m); err != nil {
|
|
t.Error(err)
|
|
}
|
|
if n.String() != "f//499k954d6OL34oL9FSTvy64sA" {
|
|
t.Error("bad nonce")
|
|
}
|
|
r := new(Realm)
|
|
if err := r.GetFrom(m); err != nil {
|
|
t.Error(err)
|
|
}
|
|
if r.String() != "example.org" {
|
|
t.Error("bad realm")
|
|
}
|
|
// checking HMAC
|
|
i := NewLongTermIntegrity(
|
|
"\u30DE\u30C8\u30EA\u30C3\u30AF\u30B9",
|
|
"example.org",
|
|
"TheMatrIX",
|
|
)
|
|
if err := i.Check(m); err != nil {
|
|
t.Error(err)
|
|
}
|
|
})
|
|
})
|
|
t.Run("Response", func(t *testing.T) {
|
|
t.Run("IPv4", func(t *testing.T) {
|
|
m := &Message{
|
|
Raw: []byte("\x01\x01\x00\x3c" +
|
|
"\x21\x12\xa4\x42" +
|
|
"\xb7\xe7\xa7\x01\xbc\x34\xd6\x86\xfa\x87\xdf\xae" +
|
|
"\x80\x22\x00\x0b" +
|
|
"\x74\x65\x73\x74\x20\x76\x65\x63\x74\x6f\x72\x20" +
|
|
"\x00\x20\x00\x08" +
|
|
"\x00\x01\xa1\x47\xe1\x12\xa6\x43" +
|
|
"\x00\x08\x00\x14" +
|
|
"\x2b\x91\xf5\x99\xfd\x9e\x90\xc3\x8c\x74\x89\xf9" +
|
|
"\x2a\xf9\xba\x53\xf0\x6b\xe7\xd7" +
|
|
"\x80\x28\x00\x04" +
|
|
"\xc0\x7d\x4c\x96",
|
|
),
|
|
}
|
|
if err := m.Decode(); err != nil {
|
|
t.Error(err)
|
|
}
|
|
software := new(Software)
|
|
if err := software.GetFrom(m); err != nil {
|
|
t.Error(err)
|
|
}
|
|
if software.String() != "test vector" {
|
|
t.Error("bad software: ", software)
|
|
}
|
|
if err := Fingerprint.Check(m); err != nil {
|
|
t.Error("Check failed: ", err)
|
|
}
|
|
addr := new(XORMappedAddress)
|
|
if err := addr.GetFrom(m); err != nil {
|
|
t.Error(err)
|
|
}
|
|
if !addr.IP.Equal(net.ParseIP("192.0.2.1")) {
|
|
t.Error("bad IP")
|
|
}
|
|
if addr.Port != 32853 {
|
|
t.Error("bad Port")
|
|
}
|
|
if err := Fingerprint.Check(m); err != nil {
|
|
t.Error("check failed: ", err)
|
|
}
|
|
})
|
|
t.Run("IPv6", func(t *testing.T) {
|
|
m := &Message{
|
|
Raw: []byte("\x01\x01\x00\x48" +
|
|
"\x21\x12\xa4\x42" +
|
|
"\xb7\xe7\xa7\x01\xbc\x34\xd6\x86\xfa\x87\xdf\xae" +
|
|
"\x80\x22\x00\x0b" +
|
|
"\x74\x65\x73\x74\x20\x76\x65\x63\x74\x6f\x72\x20" +
|
|
"\x00\x20\x00\x14" +
|
|
"\x00\x02\xa1\x47" +
|
|
"\x01\x13\xa9\xfa\xa5\xd3\xf1\x79" +
|
|
"\xbc\x25\xf4\xb5\xbe\xd2\xb9\xd9" +
|
|
"\x00\x08\x00\x14" +
|
|
"\xa3\x82\x95\x4e\x4b\xe6\x7b\xf1\x17\x84\xc9\x7c" +
|
|
"\x82\x92\xc2\x75\xbf\xe3\xed\x41" +
|
|
"\x80\x28\x00\x04" +
|
|
"\xc8\xfb\x0b\x4c",
|
|
),
|
|
}
|
|
if err := m.Decode(); err != nil {
|
|
t.Error(err)
|
|
}
|
|
software := new(Software)
|
|
if err := software.GetFrom(m); err != nil {
|
|
t.Error(err)
|
|
}
|
|
if software.String() != "test vector" {
|
|
t.Error("bad software: ", software)
|
|
}
|
|
if err := Fingerprint.Check(m); err != nil {
|
|
t.Error("Check failed: ", err)
|
|
}
|
|
addr := new(XORMappedAddress)
|
|
if err := addr.GetFrom(m); err != nil {
|
|
t.Error(err)
|
|
}
|
|
if !addr.IP.Equal(net.ParseIP("2001:db8:1234:5678:11:2233:4455:6677")) {
|
|
t.Error("bad IP")
|
|
}
|
|
if addr.Port != 32853 {
|
|
t.Error("bad Port")
|
|
}
|
|
if err := Fingerprint.Check(m); err != nil {
|
|
t.Error("check failed: ", err)
|
|
}
|
|
})
|
|
})
|
|
}
|