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

Minor bug fixes and code refactoring

Browse Source
This commit is contained in:
Bernardo Damele
2010-05-28 15:57:43 +00:00
parent 919a8345d6
commit a138dbe5f6
3 changed files with 51 additions and 56 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
lib/core/common.py
View File

@@ -37,6 +37,8 @@ import ntpath
import posixpath
import subprocess
from ConfigParser import DEFAULTSECT
from ConfigParser import RawConfigParser
from StringIO import StringIO
from subprocess import PIPE
from subprocess import Popen as execute
@@ -1365,3 +1367,22 @@ def getBruteUnicode(string):
for char in string:
retVal += unichr(ord(char))
return retVal
class UnicodeRawConfigParser(RawConfigParser):
def write(self, fp):
"""Write an .ini-format representation of the configuration state."""
if self._defaults:
fp.write("[%s]\n" % DEFAULTSECT)
for (key, value) in self._defaults.items():
fp.write("%s = %s\n" % (key, unicode(value).replace('\n', '\n\t')))
fp.write("\n")
for section in self._sections:
fp.write("[%s]\n" % section)
for (key, value) in self._sections[section].items():
if key != "__name__":
if value is None:
fp.write("%s\n" % (key))
else:
fp.write("%s = %s\n" %
(key, unicode(value).replace('\n', '\n\t')))
fp.write("\n")