frozenleaves/ddns-go
1
0
Fork 0
mirror of https://github.com/jeessy2/ddns-go.git synced 2025-10-20 15:33:46 +08:00
Code Issues Packages Projects Releases Wiki Activity

refactor: set InsecureSkipVerify only once (#822)

Browse Source
This commit is contained in:
WaterLemons2k
2023-08-17 16:37:09 +08:00
committed by GitHub
parent 98756f2124
commit 5b77583b49
2 changed files with 10 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
main.go
View File

@ -80,7 +80,7 @@ func main() {
os.Setenv(util.ConfigFilePathENV, absPath)
}
if *skipVerify {
os.Setenv(util.SkipVerifyENV, "true")
util.SetInsecureSkipVerify()
}
if *customDNSServer != "" {
util.NewDialerResolver(*customDNSServer + ":53")

18
util/http_client_util.go
View File

@ -5,12 +5,9 @@ import (
"crypto/tls"
"net"
"net/http"
"os"
"time"
)
const SkipVerifyENV = "DDNS_SKIP_VERIFY"
var dialer = &net.Dialer{
Timeout: 30 * time.Second,
KeepAlive: 30 * time.Second,
@ -31,8 +28,6 @@ var defaultTransport = &http.Transport{
// CreateHTTPClient Create Default HTTP Client
func CreateHTTPClient() *http.Client {
// SkipVerfiry
defaultTransport.TLSClientConfig = &tls.Config{InsecureSkipVerify: os.Getenv(SkipVerifyENV) == "true"}
return &http.Client{
Timeout: 30 * time.Second,
Transport: defaultTransport,
@ -74,18 +69,23 @@ var noProxyTcp6Transport = &http.Transport{
// CreateNoProxyHTTPClient Create NoProxy HTTP Client
func CreateNoProxyHTTPClient(network string) *http.Client {
if network == "tcp6" {
// SkipVerfiry
noProxyTcp6Transport.TLSClientConfig = &tls.Config{InsecureSkipVerify: os.Getenv(SkipVerifyENV) == "true"}
return &http.Client{
Timeout: 30 * time.Second,
Transport: noProxyTcp6Transport,
}
}
// SkipVerfiry
noProxyTcp4Transport.TLSClientConfig = &tls.Config{InsecureSkipVerify: os.Getenv(SkipVerifyENV) == "true"}
return &http.Client{
Timeout: 30 * time.Second,
Transport: noProxyTcp4Transport,
}
}
// SetInsecureSkipVerify 将所有 http.Transport 的 InsecureSkipVerify 设置为 true
func SetInsecureSkipVerify() {
transports := []*http.Transport{defaultTransport, noProxyTcp4Transport, noProxyTcp6Transport}
for _, transport := range transports {
transport.TLSClientConfig = &tls.Config{InsecureSkipVerify: true}
}
}