ronin-support 0.1.0.pre1

data/lib/ronin/formatting/extensions/text/string.rb

@@ -0,0 +1,206 @@
+#
+# Ronin - A Ruby platform for exploit development and security research.
+#
+# Copyright (c) 2006-2010 Hal Brodigan (postmodern.mod3 at gmail.com)
+#
+# This library is free software; you can redistribute it and/or
+# modify it under the terms of the GNU Lesser General Public
+# License as published by the Free Software Foundation; either
+# version 2.1 of the License, or (at your option) any later version.
+#
+# This library is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+# Lesser General Public License for more details.
+#
+# You should have received a copy of the GNU Lesser General Public
+# License along with this library; if not, write to the Free Software
+# Foundation, Inc., 51 Franklin Street, Fifth Floor,
+# Boston, MA  02110-1301  USA
+#
+
+require 'chars'
+
+class String
+
+  #
+  # Creates a new String by formatting each byte.
+  #
+  # @param [Hash] options
+  #   Additional options.
+  #
+  # @option options [Array, Range] :include (0x00..0xff)
+  #   The bytes to format.
+  #
+  # @option options [Array, Range] :exclude
+  #   The bytes not to format.
+  #
+  # @yield [byte]
+  #   The block which will return the formatted version of each byte
+  #   within the String.
+  #
+  # @yieldparam [Integer] byte
+  #   The byte to format.
+  #
+  # @return [String]
+  #   The formatted version of the String.
+  #
+  def format_bytes(options={})
+    included = (options[:include] || (0x00..0xff))
+    excluded = (options[:exclude] || [])
+
+    formatted = ''
+
+    self.each_byte do |b|
+      c = b.chr
+
+      if ((included.include?(b) || included.include?(c)) \
+          && !(excluded.include?(b) || excluded.include?(c)))
+        formatted << yield(b)
+      else
+        formatted << b
+      end
+    end
+
+    return formatted
+  end
+
+  #
+  # Creates a new String by formatting each character.
+  #
+  # @param [Hash] options
+  #   Additional options.
+  #
+  # @option options [Array, Range] :include (0x00..0xff)
+  #   The bytes to format.
+  #
+  # @option options [Array, Range] :exclude
+  #   The bytes not to format.
+  #
+  # @yield [char]
+  #   The block which will return the formatted version of each character
+  #   within the String.
+  #
+  # @yieldparam [String] char
+  #   The character to format.
+  #
+  # @return [String]
+  #   The formatted version of the String.
+  #
+  def format_chars(options={})
+    format_bytes(options) { |b| yield b.chr }
+  end
+
+  #
+  # Creates a new String by randomizing the case of each character in the
+  # String.
+  #
+  # @param [Hash] options
+  #   Additional options.
+  #
+  # @option options [Array, Range] :include (0x00..0xff)
+  #   The bytes to format.
+  #
+  # @option options [Array, Range] :exclude
+  #   The bytes not to format.
+  #
+  # @option options [Float] :probability (0.5)
+  #   The probability that a character will have it's case changed.
+  #
+  # @example
+  #   "get out your checkbook".random_case
+  #   # => "gEt Out YOur CHEckbook"
+  #
+  def random_case(options={})
+    prob = (options[:probability] || 0.5)
+
+    format_chars(options) do |c|
+      if rand <= prob
+        c.swapcase 
+      else
+        c
+      end
+    end
+  end
+
+  #
+  # Inserts data before the occurrence of a pattern.
+  #
+  # @param [String, Regexp] pattern
+  #   The pattern to search for.
+  #
+  # @param [String] data
+  #   The data to insert before the pattern.
+  #
+  # @return [String]
+  #   The new modified String.
+  #
+  def insert_before(pattern,data)
+    string = self.dup
+    index = string.index(pattern)
+
+    string.insert(index,data) if index
+    return string
+  end
+
+  #
+  # Inserts data after the occurrence of a pattern.
+  #
+  # @param [String, Regexp] pattern
+  #   The pattern to search for.
+  #
+  # @param [String] data
+  #   The data to insert after the pattern.
+  #
+  # @return [String]
+  #   The new modified String.
+  #
+  def insert_after(pattern,data)
+    string = self.dup
+    match = string.match(pattern)
+
+    if match
+      index = match.end(match.length - 1)
+
+      string.insert(index,data)
+    end
+
+    return string
+  end
+
+  #
+  # Creates a new String by padding the String with repeating text,
+  # out to a specified length.
+  #
+  # @param [String] padding
+  #   The text to pad the new String with.
+  #
+  # @param [String] max_length
+  #   The maximum length to pad the new String out to.
+  #
+  # @return [String]
+  #   The padded version of the String.
+  #
+  # @example
+  #   "hello".pad('A',50)
+  #   # => "helloAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA"
+  #
+  def pad(padding,max_length=self.length)
+    padding = padding.to_s
+
+    if max_length >= self.length
+      max_length -= self.length
+    else
+      max_length = 0
+    end
+
+    padded = self + (padding * (max_length / padding.length))
+
+    unless (remaining = max_length % padding.length) == 0
+      padded << padding[0...remaining]
+    end
+
+    return padded
+  end
+
+end

