HTTP GET/POST method distinguish

1 files changed, 17 insertions, 3 deletions

diff --git a/include/routing.h b/include/routing.h
index d814266..e62cbce 100644
--- a/include/routing.h
+++ b/include/routing.h
@@ -26,8 +26,13 @@ namespace crow
 
         virtual void handle(const request&, response&, const routing_params&) = 0;
 
-    protected:
+        uint32_t methods()
+        {
+            return methods_;
+        }
 
+    protected:
+        uint32_t methods_{1<<(int)HTTPMethod::GET};
     };
 
     template <typename ... Args>
@@ -147,6 +152,7 @@ namespace crow
         self_t& methods(HTTPMethod method)
         {
             methods_ = 1<<(int)method;
+            return *this;
         }
 
         template <typename ... MethodArgs>
@@ -154,6 +160,7 @@ namespace crow
         {
             methods(args_method...);
             methods_ |= 1<<(int)method;
+            return *this;
         }
 
         void validate()
@@ -257,7 +264,6 @@ namespace crow
 
         std::string rule_;
         std::string name_;
-        uint32_t methods_{1<<(int)HTTPMethod::GET};
 
         template <typename T, int Pos>
         struct call_pair
@@ -641,7 +647,15 @@ public:
             if (rule_index >= rules_.size())
                 throw std::runtime_error("Trie internal structure corrupted!");
 
-            CROW_LOG_DEBUG << "Matched rule '" << ((TaggedRule<>*)rules_[rule_index].get())->rule_ << "'";
+            if ((rules_[rule_index]->methods() & (1<<(uint32_t)req.method)) == 0)
+            {
+				CROW_LOG_DEBUG << "Rule found but method mismatch: " << editedUrl << " with " << method_name(req.method) << "(" << (uint32_t)req.method << ") / " << rules_[rule_index]->methods();
+                res = response(404);
+				res.end();
+                return;
+            }
+
+            CROW_LOG_DEBUG << "Matched rule '" << ((TaggedRule<>*)rules_[rule_index].get())->rule_ << "' " << (uint32_t)req.method << " / " << rules_[rule_index]->methods();
 
             rules_[rule_index]->handle(req, res, found.second);
         }