ronin 0.2.4 → 0.3.0

data/lib/ronin/yard/handlers/ruby/legacy/metaclass_eval_handler.rb

@@ -0,0 +1,21 @@
+require 'yard/handlers/ruby/legacy/base'
+
+module YARD
+  module Handlers
+    module Ruby
+      module Legacy
+        class MetaclassEvalHandler < Base
+
+          handles /(\A|\.)metaclass_eval(\s+|\()/
+
+          def process
+            if statement.block
+              parse_block(:namespace => namespace, :scope => :class)
+            end
+          end
+
+        end
+      end
+    end
+  end
+end

data/lib/ronin/yard/handlers/ruby/legacy/property_handler.rb

@@ -0,0 +1,41 @@
+require 'yard'
+
+module YARD
+  module Handlers
+    module Ruby
+      module Legacy
+        class PropertyHandler < Base
+
+          handles /\Aproperty\s/
+
+          def process
+            nobj = namespace
+            mscope = scope
+            name = statement.tokens[2,1].to_s[1..-1]
+
+            register MethodObject.new(nobj, name, :class) do |o|
+              o.visibility = :public
+              o.source = statement.source
+              o.signature = "def #{nobj}.#{name}(repository=nil)"
+              o.parameters = [['repository', 'nil']]
+            end
+
+            register MethodObject.new(nobj, name, mscope) do |o|
+              o.visibility = :public
+              o.source = statement.source
+              o.signature = "def #{name}"
+            end
+
+            register MethodObject.new(nobj, "#{name}=", mscope) do |o|
+              o.visibility = :public
+              o.source = statement.source
+              o.signature = "def #{name}=(value)"
+              o.parameters = [['value', nil]]
+            end
+          end
+
+        end
+      end
+    end
+  end
+end

data/lib/ronin/yard/handlers/ruby/legacy/scanner_handler.rb

@@ -0,0 +1,61 @@
+#
+# Ronin - A Ruby platform for exploit development and security research.
+#
+# Copyright (c) 2006-2009 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 'yard'
+
+module YARD
+  module Handlers
+    module Ruby
+      module Legacy
+        class ScannerHandler < Base
+
+          handles /\Ascanner(\s+|\()/
+
+          def process
+            nobj = namespace
+            mscope = scope
+            name = statement.tokens[2,1].to_s[1..-1]
+
+            register MethodObject.new(nobj, "first_#{name}", mscope) do |o|
+              o.visibility = :public
+              o.source = statement.source
+              o.signature = "def first_#{name}(options=true)"
+              o.parameters = [['options', 'true']]
+            end
+
+            register MethodObject.new(nobj, "has_#{name}?", mscope) do |o|
+              o.visibility = :public
+              o.source = statement.source
+              o.signature = "def has_#{name}?(options=true)"
+              o.parameters = [['options', 'true']]
+            end
+
+            register MethodObject.new(nobj, "#{name}_scan", mscope) do |o|
+              o.visibility = :public
+              o.source = statement.source
+              o.signature = "def #{name}_scan(options=true,&block)"
+              o.parameters = [['options', 'true'], ['&block', nil]]
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/ronin/yard/handlers/ruby/metaclass_eval_handler.rb

@@ -0,0 +1,18 @@
+require 'yard'
+
+module YARD
+  module Handlers
+    module Ruby
+      class MetaclassEvalHandler < YARD::Handlers::Ruby::Base
+
+        handles method_call(:metaclass_eval)
+
+        def process
+          if (block = statement.jump(:brace_block, :do_block).last)
+            parse_block(block, :namespace => namespace, :scope => :class)
+          end
+        end
+      end
+    end
+  end
+end

data/lib/ronin/yard/handlers/ruby/property_handler.rb

@@ -0,0 +1,45 @@
+require 'yard'
+
+require 'ronin/yard/handlers/ruby/base'
+
+module YARD
+  module Handlers
+    module Ruby
+      class PropertyHandler < Base
+
+        handles method_call(:property)
+
+        def process
+          nobj = effected_namespace
+          mscope = scope
+          name = statement.parameters[0].last
+
+          if name.type == :symbol
+            name = name.source[1..-1]
+
+            register MethodObject.new(nobj, name, :class) do |o|
+              o.visibility = :public
+              o.source = statement.source
+              o.signature = "def #{nobj}.#{name}(repository=nil)"
+              o.parameters = [['repository', 'nil']]
+            end
+
+            register MethodObject.new(nobj, name, mscope) do |o|
+              o.visibility = :public
+              o.source = statement.source
+              o.signature = "def #{name}"
+            end
+
+            register MethodObject.new(nobj, "#{name}=", mscope) do |o|
+              o.visibility = :public
+              o.source = statement.source
+              o.signature = "def #{name}=(value)"
+              o.parameters = [['value', nil]]
+            end
+          end
+        end
+
+      end
+    end
+  end
+end

data/lib/ronin/yard/handlers/ruby/scanner_handler.rb

@@ -0,0 +1,66 @@
+#
+# Ronin - A Ruby platform for exploit development and security research.
+#
+# Copyright (c) 2006-2009 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 'yard'
+
+module YARD
+  module Handlers
+    module Ruby
+      class ScannerHandler < Base
+
+        handles method_call(:scanner)
+
+        def process
+          obj = statement.parameters(false).first
+          nobj = namespace
+          mscope = scope
+          name = case obj.type
+                 when :symbol_literal
+                   obj.jump(:ident, :op, :kw, :const).source
+                 when :string_literal
+                   obj.jump(:string_content).source
+                 end
+
+          register MethodObject.new(nobj, "first_#{name}", mscope) do |o|
+            o.visibility = :public
+            o.source = statement.source
+            o.signature = "def first_#{name}(options=true)"
+            o.parameters = [['options', 'true']]
+          end
+
+          register MethodObject.new(nobj, "has_#{name}?", mscope) do |o|
+            o.visibility = :public
+            o.source = statement.source
+            o.signature = "def has_#{name}?(options=true)"
+            o.parameters = [['options', 'true']]
+          end
+
+          register MethodObject.new(nobj, "#{name}_scan", mscope) do |o|
+            o.visibility = :public
+            o.source = statement.source
+            o.signature = "def #{name}_scan(options=true,&block)"
+            o.parameters = [['options', 'true'], ['&block', nil]]
+          end
+        end
+
+      end
+    end
+  end
+end

data/spec/author_spec.rb

@@ -3,8 +3,8 @@ require 'ronin/author'
 require 'helpers/database'
 
 describe Author do
-  it "should have a default name" do
-    @author = Author.new
-    @author.name.should == Author::ANONYMOUSE
+  it "should have a String representation" do
+    author = Author.new(:name => 'test')
+    author.to_s.should == 'test'
   end
 end

data/spec/extensions/classes/some_class.rb

@@ -0,0 +1,2 @@
+class SomeClass
+end

data/spec/extensions/kernel_spec.rb

@@ -56,4 +56,21 @@ describe Kernel do
       }.should_not raise_error(StandardError)
     end
   end
+
+  describe "require_within" do
+    it "should require paths from within a directory" do
+      Object.const_defined?('SomeClass').should_not == true
+
+      require_within 'extensions/classes', 'some_class'
+
+      Object.const_defined?('SomeClass').should == true
+    end
+
+    it "should prevent directory traversal" do
+      lambda {
+        require_within 'extensions/classes',
+                       File.join('..','classes','some_class')
+      }.should raise_error(LoadError)
+    end
+  end
 end

data/spec/extensions/uri/http_spec.rb

@@ -6,35 +6,4 @@ 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/formatting/binary/integer_spec.rb

@@ -124,4 +124,8 @@ describe Integer do
       42.hex_escape.should == "\\x2a"
     end
   end
+
+  it "should alias char to the #chr method" do
+    0x41.char.should == 'A'
+  end
 end

data/spec/formatting/text/string_spec.rb

@@ -25,6 +25,54 @@ describe String do
         "_#{c}"
       }.should == "_h_e_l_l_o"
     end
