Fix windows route

This commit is contained in:
世界 2022-09-29 18:59:33 +08:00
parent 6bad0c2380
commit a93592a9b5
No known key found for this signature in database
GPG key ID: CD109927C34A63C4
9 changed files with 662 additions and 26 deletions

View file

@ -0,0 +1,158 @@
//go:build windows
package winsys
import (
"golang.org/x/sys/windows"
)
const (
AF_INET = 2
AF_INET6 = 23
)
const (
MAX_MODULE_NAME32 = 255
MAX_PATH = 260
)
// https://docs.microsoft.com/en-us/windows/desktop/api/tlhelp32/nf-tlhelp32-createtoolhelp32snapshot
const (
TH32CS_SNAPHEAPLIST = 0x00000001
TH32CS_SNAPPROCESS = 0x00000002
TH32CS_SNAPTHREAD = 0x00000004
TH32CS_SNAPMODULE = 0x00000008
TH32CS_SNAPMODULE32 = 0x00000010
TH32CS_INHERIT = 0x80000000
TH32CS_SNAPALL = TH32CS_SNAPHEAPLIST | TH32CS_SNAPMODULE | TH32CS_SNAPPROCESS | TH32CS_SNAPTHREAD
)
const (
MAX_ADAPTER_NAME = 128
MAX_INTERFACE_NAME_LEN = 256
MAXLEN_PHYSADDR = 8
MAXLEN_IFDESCR = 256
)
const (
ERROR_INSUFFICIENT_BUFFER = 122
)
const (
RPC_C_AUTHN_DEFAULT uint32 = 0xFFFFFFFF
FWPM_SESSION_FLAG_DYNAMIC uint32 = 0x00000001
)
const (
FWP_MATCH_EQUAL uint32 = 0
FWP_MATCH_GREATER = (FWP_MATCH_EQUAL + 1)
FWP_MATCH_LESS = (FWP_MATCH_GREATER + 1)
FWP_MATCH_GREATER_OR_EQUAL = (FWP_MATCH_LESS + 1)
FWP_MATCH_LESS_OR_EQUAL = (FWP_MATCH_GREATER_OR_EQUAL + 1)
FWP_MATCH_RANGE = (FWP_MATCH_LESS_OR_EQUAL + 1)
FWP_MATCH_FLAGS_ALL_SET = (FWP_MATCH_RANGE + 1)
FWP_MATCH_FLAGS_ANY_SET = (FWP_MATCH_FLAGS_ALL_SET + 1)
FWP_MATCH_FLAGS_NONE_SET = (FWP_MATCH_FLAGS_ANY_SET + 1)
FWP_MATCH_EQUAL_CASE_INSENSITIVE = (FWP_MATCH_FLAGS_NONE_SET + 1)
FWP_MATCH_NOT_EQUAL = (FWP_MATCH_EQUAL_CASE_INSENSITIVE + 1)
FWP_MATCH_PREFIX = (FWP_MATCH_NOT_EQUAL + 1)
FWP_MATCH_NOT_PREFIX = (FWP_MATCH_PREFIX + 1)
FWP_MATCH_TYPE_MAX = (FWP_MATCH_NOT_PREFIX + 1)
)
const (
FWP_EMPTY uint32 = 0
FWP_UINT8 = (FWP_EMPTY + 1)
FWP_UINT16 = (FWP_UINT8 + 1)
FWP_UINT32 = (FWP_UINT16 + 1)
FWP_UINT64 = (FWP_UINT32 + 1)
FWP_INT8 = (FWP_UINT64 + 1)
FWP_INT16 = (FWP_INT8 + 1)
FWP_INT32 = (FWP_INT16 + 1)
FWP_INT64 = (FWP_INT32 + 1)
FWP_FLOAT = (FWP_INT64 + 1)
FWP_DOUBLE = (FWP_FLOAT + 1)
FWP_BYTE_ARRAY16_TYPE = (FWP_DOUBLE + 1)
FWP_BYTE_BLOB_TYPE = (FWP_BYTE_ARRAY16_TYPE + 1)
FWP_SID = (FWP_BYTE_BLOB_TYPE + 1)
FWP_SECURITY_DESCRIPTOR_TYPE = (FWP_SID + 1)
FWP_TOKEN_INFORMATION_TYPE = (FWP_SECURITY_DESCRIPTOR_TYPE + 1)
FWP_TOKEN_ACCESS_INFORMATION_TYPE = (FWP_TOKEN_INFORMATION_TYPE + 1)
FWP_UNICODE_STRING_TYPE = (FWP_TOKEN_ACCESS_INFORMATION_TYPE + 1)
FWP_BYTE_ARRAY6_TYPE = (FWP_UNICODE_STRING_TYPE + 1)
FWP_BITMAP_INDEX_TYPE = (FWP_BYTE_ARRAY6_TYPE + 1)
FWP_BITMAP_ARRAY64_TYPE = (FWP_BITMAP_INDEX_TYPE + 1)
FWP_SINGLE_DATA_TYPE_MAX = 0xff
FWP_V4_ADDR_MASK = (FWP_SINGLE_DATA_TYPE_MAX + 1)
FWP_V6_ADDR_MASK = (FWP_V4_ADDR_MASK + 1)
FWP_RANGE_TYPE = (FWP_V6_ADDR_MASK + 1)
FWP_DATA_TYPE_MAX = (FWP_RANGE_TYPE + 1)
)
var FWPM_CONDITION_IP_PROTOCOL = windows.GUID{
Data1: 0x3971ef2b,
Data2: 0x623e,
Data3: 0x4f9a,
Data4: [8]byte{0x8c, 0xb1, 0x6e, 0x79, 0xb8, 0x06, 0xb9, 0xa7},
}
var FWPM_CONDITION_IP_REMOTE_PORT = windows.GUID{
Data1: 0xc35a604d,
Data2: 0xd22b,
Data3: 0x4e1a,
Data4: [8]byte{0x91, 0xb4, 0x68, 0xf6, 0x74, 0xee, 0x67, 0x4b},
}
var FWPM_LAYER_ALE_AUTH_CONNECT_V4 = windows.GUID{
Data1: 0xc38d57d1,
Data2: 0x05a7,
Data3: 0x4c33,
Data4: [8]byte{0x90, 0x4f, 0x7f, 0xbc, 0xee, 0xe6, 0x0e, 0x82},
}
var FWPM_CONDITION_LOCAL_INTERFACE_INDEX = windows.GUID{
Data1: 0x667fd755,
Data2: 0xd695,
Data3: 0x434a,
Data4: [8]byte{0x8a, 0xf5, 0xd3, 0x83, 0x5a, 0x12, 0x59, 0xbc},
}
var FWPM_LAYER_ALE_AUTH_CONNECT_V6 = windows.GUID{
Data1: 0x4a72393b,
Data2: 0x319f,
Data3: 0x44bc,
Data4: [8]byte{0x84, 0xc3, 0xba, 0x54, 0xdc, 0xb3, 0xb6, 0xb4},
}
var FWPM_CONDITION_ALE_APP_ID = windows.GUID{
Data1: 0xd78e1e87,
Data2: 0x8644,
Data3: 0x4ea5,
Data4: [8]byte{0x94, 0x37, 0xd8, 0x09, 0xec, 0xef, 0xc9, 0x71},
}
const (
IPPROTO_UDP uint32 = 17
)
const (
FWP_ACTION_FLAG_TERMINATING uint32 = 0x00001000
FWP_ACTION_BLOCK uint32 = (0x00000001 | FWP_ACTION_FLAG_TERMINATING)
FWP_ACTION_PERMIT uint32 = (0x00000002 | FWP_ACTION_FLAG_TERMINATING)
)
const (
FWPM_FILTER_FLAG_NONE = 0x00000000
FWPM_FILTER_FLAG_PERSISTENT = 0x00000001
FWPM_FILTER_FLAG_BOOTTIME = 0x00000002
FWPM_FILTER_FLAG_HAS_PROVIDER_CONTEXT = 0x00000004
FWPM_FILTER_FLAG_CLEAR_ACTION_RIGHT = 0x00000008
FWPM_FILTER_FLAG_PERMIT_IF_CALLOUT_UNREGISTERED = 0x00000010
FWPM_FILTER_FLAG_DISABLED = 0x00000020
FWPM_FILTER_FLAG_INDEXED = 0x00000040
FWPM_FILTER_FLAG_HAS_SECURITY_REALM_PROVIDER_CONTEXT = 0x00000080
FWPM_FILTER_FLAG_SYSTEMOS_ONLY = 0x00000100
FWPM_FILTER_FLAG_GAMEOS_ONLY = 0x00000200
FWPM_FILTER_FLAG_SILENT_MODE = 0x00000400
FWPM_FILTER_FLAG_IPSEC_NO_ACQUIRE_INITIATE = 0x00000800
)

