src/govpn/cmd/govpn-server/common.go

   1 package main
   2
   3 import (
   4         "bytes"
   5         "path"
   6         "sync"
   7         "time"
   8
   9         "govpn"
  10 )
  11
  12 type PeerState struct {
  13         peer       *govpn.Peer
  14         terminator chan struct{}
  15         tap        *govpn.TAP
  16 }
  17
  18 var (
  19         handshakes map[string]*govpn.Handshake = make(map[string]*govpn.Handshake)
  20         hsLock     sync.RWMutex
  21
  22         peers     map[string]*PeerState = make(map[string]*PeerState)
  23         peersLock sync.RWMutex
  24
  25         peersById     map[govpn.PeerId]string = make(map[govpn.PeerId]string)
  26         peersByIdLock sync.RWMutex
  27
  28         knownPeers govpn.KnownPeers
  29         kpLock     sync.RWMutex
  30 )
  31
  32 func peerReady(ps PeerState) {
  33         var data []byte
  34         heartbeat := time.NewTicker(ps.peer.Timeout)
  35 Processor:
  36         for {
  37                 select {
  38                 case <-heartbeat.C:
  39                         ps.peer.EthProcess(nil)
  40                 case <-ps.terminator:
  41                         break Processor
  42                 case data = <-ps.tap.Sink:
  43                         ps.peer.EthProcess(data)
  44                 }
  45         }
  46         close(ps.terminator)
  47         ps.peer.Zero()
  48         heartbeat.Stop()
  49 }
  50
  51 func callUp(peerId *govpn.PeerId) (string, error) {
  52         upPath := path.Join(govpn.PeersPath, peerId.String(), "up.sh")
  53         result, err := govpn.ScriptCall(upPath, "")
  54         if err != nil {
  55                 return "", err
  56         }
  57         sepIndex := bytes.Index(result, []byte{'\n'})
  58         if sepIndex < 0 {
  59                 sepIndex = len(result)
  60         }
  61         ifaceName := string(result[:sepIndex])
  62         return ifaceName, nil
  63 }