fix panic error check when internal adapter is unreachable.

iqhater · Oct 27, 2024 · 0ccaa27 · 0ccaa27
1 parent d2539a2
commit 0ccaa27
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/data/internal.go b/data/internal.go
@@ -1,6 +1,7 @@
 package data
 
 import (
+	"errors"
 	"net"
 	"strings"
 )
@@ -15,7 +16,12 @@ type InternalData struct {
 func (i *InternalData) GetInternalIP() {
 
 	conn, err := net.Dial("udp", "8.8.8.8:80")
-	checkErr(err)
+
+	defer func() {
+		if r := recover(); r != nil {
+			checkErr(errors.Join(err, r.(error)))
+		}
+	}()
 	defer conn.Close()
 
 	localAddr := conn.LocalAddr().(*net.UDPAddr)

diff --git a/data/internal_test.go b/data/internal_test.go
@@ -13,7 +13,7 @@ func TestGetInternalIP(t *testing.T) {
 
 	// assert
 	if result == "" {
-		t.Errorf("Wrong internal IP address! Shoud be fromat like this XX.XX.XX.XX %s\n", result)
+		t.Errorf("Wrong internal IP address! Shoud be format like this XX.XX.XX.XX %s\n", result)
 	}
 }