]> Cypherpunks.ru repositories - govpn.git/blobdiff - handshake.go
Raise copyright years
[govpn.git] / handshake.go
index 86e6083482a770e7e8b91d52ec5313598fac6352..d1516406d3dd4e92b99642dad7188eebb0ac8eba 100644 (file)
@@ -1,11 +1,10 @@
 /*
 GoVPN -- simple secure free software virtual private network daemon
-Copyright (C) 2014-2015 Sergey Matveev <stargrave@stargrave.org>
+Copyright (C) 2014-2020 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.
+the Free Software Foundation, version 3 of the License.
 
 This program is distributed in the hope that it will be useful,
 but WITHOUT ANY WARRANTY; without even the implied warranty of
@@ -19,36 +18,43 @@ along with this program.  If not, see <http://www.gnu.org/licenses/>.
 package govpn
 
 import (
-       "crypto/rand"
        "crypto/subtle"
        "encoding/binary"
+       "io"
        "log"
-       "net"
-       "path"
        "time"
 
+       "go.cypherpunks.ru/govpn/v7/internal/chacha20"
+
+       "github.com/agl/ed25519"
+       "github.com/agl/ed25519/extra25519"
+       "golang.org/x/crypto/blake2b"
        "golang.org/x/crypto/curve25519"
-       "golang.org/x/crypto/salsa20"
-       "golang.org/x/crypto/salsa20/salsa"
-       "golang.org/x/crypto/xtea"
+)
+
+const (
+       RSize = 8
+       SSize = 32
 )
 
 type Handshake struct {
-       addr     *net.UDPAddr
+       addr     string
+       conn     io.Writer
        LastPing time.Time
-       Id       PeerId
-       rNonce   *[8]byte
-       dhPriv   *[32]byte      // own private DH key
-       key      *[KeySize]byte // handshake encryption key
-       rServer  *[8]byte       // random string for authentication
-       rClient  *[8]byte
-       sServer  *[32]byte // secret string for main key calculation
-       sClient  *[32]byte
+       Conf     *PeerConf
+       dsaPubH  *[ed25519.PublicKeySize]byte
+       key      *[32]byte
+       rNonce   *[16]byte
+       dhPriv   *[32]byte    // own private DH key
+       rServer  *[RSize]byte // random string for authentication
+       rClient  *[RSize]byte
+       sServer  *[SSize]byte // secret string for main key calculation
+       sClient  *[SSize]byte
 }
 
-func keyFromSecrets(server, client []byte) *[KeySize]byte {
-       k := new([32]byte)
-       for i := 0; i < 32; i++ {
+func keyFromSecrets(server, client []byte) *[SSize]byte {
+       k := new([SSize]byte)
+       for i := 0; i < SSize; i++ {
                k[i] = server[i] ^ client[i]
        }
        return k
@@ -57,164 +63,268 @@ func keyFromSecrets(server, client []byte) *[KeySize]byte {
 // Zero handshake's memory state
 func (h *Handshake) Zero() {
        if h.rNonce != nil {
-               sliceZero(h.rNonce[:])
+               SliceZero(h.rNonce[:])
        }
        if h.dhPriv != nil {
-               sliceZero(h.dhPriv[:])
+               SliceZero(h.dhPriv[:])
        }
        if h.key != nil {
-               sliceZero(h.key[:])
+               SliceZero(h.key[:])
+       }
+       if h.dsaPubH != nil {
+               SliceZero(h.dsaPubH[:])
        }
        if h.rServer != nil {
-               sliceZero(h.rServer[:])
+               SliceZero(h.rServer[:])
        }
        if h.rClient != nil {
-               sliceZero(h.rClient[:])
+               SliceZero(h.rClient[:])
        }
        if h.sServer != nil {
-               sliceZero(h.sServer[:])
+               SliceZero(h.sServer[:])
        }
        if h.sClient != nil {
-               sliceZero(h.sClient[:])
+               SliceZero(h.sClient[:])
        }
 }
 
-func (h *Handshake) rNonceNext() []byte {
-       nonce := make([]byte, 8)
-       nonceCurrent, _ := binary.Uvarint(h.rNonce[:])
-       binary.PutUvarint(nonce, nonceCurrent+1)
+func (h *Handshake) rNonceNext(count uint64) *[16]byte {
+       nonce := new([16]byte)
+       nonceCurrent, _ := binary.Uvarint(h.rNonce[8:])
+       binary.PutUvarint(nonce[8:], nonceCurrent+count)
        return nonce
 }
 
-func dhPrivGen() *[32]byte {
-       dh := new([32]byte)
-       if _, err := rand.Read(dh[:]); err != nil {
-               panic("Can not read random for DH private key")
+func dhKeypairGen() (*[32]byte, *[32]byte) {
+       priv := new([32]byte)
+       pub := new([32]byte)
+       repr := new([32]byte)
+       reprFound := false
+       for !reprFound {
+               if _, err := io.ReadFull(Rand, priv[:]); err != nil {
+                       log.Fatalln("Error reading random for DH private key:", err)
+               }
+               reprFound = extra25519.ScalarBaseMult(pub, repr, priv)
        }
-       return dh
+       return priv, repr
 }
 
 func dhKeyGen(priv, pub *[32]byte) *[32]byte {
        key := new([32]byte)
        curve25519.ScalarMult(key, priv, pub)
-       salsa.HSalsa20(key, new([16]byte), key, &salsa.Sigma)
-       return key
+       hashed := blake2b.Sum256(key[:])
+       return &hashed
 }
 
 // Create new handshake state.
-func HandshakeNew(addr *net.UDPAddr) *Handshake {
+func NewHandshake(addr string, conn io.Writer, conf *PeerConf) *Handshake {
        state := Handshake{
                addr:     addr,
+               conn:     conn,
                LastPing: time.Now(),
+               Conf:     conf,
        }
+       state.dsaPubH = new([ed25519.PublicKeySize]byte)
+       copy(state.dsaPubH[:], state.Conf.Verifier.Pub[:])
+       hashed := blake2b.Sum256(state.dsaPubH[:])
+       state.dsaPubH = &hashed
        return &state
 }
 
 // Generate ID tag from client identification and data.
-func idTag(id *PeerId, data []byte) []byte {
-       ciph, err := xtea.NewCipher(id[:])
+func idTag(id *PeerID, timeSync int, data []byte) []byte {
+       enc := make([]byte, 8)
+       copy(enc, data)
+       AddTimeSync(timeSync, enc)
+       mac, err := blake2b.New256(id[:])
        if err != nil {
                panic(err)
        }
-       enc := make([]byte, xtea.BlockSize)
-       ciph.Encrypt(enc, data[:xtea.BlockSize])
-       return enc
+       mac.Write(enc)
+       sum := mac.Sum(nil)
+       return sum[len(sum)-8:]
 }
 
-// Start handshake's procedure from the client.
-// It is the entry point for starting the handshake procedure.
-// You have to specify outgoing conn address, remote's addr address,
-// our own identification and an encryption key. First handshake packet
-// will be sent immediately.
-func HandshakeStart(conn *net.UDPConn, addr *net.UDPAddr, id *PeerId, key *[32]byte) *Handshake {
-       state := HandshakeNew(addr)
-
-       state.dhPriv = dhPrivGen()
-       dhPub := new([32]byte)
-       curve25519.ScalarBaseMult(dhPub, state.dhPriv)
-
-       state.rNonce = new([8]byte)
-       if _, err := rand.Read(state.rNonce[:]); err != nil {
-               panic("Can not read random for handshake nonce")
+// Start handshake's procedure from the client. It is the entry point
+// for starting the handshake procedure.
+// First handshake packet will be sent immediately.
+func HandshakeStart(addr string, conn io.Writer, conf *PeerConf) *Handshake {
+       state := NewHandshake(addr, conn, conf)
+       var dhPubRepr *[32]byte
+       state.dhPriv, dhPubRepr = dhKeypairGen()
+
+       state.rNonce = new([16]byte)
+       if _, err := io.ReadFull(Rand, state.rNonce[8:]); err != nil {
+               log.Fatalln("Error reading random for nonce:", err)
        }
-       enc := make([]byte, 32)
-       salsa20.XORKeyStream(enc, dhPub[:], state.rNonce[:], key)
-       data := append(state.rNonce[:], enc...)
-       data = append(data, idTag(id, state.rNonce[:])...)
-       if _, err := conn.WriteTo(data, addr); err != nil {
-               panic(err)
+       var enc []byte
+       if conf.Noise {
+               enc = make([]byte, conf.MTU-8-RSize)
+       } else {
+               enc = make([]byte, 32)
        }
+       copy(enc, dhPubRepr[:])
+       if conf.Encless {
+               var err error
+               enc, err = EnclessEncode(state.dsaPubH, state.rNonce, enc)
+               if err != err {
+                       panic(err)
+               }
+       } else {
+               chacha20.XORKeyStream(enc, enc, state.rNonce, state.dsaPubH)
+       }
+       data := append(state.rNonce[8:], enc...)
+       data = append(data, idTag(state.Conf.ID, state.Conf.TimeSync, state.rNonce[8:])...)
+       state.conn.Write(data)
        return state
 }
 
 // Process handshake message on the server side.
 // This function is intended to be called on server's side.
-// Client identity, our outgoing conn connection and
-// received data are required.
 // If this is the final handshake message, then new Peer object
 // will be created and used as a transport. If no mutually
 // authenticated Peer is ready, then return nil.
-func (h *Handshake) Server(id *PeerId, conn *net.UDPConn, data []byte) *Peer {
-       // R + ENC(PSK, dh_client_pub) + IDtag
-       if len(data) == 48 && h.rNonce == nil {
-               key := KeyRead(path.Join(PeersPath, id.String(), "key"))
-               h.Id = *id
-
-               // Generate private DH key
-               h.dhPriv = dhPrivGen()
-               dhPub := new([32]byte)
-               curve25519.ScalarBaseMult(dhPub, h.dhPriv)
-
-               // Decrypt remote public key and compute shared key
-               dec := new([32]byte)
-               salsa20.XORKeyStream(dec[:], data[8:8+32], data[:8], key)
-               h.key = dhKeyGen(h.dhPriv, dec)
-
-               // Compute nonce and encrypt our public key
-               h.rNonce = new([8]byte)
-               copy(h.rNonce[:], data[:8])
+func (h *Handshake) Server(data []byte) *Peer {
+       // R + ENC(H(DSAPub), R, El(CDHPub)) + IDtag
+       if h.rNonce == nil && ((!h.Conf.Encless && len(data) >= 48) ||
+               (h.Conf.Encless && len(data) == EnclessEnlargeSize+h.Conf.MTU)) {
+               h.rNonce = new([16]byte)
+               copy(h.rNonce[8:], data[:RSize])
+
+               // Decrypt remote public key
+               cDHRepr := new([32]byte)
+               if h.Conf.Encless {
+                       out, err := EnclessDecode(
+                               h.dsaPubH,
+                               h.rNonce,
+                               data[RSize:len(data)-8],
+                       )
+                       if err != nil {
+                               log.Println("Unable to decode packet from", h.addr, err)
+                               return nil
+                       }
+                       copy(cDHRepr[:], out)
+               } else {
+                       chacha20.XORKeyStream(cDHRepr[:], data[RSize:RSize+32], h.rNonce, h.dsaPubH)
+               }
 
-               encPub := make([]byte, 32)
-               salsa20.XORKeyStream(encPub, dhPub[:], h.rNonceNext(), key)
+               // Generate DH keypair
+               var dhPubRepr *[32]byte
+               h.dhPriv, dhPubRepr = dhKeypairGen()
+
+               // Compute shared key
+               cDH := new([32]byte)
+               extra25519.RepresentativeToPublicKey(cDH, cDHRepr)
+               h.key = dhKeyGen(h.dhPriv, cDH)
+
+               var encPub []byte
+               var err error
+               if h.Conf.Encless {
+                       encPub = make([]byte, h.Conf.MTU)
+                       copy(encPub, dhPubRepr[:])
+                       encPub, err = EnclessEncode(h.dsaPubH, h.rNonceNext(1), encPub)
+                       if err != nil {
+                               panic(err)
+                       }
+               } else {
+                       encPub = make([]byte, 32)
+                       chacha20.XORKeyStream(encPub, dhPubRepr[:], h.rNonceNext(1), h.dsaPubH)
+               }
 
                // Generate R* and encrypt them
-               h.rServer = new([8]byte)
-               if _, err := rand.Read(h.rServer[:]); err != nil {
-                       panic("Can not read random for handshake random key")
+               h.rServer = new([RSize]byte)
+               if _, err = io.ReadFull(Rand, h.rServer[:]); err != nil {
+                       log.Fatalln("Error reading random for R:", err)
+               }
+               h.sServer = new([SSize]byte)
+               if _, err = io.ReadFull(Rand, h.sServer[:]); err != nil {
+                       log.Fatalln("Error reading random for S:", err)
                }
-               h.sServer = new([32]byte)
-               if _, err := rand.Read(h.sServer[:]); err != nil {
-                       panic("Can not read random for handshake shared key")
+               var encRs []byte
+               if h.Conf.Noise && !h.Conf.Encless {
+                       encRs = make([]byte, h.Conf.MTU-len(encPub)-8)
+               } else if h.Conf.Encless {
+                       encRs = make([]byte, h.Conf.MTU-8)
+               } else {
+                       encRs = make([]byte, RSize+SSize)
+               }
+               copy(encRs, append(h.rServer[:], h.sServer[:]...))
+               if h.Conf.Encless {
+                       encRs, err = EnclessEncode(h.key, h.rNonce, encRs)
+                       if err != nil {
+                               panic(err)
+                       }
+               } else {
+                       chacha20.XORKeyStream(encRs, encRs, h.rNonce, h.key)
                }
-               encRs := make([]byte, 8+32)
-               salsa20.XORKeyStream(encRs, append(h.rServer[:], h.sServer[:]...), h.rNonce[:], h.key)
 
                // Send that to client
-               if _, err := conn.WriteTo(
-                       append(encPub, append(encRs, idTag(id, encPub)...)...), h.addr); err != nil {
-                       panic(err)
-               }
+               h.conn.Write(append(encPub, append(
+                       encRs, idTag(h.Conf.ID, h.Conf.TimeSync, encPub)...,
+               )...))
                h.LastPing = time.Now()
        } else
-       // ENC(K, RS + RC + SC) + IDtag
-       if len(data) == 56 && h.rClient == nil {
-               // Decrypted Rs compare rServer
-               decRs := make([]byte, 8+8+32)
-               salsa20.XORKeyStream(decRs, data[:8+8+32], h.rNonceNext(), h.key)
-               if subtle.ConstantTimeCompare(decRs[:8], h.rServer[:]) != 1 {
+       // ENC(K, R+1, RS + RC + SC + Sign(DSAPriv, K)) + IDtag
+       if h.rClient == nil && ((!h.Conf.Encless && len(data) >= 120) ||
+               (h.Conf.Encless && len(data) == EnclessEnlargeSize+h.Conf.MTU)) {
+               var dec []byte
+               var err error
+               if h.Conf.Encless {
+                       dec, err = EnclessDecode(
+                               h.key,
+                               h.rNonceNext(1),
+                               data[:len(data)-8],
+                       )
+                       if err != nil {
+                               log.Println("Unable to decode packet from", h.addr, err)
+                               return nil
+                       }
+                       dec = dec[:RSize+RSize+SSize+ed25519.SignatureSize]
+               } else {
+                       dec = make([]byte, RSize+RSize+SSize+ed25519.SignatureSize)
+                       chacha20.XORKeyStream(
+                               dec,
+                               data[:RSize+RSize+SSize+ed25519.SignatureSize],
+                               h.rNonceNext(1),
+                               h.key,
+                       )
+               }
+               if subtle.ConstantTimeCompare(dec[:RSize], h.rServer[:]) != 1 {
                        log.Println("Invalid server's random number with", h.addr)
                        return nil
                }
+               sign := new([ed25519.SignatureSize]byte)
+               copy(sign[:], dec[RSize+RSize+SSize:])
+               if !ed25519.Verify(h.Conf.Verifier.Pub, h.key[:], sign) {
+                       log.Println("Invalid signature from", h.addr)
+                       return nil
+               }
 
                // Send final answer to client
-               enc := make([]byte, 8)
-               salsa20.XORKeyStream(enc, decRs[8:8+8], make([]byte, 8), h.key)
-               if _, err := conn.WriteTo(append(enc, idTag(id, enc)...), h.addr); err != nil {
-                       panic(err)
+               var enc []byte
+               if h.Conf.Noise {
+                       enc = make([]byte, h.Conf.MTU-8)
+               } else {
+                       enc = make([]byte, RSize)
                }
+               copy(enc, dec[RSize:RSize+RSize])
+               if h.Conf.Encless {
+                       enc, err = EnclessEncode(h.key, h.rNonceNext(2), enc)
+                       if err != nil {
+                               panic(err)
+                       }
+               } else {
+                       chacha20.XORKeyStream(enc, enc, h.rNonceNext(2), h.key)
+               }
+               h.conn.Write(append(enc, idTag(h.Conf.ID, h.Conf.TimeSync, enc)...))
 
                // Switch peer
-               peer := newPeer(h.addr, h.Id, 0, keyFromSecrets(h.sServer[:], decRs[8+8:]))
+               peer := newPeer(
+                       false,
+                       h.addr,
+                       h.conn,
+                       h.Conf,
+                       keyFromSecrets(h.sServer[:], dec[RSize+RSize:RSize+RSize+SSize]))
                h.LastPing = time.Now()
                return peer
        } else {
@@ -225,71 +335,124 @@ func (h *Handshake) Server(id *PeerId, conn *net.UDPConn, data []byte) *Peer {
 
 // Process handshake message on the client side.
 // This function is intended to be called on client's side.
-// Our outgoing conn connection, authentication
-// key and received data are required.
 // If this is the final handshake message, then new Peer object
 // will be created and used as a transport. If no mutually
 // authenticated Peer is ready, then return nil.
-func (h *Handshake) Client(id *PeerId, conn *net.UDPConn, key *[KeySize]byte, data []byte) *Peer {
-       switch len(data) {
-       case 80: // ENC(PSK, dh_server_pub) + ENC(K, RS + SS) + IDtag
-               if h.key != nil {
-                       log.Println("Invalid handshake stage from", h.addr)
-                       return nil
+func (h *Handshake) Client(data []byte) *Peer {
+       // ENC(H(DSAPub), R+1, El(SDHPub)) + ENC(K, R, RS + SS) + IDtag
+       if h.rServer == nil && h.key == nil &&
+               ((!h.Conf.Encless && len(data) >= 80) ||
+                       (h.Conf.Encless && len(data) == 2*(EnclessEnlargeSize+h.Conf.MTU))) {
+               // Decrypt remote public key
+               sDHRepr := new([32]byte)
+               var tmp []byte
+               var err error
+               if h.Conf.Encless {
+                       tmp, err = EnclessDecode(
+                               h.dsaPubH,
+                               h.rNonceNext(1),
+                               data[:len(data)/2],
+                       )
+                       if err != nil {
+                               log.Println("Unable to decode packet from", h.addr, err)
+                               return nil
+                       }
+                       copy(sDHRepr[:], tmp[:32])
+               } else {
+                       chacha20.XORKeyStream(sDHRepr[:], data[:32], h.rNonceNext(1), h.dsaPubH)
                }
 
-               // Decrypt remote public key and compute shared key
-               dec := new([32]byte)
-               salsa20.XORKeyStream(dec[:], data[:32], h.rNonceNext(), key)
-               h.key = dhKeyGen(h.dhPriv, dec)
+               // Compute shared key
+               sDH := new([32]byte)
+               extra25519.RepresentativeToPublicKey(sDH, sDHRepr)
+               h.key = dhKeyGen(h.dhPriv, sDH)
 
                // Decrypt Rs
-               decRs := make([]byte, 8+32)
-               salsa20.XORKeyStream(decRs, data[32:32+8+32], h.rNonce[:], h.key)
-               h.rServer = new([8]byte)
-               copy(h.rServer[:], decRs[:8])
-               h.sServer = new([32]byte)
-               copy(h.sServer[:], decRs[8:])
+               h.rServer = new([RSize]byte)
+               h.sServer = new([SSize]byte)
+               if h.Conf.Encless {
+                       tmp, err = EnclessDecode(h.key, h.rNonce, data[len(data)/2:len(data)-8])
+                       if err != nil {
+                               log.Println("Unable to decode packet from", h.addr, err)
+                               return nil
+                       }
+                       copy(h.rServer[:], tmp[:RSize])
+                       copy(h.sServer[:], tmp[RSize:RSize+SSize])
+               } else {
+                       decRs := make([]byte, RSize+SSize)
+                       chacha20.XORKeyStream(decRs, data[SSize:SSize+RSize+SSize], h.rNonce, h.key)
+                       copy(h.rServer[:], decRs[:RSize])
+                       copy(h.sServer[:], decRs[RSize:])
+               }
 
-               // Generate R* and encrypt them
-               h.rClient = new([8]byte)
-               if _, err := rand.Read(h.rClient[:]); err != nil {
-                       panic("Can not read random for handshake random key")
+               // Generate R* and signature and encrypt them
+               h.rClient = new([RSize]byte)
+               if _, err = io.ReadFull(Rand, h.rClient[:]); err != nil {
+                       log.Fatalln("Error reading random for R:", err)
                }
-               h.sClient = new([32]byte)
-               if _, err := rand.Read(h.sClient[:]); err != nil {
-                       panic("Can not read random for handshake shared key")
+               h.sClient = new([SSize]byte)
+               if _, err = io.ReadFull(Rand, h.sClient[:]); err != nil {
+                       log.Fatalln("Error reading random for S:", err)
                }
-               encRs := make([]byte, 8+8+32)
-               salsa20.XORKeyStream(encRs,
-                       append(h.rServer[:],
-                               append(h.rClient[:], h.sClient[:]...)...), h.rNonceNext(), h.key)
+               sign := ed25519.Sign(h.Conf.DSAPriv, h.key[:])
 
-               // Send that to server
-               if _, err := conn.WriteTo(append(encRs, idTag(id, encRs)...), h.addr); err != nil {
-                       panic(err)
+               var enc []byte
+               if h.Conf.Noise {
+                       enc = make([]byte, h.Conf.MTU-8)
+               } else {
+                       enc = make([]byte, RSize+RSize+SSize+ed25519.SignatureSize)
                }
-               h.LastPing = time.Now()
-       case 16: // ENC(K, RC) + IDtag
-               if h.key == nil {
-                       log.Println("Invalid handshake stage from", h.addr)
-                       return nil
+               copy(enc, h.rServer[:])
+               copy(enc[RSize:], h.rClient[:])
+               copy(enc[RSize+RSize:], h.sClient[:])
+               copy(enc[RSize+RSize+SSize:], sign[:])
+               if h.Conf.Encless {
+                       enc, err = EnclessEncode(h.key, h.rNonceNext(1), enc)
+                       if err != nil {
+                               panic(err)
+                       }
+               } else {
+                       chacha20.XORKeyStream(enc, enc, h.rNonceNext(1), h.key)
                }
 
+               // Send that to server
+               h.conn.Write(append(enc, idTag(h.Conf.ID, h.Conf.TimeSync, enc)...))
+               h.LastPing = time.Now()
+       } else
+       // ENC(K, R+2, RC) + IDtag
+       if h.key != nil && ((!h.Conf.Encless && len(data) >= 16) ||
+               (h.Conf.Encless && len(data) == EnclessEnlargeSize+h.Conf.MTU)) {
+               var err error
                // Decrypt rClient
-               dec := make([]byte, 8)
-               salsa20.XORKeyStream(dec, data[:8], make([]byte, 8), h.key)
+               var dec []byte
+               if h.Conf.Encless {
+                       dec, err = EnclessDecode(h.key, h.rNonceNext(2), data[:len(data)-8])
+                       if err != nil {
+                               log.Println("Unable to decode packet from", h.addr, err)
+                               return nil
+                       }
+                       dec = dec[:RSize]
+               } else {
+                       dec = make([]byte, RSize)
+                       chacha20.XORKeyStream(dec, data[:RSize], h.rNonceNext(2), h.key)
+               }
                if subtle.ConstantTimeCompare(dec, h.rClient[:]) != 1 {
                        log.Println("Invalid client's random number with", h.addr)
                        return nil
                }
 
                // Switch peer
-               peer := newPeer(h.addr, h.Id, 1, keyFromSecrets(h.sServer[:], h.sClient[:]))
+               peer := newPeer(
+                       true,
+                       h.addr,
+                       h.conn,
+                       h.Conf,
+                       keyFromSecrets(h.sServer[:], h.sClient[:]),
+               )
                h.LastPing = time.Now()
                return peer
-       default:
-               log.Println("Invalid handshake message from", h.addr)
+       } else {
+               log.Println("Invalid handshake stage from", h.addr)
        }
        return nil
 }