mirror of
https://github.com/MetaCubeX/mihomo.git
synced 2024-08-20 09:24:07 +00:00
290 lines
8.3 KiB
Go
290 lines
8.3 KiB
Go
package route
|
|
|
|
import (
|
|
"net/http"
|
|
"net/netip"
|
|
"path/filepath"
|
|
"sync"
|
|
|
|
"github.com/Dreamacro/clash/component/dialer"
|
|
"github.com/Dreamacro/clash/component/resolver"
|
|
"github.com/Dreamacro/clash/config"
|
|
"github.com/Dreamacro/clash/constant"
|
|
C "github.com/Dreamacro/clash/constant"
|
|
"github.com/Dreamacro/clash/hub/executor"
|
|
P "github.com/Dreamacro/clash/listener"
|
|
"github.com/Dreamacro/clash/log"
|
|
"github.com/Dreamacro/clash/tunnel"
|
|
|
|
"github.com/go-chi/chi/v5"
|
|
"github.com/go-chi/render"
|
|
)
|
|
|
|
var (
|
|
updateGeoMux sync.Mutex
|
|
updatingGeo = false
|
|
)
|
|
|
|
func configRouter() http.Handler {
|
|
r := chi.NewRouter()
|
|
r.Get("/", getConfigs)
|
|
r.Put("/", updateConfigs)
|
|
r.Post("/geo", updateGeoDatabases)
|
|
r.Patch("/", patchConfigs)
|
|
return r
|
|
}
|
|
|
|
type configSchema struct {
|
|
Port *int `json:"port"`
|
|
SocksPort *int `json:"socks-port"`
|
|
RedirPort *int `json:"redir-port"`
|
|
TProxyPort *int `json:"tproxy-port"`
|
|
MixedPort *int `json:"mixed-port"`
|
|
Tun *tunSchema `json:"tun"`
|
|
AllowLan *bool `json:"allow-lan"`
|
|
BindAddress *string `json:"bind-address"`
|
|
Mode *tunnel.TunnelMode `json:"mode"`
|
|
LogLevel *log.LogLevel `json:"log-level"`
|
|
IPv6 *bool `json:"ipv6"`
|
|
Sniffing *bool `json:"sniffing"`
|
|
TcpConcurrent *bool `json:"tcp-concurrent"`
|
|
InterfaceName *string `json:"interface-name"`
|
|
}
|
|
|
|
type tunSchema struct {
|
|
Enable bool `yaml:"enable" json:"enable"`
|
|
Device *string `yaml:"device" json:"device"`
|
|
Stack *C.TUNStack `yaml:"stack" json:"stack"`
|
|
DNSHijack *[]netip.AddrPort `yaml:"dns-hijack" json:"dns-hijack"`
|
|
AutoRoute *bool `yaml:"auto-route" json:"auto-route"`
|
|
AutoDetectInterface *bool `yaml:"auto-detect-interface" json:"auto-detect-interface"`
|
|
//RedirectToTun []string `yaml:"-" json:"-"`
|
|
|
|
MTU *uint32 `yaml:"mtu" json:"mtu,omitempty"`
|
|
//Inet4Address *[]config.ListenPrefix `yaml:"inet4-address" json:"inet4_address,omitempty"`
|
|
Inet6Address *[]config.ListenPrefix `yaml:"inet6-address" json:"inet6_address,omitempty"`
|
|
StrictRoute *bool `yaml:"strict-route" json:"strict_route,omitempty"`
|
|
Inet4RouteAddress *[]config.ListenPrefix `yaml:"inet4_route_address" json:"inet4_route_address,omitempty"`
|
|
Inet6RouteAddress *[]config.ListenPrefix `yaml:"inet6_route_address" json:"inet6_route_address,omitempty"`
|
|
IncludeUID *[]uint32 `yaml:"include-uid" json:"include_uid,omitempty"`
|
|
IncludeUIDRange *[]string `yaml:"include-uid-range" json:"include_uid_range,omitempty"`
|
|
ExcludeUID *[]uint32 `yaml:"exclude-uid" json:"exclude_uid,omitempty"`
|
|
ExcludeUIDRange *[]string `yaml:"exclude-uid-range" json:"exclude_uid_range,omitempty"`
|
|
IncludeAndroidUser *[]int `yaml:"include-android-user" json:"include_android_user,omitempty"`
|
|
IncludePackage *[]string `yaml:"include-package" json:"include_package,omitempty"`
|
|
ExcludePackage *[]string `yaml:"exclude-package" json:"exclude_package,omitempty"`
|
|
EndpointIndependentNat *bool `yaml:"endpoint-independent-nat" json:"endpoint_independent_nat,omitempty"`
|
|
UDPTimeout *int64 `yaml:"udp-timeout" json:"udp_timeout,omitempty"`
|
|
}
|
|
|
|
func getConfigs(w http.ResponseWriter, r *http.Request) {
|
|
general := executor.GetGeneral()
|
|
render.JSON(w, r, general)
|
|
}
|
|
|
|
func pointerOrDefault(p *int, def int) int {
|
|
if p != nil {
|
|
return *p
|
|
}
|
|
|
|
return def
|
|
}
|
|
|
|
func pointerOrDefaultTun(p *tunSchema, def config.Tun) config.Tun {
|
|
if p != nil {
|
|
def.Enable = p.Enable
|
|
if p.Device != nil {
|
|
def.Device = *p.Device
|
|
}
|
|
if p.Stack != nil {
|
|
def.Stack = *p.Stack
|
|
}
|
|
if p.DNSHijack != nil {
|
|
def.DNSHijack = *p.DNSHijack
|
|
}
|
|
if p.AutoRoute != nil {
|
|
def.AutoRoute = *p.AutoRoute
|
|
}
|
|
if p.AutoDetectInterface != nil {
|
|
def.AutoDetectInterface = *p.AutoDetectInterface
|
|
}
|
|
if p.MTU != nil {
|
|
def.MTU = *p.MTU
|
|
}
|
|
//if p.Inet4Address != nil {
|
|
// def.Inet4Address = *p.Inet4Address
|
|
//}
|
|
if p.Inet6Address != nil {
|
|
def.Inet6Address = *p.Inet6Address
|
|
}
|
|
if p.IncludeUID != nil {
|
|
def.IncludeUID = *p.IncludeUID
|
|
}
|
|
if p.IncludeUIDRange != nil {
|
|
def.IncludeUIDRange = *p.IncludeUIDRange
|
|
}
|
|
if p.ExcludeUID != nil {
|
|
def.ExcludeUID = *p.ExcludeUID
|
|
}
|
|
if p.ExcludeUIDRange != nil {
|
|
def.ExcludeUIDRange = *p.ExcludeUIDRange
|
|
}
|
|
if p.IncludeAndroidUser != nil {
|
|
def.IncludeAndroidUser = *p.IncludeAndroidUser
|
|
}
|
|
if p.IncludePackage != nil {
|
|
def.IncludePackage = *p.IncludePackage
|
|
}
|
|
if p.ExcludePackage != nil {
|
|
def.ExcludePackage = *p.ExcludePackage
|
|
}
|
|
if p.EndpointIndependentNat != nil {
|
|
def.EndpointIndependentNat = *p.EndpointIndependentNat
|
|
}
|
|
if p.UDPTimeout != nil {
|
|
def.UDPTimeout = *p.UDPTimeout
|
|
}
|
|
}
|
|
return def
|
|
}
|
|
|
|
func patchConfigs(w http.ResponseWriter, r *http.Request) {
|
|
general := &configSchema{}
|
|
if err := render.DecodeJSON(r.Body, general); err != nil {
|
|
render.Status(r, http.StatusBadRequest)
|
|
render.JSON(w, r, ErrBadRequest)
|
|
return
|
|
}
|
|
|
|
if general.AllowLan != nil {
|
|
P.SetAllowLan(*general.AllowLan)
|
|
}
|
|
|
|
if general.BindAddress != nil {
|
|
P.SetBindAddress(*general.BindAddress)
|
|
}
|
|
|
|
if general.Sniffing != nil {
|
|
tunnel.SetSniffing(*general.Sniffing)
|
|
}
|
|
|
|
if general.TcpConcurrent != nil {
|
|
dialer.SetDial(*general.TcpConcurrent)
|
|
}
|
|
|
|
if general.InterfaceName != nil {
|
|
dialer.DefaultInterface.Store(*general.InterfaceName)
|
|
}
|
|
|
|
ports := P.GetPorts()
|
|
|
|
tcpIn := tunnel.TCPIn()
|
|
udpIn := tunnel.UDPIn()
|
|
|
|
P.ReCreateHTTP(pointerOrDefault(general.Port, ports.Port), tcpIn)
|
|
P.ReCreateSocks(pointerOrDefault(general.SocksPort, ports.SocksPort), tcpIn, udpIn)
|
|
P.ReCreateRedir(pointerOrDefault(general.RedirPort, ports.RedirPort), tcpIn, udpIn)
|
|
P.ReCreateTProxy(pointerOrDefault(general.TProxyPort, ports.TProxyPort), tcpIn, udpIn)
|
|
P.ReCreateMixed(pointerOrDefault(general.MixedPort, ports.MixedPort), tcpIn, udpIn)
|
|
P.ReCreateTun(pointerOrDefaultTun(general.Tun, P.LastTunConf), tcpIn, udpIn)
|
|
|
|
if general.Mode != nil {
|
|
tunnel.SetMode(*general.Mode)
|
|
}
|
|
|
|
if general.LogLevel != nil {
|
|
log.SetLevel(*general.LogLevel)
|
|
}
|
|
|
|
if general.IPv6 != nil {
|
|
resolver.DisableIPv6 = !*general.IPv6
|
|
}
|
|
|
|
render.NoContent(w, r)
|
|
}
|
|
|
|
type updateConfigRequest struct {
|
|
Path string `json:"path"`
|
|
Payload string `json:"payload"`
|
|
}
|
|
|
|
func updateConfigs(w http.ResponseWriter, r *http.Request) {
|
|
req := updateConfigRequest{}
|
|
if err := render.DecodeJSON(r.Body, &req); err != nil {
|
|
render.Status(r, http.StatusBadRequest)
|
|
render.JSON(w, r, ErrBadRequest)
|
|
return
|
|
}
|
|
|
|
force := r.URL.Query().Get("force") == "true"
|
|
var cfg *config.Config
|
|
var err error
|
|
|
|
if req.Payload != "" {
|
|
cfg, err = executor.ParseWithBytes([]byte(req.Payload))
|
|
if err != nil {
|
|
render.Status(r, http.StatusBadRequest)
|
|
render.JSON(w, r, newError(err.Error()))
|
|
return
|
|
}
|
|
} else {
|
|
if req.Path == "" {
|
|
req.Path = constant.Path.Config()
|
|
}
|
|
if !filepath.IsAbs(req.Path) {
|
|
render.Status(r, http.StatusBadRequest)
|
|
render.JSON(w, r, newError("path is not a absolute path"))
|
|
return
|
|
}
|
|
|
|
cfg, err = executor.ParseWithPath(req.Path)
|
|
if err != nil {
|
|
render.Status(r, http.StatusBadRequest)
|
|
render.JSON(w, r, newError(err.Error()))
|
|
return
|
|
}
|
|
}
|
|
|
|
executor.ApplyConfig(cfg, force)
|
|
render.NoContent(w, r)
|
|
}
|
|
|
|
func updateGeoDatabases(w http.ResponseWriter, r *http.Request) {
|
|
updateGeoMux.Lock()
|
|
|
|
if updatingGeo {
|
|
updateGeoMux.Unlock()
|
|
render.Status(r, http.StatusBadRequest)
|
|
render.JSON(w, r, newError("updating..."))
|
|
return
|
|
}
|
|
|
|
updatingGeo = true
|
|
updateGeoMux.Unlock()
|
|
|
|
go func() {
|
|
defer func() {
|
|
updatingGeo = false
|
|
}()
|
|
|
|
log.Warnln("[REST-API] updating GEO databases...")
|
|
|
|
if err := config.UpdateGeoDatabases(); err != nil {
|
|
log.Errorln("[REST-API] update GEO databases failed: %v", err)
|
|
return
|
|
}
|
|
|
|
cfg, err := executor.ParseWithPath(constant.Path.Config())
|
|
if err != nil {
|
|
log.Errorln("[REST-API] update GEO databases failed: %v", err)
|
|
return
|
|
}
|
|
|
|
log.Warnln("[REST-API] update GEO databases successful, apply config...")
|
|
|
|
executor.ApplyConfig(cfg, false)
|
|
}()
|
|
|
|
render.NoContent(w, r)
|
|
}
|