librex 0.0.1

data/lib/rex/ui/text/input/buffer.rb

@@ -0,0 +1,75 @@
+require 'rex/ui'
+
+module Rex
+module Ui
+module Text
+
+require 'rex/io/stream_abstraction'
+
+###
+#
+# This class implements input against a socket.
+#
+###
+class Input::Buffer < Rex::Ui::Text::Input
+
+	class BufferSock
+		include Rex::IO::StreamAbstraction
+	end
+
+	def initialize
+		@sock = BufferSock.new
+		@sock.initialize_abstraction
+	end
+
+	def close
+		@sock.cleanup_abstraction
+	end
+
+	def sysread(len = 1)
+		@sock.rsock.sysread(len)
+	end
+
+	def put(msg)
+		@sock.lsock.write(msg)
+	end
+
+	#
+	# Wait for a line of input to be read from a socket.
+	#
+	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.rsock.getc
+			
+			# Append this character to the string
+			line << char
+		end
+		
+		return line
+	end
+
+	#
+	# Returns whether or not EOF has been reached on stdin.
+	#
+	def eof?
+		@sock.lsock.closed?
+	end
+
+	#
+	# Returns the file descriptor associated with a socket.
+	#
+	def fd
+		return @sock.rsock
+	end
+end
+
+end
+end
+end

data/lib/rex/ui/text/input/readline.rb

@@ -0,0 +1,129 @@
+require 'rex/ui'
+
+module Rex
+module Ui
+module Text
+
+begin 
+
+	###
+	#
+	# This class implements standard input using readline against
+	# standard input.  It supports tab completion.
+	#
+	###
+	class Input::Readline < Rex::Ui::Text::Input
+		
+		#
+		# Initializes the readline-aware Input instance for text.
+		#
+		def initialize(tab_complete_proc = nil)
+		
+		
+			if(not Object.const_defined?('Readline'))
+				begin 
+					require 'readline'
+				rescue ::LoadError
+					require 'readline_compatible'
+				end
+			end
+			
+			self.extend(::Readline)
+			
+			if (tab_complete_proc)
+				::Readline.basic_word_break_characters = "\x00"
+				::Readline.completion_proc = tab_complete_proc
+				@rl_saved_proc = tab_complete_proc
+			end
+		end
+
+		#
+		# Reattach the original completion proc
+		#
+		def reset_tab_completion(tab_complete_proc = nil)
+			::Readline.basic_word_break_characters = "\x00"
+			::Readline.completion_proc = tab_complete_proc || @rl_saved_proc
+		end
+		
+		#
+		# Whether or not the input medium supports readline.
+		#
+		def supports_readline
+			true
+		end
+		
+		#
+		# Calls sysread on the standard input handle.
+		#
+		def sysread(len = 1)
+			begin
+				self.fd.sysread(len)
+			rescue ::Errno::EINTR
+				retry
+			end
+		end
+		
+		#
+		# Read a line from stdin
+		#
+		def gets()
+			begin
+				self.fd.gets()
+			rescue ::Errno::EINTR
+				retry
+			end
+		end
+
+		#
+		# Stick readline into a low-priority thread so that the scheduler doesn't slow
+		# down other background threads. This is important when there are many active
+		# background jobs, such as when the user is running Karmetasploit
+		#
+		def pgets
+		
+			line = nil
+			orig = Thread.current.priority
+			
+			begin
+				Thread.current.priority = -20
+				output.prompting
+				line = ::Readline.readline(prompt, true)
+				::Readline::HISTORY.pop if (line and line.empty?)
+			ensure
+				Thread.current.priority = orig || 0
+			end
+			
+			line
+		end
+
+		#
+		# Returns the output pipe handle
+		#
+		def fd
+			$stdin
+		end
+		
+		#
+		# Indicates that this input medium as a shell builtin, no need 
+		# to extend.
+		#
+		def intrinsic_shell?
+			true
+		end
+
+		#
+		# The prompt that is to be displayed.
+		#
+		attr_accessor :prompt
+		#
+		# The output handle to use when displaying the prompt.
+		#
+		attr_accessor :output
+
+	end
+rescue LoadError
+end
+
+end
+end
+end

data/lib/rex/ui/text/input/socket.rb