data/lib/ronin/formatting/http.rb

@@ -0,0 +1,22 @@
+#
+# Ronin - A Ruby platform for exploit development and security research.
+#
+# Copyright (c) 2006-2010 Hal Brodigan (postmodern.mod3 at gmail.com)
+#
+# This library is free software; you can redistribute it and/or
+# modify it under the terms of the GNU Lesser General Public
+# License as published by the Free Software Foundation; either
+# version 2.1 of the License, or (at your option) any later version.
+#
+# This library is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+# Lesser General Public License for more details.
+#
+# You should have received a copy of the GNU Lesser General Public
+# License along with this library; if not, write to the Free Software
+# Foundation, Inc., 51 Franklin Street, Fifth Floor,
+# Boston, MA  02110-1301  USA
+#
+
+require 'ronin/formatting/extensions/http'

data/lib/ronin/formatting/text.rb

@@ -0,0 +1,22 @@
+#
+# Ronin - A Ruby platform for exploit development and security research.
+#
+# Copyright (c) 2006-2010 Hal Brodigan (postmodern.mod3 at gmail.com)
+#
+# This library is free software; you can redistribute it and/or
+# modify it under the terms of the GNU Lesser General Public
+# License as published by the Free Software Foundation; either
+# version 2.1 of the License, or (at your option) any later version.
+#
+# This library is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+# Lesser General Public License for more details.
+#
+# You should have received a copy of the GNU Lesser General Public
+# License along with this library; if not, write to the Free Software
+# Foundation, Inc., 51 Franklin Street, Fifth Floor,
+# Boston, MA  02110-1301  USA
+#
+
+require 'ronin/formatting/extensions/text'

data/lib/ronin/network.rb

@@ -0,0 +1,29 @@
+#
+# Ronin - A Ruby platform for exploit development and security research.
+#
+# Copyright (c) 2006-2010 Hal Brodigan (postmodern.mod3 at gmail.com)
+#
+# This library is free software; you can redistribute it and/or
+# modify it under the terms of the GNU Lesser General Public
+# License as published by the Free Software Foundation; either
+# version 2.1 of the License, or (at your option) any later version.
+#
+# This library is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+# Lesser General Public License for more details.
+#
+# You should have received a copy of the GNU Lesser General Public
+# License along with this library; if not, write to the Free Software
+# Foundation, Inc., 51 Franklin Street, Fifth Floor,
+# Boston, MA  02110-1301  USA
+#
+
+require 'ronin/network/tcp'
+require 'ronin/network/udp'
+require 'ronin/network/smtp'
+require 'ronin/network/esmtp'
+require 'ronin/network/pop3'
+require 'ronin/network/imap'
+require 'ronin/network/telnet'
+require 'ronin/network/http'

data/lib/ronin/network/esmtp.rb

@@ -0,0 +1,22 @@
+#
+# Ronin - A Ruby platform for exploit development and security research.
+#
+# Copyright (c) 2006-2010 Hal Brodigan (postmodern.mod3 at gmail.com)
+#
+# This library is free software; you can redistribute it and/or
+# modify it under the terms of the GNU Lesser General Public
+# License as published by the Free Software Foundation; either
+# version 2.1 of the License, or (at your option) any later version.
+#
+# This library is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+# Lesser General Public License for more details.
+#
+# You should have received a copy of the GNU Lesser General Public
+# License along with this library; if not, write to the Free Software
+# Foundation, Inc., 51 Franklin Street, Fifth Floor,
+# Boston, MA  02110-1301  USA
+#
+
+require 'ronin/network/extensions/esmtp'

data/lib/ronin/network/extensions.rb

