librex 0.0.68 → 0.0.70

data/lib/rex/proto/smb.rb.ts.rb

@@ -1,9 +0,0 @@
-#!/usr/bin/env ruby
-# -*- coding: binary -*-
-
-require 'rex/test'
-require 'rex/proto/smb/client.rb.ut.rb'
-require 'rex/proto/smb/constants.rb.ut.rb'
-require 'rex/proto/smb/crypt.rb.ut.rb'
-require 'rex/proto/smb/simpleclient.rb.ut.rb'
-require 'rex/proto/smb/utils.rb.ut.rb'

data/lib/rex/proto/smb/client.rb.ut.rb

@@ -1,224 +0,0 @@
-#!/usr/bin/env ruby
-# -*- coding: binary -*-
-
-$:.unshift(File.join(File.dirname(__FILE__), '..', '..', '..'))
-
-require 'rex/test'
-require 'rex/proto/smb/constants'
-require 'rex/proto/smb/exceptions'
-require 'rex/proto/smb/utils'
-require 'rex/proto/smb/client'
-require 'rex/proto/dcerpc'
-require 'rex/socket'
-
-class Rex::Proto::SMB::Client::UnitTest < Test::Unit::TestCase
-
-	Klass = Rex::Proto::SMB::Client
-
-	# Alias over the Rex DCERPC protocol modules
-	DCERPCPacket   = Rex::Proto::DCERPC::Packet
-	DCERPCClient   = Rex::Proto::DCERPC::Client
-	DCERPCResponse = Rex::Proto::DCERPC::Response
-	DCERPCUUID     = Rex::Proto::DCERPC::UUID
-
-	def test_smb_open_share
-
-		share = 'C$'
-
-		write_data = ('A' * 256)
-		filename = 'smb_test.txt'
-
-		begin
-		Timeout.timeout($_REX_TEST_TIMEOUT) {
-		s = Rex::Socket.create_tcp(
-			'PeerHost' => $_REX_TEST_SMB_HOST,
-			'PeerPort' => 139
-		)
-
-		c = Klass.new(s)
-
-		# Request a SMB session over NetBIOS
-		# puts "[*] Requesting a SMB session over NetBIOS..."
-		ok = c.session_request()
-		assert_kind_of(Rex::Struct2::CStruct, ok)
-
-		# Check for a positive session response
-		# A negative response is 0x83
-		assert_equal(ok.v['Type'], 0x82)
-
-		# puts "[*] Negotiating SMB dialects..."
-		ok = c.negotiate()
-		assert_kind_of(Rex::Struct2::CStruct, ok)
-
-		# puts "[*] Authenticating with NTLMv2..."
-		ok = c.session_setup_with_ntlmssp($_REX_TEXT_SMB_USER, $_REX_TEXT_SMB_PASS)
-		assert_kind_of(Rex::Struct2::CStruct, ok)
-		assert_not_equal(c.auth_user_id, 0)
-
-		# puts "[*] Connecting to the share..."
-		ok = c.tree_connect(share)
-		assert_kind_of(Rex::Struct2::CStruct, ok)
-		assert_not_equal(c.last_tree_id, 0)
-
-		# puts "[*] Opening a file for write..."
-		ok = c.open(filename)
-		assert_kind_of(Rex::Struct2::CStruct, ok)
-		assert_not_equal(c.last_file_id, 0)
-
-		# puts "[*] Writing data to the test file..."
-		ok = c.write(c.last_file_id, 0, write_data)
-		assert_kind_of(Rex::Struct2::CStruct, ok)
-		assert_equal(ok['Payload'].v['CountLow'], write_data.length)
-
-		# puts "[*] Closing the test file..."
-		ok = c.close(c.last_file_id)
-		assert_kind_of(Rex::Struct2::CStruct, ok)
-
-		# puts "[*] Opening a file for read..."
-		ok = c.open(filename, 1)
-		assert_kind_of(Rex::Struct2::CStruct, ok)
-		assert_not_equal(c.last_file_id, 0)
-
-		# puts "[*] Reading data from the test file..."
-		ok = c.read(c.last_file_id, 0, write_data.length)
-		assert_kind_of(Rex::Struct2::CStruct, ok)
-		assert_equal(ok['Payload'].v['DataLenLow'], write_data.length)
-
-		read_data =  ok.to_s.slice(
-			ok['Payload'].v['DataOffset'] + 4,
-			ok['Payload'].v['DataLenLow']
-		)
-		assert_equal(read_data, write_data)
-
-		# puts "[*] Closing the test file..."
-		ok = c.close(c.last_file_id)
-		assert_kind_of(Rex::Struct2::CStruct, ok)
-
-		# puts "[*] Disconnecting from the tree..."
-		ok = c.tree_disconnect
-		assert_kind_of(Rex::Struct2::CStruct, ok)
-
-		s.close
-
-
-		# Reconnect and delete the file
-		s = Rex::Socket.create_tcp(
-			'PeerHost' => $_REX_TEST_SMB_HOST,
-			'PeerPort' => 139
-		)
-
-		c = Klass.new(s)
-
-		# Request a SMB session over NetBIOS
-		# puts "[*] Requesting a SMB session over NetBIOS..."
-		ok = c.session_request()
-		assert_kind_of(Rex::Struct2::CStruct, ok)
-
-		# Check for a positive session response
-		# A negative response is 0x83
-		assert_equal(ok.v['Type'], 0x82)
-
-		# puts "[*] Negotiating SMB dialects..."
-		ok = c.negotiate()
-		assert_kind_of(Rex::Struct2::CStruct, ok)
-
-		# puts "[*] Authenticating with NTLMv2..."
-		ok = c.session_setup_with_ntlmssp($_REX_TEXT_SMB_USER, $_REX_TEXT_SMB_PASS)
-		assert_kind_of(Rex::Struct2::CStruct, ok)
-		assert_not_equal(c.auth_user_id, 0)
-
-		# puts "[*] Connecting to the share..."
-		ok = c.tree_connect(share)
-		assert_kind_of(Rex::Struct2::CStruct, ok)
-		assert_not_equal(c.last_tree_id, 0)
-
-		# puts "[*] Deleting the test file..."
-		ok = c.delete(filename)
-		assert_kind_of(Rex::Struct2::CStruct, ok)
-
-		# puts "[*] Diconnecting from the tree..."
-		ok = c.tree_disconnect
-		assert_kind_of(Rex::Struct2::CStruct, ok)
-
-		s.close
-		}
-		rescue Timeout::Error
-			flunk('timeout')
-		end
-
-	end
-
-	def test_smb_session_request
-		begin
-		Timeout.timeout($_REX_TEST_TIMEOUT) {
-		s = Rex::Socket.create_tcp(
-			'PeerHost' => $_REX_TEST_SMB_HOST,
-			'PeerPort' => 139
-		)
-
-		c = Klass.new(s)
-
-		# Request a SMB session over NetBIOS
-		# puts "[*] Requesting a SMB session over NetBIOS..."
-		ok = c.session_request()
-		assert_kind_of(Rex::Struct2::CStruct, ok)
-
-		# Check for a positive session response
-		# A negative response is 0x83
-		assert_equal(ok.v['Type'], 0x82)
-
-		# puts "[*] Negotiating SMB dialects..."
-		ok = c.negotiate()
-		assert_kind_of(Rex::Struct2::CStruct, ok)
-
-		# puts "[*] Authenticating with NTLMv2..."
-		ok = c.session_setup_with_ntlmssp
-		assert_kind_of(Rex::Struct2::CStruct, ok)
-
-		# puts "[*] Authenticating with NTLMv1..."
-		ok = c.session_setup_no_ntlmssp
-		assert_kind_of(Rex::Struct2::CStruct, ok)
-
-		# puts "[*] Authenticating with clear text passwords..."
-		begin
-			ok = c.session_setup_clear
-			assert_kind_of(Rex::Struct2::CStruct, ok)
-		rescue Rex::Proto::SMB::Exceptions::ErrorCode
-			if ($!.error_code != 0x00010002)
-				raise $!
-			end
-		end
-
-		# puts "[*] Connecting to IPC$..."
-		ok = c.tree_connect
-		assert_kind_of(Rex::Struct2::CStruct, ok)
-
-		# puts "[*] Opening the \BROWSER pipe..."
-		ok = c.create_pipe('\BROWSER')
-		assert_kind_of(Rex::Struct2::CStruct, ok)
-
-		vers = DCERPCUUID.vers_by_name('SRVSVC')
-		uuid = DCERPCUUID.uuid_by_name('SRVSVC')
-		bind, ctx = DCERPCPacket.make_bind_fake_multi(uuid, vers)
-
-		# puts "[*] Binding to the Server Service..."
-		ok = c.trans_named_pipe(c.last_file_id, bind)
-		assert_kind_of(Rex::Struct2::CStruct, ok)
-
-		data = ok.to_s.slice(
-			ok['Payload'].v['DataOffset'] + 4,
-			ok['Payload'].v['DataCount']
-		)
-		assert_not_equal(data, nil)
-
-		resp = DCERPCResponse.new(data)
-		assert_equal(resp.type, 12)
-		}
-		rescue Timeout::Error
-			flunk('timeout')
-		end
-	end
-
-
-end
-

