src/govpn/handshake.go

   1 /*
   2 GoVPN -- simple secure free software virtual private network daemon
   3 Copyright (C) 2014-2015 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         "crypto/rand"
  23         "crypto/subtle"
  24         "encoding/binary"
  25         "log"
  26         "net"
  27         "time"
  28
  29         "github.com/agl/ed25519"
  30         "github.com/agl/ed25519/extra25519"
  31         "golang.org/x/crypto/curve25519"
  32         "golang.org/x/crypto/salsa20"
  33         "golang.org/x/crypto/salsa20/salsa"
  34         "golang.org/x/crypto/xtea"
  35 )
  36
  37 const (
  38         RSize = 8
  39         SSize = 32
  40 )
  41
  42 type Handshake struct {
  43         addr     *net.UDPAddr
  44         LastPing time.Time
  45         Conf     *PeerConf
  46         dsaPubH  *[ed25519.PublicKeySize]byte
  47         key      *[32]byte
  48         rNonce   *[RSize]byte
  49         dhPriv   *[32]byte    // own private DH key
  50         rServer  *[RSize]byte // random string for authentication
  51         rClient  *[RSize]byte
  52         sServer  *[SSize]byte // secret string for main key calculation
  53         sClient  *[SSize]byte
  54 }
  55
  56 func keyFromSecrets(server, client []byte) *[SSize]byte {
  57         k := new([SSize]byte)
  58         for i := 0; i < SSize; i++ {
  59                 k[i] = server[i] ^ client[i]
  60         }
  61         return k
  62 }
  63
  64 // Apply HSalsa20 function for data. Used to hash public keys.
  65 func HApply(data *[32]byte) {
  66         salsa.HSalsa20(data, new([16]byte), data, &salsa.Sigma)
  67 }
  68
  69 // Zero handshake's memory state
  70 func (h *Handshake) Zero() {
  71         if h.rNonce != nil {
  72                 sliceZero(h.rNonce[:])
  73         }
  74         if h.dhPriv != nil {
  75                 sliceZero(h.dhPriv[:])
  76         }
  77         if h.key != nil {
  78                 sliceZero(h.key[:])
  79         }
  80         if h.dsaPubH != nil {
  81                 sliceZero(h.dsaPubH[:])
  82         }
  83         if h.rServer != nil {
  84                 sliceZero(h.rServer[:])
  85         }
  86         if h.rClient != nil {
  87                 sliceZero(h.rClient[:])
  88         }
  89         if h.sServer != nil {
  90                 sliceZero(h.sServer[:])
  91         }
  92         if h.sClient != nil {
  93                 sliceZero(h.sClient[:])
  94         }
  95 }
  96
  97 func (h *Handshake) rNonceNext(count uint64) []byte {
  98         nonce := make([]byte, RSize)
  99         nonceCurrent, _ := binary.Uvarint(h.rNonce[:])
 100         binary.PutUvarint(nonce, nonceCurrent+count)
 101         return nonce
 102 }
 103
 104 func dhKeypairGen() (*[32]byte, *[32]byte) {
 105         priv := new([32]byte)
 106         pub := new([32]byte)
 107         repr := new([32]byte)
 108         reprFound := false
 109         for !reprFound {
 110                 if _, err := rand.Read(priv[:]); err != nil {
 111                         log.Fatalln("Error reading random for DH private key:", err)
 112                 }
 113                 reprFound = extra25519.ScalarBaseMult(pub, repr, priv)
 114         }
 115         return priv, repr
 116 }
 117
 118 func dhKeyGen(priv, pub *[32]byte) *[32]byte {
 119         key := new([32]byte)
 120         curve25519.ScalarMult(key, priv, pub)
 121         HApply(key)
 122         return key
 123 }
 124
 125 // Create new handshake state.
 126 func HandshakeNew(addr *net.UDPAddr, conf *PeerConf) *Handshake {
 127         state := Handshake{
 128                 addr:     addr,
 129                 LastPing: time.Now(),
 130                 Conf:     conf,
 131         }
 132         state.dsaPubH = new([ed25519.PublicKeySize]byte)
 133         copy(state.dsaPubH[:], state.Conf.DSAPub[:])
 134         HApply(state.dsaPubH)
 135         return &state
 136 }
 137
 138 // Generate ID tag from client identification and data.
 139 func idTag(id *PeerId, data []byte) []byte {
 140         ciph, err := xtea.NewCipher(id[:])
 141         if err != nil {
 142                 panic(err)
 143         }
 144         enc := make([]byte, xtea.BlockSize)
 145         ciph.Encrypt(enc, data[:xtea.BlockSize])
 146         return enc
 147 }
 148
 149 // Start handshake's procedure from the client. It is the entry point
 150 // for starting the handshake procedure. You have to specify outgoing
 151 // conn address, remote's addr address, our own peer configuration.
 152 // First handshake packet will be sent immediately.
 153 func HandshakeStart(conf *PeerConf, conn *net.UDPConn, addr *net.UDPAddr) *Handshake {
 154         state := HandshakeNew(addr, conf)
 155
 156         var dhPubRepr *[32]byte
 157         state.dhPriv, dhPubRepr = dhKeypairGen()
 158
 159         state.rNonce = new([RSize]byte)
 160         if _, err := rand.Read(state.rNonce[:]); err != nil {
 161                 log.Fatalln("Error reading random for nonce:", err)
 162         }
 163         enc := make([]byte, 32)
 164         salsa20.XORKeyStream(enc, dhPubRepr[:], state.rNonce[:], state.dsaPubH)
 165         data := append(state.rNonce[:], enc...)
 166         data = append(data, idTag(state.Conf.Id, state.rNonce[:])...)
 167         conn.WriteToUDP(data, addr)
 168         return state
 169 }
 170
 171 // Process handshake message on the server side.
 172 // This function is intended to be called on server's side.
 173 // Our outgoing conn connection and received data are required.
 174 // If this is the final handshake message, then new Peer object
 175 // will be created and used as a transport. If no mutually
 176 // authenticated Peer is ready, then return nil.
 177 func (h *Handshake) Server(conn *net.UDPConn, data []byte) *Peer {
 178         // R + ENC(H(DSAPub), R, El(CDHPub)) + IDtag
 179         if len(data) == 48 && h.rNonce == nil {
 180                 // Generate DH keypair
 181                 var dhPubRepr *[32]byte
 182                 h.dhPriv, dhPubRepr = dhKeypairGen()
 183
 184                 h.rNonce = new([RSize]byte)
 185                 copy(h.rNonce[:], data[:RSize])
 186
 187                 // Decrypt remote public key and compute shared key
 188                 cDHRepr := new([32]byte)
 189                 salsa20.XORKeyStream(
 190                         cDHRepr[:],
 191                         data[RSize:RSize+32],
 192                         h.rNonce[:],
 193                         h.dsaPubH,
 194                 )
 195                 cDH := new([32]byte)
 196                 extra25519.RepresentativeToPublicKey(cDH, cDHRepr)
 197                 h.key = dhKeyGen(h.dhPriv, cDH)
 198
 199                 encPub := make([]byte, 32)
 200                 salsa20.XORKeyStream(encPub, dhPubRepr[:], h.rNonceNext(1), h.dsaPubH)
 201
 202                 // Generate R* and encrypt them
 203                 h.rServer = new([RSize]byte)
 204                 if _, err := rand.Read(h.rServer[:]); err != nil {
 205                         log.Fatalln("Error reading random for R:", err)
 206                 }
 207                 h.sServer = new([SSize]byte)
 208                 if _, err := rand.Read(h.sServer[:]); err != nil {
 209                         log.Fatalln("Error reading random for S:", err)
 210                 }
 211                 encRs := make([]byte, RSize+SSize)
 212                 salsa20.XORKeyStream(encRs, append(h.rServer[:], h.sServer[:]...), h.rNonce[:], h.key)
 213
 214                 // Send that to client
 215                 conn.WriteToUDP(append(encPub, append(encRs, idTag(h.Conf.Id, encPub)...)...), h.addr)
 216                 h.LastPing = time.Now()
 217         } else
 218         // ENC(K, R+1, RS + RC + SC + Sign(DSAPriv, K)) + IDtag
 219         if len(data) == 120 && h.rClient == nil {
 220                 // Decrypted Rs compare rServer
 221                 dec := make([]byte, RSize+RSize+SSize+ed25519.SignatureSize)
 222                 salsa20.XORKeyStream(
 223                         dec,
 224                         data[:RSize+RSize+SSize+ed25519.SignatureSize],
 225                         h.rNonceNext(1),
 226                         h.key,
 227                 )
 228                 if subtle.ConstantTimeCompare(dec[:RSize], h.rServer[:]) != 1 {
 229                         log.Println("Invalid server's random number with", h.addr)
 230                         return nil
 231                 }
 232                 sign := new([ed25519.SignatureSize]byte)
 233                 copy(sign[:], dec[RSize+RSize+SSize:])
 234                 if !ed25519.Verify(h.Conf.DSAPub, h.key[:], sign) {
 235                         log.Println("Invalid signature from", h.addr)
 236                         return nil
 237                 }
 238
 239                 // Send final answer to client
 240                 enc := make([]byte, RSize)
 241                 salsa20.XORKeyStream(enc, dec[RSize:RSize+RSize], h.rNonceNext(2), h.key)
 242                 conn.WriteToUDP(append(enc, idTag(h.Conf.Id, enc)...), h.addr)
 243
 244                 // Switch peer
 245                 peer := newPeer(
 246                         h.addr,
 247                         h.Conf,
 248                         0,
 249                         keyFromSecrets(h.sServer[:], dec[RSize+RSize:RSize+RSize+SSize]))
 250                 h.LastPing = time.Now()
 251                 return peer
 252         } else {
 253                 log.Println("Invalid handshake message from", h.addr)
 254         }
 255         return nil
 256 }
 257
 258 // Process handshake message on the client side.
 259 // This function is intended to be called on client's side.
 260 // Our outgoing conn connection, authentication
 261 // key and received data are required.
 262 // If this is the final handshake message, then new Peer object
 263 // will be created and used as a transport. If no mutually
 264 // authenticated Peer is ready, then return nil.
 265 func (h *Handshake) Client(conn *net.UDPConn, data []byte) *Peer {
 266         switch len(data) {
 267         case 80: // ENC(H(DSAPub), R+1, El(SDHPub)) + ENC(K, R, RS + SS) + IDtag
 268                 if h.key != nil {
 269                         log.Println("Invalid handshake stage from", h.addr)
 270                         return nil
 271                 }
 272
 273                 // Decrypt remote public key and compute shared key
 274                 sDHRepr := new([32]byte)
 275                 salsa20.XORKeyStream(sDHRepr[:], data[:32], h.rNonceNext(1), h.dsaPubH)
 276                 sDH := new([32]byte)
 277                 extra25519.RepresentativeToPublicKey(sDH, sDHRepr)
 278                 h.key = dhKeyGen(h.dhPriv, sDH)
 279
 280                 // Decrypt Rs
 281                 decRs := make([]byte, RSize+SSize)
 282                 salsa20.XORKeyStream(decRs, data[SSize:32+RSize+SSize], h.rNonce[:], h.key)
 283                 h.rServer = new([RSize]byte)
 284                 copy(h.rServer[:], decRs[:RSize])
 285                 h.sServer = new([SSize]byte)
 286                 copy(h.sServer[:], decRs[RSize:])
 287
 288                 // Generate R* and signature and encrypt them
 289                 h.rClient = new([RSize]byte)
 290                 if _, err := rand.Read(h.rClient[:]); err != nil {
 291                         log.Fatalln("Error reading random for R:", err)
 292                 }
 293                 h.sClient = new([SSize]byte)
 294                 if _, err := rand.Read(h.sClient[:]); err != nil {
 295                         log.Fatalln("Error reading random for S:", err)
 296                 }
 297                 sign := ed25519.Sign(h.Conf.DSAPriv, h.key[:])
 298
 299                 enc := make([]byte, RSize+RSize+SSize+ed25519.SignatureSize)
 300                 salsa20.XORKeyStream(enc,
 301                         append(h.rServer[:],
 302                                 append(h.rClient[:],
 303                                         append(h.sClient[:], sign[:]...)...)...), h.rNonceNext(1), h.key)
 304
 305                 // Send that to server
 306                 conn.WriteToUDP(append(enc, idTag(h.Conf.Id, enc)...), h.addr)
 307                 h.LastPing = time.Now()
 308         case 16: // ENC(K, R+2, RC) + IDtag
 309                 if h.key == nil {
 310                         log.Println("Invalid handshake stage from", h.addr)
 311                         return nil
 312                 }
 313
 314                 // Decrypt rClient
 315                 dec := make([]byte, RSize)
 316                 salsa20.XORKeyStream(dec, data[:RSize], h.rNonceNext(2), h.key)
 317                 if subtle.ConstantTimeCompare(dec, h.rClient[:]) != 1 {
 318                         log.Println("Invalid client's random number with", h.addr)
 319                         return nil
 320                 }
 321
 322                 // Switch peer
 323                 peer := newPeer(h.addr, h.Conf, 1, keyFromSecrets(h.sServer[:], h.sClient[:]))
 324                 h.LastPing = time.Now()
 325                 return peer
 326         default:
 327                 log.Println("Invalid handshake message from", h.addr)
 328         }
 329         return nil
 330 }