From 40d6b2c09059fbb94455ea8480fd3c90b649ea76 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E8=80=97=E5=AD=90?= <i@haozi.net>
Date: Sun, 29 Dec 2024 11:28:22 +0800
Subject: [PATCH] =?UTF-8?q?fix:=20=E5=AE=89=E5=85=A8=E7=99=BB=E5=BD=95?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 internal/service/user.go | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/internal/service/user.go b/internal/service/user.go
index 91bee6e8..14b3afa4 100644
--- a/internal/service/user.go
+++ b/internal/service/user.go
@@ -2,6 +2,7 @@ package service
 
 import (
 	"crypto/rsa"
+	"crypto/sha256"
 	"encoding/gob"
 	"fmt"
 	"net"
@@ -12,7 +13,6 @@ import (
 	"github.com/go-rat/sessions"
 	"github.com/knadh/koanf/v2"
 	"github.com/spf13/cast"
-	"golang.org/x/crypto/sha3"
 
 	"github.com/TheTNB/panel/internal/biz"
 	"github.com/TheTNB/panel/internal/http/request"
@@ -93,7 +93,10 @@ func (s *UserService) Login(w http.ResponseWriter, r *http.Request) {
 	}
 	if req.SafeLogin && !s.conf.Bool("http.tls") {
 		sess.Put("safe_login", true)
-		sess.Put("safe_client", fmt.Sprintf("%x", sha3.Sum256([]byte(ip))))
+		sess.Put("safe_client", fmt.Sprintf("%x", sha256.Sum256([]byte(ip))))
+	} else {
+		sess.Forget("safe_login")
+		sess.Forget("safe_client")
 	}
 
 	sess.Put("user_id", user.ID)