data/lib/rex/proto/smb/constants.rb.ut.rb

@@ -1,19 +0,0 @@
-#!/usr/bin/env ruby
-# -*- coding: binary -*-
-
-$:.unshift(File.join(File.dirname(__FILE__), '..', '..', '..'))
-
-require 'rex/test'
-require 'rex/proto/smb/constants'
-
-class Rex::Proto::SMB::Constants::UnitTest < Test::Unit::TestCase
-	
-	Klass = Rex::Proto::SMB::Constants
-
-	def test_defines
-		assert_equal(Klass::SMB_COM_CREATE_DIRECTORY,  0x00 )
-		assert_equal(Klass::SMB_COM_NT_CREATE_ANDX, 0xa2 )
-		assert_equal(Klass::NT_TRANSACT_QUERY_SECURITY_DESC, 0x06)
-	end
-	
-end

data/lib/rex/proto/smb/simpleclient.rb.ut.rb

@@ -1,129 +0,0 @@
-#!/usr/bin/env ruby
-# -*- coding: binary -*-
-
-$:.unshift(File.join(File.dirname(__FILE__), '..', '..', '..'))
-
-require 'rex/test'
-require 'rex/proto/smb'
-require 'rex/proto/dcerpc'
-require 'rex/socket'
-
-class Rex::Proto::SMB::SimpleClient::UnitTest < Test::Unit::TestCase
-
-	Klass = Rex::Proto::SMB::SimpleClient
-
-	# Alias over the Rex DCERPC protocol modules
-	DCERPCPacket   = Rex::Proto::DCERPC::Packet
-	DCERPCClient   = Rex::Proto::DCERPC::Client
-	DCERPCResponse = Rex::Proto::DCERPC::Response
-	DCERPCUUID     = Rex::Proto::DCERPC::UUID
-	XCEPT          = Rex::Proto::SMB::Exceptions
-
-	FILE_CREATE = 0x10
-	FILE_TRUNC  = 0x02
-	FILE_OPEN   = 0x01
-
-
-	def test_smb_open_share
-		user = 'SMBTest'
-		pass = 'SMBTest'
-		share = 'C$'
-
-		write_data = ('A' * (1024 * 8))
-		filename = 'smb_tester.txt'
-		begin
-		Timeout.timeout($_REX_TEST_TIMEOUT) {
-		s = Rex::Socket.create_tcp(
-			'PeerHost' => $_REX_TEST_SMB_HOST,
-			'PeerPort' => 445
-		)
-
-		c = Klass.new(s, true)
-
-		begin
-	        c.login('*SMBSERVER', user, pass)
-		rescue XCEPT::LoginError
-			flunk('login failure')
-		end
-
-		c.connect(share)
-
-		f = c.open(filename, 'rwct')
-	    f << write_data
-		f.close
-
-		f = c.open(filename, 'ro')
-	    d = f.read()
-	    f.close
-
-		c.delete(filename)
-		c.disconnect(share)
-
-		s.close
-		}
-		rescue Timeout::Error
-			flunk('timeout')
-		end
-	end
-
-	def test_smb_dcerpc
-		begin
-		Timeout.timeout($_REX_TEST_TIMEOUT) {
-		s = Rex::Socket.create_tcp(
-			'PeerHost' => $_REX_TEST_SMB_HOST,
-			'PeerPort' => 445
-		)
-
-		c = Klass.new(s, true)
-
-		user = ''
-		pass = ''
-
-		begin
-			c.login('*SMBSERVER', user, pass)
-		rescue XCEPT::LoginError
-			flunk('login failure')
-		end
-
-		c.connect('IPC$')
-		f = c.create_pipe('\BROWSER')
-
-		bind, ctx = DCERPCPacket.make_bind_fake_multi(
-			'4b324fc8-1670-01d3-1278-5a47bf6ee188', '3.0',
-			10,
-			4
-		)
-
-		# Evasion techniques:
-		# 	1) Write the bind out a few bytes at a time with a random offset
-		#	2) Read the response back a few bytes at a time with a random offset
-
-		# Write the bind request out in random chunk sizes
-		while (bind.length > 0)
-		    f.write( bind.slice!(0, (rand(20)+5)), rand(1024)+1 )
-		end
-
-		d = ''
-		# Read the response back a few bytes a time
-		begin
-		    while(true)
-			    t = (f.read((rand(20)+5), rand(1024)+1))
-				last if ! t.length
-				d << t
-			end
-		rescue XCEPT::NoReply
-		end
-
-		r = DCERPCResponse.new(d)
-		assert_equal(r.type, 12)
-		assert_equal(r.ack_result[ctx-0], 0)
-		assert_equal(r.ack_result[ctx-1], 2)
-
-		s.close
-		}
-		rescue Timeout::Error
-			flunk('timeout')
-		end
-	end
-end
-

