修改了错误的项目名称

quanguanyu@qq.com · quanguanyu@qq.com · commit afc99a784d34 · 2020-08-23T20:51:29.000+08:00
diff --git a/README.md b/README.md
@@ -2,3 +2,61 @@
 
 
 通过系统调用的方式实现Windows系统的代理配置。
+
+
+修改自 C++项目,`sysproxy.exe`
+
+- [Noisyfox/sysproxy](https://github.com/Noisyfox/sysproxy)
+
+
+## Quick Start
+
+获取
+```
+go get -u github.com/Trisia/gosysproxy
+```
+
+API
+```go
+// SetPAC 设置PAC代理模式
+// scriptLoc: 脚本地址，如: "http://127.0.0.1:7777/pac"
+func SetPAC(scriptLoc string)
+
+// SetGlobalProxy 设置全局代理
+// proxyServer: 代理服务器host:port，例如: "127.0.0.1:7890"
+// bypass: 忽略代理列表,这些配置项开头的地址不进行代理
+func SetGlobalProxy(proxyServer string, bypasses ...string) error
+
+// Off 关闭代理
+func Off() error
+
+// Flush 更新系统配置使生效
+func Flush()
+```
+
+## Demo
+
+```go
+package main
+
+import (
+	"github.com/Trisia/gosysproxy"
+	"time"
+)
+
+func main() {
+    // 设置全局代理
+    err := gosysproxy.SetGlobalProxy("127.0.0.1:7890")
+    if err{
+        panic(err)
+    }
+    
+    time.Sleep(time.Second * 60)
+    
+    err := gosysproxy.Off()
+    if err{
+        panic(err)
+    }
+}
+
+```
diff --git a/main.go b/main.go
@@ -1,6 +1,212 @@
 // +build windows
 
 // windows系统代理配置
-package gosysporxy
+package gosysproxy
 
+import (
+	"errors"
+	"fmt"
+	"math/big"
+	"strings"
+	"syscall"
+	"unsafe"
+)
 
+var (
+	wininet, _           = syscall.LoadLibrary("Wininet.dll")
+	internetSetOption, _ = syscall.GetProcAddress(wininet, "InternetSetOptionW")
+)
+
+const (
+	_INTERNET_OPTION_PER_CONNECTION_OPTION  = 75
+	_INTERNET_OPTION_PROXY_SETTINGS_CHANGED = 95
+	_INTERNET_OPTION_REFRESH                = 37
+)
+
+const (
+	_PROXY_TYPE_DIRECT         = 0x00000001 // direct to net
+	_PROXY_TYPE_PROXY          = 0x00000002 // via named proxy
+	_PROXY_TYPE_AUTO_PROXY_URL = 0x00000004 // autoproxy URL
+	_PROXY_TYPE_AUTO_DETECT    = 0x00000008 // use autoproxy detection
+)
+
+const (
+	_INTERNET_PER_CONN_FLAGS                        = 1
+	_INTERNET_PER_CONN_PROXY_SERVER                 = 2
+	_INTERNET_PER_CONN_PROXY_BYPASS                 = 3
+	_INTERNET_PER_CONN_AUTOCONFIG_URL               = 4
+	_INTERNET_PER_CONN_AUTODISCOVERY_FLAGS          = 5
+	_INTERNET_PER_CONN_AUTOCONFIG_SECONDARY_URL     = 6
+	_INTERNET_PER_CONN_AUTOCONFIG_RELOAD_DELAY_MINS = 7
+	_INTERNET_PER_CONN_AUTOCONFIG_LAST_DETECT_TIME  = 8
+	_INTERNET_PER_CONN_AUTOCONFIG_LAST_DETECT_URL   = 9
+	_INTERNET_PER_CONN_FLAGS_UI                     = 10
+)
+
+type internetPerConnOptionList struct {
+	dwSize        uint32
+	pszConnection *uint16
+	dwOptionCount uint32
+	dwOptionError uint32
+	pOptions      uintptr
+}
+
+type internetPreConnOption struct {
+	dwOption uint32
+	value    uint64
+}
+
+// stringPtrAddr 获取C字符串(UTF16)的数组第一个位置的地址
+func stringPtrAddr(str string) (uint64, error) {
+	scriptLocPtr, err := syscall.UTF16PtrFromString(str)
+	if err != nil {
+		return 0, err
+	}
+	n := new(big.Int)
+	n.SetString(fmt.Sprintf("%x\n", scriptLocPtr), 16)
+	return n.Uint64(), nil
+}
+
+// newParam 创建参数序列容器
+// n: 参数数量
+func newParam(n int) internetPerConnOptionList {
+	return internetPerConnOptionList{
+		dwSize:        4,
+		pszConnection: nil,
+		dwOptionCount: uint32(n),
+		dwOptionError: 0,
+		pOptions:      0,
+	}
+}
+
+// SetPAC 设置PAC代理模式
+// scriptLoc: 脚本地址，如: "http://127.0.0.1:7777/pac"
+func SetPAC(scriptLoc string) error {
+	if scriptLoc == "" {
+		return errors.New("PAC脚本地址(scriptLoc)配置为空")
+	}
+
+	scriptLocAddr, err := stringPtrAddr(scriptLoc)
+	if err != nil {
+		return err
+	}
+
+	param := newParam(2)
+	// 利用Golang数组地址空间连续模拟 malloc
+	options := []internetPreConnOption{
+		{dwOption: _INTERNET_PER_CONN_FLAGS, value: _PROXY_TYPE_AUTO_PROXY_URL | _PROXY_TYPE_DIRECT},
+		{dwOption: _INTERNET_PER_CONN_AUTOCONFIG_URL, value: scriptLocAddr},
+	}
+	param.pOptions = uintptr(unsafe.Pointer(&options[0]))
+	ret, _, infoPtr := syscall.Syscall6(internetSetOption,
+		4,
+		0,
+		_INTERNET_OPTION_PER_CONNECTION_OPTION,
+		uintptr(unsafe.Pointer(&param)),
+		unsafe.Sizeof(param),
+		0, 0)
+	// fmt.Printf(">> Ret [%d] Setting options: %s\n", ret, info)
+	if ret != 1 {
+		return errors.New(fmt.Sprintf("%s", infoPtr))
+	}
+
+	return Flush()
+}
+
+// SetGlobalProxy 设置全局代理
+// proxyServer: 代理服务器host:port，例如: "127.0.0.1:7890"
+// bypass: 忽略代理列表,这些配置项开头的地址不进行代理
+func SetGlobalProxy(proxyServer string, bypasses ...string) error {
+	if proxyServer == "" {
+		return errors.New("代理服务器(proxyServer)配置为空")
+	}
+
+	proxyServerPtrAddr, err := stringPtrAddr(proxyServer)
+	if err != nil {
+		return err
+	}
+
+	var bypassBuilder strings.Builder
+	// 地址过滤配置
+	if bypasses != nil {
+		for _, item := range bypasses {
+			bypassBuilder.WriteString(item)
+			bypassBuilder.WriteByte(';')
+		}
+	} else {
+		bypassBuilder.WriteString("<local>")
+	}
+	bypassAddr, err := stringPtrAddr(bypassBuilder.String())
+	if err != nil {
+		return err
+	}
+
+	param := newParam(3)
+	options := []internetPreConnOption{
+		{dwOption: _INTERNET_PER_CONN_FLAGS, value: _PROXY_TYPE_PROXY | _PROXY_TYPE_DIRECT},
+		{dwOption: _INTERNET_PER_CONN_PROXY_SERVER, value: proxyServerPtrAddr},
+		{dwOption: _INTERNET_PER_CONN_PROXY_BYPASS, value: bypassAddr},
+	}
+	param.pOptions = uintptr(unsafe.Pointer(&options[0]))
+	ret, _, infoPtr := syscall.Syscall6(internetSetOption,
+		4,
+		0,
+		_INTERNET_OPTION_PER_CONNECTION_OPTION,
+		uintptr(unsafe.Pointer(&param)),
+		unsafe.Sizeof(param),
+		0, 0)
+	// fmt.Printf(">> Ret [%d] Setting options: %s\n", ret, infoPtr)
+	if ret != 1 {
+		return errors.New(fmt.Sprintf("%s", infoPtr))
+	}
+
+	return Flush()
+}
+
+// Off 关闭代理
+func Off() error {
+	param := newParam(1)
+	option := internetPreConnOption{
+		dwOption: _INTERNET_PER_CONN_FLAGS,
+		//value:    _PROXY_TYPE_AUTO_DETECT | _PROXY_TYPE_DIRECT}
+		value: _PROXY_TYPE_DIRECT}
+	param.pOptions = uintptr(unsafe.Pointer(&option))
+	ret, _, infoPtr := syscall.Syscall6(internetSetOption,
+		4,
+		0,
+		_INTERNET_OPTION_PER_CONNECTION_OPTION,
+		uintptr(unsafe.Pointer(&param)),
+		unsafe.Sizeof(param),
+		0, 0)
+	// fmt.Printf(">> Ret [%d] Setting options: %s\n", ret, info)
+	if ret != 1 {
+		return errors.New(fmt.Sprintf("%s", infoPtr))
+	}
+	return Flush()
+}
+
+// Flush 更新系统配置使生效
+func Flush() error {
+	ret, _, infoPtr := syscall.Syscall6(internetSetOption,
+		4,
+		0,
+		_INTERNET_OPTION_PROXY_SETTINGS_CHANGED,
+		0, 0,
+		0, 0)
+	// fmt.Println(">> Propagating changes:", fmt.Sprintf("%s", errno))
+	if ret != 1 {
+		return errors.New(fmt.Sprintf("%s", infoPtr))
+	}
+
+	ret, _, infoPtr = syscall.Syscall6(internetSetOption,
+		4,
+		0,
+		_INTERNET_OPTION_REFRESH,
+		0, 0,
+		0, 0)
+	// fmt.Println(">> Refreshing:", errno)
+	if ret != 1 {
+		return errors.New(fmt.Sprintf("%s", infoPtr))
+	}
+	return nil
+}
diff --git a/main_test.go b/main_test.go
@@ -0,0 +1,48 @@
+// +build windows
+
+package gosysproxy
+
+import "testing"
+
+func TestFlush(t *testing.T) {
+	err := Flush()
+	if err != nil {
+		t.Fatal(err)
+	}
+}
+
+func TestOff(t *testing.T) {
+	err := Off()
+	if err != nil {
+		t.Fatal(err)
+	}
+}
+
+func TestSetPAC(t *testing.T) {
+	err := SetPAC("http://127.0.0.1:7777/pac")
+	if err != nil {
+		t.Fatal(err)
+	}
+}
+
+func TestSetGlobalProxy(t *testing.T) {
+	err := SetGlobalProxy("127.0.0.1:7890")
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	err = Off()
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	err = SetGlobalProxy("127.0.0.1:7890", "foo", "bar")
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	err = Off()
+	if err != nil {
+		t.Fatal(err)
+	}
+}
