ronin 0.0.9 → 0.1.0

data/lib/ronin/target.rb

@@ -36,5 +36,8 @@ module Ronin
     # Targeted architecture
     belongs_to :arch
 
+    # Validates
+    validates_present :platform, :arch
+
   end
 end

data/lib/ronin/version.rb

@@ -23,5 +23,5 @@
 
 module Ronin
   # Ronin version 
-  VERSION = '0.0.9'
+  VERSION = '0.1.0'
 end

data/spec/arch_spec.rb

@@ -0,0 +1,61 @@
+require 'ronin/arch'
+
+require 'spec_helper'
+
+describe Arch do
+  it "should require a name, endian and address_length attributes" do
+    @arch = Arch.new
+    @arch.should_not be_valid
+    
+    @arch.name = 'future'
+    @arch.should_not be_valid
+   
+    @arch.endian = 'little'
+    @arch.should_not be_valid
+
+    @arch.address_length = 4
+    @arch.should be_valid
+  end
+
+  it "should require a unique name" do
+    @first_arch = Arch.create(:name => 'cats',
+                              :endian => 'little',
+                              :address_length => 4)
+    @first_arch.should be_valid
+
+    @second_arch = Arch.new(:name => 'cats',
+                            :endian => 'big',
+                            :address_length => 4)
+    @second_arch.should_not be_valid
+  end
+
+  it "should require either 'little' or 'big' for the endian attribute" do
+    @arch = Arch.new(:name => 'test',
+                     :endian => 'lol',
+                     :address_length => 4)
+    @arch.should_not be_valid
+
+    @arch.endian = 'little'
+    @arch.should be_valid
+
+    @arch.endian = 'big'
+    @arch.should be_valid
+  end
+
+  it "should require a numeric valid for the address_length attribute" do
+    @arch = Arch.new(:name => 'test2',
+                     :endian => 'big',
+                     :address_length => 'x')
+    @arch.should_not be_valid
+
+    @arch.address_length = '4'
+    @arch.should be_valid
+
+    @arch.address_length = 4
+    @arch.should be_valid
+  end
+
+  it "should provide built-in archs" do
+    Arch.i386.should_not be_nil
+  end
+end

data/spec/author_spec.rb

@@ -0,0 +1,10 @@
+require 'ronin/author'
+
+require 'spec_helper'
+
+describe Author do
+  it "should have a default name" do
+    @author = Author.new
+    @author.name.should == Author::ANONYMOUSE
+  end
+end

data/spec/chars/chars_spec.rb

