ronin 0.0.9

data/lib/ronin/network/pop3.rb

@@ -0,0 +1,47 @@
+#
+#--
+# Ronin - A Ruby platform designed for information security and data
+# exploration tasks.
+#
+# Copyright (c) 2006-2008 Hal Brodigan (postmodern.mod3 at gmail.com)
+#
+# This program is free software; you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation; either version 2 of the License, or
+# (at your option) any later version.
+#
+# This program 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 General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program; if not, write to the Free Software
+# Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
+#++
+#
+
+require 'ronin/network/extensions/pop3'
+
+module Ronin
+  module Network
+    module POP3
+      # Default pop3 port
+      DEFAULT_PORT = 110
+
+      #
+      # Returns the default Ronin POP3 port.
+      #
+      def POP3.default_port
+        @@pop3_default_port ||= DEFAULT_PORT
+      end
+
+      #
+      # Sets the default Ronin POP3 port to the specified _port_.
+      #
+      def POP3.default_port=(port)
+        @@pop3_default_port = port
+      end
+    end
+  end
+end

data/lib/ronin/network/smtp.rb

@@ -0,0 +1,26 @@
+#
+#--
+# Ronin - A Ruby platform designed for information security and data
+# exploration tasks.
+#
+# Copyright (c) 2006-2008 Hal Brodigan (postmodern.mod3 at gmail.com)
+#
+# This program is free software; you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation; either version 2 of the License, or
+# (at your option) any later version.
+#
+# This program 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 General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program; if not, write to the Free Software
+# Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
+#++
+#
+
+require 'ronin/network/smtp/email'
+require 'ronin/network/smtp/smtp'
+require 'ronin/network/extensions/smtp'

data/lib/ronin/network/smtp/email.rb

@@ -0,0 +1,126 @@
+#
+#--
+# Ronin - A Ruby platform designed for information security and data
+# exploration tasks.
+#
+# Copyright (c) 2006-2008 Hal Brodigan (postmodern.mod3 at gmail.com)
+#
+# This program is free software; you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation; either version 2 of the License, or
+# (at your option) any later version.
+#
+# This program 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 General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program; if not, write to the Free Software
+# Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
+#++
+#
+
+module Ronin
+  module Network
+    module SMTP
+      class Email
+
+        # Sender of the email
+        attr_accessor :from
+
+        # Recipient of the email
+        attr_accessor :to
+
+        # Subject of the email
+        attr_accessor :subject
+
+        # Date of the email
+        attr_accessor :date
+
+        # Unique message-id string
+        attr_accessor :message_id
+
+        # Body of the email
+        attr_accessor :body
+
+        #
+        # Creates a new Email object with the given _options_. If a _block_
+        # is given, it will be passed the newly created Email object.
+        #
+        # _options_ must contain the following keys:
+        # <tt>:from</tt>:: The address the email is from.
+        # <tt>:to</tt>:: The address that the email should be sent to.
+        # <tt>:subject</tt>:: The subject of the email.
+        # <tt>:message_id</tt>:: 
+        #
+        # _options_ may contain the following keys:
+        # <tt>:date</tt>:: The date the email was sent on. Defaults to
+        #                  <tt>Time.now</tt>.
+        # <tt>:body</tt>:: The body of the email. Can be a String or an
+        #                  Array of Strings.
+        #
+        def initialize(options={},&block)
+          @from = options[:from]
+          @to = options[:to]
+          @subject = options[:subject]
+          @date = options[:date] || Time.now
+          @message_id = options[:message_id]
+          @body = []
+
+          if options[:body].kind_of?(Array)
+            @body += options[:body]
+          else
+            @body << options[:body]
+          end
+
+          block.call(self) if block
+        end
+
+        #
+        # Formats the email into a SMTP message as described in
+        # http://www.ruby-doc.org/stdlib/libdoc/net/smtp/rdoc/classes/Net/SMTP.html
+        #
+        def to_s
+          address = lambda { |info|
+            if info.kind_of?(Array)
+              return "#{info[0]} <#{info[1]}>"
+            elsif info.kind_of?(Hash)
+              return "#{info[:name]} <#{info[:email]}>"
+            else
+              return info
+            end
+          }
+
+          message = []
+
+          if @from
+            message << "From: #{address.call(@from)}"
+          end
+
+          if @to
+            message << "To: #{address.call(@to)}"
+          end
+
+          if @subject
+            message << "Subject: #{@subject}"
+          end
+
+          if @date
+            message << "Date: #{@date}"
+          end
+
+          if @message_id
+            message << "Message-Id: <#{@message_id}>"
+          end
+
+          message << ''
+          message += @body
+
+          return message.join("\n")
+        end
+
+      end
+    end
+  end
+end

