2008-10-15 19:38:22 +04:00
|
|
|
#!/usr/bin/env python
|
|
|
|
|
|
|
|
|
|
"""
|
2008-10-15 19:56:32 +04:00
|
|
|
$Id$
|
2008-10-15 19:38:22 +04:00
|
|
|
|
|
|
|
|
This file is part of the sqlmap project, http://sqlmap.sourceforge.net.
|
|
|
|
|
|
|
|
|
|
Copyright (c) 2006-2008 Bernardo Damele A. G. <bernardo.damele@gmail.com>
|
|
|
|
|
and Daniele Bellucci <daniele.bellucci@gmail.com>
|
|
|
|
|
|
|
|
|
|
sqlmap is free software; you can redistribute it and/or modify it under
|
|
|
|
|
the terms of the GNU General Public License as published by the Free
|
|
|
|
|
Software Foundation version 2 of the License.
|
|
|
|
|
|
|
|
|
|
sqlmap is distributed in the hope that it will be useful, but WITHOUT ANY
|
|
|
|
|
WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
|
|
|
|
|
FOR A PARTICULAR PURPOSE. See the GNU General Public License for more
|
|
|
|
|
details.
|
|
|
|
|
|
|
|
|
|
You should have received a copy of the GNU General Public License along
|
|
|
|
|
with sqlmap; if not, write to the Free Software Foundation, Inc., 51
|
|
|
|
|
Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
|
|
|
|
|
"""
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
import atexit
|
|
|
|
|
import os
|
|
|
|
|
import rlcompleter
|
|
|
|
|
|
|
|
|
|
from lib.core import readlineng as readline
|
|
|
|
|
from lib.core.data import kb
|
|
|
|
|
from lib.core.data import paths
|
|
|
|
|
from lib.core.data import queries
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def saveHistory():
|
|
|
|
|
historyPath = os.path.expanduser(paths.SQLMAP_HISTORY)
|
|
|
|
|
readline.write_history_file(historyPath)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def loadHistory():
|
|
|
|
|
historyPath = os.path.expanduser(paths.SQLMAP_HISTORY)
|
|
|
|
|
|
|
|
|
|
if os.path.exists(historyPath):
|
|
|
|
|
readline.read_history_file(historyPath)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def queriesForAutoCompletion():
|
|
|
|
|
autoComplQueries = {}
|
|
|
|
|
|
|
|
|
|
for _, query in queries[kb.dbms].items():
|
|
|
|
|
if isinstance(query, str) and len(query) > 1:
|
|
|
|
|
autoComplQuery = query
|
|
|
|
|
elif isinstance(query, dict) and "inband" in query:
|
|
|
|
|
autoComplQuery = query["inband"]["query"]
|
|
|
|
|
|
|
|
|
|
autoComplQueries[autoComplQuery] = None
|
|
|
|
|
|
|
|
|
|
return autoComplQueries
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class CompleterNG(rlcompleter.Completer):
|
|
|
|
|
def global_matches(self, text):
|
|
|
|
|
"""
|
|
|
|
|
Compute matches when text is a simple name.
|
|
|
|
|
Return a list of all names currently defined in self.namespace
|
|
|
|
|
that match.
|
|
|
|
|
"""
|
|
|
|
|
|
|
|
|
|
matches = []
|
|
|
|
|
n = len(text)
|
|
|
|
|
|
|
|
|
|
for list in [ self.namespace ]:
|
|
|
|
|
for word in list:
|
|
|
|
|
if word[:n] == text:
|
|
|
|
|
matches.append(word)
|
|
|
|
|
|
|
|
|
|
return matches
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def autoCompletion(sqlShell=False, osShell=False):
|
|
|
|
|
# First of all we check if the readline is available, by default
|
|
|
|
|
# it is not in Python default installation on Windows
|
|
|
|
|
if not readline.haveReadline:
|
|
|
|
|
return
|
|
|
|
|
|
|
|
|
|
if sqlShell:
|
|
|
|
|
completer = CompleterNG(queriesForAutoCompletion())
|
|
|
|
|
elif osShell:
|
|
|
|
|
# TODO: add more operating system commands; differentiate commands
|
|
|
|
|
# based on future operating system fingerprint
|
|
|
|
|
completer = CompleterNG({
|
|
|
|
|
"id": None, "ifconfig": None, "ls": None,
|
|
|
|
|
"netstat -natu": None, "pwd": None,
|
|
|
|
|
"uname": None, "whoami": None,
|
|
|
|
|
})
|
|
|
|
|
|
|
|
|
|
readline.set_completer(completer.complete)
|
|
|
|
|
readline.parse_and_bind("tab: complete")
|
|
|
|
|
|
|
|
|
|
loadHistory()
|
|
|
|
|
atexit.register(saveHistory)
|
