ronin-support 0.4.1 → 0.5.0.rc1

data/lib/ronin/ui/output/output.rb

@@ -26,13 +26,13 @@ module Ronin
     # Controls {Output} from Ronin.
     #
     module Output
-      @mode = if ($VERBOSE || $DEBUG || ENV['VERBOSE'])
+      @mode = if ($VERBOSE || $DEBUG)
                 :verbose
               else
                 :quiet
               end
 
-      @handler = if $stdout.tty?
+      @handler = if STDOUT.tty?
                    Terminal::Color
                  else
                    Terminal::Raw

data/lib/ronin/ui/output/terminal/color.rb

@@ -38,6 +38,12 @@ module Ronin
           # ANSI Red code
           RED = "\e[31m"
 
+          # ANSI Bright code
+          BRIGHT = "\e[1m"
+
+          # ANSI Bright-Off code
+          BRIGHT_OFF = "\e[21m"
+
           # ANSI Clear code
           CLEAR = "\e[0m"
 
@@ -66,7 +72,7 @@ module Ronin
           # @api private
           #
           def self.print_info(message)
-            $stdout.puts "#{GREEN}[-] #{message}#{CLEAR}"
+            $stdout.puts "#{GREEN}#{BRIGHT}[-]#{BRIGHT_OFF} #{message}#{CLEAR}"
           end
 
           #
@@ -80,7 +86,7 @@ module Ronin
           # @api private
           #
           def self.print_debug(message)
-            $stdout.puts "#{CYAN}[=] #{message}#{CLEAR}"
+            $stdout.puts "#{CYAN}#{BRIGHT}[=]#{BRIGHT_OFF} #{message}#{CLEAR}"
           end
 
           #
@@ -94,7 +100,7 @@ module Ronin
           # @api private
           #
           def self.print_warning(message)
-            $stdout.puts "#{YELLOW}[*] #{message}#{CLEAR}"
+            $stdout.puts "#{YELLOW}#{BRIGHT}[*]#{BRIGHT_OFF} #{message}#{CLEAR}"
           end
 
           #
@@ -108,7 +114,7 @@ module Ronin
           # @api private
           #
           def self.print_error(message)
-            $stdout.puts "#{RED}[!] #{message}#{CLEAR}"
+            $stdout.puts "#{RED}#{BRIGHT}[!]#{BRIGHT_OFF} #{message}#{CLEAR}"
           end
 
         end

data/lib/ronin/wordlist.rb

@@ -17,6 +17,7 @@
 # along with Ronin Support.  If not, see <http://www.gnu.org/licenses/>.
 #
 
+require 'ronin/extensions/regexp'
 require 'ronin/fuzzing/extensions'
 
 require 'set'
@@ -31,8 +32,15 @@ module Ronin
 
     include Enumerable
 
-    # The path to the wordlist file or a list of words
-    attr_accessor :list
+    # The path to the wordlist file
+    #
+    # @since 0.5.0
+    attr_reader :path
+
+    # The words for the list
+    #
+    # @since 0.5.0
+    attr_reader :words
 
     # Mutation rules to apply to every word in the list
     attr_reader :mutations
@@ -40,7 +48,7 @@ module Ronin
     #
     # Initializes the wordlist.
     #
-    # @param [String, Enumerable] list
+    # @param [String, Enumerable] wordlist
     #   The path of the wordlist or list of words.
     #
     # @param [Hash{Regexp,String,Symbol => Symbol,#each}] mutations
@@ -52,6 +60,9 @@ module Ronin
     # @yieldparam [Wordlist] wordlist
     #   The new wordlist object.
     #
+    # @raise [TypeError]
+    #   The list was not a path to a wordlist file, nor a list of words.
+    #
     # @example Use a file wordlist
     #   wordlist = Wordlist.new('passwords.txt')
     #
@@ -65,10 +76,19 @@ module Ronin
     #
     # @api public
     #
-    def initialize(list,mutations={})
-      @list      = list
-      @mutations = {}
-      @mutations.merge!(mutations)
+    def initialize(wordlist,mutations={})
+      case wordlist
+      when String
+        @path  = wordlist
+        @words = nil
+      when Enumerable
+        @path  = nil
+        @words = wordlist
+      else
+        raise(TypeError,"wordlist must be a path or Enumerable")
+      end
+
+      @mutations = mutations
 
       yield self if block_given?
     end
@@ -93,7 +113,7 @@ module Ronin
       words_seen = SortedSet[]
 
       text.each_line do |line|
-        line.split.each do |word|
+        line.scan(Regexp::WORD) do |word|
           if block_given?
             yield word unless words_seen.include?(word)
           end
