mirror of
				https://github.com/sqlmapproject/sqlmap.git
				synced 2025-10-31 16:07:55 +03:00 
			
		
		
		
	
		
			
				
	
	
		
			102 lines
		
	
	
		
			3.2 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
			
		
		
	
	
			102 lines
		
	
	
		
			3.2 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
| #!/usr/bin/env python
 | |
| 
 | |
| """
 | |
| $Id$
 | |
| 
 | |
| Copyright (c) 2006-2012 sqlmap developers (http://www.sqlmap.org/)
 | |
| See the file 'doc/COPYING' for copying permission
 | |
| """
 | |
| 
 | |
| import codecs
 | |
| 
 | |
| from ConfigParser import MissingSectionHeaderError
 | |
| 
 | |
| from lib.core.common import checkFile
 | |
| from lib.core.common import UnicodeRawConfigParser
 | |
| from lib.core.data import conf
 | |
| from lib.core.data import logger
 | |
| from lib.core.exception import sqlmapMissingMandatoryOptionException
 | |
| from lib.core.exception import sqlmapSyntaxException
 | |
| from lib.core.optiondict import optDict
 | |
| from lib.core.settings import UNICODE_ENCODING
 | |
| 
 | |
| config = None
 | |
| 
 | |
| def configFileProxy(section, option, boolean=False, integer=False):
 | |
|     """
 | |
|     Parse configuration file and save settings into the configuration
 | |
|     advanced dictionary.
 | |
|     """
 | |
| 
 | |
|     global config
 | |
| 
 | |
|     if config.has_option(section, option):
 | |
|         if boolean:
 | |
|             value = config.getboolean(section, option) if config.get(section, option) else False
 | |
|         elif integer:
 | |
|             value = config.getint(section, option) if config.get(section, option) else 0
 | |
|         else:
 | |
|             value = config.get(section, option)
 | |
| 
 | |
|         if value:
 | |
|             conf[option] = value
 | |
|         else:
 | |
|             conf[option] = None
 | |
|     else:
 | |
|         debugMsg = "missing requested option '%s' (section " % option
 | |
|         debugMsg += "'%s') into the configuration file, " % section
 | |
|         debugMsg += "ignoring. Skipping to next."
 | |
|         logger.debug(debugMsg)
 | |
| 
 | |
| def configFileParser(configFile):
 | |
|     """
 | |
|     Parse configuration file and save settings into the configuration
 | |
|     advanced dictionary.
 | |
|     """
 | |
| 
 | |
|     global config
 | |
| 
 | |
|     debugMsg = "parsing configuration file"
 | |
|     logger.debug(debugMsg)
 | |
| 
 | |
|     checkFile(configFile)
 | |
|     configFP = codecs.open(configFile, "rb", UNICODE_ENCODING)
 | |
| 
 | |
|     try:
 | |
|         config = UnicodeRawConfigParser()
 | |
|         config.readfp(configFP)
 | |
|     except MissingSectionHeaderError:
 | |
|         errMsg = "you have provided an invalid configuration file"
 | |
|         raise sqlmapSyntaxException, errMsg
 | |
| 
 | |
|     if not config.has_section("Target"):
 | |
|         errMsg = "missing a mandatory section 'Target' in the configuration file"
 | |
|         raise sqlmapMissingMandatoryOptionException, errMsg
 | |
| 
 | |
|     condition = not config.has_option("Target", "url")
 | |
|     condition &= not config.has_option("Target", "logFile")
 | |
|     condition &= not config.has_option("Target", "bulkFile")
 | |
|     condition &= not config.has_option("Target", "googleDork")
 | |
|     condition &= not config.has_option("Target", "requestFile")
 | |
|     condition &= not config.has_option("Target", "wizard")
 | |
| 
 | |
|     if condition:
 | |
|         errMsg = "missing a mandatory option in the configuration file "
 | |
|         errMsg += "(url, logFile, bulkFile, googleDork, requestFile or wizard)"
 | |
|         raise sqlmapMissingMandatoryOptionException, errMsg
 | |
| 
 | |
|     for family, optionData in optDict.items():
 | |
|         for option, datatype in optionData.items():
 | |
|             boolean = False
 | |
|             integer = False
 | |
| 
 | |
|             if isinstance(datatype, (list, tuple, set)):
 | |
|                 datatype = datatype[0]
 | |
| 
 | |
|             if datatype == "boolean":
 | |
|                 boolean = True
 | |
|             elif datatype == "integer":
 | |
|                 integer = True
 | |
| 
 | |
|             configFileProxy(family, option, boolean, integer)
 |