ronin-support 0.4.1 → 0.5.0.rc1

data/spec/extensions/regexp_spec.rb

@@ -2,6 +2,38 @@ require 'spec_helper'
 require 'ronin/extensions/regexp'
 
 describe Regexp do
+  describe "WORD" do
+    let(:word) { 'dog' }
+
+    subject { Regexp::WORD }
+
+    it "should not match single letters" do
+      subject.match('A').should be_nil
+    end
+
+    it "should not match numeric letters" do
+      subject.match("123#{word}123")[0].should == word
+    end
+
+    it "should not include ending periods" do
+      subject.match("#{word}.")[0].should == word
+    end
+
+    it "should not include leading punctuation" do
+      subject.match("'#{word}")[0].should == word
+    end
+
+    it "should not include tailing punctuation" do
+      subject.match("#{word}'")[0].should == word
+    end
+
+    it "should include punctuation in the middle of the word" do
+      name = "O'Brian"
+
+      subject.match(name)[0].should == name
+    end
+  end
+
   describe "OCTET" do
     subject { Regexp::OCTET }
 
@@ -172,6 +204,34 @@ describe Regexp do
     end
   end
 
+  describe "PHONE_NUMBER" do
+    subject { Regexp::PHONE_NUMBER }
+
+    it "should match 111-2222" do
+      number = '111-2222'
+
+      subject.match(number)[0].should == number
+    end
+
+    it "should match 111-2222x9" do
+      number = '111-2222x9'
+
+      subject.match(number)[0].should == number
+    end
+
+    it "should match 800-111-2222" do
+      number = '800-111-2222'
+
+      subject.match(number)[0].should == number
+    end
+
+    it "should match 1-800-111-2222" do
+      number = '1-800-111-2222'
+
+      subject.match(number)[0].should == number
+    end
+  end
+
   describe "IDENTIFIER" do
     subject { Regexp::IDENTIFIER }
 

data/spec/extensions/string_spec.rb

@@ -157,7 +157,7 @@ describe String do
     end
   end
 
-  if RUBY_VERSION < '1.9'
+  if RUBY_VERSION < '1.9.'
     describe "#dump" do
       it "should dump printable strings" do
         "hello".dump.should == '"hello"'

data/spec/formatting/binary/array_spec.rb

@@ -0,0 +1,22 @@
+require 'spec_helper'
+require 'ronin/formatting/extensions/binary/array'
+
+describe Array do
+  subject { [0x1234, "hello"] }
+
+  it "should provide Array#pack" do
+    should respond_to(:pack)
+  end
+
+  describe "#pack" do
+    let(:packed) { "\x34\x12hello\0" }
+
+    it "should pack elements using Array#pack codes" do
+      subject.pack('vZ*').should == packed
+    end
+
+    it "should pack elements using Binary::Template types" do
+      subject.pack(:uint16_le, :string).should == packed
+    end
+  end
+end

data/spec/formatting/binary/base64_spec.rb

@@ -0,0 +1,50 @@
+require 'spec_helper'
+require 'ronin/formatting/extensions/binary/base64'
+
+describe Base64 do
+  subject { described_class }
+
+  if RUBY_VERSION < '1.9.'
+    let(:string) { '>' * 64 }
+
+    let(:base64) { "Pj4+Pj4+Pj4+Pj4+Pj4+Pj4+Pj4+Pj4+Pj4+Pj4+Pj4+Pj4+Pj4+Pj4+Pj4+\nPj4+Pj4+Pj4+Pj4+Pj4+Pj4+Pg==\n" }
+    let(:strict_base64) { "Pj4+Pj4+Pj4+Pj4+Pj4+Pj4+Pj4+Pj4+Pj4+Pj4+Pj4+Pj4+Pj4+Pj4+Pj4+Pj4+Pj4+Pj4+Pj4+Pj4+Pj4+Pg==" }
+    let(:urlsafe_base64) { "Pj4-Pj4-Pj4-Pj4-Pj4-Pj4-Pj4-Pj4-Pj4-Pj4-Pj4-Pj4-Pj4-Pj4-Pj4-Pj4-Pj4-Pj4-Pj4-Pj4-Pj4-Pg==" }
+
+    describe "strict_encode64" do
+      it "should strictly encode a String" do
+        subject.strict_encode64(string).should == strict_base64
+      end
+    end
+
+    describe "strict_decode64" do
+      it "should strict decode an encoded Base64 String" do
+        subject.strict_decode64(strict_base64).should == string
+      end
+
+      it "should raise an ArgumentError for non-strictly encoded Base64" do
+        lambda {
+          subject.strict_decode64(base64)
+        }.should raise_error(ArgumentError)
+      end
+    end
+
+    describe "urlsafe_encode64" do
+      it "should encode URL-safe Base64 Strings" do
+        subject.urlsafe_encode64(string).should == urlsafe_base64
+      end
+    end
+
+    describe "urlsafe_decode64" do
+      it "should decode URL-safe Base64 String" do
+        subject.urlsafe_decode64(urlsafe_base64).should == string
+      end
+
+      it "should raise an ArgumentError for non-strictly encoded Base64" do
+        lambda {
+          subject.urlsafe_decode64(base64)
+        }.should raise_error(ArgumentError)
+      end
+    end
+  end
+end

