mirror of
https://github.com/pion/stun.git
synced 2025-09-26 20:01:18 +08:00
74 lines
1.7 KiB
Go
74 lines
1.7 KiB
Go
// SPDX-FileCopyrightText: 2023 The Pion community <https://pion.ly>
|
|
// SPDX-License-Identifier: MIT
|
|
|
|
//go:build !js
|
|
// +build !js
|
|
|
|
package stun
|
|
|
|
import (
|
|
"net"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func BenchmarkFingerprint_AddTo(b *testing.B) {
|
|
b.ReportAllocs()
|
|
msg := new(Message)
|
|
s := NewSoftware("software")
|
|
addr := &XORMappedAddress{
|
|
IP: net.IPv4(213, 1, 223, 5),
|
|
}
|
|
addAttr(b, msg, addr)
|
|
addAttr(b, msg, s)
|
|
b.SetBytes(int64(len(msg.Raw)))
|
|
for i := 0; i < b.N; i++ {
|
|
Fingerprint.AddTo(msg) //nolint:errcheck,gosec
|
|
msg.WriteLength()
|
|
msg.Length -= attributeHeaderSize + fingerprintSize
|
|
msg.Raw = msg.Raw[:msg.Length+messageHeaderSize]
|
|
msg.Attributes = msg.Attributes[:len(msg.Attributes)-1]
|
|
}
|
|
}
|
|
|
|
func TestFingerprint_Check(t *testing.T) {
|
|
m := new(Message)
|
|
addAttr(t, m, NewSoftware("software"))
|
|
m.WriteHeader()
|
|
Fingerprint.AddTo(m) //nolint:errcheck,gosec
|
|
m.WriteHeader()
|
|
assert.NoError(t, Fingerprint.Check(m))
|
|
m.Raw[3]++
|
|
assert.Error(t, Fingerprint.Check(m))
|
|
}
|
|
|
|
func TestFingerprint_CheckBad(t *testing.T) {
|
|
m := new(Message)
|
|
addAttr(t, m, NewSoftware("software"))
|
|
m.WriteHeader()
|
|
assert.Error(t, Fingerprint.Check(m))
|
|
m.Add(AttrFingerprint, []byte{1, 2, 3})
|
|
assert.True(t, IsAttrSizeInvalid(Fingerprint.Check(m)))
|
|
}
|
|
|
|
func BenchmarkFingerprint_Check(b *testing.B) {
|
|
b.ReportAllocs()
|
|
m := new(Message)
|
|
s := NewSoftware("software")
|
|
addr := &XORMappedAddress{
|
|
IP: net.IPv4(213, 1, 223, 5),
|
|
}
|
|
addAttr(b, m, addr)
|
|
addAttr(b, m, s)
|
|
m.WriteHeader()
|
|
Fingerprint.AddTo(m) //nolint:errcheck,gosec
|
|
m.WriteHeader()
|
|
b.SetBytes(int64(len(m.Raw)))
|
|
for i := 0; i < b.N; i++ {
|
|
if err := Fingerprint.Check(m); err != nil {
|
|
b.Fatal(err)
|
|
}
|
|
}
|
|
}
|