ronin-exploits 0.2.1 → 0.3.0

data/spec/generators/payloads/binary_payload_spec.rb

@@ -0,0 +1,31 @@
+require 'ronin/generators/payloads/binary_payload'
+require 'ronin/payloads/binary_payload'
+
+require 'spec_helper'
+require 'generators/payloads/payload_examples'
+
+require 'tmpdir'
+require 'fileutils'
+
+describe Generators::Payloads::BinaryPayload do
+  before(:all) do
+    @path = File.join(Dir.tmpdir,'generated_payload.rb')
+
+    Generators::Payloads::BinaryPayload.generate(
+      {},
+      [@path]
+    )
+
+    @payload = Payloads::BinaryPayload.load_from(@path)
+  end
+
+  it_should_behave_like "a Payload"
+
+  it "should define a BinaryPayload" do
+    @payload.class.should == Payloads::BinaryPayload
+  end
+
+  after(:all) do
+    FileUtils.rm(@path)
+  end
+end

data/spec/generators/payloads/nops_spec.rb

@@ -0,0 +1,31 @@
+require 'ronin/generators/payloads/nops'
+require 'ronin/payloads/nops'
+
+require 'spec_helper'
+require 'generators/payloads/payload_examples'
+
+require 'tmpdir'
+require 'fileutils'
+
+describe Generators::Payloads::Nops do
+  before(:all) do
+    @path = File.join(Dir.tmpdir,'generated_payload.rb')
+
+    Generators::Payloads::Nops.generate(
+      {},
+      [@path]
+    )
+
+    @payload = Payloads::Nops.load_from(@path)
+  end
+
+  it_should_behave_like "a Payload"
+
+  it "should define a Nops payload" do
+    @payload.class.should == Payloads::Nops
+  end
+
+  after(:all) do
+    FileUtils.rm(@path)
+  end
+end

data/spec/generators/payloads/payload_examples.rb

@@ -0,0 +1,16 @@
+require 'spec_helper'
+
+shared_examples_for "a Payload" do
+  it "should set the name property" do
+    @payload.name.should == Generators::Payloads::Payload::DEFAULT_NAME
+  end
+
+  it "should set the description property" do
+    @payload.description.should == Generators::Payloads::Payload::DEFAULT_DESCRIPTION
+  end
+
+  it "should defines an author" do
+    @payload.authors.length.should == 1
+    @payload.authors.first.name.should == Author::ANONYMOUSE
+  end
+end

data/spec/generators/payloads/payload_spec.rb

@@ -0,0 +1,31 @@
+require 'ronin/generators/payloads/payload'
+require 'ronin/payloads/payload'
+
+require 'spec_helper'
+require 'generators/payloads/payload_examples'
+
+require 'tmpdir'
+require 'fileutils'
+
+describe Generators::Payloads::Payload do
+  before(:all) do
+    @path = File.join(Dir.tmpdir,'generated_payload.rb')
+
+    Generators::Payloads::Payload.generate(
+      {},
+      [@path]
+    )
+
+    @payload = Payloads::Payload.load_from(@path)
+  end
+
+  it_should_behave_like "a Payload"
+
+  it "should define a Payload" do
+    @payload.class.should == Payloads::Payload
+  end
+
+  after(:all) do
+    FileUtils.rm(@path)
+  end
+end

data/spec/generators/payloads/shellcode_spec.rb

@@ -0,0 +1,31 @@
+require 'ronin/generators/payloads/shellcode'
+require 'ronin/payloads/shellcode'
+
+require 'spec_helper'
+require 'generators/payloads/payload_examples'
+
+require 'tmpdir'
+require 'fileutils'
+
+describe Generators::Payloads::Shellcode do
+  before(:all) do
+    @path = File.join(Dir.tmpdir,'generated_payload.rb')
+
+    Generators::Payloads::Shellcode.generate(
+      {},
+      [@path]
+    )
+
+    @payload = Payloads::Shellcode.load_from(@path)
+  end
+
+  it_should_behave_like "a Payload"
+
+  it "should define a Shellcode payload" do
+    @payload.class.should == Payloads::Shellcode
+  end
+
+  after(:all) do
+    FileUtils.rm(@path)
+  end
+end

data/spec/model/models/targets_product_model.rb

@@ -0,0 +1,11 @@
+require 'ronin/model'
+require 'ronin/model/targets_product'
+
+class TargetsProductModel
+
+  include Ronin::Model
+  include Ronin::Model::TargetsProduct
+
+  property :id, Serial
+
+end

