ultrasn0w/huso
1
0
Fork 0
mirror of https://github.com/ultrasn0w/huso.git synced 2025-12-15 21:09:52 +01:00
Code Issues Projects Releases Wiki Activity

Fix header check

Browse Source
This commit is contained in:
daru
2022-04-18 23:56:28 +02:00
parent 171fd41c1c
commit 10c9997160
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
ober.go
View File

@@ -18,7 +18,7 @@ func RunWebserv() {
r.GET("/api/season", Headers(Season)) r.GET("/api/season", Headers(Season))
r.GET("/api/auth/{user}", Headers(AuthTest)) r.GET("/api/auth/{user}", Headers(AuthTest))
r.GET("/api/anime/{id}", Headers(AnimeGet)) r.GET("/api/anime/{id}", Headers(AnimeGet))
r.GET("/api/animesearch/", Headers(AnimeSearchGet)) r.GET("/api/animesearch", Headers(AnimeSearchGet))
r.GET("/api/user/{user?}", Headers(UserGet)) r.GET("/api/user/{user?}", Headers(UserGet))
r.GET("/api/watch/{user?}", Headers(WatchGet)) r.GET("/api/watch/{user?}", Headers(WatchGet))
r.GET("/api/watchext/{user?}", Headers(WatchExtendedGet)) r.GET("/api/watchext/{user?}", Headers(WatchExtendedGet))
@@ -240,7 +240,7 @@ func WatchExtendedGet(ctx *fasthttp.RequestCtx) {
} }
func Register(ctx *fasthttp.RequestCtx) { func Register(ctx *fasthttp.RequestCtx) {
if string(ctx.Request.Header.ContentType()) != "application/json" { if !strings.Contains(string(ctx.Request.Header.ContentType()), "application/json") {
ctx.SetStatusCode(fasthttp.StatusBadRequest) ctx.SetStatusCode(fasthttp.StatusBadRequest)
return return
} }
@@ -304,7 +304,7 @@ func WatchPost(ctx *fasthttp.RequestCtx) {
} }
func UnRegister(ctx *fasthttp.RequestCtx) { func UnRegister(ctx *fasthttp.RequestCtx) {
if string(ctx.Request.Header.ContentType()) != "application/json" { if !strings.Contains(string(ctx.Request.Header.ContentType()), "application/json") {
ctx.SetStatusCode(fasthttp.StatusBadRequest) ctx.SetStatusCode(fasthttp.StatusBadRequest)
return return
} }
@@ -410,7 +410,7 @@ func watchGetLogic(ctx *fasthttp.RequestCtx) ([]AnimeUser, error) {
func processUpdateReq(ctx *fasthttp.RequestCtx, update bool) { func processUpdateReq(ctx *fasthttp.RequestCtx, update bool) {
auth := ctx.Request.Header.Peek("X-HUSO-AUTH") auth := ctx.Request.Header.Peek("X-HUSO-AUTH")
if ctx.UserValue("user") == nil || auth == nil || string(auth) == "" || string(ctx.Request.Header.ContentType()) != "application/json" { if ctx.UserValue("user") == nil || auth == nil || string(auth) == "" || !strings.Contains(string(ctx.Request.Header.ContentType()), "application/json") {
ctx.SetStatusCode(fasthttp.StatusBadRequest) ctx.SetStatusCode(fasthttp.StatusBadRequest)
return return
} }