ronin 2.0.0 → 2.0.2

data/lib/ronin/cli/commands/encode.rb

@@ -61,14 +61,14 @@ module Ronin
       class Encode < StringMethodsCommand
 
         option :base16, desc: 'Base16 encodes the data' do
-                          require 'ronin/support/encoding/base16'
-                          @method_calls << :base16_encode
-                        end
+          require 'ronin/support/encoding/base16'
+          @method_calls << :base16_encode
+        end
 
         option :base32, desc: 'Base32 encodes the data' do
-                          require 'ronin/support/encoding/base32'
-                          @method_calls << :base32_encode
-                        end
+          require 'ronin/support/encoding/base32'
+          @method_calls << :base32_encode
+        end
 
         option :base64, short: '-b',
                         equals: true,
@@ -78,11 +78,11 @@ module Ronin
                         },
                         desc: 'Base64 encodes the data' do |mode=nil|
                           require 'ronin/support/encoding/base64'
-                          if mode
-                            @method_calls << [:base64_encode, [mode]]
-                          else
-                            @method_calls << :base64_encode
-                          end
+                          @method_calls << if mode
+                                             [:base64_encode, [mode]]
+                                           else
+                                             :base64_encode
+                                           end
                         end
 
         option :zlib, short: '-z',
@@ -122,9 +122,9 @@ module Ronin
                      end
 
         option :http, desc: 'HTTP encodes the data' do
-                        require 'ronin/support/encoding/http'
-                        @method_calls << :http_encode
-                      end
+          require 'ronin/support/encoding/http'
+          @method_calls << :http_encode
+        end
 
         option :js, short: '-j',
                     desc: 'JavaScript encodes the data' do

data/lib/ronin/cli/commands/extract.rb

@@ -122,7 +122,7 @@ module Ronin
         def run(*files)
           unless @pattern
             print_error "must specify a pattern to search for"
-            exit -1
+            exit(-1)
           end
 
           super(*files)

data/lib/ronin/cli/commands/grep.rb

@@ -136,7 +136,7 @@ module Ronin
         def run(*files)
           unless @pattern
             print_error "must specify a pattern to search for"
-            exit -1
+            exit(-1)
           end
 
           super(*files)
@@ -151,8 +151,8 @@ module Ronin
         def process_input(input)
           filename = filename_of(input)
 
-          input.each_line(chomp: true).each_with_index do |line,index|
-            match_line(line, filename: filename, line_number: index+1)
+          input.each_line(chomp: true).with_index do |line,index|
+            match_line(line, filename: filename, line_number: index + 1)
           end
         end
 
@@ -184,7 +184,9 @@ module Ronin
         #
         def match_line(line,**kwargs)
           index = 0
+
           printed_prefix = false
+          only_matching  = options[:only_matching]
 
           while (match = line.match(@pattern,index))
             unless printed_prefix
@@ -195,13 +197,13 @@ module Ronin
             match_start, match_stop = match.offset(0)
 
             # print the text before the match, unless --only-matching is enabled
-            print(line[index...match_start]) unless options[:only_matching]
+            print(line[index...match_start]) unless only_matching
             print_match(match)
 
             index = match_stop
           end
 
-          unless options[:only_matching]
+          unless only_matching
             # print the rest of the line, if we've had at least one match
             puts(line[index..]) if index > 0
           end

data/lib/ronin/cli/commands/hexdump.rb

@@ -60,7 +60,7 @@ module Ronin
       # ## Arguments
       #
       #    [FILE]                           Optional file to hexdump
-      # 
+      #
       class Hexdump < FileProcessorCommand
 
         # Supported types for the `-t,--type` option.
@@ -149,7 +149,7 @@ module Ronin
         option :length, short: '-L',
                         value: {
                           type: Integer,
-                          usage: 'LEN',
+                          usage: 'LEN'
                         },
                         desc: 'Length of data to hexdump'
 
@@ -399,7 +399,7 @@ module Ronin
             end
 
             regexp = Regexp.new(value[1...index])
