PenTest/sqlmap
1
0
Fork 0
mirror of https://github.com/sqlmapproject/sqlmap.git synced 2025-12-07 05:01:30 +00:00
Code Issues Projects Releases Wiki Activity

Minor update

Browse Source
This commit is contained in:
Miroslav Stampar
2019-05-03 00:48:46 +02:00
parent a33ee69337
commit 0f4d1e79b7
4 changed files with 9 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
thirdparty/multipart/multipartpost.py vendored
View File

@@ -26,6 +26,7 @@ import os
import stat
import sys
from lib.core.common import getBytes
from lib.core.compat import choose_boundary
from lib.core.exception import SqlmapDataException
from thirdparty.six.moves import urllib as _urllib
@@ -51,7 +52,7 @@ class MultipartPostHandler(_urllib.request.BaseHandler):
try:
for(key, value) in data.items():
if isinstance(value, file) or hasattr(value, "file") or isinstance(value, io.IOBase):
if hasattr(value, "fileno") or hasattr(value, "file") or isinstance(value, io.IOBase):
v_files.append((key, value))
else:
v_vars.append((key, value))
@@ -85,7 +86,7 @@ class MultipartPostHandler(_urllib.request.BaseHandler):
buf += "\r\n\r\n" + value + "\r\n"
for (key, fd) in files:
file_size = os.fstat(fd.fileno())[stat.ST_SIZE] if isinstance(fd, file) else fd.len
file_size = os.fstat(fd.fileno())[stat.ST_SIZE] if hasattr(fd, "fileno") else fd.len
filename = fd.name.split("/")[-1] if "/" in fd.name else fd.name.split("\\")[-1]
try:
contenttype = mimetypes.guess_type(filename)[0] or "application/octet-stream"
@@ -98,8 +99,8 @@ class MultipartPostHandler(_urllib.request.BaseHandler):
# buf += "Content-Length: %s\r\n" % file_size
fd.seek(0)
buf = str(buf) if not isinstance(buf, unicode) else buf.encode("utf8")
buf += "\r\n%s\r\n" % fd.read()
buf = getBytes(buf)
buf += b"\r\n%s\r\n" % fd.read()
buf += "--%s--\r\n\r\n" % boundary