data/spec/formatting/binary/float_spec.rb

@@ -0,0 +1,30 @@
+require 'spec_helper'
+require 'ronin/formatting/extensions/binary/float'
+
+describe Float do
+  subject { 0.42 }
+
+  it "should provide Float#pack" do
+    should respond_to(:pack)
+  end
+
+  describe "#pack" do
+    let(:packed) { "\xE1z\x14\xAEG\xE1\xDA?" }
+
+    it "should unpack Strings using String#unpack template Strings" do
+      subject.pack('E').should == packed
+    end
+
+    it "should unpack Strings using a Binary::Template" do
+      subject.pack(:double_le).should == packed
+    end
+
+    context "when given non-Float Binary::Template types" do
+      it "should raise an ArgumentError" do
+        lambda {
+          subject.pack(:int)
+        }.should raise_error(ArgumentError)
+      end
+    end
+  end
+end

data/spec/formatting/binary/integer_spec.rb

@@ -1,5 +1,5 @@
 require 'spec_helper'
-require 'ronin/formatting/binary'
+require 'ronin/formatting/extensions/binary/integer'
 
 require 'ostruct'
 
@@ -81,60 +81,74 @@ describe Integer do
   end
 
   describe "#pack" do
-    let(:i386) do
-      OpenStruct.new(:endian => :little, :address_length => 4)
+    subject { 0x1337 }
+
+    let(:packed) { "7\023\000\000" }
+
+    it "should pack Integers using Array#pack codes" do
+      subject.pack('V').should == packed
     end
 
-    let(:ppc) do
-      OpenStruct.new(:endian => :big, :address_length => 4)
+    it "should pack Integers using Binary::Template types" do
+      subject.pack(:uint32_le).should == packed
     end
 
-    subject { 0x1337 }
+    context "when given non-Integer Binary::Template types" do
+      it "should raise an ArgumentError" do
+        lambda {
+          subject.pack(:float)
+        }.should raise_error(ArgumentError)
+      end
+    end
 
-    let(:i386_packed_int)   { "7\023\000\000" }
-    let(:i386_packed_short) { "7\023" }
-    let(:i386_packed_long)  { "7\023\000\000" }
-    let(:i386_packed_quad)  { "7\023\000\000\000\000\000\000" }
+    context "deprecated" do
+      let(:uint16_le) { "7\023" }
+      let(:uint32_le) { "7\023\000\000" }
+      let(:uint64_le) { "7\023\000\000\000\000\000\000" }
 
-    let(:ppc_packed_int)   { "\000\000\0237" }
-    let(:ppc_packed_short) { "\0237" }
-    let(:ppc_packed_long)  { "\000\000\0237" }
-    let(:ppc_packed_quad)  { "\000\000\000\000\000\000\0237" }
+      let(:uint16_be) { "\0237" }
+      let(:uint32_be) { "\000\000\0237" }
+      let(:uint64_be) { "\000\000\000\000\000\000\0237" }
 
-    it "should pack itself for a little-endian architecture by default" do
-      subject.pack(i386).should == i386_packed_int
-    end
+      let(:i386) do
+        OpenStruct.new(:endian => :little, :address_length => 4)
+      end
 
-    it "should pack itself as a short for a little-endian architecture" do
-      subject.pack(i386,2).should == i386_packed_short
-    end
+      let(:ppc) do
+        OpenStruct.new(:endian => :big, :address_length => 4)
+      end
 