@@ -0,0 +1,29 @@
+#
+# Ronin - A Ruby platform for exploit development and security research.
+#
+# Copyright (c) 2006-2010 Hal Brodigan (postmodern.mod3 at gmail.com)
+#
+# This library is free software; you can redistribute it and/or
+# modify it under the terms of the GNU Lesser General Public
+# License as published by the Free Software Foundation; either
+# version 2.1 of the License, or (at your option) any later version.
+#
+# This library is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+# Lesser General Public License for more details.
+#
+# You should have received a copy of the GNU Lesser General Public
+# License along with this library; if not, write to the Free Software
+# Foundation, Inc., 51 Franklin Street, Fifth Floor,
+# Boston, MA  02110-1301  USA
+#
+
+require 'ronin/network/extensions/tcp'
+require 'ronin/network/extensions/udp'
+require 'ronin/network/extensions/smtp'
+require 'ronin/network/extensions/esmtp'
+require 'ronin/network/extensions/pop3'
+require 'ronin/network/extensions/imap'
+require 'ronin/network/extensions/telnet'
+require 'ronin/network/extensions/http'

data/lib/ronin/network/extensions/esmtp.rb

@@ -0,0 +1,22 @@
+#
+# Ronin - A Ruby platform for exploit development and security research.
+#
+# Copyright (c) 2006-2010 Hal Brodigan (postmodern.mod3 at gmail.com)
+#
+# This library is free software; you can redistribute it and/or
+# modify it under the terms of the GNU Lesser General Public
+# License as published by the Free Software Foundation; either
+# version 2.1 of the License, or (at your option) any later version.
+#
+# This library is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+# Lesser General Public License for more details.
+#
+# You should have received a copy of the GNU Lesser General Public
+# License along with this library; if not, write to the Free Software
+# Foundation, Inc., 51 Franklin Street, Fifth Floor,
+# Boston, MA  02110-1301  USA
+#
+
+require 'ronin/network/extensions/esmtp/net'

data/lib/ronin/network/extensions/esmtp/net.rb

@@ -0,0 +1,100 @@
+#
+# Ronin - A Ruby platform for exploit development and security research.
+#
+# Copyright (c) 2006-2010 Hal Brodigan (postmodern.mod3 at gmail.com)
+#
+# This library is free software; you can redistribute it and/or
+# modify it under the terms of the GNU Lesser General Public
+# License as published by the Free Software Foundation; either
+# version 2.1 of the License, or (at your option) any later version.
+#
+# This library is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+# Lesser General Public License for more details.
+#
+# You should have received a copy of the GNU Lesser General Public
+# License along with this library; if not, write to the Free Software
+# Foundation, Inc., 51 Franklin Street, Fifth Floor,
+# Boston, MA  02110-1301  USA
+#
+
+require 'ronin/network/extensions/smtp'
+
+module Net
+  #
+  # @see Ronin::Network::SMTP.message
+  #
+  def Net.esmtp_message(options={},&block)
+    Net.smtp_message(options,&block)
+  end
+
+  #
+  # Creates a connection to the ESMTP server.
+  #
+  # @param [String] host
+  #   The host to connect to.
+  #
+  # @param [Hash] options
+  #   Additional options.
+  #
+  # @option options [Integer] :port (Ronin::Network::SMTP.default_port)
+  #   The port to connect to.
+  #
+  # @option options [String] :helo
+  #   The HELO domain.
+  #
+  # @option options [Symbol] :auth
+  #   The type of authentication to use. Can be either `:login`, `:plain`,
+  #   or `:cram_md5`.
+  #
+  # @option options [String] :user
+  #   The user-name to authenticate with.
+  #
+  # @option options [String] :password
+  #   The password to authenticate with.
+  #
+  # @yield [session]
+  #   If a block is given, it will be passed an ESMTP enabled session
+  #   object.
+  #
+  # @yieldparam [Net::SMTP] session
+  #   The ESMTP session.
+  #
+  # @return [Net::SMTP]
+  #   The ESMTP enabled session.
+  #
+  def Net.esmtp_connect(host,options={})
+    session = Net.smtp_connect(host,options)
+    session.esmtp = true
+
+    yield session if block_given?
+    return session
+  end
+
+  #
+  # Starts an ESMTP session with the ESMTP enabled server.
+  #
+  # @param [String] host
+  #   The host to connect to.
+  #
+  # @param [Hash] options
+  #   Additional options.
+  #
+  # @yield [session]
+  #   If a block is given, it will be passed an ESMTP enabled session
+  #   object. After the block has returned, the session will be closed.
+  #
+  # @yieldparam [Net::SMTP] session
+  #   The ESMTP session.
+  #
+  # @see Net.esmtp_connect
+  #
+  def Net.esmtp_session(host,options={})
+    Net.smtp_session(host,options) do |session|
+      session.esmtp = true
+
+      yield session if block_given?
+    end
+  end
+end