ronin-support 0.5.2 → 1.0.0.beta1

data/spec/network/proxy_spec.rb

@@ -1,121 +0,0 @@
-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]
-      )
-
-      expect(proxy.port).to eq(port)
-      expect(proxy.host).to eq('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]
-      )
-
-      expect(proxy.port).to eq(port)
-      expect(proxy.host).to eq(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]
-      )
-
-      expect(proxy.server_host).to eq(server_host)
-      expect(proxy.server_port).to eq(server_port)
-    end
-  end
-
-  describe "#on_data" do
-    it "should call on_client_data" do
-      expect(subject).to receive(:on_client_data)
-
-      subject.on_data { |client,server,data| }
-    end
-
-    it "should call on_server_data" do
-      expect(subject).to 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
-        expect {
-          subject.ignore!
-        }.to throw_symbol(:action, :ignore)
-      end
-    end
-
-    describe "#close!" do
-      it "should throw the :close action" do
-        expect {
-          subject.close!
-        }.to throw_symbol(:action, :close)
-      end
-    end
-
-    describe "#reset!" do
-      it "should throw the :reset action" do
-        expect {
-          subject.reset!
-        }.to throw_symbol(:action, :reset)
-      end
-    end
-
-    describe "#stop!" do
-      it "should throw the :stop action" do
-        expect {
-          subject.stop!
-        }.to throw_symbol(:action, :stop)
-      end
-    end
-  end
-
-  describe "#to_s" do
-    subject { proxy.to_s }
-
-    it "should include the proxy host and port" do
-      expect(subject).to include("#{host}:#{port}")
-    end
-
-    it "should include the server host and port" do
-      expect(subject).to include("#{server_host}:#{server_port}")
-    end
-  end
-
-  describe "#inspect" do
-    subject { proxy.inspect }
-
-    it "should include the output of #to_s" do
-      expect(subject).to include(proxy.to_s)
-    end
-  end
-end

data/spec/network/shared/unix_server.rb

@@ -1,31 +0,0 @@
-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/smtp/email_spec.rb

@@ -1,100 +0,0 @@
-require 'spec_helper'
-require 'ronin/network/smtp/email'
-
-require 'date'
-
-describe Network::SMTP::Email do
-  describe "#initialize" do
-    it "should default 'date' to Time.now" do
-      email = Network::SMTP::Email.new
-
-      expect(email.date).not_to be_nil
-    end
-
-    it "should accept a String body" do
-      body = 'hello'
-      email = Network::SMTP::Email.new(:body => body)
-
-      expect(email.body).to eq([body])
-    end
-
-    it "should accept an Array body" do
-      body = ['hello', 'world']
-      email = Network::SMTP::Email.new(:body => body)
-
-      expect(email.body).to eq(body)
-    end
-
-    it "should default 'body' to an empty Array" do
-      email = Network::SMTP::Email.new
-
-      expect(email.body).to be_empty
-    end
-  end
-
-  describe "#to_s" do
-    subject { Network::SMTP::Email.new }
-
-    it "should add the 'from'" do
-      subject.from = 'joe@example.com'
-
-      expect(subject.to_s).to include("From: joe@example.com\n\r")
-    end
-
-    context "when formatting 'to'" do
-      it "should accept an Array of addresses" do
-        subject.to = ['alice@example.com', 'joe@example.com']
-
-        expect(subject.to_s).to include("To: alice@example.com, joe@example.com\n\r")
-      end
-
-      it "should accept a String" do
-        subject.to = 'joe@example.com'
-
-        expect(subject.to_s).to include("To: joe@example.com\n\r")
-      end
-    end
-
-    it "should add the 'subject'" do
-      subject.subject = 'Hello'
-
-      expect(subject.to_s).to include("Subject: Hello\n\r")
-    end
-
-    it "should add the 'date'" do
-      subject.date = Date.parse('Sun Apr 24 17:22:55 PDT 2011')
-
-      expect(subject.to_s).to include("Date: #{subject.date}\n\r")
-    end
-
-    it "should add the 'message_id'" do
-      subject.message_id = '1234'
-
-      expect(subject.to_s).to include("Message-Id: <#{subject.message_id}>\n\r")
-    end
-
-    it "should add additional headers" do
-      subject.headers['X-Foo'] = 'Bar'
-      subject.headers['X-Baz'] = 'Quix'
-
-      lines = subject.to_s.split("\n\r")
-      
-      expect(lines).to include('X-Foo: Bar')
-      expect(lines).to include('X-Baz: Quix')
-    end
-
-    context "when formatting 'body'" do
-      it "should append each line with a CRLF" do
-        subject.body = ['hello', 'world']
-
-        expect(subject.to_s).to include("hello\n\rworld")
-      end
-
-      it "should add a CRLF before the body" do
-        subject.body = ['hello', 'world']
-
-        expect(subject.to_s).to include("\n\rhello")
-      end
-    end
-  end
-end

