Class: ConnectionTest

Inherits:
Test::Unit::TestCase
  • Object
show all
Defined in:
lib/rex/proto/ntlm.rb.ut.rb

Instance Method Summary collapse

Instance Method Details

#client_auth(pw) ⇒ Object 



41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
 
# File 'lib/rex/proto/ntlm.rb.ut.rb', line 41

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

#http_message(msg) ⇒ Object 



31
32
33
34
35
36
37
38
39
 
# File 'lib/rex/proto/ntlm.rb.ut.rb', line 31

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

#setup ⇒ Object 



11
12
13
14
15
16
 
# File 'lib/rex/proto/ntlm.rb.ut.rb', line 11

def setup
  @user = "admin"
  @pass = "1234"
  @domain = ""
  @host = "192.168.145.161"
end

#test_client_auth_fail ⇒ Object 



68
69
70
71
 
# File 'lib/rex/proto/ntlm.rb.ut.rb', line 68

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

#test_client_auth_success ⇒ Object 



64
65
66
 
# File 'lib/rex/proto/ntlm.rb.ut.rb', line 64

def test_client_auth_success
  assert_equal client_auth(@pass)[0,12], "HTTP/1.1 200"
end

#test_socket_connectivity ⇒ Object 



18
19
20
21
22
23
24
25
26
27
28
29
 
# File 'lib/rex/proto/ntlm.rb.ut.rb', line 18

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