-    it "should pack itself as a long for a little-endian architecture" do
-      subject.pack(i386,4).should == i386_packed_long
-    end
+      it "should pack itself for a little-endian architecture by default" do
+        subject.pack(i386).should == uint32_le
+      end
 
-    it "should pack itself as a quad for a little-endian architecture" do
-      subject.pack(i386,8).should == i386_packed_quad
-    end
+      it "should pack itself as a short for a little-endian architecture" do
+        subject.pack(i386,2).should == uint16_le
+      end
 
-    it "should pack itself for a big-endian architecture" do
-      subject.pack(ppc).should == ppc_packed_int
-    end
+      it "should pack itself as a long for a little-endian architecture" do
+        subject.pack(i386,4).should == uint32_le
+      end
 
-    it "should pack itself as a short for a big-endian architecture" do
-      subject.pack(ppc,2).should == ppc_packed_short
-    end
+      it "should pack itself as a quad for a little-endian architecture" do
+        subject.pack(i386,8).should == uint64_le
+      end
 
-    it "should pack itself as a long for a big-endian architecture" do
-      subject.pack(ppc,4).should == ppc_packed_long
-    end
+      it "should pack itself for a big-endian architecture" do
+        subject.pack(ppc).should == uint32_be
+      end
 
-    it "should pack itself as a quad for a big-endian architecture" do
-      subject.pack(ppc,8).should == ppc_packed_quad
-    end
+      it "should pack itself as a short for a big-endian architecture" do
+        subject.pack(ppc,2).should == uint16_be
+      end
+
+      it "should pack itself as a long for a big-endian architecture" do
+        subject.pack(ppc,4).should == uint32_be
+      end
 
-    it "should accept Array#pack template strings" do
-      subject.pack('L').should == i386_packed_long
+      it "should pack itself as a quad for a big-endian architecture" do
+        subject.pack(ppc,8).should == uint64_be
+      end
     end
   end
 end

data/spec/formatting/binary/string_spec.rb

@@ -1,12 +1,19 @@
 require 'spec_helper'
-require 'ronin/formatting/binary'
+require 'ronin/formatting/extensions/binary/string'
 
-require 'formatting/binary/helpers/hexdumps'
 require 'ostruct'
 
 describe String do
   subject { "hello" }
 
+  it "should provide String#unpack_original" do
+    should respond_to(:unpack_original)
+  end
+
+  it "should provide String#unpack" do
+    should respond_to(:unpack)
+  end
+
   it "should provide String#depack" do
     should respond_to(:depack)
   end
@@ -31,61 +38,77 @@ describe String do
     should respond_to(:unhexdump)
   end
 
-  describe "#depack" do
-    subject { 0x1337 }
+  describe "#unpack" do
+    subject { "\x34\x12\x00\x00hello\0" }
+
+    let(:data) { [0x1234, "hello"] }
 
-    let(:i386) do
-      OpenStruct.new(:endian => :little, :address_length => 4)
+    it "should unpack Strings using String#unpack template Strings" do
+      subject.unpack('VZ*').should == data
     end
 
-    let(:ppc) do
-      OpenStruct.new(:endian => :big, :address_length => 4)
+    it "should unpack Strings using a Binary::Template" do
+      subject.unpack(:uint32_le, :string).should == data
     end
+  end
 
-    let(:i386_packed_int)   { "7\023\000\000" }
-    let(:i386_packed_short) { "7\023" }
-    let(:i386_packed_long)  { "7\023\000\000" }
-    let(:i386_packed_quad)  { "7\023\000\000\000\000\000\000" }
+  context "deprecated" do
+    describe "#depack" do
+      subject { 0x1337 }
 
-    let(:ppc_packed_int)   { "\000\000\0237" }
-    let(:ppc_packed_short) { "\0237" }
-    let(:ppc_packed_long)  { "\000\000\0237" }
-    let(:ppc_packed_quad)  { "\000\000\000\000\000\000\0237" }
+      let(:i386) do
+        OpenStruct.new(:endian => :little, :address_length => 4)
+      end
 
-    it "should depack itself for a little-endian architecture" do
-      i386_packed_int.depack(i386).should == subject
-    end
+      let(:ppc) do
+        OpenStruct.new(:endian => :big, :address_length => 4)
+      end
 
