wireguard-go/tun.go

61 lines
1.6 KiB
Go
Raw Normal View History

package main
import (
2018-02-09 17:56:00 +00:00
"git.zx2c4.com/wireguard-go/internal/events"
2017-11-14 17:26:28 +00:00
"os"
"sync/atomic"
)
const DefaultMTU = 1420
const (
TUNEventUp = 1 << iota
TUNEventDown
TUNEventMTUUpdate
)
type TUNDevice interface {
File() *os.File // returns the file descriptor of the device
Read([]byte, int) (int, error) // read a packet from the device (without any additional headers)
Write([]byte, int) (int, error) // writes a packet to the device (without any additional headers)
MTU() (int, error) // returns the MTU of the device
Name() string // returns the current name
2018-02-09 17:56:00 +00:00
Events() chan events.Event // returns a constant channel of events related to the device
Close() error // stops the device and closes the event channel
}
func (device *Device) RoutineTUNEventReader() {
logInfo := device.log.Info
logError := device.log.Error
for event := range device.tun.device.Events() {
2018-02-09 17:56:00 +00:00
if event.Contains(TUNEventMTUUpdate) {
mtu, err := device.tun.device.MTU()
old := atomic.LoadInt32(&device.tun.mtu)
if err != nil {
logError.Println("Failed to load updated MTU of device:", err)
} else if int(old) != mtu {
if mtu+MessageTransportSize > MaxMessageSize {
logInfo.Println("MTU updated:", mtu, "(too large)")
} else {
logInfo.Println("MTU updated:", mtu)
}
2017-08-22 15:22:45 +00:00
atomic.StoreInt32(&device.tun.mtu, int32(mtu))
}
}
2018-02-09 17:56:00 +00:00
if event.Contains(TUNEventUp) && !device.isUp.Get() {
2017-12-29 16:42:09 +00:00
logInfo.Println("Interface set up")
device.Up()
}
2018-02-09 17:56:00 +00:00
if event.Contains(TUNEventDown) && device.isUp.Get() {
2017-12-29 16:42:09 +00:00
logInfo.Println("Interface set down")
device.Down()
}
2018-02-09 17:56:00 +00:00
event.Processed()
}
}