data/lib/ronin/network/smtp/smtp.rb

@@ -0,0 +1,55 @@
+#
+#--
+# Ronin - A Ruby platform designed for information security and data
+# exploration tasks.
+#
+# Copyright (c) 2006-2008 Hal Brodigan (postmodern.mod3 at gmail.com)
+#
+# This program is free software; you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation; either version 2 of the License, or
+# (at your option) any later version.
+#
+# This program 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 General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program; if not, write to the Free Software
+# Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
+#++
+#
+
+require 'ronin/network/smtp/email'
+
+module Ronin
+  module Network
+    module SMTP
+      # Default smtp port
+      DEFAULT_PORT = 25
+
+      #
+      # Returns the default Ronin SMTP port.
+      #
+      def SMTP.default_port
+        @@smtp_default_port ||= DEFAULT_PORT
+      end
+
+      #
+      # Sets the default Ronin SMTP port.
+      #
+      def SMTP.default_port=(port)
+        @@smtp_default_port = port
+      end
+
+      #
+      # Returns the String form of an Email object with the given _options_
+      # and _block_.
+      #
+      def SMTP.message(options={},&block)
+        Email.new(options,&block).to_s
+      end
+    end
+  end
+end

data/lib/ronin/network/tcp.rb

@@ -0,0 +1,24 @@
+#
+#--
+# Ronin - A Ruby platform designed for information security and data
+# exploration tasks.
+#
+# Copyright (c) 2006-2008 Hal Brodigan (postmodern.mod3 at gmail.com)
+#
+# This program is free software; you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation; either version 2 of the License, or
+# (at your option) any later version.
+#
+# This program 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 General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program; if not, write to the Free Software
+# Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
+#++
+#
+
+require 'ronin/network/extensions/tcp'

data/lib/ronin/network/telnet.rb

