ronin-support 0.4.1 → 0.5.0.rc1

data/spec/formatting/text/string_spec.rb

@@ -24,6 +24,14 @@ describe String do
     should respond_to(:insert_after)
   end
 
+  it "should provide String#escape" do
+    should respond_to(:escape)
+  end
+
+  it "should provide String#unescape" do
+    should respond_to(:unescape)
+  end
+
   describe "#format_bytes" do
     it "should format each byte in the String" do
       subject.format_bytes { |b|
@@ -129,4 +137,26 @@ describe String do
       subject.insert_after(/x/,'x').should == subject
     end
   end
+
+  describe "#unescape" do
+    it "should not unescape a normal String" do
+      "hello".unescape.should == "hello"
+    end
+
+    it "should unescape a hex String" do
+      "\\x68\\x65\\x6c\\x6c\\x6f\\x4e".unescape.should == "hello\x4e"
+    end
+
+    it "should unescape an octal String" do
+      "hello\012".unescape.should == "hello\n"
+    end
+
+    it "should unescape control characters" do
+      "hello\\n".unescape.should == "hello\n"
+    end
+
+    it "should unescape normal characters" do
+      "hell\\o".unescape.should == "hello"
+    end
+  end
 end

data/spec/network/dns_spec.rb

@@ -78,6 +78,28 @@ describe Network::DNS do
       it "should accept an additional nameserver argument" do
         subject.dns_lookup(hostname,server).should == address
       end
+
+      context "when given a block" do
+        it "should yield the resolved address" do
+          resolved_address = nil
+
+          subject.dns_lookup(hostname) do |address|
+            resolved_address = address
+          end
+
+          resolved_address.should == address
+        end
+
+        it "should not yield unresolved addresses" do
+          resolved_address = nil
+
+          subject.dns_lookup(bad_hostname) do |address|
+            resolved_address = address
+          end
+
+          resolved_address.should be_nil
+        end
+      end
     end
 
     describe "#dns_lookup_all" do
@@ -96,6 +118,16 @@ describe Network::DNS do
       it "should accept an additional nameserver argument" do
         subject.dns_lookup_all(hostname,server).should include(address)
       end
+
+      context "when given a block" do
+        it "should yield the resolved address" do
+          subject.enum_for(:dns_lookup,hostname).to_a.should == [address]
+        end
+
+        it "should not yield unresolved addresses" do
+          subject.enum_for(:dns_lookup,bad_hostname).to_a.should == []
+        end
+      end
     end
 
     describe "#dns_reverse_lookup" do
@@ -114,6 +146,28 @@ describe Network::DNS do
       it "should accept an additional nameserver argument" do
         subject.dns_reverse_lookup(address,server).should == reverse_hostname
       end
+
+      context "when given a block" do
+        it "should yield the resolved hostname" do
+          resolved_hostname = nil
+
+          subject.dns_reverse_lookup(address) do |hostname|
+            resolved_hostname = hostname
+          end
+
+          resolved_hostname.should == reverse_hostname
+        end
+
+        it "should not yield unresolved hostnames" do
+          resolved_hostname = nil
+
+          subject.dns_reverse_lookup(bad_address) do |hostname|
+            resolved_hostname = hostname
+          end
+
+          resolved_hostname.should be_nil
+        end
+      end
     end
 
     describe "#dns_reverse_lookup_all" do
@@ -132,6 +186,16 @@ describe Network::DNS do
       it "should accept an additional nameserver argument" do
         subject.dns_reverse_lookup_all(address,server).should include(reverse_hostname)
       end
+
+      context "when given a block" do
+        it "should yield the resolved hostnames" do
+          subject.enum_for(:dns_reverse_lookup_all,address).to_a.should == [reverse_hostname]
+        end
+
+        it "should not yield unresolved hostnames" do
+          subject.enum_for(:dns_reverse_lookup_all,bad_address).to_a.should == []
+        end
+      end
     end
   end
 end

data/spec/network/ftp_spec.rb

@@ -0,0 +1,65 @@
+require 'spec_helper'
+require 'ronin/network/ftp'
+
+describe Network::FTP do
+  describe "helpers", :network do
+    subject do
+      obj = Object.new
+      obj.extend described_class
+      obj
+    end
+
+    let(:host) { 'ftp.kernel.org' }
+
+    describe "#ftp_connect" do
+      it "should return a Net::FTP object" do
+        ftp = subject.ftp_connect(host)
+
+        ftp.should be_kind_of(Net::FTP)
+        ftp.close
+      end
+
+      it "should connect to an FTP service" do
+        ftp = subject.ftp_connect(host)
+
+        ftp.should_not be_closed
+        ftp.close
+      end
+
+      context "when given a block" do
+        it "should yield the new Net::FTP object" do
+          ftp = subject.ftp_connect(host) do |ftp|
+            ftp.should be_kind_of(Net::FTP)
+          end
+
+          ftp.close
+        end
+      end
+    end
+
+    describe "#ftp_session" do
+      it "should yield a new Net::FTP object" do
+        yielded_ftp = nil
+
+        subject.ftp_session(host) do |ftp|
+          yielded_ftp = ftp
+        end
+
+        yielded_ftp.should be_kind_of(Net::FTP)
+      end
+
+      it "should close the FTP session after yielding it" do
+        session  = nil
+        was_open = nil
+
+        subject.ftp_session(host) do |ftp|
+          session   = ftp
+          was_open  = !ftp.closed?
+        end
+
+        was_open.should == true
+        session.should be_closed
+      end
+    end
+  end
+end

data/spec/network/proxy_spec.rb

@@ -0,0 +1,121 @@
+require 'spec_helper'
+require 'ronin/network/proxy'
+
+describe Network::Proxy do
+  let(:port)  { 1337              }
+  let(:host)  { '127.0.0.1'       }
+  let(:server_host) { 'www.example.com' }
+  let(:server_port) { 80                }
+
+  let(:proxy) do
+    described_class.new(
+      :port   => port,
+      :host   => host,
+      :server => [server_host, server_port]
+    )
+  end
+
+  subject { proxy }
+
+  describe "#initialize" do
+    it "should default host to '0.0.0.0'" do
+      proxy = described_class.new(
+        :port   => port,
+        :server => [server_host, server_port]
+      )
+
+      proxy.port.should == port
+      proxy.host.should == '0.0.0.0'
+    end
+
+    it "should allow setting both the host and port" do
+      proxy = described_class.new(
+        :port   => port,
+        :host   => host,
+        :server => [server_host, server_port]
+      )
+
+      proxy.port.should == port
+      proxy.host.should == host
+    end
+
+    it "should set the server_host and server_port" do
+      proxy = described_class.new(
+        :port   => port,
+        :host   => host,
+        :server => [server_host, server_port]
+      )
+
+      proxy.server_host.should == server_host
+      proxy.server_port.should == server_port
+    end
+  end
+
+  describe "#on_data" do
+    it "should call on_client_data" do
+      subject.should_receive(:on_client_data)
+
+      subject.on_data { |client,server,data| }
+    end
+
+    it "should call on_server_data" do
+      subject.should_receive(:on_server_data)
+
+      subject.on_data { |client,server,data| }
+    end
+  end
+
+  describe "actions" do
+    describe "#ignore!" do
+      it "should throw the :ignore action" do
+        lambda {
+          subject.ignore!
+        }.should throw_symbol(:action, :ignore)
+      end
+    end
+
+    describe "#close!" do
+      it "should throw the :close action" do
+        lambda {
+          subject.close!
+        }.should throw_symbol(:action, :close)
+      end
+    end
+
+    describe "#reset!" do
+      it "should throw the :reset action" do
+        lambda {
+          subject.reset!
+        }.should throw_symbol(:action, :reset)
+      end
+    end
+
+    describe "#stop!" do
+      it "should throw the :stop action" do
+        lambda {
+          subject.stop!
+        }.should throw_symbol(:action, :stop)
+      end
+    end
+  end
+
+  describe "#to_s" do
+    subject { proxy.to_s }
+
+    it "should include the proxy host and port" do
+      subject.should include("#{host}:#{port}")
+    end
+
+    it "should include the server host and port" do
+      subject.should include("#{server_host}:#{server_port}")
+    end
+  end
+
+  describe "#inspect" do
+    subject { proxy.inspect }
+
+    it "should include the output of #to_s" do
+      subject.should include(proxy.to_s)
+    end
+  end
+end

data/spec/network/shared/unix_server.rb

@@ -0,0 +1,31 @@
+require 'rspec'
+
+require 'tmpdir'
+require 'fileutils'
+require 'socket'
+
+shared_context "UNIX Server" do
+  def socket_path(name); File.join(Dir.tmpdir,name); end
+
+  let(:path) { socket_path('ronin_unix_server') }
+
+  before(:each) do
+    @server = UNIXServer.new(path)
+    @server_thread = Thread.new do
+      socket = @server.accept
+
+      begin
+        socket.puts socket.readline
+      ensure
+        socket.close
+      end
+    end
+  end
+
+  after(:each) do
+    @server_thread.kill
+    @server.close
+
+    FileUtils.rm(path)
+  end
+end

data/spec/network/tcp/proxy_spec.rb

@@ -0,0 +1,116 @@
+require 'spec_helper'
+require 'ronin/network/tcp/proxy'
+
+describe Network::TCP::Proxy, :network => true do
+  let(:port)   { 1337                    }
+  let(:host)   { 'localhost'             }
+  let(:server) { ['www.example.com', 80] }
+
+  before(:each) do
+    @proxy  = described_class.new(
+      :port   => port,
+      :host   => host,
+      :server => server
+    )
+    @thread = Thread.new { @proxy.start }
+
+    sleep 0.1
+  end
+
+  describe "#on_client_connect" do
+    let(:injection) { "Client connected\r\n" }
+
+    before do
+      @proxy.on_client_connect do |client|
+        client.write(injection)
+      end
+
+      @socket = TCPSocket.new(host,port)
+    end
+
+    it "should trigger when a new client connects" do
+      @socket.readline.should == injection
+    end
+
+    after { @socket.close }
+  end
+
+  describe "#on_server_connect" do
+    let(:injection) { "Server connected\r\n" }
+
+    before do
+      @proxy.on_server_connect do |client,server|
+        client.write(injection)
+      end
+
+      @socket = TCPSocket.new(host,port)
+    end
+
+    it "should trigger after a new client connects" do
+      @socket.readline.should == injection
+    end
+
+    after { @socket.close }
+  end
+
+  describe "#on_client_data" do
+    before do
+      @proxy.on_client_data do |client,server,data|
+        data.gsub!(/HTTP\/1.1/,'HTTP/1.0')
+      end
+
+      @socket = TCPSocket.new(host,port)
+    end
+
+    it "should trigger when the client sends data" do
+      @socket.write("GET / HTTP/1.1\r\n\r\n")
+
+      @socket.readline.should == "HTTP/1.0 302 Found\r\n"
+    end
+
+    after { @socket.close }
+  end
+
+  describe "#on_server_data" do
+    before do
+      @proxy.on_server_data do |client,server,data|
+        data.gsub!(/Connection: \S+/,'Connection: keep-alive')
+      end
+
+      @socket = TCPSocket.new(host,port)
+    end
+
+    it "should trigger when the server sends data" do
+      @socket.write("GET / HTTP/1.0\r\n\r\n")
+
+      @socket.read.should include("Connection: keep-alive\r\n")
+    end
+
+    after { @socket.close }
+  end
+
+  describe "#on_server_disconnect" do
+    let(:injection) { "Haha Internet!\r\n" }
+
+    before do
+      @proxy.on_server_disconnect do |client,server|
+        client.write(injection)
+      end
+
+      @socket = TCPSocket.new(host,port)
+    end
+
+    it "should trigger when the server closes the connection" do
+      @socket.write("GET / HTTP/1.0\r\n\r\n")
+
+      @socket.read.end_with?(injection).should be_true
+    end
+
+    after { @socket.close }
+  end
+
+  after(:each) do
+    @thread.kill
+    @proxy.close
+  end
+end