@@ -0,0 +1,95 @@
+require 'rex/ui'
+
+module Rex
+module Ui
+module Text
+
+###
+#
+# This class implements input against a socket.
+#
+###
+class Input::Socket < Rex::Ui::Text::Input
+
+	def initialize(sock)
+		@sock = sock
+	end
+
+	#
+	# Sockets do not currently support readline.
+	#
+	def supports_readline
+		false
+	end
+
+	#
+	# Reads input from the raw socket.
+	#
+	def sysread(len = 1)
+		@sock.sysread(len)
+	end
+
+	#
+	# Wait for a line of input to be read from a socket.
+	#
+	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
+
+	#
+	# Returns whether or not EOF has been reached on stdin.
+	#
+	def eof?
+		@sock.closed?
+	end
+
+	#
+	# Returns the file descriptor associated with a socket.
+	#
+	def fd
+		return @sock
+	end
+end
+
+end
+end
+end

data/lib/rex/ui/text/input/stdio.rb

@@ -0,0 +1,45 @@
+require 'rex/ui'
+
+module Rex
+module Ui
+module Text
+
+###
+#
+# This class implements input against standard in.
+#
+###
+class Input::Stdio < Rex::Ui::Text::Input
+
+	#
+	# Reads text from standard input.
+	#
+	def sysread(len = 1)
+		$stdin.sysread(len)
+	end
+
+	#
+	# Wait for a line of input to be read from standard input.
+	#
+	def gets
+		return $stdin.gets
+	end
+
+	#
+	# Returns whether or not EOF has been reached on stdin.
+	#
+	def eof?
+		$stdin.closed?
+	end
+
+	#
+	# Returns the file descriptor associated with standard input.
+	#
+	def fd
+		return $stdin
+	end
+end
+
+end
+end
+end

data/lib/rex/ui/text/irb_shell.rb

@@ -0,0 +1,55 @@
+module Rex
+module Ui
+module Text
+
+###
+#
+# This class wraps the creation of an IRB shell.
+#
+###
+class IrbShell
+
+	@@IrbInitialized = false
+
+	def initialize(binding)
+		@binding_ctx = binding
+	end
+
+	#
+	# Runs the IRB shell until completion.  The binding parameter initializes
+	# IRB to the appropriate binding context.
+	#
+	def run
+		# Initialize IRB by setting up its internal configuration hash and
+		# stuff.
+		if (@@IrbInitialized == false)
+			load('irb.rb')
+
+			IRB.setup(nil)
+			IRB.conf[:PROMPT_MODE]  = :SIMPLE
+
+			@@IrbInitialized = true
+		end
+
+		# Create a new IRB instance
+		irb = IRB::Irb.new(IRB::WorkSpace.new(@binding_ctx))
+
+		# Set the primary irb context so that exit and other intrinsic
+		# commands will work.
+		IRB.conf[:MAIN_CONTEXT] = irb.context
+
+		# Trap interrupt
+		trap("SIGINT") do
+			irb.signal_handle
+		end
+
+		# Keep processing input until the cows come home...
+		catch(:IRB_EXIT) do
+			irb.eval_input
+		end
+	end
+
+end
+end
+end
+end

data/lib/rex/ui/text/output.rb

@@ -0,0 +1,80 @@
+require 'rex/ui'
+
+module Rex
+module Ui
+module Text
+
+###
+#
+# This class implements text-based output but is not
+# tied to an output medium.
+#
+###
+class Output < Rex::Ui::Output
+
+	require 'rex/ui/text/output/stdio'
+	require 'rex/ui/text/output/socket'
+	require 'rex/ui/text/output/buffer'
+	require 'rex/ui/text/output/file'
+	require 'rex/ui/text/color'
+
+	include Rex::Ui::Text::Color
+
+	def initialize
+		@config = {
+			:color => :auto, # true, false, :auto
+		}
+		super
+	end
+	attr_reader :config
+
+	def disable_color
+		@config[:color] = false
+	end
+
+	def enable_color
+		@config[:color] = true
+	end
+
+	def auto_color
+		@config[:color] = :auto
+	end
+
+	def update_prompt(prompt = nil)
+		return if prompt.nil?
+		substitute_colors(prompt, true)
+	end
+
+	def print_error(msg = '')
+		print_line("%bld%red[-]%clr #{msg}")
+	end
+
+	def print_good(msg = '')
+		print_line("%bld%grn[+]%clr #{msg}")
+	end
+
+	def print_debug(msg = '')
+		print_line("%bld%cya[!]%clr #{msg}")
+	end
+
+	def print_status(msg = '')
+		print_line("%bld%blu[*]%clr #{msg}")
+	end
+
+	def print_line(msg = '')
+		print(msg + "\n")
+	end
+
+	def print(msg = '')
+		print_raw(substitute_colors(msg))
+	end
+
+	def reset
+	end
+
+end
+
+end
+end
+end
+