Minor bug fixes to -d

This commit is contained in:
Bernardo Damele 2010-06-25 13:24:43 +00:00
parent ccfc9b0fec
commit 9ea72f9640
2 changed files with 2 additions and 2 deletions

View File

@ -72,7 +72,7 @@ def direct(query, content=True):
out = list(output)[0][0]
if isinstance(out, str):
out = utf8decode(out)
return out
return getUnicode(out)
else:
return list(output)
else:

View File

@ -66,7 +66,7 @@ class Filesystem(GenericFilesystem):
length = inject.getValue("SELECT LENGTH(%s) FROM %s" % (self.tblField, self.fileTblName), sort=False, resumeValue=False, charsetType=2)
if not length.isdigit() or not len(length) or length in ( "0", "1" ):
if length is None or not length.isdigit() or not len(length) or length in ( "0", "1" ):
errMsg = "unable to retrieve the content of the "
errMsg += "file '%s'" % rFile
raise sqlmapNoneDataException, errMsg