class Bot
class Wordlist
def self.wordlist_base
- @@wordlist_base ||= File.join(Utils.bot.botclass, 'wordlists')
+ @@wordlist_base ||= Utils.bot.path 'wordlists'
end
- def self.get(path, options={})
+ def self.get(where, options={})
opts = { :spaces => false }.merge(options)
- wordlist_path = File.join(wordlist_base, path)
+ wordlist_path = File.join(wordlist_base, where)
raise "wordlist not found: #{wordlist_path}" unless File.exist?(wordlist_path)
# Location is a directory -> combine all lists beneath it
File.readlines(wordlist_path)
end
- wordlist.map! { |l| l.strip }
+ # wordlists are assumed to be UTF-8, but we need to strip the BOM, if present
+ wordlist.map! { |l| l.sub("\xef\xbb\xbf",'').strip }
wordlist.reject do |word|
word =~ /\s/ && !opts[:spaces] ||
word.empty?
name[striplen..-1]
}
end
+
+ def self.exist?(path)
+ fn = path.to_s
+ # refuse to check outside of the wordlist base directory
+ return false if fn =~ /\.\.\//
+ File.exist?(File.join(self.wordlist_base, fn))
+ end
+
end
end
end