@@ -73,11 +73,19 @@ def tableExists(tableFile, regex=None):
7373 errMsg += "to distinguish erroneous results)"
7474 raise SqlmapDataException (errMsg )
7575
76- tables = getFileItems (tableFile , lowercase = Backend .getIdentifiedDbms () in (DBMS .ACCESS ,), unique = True )
76+ message = "which common tables (wordlist) file do you want to use?\n "
77+ message += "[1] default '%s' (press Enter)\n " % tableFile
78+ message += "[2] custom"
79+ choice = readInput (message , default = '1' )
80+
81+ if choice == '2' :
82+ message = "what's the custom common tables file location?\n "
83+ tableFile = readInput (message ) or tableFile
7784
7885 infoMsg = "checking table existence using items from '%s'" % tableFile
7986 logger .info (infoMsg )
8087
88+ tables = getFileItems (tableFile , lowercase = Backend .getIdentifiedDbms () in (DBMS .ACCESS ,), unique = True )
8189 tables .extend (_addPageTextWords ())
8290 tables = filterListValue (tables , regex )
8391
@@ -180,6 +188,15 @@ def columnExists(columnFile, regex=None):
180188 errMsg += "to distinguish erroneous results)"
181189 raise SqlmapDataException (errMsg )
182190
191+ message = "which common columns (wordlist) file do you want to use?\n "
192+ message += "[1] default '%s' (press Enter)\n " % columnFile
193+ message += "[2] custom"
194+ choice = readInput (message , default = '1' )
195+
196+ if choice == '2' :
197+ message = "what's the custom common columns file location?\n "
198+ columnFile = readInput (message ) or columnFile
199+
183200 infoMsg = "checking column existence using items from '%s'" % columnFile
184201 logger .info (infoMsg )
185202
0 commit comments