File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -935,7 +935,7 @@ def setPaths():
935935 paths .SQL_KEYWORDS = os .path .join (paths .SQLMAP_TXT_PATH , "keywords.txt" )
936936 paths .SMALL_DICT = os .path .join (paths .SQLMAP_TXT_PATH , "smalldict.txt" )
937937 paths .USER_AGENTS = os .path .join (paths .SQLMAP_TXT_PATH , "user-agents.txt" )
938- paths .WORDLIST = os .path .join (paths .SQLMAP_TXT_PATH , "wordlist.txt " )
938+ paths .WORDLIST = os .path .join (paths .SQLMAP_TXT_PATH , "wordlist.zip " )
939939 paths .PHPIDS_RULES_XML = os .path .join (paths .SQLMAP_XML_PATH , "phpids_rules.xml" )
940940 paths .ERRORS_XML = os .path .join (paths .SQLMAP_XML_PATH , "errors.xml" )
941941 paths .PAYLOADS_XML = os .path .join (paths .SQLMAP_XML_PATH , "payloads.xml" )
Original file line number Diff line number Diff line change 55See the file 'doc/COPYING' for copying permission
66"""
77
8+ import os
9+ import zipfile
10+
11+ from lib .core .exception import sqlmapDataException
12+
813class Wordlist :
914 """
1015 Iterator for looping over a large dictionaries
@@ -32,7 +37,14 @@ def adjust(self):
3237 self .iter = iter (self .custom )
3338 else :
3439 current = self .filenames [self .index ]
35- self .fp = open (current , "r" )
40+ if os .path .splitext (current )[1 ].lower () == ".zip" :
41+ _ = zipfile .ZipFile (current , 'r' )
42+ if len (_ .namelist ()) == 0 :
43+ errMsg = "no file(s) inside '%s'" % current
44+ raise sqlmapDataException , errMsg
45+ self .fp = _ .open (_ .namelist ()[0 ])
46+ else :
47+ self .fp = open (current , 'r' )
3648 self .iter = iter (self .fp )
3749
3850 self .index += 1
You can’t perform that action at this time.
0 commit comments