46
internal/winsys/helper.go Normal file
View file

@ -0,0 +1,46 @@
//go:build windows
package winsys
import (
"os"
"unsafe"
"github.com/sagernet/sing/common"
"golang.org/x/sys/windows"
)
func CreateDisplayData(name, description string) FWPM_DISPLAY_DATA0 {
namePtr, err := windows.UTF16PtrFromString(name)
common.Must(err)
descriptionPtr, err := windows.UTF16PtrFromString(description)
common.Must(err)
return FWPM_DISPLAY_DATA0{
Name: namePtr,
Description: descriptionPtr,
}
}
func GetCurrentProcessAppID() (*FWP_BYTE_BLOB, error) {
currentFile, err := os.Executable()
if err != nil {
return nil, err
}
curFilePtr, err := windows.UTF16PtrFromString(currentFile)
if err != nil {
return nil, err
}
windows.GetCurrentProcessId()
var appID *FWP_BYTE_BLOB
err = FwpmGetAppIdFromFileName0(curFilePtr, unsafe.Pointer(&appID))
if err != nil {
return nil, err
}
return appID, nil
}

View file

@ -0,0 +1,5 @@
//go:build windows
package winsys
//go:generate go run golang.org/x/sys/windows/mkwinsyscall -output zsyscall_windows.go syscall_windows.go

