ronin-support 0.1.0.pre1

data/lib/ronin/formatting/extensions/digest.rb

@@ -0,0 +1,23 @@
+#
+# 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/digest/string'
+require 'ronin/formatting/extensions/digest/file'

data/lib/ronin/formatting/extensions/digest/file.rb

@@ -0,0 +1,117 @@
+#
+# 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 'digest/md5'
+require 'digest/sha1'
+require 'digest/sha2'
+
+class File
+
+  #
+  # Calculates the MD5 checksum of a file.
+  #
+  # @param [String] path
+  #   The path to the file.
+  #
+  # @return [String]
+  #   The MD5 checksum of the file.
+  #
+  # @example
+  #   File.md5('data.txt')
+  #   # => "5d41402abc4b2a76b9719d911017c592"
+  #
+  def File.md5(path)
+    Digest::MD5.file(path).hexdigest
+  end
+
+  #
+  # Calculates the SHA1 checksum of a file.
+  #
+  # @param [String] path
+  #   The path to the file.
+  #
+  # @return [String]
+  #   The SHA1 checksum of the file.
+  #
+  # @example
+  #   File.sha1('data.txt')
+  #   # => "aaf4c61ddcc5e8a2dabede0f3b482cd9aea9434d"
+  #
+  def File.sha1(path)
+    Digest::SHA1.file(path).hexdigest
+  end
+
+  #
+  # @see File.sha1
+  #
+  def File.sha128(path)
+    File.sha1(path)
+  end
+
+  #
+  # Calculates the SHA256 checksum of a file.
+  #
+  # @param [String] path
+  #   The path to the file.
+  #
+  # @return [String]
+  #   The SHA256 checksum of the file.
+  #
+  # @example
+  #   File.sha256('data.txt')
+  #   # => "2cf24dba5fb0a30e26e83b2ac5b9e29e1b161e5c1fa7425e73043362938b9824"
+  #
+  def File.sha256(path)
+    Digest::SHA256.file(path).hexdigest
+  end
+
+  #
+  # @see File.sha256
+  #
+  def File.sha2(path)
+    File.sha256(path)
+  end
+
+  #
+  # Calculates the SHA512 checksum of a file.
+  #
+  # @param [String] path
+  #   The path to the file.
+  #
+  # @return [String]
+  #   The SHA512 checksum of the file.
+  #
+  # @example
+  #   File.sha512('data.txt')
+  #   # => "9b71d224bd62f3785d96d46ad3ea3d73319bfbc2890caadae2dff72519673ca72323c3d99ba5c11d7c7acc6e14b8c5da0c4663475c2e5c3adef46f73bcdec043"
+  #
+  def File.sha512(path)
+    Digest::SHA512.file(path).hexdigest
+  end
+
+  #
+  # @see File.sha512
+  #
+  def File.sha5(path)
+    File.sha512(path)
+  end
+
+end

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

@@ -0,0 +1,80 @@
+#
+# 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 'digest/md5'
+require 'digest/sha1'
+require 'digest/sha2'
+
+class String
+
+  #
+  # @return [String]
+  #   The MD5 checksum of the String.
+  #
+  # @example
+  #   "hello".md5
+  #   # => "5d41402abc4b2a76b9719d911017c592"
+  #
+  def md5
+    Digest::MD5.hexdigest(self)
+  end
+
+  #
+  # @return [String]
+  #   The SHA1 checksum of the String.
+  #
+  # @example
+  #   "hello".sha1
+  #   # => "aaf4c61ddcc5e8a2dabede0f3b482cd9aea9434d"
+  #
+  def sha1
+    Digest::SHA1.hexdigest(self)
+  end
+
+  alias sha128 sha1
+
+  #
+  # @return [String]
+  #   The SHA2 checksum of the String.
+  #
+  # @example
+  #   "hello".sha2
+  #   # => "2cf24dba5fb0a30e26e83b2ac5b9e29e1b161e5c1fa7425e73043362938b9824"
+  #
+  def sha256
+    Digest::SHA256.hexdigest(self)
+  end
+
+  alias sha2 sha256
+
+  #
+  # @return [String]
+  #   The SHA512 checksum of the String.
+  #
+  # @example
+  #   "hello".sha512
+  #   # => "9b71d224bd62f3785d96d46ad3ea3d73319bfbc2890caadae2dff72519673ca72323c3d99ba5c11d7c7acc6e14b8c5da0c4663475c2e5c3adef46f73bcdec043"
+  #
+  def sha512
+    Digest::SHA512.hexdigest(self)
+  end
+
+end

data/lib/ronin/formatting/extensions/http.rb

@@ -0,0 +1,23 @@
+#
+# 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/integer'
+require 'ronin/formatting/extensions/http/string'

data/lib/ronin/formatting/extensions/http/integer.rb

