+/*
+GoVPN -- simple secure free software virtual private network daemon
+Copyright (C) 2014-2016 Sergey Matveev <stargrave@stargrave.org>
+
+This program is free software: you can redistribute it and/or modify
+it under the terms of the GNU General Public License as published by
+the Free Software Foundation, either version 3 of the License, or
+(at your option) any later version.
+
+This program is distributed in the hope that it will be useful,
+but WITHOUT ANY WARRANTY; without even the implied warranty of
+MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+GNU General Public License for more details.
+
+You should have received a copy of the GNU General Public License
+along with this program. If not, see <http://www.gnu.org/licenses/>.
+*/
+
package govpn
import (
"testing"
+ "testing/quick"
"time"
)
var (
- peer *Peer
- plaintext []byte
- ciphertext []byte
- peerId *PeerId
- conf *PeerConf
+ testPeer *Peer
+ testPt []byte
+ testCt []byte
+ testPeerId PeerId
+ testConf *PeerConf
)
type Dummy struct {
}
func init() {
- MTU = 1500
- peerId, _ = IDDecode("ffffffffffffffffffffffffffffffff")
- conf = &PeerConf{
- Id: peerId,
- Timeout: time.Second * time.Duration(TimeoutDefault),
- NoiseEnable: false,
- CPR: 0,
+ id := new([IDSize]byte)
+ testPeerId = PeerId(*id)
+ testConf = &PeerConf{
+ Id: &testPeerId,
+ MTU: MTUDefault,
+ Timeout: time.Second * time.Duration(TimeoutDefault),
}
- peer = newPeer(true, "foo", Dummy{&ciphertext}, conf, new([SSize]byte))
- plaintext = make([]byte, 789)
+ testPeer = newPeer(true, "foo", Dummy{&testCt}, testConf, new([SSize]byte))
+ testPt = make([]byte, 789)
+}
+
+func TestTransportSymmetric(t *testing.T) {
+ peerd := newPeer(true, "foo", Dummy{nil}, testConf, new([SSize]byte))
+ f := func(payload []byte) bool {
+ if len(payload) == 0 {
+ return true
+ }
+ testPeer.EthProcess(payload)
+ return peerd.PktProcess(testCt, Dummy{nil}, true)
+ }
+ if err := quick.Check(f, nil); err != nil {
+ t.Error(err)
+ }
+}
+
+func TestTransportSymmetricNoise(t *testing.T) {
+ peerd := newPeer(true, "foo", Dummy{nil}, testConf, new([SSize]byte))
+ testPeer.NoiseEnable = true
+ peerd.NoiseEnable = true
+ f := func(payload []byte) bool {
+ if len(payload) == 0 {
+ return true
+ }
+ testPeer.EthProcess(payload)
+ return peerd.PktProcess(testCt, Dummy{nil}, true)
+ }
+ if err := quick.Check(f, nil); err != nil {
+ t.Error(err)
+ }
+ testPeer.NoiseEnable = true
+}
+
+func TestTransportSymmetricEncless(t *testing.T) {
+ peerd := newPeer(true, "foo", Dummy{nil}, testConf, new([SSize]byte))
+ testPeer.Encless = true
+ testPeer.NoiseEnable = true
+ peerd.Encless = true
+ peerd.NoiseEnable = true
+ f := func(payload []byte) bool {
+ if len(payload) == 0 {
+ return true
+ }
+ testPeer.EthProcess(payload)
+ return peerd.PktProcess(testCt, Dummy{nil}, true)
+ }
+ if err := quick.Check(f, nil); err != nil {
+ t.Error(err)
+ }
+ testPeer.NoiseEnable = false
+ testPeer.Encless = false
}
func BenchmarkEnc(b *testing.B) {
b.ResetTimer()
for i := 0; i < b.N; i++ {
- peer.EthProcess(plaintext)
+ testPeer.EthProcess(testPt)
}
}
func BenchmarkDec(b *testing.B) {
- peer = newPeer(true, "foo", Dummy{&ciphertext}, conf, new([SSize]byte))
- peer.EthProcess(plaintext)
- peer = newPeer(true, "foo", Dummy{nil}, conf, new([SSize]byte))
- orig := make([]byte, len(ciphertext))
- copy(orig, ciphertext)
+ testPeer = newPeer(true, "foo", Dummy{&testCt}, testConf, new([SSize]byte))
+ testPeer.EthProcess(testPt)
+ testPeer = newPeer(true, "foo", Dummy{nil}, testConf, new([SSize]byte))
+ orig := make([]byte, len(testCt))
+ copy(orig, testCt)
b.ResetTimer()
for i := 0; i < b.N; i++ {
- peer.nonceBucket0 = make(map[uint64]struct{}, 1)
- peer.nonceBucket1 = make(map[uint64]struct{}, 1)
- copy(ciphertext, orig)
- if !peer.PktProcess(ciphertext, Dummy{nil}, true) {
+ testPeer.nonceBucket0 = make(map[uint64]struct{}, 1)
+ testPeer.nonceBucket1 = make(map[uint64]struct{}, 1)
+ copy(testCt, orig)
+ if !testPeer.PktProcess(testCt, Dummy{nil}, true) {
b.Fail()
}
}
}
+
+func TestTransportBigger(t *testing.T) {
+ tmp := make([]byte, MTUMax*4)
+ Rand.Read(tmp)
+ testPeer.PktProcess(tmp, Dummy{nil}, true)
+}