Use built-in cross-origin protection

1 files changed, 20 insertions, 58 deletions

diff --git a/roseh.moe.go b/roseh.moe.go
index 3d357fd..6d1ccea 100644
--- a/roseh.moe.go
+++ b/roseh.moe.go
@@ -421,42 +421,6 @@ func cookieAuth(w http.ResponseWriter, r *http.Request) bool {
 	return true
 }
 
-const csrfCookieName = "roseh.moe.csrf-token"
-
-func reqCSRFToken(w http.ResponseWriter, r *http.Request) string {
-	const csrfTokenLen = 32
-	if csrfCookie, err := r.Cookie(csrfCookieName); err == nil {
-		return csrfCookie.Value
-	} else {
-		buf := make([]byte, csrfTokenLen)
-		rand.Read(buf)
-		csrfToken := base64.RawURLEncoding.EncodeToString(buf)
-		http.SetCookie(w, &http.Cookie{
-			Name:        csrfCookieName,
-			Value:       csrfToken,
-			Path:        "/",
-			Secure:      true,
-			HttpOnly:    true,
-			SameSite:    http.SameSiteStrictMode,
-			Partitioned: true,
-		})
-		return csrfToken
-	}
-}
-
-func checkCSRFToken(r *http.Request) error {
-	cookie, err := r.Cookie(csrfCookieName)
-	if err != nil {
-		return err
-	}
-	cookieHash := sha256.Sum256([]byte(cookie.Value))
-	formValueHash := sha256.Sum256([]byte(r.FormValue("csrf-token")))
-	if subtle.ConstantTimeCompare(cookieHash[:], formValueHash[:]) == 0 {
-		return fmt.Errorf("bad CSRF token")
-	}
-	return nil
-}
-
 func checkPassword(password string) bool {
 	const pwHashSize = targetSecurityLevel
 	hash := sha256.Sum256([]byte(password))
@@ -529,8 +493,7 @@ var (
 )
 
 type notepadTemplateArgs struct {
-	Content   string
-	CSRFToken string
+	Content string
 }
 
 func notepad(w http.ResponseWriter, r *http.Request) {
@@ -541,15 +504,12 @@ func notepad(w http.ResponseWriter, r *http.Request) {
 	notepadContentsMu.Lock()
 	currentContent := notepadContents
 	notepadContentsMu.Unlock()
-	if err := notepadTemplate.Execute(w, notepadTemplateArgs{Content: currentContent, CSRFToken: reqCSRFToken(w, r)}); err != nil {
+	if err := notepadTemplate.Execute(w, notepadTemplateArgs{Content: currentContent}); err != nil {
 		log.Printf("Warning: notepad: %s", err)
 	}
 }
 
 func saveNote(w http.ResponseWriter, r *http.Request) error {
-	if err := checkCSRFToken(r); err != nil {
-		return err
-	}
 	if !cookieAuth(w, r) {
 		return fmt.Errorf("not logged in")
 	}
@@ -625,22 +585,24 @@ func main() {
 		log.Fatal(err)
 	}
 
-	http.HandleFunc("GET /pong", pong)
-	http.HandleFunc("GET /wormhole", newWormhole)
-	http.HandleFunc("GET /wormhole/{hole}/upload", wormhole)
-	http.HandleFunc("POST /wormhole/{hole}/upload", wormholeSend)
-	http.HandleFunc("GET /wormhole/{hole}", wormholeRecv)
-	http.HandleFunc("GET /wormhole/{hole}/qr.png", wormholeQR)
-	http.HandleFunc("GET /login", serveLogin)
-	http.HandleFunc("POST /login", login)
-	http.HandleFunc("GET /notepad", notepad)
-	http.HandleFunc("POST /notepad", autosave)
-	http.HandleFunc("GET /cmd/{cmd}", cmd)
-	http.HandleFunc("GET /pkg/{pkg}", pkg)
-	http.HandleFunc("GET /static/", static)
-	http.HandleFunc("GET /favicon.ico", favicon)
-	http.HandleFunc("GET /{$}", index)
-	http.HandleFunc("GET /", notFound)
+	mux := http.NewServeMux()
+	mux.HandleFunc("GET /pong", pong)
+	mux.HandleFunc("GET /wormhole", newWormhole)
+	mux.HandleFunc("GET /wormhole/{hole}/upload", wormhole)
+	mux.HandleFunc("POST /wormhole/{hole}/upload", wormholeSend)
+	mux.HandleFunc("GET /wormhole/{hole}", wormholeRecv)
+	mux.HandleFunc("GET /wormhole/{hole}/qr.png", wormholeQR)
+	mux.HandleFunc("GET /login", serveLogin)
+	mux.HandleFunc("POST /login", login)
+	mux.HandleFunc("GET /notepad", notepad)
+	mux.HandleFunc("POST /notepad", autosave)
+	mux.HandleFunc("GET /cmd/{cmd}", cmd)
+	mux.HandleFunc("GET /pkg/{pkg}", pkg)
+	mux.HandleFunc("GET /static/", static)
+	mux.HandleFunc("GET /favicon.ico", favicon)
+	mux.HandleFunc("GET /{$}", index)
+	mux.HandleFunc("GET /", notFound)
+	http.Handle("/", http.NewCrossOriginProtection().Handler(mux))
 	jellyfinURL, err := url.Parse("http://127.0.0.1:8096")
 	if err != nil {
 		log.Fatal(err)