buyfakett/server_torii
1
0
Fork 0
mirror of https://github.com/Rayzggz/server_torii.git synced 2025-06-21 22:41:31 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: CAPTCHA SESSION

Browse Source
This commit is contained in:
Roi Feng
2025-03-08 23:09:06 -05:00
parent 1a8d40bc82
commit a1f15f37c6
6 changed files with 62 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
internal/server/checker.go
View File

@ -70,6 +70,7 @@ func CheckMain(w http.ResponseWriter, userRequestData dataType.UserRequest, rule
http.Error(w, "500 - Internal Server Error", http.StatusInternalServerError)
return
}
w.Header().Set("Set-Cookie", "__torii_session_id="+string(decision.ResponseData)+"; Path=/; Path=/; Max-Age=86400; Priority=High; HttpOnly;")
w.Header().Set("Content-Type", "text/html; charset=utf-8")
w.WriteHeader(http.StatusServiceUnavailable)
if err = tpl.Execute(w, nil); err != nil {

8
internal/server/torii.go
View File

@ -28,6 +28,14 @@ func CheckTorii(w http.ResponseWriter, r *http.Request, reqData dataType.UserReq
return
}
return
} else if bytes.Compare(decision.ResponseData, []byte("badSession")) == 0 {
w.WriteHeader(http.StatusOK)
_, err := w.Write([]byte("badSession"))
if err != nil {
log.Printf("Error writing response: %v", err)
return
}
return
} else if bytes.Compare(decision.ResponseData, []byte("good")) == 0 {
w.Header().Set("Set-Cookie", "__torii_clearance="+string(check.GenClearance(reqData, *ruleSet))+"; Path=/; Max-Age=86400; Priority=High; HttpOnly;")
w.WriteHeader(http.StatusOK)