View file

@ -0,0 +1,19 @@
package winsys
// https://docs.microsoft.com/en-us/windows/desktop/api/fwpmu/nf-fwpmu-fwpmengineopen0
//sys FwpmEngineOpen0(serverName *uint16, authnService uint32, authIdentity *uintptr, session *FWPM_SESSION0, engineHandle unsafe.Pointer) (err error) [failretval!=0] = fwpuclnt.FwpmEngineOpen0
// https://learn.microsoft.com/en-us/windows/win32/api/fwpmu/nf-fwpmu-fwpmengineclose0
//sys FwpmEngineClose0(engineHandle uintptr) (err error) [failretval!=0] = fwpuclnt.FwpmEngineClose0
// https://docs.microsoft.com/en-us/windows/desktop/api/fwpmu/nf-fwpmu-fwpmsublayeradd0
//sys FwpmSubLayerAdd0(engineHandle uintptr, subLayer *FWPM_SUBLAYER0, sd uintptr) (err error) [failretval!=0] = fwpuclnt.FwpmSubLayerAdd0
// https://docs.microsoft.com/en-us/windows/desktop/api/fwpmu/nf-fwpmu-fwpmfilteradd0
//sys FwpmFilterAdd0(engineHandle uintptr, filter *FWPM_FILTER0, sd uintptr, id *uint64) (err error) [failretval!=0] = fwpuclnt.FwpmFilterAdd0
// https://docs.microsoft.com/en-us/windows/desktop/api/fwpmu/nf-fwpmu-fwpmgetappidfromfilename0
//sys FwpmGetAppIdFromFileName0(fileName *uint16, appID unsafe.Pointer) (err error) [failretval!=0] = fwpuclnt.FwpmGetAppIdFromFileName0
// https://docs.microsoft.com/en-us/windows/desktop/api/fwpmu/nf-fwpmu-fwpmfreememory0
//sys FwpmFreeMemory0(p unsafe.Pointer) = fwpuclnt.FwpmFreeMemory0

View file

@ -0,0 +1,99 @@
package winsys
import (
"golang.org/x/sys/windows"
)
type (
BOOL int32
HANDLE uintptr
DWORD uint32
PDWORD uintptr
ULONG uint32
ULONG_PTR uintptr
HMODULE HANDLE
)
type MIB_IPFORWARDROW struct {
ForwardDest uint32
ForwardMask uint32
ForwardPolicy uint32
ForwardNextHop uint32
ForwardIfIndex uint32
ForwardType uint32
ForwardProto uint32
ForwardAge uint32
ForwardNextHopAS uint32
ForwardMetric1 uint32
ForwardMetric2 uint32
ForwardMetric3 uint32
ForwardMetric4 uint32
ForwardMetric5 uint32
}
type FWPM_DISPLAY_DATA0 struct {
Name *uint16
Description *uint16
}
type FWPM_SESSION0 struct {
SessionKey windows.GUID
DisplayData FWPM_DISPLAY_DATA0
Flags uint32
TxnWaitTimeoutInMSec uint32
ProcessId uint32
Sid *windows.SID
Username *uint16
KernelMode int32
}
type FWP_BYTE_BLOB struct {
size uint32
data *uint8
}
type FWPM_SUBLAYER0 struct {
SubLayerKey windows.GUID // Windows type: GUID
DisplayData FWPM_DISPLAY_DATA0
Flags uint32
ProviderKey *windows.GUID // Windows type: *GUID
ProviderData FWP_BYTE_BLOB
Weight uint16
}
type FWP_VALUE0 struct {
Type uint32
Value uintptr
}
type FWP_CONDITION_VALUE0 FWP_VALUE0
type FWPM_FILTER_CONDITION0 struct {
FieldKey windows.GUID // Windows type: GUID
MatchType uint32
ConditionValue FWP_CONDITION_VALUE0
}
type FWPM_ACTION0 struct {
Type uint32
Value windows.GUID
}
type FWPM_FILTER0 struct {
FilterKey windows.GUID
DisplayData FWPM_DISPLAY_DATA0
Flags uint32
ProviderKey *windows.GUID
ProviderData FWP_BYTE_BLOB
LayerKey windows.GUID
SubLayerKey windows.GUID
Weight FWP_VALUE0
NumFilterConditions uint32
FilterCondition *FWPM_FILTER_CONDITION0
Action FWPM_ACTION0
Offset1 [4]byte
Context windows.GUID
Reserved *windows.GUID
FilterId uint64
EffectiveWeight FWP_VALUE0
}

