Class: Rex::Ui::Text::Input::Socket

Inherits:
Rex::Ui::Text::Input show all
Defined in:
lib/rex/ui/text/input/socket.rb

Overview

This class implements input against a socket.

Constant Summary

Constants included from Color

Color::AnsiAttributes

Instance Attribute Summary

Attributes inherited from Rex::Ui::Text::Input

#config, #eof, #prompt, #prompt_char

Instance Method Summary collapse

Methods inherited from Rex::Ui::Text::Input

#auto_color, #disable_color, #enable_color, #intrinsic_shell?, #reset_color, #reset_tab_completion, #update_prompt

Methods included from Color

#ansi, #colorize, #do_colorize, #reset_color, #substitute_colors

Constructor Details

#initialize(sock) ⇒ Socket

Returns a new instance of Socket.



15
16
17
 
# File 'lib/rex/ui/text/input/socket.rb', line 15

def initialize(sock)
	@sock = sock
end

Instance Method Details

#eof? ⇒ Boolean

Returns whether or not EOF has been reached on stdin.

Returns:

  • (Boolean) 


82
83
84
 
# File 'lib/rex/ui/text/input/socket.rb', line 82

def eof?
	@sock.closed?
end

#fd ⇒ Object

Returns the file descriptor associated with a socket.



89
90
91
 
# File 'lib/rex/ui/text/input/socket.rb', line 89

def fd
	return @sock
end

#gets ⇒ Object

Wait for a line of input to be read from a socket.



36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
 
# File 'lib/rex/ui/text/input/socket.rb', line 36

def gets

	# Initialize the line buffer
	line = ''
	
	# Read data one byte at a time until we see a LF
	while (true)

		break if line.include?("\n")
		
		# Read another character of input
		char = @sock.getc
		if char.nil?
			@sock.close
			return
		end
		
		# Telnet sends 0x04 as EOF
		if (char == 4)
			@sock.write("[*] Caught ^D, closing the socket...\n")
			@sock.close
			return
		end
		
		# Append this character to the string
		line << char

		# Handle telnet sequences
		case line
			when /\xff\xf4\xff\xfd\x06/
				@sock.write("[*] Caught ^C, closing the socket...\n")
				@sock.close
				return
				
			when /\xff\xed\xff\xfd\x06/
				@sock.write("[*] Caught ^Z\n")
				return		
		end
	end
	
	return line
end

#supports_readline ⇒ Object

Sockets do not currently support readline.



22
23
24
 
# File 'lib/rex/ui/text/input/socket.rb', line 22

def supports_readline
	false
end

#sysread(len = 1) ⇒ Object

Reads input from the raw socket.



29
30
31
 
# File 'lib/rex/ui/text/input/socket.rb', line 29

def sysread(len = 1)
	@sock.sysread(len)
end