ronin 0.3.0 → 1.0.0.pre1

data/lib/ronin/model/has_version/class_methods.rb

@@ -0,0 +1,47 @@
+#
+# Ronin - A Ruby platform for exploit development and security research.
+#
+# Copyright (c) 2006-2010 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 Model
+    module HasVersion
+      module ClassMethods
+        #
+        # Finds all models with a specific version.
+        #
+        # @param [String] version
+        #   The specific version to search for.
+        #
+        # @return [Array]
+        #   The models with the specific version.
+        #
+        def revision(version)
+          all(:version => version.to_s)
+        end
+
+        #
+        # Finds latest version of the model.
+        #
+        def latest
+          self.first(:order => [:version.desc])
+        end
+      end
+    end
+  end
+end

data/lib/ronin/model/has_version/has_version.rb

@@ -0,0 +1,41 @@
+#
+# Ronin - A Ruby platform for exploit development and security research.
+#
+# Copyright (c) 2006-2010 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/model/has_version/class_methods'
+require 'ronin/model/model'
+
+module Ronin
+  module Model
+    #
+    # Adds a `version` property to a model.
+    #
+    module HasVersion
+      def self.included(base)
+        base.send :include, Model
+        base.send :extend, ClassMethods
+
+        base.module_eval do
+          # The version of the model
+          property :version, String, :default => '0.1', :index => true
+        end
+      end
+    end
+  end
+end

data/lib/ronin/model/has_version.rb

@@ -1,7 +1,7 @@
 #
 # Ronin - A Ruby platform for exploit development and security research.
 #
-# Copyright (c) 2006-2009 Hal Brodigan (postmodern.mod3 at gmail.com)
+# Copyright (c) 2006-2010 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
@@ -18,39 +18,5 @@
 # Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
 #
 
-require 'ronin/model'
-
-module Ronin
-  module Model
-    module HasVersion
-      def self.included(base)
-        base.module_eval do
-          include Ronin::Model
-
-          # The version of the model
-          property :version, String, :default => '0.1', :index => true
-
-          #
-          # Finds all models with a specific version.
-          #
-          # @param [String] version
-          #   The specific version to search for.
-          #
-          # @return [Array]
-          #   The models with the specific version.
-          #
-          def self.revision(version)
-            self.all(:version => version.to_s)
-          end
-
-          #
-          # Finds latest version of the model.
-          #
-          def self.latest
-            self.first(:order => [:version.desc])
-          end
-        end
-      end
-    end
-  end
-end
+require 'ronin/model/has_version/class_methods'
+require 'ronin/model/has_version/has_version'

data/lib/ronin/model/model.rb

@@ -1,7 +1,7 @@
 #
 # Ronin - A Ruby platform for exploit development and security research.
 #
-# Copyright (c) 2006-2009 Hal Brodigan (postmodern.mod3 at gmail.com)
+# Copyright (c) 2006-2010 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
@@ -18,43 +18,35 @@
 # Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
 #
 
-require 'extlib'
+require 'ronin/model/class_methods'
+require 'ronin/model/types'
+require 'ronin/support/inflector'
+
 require 'dm-core'
 require 'dm-types'
+require 'dm-migrations'
 require 'dm-validations'
+require 'dm-aggregates'
+require 'dm-serializer'
 
 module Ronin
+  #
+  # Modifies a class so that it can represent a DataMapper model in the
+  # {Database}.
+  #
   module Model
-    include DataMapper::Types
-
-    # Name of Ronin's DataMapper repository
-    REPOSITORY_NAME = :ronin
+    include Model::Types
 
     def self.included(base)
       unless base.ancestors.include?(DataMapper::Resource)
-        base.module_eval do
-          include DataMapper::Resource
-          include DataMapper::Migrations
-
-          def self.allocate
-            resource = super
-            resource.instance_eval("initialize()")
-
-            return resource
-          end
-
-          #
-          # @return [Symbol]
-          #   The default repository name used for the model.
-          #
-          def self.default_repository_name
-            Model::REPOSITORY_NAME
-          end
-
-          # The class type property
-          property :type, Discriminator
-        end
+        base.send :include, DataMapper::Resource
       end
+
+      base.send :include, DataMapper,
+                          DataMapper::Migrations,
+                          DataMapper::Serialize,
+                          Model::Types
+      base.send :extend, Model::ClassMethods
     end
 
     #
@@ -80,7 +72,7 @@ module Ronin
     # @return [Hash{String => String}]
     #   A hash of the humanly readable names and values of the attributes.
     #
-    def humanize_attributes(options={},&block)
+    def humanize_attributes(options={})
       exclude = [:id, :type]
 
       if options[:exclude]
@@ -91,7 +83,7 @@ module Ronin
         if value.kind_of?(Array)
           value.map(&formatter).join(', ')
         elsif value.kind_of?(Symbol)
