librex 0.0.68 → 0.0.70

data/lib/rex/proto/http/response.rb.ut.rb

@@ -1,150 +0,0 @@
-#!/usr/bin/env ruby
-# -*- coding: binary -*-
-
-$:.unshift(File.join(File.dirname(__FILE__), '..', '..', '..'))
-
-require 'test/unit'
-require 'rex/proto/http'
-
-class Rex::Proto::Http::Response::UnitTest < Test::Unit::TestCase
-
-	Klass = Rex::Proto::Http::Response
-
-	def test_deflate
-		h = Klass.new
-		h.body = 'hi mom'
-		h.compress = 'deflate'
-		assert_equal(
-			"HTTP/1.1 200 OK\r\n" +
-			"Content-Length: 14\r\n" +
-			"Content-Encoding: deflate\r\n\r\n" +
-			"x\234\313\310T\310\315\317\005\000\a\225\002;",
-			h.to_s, 'deflate'
-			)
-	end
-
-	def test_gzip
-		h = Klass.new
-		h.body = 'hi mom'
-		h.compress = 'gzip'
-		srand(0)
-
-		response = h.to_s
-
-		http_header = response.slice!(0,63)
-
-		assert_equal(
-			"HTTP/1.1 200 OK\r\n" +
-			"Content-Length: 26\r\n"+
-			"Content-Encoding: gzip\r\n\r\n",
-			http_header, 'http headers'
-			)
-
-		assert_equal("\x1f\x8b\x08\x00", response.slice!(0,4), 'gzip headers')
-
-		# skip the next 6 bytes as it is host & time specific (zlib's example gun does, so why not us too?)
-		response.slice!(0,6)
-
-		assert_equal("\xcb\xc8\x54\xc8\xcd\xcf\x05\x00\x68\xa4\x1c\xf0\x06\x00\x00\x00", response, 'gzip data')
-	end
-
-
-	def test_to_s
-		h = Klass.new
-
-		h.headers['Foo']     = 'Fishing'
-		h.headers['Chicken'] = 47
-
-		assert_equal(
-			"HTTP/1.1 200 OK\r\n" +
-			"Content-Length: 0\r\n" +
-			"Foo: Fishing\r\n" +
-			"Chicken: 47\r\n\r\n", h.to_s, 'to_s w/o body')
-
-		h.body = 'hi mom'
-		assert_equal(
-			"HTTP/1.1 200 OK\r\n" +
-			"Content-Length: 6\r\n" +
-			"Foo: Fishing\r\n" +
-			"Chicken: 47\r\n\r\nhi mom", h.to_s, 'to_s w/ body')
-	end
-
-
-	def test_chunked
-		h = Klass.new
-
-		h.headers['Foo']     = 'Fishing'
-		h.headers['Chicken'] = 47
-		h.auto_cl = false
-		h.transfer_chunked = true
-
-
-		assert_equal(
-			"HTTP/1.1 200 OK\r\n" +
-			"Transfer-Encoding: chunked\r\n" +
-			"Foo: Fishing\r\n" +
-			"Chicken: 47\r\n\r\n0\r\n\r\n", h.to_s, 'chunked w/o body'
-			)
-
-		srand(0)
-		h.body = Rex::Text.rand_text_alphanumeric(100)
-		assert_equal(
-			"HTTP/1.1 200 OK\r\n" +
-			"Transfer-Encoding: chunked\r\n" +
-			"Foo: Fishing\r\n" +
-			"Chicken: 47\r\n\r\n" +
-			"5\r\nsv1AD\r\n7\r\n7DnJTVy\r\n5\r\nkXGYY\r\n5\r\nM6Bmn\r\n4\r\nXuYR\r\n5\r\nlZNIJ\r\n5\r\nUzQzF\r\n9\r\nPvASjYxzd\r\n5\r\nTTOng\r\n4\r\nBJ5g\r\n8\r\nfK0XjLy3\r\n6\r\nciAAk1\r\n6\r\nFmo0RP\r\n1\r\nE\r\n2\r\npq\r\n6\r\n6f4BBn\r\n4\r\np5jm\r\n1\r\n3\r\n6\r\nLuSbAO\r\n1\r\nj\r\n2\r\n1M\r\n3\r\n5qU\r\n0\r\n\r\n",
-			h.to_s, 'random chunk sizes'
-			)
-
-		h.chunk_max_size = 1
-		h.body = 'hi mom'
-		assert_equal(
-			"HTTP/1.1 200 OK\r\n" +
-			"Transfer-Encoding: chunked\r\n" +
-			"Foo: Fishing\r\n" +
-			"Chicken: 47\r\n\r\n" +
-			"1\r\nh\r\n1\r\ni\r\n1\r\n \r\n1\r\nm\r\n1\r\no\r\n1\r\nm\r\n0\r\n\r\n",
-			h.to_s, '1 byte chunks'
-			)
-
-		h.chunk_min_size = 2
-		assert_equal(
-			"HTTP/1.1 200 OK\r\n" +
-			"Transfer-Encoding: chunked\r\n" +
-			"Foo: Fishing\r\n" +
-			"Chicken: 47\r\n\r\n" +
-			"2\r\nhi\r\n2\r\n m\r\n2\r\nom\r\n0\r\n\r\n",
-			h.to_s, '2 byte chunks'
-			)
-
-		h = Klass.new(200, 'OK', '1.0')
-		h.body = 'hi mom'
-		h.auto_cl = false
-		h.transfer_chunked = true
-		assert_raise(Rex::RuntimeError, 'chunked encoding via 1.0') {
-			h.to_s
-		}
-
-	end
-
-	def test_from_s
-		h = Klass.new
-
-		h.from_s(
-			"HTTP/1.0 404 File not found\r\n" +
-			"Lucifer: Beast\r\n" +
-			"HoHo: Satan\r\n" +
-			"Eat: Babies\r\n" +
-			"\r\n")
-
-		assert_equal(404, h.code)
-		assert_equal('File not found', h.message)
-		assert_equal('1.0', h.proto)
-		assert_equal("HTTP/1.0 404 File not found\r\n", h.cmd_string)
-		h.code = 470
-		assert_equal("HTTP/1.0 470 File not found\r\n", h.cmd_string)
-		assert_equal('Babies', h['Eat'])
-	end
-
-end