+
+    it "should format specific bytes in a String" do
+      @string.format_chars(:included => [104, 108]) { |c|
+        c.upcase
+      }.should == 'HeLLo'
+    end
+
+    it "should not format specific bytes in a String" do
+      @string.format_chars(:excluded => [101, 111]) { |c|
+        c.upcase
+      }.should == 'HeLLo'
+    end
+
+    it "should format ranges of bytes in a String" do
+      @string.format_chars(:included => (104..108)) { |c|
+        c.upcase
+      }.should == 'HeLLo'
+    end
+
+    it "should not format ranges of bytes in a String" do
+      @string.format_chars(:excluded => (104..108)) { |c|
+        c.upcase
+      }.should == 'hEllO'
+    end
+
+    it "should format specific chars in a String" do
+      @string.format_chars(:included => ['h', 'l']) { |c|
+        c.upcase
+      }.should == 'HeLLo'
+    end
+
+    it "should not format specific bytes in a String" do
+      @string.format_chars(:excluded => ['e', 'o']) { |c|
+        c.upcase
+      }.should == 'HeLLo'
+    end
+
+    it "should format ranges of chars in a String" do
+      @string.format_chars(:included => ('h'..'l')) { |c|
+        c.upcase
+      }.should == 'HeLLo'
+    end
+
+    it "should not format ranges of chars in a String" do
+      @string.format_chars(:excluded => ('h'..'l')) { |c|
+        c.upcase
+      }.should == 'hEllO'
+    end
   end
 
   describe "format_bytes" do
@@ -33,6 +81,54 @@ describe String do
         sprintf("%%%x",b)
       }.should == "%68%65%6c%6c%6f"
     end
+
+    it "should format specific bytes in a String" do
+      @string.format_bytes(:included => [104, 108]) { |b|
+        b - 32
+      }.should == 'HeLLo'
+    end
+
+    it "should not format specific bytes in a String" do
+      @string.format_bytes(:excluded => [101, 111]) { |b|
+        b - 32
+      }.should == 'HeLLo'
+    end
+
+    it "should format ranges of bytes in a String" do
+      @string.format_bytes(:included => (104..108)) { |b|
+        b - 32
+      }.should == 'HeLLo'
+    end
+
+    it "should not format ranges of bytes in a String" do
+      @string.format_bytes(:excluded => (104..108)) { |b|
+        b - 32
+      }.should == 'hEllO'
+    end
+
+    it "should format specific chars in a String" do
+      @string.format_bytes(:included => ['h', 'l']) { |b|
+        b - 32
+      }.should == 'HeLLo'
+    end
+
+    it "should not format specific bytes in a String" do
+      @string.format_bytes(:excluded => ['e', 'o']) { |b|
+        b - 32
+      }.should == 'HeLLo'
+    end
+
+    it "should format ranges of chars in a String" do
+      @string.format_bytes(:included => ('h'..'l')) { |b|
+        b - 32
+      }.should == 'HeLLo'
+    end
+
+    it "should not format ranges of chars in a String" do
+      @string.format_bytes(:excluded => ('h'..'l')) { |b|
+        b - 32
+      }.should == 'hEllO'
+    end
   end
 
   describe "random_case" do