ronin 1.0.0.pre3 → 1.0.0.pre4

data/lib/ronin/url_query_param.rb

@@ -1,26 +1,27 @@
 #
-# Ronin - A Ruby platform for exploit development and security research.
+# Copyright (c) 2006-2011 Hal Brodigan (postmodern.mod3 at gmail.com)
 #
-# Copyright (c) 2006-2010 Hal Brodigan (postmodern.mod3 at gmail.com)
+# This file is part of Ronin.
 #
-# This program is free software; you can redistribute it and/or modify
+# Ronin 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
+# the Free Software Foundation, either version 3 of the License, or
 # (at your option) any later version.
 #
-# This program is distributed in the hope that it will be useful,
+# Ronin 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
+# along with Ronin.  If not, see <http://www.gnu.org/licenses/>.
 #
 
 require 'ronin/model/has_name'
 require 'ronin/model'
 
+require 'uri/query_params'
+
 module Ronin
   class URLQueryParam
 
@@ -36,5 +37,29 @@ module Ronin
     # The URL
     belongs_to :url, :model => 'URL'
 
+    #
+    # Converts the URL query param to a String.
+    #
+    # @return [String]
+    #   The dumped URL query param.
+    #
+    # @since 1.0.0
+    #
+    def to_s
+      URI::QueryParams.dump(self.name => self.value)
+    end
+
+    #
+    # Inspects the URL query param.
+    #
+    # @return [String]
+    #   The inspected URL query param.
+    #
+    # @since 1.0.0
+    #
+    def inspect
+      "#<#{self.class}: #{self}>"
+    end
+
   end
 end

data/lib/ronin/url_scheme.rb

@@ -1,21 +1,20 @@
 #
-# Ronin - A Ruby platform for exploit development and security research.
+# Copyright (c) 2006-2011 Hal Brodigan (postmodern.mod3 at gmail.com)
 #
-# Copyright (c) 2006-2010 Hal Brodigan (postmodern.mod3 at gmail.com)
+# This file is part of Ronin.
 #
-# This program is free software; you can redistribute it and/or modify
+# Ronin 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
+# the Free Software Foundation, either version 3 of the License, or
 # (at your option) any later version.
 #
-# This program is distributed in the hope that it will be useful,
+# Ronin 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
+# along with Ronin.  If not, see <http://www.gnu.org/licenses/>.
 #
 
 require 'ronin/model'

data/lib/ronin/user_name.rb

@@ -1,21 +1,20 @@
 #
-# Ronin - A Ruby platform for exploit development and security research.
+# Copyright (c) 2006-2011 Hal Brodigan (postmodern.mod3 at gmail.com)
 #
-# Copyright (c) 2006-2010 Hal Brodigan (postmodern.mod3 at gmail.com)
+# This file is part of Ronin.
 #
-# This program is free software; you can redistribute it and/or modify
+# Ronin 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
+# the Free Software Foundation, either version 3 of the License, or
 # (at your option) any later version.
 #
-# This program is distributed in the hope that it will be useful,
+# Ronin 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
+# along with Ronin.  If not, see <http://www.gnu.org/licenses/>.
 #
 
 require 'ronin/model'

data/lib/ronin/vendor.rb

@@ -1,21 +1,20 @@
 #
-# Ronin - A Ruby platform for exploit development and security research.
+# Copyright (c) 2006-2011 Hal Brodigan (postmodern.mod3 at gmail.com)
 #
-# Copyright (c) 2006-2010 Hal Brodigan (postmodern.mod3 at gmail.com)
+# This file is part of Ronin.
 #
-# This program is free software; you can redistribute it and/or modify
+# Ronin 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
+# the Free Software Foundation, either version 3 of the License, or
 # (at your option) any later version.
 #
-# This program is distributed in the hope that it will be useful,
+# Ronin 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
+# along with Ronin.  If not, see <http://www.gnu.org/licenses/>.
 #
 
 require 'ronin/model'

data/lib/ronin/version.rb

@@ -1,24 +1,23 @@
 #
