sqlmapproject/sqlmap
1
1
Fork 0
mirror of https://github.com/sqlmapproject/sqlmap.git synced 2024-11-22 01:26:42 +03:00
Code Issues Packages Projects Releases Wiki Activity

i was on some heavy drugs (sys.stdout = fpOut)

Browse Source
This commit is contained in:
Miroslav Stampar 2011-04-15 12:58:56 +00:00
parent 0387654166
commit a883316e22
1 changed files with 11 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
extra/cloak/cloak.py
View File

@ -35,8 +35,13 @@ def cloak(inputFile):
def decloak(inputFile):
f = open(inputFile, 'rb')
data = bz2.decompress(hideAscii(f.read()))
f.close()
try:
data = bz2.decompress(hideAscii(f.read()))
except:
print 'ERROR: the provided input file \'%s\' does not contain valid cloaked content' % inputFile
sys.exit(1)
finally:
f.close()
return data
@ -58,7 +63,7 @@ def main():
parser.error(e)
if not os.path.isfile(args.inputFile):
print 'ERROR: the provided input file \'%s\' is not a regular file' % args.inputFile
print 'ERROR: the provided input file \'%s\' is non existent' % args.inputFile
sys.exit(1)
if not args.decrypt:
@ -72,10 +77,9 @@ def main():
else:
args.outputFile = args.inputFile[:-1]
fpOut = open(args.outputFile, 'wb')
sys.stdout = fpOut
sys.stdout.write(data)
sys.stdout.close()
f = open(args.outputFile, 'wb')
f.write(data)
f.close()
if __name__ == '__main__':
main()