sqlmapproject/sqlmap
1
1
Fork 0
mirror of https://github.com/sqlmapproject/sqlmap.git synced 2025-03-03 11:45:46 +03:00
Code Issues Packages Projects Releases Wiki Activity

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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
lib/core/bigarray.py
View File

@ -7,6 +7,7 @@ Copyright (c) 2006-2012 sqlmap developers (http://www.sqlmap.org/)
See the file 'doc/COPYING' for copying permission
"""
import os
import pickle
import tempfile

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

1
lib/core/threads.py
View File

@ -61,6 +61,7 @@ def getCurrentThreadUID():
return hash(threading.currentThread())
def readInput(message, default=None):
# It will be overwritten by original from lib.core.common
pass
def getCurrentThreadData():