minor cosmetics

This commit is contained in:
Miroslav Stampar 2011-01-19 22:48:06 +00:00
parent c106dc829a
commit 4bdc19d879

View File

@ -33,13 +33,14 @@ from lib.utils.resume import resume
reqCount = 0 reqCount = 0
def __configUnionChar(char): def configUnion(char=None, columns=None):
def __configUnionChar(char):
if char.isdigit() or char == "NULL": if char.isdigit() or char == "NULL":
conf.uChar = char conf.uChar = char
elif not char.startswith("'") or not char.endswith("'"): elif not char.startswith("'") or not char.endswith("'"):
conf.uChar = "'%s'" % char conf.uChar = "'%s'" % char
def __configUnionCols(columns): def __configUnionCols(columns):
if "-" not in columns or len(columns.split("-")) != 2: if "-" not in columns or len(columns.split("-")) != 2:
raise sqlmapSyntaxException, "--union-cols must be a range with hyphon (e.g. 1-10)" raise sqlmapSyntaxException, "--union-cols must be a range with hyphon (e.g. 1-10)"
@ -57,7 +58,6 @@ def __configUnionCols(columns):
errMsg += "higher number of columns" errMsg += "higher number of columns"
raise sqlmapSyntaxException, errMsg raise sqlmapSyntaxException, errMsg
def configUnion(char=None, columns=None):
if isinstance(conf.uChar, basestring): if isinstance(conf.uChar, basestring):
__configUnionChar(conf.uChar) __configUnionChar(conf.uChar)
elif isinstance(char, basestring): elif isinstance(char, basestring):