@@ -0,0 +1,95 @@
+#
+#--
+# Ronin - A Ruby platform designed for information security and data
+# exploration tasks.
+#
+# Copyright (c) 2006-2008 Hal Brodigan (postmodern.mod3 at gmail.com)
+#
+# This program is free software; you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation; either version 2 of the License, or
+# (at your option) any later version.
+#
+# This program 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 General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program; if not, write to the Free Software
+# Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
+#++
+#
+
+require 'ronin/network/extensions/telnet'
+
+module Ronin
+  module Network
+    module Telnet
+      # Default telnet port
+      DEFAULT_PORT = 23
+
+      # The default prompt regular expression
+      DEFAULT_PROMPT = /[$%#>] \z/n
+
+      # The default timeout
+      DEFAULT_TIMEOUT = 10
+
+      #
+      # Returns the default Ronin Telnet port.
+      #
+      def Telnet.default_port
+        @@telnet_default_port ||= DEFAULT_PORT
+      end
+
+      #
+      # Sets the default Ronin Telnet port to the specified _port_.
+      #
+      def Telnet.default_port=(port)
+        @@telnet_default_port = port
+      end
+
+      #
+      # Returns the default Ronin Telnet prompt.
+      #
+      def Telnet.default_prompt
+        @@telnet_default_prompt ||= DEFAULT_PROMPT
+      end
+
+      #
+      # Sets the default Ronin Telnet prompt to the specified _prompt_.
+      #
+      def Telnet.default_prompt=(prompt)
+        @@telnet_default_prompt = prompt
+      end
+
+      #
+      # Returns the default Ronin Telnet timeout.
+      #
+      def Telnet.default_timeout
+        @@telnet_default_timeout ||= DEFAULT_TIMEOUT
+      end
+
+      #
+      # Sets the default Ronin Telnet timeout to the specified _timeout_.
+      #
+      def Telnet.default_timeout=(timeout)
+        @@telnet_default_timeout = timeout
+      end
+
+      #
+      # Returns the Ronin Telnet proxy.
+      #
+      def Telnet.proxy
+        @@telnet_proxy ||= nil
+      end
+
+      #
+      # Sets the Ronin Telnet proxy to _new_proxy_.
+      #
+      def Telnet.proxy=(new_proxy)
+        @@telnet_proxy = new_proxy
+      end
+    end
+  end
+end

data/lib/ronin/network/udp.rb

@@ -0,0 +1,24 @@
+#
+#--
+# Ronin - A Ruby platform designed for information security and data
+# exploration tasks.
+#
+# Copyright (c) 2006-2008 Hal Brodigan (postmodern.mod3 at gmail.com)
+#
+# This program is free software; you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation; either version 2 of the License, or
+# (at your option) any later version.
+#
+# This program 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 General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program; if not, write to the Free Software
+# Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
+#++
+#
+
+require 'ronin/network/extensions/udp'

data/lib/ronin/object_context.rb

@@ -0,0 +1,257 @@
+#
+#--
+# Ronin - A Ruby platform designed for information security and data
+# exploration tasks.
+#
+# Copyright (c) 2006-2008 Hal Brodigan (postmodern.mod3 at gmail.com)
+#
+# This program is free software; you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation; either version 2 of the License, or
+# (at your option) any later version.
+#
+# This program 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 General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program; if not, write to the Free Software
+# Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
+#++
+#
+
+require 'ronin/exceptions/unknown_object_context'
+require 'ronin/exceptions/object_context_not_found'
+require 'ronin/extensions/meta'
+require 'ronin/context'
+
+require 'dm-core'
+
+module Ronin
+  module ObjectContext
+    include DataMapper::Types
+
+    def self.included(base)
+      base.class_eval do
+        include Model
+        include Context
+
+        metaclass_def(:object_contextify) do |name|
+          ObjectContext.object_contexts[name] = self
+
+          contextify name
+
+          # The Path to the object context
+          property :object_path, String, :key => true
+
+          # The modification timestamp of the object context
+          property :object_timestamp, EpochTime
+
+          meta_def(:load_object) do |path,*args|
+            ObjectContext.load_object(context_name,path,*args)
+          end
+
+          meta_def(:cache) do |path,*args|
+            load_object(path,*args).cache
+          end
+
+          meta_def(:mirror) do |path,*args|
+            path = File.expand_path(path)
+            existing_obj = first(:object_path => path)
+
+            if existing_obj
+              return existing_obj.mirror
+            else
+              return cache(path,*args)
+            end
+          end
+
+          meta_def(:search) do |*attribs|
+            all(*attribs).map { |obj| obj.object }
+          end
+
+          class_def(:object) do
+            self.class.load_object(self.object_path)
+          end
+
+          class_def(:missing?) do
+            if self.object_path
+              return !(File.file?(self.object_path))
+            end
+
+            return false
+          end
+
+          class_def(:stale?) do
+            if self.object_timestamp
+              return File.mtime(self.object_path) > self.object_timestamp
+            end
+
+            return false
+          end
+
+          class_def(:cache) do
+            if self.object_path
+              self.object_timestamp = File.mtime(self.object_path)
+              return save
+            end
+
+            return false
+          end
+
+          class_def(:mirror) do
+            if self.object_path
+              unless File.file?(self.object_path)
+                return destroy
+              else
+                if (!(dirty?) && stale?)
+                  destroy
+                  return object.cache
+                end
+              end
+            end
+
+            return false
+          end
+
+          # define Repo-level object loader method
+          Ronin.module_eval %{
+            def ronin_load_#{name}(path,*args,&block)
+              new_obj = #{self}.load_object(path,*args)
+              
+              block.call(new_obj) if block
+              return new_obj
+            end
+          }
+        end
+      end
+    end
+
+    #
+    # Returns the Hash of all defined object-contexts.
+    #
+    def ObjectContext.object_contexts
+      @@ronin_object_contexts ||= {}
+    end
+
+    #
+    # Returns +true+ if there is an object-context defined with the
+    # specified _name_, returns +false+ otherwise.
+    #
+    def ObjectContext.is_object_context?(name)
+      ObjectContext.object_contexts.has_key?(name.to_sym)
+    end
+
+    #
+    # Loads the object context of the specified _name_ and from the
+    # specified _path_ with the given _args_. If no object contexts were
+    # defined with the specified _name_, an UnknownObjectContext
+    # exception will be raised.
+    #
+    #   ObjectContext.load_object(:note,'/path/to/my_notes.rb') # => Note
+    #
+    def ObjectContext.load_object(name,path,*args,&block)
+      name = name.to_sym
+
+      unless ObjectContext.is_object_context?(name)
+        raise(UnknownObjectContext,"unknown object context '#{name}'",caller)
+      end
+
+      path = File.expand_path(path)
+
+      unless File.file?(path)
+        raise(ObjectContextNotFound,"object context #{path.dump} does not exist",caller)
+      end
+
+      new_obj = Context.load_context(name,path,*args)
+      new_obj.object_path = path
+
+      block.call(new_obj) if block
+      return new_obj
+    end
+
+    #
+    # Loads all object contexts from the specified _path_ returning an
+    # +Array+ of loaded object contexts. If a _block_ is given, it will
+    # be passed each loaded object context.
+    #
+    #   Context.load_contexts('/path/to/misc_contexts.rb') # => [...]
+    #
+    def ObjectContext.load_objects(path,&block)
+      path = File.expand_path(path)
+
+      unless File.file?(path)
+        raise(ObjectContextNotFound,"object context #{path.dump} does not exist",caller)
+      end
+
+      return Context.load_contexts(path) do |new_obj|
+        new_obj.object_path = path
+
+        block.call(new_obj) if block
+      end
+    end
+
+    #
+    # Cache all objects loaded from the specified _path_.
+    #
+    def ObjectContext.cache_objects(path)
+      ObjectContext.load_objects(path).each do |obj|
+        obj.cache
+      end
+    end
+
+    #
+    # Cache all objects loaded from the paths within the specified
+    # _directory_.
+    #
+    def ObjectContext.cache_objects_in(directory)
+      directory = File.expand_path(directory)
+      paths = Dir[File.join(directory,'**','*.rb')]
+
+      paths.each do |path|
+        ObjectContext.load_objects(path).each do |obj|
+          obj.cache
+        end
+      end
+
+      return nil
+    end
+
+    #
+    # Mirror all objects that were previously cached from paths within
+    # the specified _directory_. Also cache objects which have yet to
+    # be cached.
+    #
+    def ObjectContext.mirror_objects_in(directory)
+      directory = File.expand_path(directory)
+      paths = Dir[File.join(directory,'**','*.rb')]
+
+      ObjectContext.object_contexts.each_value do |base|
+        objects = base.all(:object_path.like => "#{directory}%")
+        paths -= objects.map { |obj| obj.object_path }
+
+        objects.each { |obj| obj.mirror }
+      end
+
+      paths.each do |path|
+        ObjectContext.cache_objects(path)
+      end
+
+      return nil
+    end
+
+    #
+    # Deletes all cached objects that existed in the specified _directory_.
+    #
+    def ObjectContext.expunge_objects_from(directory)
+      directory = File.expand_path(directory)
+
+      ObjectContext.object_contexts.each_value do |base|
+        base.all(:object_path.like => "#{directory}%").destroy!
+      end
+
+      return nil
+    end
+  end
+end