zishuo/stun
1
0
Fork 0
mirror of https://github.com/pion/stun.git synced 2025-09-27 04:05:57 +08:00
Code Issues Packages Projects Releases Wiki Activity

a:integrity: fix case with fingerprint

Browse Source
This commit is contained in:
Aleksandr Razumov
2017-02-21 03:41:26 +03:00
parent 0c49de1e6f
commit d11678fcfd
2 changed files with 33 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
integrity_test.go
View File

@@ -24,6 +24,7 @@ func TestMessageIntegrity_AddTo_Simple(t *testing.T) {
if err := i.AddTo(m); err != nil {
t.Error(err)
}
NewSoftware("software").AddTo(m)
m.WriteHeader()
dM := new(Message)
dM.Raw = m.Raw
@@ -33,13 +34,30 @@ func TestMessageIntegrity_AddTo_Simple(t *testing.T) {
if err := i.Check(dM); err != nil {
t.Error(err)
}
m.Raw[3] = m.Raw[3] + 12 // HMAC now invalid
dM.Raw[24] += 12 // HMAC now invalid
if err, ok := i.Check(dM).(*IntegrityErr); !ok {
t.Error(err, "should be *IntegrityErr")
}
})
}
func TestMessageIntegrityWithFingerprint(t *testing.T) {
m := new(Message)
m.TransactionID = [transactionIDSize]byte{1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11}
m.WriteHeader()
NewSoftware("software").AddTo(m)
i := NewShortTermIntegrity("pwd")
if err := i.AddTo(m); err != nil {
t.Fatal(err)
}
if err := Fingerprint.AddTo(m); err != nil {
t.Fatal(err)
}
if err := i.Check(m); err != nil {
t.Fatal(err)
}
}
func TestMessageIntegrity(t *testing.T) {
m := new(Message)
//NewSoftware("software")