require 'rex/sslscan/result'

describe Rex::SSLScan::Result do

	subject{Rex::SSLScan::Result.new}

      it { should respond_to :accepted }
      it { should respond_to :cert }
      it { should respond_to :ciphers }
      it { should respond_to :rejected }
      it { should respond_to :sslv2 }
      it { should respond_to :sslv3 }
      it { should respond_to :standards_compliant? }
      it { should respond_to :strong_ciphers }
      it { should respond_to :supports_ssl? }
      it { should respond_to :supports_sslv2? }
      it { should respond_to :supports_sslv3? }
      it { should respond_to :supports_tlsv1? }
      it { should respond_to :supports_weak_ciphers? }
      it { should respond_to :tlsv1 }
      it { should respond_to :weak_ciphers }

	context "with no values set" do
		it "should return nil for the cert" do
			subject.cert.should == nil
		end

		it "should return an empty set for ciphers" do
			subject.ciphers.should be_empty
		end

		it "should return an empty array for accepted" do
			subject.accepted.should == []
		end

		it "should return an empty array for rejected" do
			subject.rejected.should == []
		end

		it "should return an empty array for #sslv2" do
			subject.sslv2.should == []
		end

		it "should return an empty array for #sslv3" do
			subject.sslv3.should == []
		end

		it "should return an empty array for #tlsv1" do
			subject.tlsv1.should == []
		end

		it "should return an empty array for #weak_ciphers" do
			subject.weak_ciphers.should == []
		end

		it "should return an empty array for #strong_ciphers" do
			subject.strong_ciphers.should == []
		end

		it "should return false for #supports_ssl?" do
			subject.supports_ssl?.should == false
		end

		it "should return false for #supports_ssl?v2" do
			subject.supports_sslv2?.should == false
		end

		it "should return false for #supports_sslv3?" do
			subject.supports_sslv3?.should == false
		end

		it "should return false for #supports_tlsv1?" do
			subject.supports_tlsv1?.should == false
		end

		it "should return false for #supports_weak_ciphers?" do
			subject.supports_weak_ciphers?.should == false
		end

		it "should return true for #standards_compliant?" do
			subject.standards_compliant?.should == true
		end
	end

	context "setting the cert" do
		it "should accept nil" do
			subject.cert = nil
			subject.cert.should == nil
		end

		it "should accept an X509 cert" do
			cert = OpenSSL::X509::Certificate.new
			subject.cert = cert
			subject.cert.should == cert
		end

		it "should raise an exception for anything else" do
			expect{subject.cert = "foo"}.to raise_error
		end
	end

	context "adding a cipher result" do
		context "should raise an exception if" do
			it "given an invalid SSL version" do
				expect{subject.add_cipher(:ssl3, 'AES256-SHA', 256, :accepted )}.to raise_error
			end

			it "given SSL version as a string" do
				expect{subject.add_cipher('sslv3', 'AES256-SHA', 256, :accepted )}.to raise_error
			end

			it "given an invalid SSL cipher" do
				expect{subject.add_cipher(:SSLv3, 'FOO256-SHA', 256, :accepted )}.to raise_error
			end

			it "given an unsupported cipher for the version" do
				expect{subject.add_cipher(:SSLv3, 'DES-CBC3-MD5', 256, :accepted )}.to raise_error
			end

			it "given a non-number for key length" do
				expect{subject.add_cipher(:SSLv3, 'AES256-SHA', "256", :accepted )}.to raise_error
			end

			it "given a decimal key length" do
				expect{subject.add_cipher(:SSLv3, 'AES256-SHA', 25.6, :accepted )}.to raise_error
			end

			it "given an invalid status" do
				expect{subject.add_cipher(:SSLv3, 'AES256-SHA', 256, :good )}.to raise_error
			end

			it "given status as a string" do
				expect{subject.add_cipher(:SSLv3, 'AES256-SHA', 256, "accepted" )}.to raise_error
			end
		end
		context "that was accepted" do
			it "should add an SSLv2 cipher result to the SSLv2 Accepted array or generate an SSLv2 exception" do
				begin
					subject.add_cipher(:SSLv2, "DES-CBC3-MD5", 168, :accepted)
					subject.accepted(:SSLv2).should include({
						:version => :SSLv2,
						:cipher=>"DES-CBC3-MD5",
						:key_length=>168,
						:weak=> false,
						:status => :accepted})
				rescue ArgumentError => e
					e.message.should == "unknown SSL method `SSLv2'."
				end
			end

			it "should add an SSLv3 cipher result to the SSLv3 Accepted array" do
				subject.add_cipher(:SSLv3, "AES256-SHA", 256, :accepted)
				subject.accepted(:SSLv3).should include({
					:version => :SSLv3,
					:cipher=>"AES256-SHA",
					:key_length=>256,
					:weak=> false,
					:status => :accepted})
			end

			it "should add an TLSv1 cipher result to the TLSv1 Accepted array" do
				subject.add_cipher(:TLSv1, "AES256-SHA", 256, :accepted)
				subject.accepted(:TLSv1).should include({
					:version => :TLSv1,
					:cipher=>"AES256-SHA",
					:key_length=>256,
					:weak=> false,
					:status => :accepted})
			end

			it "should successfully add multiple entries in a row" do
				subject.add_cipher(:SSLv3, "AES128-SHA", 128, :accepted)
				subject.add_cipher(:SSLv3, "AES256-SHA", 256, :accepted)
				subject.accepted(:SSLv3).should include({
					:version => :SSLv3,
					:cipher=>"AES256-SHA",
					:key_length=>256,
					:weak=> false,
					:status => :accepted})
				subject.accepted(:SSLv3).should include({
					:version => :SSLv3,
					:cipher=>"AES256-SHA",
					:key_length=>256,
					:weak=> false,
					:status => :accepted})
			end

			it "should not add duplicate entries" do
				subject.add_cipher(:SSLv3, "AES128-SHA", 128, :accepted)
				subject.add_cipher(:SSLv3, "AES128-SHA", 128, :accepted)
				subject.accepted(:SSLv3).count.should == 1
			end
		end
		context "that was rejected" do
			it "should add an SSLv2 cipher result to the SSLv2 Rejected array or generate an SSLv2 exception" do
				begin
					subject.add_cipher(:SSLv2, "DES-CBC3-MD5", 168, :rejected)
					subject.rejected(:SSLv2).should include({
						:version => :SSLv2,
						:cipher=>"DES-CBC3-MD5",
						:key_length=>168,
						:weak=> false,
						:status => :rejected})
				rescue ArgumentError => e
					e.message.should == "unknown SSL method `SSLv2'."
				end
			end

			it "should add an SSLv3 cipher result to the SSLv3 Rejected array" do
				subject.add_cipher(:SSLv3, "AES256-SHA", 256, :rejected)
				subject.rejected(:SSLv3).should include({
					:version => :SSLv3,
					:cipher=>"AES256-SHA",
					:key_length=>256,
					:weak=> false,
					:status => :rejected})
			end

			it "should add an TLSv1 cipher result to the TLSv1 Rejected array" do
				subject.add_cipher(:TLSv1, "AES256-SHA", 256, :rejected)
				subject.rejected(:TLSv1).should include({
					:version => :TLSv1,
					:cipher=>"AES256-SHA",
					:key_length=>256,
					:weak=> false,
					:status => :rejected})
			end

			it "should successfully add multiple entries in a row" do
				subject.add_cipher(:SSLv3, "AES128-SHA", 128, :rejected)
				subject.add_cipher(:SSLv3, "AES256-SHA", 256, :rejected)
				subject.rejected(:SSLv3).should include({
					:version => :SSLv3,
					:cipher=>"AES256-SHA",
					:key_length=>256,
					:weak=> false,
					:status => :rejected})
				subject.rejected(:SSLv3).should include({
					:version => :SSLv3,
					:cipher=>"AES128-SHA",
					:key_length=>128,
					:weak=> false,
					:status => :rejected})
			end

			it "should not add duplicate entries" do
				subject.add_cipher(:SSLv3, "AES128-SHA", 128, :rejected)
				subject.add_cipher(:SSLv3, "AES128-SHA", 128, :rejected)
				subject.rejected(:SSLv3).count.should == 1
			end
		end
	end

	context "enumerating all accepted ciphers" do
		before(:each) do
			subject.add_cipher(:SSLv3, "AES256-SHA", 256, :accepted)
			subject.add_cipher(:TLSv1, "AES256-SHA", 256, :accepted)
			subject.add_cipher(:SSLv3, "AES128-SHA", 128, :accepted)
		end

		context "with no version selected" do
			it "should return an array of cipher detail hashes" do
				subject.each_accepted do |cipher_details|
					cipher_details.should include(:version, :cipher, :key_length, :status, :weak)
				end
			end

			it "should return all of the accepted cipher details" do
				count = 0
				subject.each_accepted do |cipher_details|
					count = count+1
				end
				count.should == 3
			end
		end

		context "when specifying one SSL version" do
			it "should raise an exception if not given a symbol" do
				expect{ subject.each_accepted('sslv2')}.to raise_error
			end

			it "should raise an exception if given an invalid SSL version" do
				expect{ subject.each_accepted(:TLSv3)}.to raise_error
			end

			it "should return only ciphers matching the version" do
				subject.each_accepted(:SSLv3) do |cipher_details|
					cipher_details[:version].should == :SSLv3
				end
			end
		end

		context "when specifying multiple SSL Versions in an array" do
			it "should return all versions if no valid versions were supplied" do
				count = 0
				subject.each_accepted([:TLSv3, :TLSv4]) do |cipher_details|
					count = count+1
				end
				count.should == 3
			end

			it "should return only the ciphers for the specified version" do
				subject.each_accepted([:SSLv3,:TLSv1]) do |cipher_details|
					cipher_details[:version].should_not == :SSLv2
				end
			end
		end
	end

	context "enumerating all rejected ciphers" do
		before(:each) do
			subject.add_cipher(:SSLv3, "AES256-SHA", 256, :rejected)
			subject.add_cipher(:TLSv1, "AES256-SHA", 256, :rejected)
			subject.add_cipher(:SSLv3, "AES128-SHA", 128, :rejected)
		end

		context "with no version selected" do
			it "should return an array of cipher detail hashes" do
				subject.each_rejected do |cipher_details|
					cipher_details.should include(:version, :cipher, :key_length, :status, :weak)
				end
			end

			it "should return all of the rejected cipher details" do
				count = 0
				subject.each_rejected do |cipher_details|
					count = count+1
				end
				count.should == 3
			end
		end

		context "when specifying one SSL version" do
			it "should raise an exception if not given a symbol" do
				expect{ subject.each_rejected('sslv2')}.to raise_error
			end

			it "should raise an exception if given an invalid SSL version" do
				expect{ subject.each_rejected(:TLSv3)}.to raise_error
			end

			it "should return only ciphers matching the version" do
				subject.each_rejected(:SSLv3) do |cipher_details|
					cipher_details[:version].should == :SSLv3
				end
			end
		end

		context "when specifying multiple SSL Versions in an array" do
			it "should return all versions if no valid versions were supplied" do
				count = 0
				subject.each_rejected([:TLSv3, :TLSv4]) do |cipher_details|
					count = count+1
				end
				count.should == 3
			end

			it "should return only the ciphers for the specified version" do
				subject.each_rejected([:SSLv3,:TLSv1]) do |cipher_details|
					cipher_details[:version].should_not == :SSLv2
				end
			end
		end
	end

	context "checking SSL support" do
		context "for SSLv2" do
			it "should return false if there are no accepted ciphers" do
				subject.supports_sslv2?.should == false
			end
			it "should return true if there are accepted ciphers or raise an SSLv2 exception" do
				begin
					subject.add_cipher(:SSLv2, "DES-CBC3-MD5", 168, :accepted)
					subject.supports_sslv2?.should == true
				rescue ArgumentError => e
					e.message.should == "unknown SSL method `SSLv2'."
				end
			end
		end
		context "for SSLv3" do
			it "should return false if there are no accepted ciphers" do
				subject.supports_sslv3?.should == false
			end
			it "should return true if there are accepted ciphers" do
				subject.add_cipher(:SSLv3, "AES256-SHA", 256, :accepted)
				subject.supports_sslv3?.should == true
			end
		end
		context "for TLSv1" do
			it "should return false if there are no accepted ciphers" do
				subject.supports_tlsv1?.should == false
			end
			it "should return true if there are accepted ciphers" do
				subject.add_cipher(:TLSv1, "AES256-SHA", 256, :accepted)
				subject.supports_tlsv1?.should == true
			end
		end
		context "for SSL at large" do
			it "should return false if there are no accepted ciphers" do
				subject.supports_ssl?.should == false
			end
			it "should return true if there are accepted ciphers" do
				subject.add_cipher(:TLSv1, "AES256-SHA", 256, :accepted)
				subject.supports_ssl?.should == true
			end
		end
	end

	context "checking for weak ciphers" do
		context "when weak ciphers are supported" do
			before(:each) do
				subject.add_cipher(:SSLv3, "EXP-RC4-MD5", 40, :accepted)
				subject.add_cipher(:SSLv3, "DES-CBC-SHA", 56, :accepted)
			end
			it "should return an array of weak ciphers from #weak_ciphers" do
				weak = subject.weak_ciphers
				weak.class.should == Array
				weak.each do |cipher|
					cipher[:weak].should == true
				end
				weak.count.should == 2
			end

			it "should return true from #supports_weak_ciphers" do
				subject.supports_weak_ciphers?.should == true
			end
		end

		context "when no weak ciphers are supported" do
			before(:each) do
				subject.add_cipher(:SSLv3, "AES256-SHA", 256, :accepted)
				subject.add_cipher(:TLSv1, "AES256-SHA", 256, :accepted)
				subject.add_cipher(:SSLv3, "AES128-SHA", 128, :accepted)
			end
			it "should return an empty array from #weak_ciphers" do
				subject.weak_ciphers.should == []
			end

			it "should return false from #supports_weak_ciphers" do
				subject.supports_weak_ciphers?.should == false
			end
		end
	end

	context "checking for standards compliance" do
		it "should return true if there is no SSL support" do
			subject.standards_compliant?.should == true
		end

		it "should return false if SSLv2 is supported or raise an SSLv2 exception" do
			begin
				subject.add_cipher(:SSLv2, "DES-CBC3-MD5", 168, :accepted)
				subject.standards_compliant?.should == false
			rescue ArgumentError => e
				e.message.should == "unknown SSL method `SSLv2'."
			end
		end

		it "should return false if weak ciphers are supported" do
			subject.add_cipher(:SSLv3, "EXP-RC2-CBC-MD5", 40, :accepted)
			subject.standards_compliant?.should == false
		end

		it "should return true if SSLv2 and Weak Ciphers are disabled" do
			subject.add_cipher(:SSLv3, "AES256-SHA", 256, :accepted)
			subject.add_cipher(:TLSv1, "AES256-SHA", 256, :accepted)
			subject.add_cipher(:SSLv3, "AES128-SHA", 128, :accepted)
			subject.standards_compliant?.should == true
		end
	end

	context "when printing the results" do
		context "when OpenSSL is compiled without SSLv2" do
			before(:each) do
				subject.add_cipher(:SSLv3, "AES256-SHA", 256, :accepted)
				subject.add_cipher(:TLSv1, "AES256-SHA", 256, :accepted)
				subject.add_cipher(:SSLv3, "AES128-SHA", 128, :accepted)
				subject.openssl_sslv2 = false
			end
			it "should warn the user" do
				subject.to_s.should include "*** WARNING: Your OS hates freedom! Your OpenSSL libs are compiled without SSLv2 support!"
			end
		end

		context "when we have SSL results" do
			before(:each) do
				subject.add_cipher(:SSLv3, "AES256-SHA", 256, :accepted)
				subject.add_cipher(:TLSv1, "AES256-SHA", 256, :accepted)
				subject.add_cipher(:SSLv3, "AES128-SHA", 128, :accepted)
				subject.add_cipher(:SSLv3, "EXP-RC2-CBC-MD5", 40, :accepted)

				cert = OpenSSL::X509::Certificate.new
				key = OpenSSL::PKey::RSA.new 2048
				cert.version = 2 #
				cert.serial = 1
				cert.subject = OpenSSL::X509::Name.parse "/DC=org/DC=ruby-lang/CN=Ruby CA"
				cert.issuer = cert.subject
				cert.public_key = key.public_key
				cert.not_before = Time.now
				cert.not_after = cert.not_before + 2 * 365 * 24 * 60 * 60 # 2

				subject.cert = cert
			end

			it "should contain the certificate" do
				subject.to_s.should include "Issuer: DC=org, DC=ruby-lang, CN=Ruby CA"
				subject.to_s.should include "Subject: DC=org, DC=ruby-lang, CN=Ruby CA"
			end

			it "should have a table with our SSL Cipher Results" do
				subject.to_s.should include "Accepted  *     SSLv3        40          EXP-RC2-CBC-MD5"
				subject.to_s.should include "Accepted        SSLv3        128         AES128-SHA"
				subject.to_s.should include "Accepted        SSLv3        256         AES256-SHA"
				subject.to_s.should include "Accepted        TLSv1        256         AES256-SHA"
			end
		end

		it "should return an appropriate message when SSL is not supported" do
			subject.stub(:supports_ssl?).and_return(false)
			subject.to_s.should == "Server does not appear to support SSL on this port!"
		end


	end

end