data/spec/model/targets_arch_spec.rb

@@ -5,7 +5,17 @@ require 'model/models/targets_arch_model'
 
 describe Model::TargetsArch do
   it "should define an arch relation" do
-    TargetsArchModel.relationships.has_key?('arch')
+    relationship = TargetsArchModel.relationships['arch']
+
+    relationship.should_not be_nil
+    relationship.parent_model.should == Arch
+  end
+
+  it "should define a relationship with Arch" do
+    relationship = Arch.relationships['targets_arch_models']
+
+    relationship.should_not be_nil
+    relationship.child_model.should == TargetsArchModel
   end
 
   it "should provide access to the targeted arch" do

data/spec/model/targets_os_spec.rb

@@ -5,7 +5,17 @@ require 'model/models/targets_os_model'
 
 describe Model::TargetsOS do
   it "should define an os relation" do
-    TargetsOSModel.relationships.has_key?('os')
+    relationship = TargetsOSModel.relationships['os']
+
+    relationship.should_not be_nil
+    relationship.parent_model.should == OS
+  end
+
+  it "should define a relationship with Arch" do
+    relationship = OS.relationships['targets_os_models']
+
+    relationship.should_not be_nil
+    relationship.child_model.should == TargetsOSModel
   end
 
   it "should provide access to the targeted os" do

data/spec/model/targets_product_spec.rb

@@ -0,0 +1,35 @@
+require 'ronin/model/targets_product'
+
+require 'spec_helper'
+require 'model/models/targets_product_model'
+
+describe Model::TargetsArch do
+  it "should define a product relation" do
+    relationship = TargetsProductModel.relationships['product']
+
+    relationship.should_not be_nil
+    relationship.parent_model.should == Product
+  end
+
+  it "should define a relationship with Product" do
+    relationship = Product.relationships['targets_product_models']
+
+    relationship.should_not be_nil
+    relationship.child_model.should == TargetsProductModel
+  end
+
+  it "should provide access to the targeted product" do
+    model = TargetsProductModel.new(
+      :product => Product.new(:name => 'Toaster')
+    )
+
+    model.product.name.should == 'Toaster'
+  end
+
+  it "should allow the product to be set via the getter method" do
+    model = TargetsProductModel.new
+    model.product :name => 'Toaster'
+
+    model.product.name.should == 'Toaster'
+  end
+end

data/spec/objects/exploits/{test.rb → simple.rb}

@@ -1,6 +1,6 @@
 ronin_exploit do
   cache do
-    self.name = 'test'
+    self.name = 'simple'
     self.version = '0.2'
 
     author :name => 'Anonymous', :email => 'anonymous@example.com'

data/spec/objects/payloads/{test.rb → simple.rb}

@@ -4,12 +4,16 @@ ronin_payload do
             :description => 'Custom value to use in building the payload'
 
   cache do
-    self.name = 'test'
+    self.name = 'simple'
 
     author :name => 'Anonymous', :email => 'anonymous@example.com'
   end
 
   def build
-    @payload = "code.#{@custom}"
+    @raw_payload = "code.#{@custom}"
+  end
+
+  def some_control
+    'control data'
   end
 end

data/spec/payloads/encoders/encoder_spec.rb

@@ -0,0 +1,30 @@
+require 'ronin/payloads/encoders/encoder'
+
+require 'spec_helper'
+
+describe Payloads::Encoders::Encoder do
+  before(:all) do
+    @encoder = Payloads::Encoders::Encoder.new(:name => 'test')
+    @data = 'some data'
+  end
+
+  it "should require a name" do
+    encoder = Payloads::Encoders::Encoder.new
+    encoder.should_not be_valid
+
+    encoder = Payloads::Encoders::Encoder.new(:name => 'encoder')
+    encoder.should be_valid
+  end
+
+  it "should provide a #encode method" do
+    @encoder.respond_to?(:encode).should == true
+  end
+
+  it "should return the data to be encoded by default" do
+    @encoder.encode(@data).should == @data
+  end
+
+  it "should have a custom inspect method" do
+    @encoder.inspect.should == '#<Ronin::Payloads::Encoders::Encoder: test>'
+  end
+end

data/spec/payloads/encoders/xor_spec.rb

@@ -12,7 +12,7 @@ describe Ronin do
       disallow = [0x00, 0x01, 0x90]
       xor = Payloads::Encoders::XOR.new(:disallow => disallow)
 
