优化代码

bjdgyc · Oct 25, 2024 · 96fd114 · 96fd114
1 parent bd6ee0b
commit 96fd114
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/server/handler/link_auth_otp.go b/server/handler/link_auth_otp.go
@@ -172,6 +172,8 @@ func LinkAuth_otp(w http.ResponseWriter, r *http.Request) {
 
 	body, err := io.ReadAll(r.Body)
 	if err != nil {
+		base.Error(err)
+		SessStore.DeleteAuthSession(sessionID)
 		w.WriteHeader(http.StatusBadRequest)
 		return
 	}
@@ -181,6 +183,7 @@ func LinkAuth_otp(w http.ResponseWriter, r *http.Request) {
 	err = xml.Unmarshal(body, &cr)
 	if err != nil {
 		base.Error(err)
+		SessStore.DeleteAuthSession(sessionID)
 		w.WriteHeader(http.StatusBadRequest)
 		return
 	}
@@ -193,6 +196,7 @@ func LinkAuth_otp(w http.ResponseWriter, r *http.Request) {
 	// 动态码错误
 	if !dbdata.CheckOtp(username, otp, otpSecret) {
 		if sessionData.AddOtpErrCount(1) > maxOtpErrCount {
+			SessStore.DeleteAuthSession(sessionID)
 			http.Error(w, "TooManyError, please login again", http.StatusBadRequest)
 			return
 		}