feat: add support for getting IP addr by scripts

README.md

@@ -22,7 +22,7 @@
 
 - 支持Mac、Windows、Linux系统，支持ARM、x86架构
 - 支持的域名服务商 `Alidns(阿里云)` `Dnspod(腾讯云)` `Cloudflare` `华为云` `Callback` `百度云` `porkbun` `GoDaddy` `Google Domains`
-- 支持接口/网卡获取IP
+- 支持接口/网卡/命令（Windows：powershell 脚本、其它：bash 脚本）获取IP
 - 支持以服务的方式运行
 - 默认间隔5分钟同步一次
 - 支持多个域名同时解析，公司必备
@@ -33,6 +33,9 @@
 - 支持TTL
 - 支持部分dns服务商传递自定义参数，实现地域解析等功能
 
+> **Note**
+> 建议在启用公网访问时，使用 Nginx 等反向代理软件启用 HTTPS 访问，以保证安全性。
+
 ## 系统中使用
 
 - 下载并解压[https://github.com/jeessy2/ddns-go/releases](https://github.com/jeessy2/ddns-go/releases)

config/config.go

@@ -1,10 +1,12 @@
 package config
 
 import (
-	"io/ioutil"
+	"io"
 	"log"
 	"os"
+	"os/exec"
 	"regexp"
+	"runtime"
 	"strconv"
 	"strings"
 	"sync"
@@ -14,10 +16,10 @@ import (
 )
 
 // Ipv4Reg IPv4正则
-const Ipv4Reg = `((25[0-5]|(2[0-4]|1{0,1}[0-9]){0,1}[0-9])\.){3,3}(25[0-5]|(2[0-4]|1{0,1}[0-9]){0,1}[0-9])`
+var Ipv4Reg = regexp.MustCompile(`((25[0-5]|(2[0-4]|1{0,1}[0-9]){0,1}[0-9])\.){3,3}(25[0-5]|(2[0-4]|1{0,1}[0-9]){0,1}[0-9])`)
 
 // Ipv6Reg IPv6正则
-const Ipv6Reg = `((([0-9A-Fa-f]{1,4}:){7}([0-9A-Fa-f]{1,4}|:))|(([0-9A-Fa-f]{1,4}:){6}(:[0-9A-Fa-f]{1,4}|((25[0-5]|2[0-4]\d|1\d\d|[1-9]?\d)(\.(25[0-5]|2[0-4]\d|1\d\d|[1-9]?\d)){3})|:))|(([0-9A-Fa-f]{1,4}:){5}(((:[0-9A-Fa-f]{1,4}){1,2})|:((25[0-5]|2[0-4]\d|1\d\d|[1-9]?\d)(\.(25[0-5]|2[0-4]\d|1\d\d|[1-9]?\d)){3})|:))|(([0-9A-Fa-f]{1,4}:){4}(((:[0-9A-Fa-f]{1,4}){1,3})|((:[0-9A-Fa-f]{1,4})?:((25[0-5]|2[0-4]\d|1\d\d|[1-9]?\d)(\.(25[0-5]|2[0-4]\d|1\d\d|[1-9]?\d)){3}))|:))|(([0-9A-Fa-f]{1,4}:){3}(((:[0-9A-Fa-f]{1,4}){1,4})|((:[0-9A-Fa-f]{1,4}){0,2}:((25[0-5]|2[0-4]\d|1\d\d|[1-9]?\d)(\.(25[0-5]|2[0-4]\d|1\d\d|[1-9]?\d)){3}))|:))|(([0-9A-Fa-f]{1,4}:){2}(((:[0-9A-Fa-f]{1,4}){1,5})|((:[0-9A-Fa-f]{1,4}){0,3}:((25[0-5]|2[0-4]\d|1\d\d|[1-9]?\d)(\.(25[0-5]|2[0-4]\d|1\d\d|[1-9]?\d)){3}))|:))|(([0-9A-Fa-f]{1,4}:){1}(((:[0-9A-Fa-f]{1,4}){1,6})|((:[0-9A-Fa-f]{1,4}){0,4}:((25[0-5]|2[0-4]\d|1\d\d|[1-9]?\d)(\.(25[0-5]|2[0-4]\d|1\d\d|[1-9]?\d)){3}))|:))|(:(((:[0-9A-Fa-f]{1,4}){1,7})|((:[0-9A-Fa-f]{1,4}){0,5}:((25[0-5]|2[0-4]\d|1\d\d|[1-9]?\d)(\.(25[0-5]|2[0-4]\d|1\d\d|[1-9]?\d)){3}))|:)))`
+var Ipv6Reg = regexp.MustCompile(`((([0-9A-Fa-f]{1,4}:){7}([0-9A-Fa-f]{1,4}|:))|(([0-9A-Fa-f]{1,4}:){6}(:[0-9A-Fa-f]{1,4}|((25[0-5]|2[0-4]\d|1\d\d|[1-9]?\d)(\.(25[0-5]|2[0-4]\d|1\d\d|[1-9]?\d)){3})|:))|(([0-9A-Fa-f]{1,4}:){5}(((:[0-9A-Fa-f]{1,4}){1,2})|:((25[0-5]|2[0-4]\d|1\d\d|[1-9]?\d)(\.(25[0-5]|2[0-4]\d|1\d\d|[1-9]?\d)){3})|:))|(([0-9A-Fa-f]{1,4}:){4}(((:[0-9A-Fa-f]{1,4}){1,3})|((:[0-9A-Fa-f]{1,4})?:((25[0-5]|2[0-4]\d|1\d\d|[1-9]?\d)(\.(25[0-5]|2[0-4]\d|1\d\d|[1-9]?\d)){3}))|:))|(([0-9A-Fa-f]{1,4}:){3}(((:[0-9A-Fa-f]{1,4}){1,4})|((:[0-9A-Fa-f]{1,4}){0,2}:((25[0-5]|2[0-4]\d|1\d\d|[1-9]?\d)(\.(25[0-5]|2[0-4]\d|1\d\d|[1-9]?\d)){3}))|:))|(([0-9A-Fa-f]{1,4}:){2}(((:[0-9A-Fa-f]{1,4}){1,5})|((:[0-9A-Fa-f]{1,4}){0,3}:((25[0-5]|2[0-4]\d|1\d\d|[1-9]?\d)(\.(25[0-5]|2[0-4]\d|1\d\d|[1-9]?\d)){3}))|:))|(([0-9A-Fa-f]{1,4}:){1}(((:[0-9A-Fa-f]{1,4}){1,6})|((:[0-9A-Fa-f]{1,4}){0,4}:((25[0-5]|2[0-4]\d|1\d\d|[1-9]?\d)(\.(25[0-5]|2[0-4]\d|1\d\d|[1-9]?\d)){3}))|:))|(:(((:[0-9A-Fa-f]{1,4}){1,7})|((:[0-9A-Fa-f]{1,4}){0,5}:((25[0-5]|2[0-4]\d|1\d\d|[1-9]?\d)(\.(25[0-5]|2[0-4]\d|1\d\d|[1-9]?\d)){3}))|:)))`)
 
 // Config 配置
 type Config struct {
@@ -27,6 +29,7 @@ type Config struct {
 		GetType      string
 		URL          string
 		NetInterface string
+		Cmd          string
 		Domains      []string
 	}
 	Ipv6 struct {
@@ -35,6 +38,7 @@ type Config struct {
 		GetType      string
 		URL          string
 		NetInterface string
+		Cmd          string
 		IPv6Reg      string // ipv6匹配正则表达式
 		Domains      []string
 	}
@@ -82,7 +86,7 @@ func GetConfigCache() (conf Config, err error) {
 		return *cache.ConfigSingle, err
 	}
 
-	byt, err := ioutil.ReadFile(configFilePath)
+	byt, err := os.ReadFile(configFilePath)
 	if err != nil {
 		log.Println("config.yaml读取失败")
 		cache.Err = err
@@ -112,7 +116,7 @@ func (conf *Config) SaveConfig() (err error) {
 	}
 
 	configFilePath := util.GetConfigFilePath()
-	err = ioutil.WriteFile(configFilePath, byt, 0600)
+	err = os.WriteFile(configFilePath, byt, 0600)
 	if err != nil {
 		log.Println(err)
 		return
@@ -126,27 +130,24 @@ func (conf *Config) SaveConfig() (err error) {
 	return
 }
 
-// GetIpv4Addr 获得IPv4地址
-func (conf *Config) GetIpv4Addr() (result string) {
-	// 判断从哪里获取IP
-	if conf.Ipv4.GetType == "netInterface" {
-		// 从网卡获取IP
-		ipv4, _, err := GetNetInterface()
-		if err != nil {
-			log.Println("从网卡获得IPv4失败!")
-			return
-		}
+func (conf *Config) getIpv4AddrFromInterface() string {
+	ipv4, _, err := GetNetInterface()
+	if err != nil {
+		log.Println("从网卡获得IPv4失败!")
+		return ""
+	}
 
-		for _, netInterface := range ipv4 {
-			if netInterface.Name == conf.Ipv4.NetInterface && len(netInterface.Address) > 0 {
-				return netInterface.Address[0]
-			}
+	for _, netInterface := range ipv4 {
+		if netInterface.Name == conf.Ipv4.NetInterface && len(netInterface.Address) > 0 {
+			return netInterface.Address[0]
 		}
-
-		log.Println("从网卡中获得IPv4失败! 网卡名: ", conf.Ipv4.NetInterface)
-		return
 	}
 
+	log.Println("从网卡中获得IPv4失败! 网卡名: ", conf.Ipv4.NetInterface)
+	return ""
+}
+
+func (conf *Config) getIpv4AddrFromUrl() string {
 	client := util.CreateNoProxyHTTPClient("tcp4")
 	urls := strings.Split(conf.Ipv4.URL, ",")
 	for _, url := range urls {
@@ -158,73 +159,120 @@ func (conf *Config) GetIpv4Addr() (result string) {
 			continue
 		}
 		defer resp.Body.Close()
-		body, err := ioutil.ReadAll(resp.Body)
+		body, err := io.ReadAll(resp.Body)
 		if err != nil {
 			log.Println("读取IPv4结果失败! 接口: ", url)
 			continue
 		}
-		comp := regexp.MustCompile(Ipv4Reg)
-		result = comp.FindString(string(body))
-		if result != "" {
-			return
-		} else {
+		result := Ipv4Reg.FindString(string(body))
+		if result == "" {
 			log.Printf("获取IPv4结果失败! 接口: %s ,返回值: %s\n", url, result)
 		}
+		return result
 	}
+	return ""
+}
 
-	return
+func (conf *Config) getAddrFromCmd(addrType string) string {
+	var cmd string
+	var comp *regexp.Regexp
+	if addrType == "IPv4" {
+		cmd = conf.Ipv4.Cmd
+		comp = Ipv4Reg
+	} else {
+		cmd = conf.Ipv6.Cmd
+		comp = Ipv6Reg
+	}
+	// cmd is empty
+	if cmd == "" {
+		return ""
+	}
+	// run cmd with proper shell
+	var execCmd *exec.Cmd
+	if runtime.GOOS == "windows" {
+		execCmd = exec.Command("powershell", "-Command", cmd)
+	} else {
+		execCmd = exec.Command("sh", "-c", cmd)
+	}
+	// run cmd
+	out, err := execCmd.Output()
+	if err != nil {
+		log.Printf("获取%s结果失败! 未能成功执行命令：%s，错误：%q\n", addrType, execCmd.String(), err.Error())
+		return ""
+	}
+	str := string(out)
+	// get result
+	result := comp.FindString(str)
+	if result == "" {
+		log.Printf("获取%s结果失败! 命令：%s，标准输出：%q\n", addrType, execCmd.String(), str)
+	}
+	return result
 }
 
-// GetIpv6Addr 获得IPv6地址
-func (conf *Config) GetIpv6Addr() (result string) {
+// GetIpv4Addr 获得IPv4地址
+func (conf *Config) GetIpv4Addr() string {
 	// 判断从哪里获取IP
-	if conf.Ipv6.GetType == "netInterface" {
-		// 从网卡获取IP
-		_, ipv6, err := GetNetInterface()
-		if err != nil {
-			log.Println("从网卡获得IPv6失败!")
-			return
-		}
+	switch conf.Ipv4.GetType {
+	case "netInterface":
+		// 从网卡获取 IP
+		return conf.getIpv4AddrFromInterface()
+	case "url":
+		// 从 URL 获取 IP
+		return conf.getIpv4AddrFromUrl()
+	case "cmd":
+		// 从命令行获取 IP
+		return conf.getAddrFromCmd("IPv4")
+	}
+	return "" // unknown type
+}
 
-		for _, netInterface := range ipv6 {
-			if netInterface.Name == conf.Ipv6.NetInterface && len(netInterface.Address) > 0 {
-				if conf.Ipv6.IPv6Reg != "" {
-					// 匹配第几个IPv6
-					if match, err := regexp.MatchString("@\\d", conf.Ipv6.IPv6Reg); err == nil && match {
-						num, err := strconv.Atoi(conf.Ipv6.IPv6Reg[1:])
-						if err == nil {
-							if num > 0 {
-								log.Printf("IPv6将使用第 %d 个IPv6地址\n", num)
-								if num <= len(netInterface.Address) {
-									return netInterface.Address[num-1]
-								}
-								log.Printf("未找到第 %d 个IPv6地址! 将使用第一个IPv6地址\n", num)
-								return netInterface.Address[0]
+func (conf *Config) getIpv6AddrFromInterface() string {
+	_, ipv6, err := GetNetInterface()
+	if err != nil {
+		log.Println("从网卡获得IPv6失败!")
+		return ""
+	}
+
+	for _, netInterface := range ipv6 {
+		if netInterface.Name == conf.Ipv6.NetInterface && len(netInterface.Address) > 0 {
+			if conf.Ipv6.IPv6Reg != "" {
+				// 匹配第几个IPv6
+				if match, err := regexp.MatchString("@\\d", conf.Ipv6.IPv6Reg); err == nil && match {
+					num, err := strconv.Atoi(conf.Ipv6.IPv6Reg[1:])
+					if err == nil {
+						if num > 0 {
+							log.Printf("IPv6将使用第 %d 个IPv6地址\n", num)
+							if num <= len(netInterface.Address) {
+								return netInterface.Address[num-1]
 							}
-							log.Printf("IPv6匹配表达式 %s 不正确! 最小从1开始\n", conf.Ipv6.IPv6Reg)
-							return
+							log.Printf("未找到第 %d 个IPv6地址! 将使用第一个IPv6地址\n", num)
+							return netInterface.Address[0]
 						}
+						log.Printf("IPv6匹配表达式 %s 不正确! 最小从1开始\n", conf.Ipv6.IPv6Reg)
+						return ""
 					}
-					// 正则表达式匹配
-					log.Printf("IPv6将使用正则表达式 %s 进行匹配\n", conf.Ipv6.IPv6Reg)
-					for i := 0; i < len(netInterface.Address); i++ {
-						matched, err := regexp.MatchString(conf.Ipv6.IPv6Reg, netInterface.Address[i])
-						if matched && err == nil {
-							log.Println("匹配成功! 匹配到地址: ", netInterface.Address[i])
-							return netInterface.Address[i]
-						}
-						log.Printf("第 %d 个地址 %s 不匹配, 将匹配下一个地址\n", i+1, netInterface.Address[i])
+				}
+				// 正则表达式匹配
+				log.Printf("IPv6将使用正则表达式 %s 进行匹配\n", conf.Ipv6.IPv6Reg)
+				for i := 0; i < len(netInterface.Address); i++ {
+					matched, err := regexp.MatchString(conf.Ipv6.IPv6Reg, netInterface.Address[i])
+					if matched && err == nil {
+						log.Println("匹配成功! 匹配到地址: ", netInterface.Address[i])
+						return netInterface.Address[i]
 					}
-					log.Println("没有匹配到任何一个IPv6地址, 将使用第一个地址")
+					log.Printf("第 %d 个地址 %s 不匹配, 将匹配下一个地址\n", i+1, netInterface.Address[i])
 				}
-				return netInterface.Address[0]
+				log.Println("没有匹配到任何一个IPv6地址, 将使用第一个地址")
 			}
+			return netInterface.Address[0]
 		}
-
-		log.Println("从网卡中获得IPv6失败! 网卡名: ", conf.Ipv6.NetInterface)
-		return
 	}
 
+	log.Println("从网卡中获得IPv6失败! 网卡名: ", conf.Ipv6.NetInterface)
+	return ""
+}
+
+func (conf *Config) getIpv6AddrFromUrl() string {
 	client := util.CreateNoProxyHTTPClient("tcp6")
 	urls := strings.Split(conf.Ipv6.URL, ",")
 	for _, url := range urls {
@@ -237,19 +285,33 @@ func (conf *Config) GetIpv6Addr() (result string) {
 		}
 
 		defer resp.Body.Close()
-		body, err := ioutil.ReadAll(resp.Body)
+		body, err := io.ReadAll(resp.Body)
 		if err != nil {
 			log.Println("读取IPv6结果失败! 接口: ", url)
 			continue
 		}
-		comp := regexp.MustCompile(Ipv6Reg)
-		result = comp.FindString(string(body))
-		if result != "" {
-			return
-		} else {
+		result := Ipv6Reg.FindString(string(body))
+		if result == "" {
 			log.Printf("获取IPv6结果失败! 接口: %s ,返回值: %s\n", url, result)
 		}
+		return result
 	}
+	return ""
+}
 
-	return
+// GetIpv6Addr 获得IPv6地址
+func (conf *Config) GetIpv6Addr() (result string) {
+	// 判断从哪里获取IP
+	switch conf.Ipv6.GetType {
+	case "netInterface":
+		// 从网卡获取 IP
+		return conf.getIpv6AddrFromInterface()
+	case "url":
+		// 从 URL 获取 IP
+		return conf.getIpv6AddrFromUrl()
+	case "cmd":
+		// 从命令行获取 IP
+		return conf.getAddrFromCmd("IPv6")
+	}
+	return "" // unknown type
 }

config/domains.go

@@ -66,7 +66,7 @@ func (d Domain) GetCustomParams() url.Values {
 	return url.Values{}
 }
 
-// GetNewIp 接口/网卡获得ip并校验用户输入的域名
+// GetNewIp 接口/网卡/命令获得 ip 并校验用户输入的域名
 func (domains *Domains) GetNewIp(conf *Config) {
 	domains.Ipv4Domains = checkParseDomains(conf.Ipv4.Domains)
 	domains.Ipv6Domains = checkParseDomains(conf.Ipv6.Domains)

config/webhook.go

@@ -63,9 +63,9 @@ func ExecWebhook(domains *Domains, conf *Config) {
 		resp, err := clt.Do(req)
 		body, err := util.GetHTTPResponseOrg(resp, requestURL, err)
 		if err == nil {
-			log.Println(fmt.Sprintf("Webhook调用成功, 返回数据: %s", string(body)))
+			log.Printf("Webhook调用成功, 返回数据: %q\n", string(body))
 		} else {
-			log.Println(fmt.Sprintf("Webhook调用失败，Err：%s", err))
+			log.Printf("Webhook调用失败，Err：%s\n", err)
 		}
 	}
 }

dns/google_domain.go

@@ -1,7 +1,7 @@
 package dns
 
 import (
-	"io/ioutil"
+	"io"
 	"log"
 	"net/http"
 	"net/url"
@@ -105,7 +105,7 @@ func (gd *GoogleDomain) request(params url.Values, result *GoogleDomainResp) (er
 	}
 
 	defer resp.Body.Close()
-	data, err := ioutil.ReadAll(resp.Body)
+	data, err := io.ReadAll(resp.Body)
 	if err != nil {
 		return err
 	}

util/http_util.go

@@ -3,7 +3,7 @@ package util
 import (
 	"encoding/json"
 	"fmt"
-	"io/ioutil"
+	"io"
 	"log"
 	"net/http"
 )
@@ -36,7 +36,7 @@ func GetHTTPResponseOrg(resp *http.Response, url string, err error) ([]byte, err
 	}
 
 	defer resp.Body.Close()
-	body, err := ioutil.ReadAll(resp.Body)
+	body, err := io.ReadAll(resp.Body)
 
 	if err != nil {
 		Ipv4Cache.ForceCompare = true