From cc3f387551b406a2eb044fa0bd8e73dae70708de Mon Sep 17 00:00:00 2001 From: Miroslav Stampar Date: Fri, 5 Oct 2012 10:49:31 +0200 Subject: [PATCH] Patch for an Issue #127 --- lib/controller/action.py | 29 +++++++++++++++++++++++------ 1 file changed, 23 insertions(+), 6 deletions(-) diff --git a/lib/controller/action.py b/lib/controller/action.py index 8815016a6..1b73e0b50 100644 --- a/lib/controller/action.py +++ b/lib/controller/action.py @@ -11,7 +11,9 @@ from lib.core.common import Format from lib.core.common import dataToStdout from lib.core.data import conf from lib.core.data import kb +from lib.core.data import logger from lib.core.data import paths +from lib.core.exception import sqlmapNoneDataException from lib.core.exception import sqlmapUnsupportedDBMSException from lib.core.settings import SUPPORTED_DBMS from lib.techniques.brute.use import columnExists @@ -74,16 +76,31 @@ def action(): conf.dumper.users(conf.dbmsHandler.getUsers()) if conf.getPasswordHashes: - conf.dumper.userSettings("database management system users password hashes", - conf.dbmsHandler.getPasswordHashes(), "password hash") + try: + conf.dumper.userSettings("database management system users password hashes", + conf.dbmsHandler.getPasswordHashes(), "password hash") + except sqlmapNoneDataException, ex: + logger.critical(ex) + except: + raise if conf.getPrivileges: - conf.dumper.userSettings("database management system users privileges", - conf.dbmsHandler.getPrivileges(), "privilege") + try: + conf.dumper.userSettings("database management system users privileges", + conf.dbmsHandler.getPrivileges(), "privilege") + except sqlmapNoneDataException, ex: + logger.critical(ex) + except: + raise if conf.getRoles: - conf.dumper.userSettings("database management system users roles", - conf.dbmsHandler.getRoles(), "role") + try: + conf.dumper.userSettings("database management system users roles", + conf.dbmsHandler.getRoles(), "role") + except sqlmapNoneDataException, ex: + logger.critical(ex) + except: + raise if conf.getDbs: conf.dumper.dbs(conf.dbmsHandler.getDbs())