Hunter0x7c7
2022-08-11 b8230139fb40edea387617b6accd8371e37eda58
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
package aead
 
import (
    "bytes"
    "fmt"
    "io"
    "testing"
 
    "github.com/stretchr/testify/assert"
)
 
func TestOpenVMessAEADHeader(t *testing.T) {
    TestHeader := []byte("Test Header")
    key := KDF16([]byte("Demo Key for Auth ID Test"), "Demo Path for Auth ID Test")
    var keyw [16]byte
    copy(keyw[:], key)
    sealed := SealVMessAEADHeader(keyw, TestHeader)
 
    AEADR := bytes.NewReader(sealed)
 
    var authid [16]byte
 
    io.ReadFull(AEADR, authid[:])
 
    out, _, _, err := OpenVMessAEADHeader(keyw, authid, AEADR)
 
    fmt.Println(string(out))
    fmt.Println(err)
}
 
func TestOpenVMessAEADHeader2(t *testing.T) {
    TestHeader := []byte("Test Header")
    key := KDF16([]byte("Demo Key for Auth ID Test"), "Demo Path for Auth ID Test")
    var keyw [16]byte
    copy(keyw[:], key)
    sealed := SealVMessAEADHeader(keyw, TestHeader)
 
    AEADR := bytes.NewReader(sealed)
 
    var authid [16]byte
 
    io.ReadFull(AEADR, authid[:])
 
    out, _, readen, err := OpenVMessAEADHeader(keyw, authid, AEADR)
    assert.Equal(t, len(sealed)-16-AEADR.Len(), readen)
    assert.Equal(t, string(TestHeader), string(out))
    assert.Nil(t, err)
}
 
func TestOpenVMessAEADHeader4(t *testing.T) {
    for i := 0; i <= 60; i++ {
        TestHeader := []byte("Test Header")
        key := KDF16([]byte("Demo Key for Auth ID Test"), "Demo Path for Auth ID Test")
        var keyw [16]byte
        copy(keyw[:], key)
        sealed := SealVMessAEADHeader(keyw, TestHeader)
        var sealedm [16]byte
        copy(sealedm[:], sealed)
        sealed[i] ^= 0xff
        AEADR := bytes.NewReader(sealed)
 
        var authid [16]byte
 
        io.ReadFull(AEADR, authid[:])
 
        out, drain, readen, err := OpenVMessAEADHeader(keyw, authid, AEADR)
        assert.Equal(t, len(sealed)-16-AEADR.Len(), readen)
        assert.Equal(t, true, drain)
        assert.NotNil(t, err)
        if err == nil {
            fmt.Println(">")
        }
        assert.Nil(t, out)
    }
}
 
func TestOpenVMessAEADHeader4Massive(t *testing.T) {
    for j := 0; j < 1000; j++ {
        for i := 0; i <= 60; i++ {
            TestHeader := []byte("Test Header")
            key := KDF16([]byte("Demo Key for Auth ID Test"), "Demo Path for Auth ID Test")
            var keyw [16]byte
            copy(keyw[:], key)
            sealed := SealVMessAEADHeader(keyw, TestHeader)
            var sealedm [16]byte
            copy(sealedm[:], sealed)
            sealed[i] ^= 0xff
            AEADR := bytes.NewReader(sealed)
 
            var authid [16]byte
 
            io.ReadFull(AEADR, authid[:])
 
            out, drain, readen, err := OpenVMessAEADHeader(keyw, authid, AEADR)
            assert.Equal(t, len(sealed)-16-AEADR.Len(), readen)
            assert.Equal(t, true, drain)
            assert.NotNil(t, err)
            if err == nil {
                fmt.Println(">")
            }
            assert.Nil(t, out)
        }
    }
}