From fb13ce3ed7fd5cd2c1eb6684a50057f2b0caf463 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E7=99=BE=E9=87=8C=28barry=29?= <kooksee@163.com>
Date: Fri, 12 Jul 2024 17:40:04 +0800
Subject: [PATCH] fix: barry 2024-07-12 17:40:04

---
 server/service.go | 12 +++++++++++-
 1 file changed, 11 insertions(+), 1 deletion(-)

diff --git a/server/service.go b/server/service.go
index 29a99c0c..a891e44d 100644
--- a/server/service.go
+++ b/server/service.go
@@ -704,12 +704,21 @@ func (m authMiddleware) ServeHTTP(writer http.ResponseWriter, request *http.Requ
 	if setCookie != "" {
 		var cc Cookie
 		if err := json.Unmarshal([]byte(setCookie), &cc); err != nil {
-			log.Errorf("failed to decode cookie json data, cookie=%s", setCookie)
+			err = fmt.Errorf("failed to decode cookie json data, cookie=%s", setCookie)
+			log.Errorf(err.Error())
+			writer.WriteHeader(http.StatusBadRequest)
+			writer.Write([]byte(err.Error()))
+			return
 		}
 
 		var expiredAt = time.Now().Add(time.Hour)
 		if ee, err := strconv.ParseInt(cc.ExpiredAt, 10, 64); err == nil {
 			expiredAt = time.Unix(ee, 0)
+		} else {
+			err = fmt.Errorf("failed to parse expiredAt field, expiredAt=%s", cc.ExpiredAt)
+			writer.WriteHeader(http.StatusBadRequest)
+			writer.Write([]byte(err.Error()))
+			return
 		}
 
 		http.SetCookie(writer, &http.Cookie{
@@ -719,6 +728,7 @@ func (m authMiddleware) ServeHTTP(writer http.ResponseWriter, request *http.Requ
 			Domain:  request.Host,
 			Expires: expiredAt,
 		})
+		writer.WriteHeader(http.StatusOK)
 		writer.Write([]byte("ok"))
 		return
 	}