ronin-support 0.4.0.rc1 → 0.4.0.rc2

data/lib/ronin/network/telnet.rb

@@ -1,5 +1,5 @@
 #
-# Copyright (c) 2006-2011 Hal Brodigan (postmodern.mod3 at gmail.com)
+# Copyright (c) 2006-2012 Hal Brodigan (postmodern.mod3 at gmail.com)
 #
 # This file is part of Ronin Support.
 #

data/lib/ronin/network/udp.rb

@@ -1,5 +1,5 @@
 #
-# Copyright (c) 2006-2011 Hal Brodigan (postmodern.mod3 at gmail.com)
+# Copyright (c) 2006-2012 Hal Brodigan (postmodern.mod3 at gmail.com)
 #
 # This file is part of Ronin Support.
 #
@@ -54,17 +54,15 @@ module Ronin
       #   # => UDPSocket
       #
       # @example
-      #   udp_connect('www.wired.com',80) do |sock|
-      #     puts sock.readlines
+      #   udp_connect('www.wired.com',80) do |socket|
+      #     puts socket.readlines
       #   end
       #
       # @api public
       #
       def udp_connect(host,port,local_host=nil,local_port=nil)
-        host = host.to_s
-        local_host = if local_host
-                       local_host.to_s
-                     end
+        host       = host.to_s
+        local_host = (local_host || '0.0.0.0').to_s
 
         socket = UDPSocket.new
         socket.bind(local_host,local_port) if (local_host && local_port)
@@ -105,11 +103,11 @@ module Ronin
       # @api public
       #
       def udp_connect_and_send(data,host,port,local_host=nil,local_port=nil)
-        sock = udp_connect(host,port,local_host,local_port)
-        sock.write(data)
+        socket = udp_connect(host,port,local_host,local_port)
+        socket.write(data)
 
-        yield sock if block_given?
-        return sock
+        yield socket if block_given?
+        return socket
       end
 
       #
@@ -140,14 +138,53 @@ module Ronin
       # @api public
       #
       def udp_session(host,port,local_host=nil,local_port=nil)
-        sock = udp_connect(host,port,local_host,local_port)
+        socket = udp_connect(host,port,local_host,local_port)
 
-        yield sock if block_given?
+        yield socket if block_given?
 
-        sock.close
+        socket.close
         return nil
       end
 
+      #
+      # Connects to a specified host and port, sends the given data and then
+      # closes the connection.
+      #
+      # @param [String] data
+      #   The data to send through the connection.
+      #
+      # @param [String] host
+      #   The host to connect to.
+      #
+      # @param [Integer] port
+      #   The port to connect to.
+      #
+      # @param [String] local_host (nil)
+      #   The local host to bind to.
+      #
+      # @param [Integer] local_port (nil)
+      #   The local port to bind to.
+      #
+      # @return [true]
+      #   The data was successfully sent.
+      #
+      # @example
+      #   buffer = "GET /" + ('A' * 4096) + "\n\r"
+      #   udp_send(buffer,'victim.com',80)
+      #   # => true
+      #
+      # @api public
+      #
+      # @since 0.4.0
+      #
+      def udp_send(data,host,port,local_host=nil,local_port=nil)
+        udp_session(host,port,local_host,local_port) do |socket|
+          socket.write(data)
+        end
+
+        return true
+      end
+
       #
       # Reads the banner from the service running on the given host and port.
       #
@@ -177,8 +214,8 @@ module Ronin
       def udp_banner(host,port,local_host=nil,local_port=nil)
         banner = nil
 
-        udp_session(host,port,local_host,local_port) do |sock|
-          banner = sock.readline
+        udp_session(host,port,local_host,local_port) do |socket|
+          banner = socket.readline
         end
 
         yield banner if block_given?
@@ -202,8 +239,8 @@ module Ronin
       #
       # @api public
       #
-      def udp_server(port,host='0.0.0.0')
-        host   = host.to_s
+      def udp_server(port=nil,host=nil)
+        host   = (host || '0.0.0.0').to_s
         server = UDPSocket.new
 
         server.bind(host,port)
@@ -237,7 +274,7 @@ module Ronin
       #
       # @api public
       #
-      def udp_server_session(port,host='0.0.0.0',&block)
+      def udp_server_session(port=nil,host=nil,&block)
         server = udp_server(port,host,&block)
 
         server.close()

data/lib/ronin/path.rb

@@ -1,5 +1,5 @@
 #
-# Copyright (c) 2006-2011 Hal Brodigan (postmodern.mod3 at gmail.com)
+# Copyright (c) 2006-2012 Hal Brodigan (postmodern.mod3 at gmail.com)
 #
 # This file is part of Ronin Support.
 #

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

@@ -1,5 +1,5 @@
 #
-# Copyright (c) 2006-2011 Hal Brodigan (postmodern.mod3 at gmail.com)
+# Copyright (c) 2006-2012 Hal Brodigan (postmodern.mod3 at gmail.com)
 #
 # This file is part of Ronin Support.
 #

