dwelch-r7
143 lines
2.9 KiB
Ruby
143 lines
2.9 KiB
Ruby
# -*- coding: binary -*-
|
|
module Rex
|
|
module Post
|
|
module Meterpreter
|
|
module Ui
|
|
|
|
###
|
|
#
|
|
# Mixin that is meant to extend the base channel class from meterpreter in a
|
|
# manner that adds interactive capabilities.
|
|
#
|
|
###
|
|
module Console::InteractiveChannel
|
|
|
|
include Rex::Ui::Interactive
|
|
|
|
attr_accessor :raw
|
|
#
|
|
# Interacts with self.
|
|
#
|
|
def _interact
|
|
# If the channel has a left-side socket, then we can interact with it.
|
|
if (self.lsock)
|
|
self.interactive(true)
|
|
if raw
|
|
update_term_size
|
|
_local_fd.raw do
|
|
interact_stream(self)
|
|
end
|
|
else
|
|
interact_stream(self)
|
|
end
|
|
|
|
self.interactive(false)
|
|
else
|
|
print_error("Channel #{self.cid} does not support interaction.")
|
|
|
|
self.interacting = false
|
|
end
|
|
end
|
|
|
|
#
|
|
# Called when an interrupt is sent.
|
|
#
|
|
def _interrupt
|
|
prompt_yesno("Terminate channel #{self.cid}?")
|
|
end
|
|
|
|
#
|
|
# Suspends interaction with the channel.
|
|
#
|
|
def _suspend
|
|
# Ask the user if they would like to background the session
|
|
if (prompt_yesno("Background channel #{self.cid}?") == true)
|
|
self.interactive(false)
|
|
|
|
self.interacting = false
|
|
end
|
|
end
|
|
|
|
#
|
|
# Closes the channel like it aint no thang.
|
|
#
|
|
def _interact_complete
|
|
begin
|
|
self.interactive(false)
|
|
|
|
self.close
|
|
rescue IOError
|
|
end
|
|
end
|
|
|
|
#
|
|
# Reads data from local input and writes it remotely.
|
|
#
|
|
def _stream_read_local_write_remote(channel)
|
|
if raw
|
|
data = user_input.sysread(1024)
|
|
else
|
|
data = user_input.gets
|
|
end
|
|
return if not data
|
|
|
|
self.on_command_proc.call(data.strip) if self.on_command_proc
|
|
self.write(data)
|
|
end
|
|
|
|
#
|
|
# Reads from the channel and writes locally.
|
|
#
|
|
def _stream_read_remote_write_local(channel)
|
|
data = self.lsock.sysread(16384)
|
|
|
|
self.on_print_proc.call(data.strip) if self.on_print_proc
|
|
self.on_log_proc.call(data.strip) if self.on_log_proc
|
|
user_output.print(data)
|
|
end
|
|
|
|
#
|
|
# Returns the remote file descriptor to select on
|
|
#
|
|
def _remote_fd(stream)
|
|
self.lsock
|
|
end
|
|
|
|
attr_accessor :rows
|
|
attr_accessor :cols
|
|
|
|
def _winch
|
|
update_term_size
|
|
end
|
|
|
|
def update_term_size
|
|
return unless self.client.commands.include?(Extensions::Stdapi::COMMAND_ID_STDAPI_SYS_PROCESS_SET_TERM_SIZE)
|
|
rows, cols = ::IO.console.winsize
|
|
unless rows == self.rows && cols == self.cols
|
|
set_term_size(rows, cols)
|
|
self.rows = rows
|
|
self.cols = cols
|
|
end
|
|
end
|
|
|
|
def set_term_size(rows, columns)
|
|
if self.cid.nil?
|
|
raise IOError, 'Channel has been closed.', caller
|
|
end
|
|
|
|
request = Packet.create_request(Extensions::Stdapi::COMMAND_ID_STDAPI_SYS_PROCESS_SET_TERM_SIZE)
|
|
request.add_tlv(TLV_TYPE_CHANNEL_ID, self.cid)
|
|
request.add_tlv(Extensions::Stdapi::TLV_TYPE_TERMINAL_ROWS, rows)
|
|
request.add_tlv(Extensions::Stdapi::TLV_TYPE_TERMINAL_COLUMNS, columns)
|
|
self.client.send_packet(request)
|
|
end
|
|
|
|
attr_accessor :on_log_proc
|
|
|
|
end
|
|
|
|
end
|
|
end
|
|
end
|
|
end
|