raptor-io 0.0.1

data/spec/raptor-io/protocol/http/headers_spec.rb

@@ -0,0 +1,189 @@
+require 'spec_helper'
+
+describe RaptorIO::Protocol::HTTP::Headers do
+
+  describe '#to_s' do
+    it 'supports multiple headers' do
+      headers = described_class.new( 'X-Stuff' => %w(1 2 3) )
+      headers.to_s.should == "X-Stuff: 1\r\nX-Stuff: 2\r\nX-Stuff: 3"
+    end
+
+    it 'formats headers for HTTP transmission' do
+      options = {
+        'x-morE-stUfF' => 'blah'
+      }
+      described_class.new( options ).to_s.should ==
+              "X-More-Stuff: blah"
+    end
+  end
+
+  describe '#delete' do
+    it 'deleted a header field' do
+      h = described_class.new( 'x-my-field' => 'stuff' )
+      h.delete( 'X-My-Field' ).should == 'stuff'
+    end
+  end
+
+  describe '#include?' do
+    context 'when the field is included' do
+      it 'returns true' do
+        h = described_class.new( 'X-My-Field' => 'stuff' )
+        h.include?( 'x-my-field' ).should be_true
+      end
+    end
+    context 'when the field is not included' do
+      it 'returns false' do
+        described_class.new.include?( 'x-my-field' ).should be_false
+      end
+    end
+  end
+
+  describe '#set_cookie' do
+    context 'when there are no set-cookie fields' do
+      it 'returns an empty array' do
+        described_class.new.cookies.should == []
+      end
+    end
+
+    it 'returns an array of set-cookie strings' do
+      set_coookies = [
+          'name=value; Expires=Wed, 09 Jun 2020 10:18:14 GMT',
+          'name2=value2; Expires=Wed, 09 Jun 2021 10:18:14 GMT'
+      ]
+
+      described_class.new( 'Set-Cookie' => set_coookies ).set_cookie.should == set_coookies
+    end
+  end
+
+  describe '#parsed_set_cookie' do
+    context 'when there are no Set-cookie fields' do
+      it 'returns an empty array' do
+        described_class.new.parsed_set_cookie.should == []
+      end
+    end
+
+    it 'returns an array of cookies as hashes' do
+      described_class.new(
+          'Set-Cookie' => [
+              'name=value; Expires=Wed, 09 Jun 2020 10:18:14 GMT',
+              'name2=value2; Expires=Wed, 09 Jun 2021 10:18:14 GMT'
+          ]
+      ).parsed_set_cookie.should == [
+          {
+              name:         'name',
+              value:        'value',
+              version:      0,
+              port:         nil,
+              discard:      nil,
+              comment_url:  nil,
+              expires:      Time.parse( '2020-06-09 13:18:14 +0300' ),
+              max_age:      nil,
+              comment:      nil,
+              secure:       nil,
+              path:         nil,
+              domain:       nil
+          },
+          {
+              name:         'name2',
+              value:        'value2',
+              version:      0,
+              port:         nil,
+              discard:      nil,
+              comment_url:  nil,
+              expires:      Time.parse( '2021-06-09 13:18:14 +0300' ),
+              max_age:      nil,
+              comment:      nil,
+              secure:       nil,
+              path:         nil,
+              domain:       nil
+          }
+      ]
+    end
+  end
+
+  describe '#cookies' do
+    context 'when there is no Cookie fied' do
+      it 'returns an empty array' do
+        described_class.new.cookies.should == []
+      end
+    end
+
+    it 'returns an array of cookies as hashes' do
+      described_class.new(
+          'Cookie' => 'cname=cvalue; c2name=c2value'
+      ).cookies.should == [
+          {
+              name: 'cname',
+              value: 'cvalue',
+              version: 0,
+              port: nil,
+              discard: nil,
+              comment_url: nil,
+              expires: nil,
+              max_age: nil,
+              comment: nil,
+              secure: nil,
+              path: nil,
+              domain: nil
+          },
+          {
+              name: 'c2name',
+              value: 'c2value',
+              version: 0,
+              port: nil,
+              discard: nil,
+              comment_url: nil,
+              expires: nil,
+              max_age: nil,
+              comment: nil,
+              secure: nil,
+              path: nil,
+              domain: nil
+          }
+      ]
+    end
+  end
+
+  describe '.parse' do
+    context 'when passed an empty string' do
+      it 'returns empty Headers' do
+        described_class.parse( '' ).should be_empty
+      end
+    end
+
+    it 'supports multiple headers' do
+      headers_string = "X-Stuff: 1\r\nX-Stuff: 2\r\n\r\n"
+
+      headers = described_class.parse( headers_string )
+      headers['x-stuff'].should == %w(1 2)
+      headers.class.should == described_class
+    end
+
+    it 'supports CRLF terminators' do
+      headers_string = "content-Type: text/html;charset=utf-8\r\n" +
+        "Content-length: 431\r\n\r\n"
+
+      headers = described_class.parse( headers_string )
+      headers.should ==
+          {
+              'Content-Type'      => 'text/html;charset=utf-8',
+              'Content-Length'    => '431'
+          }
+      headers.class.should == described_class
+    end
+
+    it 'supports CR terminators' do
+      headers_string = "content-Type: text/html;charset=utf-8\n" +
+          "Content-length: 431\n\n"
+
+      headers = described_class.parse( headers_string )
+      headers.should ==
+          {
+              'Content-Type'      => 'text/html;charset=utf-8',
+              'Content-Length'    => '431'
+          }
+      headers.class.should == described_class
+    end
+  end
+
+end

