From 6b192690bd5512aa64f6b48734ff062257a4d78c Mon Sep 17 00:00:00 2001
From: Cameron Clark <cameron@gremlin.com>
Date: Fri, 9 Apr 2021 12:14:46 -0400
Subject: [PATCH] Update csrf_field matching
---
src/main.rs | 17 +++++++----------
1 file changed, 7 insertions(+), 10 deletions(-)
diff --git a/src/main.rs b/src/main.rs
index 5ca5c6c..67f67ca 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -509,16 +509,13 @@ impl MainHandler {
match multipart.save().size_limit(self.upload_size_limit).temp() {
SaveResult::Full(entries) => {
// Pull out csrf field to check if token matches one generated
- let csrf_field = match entries.fields.get("csrf") {
- Some(fields) => match fields.first() {
- Some(field) => field,
- None => {
- return Err((
- status::BadRequest,
- String::from("csrf parameter not provided"),
- ))
- }
- },
+ let csrf_field = match entries
+ .fields
+ .get("csrf")
+ .map(|fields| fields.first())
+ .unwrap_or(None)
+ {
+ Some(field) => field,
None => {
return Err((
status::BadRequest,
--
GitLab