@@ -121,6 +141,41 @@ module Ronin
       new(parse(text),mutations)
     end
 
+    #
+    # Creates a new wordlist file.
+    #
+    # @param [String] path
+    #   The path to the wordlist file.
+    #
+    # @param [String] text
+    #   The text to parse.
+    #
+    # @param [Hash{Regexp,String,Symbol => Symbol,#each}] mutations
+    #   Additional mutations for the wordlist.
+    #
+    # @return [Wordlist]
+    #   The newly built wordlist.
+    #
+    # @since 0.5.0
+    #
+    def self.create(path,text,mutations={})
+      wordlist = build(text,mutations)
+
+      return wordlist.save(path)
+    end
+
+    #
+    # The wordlist file or list of words.
+    #
+    # @return [String, Enumerable]
+    #   The path to the wordlist file or list of words.
+    #
+    # @api semipublic
+    #
+    def list
+      @path || @words
+    end
+
     #
     # Iterates over each word in the list.
     #
@@ -133,25 +188,22 @@ module Ronin
     # @return [Enumerator]
     #   If no block is given, an Enumerator will be returned.
     #
-    # @raise [TypeError]
-    #   The list was not a path to a wordlist file, nor a list of words.
+    # @raise [RuntimeError]
+    #   {#path} or {#words} must be set.
     #
     # @api public
     #
     def each_word(&block)
       return enum_for(:each_word) unless block
 
-      case @list
-      when String
-        File.open(@list) do |file|
+      if @path
+        File.open(@path) do |file|
           file.each_line do |line|
             yield line.chomp
           end
         end
-      when Enumerable
-        @list.each(&block)
-      else
-        raise(TypeError,"list must be a path or Enumerable")
+      elsif @words
+        @words.each(&block)
       end
     end
 
@@ -203,5 +255,28 @@ module Ronin
       String.generate([each, n],&block)
     end
 
+    #
+    # Saves the words to a new file.
+    #
+    # @param [String] path
+    #   The path to the new wordlist file.
+    #
+    # @return [Wordlist]
+    #   The wordlist object.
+    #
+    # @see #each
+    #
+    # @since 0.5.0
+    #
+    # @api public
+    #
+    def save(path)
+      File.open(path,'w') do |file|
+        each { |word| file.puts word }
+      end
+
+      return self
+    end
+
   end
 end

data/ronin-support.gemspec

@@ -2,130 +2,59 @@
 
 require 'yaml'
 
-Gem::Specification.new do |gemspec|
-  root = File.dirname(__FILE__)
-  lib_dir = File.join(root,'lib')
-  files = if File.directory?('.git')
-            `git ls-files`.split($/)
-          elsif File.directory?('.hg')
-            `hg manifest`.split($/)
-          elsif File.directory?('.svn')
-            `svn ls -R`.split($/).select { |path| File.file?(path) }
-          else
-            Dir['{**/}{.*,*}'].select { |path| File.file?(path) }
-          end
+Gem::Specification.new do |gem|
+  gemspec = YAML.load_file('gemspec.yml')
 
-  filter_files = lambda { |paths|
-    case paths
-    when Array
-      (files & paths)
-    when String
-      (files & Dir[paths])
-    end
-  }
-
-  version = {
-    :file => 'ronin/support/version',
-    :constant => 'Ronin::Support::VERSION'
-  }
-
-  defaults = {
-    'name' => File.basename(root),
-    'files' => files,
-    'executables' => filter_files['bin/*'].map { |path| File.basename(path) },
-    'test_files' => filter_files['{test/{**/}*_test.rb,spec/{**/}*_spec.rb}'],
-    'extra_doc_files' => filter_files['*.{txt,rdoc,md,markdown,tt,textile}'],
-  }
+  gem.name    = gemspec.fetch('name')
+  gem.version = gemspec.fetch('version') do
+                  lib_dir = File.join(File.dirname(__FILE__),'lib')
+                  $LOAD_PATH << lib_dir unless $LOAD_PATH.include?(lib_dir)
 
-  metadata = defaults.merge(YAML.load_file('gemspec.yml'))
+                  require 'ronin/support/version'
+                  Ronin::Support::VERSION
+                end
 
-  gemspec.name = metadata.fetch('name',defaults[:name])
-  gemspec.version = if metadata['version']
-                      metadata['version']
-                    else
-                      $LOAD_PATH << lib_dir unless $LOAD_PATH.include?(lib_dir)
+  gem.summary     = gemspec['summary']
+  gem.description = gemspec['description']
+  gem.licenses    = Array(gemspec['license'])
+  gem.authors     = Array(gemspec['authors'])
+  gem.email       = gemspec['email']
+  gem.homepage    = gemspec['homepage']
 
