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         ps.peer.Zero()
  47         heartbeat.Stop()
  48 }
  49
  50 func callUp(peerId *govpn.PeerId) (string, error) {
  51         upPath := path.Join(govpn.PeersPath, peerId.String(), "up.sh")
  52         result, err := govpn.ScriptCall(upPath, "")
  53         if err != nil {
  54                 return "", err
  55         }
  56         sepIndex := bytes.Index(result, []byte{'\n'})
  57         if sepIndex < 0 {
  58                 sepIndex = len(result)
  59         }
  60         ifaceName := string(result[:sepIndex])
  61         return ifaceName, nil
  62 }