data/lib/rex/proto/http/server.rb.ut.rb

@@ -1,80 +0,0 @@
-#!/usr/bin/env ruby
-# -*- coding: binary -*-
-
-$:.unshift(File.join(File.dirname(__FILE__), '..', '..', '..'))
-
-require 'test/unit'
-require 'rex/proto/http'
-
-class Rex::Proto::Http::Server::UnitTest < Test::Unit::TestCase
-
-	ListenPort = 8090
-	ListenHost = '127.0.0.1'
-
-	SrvKlass = Rex::Proto::Http::Server
-	CliKlass = Rex::Proto::Http::Client
-
-	def test_server
-		begin
-			s   = start_srv
-			c   = CliKlass.new(ListenHost, ListenPort)
-
-			1.upto(10) {
-				req = c.request_raw('uri' => '/')
-				res = c.send_recv(req)
-				assert_not_nil(res)
-				assert_equal(404, res.code)
-			}
-		ensure
-			stop_srv
-		end
-	end
-
-	def test_resource
-		begin
-			s   = start_srv
-			c   = CliKlass.new(ListenHost, ListenPort)
-
-			s.add_resource('/foo',
-				'Proc' => Proc.new { |cli, req|
-					resp = Rex::Proto::Http::Response::OK.new
-
-					resp.body = "Chickens everywhere"
-
-					cli.send_response(resp)
-				})
-
-			1.upto(10) {
-				req = c.request_raw('uri' => '/foo')
-				res = c.send_recv(req)
-				assert_not_nil(res)
-				assert_equal(200, res.code)
-				assert_equal("Chickens everywhere", res.body)
-			}
-
-			s.remove_resource('/foo')
-
-			req = c.request_raw('uri' => '/foo')
-			res = c.send_recv(req)
-			assert_not_nil(res)
-			assert_equal(404, res.code)
-		ensure
-			stop_srv
-		end
-	end
-
-protected
-
-	def start_srv
-		self.srv = SrvKlass.new(ListenPort, ListenHost)
-		self.srv.start
-		self.srv
-	end
-
-	def stop_srv
-		self.srv.stop if (self.srv)
-	end
-
-attr_accessor :srv
-
-end

data/lib/rex/proto/ntlm.rb.ut.rb