data/lib/rex/proto/smb/utils.rb.ut.rb

@@ -1,21 +0,0 @@
-#!/usr/bin/env ruby
-# -*- coding: binary -*-
-
-$:.unshift(File.join(File.dirname(__FILE__), '..', '..', '..'))
-
-require 'rex/test'
-require 'rex/proto/smb/utils'
-
-class Rex::Proto::SMB::Utils::UnitTest < Test::Unit::TestCase
-	
-	Klass = Rex::Proto::SMB::Utils
-
-	def test_nbname
-	
-		nbdecoded = 'METASPLOITROCKS!'
-		nbencoded = 'ENEFFEEBFDFAEMEPEJFEFCEPEDELFDCB'
-		
-		assert_equal(Klass.nbname_encode(nbdecoded),  nbencoded )
-		assert_equal(Klass.nbname_decode(nbencoded),  nbdecoded )
-	end
-end

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

@@ -1,29 +0,0 @@
-#!/usr/bin/env ruby
-# -*- coding: binary -*-
-#
-# $Revision: 15548 $
-#
-# $Id: server.rb.ut.rb 15548 2012-06-29 06:08:20Z rapid7 $
-#
-
-require 'rex/compat'
-require 'rex/proto/tftp'
-
-content = nil
-
-fn = ARGV.shift
-if (fn and fn.length > 0)
-	File.open(fn, "rb") do |fd|
-		content = fd.read(fd.stat.size)
-	end
-end
-
-content ||= "A" * (1024*1024)
-
-
-tftp = Rex::Proto::TFTP::Server.new
-tftp.register_file("poo", content)
-tftp.start
-
-#loop { break if not tftp.thread.alive? }
-tftp.thread.join

data/lib/rex/service_manager.rb.ut.rb

@@ -1,33 +0,0 @@
-#!/usr/bin/env ruby
-# -*- coding: binary -*-
-
-$:.unshift(File.join(File.dirname(__FILE__), '..'))
-
-require 'test/unit'
-require 'rex/service_manager'
-
-class Rex::ServiceManager::UnitTest < Test::Unit::TestCase
-
-	Klass = Rex::ServiceManager
-
-	def test_svcm
-		begin
-			c = Klass
-			s = c.start(Rex::Proto::Http::Server, 8090)
-			assert_not_nil(s)
-			t = c.start(Rex::Proto::Http::Server, 8090)
-			assert_not_nil(t)
-			assert_equal(s, t)
-			z = c.start(Rex::Proto::Http::Server, 8091)
-			assert_not_equal(t, z)
-			assert_equal("HTTP Server", s.alias)
-			assert_equal("HTTP Server 1", z.alias)
-		ensure
-			c.stop_by_alias(s.alias) if (s)
-			c.stop_by_alias(z.alias) if (z)
-			c.stop_by_alias(t.alias) if (t)
-		end
-
-	end
-
-end