mirror of
https://github.com/sqlmapproject/sqlmap.git
synced 2025-02-03 05:04:11 +03:00
Merge branch 'master' of github.com:sqlmapproject/sqlmap
This commit is contained in:
commit
92ea8841f8
|
@ -37,6 +37,8 @@ from lib.core.settings import TRIM_STDOUT_DUMP_SIZE
|
|||
from lib.core.settings import UNICODE_ENCODING
|
||||
from thirdparty.magic import magic
|
||||
|
||||
from extra.safe2bin.safe2bin import safechardecode
|
||||
|
||||
class Dump(object):
|
||||
"""
|
||||
This class defines methods used to parse and output the results
|
||||
|
@ -478,13 +480,11 @@ class Dump(object):
|
|||
self._write("| %s%s" % (value, blank), newline=False, console=console)
|
||||
|
||||
#if len(value) > 10 and r'\x' in value:
|
||||
#mimetype = magic.from_buffer(value, mime=True)
|
||||
#if mimetype.startswith("application") or mimetype.startswith("image"):
|
||||
#with open("%s%s%s" % (dumpDbPath, os.sep, "%s-%d.bin" % (column, randomInt(8))), "wb") as f:
|
||||
#from extra.safe2bin.safe2bin import safechardecode
|
||||
#_ = _.encode(UNICODE_ENCODING)
|
||||
#_ = safechardecode(value)
|
||||
#f.write(_)
|
||||
# mimetype = magic.from_buffer(value, mime=True)
|
||||
# if mimetype.startswith("application") or mimetype.startswith("image"):
|
||||
# with codecs.open("%s%s%s" % (dumpDbPath, os.sep, "%s-%d.bin" % (column, randomInt(8))), "wb", UNICODE_ENCODING) as f:
|
||||
# _ = safechardecode(value)
|
||||
# f.write(_)
|
||||
|
||||
if conf.dumpFormat == DUMP_FORMAT.CSV:
|
||||
if field == fields:
|
||||
|
|
|
@ -51,24 +51,31 @@ class Connector(GenericConnector):
|
|||
try:
|
||||
return self.cursor.fetchall()
|
||||
except (pymssql.ProgrammingError, pymssql.OperationalError, _mssql.MssqlDatabaseException), msg:
|
||||
logger.log(logging.WARN if conf.dbmsHandler else logging.DEBUG, "(remote) %s" % msg)
|
||||
logger.log(logging.WARN if conf.dbmsHandler else logging.DEBUG, "(remote) %s" % str(msg).replace("\n", " "))
|
||||
return None
|
||||
|
||||
def execute(self, query):
|
||||
retVal = False
|
||||
|
||||
try:
|
||||
self.cursor.execute(utf8encode(query))
|
||||
retVal = True
|
||||
except (pymssql.OperationalError, pymssql.ProgrammingError), msg:
|
||||
logger.log(logging.WARN if conf.dbmsHandler else logging.DEBUG, "(remote) %s" % msg)
|
||||
logger.log(logging.WARN if conf.dbmsHandler else logging.DEBUG, "(remote) %s" % str(msg).replace("\n", " "))
|
||||
except pymssql.InternalError, msg:
|
||||
raise SqlmapConnectionException(msg)
|
||||
|
||||
return retVal
|
||||
|
||||
def select(self, query):
|
||||
self.execute(query)
|
||||
value = self.fetchall()
|
||||
retVal = None
|
||||
|
||||
try:
|
||||
self.connector.commit()
|
||||
except pymssql.OperationalError:
|
||||
pass
|
||||
if self.execute(query):
|
||||
retVal = self.fetchall()
|
||||
|
||||
return value
|
||||
try:
|
||||
self.connector.commit()
|
||||
except pymssql.OperationalError:
|
||||
pass
|
||||
|
||||
return retVal
|
||||
|
|
0
sqlmapapi.py
Normal file → Executable file
0
sqlmapapi.py
Normal file → Executable file
Loading…
Reference in New Issue
Block a user