-# Ronin - A Ruby platform for exploit development and security research.
+# Copyright (c) 2006-2011 Hal Brodigan (postmodern.mod3 at gmail.com)
 #
-# Copyright (c) 2006-2010 Hal Brodigan (postmodern.mod3 at gmail.com)
+# This file is part of Ronin.
 #
-# This program is free software; you can redistribute it and/or modify
+# Ronin 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
+# the Free Software Foundation, either version 3 of the License, or
 # (at your option) any later version.
 #
-# This program is distributed in the hope that it will be useful,
+# Ronin 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
+# along with Ronin.  If not, see <http://www.gnu.org/licenses/>.
 #
 
 module Ronin
   # Ronin version 
-  VERSION = '1.0.0.pre3'
+  VERSION = '1.0.0.pre4'
 end

data/lib/ronin/web_credential.rb

@@ -1,32 +1,76 @@
 #
-# Ronin - A Ruby platform for exploit development and security research.
+# Copyright (c) 2006-2011 Hal Brodigan (postmodern.mod3 at gmail.com)
 #
-# Copyright (c) 2009-2010 Hal Brodigan (postmodern.mod3 at gmail.com)
+# This file is part of Ronin.
 #
-# This program is free software; you can redistribute it and/or modify
+# Ronin 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
+# the Free Software Foundation, either version 3 of the License, or
 # (at your option) any later version.
 #
-# This program is distributed in the hope that it will be useful,
+# Ronin 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
+# along with Ronin.  If not, see <http://www.gnu.org/licenses/>.
 #
 
 require 'ronin/credential'
+require 'ronin/email_address'
 require 'ronin/url'
 
 module Ronin
   class WebCredential < Credential
 
-    # the url the credential can be used with
+    # The optional email address the credential is associated with.
+    belongs_to :email_address, :required => false
+
+    # The url the credential can be used with.
     belongs_to :url, :required => false,
                      :model => 'URL'
 
+    #
+    # Searches all web credentials that are associated with an
+    # email address.
+    #
+    # @param [String] email
+    #   The email address to search for.
+    #
+    # @return [Array<WebCredential>]
+    #   The web credentials associated with the email address.
+    #
+    # @since 1.0.0
+    #
+    def self.with_email(email)
+      unless email.include?('@')
+        raise("invalid email address #{email.dump}")
+      end
+
+      user, domain = email.split('@',2)
+
+      return all(
+        'email_address.user_name.name' => user,
+        'email_address.host_name.address' => domain
+      )
+    end
+
+    #
+    # Converts the web credential to a String.
+    #
+    # @return [String]
+    #   The user name, clear-text password and the optional email address.
+    #
+    # @since 1.0.0
+    #
+    def to_s
+      if self.email_address
+        "#{super} (#{self.email_address})"
+      else
+        super
+      end
+    end
+
   end
 end

data/spec/arch_spec.rb

@@ -57,4 +57,11 @@ describe Arch do
   it "should provide built-in archs" do
     Arch.i386.should_not be_nil
   end
+
+  it "should implicitly splat the endian and address length" do
+    endian, address_length = Arch.i386
+
+    endian.should == Arch.i386.endian
+    address_length.should == Arch.i386.address_length
+  end
 end

data/spec/author_spec.rb

@@ -7,7 +7,7 @@ describe Author do
     author.to_s.should == 'test'
   end
 
-  describe "to_s" do
+  describe "#to_s" do
     it "should return the name when their is no email" do
       author = Author.new(:name => 'anonymous')
 

data/spec/campaign_spec.rb

@@ -2,7 +2,7 @@ require 'spec_helper'
 require 'ronin/campaign'
 
 describe Campaign do
-  describe "filename" do
+  describe "#filename" do
     it "should return nil if no name is set" do
       campaign = Campaign.new
       

data/spec/credential_spec.rb

