cyonjan/frp
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: barry 2024-07-12 17:40:04

Browse Source
This commit is contained in:
百里(barry) 2024-07-12 17:40:04 +08:00
parent 151f2e89b2
commit fb13ce3ed7
1 changed files with 11 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
server/service.go
View File

@ -704,12 +704,21 @@ func (m authMiddleware) ServeHTTP(writer http.ResponseWriter, request *http.Requ
if setCookie != "" { if setCookie != "" {
var cc Cookie var cc Cookie
if err := json.Unmarshal([]byte(setCookie), &cc); err != nil { 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) var expiredAt = time.Now().Add(time.Hour)
if ee, err := strconv.ParseInt(cc.ExpiredAt, 10, 64); err == nil { if ee, err := strconv.ParseInt(cc.ExpiredAt, 10, 64); err == nil {
expiredAt = time.Unix(ee, 0) 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{ http.SetCookie(writer, &http.Cookie{
@ -719,6 +728,7 @@ func (m authMiddleware) ServeHTTP(writer http.ResponseWriter, request *http.Requ
Domain: request.Host, Domain: request.Host,
Expires: expiredAt, Expires: expiredAt,
}) })
writer.WriteHeader(http.StatusOK)
writer.Write([]byte("ok")) writer.Write([]byte("ok"))
return return
} }