-            style  = parse_style(value[index+2..])
+            style  = parse_style(value[(index + 2)..])
 
             return [regexp, style]
           else
@@ -408,7 +408,7 @@ module Ronin
             end
 
             pattern = value[0...index]
-            style   = parse_style(value[index+1..])
+            style   = parse_style(value[(index + 1)..])
 
             return [pattern, style]
           end
@@ -417,6 +417,7 @@ module Ronin
         # List of command `options` that directly map to the keyword arguments
         # of `Hexdump.hexdump`.
         HEXDUMP_OPTIONS = [
+          :type,
           :format,
           :offset,
           :length,
@@ -445,34 +446,65 @@ module Ronin
             kwargs[key] = options[key] if options.has_key?(key)
           end
 
-          if (index_style   = options[:style_index]) ||
-             (numeric_style = options[:style_numeric]) ||
-             (chars_style   = options[:style_chars])
-            kwargs[:style] = {}
-            kwargs[:style][:index]   = index_style   if index_style
-            kwargs[:style][:numeric] = numeric_style if numeric_style
-            kwargs[:style][:chars]   = chars_style   if chars_style
+          if options.has_key?(:style_index)   ||
+             options.has_key?(:style_numeric) ||
+             options.has_key?(:style_chars)
+            kwargs[:style] = hexdump_style_kwargs
           end
 
           if !@highlight_index.empty?   ||
              !@highlight_numeric.empty? ||
              !@highlight_chars.empty?
-            kwargs[:highlights] = {}
+            kwargs[:highlights] = hexdump_highlights_kwargs
+          end
 
-            unless @highlight_index.empty?
-              kwargs[:highlights][:index] = @highlight_index
-            end
+          return kwargs
+        end
 
-            unless @highlight_numeric.empty?
-              kwargs[:highlights][:numeric] = @highlight_numeric
-            end
+        #
+        # The hexdump `style:` keyword arguments.
+        #
+        # @return [Hash{Symbol => Object}]
+        #
+        def hexdump_style_kwargs
+          style = {}
 
-            unless @highlight_chars.empty?
-              kwargs[:highlights][:chars] = @highlight_chars
-            end
+          if (index_style = options[:style_index])
+            style[:index] = index_style
           end
 
-          return kwargs
+          if (numeric_style = options[:style_numeric])
+            style[:numeric] = numeric_style
+          end
+
+          if (chars_style = options[:style_chars])
+            style[:chars] = chars_style
+          end
+
+          return style
+        end
+
+        #
+        # The hexdump `highlights:` keyword arguments.
+        #
+        # @return [Hash{Symbol => Object}]
+        #
+        def hexdump_highlights_kwargs
+          highlights = {}
+
+          unless @highlight_index.empty?
+            highlights[:index] = @highlight_index
+          end
+
+          unless @highlight_numeric.empty?
+            highlights[:numeric] = @highlight_numeric
+          end
+
+          unless @highlight_chars.empty?
+            highlights[:chars] = @highlight_chars
+          end
+
+          return highlights
         end
 
       end

data/lib/ronin/cli/commands/hmac.rb

@@ -57,7 +57,7 @@ module Ronin
         option :hash, short: '-H',
                       value: {
                         type:    [:md5, :sha1, :sha256, :sha512],
-                        default: :sha1,
+                        default: :sha1
                       },
                       desc: 'Hash algorithm to use'
 

data/lib/ronin/cli/commands/host.rb

@@ -30,17 +30,17 @@ module Ronin
       #
       # ## Usage
       #
-      #     ronin host [options] [HOST ...]
+      #     ronin host [options] {HOST ... | --file FILE}
       #
       # ## Options
       #
       #     -f, --file FILE                  Optional file to read values from
       #         --subdomain SUBNAME          Converts the hostname to a sub-domain
       #     -d, --domain                     Converts the hostname to a domain
-      #     -T, --tld                        Converts the hostname to it's TLD 
+      #     -T, --tld                        Converts the hostname to it's TLD
       #     -s, --suffix                     Converts the hostname to it's suffix
       #     -S, --change-suffix SUFFIX       Changes the suffix of each hostname
-      #         --enum-tlds                  Enumerates over every TLD 
+      #         --enum-tlds                  Enumerates over every TLD
       #         --enum-suffixes[={icann|private}]
       #                                      Enumerates over every domain suffix
       #         --enum-subdomains FILE       Enumerates over every subdomain in the wordlist
@@ -63,7 +63,7 @@ module Ronin
 
         include DNS
 
-        usage '[options] [HOST ...]'
+        usage '[options] {HOST ... | --file FILE}'
 
         option :subdomain, value: {
                              type:  String,
@@ -117,20 +117,20 @@ module Ronin
         option :has_records, short: '-H',
                              value: {
                                type: {
-                                 A: :a,
-                                 AAAA: :aaaa,
-                                 ANY: :any,
+                                 A:     :a,
+                                 AAAA:  :aaaa,
+                                 ANY:   :any,
                                  CNAME: :cname,
                                  HINFO: :hinfo,
-                                 LOC: :loc,
+                                 LOC:   :loc,
                                  MINFO: :minfo,
-                                 MX: :mx,
-                                 NS: :ns,
-                                 PTR: :ptr,
-                                 SOA: :soa,
-                                 SRV: :srv,
-                                 TXT: :txt,
-                                 WKS: :wks
+                                 MX:    :mx,
+                                 NS:    :ns,
+                                 PTR:   :ptr,
+                                 SOA:   :soa,
+                                 SRV:   :srv,
+                                 TXT:   :txt,
+                                 WKS:   :wks
                                }
                              },
                              desc: 'Filters hostnames that have a certain DNS record type'

data/lib/ronin/cli/commands/http.rb

@@ -22,6 +22,7 @@ require 'ronin/cli/http_shell'
 require 'ronin/support/network/http'
 
 require 'command_kit/options/verbose'
+require 'addressable/uri'
 
 module Ronin
   class CLI
@@ -74,6 +75,9 @@ module Ronin
         include CommandKit::Options::Verbose
         include Printing::HTTP
 
+        # `http://` and `https://` URL validation regex.
+        URL_REGEX = URI::DEFAULT_PARSER.make_regexp(%w[http https])
+
         usage '[options] {URL [...] | --shell URL}'
 
         option :method, value: {
@@ -147,10 +151,14 @@ module Ronin
         end
 
         option :shell, value: {
-                         type:  String,
+                         type:  URL_REGEX,
                          usage: 'URL'
                        },
-                       desc: 'Open an interactive HTTP shell'
+                       desc: 'Open an interactive HTTP shell' do |url|
+                         options[:shell] = Addressable::URI.parse(url)
+                       rescue Addressable::URI::InvalidURIError => error
+                         raise(OptionParser::InvalidArgument,"invalid URL: #{error.message}")
+                       end
 
         option :proxy, short: '-P',
                        value: {
@@ -172,11 +180,9 @@ module Ronin
 
         option :user_agent, short: '-u',
                             value: {
-                              type: Hash[
-                                Support::Network::HTTP::UserAgents::ALIASES.keys.map { |key|
-                                  [key.to_s.tr('_','-'), key]
-                                }
-                              ]
+                              type: Support::Network::HTTP::UserAgents::ALIASES.transform_keys { |key|
+                                key.to_s.tr('_','-')
+                              }
                             },
                             desc: 'The User-Agent to use' do |name|
                               @user_agent = name
@@ -326,25 +332,36 @@ module Ronin
         #   The URL to request.
         #
         def process_value(url)
-          url = URI(url)
-
-          Support::Network::HTTP.request(
-            @http_method, url, proxy:        @proxy,
-                               user_agent:   @user_agent,
-                               user:         url.user,
-                               password:     url.password,
-                               query_params: @query_params,
-                               headers:      @headers,
-                               body:         @body,
-                               form_data:    @form_data
-          ) do |response|
-            # NOTE: we must call HTTP.request with a block to avoid causing
-            # #read_body to be called twice.
-            print_response(response)
+          unless url =~ URL_REGEX
+            print_error "invalid URL: #{url.inspect}"
+            return
+          end
+
+          uri = begin
+                  Addressable::URI.parse(url)
+                rescue Addressable::URI::InvalidURIError => error
+                  print_error "invalid URL: #{error.message}"
+                  return
+                end
+
+          begin
+            Support::Network::HTTP.request(
+              @http_method, uri, proxy:        @proxy,
+                                 user_agent:   @user_agent,
+                                 query_params: @query_params,
+                                 headers:      @headers,
+                                 body:         @body,
+                                 form_data:    @form_data
+            ) do |response|
+              # NOTE: we must call HTTP.request with a block to avoid causing
+              # #read_body to be called twice.
+              print_response(response)
+            end
+          rescue StandardError => error
+            if verbose? then print_exception(error)
+            else             print_error(error.message)
+            end
           end
-        rescue => error
-          print_error(error.message)
-          exit(1)
         end
 
         #

data/lib/ronin/cli/commands/ip.rb

@@ -135,12 +135,12 @@ module Ronin
                            desc: 'The query string for the URI'
 
         option :http, desc: 'Converts the IP address into a http:// URI' do
-          options[:uri] = true
+          options[:uri]        = true
           options[:uri_scheme] = 'http'
         end
 
         option :https, desc: 'Converts the IP address into a https:// URI' do
-          options[:uri] = true
+          options[:uri]        = true
           options[:uri_scheme] = 'https'
         end
 
@@ -197,7 +197,9 @@ module Ronin
 
             puts "#{ip}/#{ip.prefix}"
           elsif options[:host]
-            puts ip.host
+            if (host = ip.host)
+              puts host
+            end
           elsif options[:port]
             puts "#{format_ip(ip)}:#{options[:port]}"
           elsif options[:uri]

data/lib/ronin/cli/commands/iprange.rb

@@ -47,7 +47,7 @@ module Ronin
       #    ronin iprange 1.1.2-4.10-50
       #    ronin iprange --start 1.1.1.10 --stop 1.1.4.100
       #    ronin iprange --file list.txt
-      # 
+      #
       class Iprange < ValueProcessorCommand
 
         usage '[options] [IP_RANGE ... | --start IP --stop IP]'
@@ -104,7 +104,7 @@ module Ronin
           if !@start.empty? && !@stop.empty?
             unless @start.length == @stop.length
               print_error "must specify an equal number of --start and --stop options"
-              exit -1
+              exit(-1)
             end
 
             @start.zip(@stop).each do |(start,stop)|

data/lib/ronin/cli/commands/netcat.rb

@@ -132,16 +132,14 @@ module Ronin
 
         option :ssl_key, value: {
                            type: String,
-                           usage: 'FILE',
+                           usage: 'FILE'
                          },
                          desc: 'Specifies the SSL key file'
 
         option :ssl_verify, value: {
-                              type: Hash[
-                                Support::Network::SSL::VERIFY.keys.map { |key|
-                                  [key.to_s.tr('_','-'), key]
-                                }
-                              ]
+                              type: Support::Network::SSL::VERIFY.transform_keys { |key|
+                                key.to_s.tr('_','-')
+                              }
                             },
                             desc: 'SSL verification mode'
 
@@ -301,16 +299,15 @@ module Ronin
         # The client event loop.
         #
         def client_loop
-          endpoint = async_endpoint
-          stdin = async_stdin
-          finished = Async::Notification.new
-
+          finished    = Async::Notification.new
+          endpoint    = async_endpoint
+          stdin       = async_stdin
           buffer_size = options[:buffer_size]
 
           Async do |task|
             socket = begin
                        endpoint.connect
-                     rescue => error
+                     rescue StandardError => error
                        print_error(error.message)
                        exit(1)
                      end
@@ -319,25 +316,23 @@ module Ronin
 
             begin
               client = task.async do
-                begin
-                  while (data = stream.read_partial(buffer_size))
-                    print_data(data)
-                  end
-                rescue EOFError
-                ensure
-                  finished.signal
+                while (data = stream.read_partial(buffer_size))
+                  print_data(data)
                 end
+              rescue EOFError
+                # ignore EOFError
+              ensure
+                finished.signal
               end
 
               user = task.async do
-                begin
-                  while (data = stdin.read_partial(buffer_size))
-                    socket.write(data)
-                  end
-                rescue EOFError
-                ensure
-                  finished.signal
+                while (data = stdin.read_partial(buffer_size))
+                  socket.write(data)
                 end
+              rescue EOFError
+                # ignore EOFError
+              ensure
+                finished.signal
               end
 
               finished.wait
@@ -353,12 +348,11 @@ module Ronin
         # The server event loop.
         #
         def server_loop
-          endpoint = async_endpoint
-          stdin = async_stdin
-          finished = Async::Notification.new
-
+          finished    = Async::Notification.new
+          endpoint    = async_endpoint
+          stdin       = async_stdin
+          clients     = []
           buffer_size = options[:buffer_size]
-          clients = []
 
           Async do |task|
             endpoint.accept do |socket|
@@ -374,6 +368,7 @@ module Ronin
                   print_data(data)
                 end
               rescue EOFError
+                # ignore EOFError
               end
 
               clients.delete(socket)
@@ -384,18 +379,17 @@ module Ronin
             end
 
             task.async do
-              begin
-                while (data = stdin.read_partial(buffer_size))
-                  clients.each { |client| client.write(data) }
-                end
-              rescue EOFError
-              ensure
-                finished.signal
+              while (data = stdin.read_partial(buffer_size))
+                clients.each { |client| client.write(data) }
               end
+            rescue EOFError
+              # ignore EOFError
+            ensure
+              finished.signal
             end
 
             finished.wait
-          rescue => error
+          rescue StandardError => error
             print_error(error.message)
             exit(1)
           ensure

data/lib/ronin/cli/commands/new/project.rb

@@ -23,26 +23,26 @@ require 'ronin/root'
 module Ronin
   class CLI
     module Commands
-      #
-      # Creates new Ruby project directory.
-      #
-      # ## Usage
-      #
-      #     ronin new project [options] DIR
-      #
-      # ## Options
-      #
-      #         --git                        Initializes a git repo
-      #         --ruby-version VERSION       The desired ruby version for the project
-      #         --rakefile                   Creates a Rakefile
-      #     -D, --dockerfile                 Adds a Dockerfile to the new project
-      #     -h, --help                       Print help information
-      #
-      # ## Arguments
-      # 
-      #     PATH                             The directory to create
-      #
       class New < Command
+        #
+        # Creates new Ruby project directory.
+        #
+        # ## Usage
+        #
+        #     ronin new project [options] DIR
+        #
+        # ## Options
+        #
+        #         --git                        Initializes a git repo
+        #         --ruby-version VERSION       The desired ruby version for the project
+        #         --rakefile                   Creates a Rakefile
+        #     -D, --dockerfile                 Adds a Dockerfile to the new project
+        #     -h, --help                       Print help information
+        #
+        # ## Arguments
+        #
+        #     PATH                             The directory to create
+        #
         class Project < Command
 
           include Core::CLI::Generator

data/lib/ronin/cli/commands/new/script.rb

@@ -23,25 +23,25 @@ require 'ronin/root'
 module Ronin
   class CLI
     module Commands
-      #
-      # Creates a new standalone Ruby script.
-      #
-      # ## Usage
-      #
-      #     ronin new script PATH
-      #
-      # ## Arguments
-      #
-      #     PATH                             The script file to create
-      #
       class New < Command
+        #
+        # Creates a new standalone Ruby script.
+        #
+        # ## Usage
+        #
+        #     ronin new script PATH
+        #
+        # ## Arguments
+        #
+        #     PATH                             The script file to create
+        #
         class Script < Command
 
           include Core::CLI::Generator
 
           template_dir File.join(ROOT,'data','new')
 
-          usage' PATH'
+          usage 'PATH'
 
           argument :path, required: true,
                           desc:     'The script file to create'