@@ -0,0 +1,33 @@
+require 'spec_helper'
+require 'ronin/credential'
+
+describe Credential do
+  let(:name) { 'alice' }
+  let(:secret) { 'secret' }
+
+  subject do
+    Credential.new(
+      :user_name => {:name => name},
+      :password => {:clear_text => secret}
+    )
+  end
+
+  it "should provide the user-name" do
+    subject.user.should == name
+  end
+
+  it "should provide the clear-text password" do
+    subject.clear_text.should == secret
+  end
+
+  it "should be convertable to a String" do
+    subject.to_s.should == "#{name}:#{secret}"
+  end
+
+  it "should implicitly splat the user-name and password" do
+    user_name, password = subject
+
+    user_name.should == name
+    password.should == secret
+  end
+end

data/spec/database_spec.rb

@@ -2,4 +2,27 @@ require 'spec_helper'
 require 'ronin/database'
 
 describe Database do
+  describe "repositories" do
+    it "should not be empty" do
+      subject.repositories.should_not be_empty
+    end
+
+    it "should have a ':default' repository" do
+      subject.repositories[:default].should_not be_nil
+    end
+  end
+
+  it "shold determine if a repository is defined" do
+    subject.repository?(:default).should == true
+  end
+
+  it "should determine when a repository is setup" do
+    subject.setup?(:default).should == true
+  end
+
+  it "should not allow switching to unknown repositories" do
+    lambda {
+      subject.repository(:foo) { }
+    }.should raise_error(Database::UnknownRepository)
+  end
 end

data/spec/email_address_spec.rb

@@ -29,7 +29,22 @@ describe EmailAddress do
     end
   end
 
+  it "should provide the user-name" do
+    subject.user.should == user
+  end
+
+  it "should provide the host-name" do
+    subject.host.should == host
+  end
+
   it "should convert to a String" do
     subject.to_s.should == email
   end
+
+  it "should implicitly splat the user-name and host-name" do
+    user_name, host_name = subject
+
+    user_name.should == user
+    host_name.should == host
+  end
 end

data/spec/engine/buildable_spec.rb

@@ -20,7 +20,7 @@ describe Engine::Buildable do
     subject.class.included_modules.should include(Engine::Verifiable)
   end
 
-  describe "build!" do
+  describe "#build!" do
     it "should call the build block" do
       subject.build!
 
@@ -41,7 +41,7 @@ describe Engine::Buildable do
     end
   end
 
-  describe "verify!" do
+  describe "#verify!" do
     it "should raise a NotBuilt exception when verifying unbuilt engines" do
       lambda {
         subject.verify!

data/spec/engine/deployable_spec.rb

@@ -23,7 +23,7 @@ describe Engine::Deployable do
     subject.should_not be_deployed
   end
 
-  describe "deploy!" do
+  describe "#deploy!" do
     it "should verify! the engine before deploying it" do
       subject.var = -1
 
@@ -45,7 +45,7 @@ describe Engine::Deployable do
     end
   end
 
-  describe "evacuate!" do
+  describe "#evacuate!" do
     it "should mark the engine as evacuated" do
       subject.evacuate!
 

data/spec/host_name_spec.rb

@@ -13,6 +13,10 @@ describe HostName do
     host_name.should_not be_valid
   end
 
+  it "should alias #name to #address" do
+    subject.name.should == subject.address
+  end
+
   describe "lookup" do
     subject { HostName }
 
@@ -25,6 +29,14 @@ describe HostName do
       host_names[0].address.should == example_domain
     end
 
+    it "should associate the host names with the original IP address" do
+      host_names = subject.lookup(example_ip)
+      
+      host_names.each do |host|
+        host.ip_addresses[0].address.should == example_ip
+      end
+    end
+
     it "should return an empty Array for unknown host names" do
       host_names = subject.lookup(bad_ip)
       
@@ -42,6 +54,14 @@ describe HostName do
       ips[0].address.should == example_ip
     end
 
+    it "should associate the IP addresses with the original host name" do
+      ips = subject.lookup!
+      
+      ips.each do |ip|
+        ip.host_names[0].address.should == example_domain
+      end
+    end
+
     it "should return an empty Array for unknown host names" do
       ips = HostName.new(:address => bad_domain).lookup!