@@ -0,0 +1,82 @@
+require 'ronin/chars/chars'
+
+require 'spec_helper'
+
+describe Ronin do
+  describe Chars do
+    before(:all) do
+      @numeric_string = Chars.numeric.random_string(10)
+      @octal_string = Chars.octal.random_string(10)
+      @uppercase_hex_string = Chars.uppercase_hexadecimal.random_string(10)
+      @lowercase_hex_string = Chars.lowercase_hexadecimal.random_string(10)
+      @hex_string = Chars.hexadecimal.random_string(10)
+      @uppercase_alpha_string = Chars.uppercase_alpha.random_string(10)
+      @lowercase_alpha_string = Chars.lowercase_alpha.random_string(10)
+      @alpha_string = Chars.alpha.random_string(10)
+      @alpha_numeric = Chars.alpha_numeric.random_string(10)
+      @space_string = Chars.space.random_string(10)
+      @punctuation_string = Chars.punctuation.random_string(10)
+      @symbols_string = Chars.symbols.random_string(10)
+      @control_string = Chars.control.random_string(10)
+      @ascii_string = Chars.ascii.random_string(10)
+      @all_string = Chars.all.random_string(10)
+    end
+
+    it "should provide a numeric CharSet" do
+      (@numeric_string =~ /[0-9]{10}/).should_not be_nil
+    end
+
+    it "should provide an octal CharSet" do
+      (@octal_string =~ /[0-7]{10}/).should_not be_nil
+    end
+
+    it "should provide an upper-case hexadecimal CharSet" do
+      (@uppercase_hex_string =~ /[0-9A-F]{10}/).should_not be_nil
+    end
+
+    it "should provide a lower-case hexadecimal CharSet" do
+      (@lowercase_hex_string =~ /[0-9a-f]{10}/).should_not be_nil
+    end
+
+    it "should provide a hexadecimal CharSet" do
+      (@hex_string =~ /[0-9A-Fa-f]{10}/).should_not be_nil
+    end
+
+    it "should provide an upper-case alpha CharSet" do
+      (@uppercase_alpha_string =~ /[A-Z]{10}/).should_not be_nil
+    end
+
+    it "should provide a lower-case alpha CharSet" do
+      (@lowercase_alpha_string =~ /[a-z]{10}/).should_not be_nil
+    end
+
+    it "should provide an alpha CharSet" do
+      (@alpha_string =~ /[A-Za-z]{10}/).should_not be_nil
+    end
+
+    it "should provide an alpha-numeric CharSet" do
+      (@alpha_numeric_string =~ /[A-Za-z0-9]{10}/).should_not be_nil
+    end
+
+    it "should provide a space CharSet" do
+      (@space_string =~ /[ \f\n\r\t\v]{10}/).should_not be_nil
+    end
+
+    it "should provide a punctuation CharSet" do
+      (@punctuation_string =~ /[ \'\"\`\,\;\:\~\-\(\)\[\]\{\}\.\?\!]{10}/).should_not be_nil
+    end
+
+    it "should provide a symbols CharSet" do
+      #(@symbols_string =~ /[\@\#\$\%\^\&\*\_\+\=\|\\\<\>\/]{10}/).should_not be_nil
+    end
+
+    it "should provide a control CharSet" do
+    end
+
+    it "should provide an ascii CharSet" do
+    end
+
+    it "should provide an 'all' CharSet" do
+    end
+  end
+end

data/spec/context/context_spec.rb

@@ -0,0 +1,84 @@
+require 'ronin/context'
+
+require 'spec_helper'
+
+describe Ronin do
+  describe Context do
+    before(:all) do
+      require 'context/helpers/book_context'
+      require 'context/helpers/book_review_context'
+
+      @contexts_dir = File.expand_path(File.join(File.dirname(__FILE__),'helpers','contexts'))
+      @snow_crash_path = File.join(@contexts_dir,'snow_crash.rb')
+      @neuromancer_path = File.join(@contexts_dir,'neuromancer_review.rb')
+    end
+
+    it "should contain defined contexts" do
+      Context.is_context?(:book).should == true
+    end
+
+    it "should create a class-level context name" do
+      Book.context_name.should == :book
+    end
+
+    it "should raise an UnknownContext exception when loading unknwon-contexts" do
+      lambda {
+        Context.load_context(:nothing, 'some_path.rb')
+      }.should raise_error(UnknownContext)
+    end
+
+    it "should raise a ContextNotFound exception when loading from non-existant files" do
+      lambda {
+        Context.load_context(:book, 'not_here.rb')
+      }.should raise_error(ContextNotFound)
+    end
+
+    it "should load contexts by context-name from a file" do
+      @book = Context.load_context(:book, @snow_crash_path)
+      @book.should_not be_nil
+      @book.class.should == Book
+    end
+
+    it "should load a specific context from a file with multiple contexts" do
+      @book = Context.load_context(:book, @neuromancer_path)
+      @book.should_not be_nil
+      @book.class.should == Book
+
+      @review = Context.load_context(:book_review, @neuromancer_path)
+      @review.should_not be_nil
+      @review.class.should == BookReview
+    end
+
+    it "should provide class-level methods for loading a context" do
+      @book = Book.load_context(@snow_crash_path)
+      @book.should_not be_nil
+      @book.class.should == Book
+    end
+
+    it "should provide top-level ronin methods for loading a context" do
+      @book = ronin_load_book(@snow_crash_path)
+      @book.should_not be_nil
+      @book.class.should == Book
+    end
+
+    describe "loaded contexts" do
+      before(:all) do
+        @book = Book.load_context(@snow_crash_path)
+      end
+
+      it "should have a context name" do
+        @book.context_name.should == :book
+      end
+
+      it "should have attributes" do
+        @book.title.should == 'Snow Crash'
+        @book.author.should == 'Neal Stephenson'
+      end
+
+      it "should have instance methods" do
+        @book.methods.include?('rating').should == true
+        @book.rating.should == 10
+      end
+    end
+  end
+end

data/spec/context/helpers/book_context.rb

@@ -0,0 +1,15 @@
+require 'ronin/context'
+
+class Book
+
+  include Context
+
+  contextify :book
+
+  # Title of the book
+  attr_accessor :title
+
+  # Author of the book
+  attr_accessor :author
+
+end

data/spec/context/helpers/book_review_context.rb

@@ -0,0 +1,21 @@
+require 'ronin/context'
+
+class BookReview
+
+  include Context
+
+  contextify :book_review
+
+  # Title of the book
+  attr_accessor :book_title
+
+  # Author of the book
+  attr_accessor :book_author
+
+  # Author of this review
+  attr_accessor :author
+
+  # Summary
+  attr_accessor :summary
+
+end

data/spec/context/helpers/contexts/neuromancer_review.rb

@@ -0,0 +1,15 @@
+ronin_book do
+  @title = 'Neuromancer'
+  @author = 'William Gibson'
+end
+
+ronin_book_review do
+  @book_title = 'Neuromancer'
+  @book_author = 'William Gibson'
+
+  @author = 'postmodern'
+  @summary = %{
+    Classic cyber-punk book. Provides you with a very realistic and gritty
+    vision of the future and the characters that inhabit it.
+  }
+end

data/spec/context/helpers/contexts/snow_crash.rb

@@ -0,0 +1,8 @@
+ronin_book do
+  @title = 'Snow Crash'
+  @author = 'Neal Stephenson'
+
+  def rating
+    10
+  end
+end

data/spec/extensions/hash_spec.rb

@@ -0,0 +1,38 @@
+require 'ronin/extensions/hash'
+
+require 'spec_helper'
+
+describe Hash do
+  before(:all) do
+    @hash = {:a => 1, :b => 2, :c => 3}
+  end
+
+  it "should be able to be exploded" do
+    @exploded = @hash.explode(:x)
+    @exploded.length.should == 3
+
+    @exploded.each do |key,new_hash|
+      new_hash[key].should == :x
+    end
+  end
+
+  it "should explode only with specific keys" do
+    @keys = [:a, :c]
+    @exploded = @hash.explode(:x, :included => @keys)
+
+    @exploded.each do |key,new_hash|
+      @keys.include?(key).should == true
+      new_hash[key].should == :x
+    end
+  end
+
+  it "should not explode on specified keys" do
+    @keys = [:b]
+    @exploded = @hash.explode(:x, :excluded => @keys)
+
+    @exploded.each do |key,new_hash|
+      @keys.include?(key).should == false
+      new_hash[key].should == :x
+    end
+  end
+end

data/spec/extensions/string_spec.rb

@@ -0,0 +1,13 @@
+require 'ronin/extensions/string'
+
+require 'spec_helper'
+
+describe String do
+  it "should be able to convert proper names to method names" do
+    "Proper Name".to_method_name.should == 'proper_name'
+  end
+
+  it "should be able to convert Class names to method names" do
+    "Namespace::Test".to_method_name.should == 'namespace_test'
+  end
+end

data/spec/extensions/uri/http_spec.rb

@@ -0,0 +1,40 @@
+require 'ronin/extensions/uri'
+
+require 'spec_helper'
+
+describe URI::HTTP do
+  it "should include QueryParams" do
+    URI::HTTP.include?(URI::QueryParams).should == true
+  end
+
+  it "should have explodable query params" do
+    @url = URI('http://search.dhgate.com/search.do?dkp=1&searchkey=yarn&catalog=')
+
+    @urls = @url.explode_query_params('x')
+    @urls.each do |param,new_url|
+      new_url.query_params[param].should == 'x'
+    end
+  end
+
+  it "should only explode query params with specified params" do
+    @params = ['dkp', 'catalog']
+    @url = URI('http://search.dhgate.com/search.do?dkp=1&searchkey=yarn&catalog=')
+
+    @urls = @url.explode_query_params('x', :included => @params)
+    @urls.each do |param,new_url|
+      @params.include?(param).should == true
+      new_url.query_params[param].should == 'x'
+    end
+  end
+
+  it "should not explode query params with certain params" do
+    @params = ['searchkey']
+    @url = URI('http://search.dhgate.com/search.do?dkp=1&searchkey=yarn&catalog=')
+
+    @urls = @url.explode_query_params('x', :excluded => @params)
+    @urls.each do |param,new_url|
+      @params.include?(param).should == false
+      new_url.query_params[param].should == 'x'
+    end
+  end
+end

data/spec/extensions/uri/query_params_spec.rb

@@ -0,0 +1,38 @@
+require 'ronin/extensions/uri'
+
+require 'spec_helper'
+
+describe URI::QueryParams do
+  before(:each) do
+    @uri = URI('http://www.test.com/page.php?x=1&y=one%20two&z')
+  end
+
+  it "should provide #query_params" do
+    @uri.should respond_to(:query_params)
+  end
+
+  it "#query_params should be a Hash" do
+    @uri.query_params.class.should == Hash
+  end
+
+  it "#query_params should contain params" do
+    @uri.query_params.empty?.should == false
+  end
+
+  it "#query_params can contain single-word params" do
+    @uri.query_params['x'].should == '1'
+  end
+
+  it "#query_params can contain multi-word params" do
+    @uri.query_params['y'].should == 'one two'
+  end
+
+  it "#query_params can contain empty params" do
+    @uri.query_params['z'].should be_nil
+  end
+
+  it "should update #query_params along with #query=" do
+    @uri.query = 'u=3'
+    @uri.query_params['u'].should == '3'
+  end
+end