data/lib/ronin/support.rb

@@ -1,5 +1,5 @@
 #
-# Copyright (c) 2006-2011 Hal Brodigan (postmodern.mod3 at gmail.com)
+# Copyright (c) 2006-2012 Hal Brodigan (postmodern.mod3 at gmail.com)
 #
 # This file is part of Ronin Support.
 #

data/lib/ronin/support/inflector.rb

@@ -1,5 +1,5 @@
 #
-# Copyright (c) 2006-2011 Hal Brodigan (postmodern.mod3 at gmail.com)
+# Copyright (c) 2006-2012 Hal Brodigan (postmodern.mod3 at gmail.com)
 #
 # This file is part of Ronin Support.
 #

data/lib/ronin/support/support.rb

@@ -1,5 +1,5 @@
 #
-# Copyright (c) 2006-2011 Hal Brodigan (postmodern.mod3 at gmail.com)
+# Copyright (c) 2006-2012 Hal Brodigan (postmodern.mod3 at gmail.com)
 #
 # This file is part of Ronin Support.
 #

data/lib/ronin/support/version.rb

@@ -1,5 +1,5 @@
 #
-# Copyright (c) 2006-2011 Hal Brodigan (postmodern.mod3 at gmail.com)
+# Copyright (c) 2006-2012 Hal Brodigan (postmodern.mod3 at gmail.com)
 #
 # This file is part of Ronin Support.
 #
@@ -20,6 +20,6 @@
 module Ronin
   module Support
     # ronin-support version
-    VERSION = '0.4.0.rc1'
+    VERSION = '0.4.0.rc2'
   end
 end

data/lib/ronin/templates.rb

@@ -1,5 +1,5 @@
 #
-# Copyright (c) 2006-2011 Hal Brodigan (postmodern.mod3 at gmail.com)
+# Copyright (c) 2006-2012 Hal Brodigan (postmodern.mod3 at gmail.com)
 #
 # This file is part of Ronin Support.
 #

data/lib/ronin/templates/erb.rb

@@ -1,5 +1,5 @@
 #
-# Copyright (c) 2006-2011 Hal Brodigan (postmodern.mod3 at gmail.com)
+# Copyright (c) 2006-2012 Hal Brodigan (postmodern.mod3 at gmail.com)
 #
 # This file is part of Ronin Support.
 #

data/lib/ronin/templates/template.rb

@@ -1,5 +1,5 @@
 #
-# Copyright (c) 2006-2011 Hal Brodigan (postmodern.mod3 at gmail.com)
+# Copyright (c) 2006-2012 Hal Brodigan (postmodern.mod3 at gmail.com)
 #
 # This file is part of Ronin Support.
 #

data/lib/ronin/ui/output.rb

@@ -1,5 +1,5 @@
 #
-# Copyright (c) 2006-2011 Hal Brodigan (postmodern.mod3 at gmail.com)
+# Copyright (c) 2006-2012 Hal Brodigan (postmodern.mod3 at gmail.com)
 #
 # This file is part of Ronin Support.
 #

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

@@ -1,5 +1,5 @@
 #
-# Copyright (c) 2006-2011 Hal Brodigan (postmodern.mod3 at gmail.com)
+# Copyright (c) 2006-2012 Hal Brodigan (postmodern.mod3 at gmail.com)
 #
 # This file is part of Ronin Support.
 #

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

@@ -1,5 +1,5 @@
 #
-# Copyright (c) 2006-2011 Hal Brodigan (postmodern.mod3 at gmail.com)
+# Copyright (c) 2006-2012 Hal Brodigan (postmodern.mod3 at gmail.com)
 #
 # This file is part of Ronin Support.
 #

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

@@ -1,5 +1,5 @@
 #
-# Copyright (c) 2006-2011 Hal Brodigan (postmodern.mod3 at gmail.com)
+# Copyright (c) 2006-2012 Hal Brodigan (postmodern.mod3 at gmail.com)
 #
 # This file is part of Ronin Support.
 #

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

@@ -1,5 +1,5 @@
 #
-# Copyright (c) 2006-2011 Hal Brodigan (postmodern.mod3 at gmail.com)
+# Copyright (c) 2006-2012 Hal Brodigan (postmodern.mod3 at gmail.com)
 #
 # This file is part of Ronin Support.
 #

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

@@ -1,5 +1,5 @@
 #
-# Copyright (c) 2006-2011 Hal Brodigan (postmodern.mod3 at gmail.com)
+# Copyright (c) 2006-2012 Hal Brodigan (postmodern.mod3 at gmail.com)
 #
 # This file is part of Ronin Support.
 #

data/lib/ronin/ui/shell.rb

@@ -1,5 +1,5 @@
 #
-# Copyright (c) 2006-2011 Hal Brodigan (postmodern.mod3 at gmail.com)
+# Copyright (c) 2006-2012 Hal Brodigan (postmodern.mod3 at gmail.com)
 #
 # This file is part of Ronin Support.
 #
