refactored post limit management

unbit · Apr 5, 2011 · b0db67d · b0db67d
1 parent 703cad6
commit b0db67d
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 8 deletions.
diff --git a/plugins/python/wsgi_handlers.c b/plugins/python/wsgi_handlers.c
@@ -266,14 +266,6 @@ int uwsgi_request_wsgi(struct wsgi_request *wsgi_req) {
 	}
 
 
-	if (uwsgi.limit_post) {
-		if (wsgi_req->post_cl > uwsgi.limit_post) {
-			uwsgi_log("Invalid (too big) CONTENT_LENGTH. skip.\n");
-			return -1;
-		}
-	}
-
-
 	if (!up.ignore_script_name) {
 
 		if (!wsgi_req->script_name)

diff --git a/protocol.c b/protocol.c
@@ -675,6 +675,13 @@ int uwsgi_parse_vars(struct wsgi_request *wsgi_req) {
 						}
 						else if (!uwsgi_strncmp("CONTENT_LENGTH", 14, wsgi_req->hvec[wsgi_req->var_cnt].iov_base, wsgi_req->hvec[wsgi_req->var_cnt].iov_len)) {
 							wsgi_req->post_cl = get_content_length(ptrbuf, strsize);
+							if (uwsgi.limit_post) {
+                						if (wsgi_req->post_cl > uwsgi.limit_post) {
+                        						uwsgi_log("Invalid (too big) CONTENT_LENGTH. skip.\n");
+                        						return -1;
+                						}
+        						}
+
 						}
 
 						if (wsgi_req->var_cnt < uwsgi.vec_size - (4 + 1)) {