src/govpn/peer_test.go

   1 /*
   2 GoVPN -- simple secure free software virtual private network daemon
   3 Copyright (C) 2014-2016 Sergey Matveev <stargrave@stargrave.org>
   4
   5 This program is free software: you can redistribute it and/or modify
   6 it under the terms of the GNU General Public License as published by
   7 the Free Software Foundation, either version 3 of the License, or
   8 (at your option) any later version.
   9
  10 This program is distributed in the hope that it will be useful,
  11 but WITHOUT ANY WARRANTY; without even the implied warranty of
  12 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13 GNU General Public License for more details.
  14
  15 You should have received a copy of the GNU General Public License
  16 along with this program.  If not, see <http://www.gnu.org/licenses/>.
  17 */
  18
  19 package govpn
  20
  21 import (
  22         "testing"
  23         "time"
  24 )
  25
  26 var (
  27         peer       *Peer
  28         plaintext  []byte
  29         ciphertext []byte
  30         peerId     *PeerId
  31         conf       *PeerConf
  32 )
  33
  34 type Dummy struct {
  35         dst *[]byte
  36 }
  37
  38 func (d Dummy) Write(b []byte) (int, error) {
  39         if d.dst != nil {
  40                 *d.dst = b
  41         }
  42         return len(b), nil
  43 }
  44
  45 func init() {
  46         MTU = 1500
  47         id := new([IDSize]byte)
  48         peerId := PeerId(*id)
  49         conf = &PeerConf{
  50                 Id:      &peerId,
  51                 Timeout: time.Second * time.Duration(TimeoutDefault),
  52                 Noise:   false,
  53                 CPR:     0,
  54         }
  55         peer = newPeer(true, "foo", Dummy{&ciphertext}, conf, new([SSize]byte))
  56         plaintext = make([]byte, 789)
  57 }
  58
  59 func BenchmarkEnc(b *testing.B) {
  60         b.ResetTimer()
  61         for i := 0; i < b.N; i++ {
  62                 peer.EthProcess(plaintext)
  63         }
  64 }
  65
  66 func BenchmarkDec(b *testing.B) {
  67         peer = newPeer(true, "foo", Dummy{&ciphertext}, conf, new([SSize]byte))
  68         peer.EthProcess(plaintext)
  69         peer = newPeer(true, "foo", Dummy{nil}, conf, new([SSize]byte))
  70         orig := make([]byte, len(ciphertext))
  71         copy(orig, ciphertext)
  72         b.ResetTimer()
  73         for i := 0; i < b.N; i++ {
  74                 peer.nonceBucket0 = make(map[uint64]struct{}, 1)
  75                 peer.nonceBucket1 = make(map[uint64]struct{}, 1)
  76                 copy(ciphertext, orig)
  77                 if !peer.PktProcess(ciphertext, Dummy{nil}, true) {
  78                         b.Fail()
  79                 }
  80         }
  81 }