Browse Source

chore: rename DeviceCreator

AkinoKaede 2 years ago
parent
commit
4276ee9464
3 changed files with 5 additions and 4 deletions
  1. 1 1
      app/tun/device/device.go
  2. 2 0
      app/tun/stack.go
  3. 2 3
      app/tun/tun.go

+ 1 - 1
app/tun/device/device.go

@@ -16,4 +16,4 @@ type Options struct {
 	MTU  uint32
 }
 
-type NewTUNFunc func(Options) (Device, error)
+type DeviceCreator func(Options) (Device, error)

+ 2 - 0
app/tun/stack.go

@@ -9,6 +9,8 @@ import (
 	"gvisor.dev/gvisor/pkg/tcpip/transport/udp"
 )
 
+type StackOption func(*stack.Stack) error
+
 func CreateStack(_ stack.LinkEndpoint) (*stack.Stack, error) {
 	s := stack.New(stack.Options{
 		NetworkProtocols: []stack.NetworkProtocolFactory{

+ 2 - 3
app/tun/tun.go

@@ -31,9 +31,8 @@ func (t *TUN) Type() interface{} {
 }
 
 func (t *TUN) Start() error {
-	var newDeviceFunc device.NewTUNFunc
-	newDeviceFunc = tun.New
-	device, err := newDeviceFunc(device.Options{
+	DeviceCreator := tun.New
+	device, err := DeviceCreator(device.Options{
 		Name: t.config.Name,
 		MTU:  t.config.Mtu,
 	})