Code: refresh code

listener/listener.go

@@ -3,16 +3,20 @@ package proxy
 import (
 	"fmt"
 	"net"
+	"runtime"
 	"strconv"
 	"sync"
 
 	"github.com/Dreamacro/clash/adapter/inbound"
+	"github.com/Dreamacro/clash/config"
 	C "github.com/Dreamacro/clash/constant"
 	"github.com/Dreamacro/clash/listener/http"
 	"github.com/Dreamacro/clash/listener/mixed"
 	"github.com/Dreamacro/clash/listener/redir"
 	"github.com/Dreamacro/clash/listener/socks"
 	"github.com/Dreamacro/clash/listener/tproxy"
+	"github.com/Dreamacro/clash/listener/tun"
+	"github.com/Dreamacro/clash/listener/tun/ipstack"
 	"github.com/Dreamacro/clash/log"
 )
 
@@ -29,6 +33,7 @@ var (
 	tproxyUDPListener *tproxy.UDPListener
 	mixedListener     *mixed.Listener
 	mixedUDPLister    *socks.UDPListener
+	tunAdapter        ipstack.TunAdapter
 
 	// lock for recreate function
 	socksMux  sync.Mutex
@@ -36,6 +41,7 @@ var (
 	redirMux  sync.Mutex
 	tproxyMux sync.Mutex
 	mixedMux  sync.Mutex
+	tunMux    sync.Mutex
 )
 
 type Ports struct {
@@ -58,6 +64,18 @@ func SetAllowLan(al bool) {
 	allowLan = al
 }
 
+func Tun() config.Tun {
+	if tunAdapter == nil {
+		return config.Tun{}
+	}
+	return config.Tun{
+		Enable:    true,
+		Stack:     tunAdapter.Stack(),
+		DNSListen: tunAdapter.DNSListen(),
+		AutoRoute: tunAdapter.AutoRoute(),
+	}
+}
+
 func SetBindAddress(host string) {
 	bindAddress = host
 }
@@ -275,6 +293,25 @@ func ReCreateMixed(port int, tcpIn chan<- C.ConnContext, udpIn chan<- *inbound.P
 	return nil
 }
 
+func ReCreateTun(conf config.Tun, tcpIn chan<- C.ConnContext, udpIn chan<- *inbound.PacketAdapter) error {
+	tunMux.Lock()
+	defer tunMux.Unlock()
+
+	if tunAdapter != nil {
+		tunAdapter.Close()
+		tunAdapter = nil
+	}
+
+	if !conf.Enable {
+		return nil
+	}
+
+	var err error
+	tunAdapter, err = tun.New(conf, tcpIn, udpIn)
+
+	return err
+}
+
 // GetPorts return the ports of proxy servers
 func GetPorts() *Ports {
 	ports := &Ports{}
@@ -330,3 +367,12 @@ func genAddr(host string, port int, allowLan bool) string {
 
 	return fmt.Sprintf("127.0.0.1:%d", port)
 }
+
+// CleanUp clean up something
+func CleanUp() {
+	if runtime.GOOS == "windows" {
+		if tunAdapter != nil {
+			tunAdapter.Close()
+		}
+	}
+}