-      xor.call(@data).each_byte do |b|
+      xor.encode(@data).each_byte do |b|
         disallow.include?(b).should_not == true
       end
     end

data/spec/payloads/payload_spec.rb

@@ -2,13 +2,16 @@ require 'ronin/payloads/payload'
 
 require 'spec_helper'
 require 'helpers/objects'
+require 'controls/behaviors_examples'
 
 describe Payloads::Payload do
   before(:each) do
-    @payload = load_payload('test')
-    @exploit = load_exploit('example')
+    @payload = load_payload('simple')
+    @controler = @payload
   end
 
+  it_should_behave_like "controls behaviors"
+
   it "should require a name attribute" do
     payload = Payloads::Payload.new
     payload.should_not be_valid
@@ -41,19 +44,6 @@ describe Payloads::Payload do
     @payload.params.should_not be_empty
   end
 
-  it "should not have any controls by default" do
-    @payload.controls.should be_empty
-  end
-
-  it "should specify what behaviors the payload controls" do
-    @payload.controlling :memory_read, :memory_write
-
-    @payload.behaviors.should == [
-      Vuln::Behavior[:memory_read],
-      Vuln::Behavior[:memory_write]
-    ]
-  end
-
   it "should not have an Arch by default" do
     @payload.arch.should be_nil
   end
@@ -76,7 +66,7 @@ describe Payloads::Payload do
   end
 
   it "should allow for the extending of Helper modules" do
-    @payload.instance_eval { helper :shell }.should == true
+    @payload.instance_eval { helper :rpc }.should == true
   end
 
   it "should raise an UnknownHelper when extending an unknown helper" do
@@ -89,24 +79,46 @@ describe Payloads::Payload do
     @payload.exploit.should be_nil
   end
 
-  it "should have 'built' and 'unbiult' states" do
+  it "should be 'unbuilt' by default" do
+    @payload.should_not be_built
+  end
+
+  it "should have 'built' and 'unbuilt' states" do
     @payload.should_not be_built
     @payload.build!
     @payload.should be_built
   end
 
-  it "should store the built payload in the @payload instance variable" do
+  it "should store the raw payload in the @raw_payload instance variable" do
+    @payload.build!
+    @payload.raw_payload.should == "code.func"
+  end
+
+  it "should return the raw payload when calling build!" do
     @payload.build!
-    @payload.payload.should == "code.func"
+    @payload.raw_payload.should == "code.func"
   end
 
-  it "should return the built payload when calling build!" do
-    @payload.build!.should == "code.func"
+  it "should pass the raw payload to the block given to build!" do
+    @payload.build! do |payload|
+      payload.should == @payload
+    end
+  end
+
+  it "should respect the arity of blocks passed to build!" do
+    @payload.build do
+      @payload.should be_built
+    end
   end
 
   it "should use parameters in the building of the payload" do
     @payload.custom = 'hello'
-    @payload.build!.should == "code.hello"
+    @payload.build!
+    @payload.raw_payload.should == "code.hello"
+  end
+
+  it "should be 'undeployed' by default" do
+    @payload.should_not be_deployed
   end
 
   it "should have 'deployed' and 'undeployed' states" do
@@ -122,37 +134,23 @@ describe Payloads::Payload do
     end
   end
 
-  it "should pass the built payload to the exploit when deploying" do
-    @payload.exploit = @exploit
-
-    @payload.build!
+  it "should pass the deployed payload to the block given to deploy!" do
     @payload.deploy! do |payload|
-      payload.exploit.payload.should == "code.func"
+      payload.should be_deployed
     end
   end
 
-  it "should call the exploit when deployed" do
-    @payload.exploit = @exploit
-
-    @payload.build!
-    @payload.deploy! do |payload|
-      payload.exploit.should be_built
-      payload.exploit.should be_deployed
+  it "should respect the arity of blocks passed to deploy!" do
+    @payload.deploy! do
+      @payload.should be_deployed
     end
   end
 
-  it "should build and deploy the payload when called" do
-    @payload.call
-
-    @payload.should be_built
-    @payload.should be_deployed
-  end
-
   it "should return the name and the version when calling to_s" do
-    @payload.to_s.should == 'test 0.1'
+    @payload.to_s.should == 'simple 0.1'
   end
 
   it "should have a custom inspect method" do
-    @payload.inspect.should == '#<Ronin::Payloads::Payload: test 0.1 {:custom=>"func"}>'
+    @payload.inspect.should == '#<Ronin::Payloads::Payload: simple 0.1 {:custom=>"func"}>'
   end
 end