data/spec/network/ssl_spec.rb

@@ -1,70 +0,0 @@
-require 'spec_helper'
-require 'ronin/network/ssl'
-
-describe Network::SSL do
-  describe 'VERIFY' do
-    subject { Network::SSL::VERIFY }
-
-    it "should map verify mode names to OpenSSL VERIFY_* constants" do
-      expect(subject[:peer]).to eq(OpenSSL::SSL::VERIFY_PEER)
-    end
-
-    it "should default to VERIFY_NONE if no verify mode name is given" do
-      expect(subject[nil]).to eq(OpenSSL::SSL::VERIFY_NONE)
-    end
-
-    it "should raise an exception for unknown verify modes" do
-      expect { subject[:foo_bar] }.to raise_error
-    end
-  end
-
-  describe "helpers", :network do
-    let(:host) { 'github.com' }
-    let(:port) { 443 }
-
-    subject do
-      obj = Object.new
-      obj.extend described_class
-      obj
-    end
-
-    describe "#ssl_connect" do
-      it "should connect to an SSL protected port" do
-        expect {
-          subject.ssl_connect(host,port)
-        }.not_to raise_error
-      end
-
-      it "should return an OpenSSL::SSL::SSLSocket" do
-        socket = subject.ssl_connect(host,port)
-
-        expect(socket).to be_kind_of(OpenSSL::SSL::SSLSocket)
-      end
-
-      context "when a block is given" do
-        it "should yield the OpenSSL::SSL::SSLSocket" do
-          socket = nil
-
-          subject.ssl_connect(host,port) do |yielded_socket|
-            socket = yielded_socket
-          end
-
-          expect(socket).to be_kind_of(OpenSSL::SSL::SSLSocket)
-        end
-      end
-    end
-
-    describe "#ssl_session" do
-      it "should open then close a OpenSSL::SSL::SSLSocket" do
-        socket = nil
-
-        subject.ssl_session(host,port) do |yielded_socket|
-          socket = yielded_socket
-        end
-
-        expect(socket).to be_kind_of(OpenSSL::SSL::SSLSocket)
-        expect(socket).to be_closed
-      end
-    end
-  end
-end

data/spec/network/tcp/proxy_spec.rb

@@ -1,118 +0,0 @@
-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
-      expect(@socket.readline).to eq(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
-      expect(@socket.readline).to eq(injection)
-    end
-
-    after { @socket.close }
-  end
-
-  describe "#on_client_data" do
-    before do
-      @proxy.on_client_data do |client,server,data|
-        if (match = data.match(/HTTP\/(\d\.\d)/))
-          client.write("HTTP #{match[1]} detected!\r\n")
-        end
-      end
-
-      @socket = TCPSocket.new(host,port)
-    end
-
-    it "should trigger when the client sends data" do
-      @socket.write("GET /placeholder HTTP/1.0\r\n\r\n")
-
-      expect(@socket.readline).to eq("HTTP 1.0 detected!\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")
-
-      expect(@socket.read).to 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")
-
-      expect(@socket.read.end_with?(injection)).to be(true)
-    end
-
-    after { @socket.close }
-  end
-
-  after(:each) do
-    @thread.kill
-    @proxy.close
-  end
-end