src/cypherpunks.ru/govpn/handshake_test.go

   1 /*
   2 GoVPN -- simple secure free software virtual private network daemon
   3 Copyright (C) 2014-2017 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 )
  24
  25 func testHandshake(t *testing.T, cl, srv *Handshake) {
  26         var err error
  27         if _, err = srv.Server(testCt); err != nil {
  28                 t.Error(err)
  29         }
  30         if _, err = cl.Client(testCt); err != nil {
  31                 t.Error(err)
  32         }
  33         p, err := srv.Server(testCt)
  34         if p == nil {
  35                 t.Fail()
  36         }
  37         if err != nil {
  38                 t.Error(err)
  39         }
  40         p, err = cl.Client(testCt)
  41         if p == nil {
  42                 t.Fail()
  43         }
  44         if err != nil {
  45                 t.Error(err)
  46         }
  47 }
  48
  49 func TestHandshakeSymmetric(t *testing.T) {
  50         var err error
  51         // initial values are taken from peer_test.go's init()
  52         v := VerifierNew(1<<10, 1<<4, 1, &testPeerID)
  53         testConf.Verifier = v
  54         testConf.DSAPriv, err = v.PasswordApply("does not matter")
  55         if err != nil {
  56                 t.Error(err)
  57         }
  58         hsS := NewHandshake("server", Dummy{&testCt}, testConf)
  59         hsC, err := HandshakeStart("client", Dummy{&testCt}, testConf)
  60         if err != nil {
  61                 t.Error(err)
  62         }
  63         testHandshake(t, hsC, hsS)
  64 }
  65
  66 func TestHandshakeNoiseSymmetric(t *testing.T) {
  67         var err error
  68         // initial values are taken from peer_test.go's init()
  69         v := VerifierNew(1<<10, 1<<4, 1, &testPeerID)
  70         testConf.Verifier = v
  71         testConf.DSAPriv, err = v.PasswordApply("does not matter")
  72         if err != nil {
  73                 t.Error(err)
  74         }
  75         testConf.Noise = true
  76         hsS := NewHandshake("server", Dummy{&testCt}, testConf)
  77         hsC, err := HandshakeStart("client", Dummy{&testCt}, testConf)
  78         if err != nil {
  79                 t.Error(err)
  80         }
  81         testHandshake(t, hsC, hsS)
  82         testConf.Noise = false
  83 }
  84
  85 func TestHandshakeEnclessSymmetric(t *testing.T) {
  86         var err error
  87         // initial values are taken from peer_test.go's init()
  88         v := VerifierNew(1<<10, 1<<4, 1, &testPeerID)
  89         testConf.Verifier = v
  90         testConf.DSAPriv, err = v.PasswordApply("does not matter")
  91         if err != nil {
  92                 t.Error(err)
  93         }
  94         testConf.Encless = true
  95         testConf.Noise = true
  96         hsS := NewHandshake("server", Dummy{&testCt}, testConf)
  97         hsC, err := HandshakeStart("client", Dummy{&testCt}, testConf)
  98         if err != nil {
  99                 t.Error(err)
 100         }
 101         testHandshake(t, hsC, hsS)
 102         testConf.Encless = false
 103         testConf.Noise = false
 104 }