src/cypherpunks.ru/govpn/cmd/govpn-server/common.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 main
  20
  21 import (
  22         "bytes"
  23         "sync"
  24         "time"
  25
  26         "cypherpunks.ru/govpn"
  27 )
  28
  29 type PeerState struct {
  30         peer       *govpn.Peer
  31         terminator chan struct{}
  32         tap        *govpn.TAP
  33 }
  34
  35 var (
  36         handshakes map[string]*govpn.Handshake = make(map[string]*govpn.Handshake)
  37         hsLock     sync.RWMutex
  38
  39         peers     map[string]*PeerState = make(map[string]*PeerState)
  40         peersLock sync.RWMutex
  41
  42         peersById     map[govpn.PeerId]string = make(map[govpn.PeerId]string)
  43         peersByIdLock sync.RWMutex
  44
  45         knownPeers govpn.KnownPeers
  46         kpLock     sync.RWMutex
  47 )
  48
  49 func peerReady(ps PeerState) {
  50         var data []byte
  51         heartbeat := time.NewTicker(ps.peer.Timeout)
  52 Processor:
  53         for {
  54                 select {
  55                 case <-heartbeat.C:
  56                         ps.peer.EthProcess(nil)
  57                 case <-ps.terminator:
  58                         break Processor
  59                 case data = <-ps.tap.Sink:
  60                         ps.peer.EthProcess(data)
  61                 }
  62         }
  63         close(ps.terminator)
  64         ps.peer.Zero()
  65         heartbeat.Stop()
  66 }
  67
  68 func callUp(peerId *govpn.PeerId, remoteAddr string) (string, error) {
  69         ifaceName := confs[*peerId].Iface
  70         if confs[*peerId].Up != "" {
  71                 result, err := govpn.ScriptCall(confs[*peerId].Up, ifaceName, remoteAddr)
  72                 if err != nil {
  73                         govpn.Printf(`[script-failed bind="%s" path="%s" err="%s"]`, *bindAddr, confs[*peerId].Up, err)
  74                         return "", err
  75                 }
  76                 if ifaceName == "" {
  77                         sepIndex := bytes.Index(result, []byte{'\n'})
  78                         if sepIndex < 0 {
  79                                 sepIndex = len(result)
  80                         }
  81                         ifaceName = string(result[:sepIndex])
  82                 }
  83         }
  84         if ifaceName == "" {
  85                 govpn.Printf(`[tap-failed bind="%s" peer="%s"]`, *bindAddr, *peerId)
  86         }
  87         return ifaceName, nil
  88 }