src/govpn/netconn.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         "net"
  23         "time"
  24 )
  25
  26 type UDPPkt struct {
  27         Addr *net.UDPAddr
  28         Data []byte
  29 }
  30
  31 // Create UDP listening goroutine.
  32 // This function takes already listening UDP socket and a buffer where
  33 // all UDP packet data will be saved, channel where information about
  34 // remote address and number of written bytes are stored, and a channel
  35 // used to tell that buffer is ready to be overwritten.
  36 func ConnListenUDP(conn *net.UDPConn) (chan UDPPkt, chan struct{}) {
  37         buf := make([]byte, MTU)
  38         sink := make(chan UDPPkt)
  39         sinkReady := make(chan struct{})
  40         go func(conn *net.UDPConn) {
  41                 var n int
  42                 var addr *net.UDPAddr
  43                 var err error
  44                 for {
  45                         <-sinkReady
  46                         conn.SetReadDeadline(time.Now().Add(time.Second))
  47                         n, addr, err = conn.ReadFromUDP(buf)
  48                         if err != nil {
  49                                 // This is needed for ticking the timeouts counter outside
  50                                 sink <- UDPPkt{nil, nil}
  51                                 continue
  52                         }
  53                         sink <- UDPPkt{addr, buf[:n]}
  54                 }
  55         }(conn)
  56         sinkReady <- struct{}{}
  57         return sink, sinkReady
  58 }