Add cookie option in CSRF

rookie-ninja · Apr 9, 2024 · 32a6cef · 32a6cef
1 parent 1f210eb
commit 32a6cef
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/middleware/csrf/options.go b/middleware/csrf/options.go
@@ -127,6 +127,8 @@ func NewOptionSet(opts ...Option) OptionSetInterface {
 	parts := strings.Split(set.tokenLookup, ":")
 	set.extractor = csrfTokenFromHeader(parts[1])
 	switch parts[0] {
+	case "cookie":
+		set.extractor = csrfTokenFromCookie(parts[1])
 	case "form":
 		set.extractor = csrfTokenFromForm(parts[1])
 	case "query":