data/spec/raptor-io/protocol/http/message_spec.rb

@@ -0,0 +1,5 @@
+require 'spec_helper'
+
+describe RaptorIO::Protocol::HTTP::Message do
+  it_should_behave_like 'RaptorIO::Protocol::HTTP::Message'
+end

data/spec/raptor-io/protocol/http/request/manipulators/authenticator_spec.rb

@@ -0,0 +1,193 @@
+require 'spec_helper'
+
+describe 'RaptorIO::Protocol::HTTP::Request::Manipulators::Authenticator' do
+  before :all do
+    WebServers.start :basic
+    @basic_url = WebServers.url_for( :basic )
+
+    WebServers.start :digest
+    @digest_url = WebServers.url_for( :digest )
+
+    @iis_address   = ENV['IIS']
+    @ntlm_url      = "http://#{@iis_address}/ntlm/"
+    @negotiate_url = "http://#{@iis_address}/negotiate/"
+  end
+
+  before( :each ) do
+    RaptorIO::Protocol::HTTP::Request::Manipulators.reset
+    RaptorIO::Protocol::HTTP::Client.reset
+  end
+
+
+  let(:client) do
+    RaptorIO::Protocol::HTTP::Client.new(
+        switch_board: RaptorIO::Socket::SwitchBoard.new,
+        manipulators: {
+            'authenticator' =>
+                {
+                    username: 'admin',
+                    password: 'secret'
+                }
+        }
+    )
+  end
+
+  context 'when authentication is of type' do
+    context 'Basic' do
+      it 'provides Basic authentication' do
+        opts = { mode: :sync }
+        2.times do
+          client.get( @basic_url, opts ).code.should == 200
+        end
+      end
+    end
+
+    context 'Digest' do
+      it 'provides Digest authentication' do
+        opts = { mode: :sync }
+
+        2.times do
+          client.get( @digest_url, opts ).code.should == 200
+        end
+      end
+
+      it 'doesn\'t run any queued request until the auth finishes' do
+        cnt = 0
+
+        50.times do |i|
+          client.get( @digest_url ) do |response|
+            response.code.should == 200
+            cnt += 1
+          end
+        end
+
+        client.run
+        client.datastore['authenticator'][:tries].should == 1
+        cnt.should == 50
+      end
+    end
+
+    context 'Negotiate' do
+      let(:client) do
+        RaptorIO::Protocol::HTTP::Client.new(
+            manipulators: {
+                'authenticator' =>
+                    {
+                        username: 'msfadmin',
+                        password: 'msfadmin'
+                    }
+            }
+        )
+      end
+
+      it 'provides Negotiate authentication' do
+        pending if !ENV['IIS']
+
+        opts = { mode: :sync }
+
+        2.times do
+          client.get( @negotiate_url, opts ).code.should == 200
+        end
+      end
+
+      it 'doesn\'t run any queued request until the auth finishes' do
+        pending if !ENV['IIS']
+
+        cnt = 0
+
+        50.times do |i|
+          client.get( @negotiate_url ) do |response|
+            response.code.should == 200
+            cnt += 1
+          end
+        end
+
+        client.run
+        cnt.should == 50
+      end
+
+      context 'on wrong credentials' do
+        it 'returns a 401' do
+          pending if !ENV['IIS']
+
+          opts = {
+              mode: :sync, manipulators: {
+                  'authenticator' =>
+                      {
+                          username: 'blah',
+                          password: 'blah'
+                      }
+              }
+          }
+
+          2.times do
+            client.get( @negotiate_url, opts ).code.should == 401
+          end
+        end
+      end
+    end
+
+    context 'NTLM' do
+      let(:client) do
+        RaptorIO::Protocol::HTTP::Client.new(
+            manipulators: {
+                'authenticator' =>
+                    {
+                        username: 'msfadmin',
+                        password: 'msfadmin'
+                    }
+            }
+        )
+      end
+
+      it 'provides NTLM authentication' do
+        pending if !ENV['IIS']
+
+        opts = { mode: :sync }
+
+        2.times do
+          client.get( @ntlm_url, opts ).code.should == 200
+        end
+      end
+
+      it 'doesn\'t run any queued request until the auth finishes' do
+        pending if !ENV['IIS']
+
+        cnt = 0
+
+        50.times do |i|
+          client.get( @ntlm_url ) do |response|
+            response.code.should == 200
+            cnt += 1
+          end
+        end
+
+        client.run
+        cnt.should == 50
+      end
+
+      context 'on wrong credentials' do
+        it 'returns a 401' do
+          pending if !ENV['IIS']
+
+          opts = {
+              mode: :sync, manipulators: {
+                  'authenticator' =>
+                      {
+                          username: 'blah',
+                          password: 'blah'
+                      }
+              }
+          }
+
+          2.times do
+            client.get( @ntlm_url, opts ).code.should == 401
+          end
+        end
+      end
+    end
+
+  end
+
+end
+

