After the storm, a restore..

lib/core/shell.py

@@ -0,0 +1,103 @@
+#!/usr/bin/env python
+
+"""
+$Id: shell.py 259 2008-07-20 22:25:50Z inquisb $
+
+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)