mirrors/SpoofDPI
1
0
Fork 0
mirror of https://github.com/xvzc/SpoofDPI.git synced 2024-12-22 06:15:51 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Refactor proxy setting. (#191)

Browse Source
This commit is contained in:
Ledorub 2024-08-25 09:57:38 +03:00 committed by GitHub
parent 03d63162be
commit d397d30fb5
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 50 additions and 31 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
cmd/spoof-dpi/main.go
View File

@ -36,7 +36,7 @@ func main() {
}
if config.SystemProxy {
if err := util.SetOsProxy(config.Port); err != nil {
if err := util.SetOsProxy(uint16(config.Port)); err != nil {
logger.Fatal().Msgf("error while changing proxy settings: %s", err)
}
defer func() {

79
util/os.go
View File

@ -5,15 +5,19 @@ import (
"fmt"
"os/exec"
"runtime"
"strconv"
"strings"
)
const getDefaultNetworkCMD = "networksetup -listnetworkserviceorder | grep" +
" `(route -n get default | grep 'interface' || route -n get -inet6 default | grep 'interface') | cut -d ':' -f2`" +
" -B 1 | head -n 1 | cut -d ' ' -f 2-"
const (
getDefaultNetworkCMD = "networksetup -listnetworkserviceorder | grep" +
" `(route -n get default | grep 'interface' || route -n get -inet6 default | grep 'interface') | cut -d ':' -f2`" +
" -B 1 | head -n 1 | cut -d ' ' -f 2-"
darwinOS = "darwin"
)
func SetOsProxy(port int) error {
if runtime.GOOS != "darwin" {
func SetOsProxy(port uint16) error {
if runtime.GOOS != darwinOS {
return nil
}
@ -22,23 +26,11 @@ func SetOsProxy(port int) error {
return err
}
args := fmt.Sprintf("'%s' 127.0.0.1 %d", network, port)
_, err = exec.Command("sh", "-c", "networksetup -setwebproxy "+args).Output()
if err != nil {
return err
}
_, err = exec.Command("sh", "-c", "networksetup -setsecurewebproxy "+args).Output()
if err != nil {
return err
}
return nil
return setProxy(getProxyTypes(), network, "127.0.0.1", port)
}
func UnsetOsProxy() error {
if runtime.GOOS != "darwin" {
if runtime.GOOS != darwinOS {
return nil
}
@ -47,17 +39,7 @@ func UnsetOsProxy() error {
return err
}
_, err = exec.Command("sh", "-c", "networksetup -setwebproxystate "+"'"+network+"'"+" off").Output()
if err != nil {
return err
}
_, err = exec.Command("sh", "-c", "networksetup -setsecurewebproxystate "+"'"+network+"'"+" off").Output()
if err != nil {
return err
}
return nil
return unsetProxy(getProxyTypes(), network)
}
func getDefaultNetwork() (string, error) {
@ -69,3 +51,40 @@ func getDefaultNetwork() (string, error) {
}
return strings.TrimSpace(string(network)), nil
}
func getProxyTypes() []string {
return []string{"webproxy", "securewebproxy"}
}
func setProxy(proxyTypes []string, network, domain string, port uint16) error {
args := []string{"", network, domain, strconv.FormatUint(uint64(port), 10)}
for _, proxyType := range proxyTypes {
args[0] = "-set" + proxyType
if err := networkSetup(args); err != nil {
return fmt.Errorf("setting %s: %w", proxyType, err)
}
}
return nil
}
func unsetProxy(proxyTypes []string, network string) error {
args := []string{"", network, "off"}
for _, proxyType := range proxyTypes {
args[0] = "-set" + proxyType + "state"
if err := networkSetup(args); err != nil {
return fmt.Errorf("unsetting %s: %w", proxyType, err)
}
}
return nil
}
func networkSetup(args []string) error {
cmd := exec.Command("networksetup", args...)
out, err := cmd.CombinedOutput()
if err != nil {
return fmt.Errorf("%s: %s", cmd.String(), out)
}
return nil
}