From 87513987a23ecd75cd21015ed215bae3b279d8c5 Mon Sep 17 00:00:00 2001
From: 杨宇千 <crupest@outlook.com>
Date: Mon, 19 Aug 2019 22:52:01 +0800
Subject: Add check for content in avatar put.

---
 Timeline/Filters/ContentHeaderAttributes.cs | 48 +++++++++++++++++++++++++++++
 1 file changed, 48 insertions(+)
 create mode 100644 Timeline/Filters/ContentHeaderAttributes.cs

(limited to 'Timeline/Filters')

diff --git a/Timeline/Filters/ContentHeaderAttributes.cs b/Timeline/Filters/ContentHeaderAttributes.cs
new file mode 100644
index 00000000..14685a01
--- /dev/null
+++ b/Timeline/Filters/ContentHeaderAttributes.cs
@@ -0,0 +1,48 @@
+using Microsoft.AspNetCore.Mvc;
+using Microsoft.AspNetCore.Mvc.Filters;
+using Timeline.Models.Http;
+
+namespace Timeline.Filters
+{
+    public class RequireContentTypeAttribute : ActionFilterAttribute
+    {
+        public override void OnActionExecuting(ActionExecutingContext context)
+        {
+            if (context.HttpContext.Request.ContentType == null)
+            {
+                context.Result = new BadRequestObjectResult(CommonResponse.MissingContentType());
+            }
+        }
+    }
+
+    public class RequireContentLengthAttribute : ActionFilterAttribute
+    {
+        public RequireContentLengthAttribute()
+            : this(true)
+        {
+
+        }
+
+        public RequireContentLengthAttribute(bool requireNonZero)
+        {
+            RequireNonZero = requireNonZero;
+        }
+
+        public bool RequireNonZero { get; set; }
+
+        public override void OnActionExecuting(ActionExecutingContext context)
+        {
+            if (context.HttpContext.Request.ContentLength == null)
+            {
+                context.Result = new BadRequestObjectResult(CommonResponse.MissingContentLength());
+                return;
+            }
+
+            if (RequireNonZero && context.HttpContext.Request.ContentLength.Value == 0)
+            {
+                context.Result = new BadRequestObjectResult(CommonResponse.ZeroContentLength());
+                return;
+            }
+        }
+    }
+}
-- 
cgit v1.2.3