-                      require version[:file]
-                      eval(version[:constant])
-                    end
+  glob = lambda { |patterns| gem.files & Dir[*patterns] }
 
-  gemspec.summary = metadata.fetch('summary',metadata['description'])
-  gemspec.description = metadata.fetch('description',metadata['summary'])
+  gem.files = `git ls-files`.split($/)
+  gem.files = glob[gemspec['files']] if gemspec['files']
 
-  case metadata['license']
-  when Array
-    gemspec.licenses = metadata['license']
-  when String
-    gemspec.license = metadata['license']
+  gem.executables = gemspec.fetch('executables') do
+    glob['bin/*'].map { |path| File.basename(path) }
   end
+  gem.default_executable = gem.executables.first if Gem::VERSION < '1.7.'
 
-  case metadata['authors']
-  when Array
-    gemspec.authors = metadata['authors']
-  when String
-    gemspec.author = metadata['authors']
-  end
-
-  gemspec.email = metadata['email']
-  gemspec.homepage = metadata['homepage']
+  gem.extensions       = glob[gemspec['extensions'] || 'ext/**/extconf.rb']
+  gem.test_files       = glob[gemspec['test_files'] || '{test/{**/}*_test.rb']
+  gem.extra_rdoc_files = glob[gemspec['extra_doc_files'] || '*.{txt,md}']
 
-  case metadata['require_paths']
-  when Array
-    gemspec.require_paths = metadata['require_paths']
-  when String
-    gemspec.require_path = metadata['require_paths']
-  end
+  gem.require_paths = Array(gemspec.fetch('require_paths') {
+    %w[ext lib].select { |dir| File.directory?(dir) }
+  })
 
-  gemspec.files = filter_files[metadata['files']]
+  gem.requirements              = gemspec['requirements']
+  gem.required_ruby_version     = gemspec['required_ruby_version']
+  gem.required_rubygems_version = gemspec['required_rubygems_version']
+  gem.post_install_message      = gemspec['post_install_message']
 
-  gemspec.executables = metadata['executables']
-  gemspec.extensions = metadata['extensions']
-
-  if Gem::VERSION < '1.7.'
-    gemspec.default_executable = gemspec.executables.first
-  end
-
-  gemspec.test_files = filter_files[metadata['test_files']]
-
-  unless gemspec.files.include?('.document')
-    gemspec.extra_rdoc_files = metadata['extra_doc_files']
-  end
-
-  gemspec.post_install_message = metadata['post_install_message']
-  gemspec.requirements = metadata['requirements']
-
-  if gemspec.respond_to?(:required_ruby_version=)
-    gemspec.required_ruby_version = metadata['required_ruby_version']
-  end
-
-  if gemspec.respond_to?(:required_rubygems_version=)
-    gemspec.required_rubygems_version = metadata['required_rubygems_version']
-  end
-
-  parse_versions = lambda { |versions|
-    case versions
-    when Array
-      versions.map { |v| v.to_s }
-    when String
-      versions.split(/,\s*/)
-    end
-  }
-
-  if metadata['dependencies']
-    metadata['dependencies'].each do |name,versions|
-      gemspec.add_dependency(name,parse_versions[versions])  
-    end
-  end
+  split = lambda { |string| string.split(/,\s*/) }
 
-  if metadata['runtime_dependencies']
-    metadata['runtime_dependencies'].each do |name,versions|
-      gemspec.add_runtime_dependency(name,parse_versions[versions])  
+  if gemspec['dependencies']
+    gemspec['dependencies'].each do |name,versions|
+      gem.add_dependency(name,split[versions])
     end
   end
 
-  if metadata['development_dependencies']
-    metadata['development_dependencies'].each do |name,versions|
-      gemspec.add_development_dependency(name,parse_versions[versions])  
+  if gemspec['development_dependencies']
+    gemspec['development_dependencies'].each do |name,versions|
+      gem.add_development_dependency(name,split[versions])
     end
   end
 end

data/spec/binary/hexdump/helpers/hexdumps.rb

@@ -0,0 +1,13 @@
+module Helpers
+  DIRECTORY = File.join(File.dirname(__FILE__),'hexdumps')
+
+  def load_binary_data(name)
+    File.open(File.join(DIRECTORY,"#{name}.bin"),'rb') do |file|
+      file.read
+    end
+  end
+
+  def load_hexdump(name)
+    File.read(File.join(DIRECTORY,"#{name}.txt"))
+  end
+end