data/spec/raptor-io/protocol/http/request/manipulators/authenticators/basic_spec.rb

@@ -0,0 +1,32 @@
+require 'spec_helper'
+require 'raptor-io/socket'
+
+describe 'RaptorIO::Protocol::HTTP::Request::Manipulators::Authenticators::Basic' do
+  before :all do
+    WebServers.start :basic
+    @url = WebServers.url_for( :basic )
+  end
+
+  before( :each ) do
+    RaptorIO::Protocol::HTTP::Request::Manipulators.reset
+  end
+
+  let(:client) { RaptorIO::Protocol::HTTP::Client.new(switch_board:RaptorIO::Socket::SwitchBoard.new) }
+
+  it 'provides Basic authentication' do
+    opts = {
+        mode: :sync, manipulators: {
+            'authenticators/basic' =>
+                {
+                    username: 'admin',
+                    password: 'secret'
+                }
+        }
+    }
+
+    2.times do
+      client.get( @url, opts ).code.should == 200
+    end
+  end
+end
+

data/spec/raptor-io/protocol/http/request/manipulators/authenticators/digest_spec.rb

@@ -0,0 +1,76 @@
+require 'spec_helper'
+require 'raptor-io/socket'
+
+describe 'RaptorIO::Protocol::HTTP::Request::Manipulators::Authenticators::Digest' do
+  before :all do
+    WebServers.start :digest
+    @url = WebServers.url_for( :digest )
+  end
+
+  let( :manipulators ) { RaptorIO::Protocol::HTTP::Request::Manipulators }
+  before( :each ) do
+    RaptorIO::Protocol::HTTP::Request::Manipulators.reset
+  end
+
+  let(:client) do
+    sb = RaptorIO::Socket::SwitchBoard.new
+    RaptorIO::Protocol::HTTP::Client.new(switch_board: sb)
+  end
+
+  def response( algo )
+    RaptorIO::Protocol::HTTP::Response.parse "HTTP/1.1 401 Unauthorized
+Content-Type: text/plain
+Content-Length: 0
+Www-Authenticate: Digest realm=\"Protected Area\", algorithm=\"#{algo}\", nonce=\"MTM3MzI5OTYxNiAxYzk2ZDM4OWY1MTY2ZGM3ODllNGQ2N2RjZDIyYzk1ZA==\", opaque=\"610a2ee688cda9e724885e23cd2cfdee\", qop=\"auth\"
+Connection: keep-alive
+Server: thin 1.5.1 codename Straight Razor\r\n\r\n"
+  end
+
+  it 'provides Digest authentication' do
+    opts = {
+        mode: :sync, manipulators: {
+            'authenticators/digest' =>
+                {
+                    response: client.get( @url, mode: :sync ),
+                    username: 'admin',
+                    password: 'secret'
+                }
+        }
+    }
+
+    2.times do
+      client.get( @url, opts ).code.should == 200
+    end
+  end
+
+  %w(MD5 SHA1 SHA2 SHA256 SHA384 SHA512 RMD160).each do |algo|
+    it "supports #{algo}" do
+      manipulators.process(
+          'authenticators/digest',
+          client,
+          RaptorIO::Protocol::HTTP::Request.new( url: @url ),
+          {
+              response: response( algo ),
+              username: 'admin',
+              password: 'secret'
+          }
+      )
+    end
+  end
+
+  it 'raises error on unknown algorithm' do
+    expect do
+      manipulators.process(
+          'authenticators/digest',
+          client,
+          RaptorIO::Protocol::HTTP::Request.new( url: @url ),
+          {
+              response: response( 'stuff' ),
+              username: 'admin',
+              password: 'secret'
+          }
+      )
+    end.to raise_error RaptorIO::Protocol::HTTP::Error
+  end
+end
+