src/govpn/stats.go

   1 package govpn
   2
   3 import (
   4         "encoding/json"
   5         "log"
   6         "net"
   7         "time"
   8 )
   9
  10 const (
  11         RWTimeout = 10 * time.Second
  12 )
  13
  14 type KnownPeers map[string]**Peer
  15
  16 // StatsProcessor is assumed to be run in background. It accepts
  17 // connection on statsPort, reads anything one send to them and show
  18 // information about known peers in serialized JSON format. peers
  19 // argument is a reference to the map with references to the peers as
  20 // values. Map is used here because of ease of adding and removing
  21 // elements in it.
  22 func StatsProcessor(statsPort net.Listener, peers *KnownPeers) {
  23         var conn net.Conn
  24         var err error
  25         var data []byte
  26         buf := make([]byte, 2<<8)
  27         for {
  28                 conn, err = statsPort.Accept()
  29                 if err != nil {
  30                         log.Println("Error during accepting connection", err.Error())
  31                         continue
  32                 }
  33                 conn.SetDeadline(time.Now().Add(RWTimeout))
  34                 conn.Read(buf)
  35                 conn.Write([]byte("HTTP/1.0 200 OK\r\nContent-Type: application/json\r\n\r\n"))
  36                 var peersList []*Peer
  37                 for _, peer := range *peers {
  38                         peersList = append(peersList, *peer)
  39                 }
  40                 data, err = json.Marshal(peersList)
  41                 if err != nil {
  42                         panic(err)
  43                 }
  44                 conn.Write(data)
  45                 conn.Close()
  46         }
  47 }