feat: 优化客户端hash使用sha256计算

TheTNB · Dec 9, 2024 · 1902a59 · 1902a59
1 parent 939d108
commit 1902a59
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/internal/http/middleware/must_login.go b/internal/http/middleware/must_login.go
@@ -2,6 +2,7 @@ package middleware
 
 import (
 	"context"
+	"crypto/sha256"
 	"fmt"
 	"net"
 	"net/http"
@@ -10,7 +11,6 @@ import (
 
 	"github.com/go-rat/chix"
 	"github.com/spf13/cast"
-	"golang.org/x/crypto/sha3"
 
 	"github.com/TheTNB/panel/internal/app"
 )
@@ -65,7 +65,7 @@ func MustLogin(next http.Handler) http.Handler {
 		if safeLogin {
 			safeClientHash := cast.ToString(sess.Get("safe_client"))
 			ip, _, _ := net.SplitHostPort(strings.TrimSpace(r.RemoteAddr))
-			clientHash := fmt.Sprintf("%x", sha3.Sum256([]byte(ip)))
+			clientHash := fmt.Sprintf("%x", sha256.Sum256([]byte(ip)))
 			if safeClientHash != clientHash || safeClientHash == "" {
 				render := chix.NewRender(w)
 				render.Status(http.StatusUnauthorized)