@@ -26,6 +26,68 @@ module Ronin
     #
     # Spawns a ReadLine powered interactive Shell.
     #
+    # ## Simple Shell
+    #
+    #     require 'ronin/ui/shell'
+    #     require 'ronin/network/tcp'
+    #
+    #     include Ronin::Network::TCP
+    #
+    #     tcp_session('victim.com',1337) do |socket|
+    #       UI::Shell.new(:name => 'bind_shell') do |shell,line|
+    #         socket.puts "#{line}; echo 'EOC'"
+    #
+    #         socket.each_line do |output|
+    #           puts output
+    #
+    #           break if output.chomp == 'EOC'
+    #         end
+    #       end
+    #     end
+    #
+    # ## Shell with Commands
+    #
+    #     require 'ronin/ui/shell'
+    #     require 'ronin/network/http'
+    #
+    #     class HTTPShell < Ronin::UI::Shell
+    #
+    #       include Ronin::Network::HTTP
+    #
+    #       def initialize(host)
+    #         super(:name => host)
+    #
+    #         @host = host
+    #       end
+    #
+    #       protected
+    #
+    #       def get(path)
+    #         print_response http_get(:host => @host, :path => path)
+    #       end
+    #
+    #       def post(path,*params)
+    #         print_response http_post(
+    #           :host      => @host,
+    #           :path      => path,
+    #           :post_data => Hash[params.map { |param| param.split('=') }]
+    #         )
+    #       end
+    #
+    #       private
+    #
+    #       def print_response(response)
+    #         response.canonical_each do |name,value|
+    #           puts "#{name}: #{value}"
+    #         end
+    #
+    #         puts
+    #
+    #         puts response.body
+    #       end
+    #
+    #     end
+    #
     # @api semipublic
     #
     class Shell
@@ -70,15 +132,15 @@ module Ronin
       # @since 0.3.0
       #
       def initialize(options={},&block)
-        @name     = options.fetch(:name,'')
+        @name     = options[:name]
         @prompt   = options.fetch(:prompt,DEFAULT_PROMPT)
 
-        @commands = Set[:help, :exit]
+        @commands = Set['help', 'exit']
 
         self.class.ancestors.each do |subclass|
           if subclass < Shell
             subclass.protected_instance_methods(false).each do |name|
-              @commands << name.to_sym
+              @commands << name.to_s
             end
           end
         end
@@ -163,10 +225,8 @@ module Ronin
           # ignore empty lines
           return false unless command
 
-          command = command.to_sym
-
           # no explicitly calling handler
-          return false if command == :handler
+          return false if command == 'handler'
 
           unless @commands.include?(command)
             print_error "Invalid command: #{command}"

data/lib/ronin/wordlist.rb

@@ -1,5 +1,5 @@
 #
-# Copyright (c) 2006-2011 Hal Brodigan (postmodern.mod3 at gmail.com)
+# Copyright (c) 2006-2012 Hal Brodigan (postmodern.mod3 at gmail.com)
 #
 # This file is part of Ronin Support.
 #
@@ -19,6 +19,8 @@
 
 require 'ronin/fuzzing/extensions'
 
+require 'set'
+
 module Ronin
   #
   # An Enumerable class for iterating over wordlist files or lists of words.
@@ -71,6 +73,54 @@ module Ronin
       yield self if block_given?
     end
 
+    #
+    # Parses the text into a unique Set of words.
+    #
+    # @param [#each_line] text
+    #   The text to parse.
+    #
+    # @yield [word]
+    #   If a block is given, it will be passed every unique word,
+    #   the first time it is seen.
+    #
+    # @yieldparam [String] word
+    #   A unique word from the text.
+    #
+    # @return [SortedSet]
+    #   The unique set of words from the text.
+    #
+    def self.parse(text)
+      words_seen = SortedSet[]
+
+      text.each_line do |line|
+        line.split.each do |word|
+          if block_given?
+            yield word unless words_seen.include?(word)
+          end
+
+          words_seen << word
+        end
+      end
+
+      return words_seen
+    end
+
+    #
+    # Builds a new wordlist from the text.
+    #
+    # @param [#each_line] text
+    #   The text to parse.
+    #
+    # @param [Hash{Regexp,String,Symbol => Symbol,#each}] mutations
+    #   Additional mutations for the wordlist.
+    #
+    # @return [Wordlist]
+    #   The newly build wordlist.
+    #
+    def self.build(text,mutations={})
+      new(parse(text),mutations)
+    end
+
     #
     # Iterates over each word in the list.
     #

data/spec/extensions/ip_addr_spec.rb

@@ -159,7 +159,7 @@ describe IPAddr do
   let(:ip) { IPAddr.new('127.0.0.1') }
   let(:bad_ip) { IPAddr.new('0.0.0.0') }
 
-  describe "#lookup" do
+  describe "#lookup", :network do
     let(:nameserver) { '4.2.2.1' }
 
     it "should lookup the host-name for an IP" do