@@ -0,0 +1,57 @@
+#
+# 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 'uri'
+require 'cgi'
+
+class Integer
+
+  #
+  # URI encodes the byte.
+  #
+  # @return [String]
+  #   The URI encoded byte.
+  #
+  def uri_encode
+    URI.encode(self.chr)
+  end
+
+  #
+  # URI escapes the byte.
+  #
+  # @return [String]
+  #   The URI escaped byte.
+  #
+  def uri_escape
+    CGI.escape(self.chr)
+  end
+
+  #
+  # Formats the byte for HTTP.
+  #
+  # @return [String]
+  #   The formatted byte.
+  #
+  def format_http
+    "%%%x" % self
+  end
+
+end

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

@@ -0,0 +1,102 @@
+#
+# 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/integer'
+require 'ronin/formatting/extensions/text'
+
+require 'uri'
+require 'cgi'
+
+class String
+
+  #
+  # URI encodes the String.
+  #
+  # @return [String]
+  #   The URI encoded form of the String.
+  #
+  # @example
+  #   "art is graffiti".uri_encode
+  #   # => "art%20is%20graffiti"
+  #
+  def uri_encode
+    URI.encode(self)
+  end
+
+  #
+  # URI decodes the String.
+  #
+  # @return [String]
+  #   The decoded URI form of the String.
+  #
+  # @example
+  #   "genre%3f".uri_decode
+  #   # => "genre?"
+  #
+  def uri_decode
+    URI.decode(self)
+  end
+
+  #
+  # URI escapes the String.
+  #
+  # @return [String]
+  #   The URI escaped form of the String.
+  #
+  # @example
+  #   "x > y".uri_escape
+  #   # => "x+%3E+y"
+  #
+  def uri_escape
+    CGI.escape(self)
+  end
+
+  #
+  # URI unescapes the String.
+  #
+  # @return [String]
+  #   The unescaped URI form of the String.
+  #
+  # @example
+  #   "sweet+%26+sour".uri_unescape
+  #   # => "sweet & sour"
+  #
+  def uri_unescape
+    CGI.unescape(self)
+  end
+
+  #
+  # Formats the bytes of the String.
+  #
+  # @return [String]
+  #   The HTTP hexidecimal encoded form of the String.
+  #
+  # @example
+  #   "hello".format_http
+  #   # => "%68%65%6c%6c%6f"
+  #
+  # @see String#format_bytes
+  #
+  def format_http(options={})
+    format_bytes(options) { |b| b.format_http }
+  end
+
+end

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

@@ -0,0 +1,23 @@
+#
+# 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/string'
+require 'ronin/formatting/extensions/text/array'

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

@@ -0,0 +1,125 @@
+#
+# 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
+#
+
+class Array
+
+  #
+  # Decodes the bytes contained within the Array. The Array may contain
+  # both Integer and String objects.
+  #
+  # @return [Array]
+  #   The bytes contained within the Array.
+  #
+  # @example
+  #   [0x41, 0x41, 0x20].bytes
+  #   # => [0x41, 0x41, 0x20]
+  #
+  # @example
+  #   ['A', 'BB', 0x90].bytes
+  #   # => [0x41, 0x42, 0x42, 0x90]
+  #
+  def bytes
+    self.inject([]) do |accum,elem|
+      if elem.kind_of?(Integer)
+        accum << elem
+      else
+        elem.to_s.each_byte { |b| accum << b }
+      end
+
+      accum
+    end
+  end
+
+  #
+  # Decodes the characters contained within the Array. The Array may contain
+  # either Integer or String objects.
+  #
+  # @return [Array]
+  #   The characters generated from the array.
+  #
+  # @example
+  #   [0x41, 0x41, 0x20].chars
+  #   # => ["A", "A", " "]
+  #
+  def chars
+    array_bytes = self.bytes
+
+    array_bytes.map! { |b| b.chr }
+    return array_bytes
+  end
+
+  #
+  # @return [String]
+  #   The String created from the characters within the Array.
+  #
+  # @example
+  #   [0x41, 0x41, 0x20].char_string
+  #   # => "AA "
+  #
+  def char_string
+    chars.join
+  end
+
+  #
+  # Decodes the bytes contained with the Array, and escapes them as
+  # hexadecimal characters.
+  #
+  # @return [Array<String>]
+  #   The hexidecimal characters contained within the Array.
+  #
+  # @example
+  #   [0x41, 0x41, 0x20].bytes
+  #   # => ['\x41', '\x41', '\x20']
+  #
+  # @example
+  #   ['A', 'BB', 0x90].bytes
+  #   # => ['\x41', '\x42', '\x42', '\x90']
+  #
+  def hex_chars
+    array_bytes = self.bytes
+    
+    array_bytes.map! { |b| '\x%x' % b }
+    return array_bytes
+  end
+
+  #
+  # Decodes the bytes contained with the Array, and escapes them as
+  # hexadecimal integers.
+  #
+  # @return [Array<String>]
+  #   The hexidecimal integers contained within the Array.
+  #
+  # @example
+  #   [0x41, 0x41, 0x20].bytes
+  #   # => ['0x41', '0x41', '0x20']
+  #
+  # @example
+  #   ['A', 'BB', 0x90].bytes
+  #   # => ['0x41', '0x42', '0x42', '0x90']
+  #
+  def hex_integers
+    array_bytes = self.bytes
+    
+    array_bytes.map! { |b| '0x%x' % b }
+    return array_bytes
+  end
+
+end