diff options
author | Krzysztof Bielicki <krzysztof.bielicki@allegro.pl> | 2015-03-10 10:44:06 +0100 |
---|---|---|
committer | Krzysztof Bielicki <krzysztof.bielicki@allegro.pl> | 2015-03-10 10:44:06 +0100 |
commit | 244ef243d75145a01d9029589de65be51299b3f3 (patch) | |
tree | 13ef9d5e83938149c1c629f90dfb6d2b73076a5e /libmproxy/protocol | |
parent | ec4a6ec4e52b36860fbb37ea81e6fd4027c38dfc (diff) | |
download | mitmproxy-244ef243d75145a01d9029589de65be51299b3f3.tar.gz mitmproxy-244ef243d75145a01d9029589de65be51299b3f3.tar.bz2 mitmproxy-244ef243d75145a01d9029589de65be51299b3f3.zip |
[#514] Add support for ignoring payload params in multipart/form-data
Diffstat (limited to 'libmproxy/protocol')
-rw-r--r-- | libmproxy/protocol/http.py | 21 |
1 files changed, 20 insertions, 1 deletions
diff --git a/libmproxy/protocol/http.py b/libmproxy/protocol/http.py index 49310ec3..512cf75b 100644 --- a/libmproxy/protocol/http.py +++ b/libmproxy/protocol/http.py @@ -15,6 +15,7 @@ from ..proxy.connection import ServerConnection from .. import encoding, utils, controller, stateobject, proxy HDR_FORM_URLENCODED = "application/x-www-form-urlencoded" +HDR_FORM_MULTIPART = "multipart/form-data" CONTENT_MISSING = 0 @@ -507,6 +508,19 @@ class HTTPRequest(HTTPMessage): """ self.headers["Host"] = [self.host] + def get_form(self): + """ + Retrieves the URL-encoded or multipart form data, returning an ODict object. + Returns an empty ODict if there is no data or the content-type + indicates non-form data. + """ + if self.content: + if self.headers.in_any("content-type", HDR_FORM_URLENCODED, True): + return self.get_form_urlencoded() + elif self.headers.in_any("content-type", HDR_FORM_MULTIPART, True): + return self.get_form_multipart() + return ODict([]) + def get_form_urlencoded(self): """ Retrieves the URL-encoded form data, returning an ODict object. @@ -514,7 +528,12 @@ class HTTPRequest(HTTPMessage): indicates non-form data. """ if self.content and self.headers.in_any("content-type", HDR_FORM_URLENCODED, True): - return ODict(utils.urldecode(self.content)) + return ODict(utils.urldecode(self.content)) + return ODict([]) + + def get_form_multipart(self): + if self.content and self.headers.in_any("content-type", HDR_FORM_MULTIPART, True): + return ODict(utils.multipartdecode(self.headers, self.content)) return ODict([]) def set_form_urlencoded(self, odict): |