sysproxy: add probe for state

5aaee9 · Apr 16, 2024 · e812fbd · e812fbd
1 parent f9d5a46
commit e812fbd
Show file tree

Hide file tree

Showing 5 changed files with 59 additions and 2 deletions.
diff --git a/pkgs/proxy/sysproxy/sysproxy.go b/pkgs/proxy/sysproxy/sysproxy.go
@@ -1,6 +1,12 @@
 package sysproxy
 
+type SystemProxyStatus struct {
+	State bool `json:"state"`
+}
+
 type SystemProxy interface {
 	TurnOn(addrPort string) error
 	TurnOff() error
+
+	Status() (*SystemProxyStatus, error)
 }
diff --git a/pkgs/proxy/sysproxy/sysproxy_darwin.go b/pkgs/proxy/sysproxy/sysproxy_darwin.go
@@ -241,6 +241,10 @@ func (p *DarwinSystemProxy) TurnOff() error {
 	return nil
 }
 
+func (p *DarwinSystemProxy) Status() (*SystemProxyStatus, error) {
+	return nil, nil
+}
+
 func (p *DarwinSystemProxy) TurnOn(addrport string) error {
 	host, port, err := net.SplitHostPort(addrport)
 	if err != nil {

diff --git a/pkgs/proxy/sysproxy/sysproxy_other.go b/pkgs/proxy/sysproxy/sysproxy_other.go
@@ -19,3 +19,7 @@ func (p *OtherSystemProxy) TurnOn(addrport string) error {
 func NewSystemProxy() SystemProxy {
 	return &OtherSystemProxy{}
 }
+
+func (p *OtherSystemProxy) Status() (*SystemProxyStatus, error) {
+	return nil, errors.New("not implemented")
+}
diff --git a/pkgs/proxy/sysproxy/sysproxy_test.go b/pkgs/proxy/sysproxy/sysproxy_test.go
@@ -0,0 +1,15 @@
+package sysproxy
+
+import (
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func Test_SystemProxyStatus(t *testing.T) {
+	proxy := NewSystemProxy()
+
+	status, err := proxy.Status()
+	assert.NoError(t, err)
+	assert.NotNil(t, status)
+}
diff --git a/pkgs/proxy/sysproxy/sysproxy_windows.go b/pkgs/proxy/sysproxy/sysproxy_windows.go
@@ -17,15 +17,17 @@ func NewSystemProxy() SystemProxy {
 var _ SystemProxy = (*WindowsSystemProxy)(nil)
 
 var (
-	modwininet             = windows.NewLazySystemDLL("wininet.dll")
-	procInternetSetOptionW = modwininet.NewProc("InternetSetOptionW")
+	modwininet               = windows.NewLazySystemDLL("wininet.dll")
+	procInternetSetOptionW   = modwininet.NewProc("InternetSetOptionW")
+	procInternetQueryOptionA = modwininet.NewProc("InternetQueryOptionA")
 )
 
 const (
 	internetOptionPerConnectionOption  = 75
 	internetOptionSettingsChanged      = 39
 	internetOptionRefresh              = 37
 	internetOptionProxySettingsChanged = 95
+	internetOptionProxy                = 38
 )
 
 const (
@@ -63,6 +65,12 @@ type internetPerConnOption struct {
 	value    uint64
 }
 
+type internetProxyInfo struct {
+	dwAccessType    uint32
+	lpszProxy       *uint16
+	lpszProxyBypass *uint16
+}
+
 func internetSetOption(option uintptr, lpBuffer uintptr, dwBufferSize uintptr) error {
 	r0, _, err := syscall.SyscallN(procInternetSetOptionW.Addr(), 0, option, lpBuffer, dwBufferSize)
 	if r0 != 1 {
@@ -126,3 +134,23 @@ func (p *WindowsSystemProxy) TurnOn(addrport string) error {
 func (p *WindowsSystemProxy) TurnOff() error {
 	return ClearSystemProxy()
 }
+
+func (p *WindowsSystemProxy) Status() (*SystemProxyStatus, error) {
+	var bufferLength uint32 = 1024 * 10
+	buffer := make([]byte, bufferLength)
+
+	r0, _, err := syscall.SyscallN(procInternetQueryOptionA.Addr(), 0, internetOptionProxy,
+		uintptr(unsafe.Pointer(&buffer[0])), uintptr(unsafe.Pointer(&bufferLength)), 0, 0)
+
+	if r0 != 1 {
+		return nil, err
+	}
+
+	proxyInfo := (*internetProxyInfo)(unsafe.Pointer(&buffer[0]))
+
+	res := &SystemProxyStatus{}
+
+	res.State = proxyInfo.dwAccessType != 1
+
+	return res, nil
+}