View file

@ -0,0 +1,94 @@
// Code generated by 'go generate'; DO NOT EDIT.
package winsys
import (
"syscall"
"unsafe"
"golang.org/x/sys/windows"
)
var _ unsafe.Pointer
// Do the interface allocations only once for common
// Errno values.
const (
errnoERROR_IO_PENDING = 997
)
var (
errERROR_IO_PENDING error = syscall.Errno(errnoERROR_IO_PENDING)
errERROR_EINVAL error = syscall.EINVAL
)
// errnoErr returns common boxed Errno values, to prevent
// allocations at runtime.
func errnoErr(e syscall.Errno) error {
switch e {
case 0:
return errERROR_EINVAL
case errnoERROR_IO_PENDING:
return errERROR_IO_PENDING
}
// TODO: add more here, after collecting data on the common
// error values see on Windows. (perhaps when running
// all.bat?)
return e
}
var (
modfwpuclnt = windows.NewLazySystemDLL("fwpuclnt.dll")
procFwpmEngineClose0 = modfwpuclnt.NewProc("FwpmEngineClose0")
procFwpmEngineOpen0 = modfwpuclnt.NewProc("FwpmEngineOpen0")
procFwpmFilterAdd0 = modfwpuclnt.NewProc("FwpmFilterAdd0")
procFwpmFreeMemory0 = modfwpuclnt.NewProc("FwpmFreeMemory0")
procFwpmGetAppIdFromFileName0 = modfwpuclnt.NewProc("FwpmGetAppIdFromFileName0")
procFwpmSubLayerAdd0 = modfwpuclnt.NewProc("FwpmSubLayerAdd0")
)
func FwpmEngineClose0(engineHandle uintptr) (err error) {
r1, _, e1 := syscall.Syscall(procFwpmEngineClose0.Addr(), 1, uintptr(engineHandle), 0, 0)
if r1 != 0 {
err = errnoErr(e1)
}
return
}
func FwpmEngineOpen0(serverName *uint16, authnService uint32, authIdentity *uintptr, session *FWPM_SESSION0, engineHandle unsafe.Pointer) (err error) {
r1, _, e1 := syscall.Syscall6(procFwpmEngineOpen0.Addr(), 5, uintptr(unsafe.Pointer(serverName)), uintptr(authnService), uintptr(unsafe.Pointer(authIdentity)), uintptr(unsafe.Pointer(session)), uintptr(engineHandle), 0)
if r1 != 0 {
err = errnoErr(e1)
}
return
}
func FwpmFilterAdd0(engineHandle uintptr, filter *FWPM_FILTER0, sd uintptr, id *uint64) (err error) {
r1, _, e1 := syscall.Syscall6(procFwpmFilterAdd0.Addr(), 4, uintptr(engineHandle), uintptr(unsafe.Pointer(filter)), uintptr(sd), uintptr(unsafe.Pointer(id)), 0, 0)
if r1 != 0 {
err = errnoErr(e1)
}
return
}
func FwpmFreeMemory0(p unsafe.Pointer) {
syscall.Syscall(procFwpmFreeMemory0.Addr(), 1, uintptr(p), 0, 0)
return
}
func FwpmGetAppIdFromFileName0(fileName *uint16, appID unsafe.Pointer) (err error) {
r1, _, e1 := syscall.Syscall(procFwpmGetAppIdFromFileName0.Addr(), 2, uintptr(unsafe.Pointer(fileName)), uintptr(appID), 0)
if r1 != 0 {
err = errnoErr(e1)
}
return
}
func FwpmSubLayerAdd0(engineHandle uintptr, subLayer *FWPM_SUBLAYER0, sd uintptr) (err error) {
r1, _, e1 := syscall.Syscall(procFwpmSubLayerAdd0.Addr(), 3, uintptr(engineHandle), uintptr(unsafe.Pointer(subLayer)), uintptr(sd))
if r1 != 0 {
err = errnoErr(e1)
}
return
}