@@ -1,181 +0,0 @@
-#!/usr/bin/env ruby
-# -*- coding: binary -*-
-
-$:.unshift(File.join(File.dirname(__FILE__), '..', '..'))
-
-require 'test/unit'
-require 'rex/proto/ntlm'
-require 'rex/socket'
-
-class ConnectionTest < Test::Unit::TestCase
-	def setup
-		@user = "admin"
-		@pass = "1234"
-		@domain = ""
-		@host = "192.168.145.161"
-	end
-
-	def test_socket_connectivity
-		assert_nothing_raised do
-			socket = Rex::Socket.create_tcp(
-				'PeerHost' => @host,
-				'PeerPort' => 80
-			)
-			assert_kind_of Socket, socket
-			assert !socket.closed?
-			socket.close
-			assert socket.closed?
-		end
-	end
-
-	def http_message(msg)
-		get_req = "GET / HTTP/1.1\r\n"
-		get_req += "Host: #{@host}\r\n"
-		get_req += "User-Agent: Mozilla/4.0 (compatible; MSIE 6.0; Windows NT 5.1)\r\n"
-		get_req += "Authorization: NTLM #{msg.encode64}\r\n"
-		get_req += "Content-type: application/x-www-form-urlencoded\r\n"
-		get_req += "Content-Length: 0\r\n"
-		get_req += "\r\n"
-	end
-
-	def client_auth(pw)
-		msg_1 = Rex::Proto::NTLM::Message::Type1.new
-		get_req = http_message(msg_1)
-		socket = Rex::Socket.create_tcp(
-			'PeerHost' => @host,
-			'PeerPort' => 80
-		)
-		socket.put get_req
-		res = socket.get(3)
-		assert res =~ /WWW-Authenticate: NTLM TlRM/
-			res_ntlm = res.match(/WWW-Authenticate: NTLM ([A-Z0-9\x2b\x2f=]+)/i)[1]
-		assert_operator res_ntlm.size, :>=, 24
-		msg_2 = Rex::Proto::NTLM::Message.decode64(res_ntlm)
-		assert msg_2
-		msg_3 = msg_2.response({:user => @user, :password => pw}, {:ntlmv2 => true})
-		assert msg_3
-		auth_req = http_message(msg_3)
-		socket.put auth_req
-		auth_res = socket.get(3)
-		socket.close
-		return auth_res
-	end
-
-	def test_client_auth_success
-		assert_equal client_auth(@pass)[0,12], "HTTP/1.1 200"
-	end
-
-	def test_client_auth_fail
-		assert_not_equal client_auth("badpass")[0,12], "HTTP/1.1 200"
-		assert_equal client_auth("badpass")[0,12], "HTTP/1.1 401"
-	end
-end
-
-# FunctionTest by Minero Aoki
-
-class FunctionTest < Test::Unit::TestCase #:nodoc:
-	def setup
-		@passwd = "SecREt01"
-		@user   = "user"
-		@domain = "domain"
-		@challenge = ["0123456789abcdef"].pack("H*")
-		@client_ch = ["ffffff0011223344"].pack("H*")
-		@timestamp = 1055844000
-		@trgt_info = [
-			"02000c0044004f004d00410049004e00" +
-			"01000c00530045005200560045005200" +
-			"0400140064006f006d00610069006e00" +
-			"2e0063006f006d000300220073006500" +
-			"72007600650072002e0064006f006d00" +
-			"610069006e002e0063006f006d000000" +
-			"0000"
-		].pack("H*")
-	end
-
-	def test_lm_hash
-		ahash = ["ff3750bcc2b22412c2265b23734e0dac"].pack("H*")
-		assert_equal ahash, Rex::Proto::NTLM::Crypt::lm_hash(@passwd)
-	end
-
-	def test_ntlm_hash
-		ahash = ["cd06ca7c7e10c99b1d33b7485a2ed808"].pack("H*")
-		assert_equal ahash, Rex::Proto::NTLM::Crypt::ntlm_hash(@passwd)
-	end
-
-	def test_ntlmv2_hash
-		ahash = ["04b8e0ba74289cc540826bab1dee63ae"].pack("H*")
-		assert_equal ahash, Rex::Proto::NTLM::Crypt::ntlmv2_hash(@user, @passwd, @domain)
-	end
-
-	def test_lm_response
-		ares = ["c337cd5cbd44fc9782a667af6d427c6de67c20c2d3e77c56"].pack("H*")
-		assert_equal ares, Rex::Proto::NTLM::Crypt::lm_response(
-			{
-			:lm_hash => Rex::Proto::NTLM::Crypt::lm_hash(@passwd),
-			:challenge => @challenge
-		}
-		)
-	end
-
-	def test_ntlm_response
-		ares = ["25a98c1c31e81847466b29b2df4680f39958fb8c213a9cc6"].pack("H*")
-		ntlm_hash = Rex::Proto::NTLM::Crypt::ntlm_hash(@passwd)
-		assert_equal ares, Rex::Proto::NTLM::Crypt::ntlm_response(
-			{
-			:ntlm_hash => ntlm_hash,
-			:challenge => @challenge
-		}
-		)
-	end
-
-	def test_lmv2_response
-		ares = ["d6e6152ea25d03b7c6ba6629c2d6aaf0ffffff0011223344"].pack("H*")
-		assert_equal ares, Rex::Proto::NTLM::Crypt::lmv2_response(
-			{
-			:ntlmv2_hash => Rex::Proto::NTLM::Crypt::ntlmv2_hash(@user, @passwd, @domain),
-			:challenge => @challenge
-		},
-			{ :client_challenge => @client_ch }
-		)
-	end
-
-	def test_ntlmv2_response
-		ares = [
-			"cbabbca713eb795d04c97abc01ee4983" +
-			"01010000000000000090d336b734c301" +
-			"ffffff00112233440000000002000c00" +
-			"44004f004d00410049004e0001000c00" +
-			"53004500520056004500520004001400" +
-			"64006f006d00610069006e002e006300" +
-			"6f006d00030022007300650072007600" +
-			"650072002e0064006f006d0061006900" +
-			"6e002e0063006f006d00000000000000" +
-			"0000"
-		].pack("H*")
-		assert_equal ares, Rex::Proto::NTLM::Crypt::ntlmv2_response(
-			{
-			:ntlmv2_hash => Rex::Proto::NTLM::Crypt::ntlmv2_hash(@user, @passwd, @domain),
-			:challenge => @challenge,
-			:target_info => @trgt_info
-		},
-			{
-			:timestamp => @timestamp,
-			:client_challenge => @client_ch
-		}
-		)
-	end
-
-	def test_ntlm2_session
-		acha = ["ffffff001122334400000000000000000000000000000000"].pack("H*")
-		ares = ["10d550832d12b2ccb79d5ad1f4eed3df82aca4c3681dd455"].pack("H*")
-		session = Rex::Proto::NTLM::Crypt::ntlm2_session(
-			{
-			:ntlm_hash => Rex::Proto::NTLM::Crypt::ntlm_hash(@passwd),
-			:challenge => @challenge
-		},
-			{ :client_challenge => @client_ch }
-		)
-		assert_equal acha, session[0]
-		assert_equal ares, session[1]
-	end
-end

data/lib/rex/proto/rfb.rb.ut.rb

@@ -1,40 +0,0 @@
-#!/usr/bin/env ruby
-# -*- coding: binary -*-
-
-$:.unshift(File.join(File.dirname(__FILE__), '..', '..'))
-
-##
-# $Id: $
-##
-
-##
-#
-# RFB protocol support
-#
-# by Joshua J. Drake <jduck>
-#
-# Based on:
-# vnc_auth_none contributed by Matteo Cantoni <goony[at]nothink.org>
-# vnc_auth_login contributed by carstein <carstein.sec[at]gmail.com>
-#
-##
-
-require 'rex/socket'
-require 'rex/proto/rfb'
-
-sd = Rex::Socket::Tcp.create('PeerHost' => ENV["VNCHOST"], 'PeerPort' => Rex::Proto::RFB::DefaultPort)
-
-v = Rex::Proto::RFB::Client.new(sd)
-if not v.connect('password')
-	$stderr.puts v.error
-	exit(1)
-end
-
-loop {
-	sret = select([sd],nil,nil,10)
-	puts sret.inspect
-	if sret and sret[0].include? sd
-		buf = sd.sysread(8192)
-		puts "read #{buf.length} bytes: #{buf.inspect}"
-	end
-}