-    it "should depack itself as a short for a little-endian architecture" do
-      i386_packed_short.depack(i386,2).should == subject
-    end
+      let(:i386_packed_int)   { "7\023\000\000" }
+      let(:i386_packed_short) { "7\023" }
+      let(:i386_packed_long)  { "7\023\000\000" }
+      let(:i386_packed_quad)  { "7\023\000\000\000\000\000\000" }
 
-    it "should depack itself as a long for a little-endian architecture" do
-      i386_packed_long.depack(i386,4).should == subject
-    end
+      let(:ppc_packed_int)   { "\000\000\0237" }
+      let(:ppc_packed_short) { "\0237" }
+      let(:ppc_packed_long)  { "\000\000\0237" }
+      let(:ppc_packed_quad)  { "\000\000\000\000\000\000\0237" }
 
-    it "should depack itself as a quad for a little-endian architecture" do
-      i386_packed_quad.depack(i386,8).should == subject
-    end
+      it "should depack itself for a little-endian architecture" do
+        i386_packed_int.depack(i386).should == subject
+      end
 
-    it "should depack itself for a big-endian architecture" do
-      ppc_packed_int.depack(ppc).should == subject
-    end
+      it "should depack itself as a short for a little-endian architecture" do
+        i386_packed_short.depack(i386,2).should == subject
+      end
 
-    it "should depack itself as a short for a big-endian architecture" do
-      ppc_packed_short.depack(ppc,2).should == subject
-    end
+      it "should depack itself as a long for a little-endian architecture" do
+        i386_packed_long.depack(i386,4).should == subject
+      end
 
-    it "should depack itself as a long for a big-endian architecture" do
-      ppc_packed_long.depack(ppc,4).should == subject
-    end
+      it "should depack itself as a quad for a little-endian architecture" do
+        i386_packed_quad.depack(i386,8).should == subject
+      end
 
-    it "should depack itself as a quad for a big-endian architecture" do
-      ppc_packed_quad.depack(ppc,8).should == subject
-    end
+      it "should depack itself for a big-endian architecture" do
+        ppc_packed_int.depack(ppc).should == subject
+      end
+
+      it "should depack itself as a short for a big-endian architecture" do
+        ppc_packed_short.depack(ppc,2).should == subject
+      end
+
+      it "should depack itself as a long for a big-endian architecture" do
+        ppc_packed_long.depack(ppc,4).should == subject
+      end
+
+      it "should depack itself as a quad for a big-endian architecture" do
+        ppc_packed_quad.depack(ppc,8).should == subject
+      end
 
-    it "should accept String#unpack template strings" do
-      i386_packed_long.depack('L').should == [subject]
+      it "should accept String#unpack template strings" do
+        i386_packed_long.depack('V').should == [subject]
+      end
     end
   end
 
@@ -131,24 +154,6 @@ describe String do
     end
   end
 
-  describe "#hex_unescape" do
-    it "should unescape a normal String" do
-      "hello".hex_unescape.should == "hello"
-    end
-
-    it "should unescape a hex String" do
-      "\\x68\\x65\\x6c\\x6c\\x6f\\x4e".hex_unescape.should == "hello\x4e"
-    end
-
-    it "should unescape an octal String" do
-      "hello\012".hex_unescape.should == "hello\n"
-    end
-
-    it "should unescape control characters" do
-      "hello\\n".hex_unescape.should == "hello\n"
-    end
-  end
-
   describe "#xor" do
     subject { 'hello' }
 
@@ -177,130 +182,12 @@ describe String do
   end
 
   describe "#unhexdump" do
-    include Helpers
-
-    context "GNU hexdump" do
-      let(:ascii) { load_binary_data('ascii') }
-      let(:repeated) { load_binary_data('repeated') }
-
-      it "should unhexdump octal-byte hexdump output" do
-        hexdump = load_hexdump('hexdump_octal_bytes')
-
-        hexdump.unhexdump(:format => :hexdump, :encoding => :octal_bytes).should == ascii
-      end
-
-      it "should unhexdump hex-byte hexdump output" do
-        hexdump = load_hexdump('hexdump_hex_bytes')
+    subject { "00000000  23 20 52 6f 6e 69 6e 20  53 75 70 70 6f 72 74 0a  |# Ronin Support.|\n00000010\n" }
 
