char
/
live.umm.gay
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Fix authorization (real)

Charlotte Som 2023-02-23 16:55:06 +00:00
parent 57a75ef495
commit 681cb1ca4f
2 changed files with 7 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
wish-server/.gitignore vendored
View File

@ -1 +1,2 @@
/streams.db /streams.db
/wish-server

9
wish-server/main.go
View File

@ -133,10 +133,13 @@ func HandleWHIP(res http.ResponseWriter, req *http.Request) {
logHTTPError(res, "Authorization was not set", http.StatusBadRequest) logHTTPError(res, "Authorization was not set", http.StatusBadRequest)
return return
} }
streamName, streamPassword, _ := strings.Cut(authorization, ":") streamName, streamPassword, _ := strings.Cut(authorization, ":")
if err := db.QueryRow("SELECT * FROM streams WHERE stream = ? AND password = ?", streamName, streamPassword).Scan(); err != nil { streamName, _ = strings.CutPrefix(strings.ToLower(streamName), "Bearer ")
logHTTPError(res, "Invalid stream authorization", http.StatusUnauthorized)
var qN string
var qP string
if err := db.QueryRow("SELECT * FROM streams WHERE stream = ? AND password = ?", streamName, streamPassword).Scan(&qN, &qP); err != nil {
logHTTPError(res, "Invalid stream authorization for: "+streamName+" - "+err.Error(), http.StatusUnauthorized)
return return
} }