CERBERUS/sqlmap
1
0
Fork 0
mirror of https://github.com/sqlmapproject/sqlmap.git synced 2026-08-04 06:50:14 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

some more refactoring

Browse source
This commit is contained in:
Miroslav Stampar 2012-02-16 09:54:29 +00:00
parent e1f86c97c4
commit 0e23521adc
3 changed files with 13 additions and 11 deletions
Download patch file Download diff file Expand all files Collapse all files

22
lib/core/common.py
View file

@ -286,12 +286,12 @@ class Backend:
msg += "correct [%s (default)/%s] " % (kb.dbms, dbms)
while True:
inp = readInput(msg, default=kb.dbms)
_ = readInput(msg, default=kb.dbms)
if aliasToDbmsEnum(inp) == kb.dbms:
if aliasToDbmsEnum(_) == kb.dbms:
break
elif aliasToDbmsEnum(inp) == dbms:
kb.dbms = aliasToDbmsEnum(inp)
elif aliasToDbmsEnum(_) == dbms:
kb.dbms = aliasToDbmsEnum(_)
break
else:
warnMsg = "invalid value"
@ -344,12 +344,12 @@ class Backend:
msg += "correct [%s (default)/%s] " % (kb.os, os)
while True:
inp = readInput(msg, default=kb.os)
_ = readInput(msg, default=kb.os)
if inp == kb.os:
if _ == kb.os:
break
elif inp == os:
kb.os = inp.capitalize()
elif _ == os:
kb.os = _.capitalize()
break
else:
warnMsg = "invalid value"
@ -383,10 +383,10 @@ class Backend:
msg += "\n[2] 64-bit"
while True:
arch = readInput(msg, default='1')
_ = readInput(msg, default='1')
if isinstance(arch, basestring) and arch.isdigit() and int(arch) in ( 1, 2 ):
if int(arch) == 1:
if isinstance(_, basestring) and _.isdigit() and int(_) in ( 1, 2 ):
if int(_) == 1:
kb.arch = 32
else:
kb.arch = 64