src/cypherpunks.ru/govpn/tap.go

   1 /*
   2 GoVPN -- simple secure free software virtual private network daemon
   3 Copyright (C) 2014-2017 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 // TAP is a TUN or a TAP interface.
  26 type TAP struct {
  27         Name string
  28         Sink chan []byte
  29         dev  io.ReadWriter
  30 }
  31
  32 var (
  33         taps = make(map[string]*TAP)
  34 )
  35
  36 // NewTAP creates a new TUN/TAP virtual interface
  37 func NewTAP(ifaceName string, mtu int) (*TAP, error) {
  38         tapRaw, err := newTAPer(ifaceName)
  39         if err != nil {
  40                 return nil, err
  41         }
  42         tap := TAP{
  43                 Name: ifaceName,
  44                 dev:  tapRaw,
  45                 Sink: make(chan []byte),
  46         }
  47         go func() {
  48                 var n int
  49                 var err error
  50                 var buf []byte
  51                 buf0 := make([]byte, mtu)
  52                 buf1 := make([]byte, mtu)
  53                 bufZ := false
  54                 for {
  55                         if bufZ {
  56                                 buf = buf0
  57                         } else {
  58                                 buf = buf1
  59                         }
  60                         bufZ = !bufZ
  61                         n, err = tap.dev.Read(buf)
  62                         if err != nil {
  63                                 panic("Reading TUN/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 // TAPListen opens an existing TAP (creates if none exists)
  76 func TAPListen(ifaceName string, mtu int) (*TAP, error) {
  77         tap, exists := taps[ifaceName]
  78         if exists {
  79                 return tap, nil
  80         }
  81         tap, err := NewTAP(ifaceName, mtu)
  82         if err != nil {
  83                 return nil, err
  84         }
  85         taps[ifaceName] = tap
  86         return tap, nil
  87 }