-        hexdump.unhexdump(:format => :hexdump, :encoding => :hex_bytes).should == ascii
-      end
-
-      it "should unhexdump decimal-short hexdump output" do
-        hexdump = load_hexdump('hexdump_decimal_shorts')
-
-        hexdump.unhexdump(:format => :hexdump, :encoding => :decimal_shorts).should == ascii
-      end
-
-      it "should unhexdump octal-short hexdump output" do
-        hexdump = load_hexdump('hexdump_octal_shorts')
+    let(:raw) { "# Ronin Support\n" }
 
-        hexdump.unhexdump(:format => :hexdump, :encoding => :octal_shorts).should == ascii
-      end
-
-      it "should unhexdump hex-short hexdump output" do
-        hexdump = load_hexdump('hexdump_hex_shorts')
-
-        hexdump.unhexdump(:format => :hexdump, :encoding => :hex_shorts).should == ascii
-      end
-
-      it "should unhexdump repeated hexdump output" do
-        hexdump = load_hexdump('hexdump_repeated')
-
-        hexdump.unhexdump(:format => :hexdump, :encoding => :hex_bytes).should == repeated
-      end
-    end
-
-    context "od" do
-      let(:ascii) { load_binary_data('ascii') }
-      let(:repeated) { load_binary_data('repeated') }
-
-      it "should unhexdump octal-byte hexdump output" do
-        hexdump = load_hexdump('od_octal_bytes')
-
-        hexdump.unhexdump(:format => :od, :encoding => :octal_bytes).should == ascii
-      end
-
-      it "should unhexdump octal-shorts hexdump output" do
-        hexdump = load_hexdump('od_octal_shorts')
-
-        hexdump.unhexdump(:format => :od, :encoding => :octal_shorts).should == ascii
-      end
-
-      it "should unhexdump octal-ints hexdump output" do
-        hexdump = load_hexdump('od_octal_ints')
-
-        hexdump.unhexdump(:format => :od, :encoding => :octal_ints).should == ascii
-      end
-
-      it "should unhexdump octal-quads hexdump output" do
-        hexdump = load_hexdump('od_octal_quads')
-
-        hexdump.unhexdump(:format => :od, :encoding => :octal_quads).should == ascii
-      end
-
-      it "should unhexdump decimal-byte hexdump output" do
-        hexdump = load_hexdump('od_decimal_bytes')
-
-        hexdump.unhexdump(:format => :od, :encoding => :decimal_bytes).should == ascii
-      end
-
-      it "should unhexdump decimal-shorts hexdump output" do
-        hexdump = load_hexdump('od_decimal_shorts')
-
-        hexdump.unhexdump(:format => :od, :encoding => :decimal_shorts).should == ascii
-      end
-
-      it "should unhexdump decimal-ints hexdump output" do
-        hexdump = load_hexdump('od_decimal_ints')
-
-        hexdump.unhexdump(:format => :od, :encoding => :decimal_ints).should == ascii
-      end
-
-      it "should unhexdump decimal-quads hexdump output" do
-        hexdump = load_hexdump('od_decimal_quads')
-
-        hexdump.unhexdump(:format => :od, :encoding => :decimal_quads).should == ascii
-      end
-
-      it "should unhexdump hex-byte hexdump output" do
-        hexdump = load_hexdump('od_hex_bytes')
-
-        hexdump.unhexdump(:format => :od, :encoding => :hex_bytes).should == ascii
-      end
-
-      it "should unhexdump hex-shorts hexdump output" do
-        hexdump = load_hexdump('od_hex_shorts')
-
-        hexdump.unhexdump(:format => :od, :encoding => :hex_shorts).should == ascii
-      end
-
-      it "should unhexdump hex-ints hexdump output" do
-        hexdump = load_hexdump('od_hex_ints')
-
-        hexdump.unhexdump(:format => :od, :encoding => :hex_ints).should == ascii
-      end
-
-      it "should unhexdump hex-quads hexdump output" do
-        hexdump = load_hexdump('od_hex_quads')
-
-        hexdump.unhexdump(:format => :od, :encoding => :hex_quads).should == ascii
-      end
-
-      it "should unhexdump repeated hexdump output" do
-        hexdump = load_hexdump('od_repeated')
-
-        hexdump.unhexdump(:format => :od, :encoding => :octal_shorts).should == repeated
-      end
+    it "should unhexdump a String" do
+      subject.unhexdump.should == raw
     end
   end
 end