src/cypherpunks.ru/govpn/tap.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 govpn
  20
  21 import (
  22         "io"
  23 )
  24
  25 type TAP struct {
  26         Name string
  27         Sink chan []byte
  28         dev  io.ReadWriter
  29         buf0 []byte
  30         buf1 []byte
  31         bufZ bool
  32 }
  33
  34 var (
  35         taps = make(map[string]*TAP)
  36 )
  37
  38 func NewTAP(ifaceName string, mtu int) (*TAP, error) {
  39         tapRaw, err := newTAPer(ifaceName)
  40         if err != nil {
  41                 return nil, err
  42         }
  43         tap := TAP{
  44                 Name: ifaceName,
  45                 dev:  tapRaw,
  46                 buf0: make([]byte, mtu),
  47                 buf1: make([]byte, mtu),
  48                 Sink: make(chan []byte),
  49         }
  50         go func() {
  51                 var n int
  52                 var err error
  53                 var buf []byte
  54                 for {
  55                         if tap.bufZ {
  56                                 buf = tap.buf0
  57                         } else {
  58                                 buf = tap.buf1
  59                         }
  60                         tap.bufZ = !tap.bufZ
  61                         n, err = tap.dev.Read(buf)
  62                         if err != nil {
  63                                 panic("Reading TAP:" + err.Error())
  64                         }
  65                         tap.Sink <- buf[:n]
  66                 }
  67         }()
  68         return &tap, nil
  69 }
  70
  71 func (t *TAP) Write(data []byte) (n int, err error) {
  72         return t.dev.Write(data)
  73 }
  74
  75 func TAPListen(ifaceName string, mtu int) (*TAP, error) {
  76         tap, exists := taps[ifaceName]
  77         if exists {
  78                 return tap, nil
  79         }
  80         tap, err := NewTAP(ifaceName, mtu)
  81         if err != nil {
  82                 return nil, err
  83         }
  84         taps[ifaceName] = tap
  85         return tap, nil
  86 }