Minor fix

This commit is contained in:
Miroslav Stampar
2012-12-30 11:22:23 +01:00
parent 75edb84a71
commit 0795760255
2 changed files with 14 additions and 9 deletions

View File

@@ -730,8 +730,13 @@ def dataToTrafficFile(data):
if not conf.trafficFile:
return
conf.trafficFP.write(data)
conf.trafficFP.flush()
try:
conf.trafficFP.write(data)
conf.trafficFP.flush()
except IOError, ex:
errMsg = "something went wrong while trying "
errMsg += "to write to the traffic file '%s' ('%s')" % (conf.trafficFile, ex)
raise SqlmapGenericException, errMsg
def dataToDumpFile(dumpFile, data):
dumpFile.write(data)