-          Extlib::Inflection.humanize(value)
+          Support::Inflector.humanize(value)
         else
           value.to_s
         end
@@ -106,10 +98,13 @@ module Ronin
           name = name.to_s
 
           unless name[-3..-1] == '_id'
-            name = Extlib::Inflection.humanize(name)
+            name = Support::Inflector.humanize(name)
             value = formatter.call(value)
 
-            block.call(name,value) if block
+            if block_given?
+              yield name, value
+            end
+
             formatted[name] = value
           end
         end

data/lib/ronin/model/types/description.rb

@@ -0,0 +1,60 @@
+#
+# Ronin - A Ruby platform for exploit development and security research.
+#
+# Copyright (c) 2006-2010 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 'dm-core'
+
+module Ronin
+  module Model
+    module Types
+      class Description < DataMapper::Property::Text
+
+        #
+        # Typecasts the description.
+        #
+        # @param [Object] value
+        #   The text of the description.
+        #
+        # @param [DataMapper::Property] property
+        #   The description property.
+        #
+        # @return [String, nil]
+        #   The typecasted description.
+        #
+        # @since 1.0.0
+        #
+        def typecast(value)
+          case value
+          when nil
+            nil
+          else
+            sanitized_lines = []
+
+            value.to_s.each_line do |line|
+              sanitized_lines << line.strip
+            end
+
+            return sanitized_lines.join("\n").strip
+          end
+        end
+
+      end
+    end
+  end
+end

data/lib/ronin/{extensions/meta.rb → model/types.rb}

@@ -1,7 +1,7 @@
 #
 # Ronin - A Ruby platform for exploit development and security research.
 #
-# Copyright (c) 2006-2009 Hal Brodigan (postmodern.mod3 at gmail.com)
+# Copyright (c) 2006-2010 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
@@ -18,4 +18,4 @@
 # Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
 #
 
-require 'ronin/extensions/meta/object'
+require 'ronin/model/types/description'

data/lib/ronin/model.rb

@@ -1,7 +1,7 @@
 #
 # Ronin - A Ruby platform for exploit development and security research.
 #
-# Copyright (c) 2006-2009 Hal Brodigan (postmodern.mod3 at gmail.com)
+# Copyright (c) 2006-2010 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

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

@@ -0,0 +1,152 @@
+#
+# 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/esmtp'
+
+require 'parameters'
+
+module Ronin
+  module Network
+    module Mixins
+      #
+      # Adds ESMTP convenience methods and connection parameters to a class.
+      #
+      module ESMTP
+        include Parameters
+
+        # ESMTP host
+        parameter :host,
+                  :type => String,
+                  :description => 'ESMTP host'
+
+        # ESMTP port
+        parameter :port,
+                  :type => Integer,
+                  :description => 'ESMTP port'
+
+        # ESMTP authentication method to use
+        parameter :esmtp_login,
+                  :type => String,
+                  :description => 'ESMTP authentication method to use'
+
+        # ESMTP user to login as
+        parameter :esmtp_user,
+                  :type => String,
+                  :description => 'ESMTP user to login as'
+
+        # ESMTP password to login with
+        parameter :esmtp_password,
+                  :type => String,
+                  :description => 'ESMTP password to login with'
+
+        protected
+
+        #
+        # @see Ronin::Network::SMTP.message.
+        #
+        def esmtp_message(options={},&block)
+          Network::SMTP.message(options,&block)
+        end
+
+        #
+        # Creates a connection to the ESMTP server. The {#host}, {#port},
+        # {#esmtp_login}, {#esmtp_user} and {#esmtp_password} parameters
+        # will also be used to connect to the ESMTP server.
+        #
+        # @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 esmtp_connect(options={},&block)
+          options[:port] ||= self.port
+          options[:login] ||= self.esmtp_login
+          options[:user] ||= self.esmtp_user
+          options[:password] ||= self.esmtp_password
+
+          if self.port
+            print_info "Connecting to #{self.host}:#{self.port} ..."
+          else
+            print_info "Connecting to #{self.host} ..."
+          end
+
+          return ::Net.esmtp_connect(self.host,options,&block)
+        end
+
+        #
+        # Starts a session with the ESMTP server. The {#host}, {#port},
+        # {#esmtp_login}, {#esmtp_user} and {#esmtp_password} parameters
+        # will also be used to connect to the ESMTP server.
+        #
+        # @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 esmtp_connect
+        #
+        def esmtp_session(options={})
+          esmtp_connect(options) do |sess|
+            yield sess if block_given?
+
+            sess.close
+
+            if self.port
+              print_info "Disconnecting from #{self.host}:#{self.port}"
+            else
+              print_info "Disconnecting from #{self.host}"
+            end
+          end
+        end
+      end
+    end
+  end
+end