156 lines
2.7 KiB
Ruby
156 lines
2.7 KiB
Ruby
module Msf
|
|
module Ui
|
|
module Web
|
|
|
|
###
|
|
#
|
|
# This class implements a console instance for use by the web interface
|
|
#
|
|
###
|
|
|
|
class WebConsole
|
|
attr_accessor :pipe
|
|
attr_accessor :console
|
|
attr_accessor :console_id
|
|
attr_accessor :last_access
|
|
attr_accessor :framework
|
|
attr_accessor :thread
|
|
|
|
# Wrapper class in case we need to extend the pipe
|
|
class WebConsolePipe < Rex::IO::BidirectionalPipe
|
|
def prompting?
|
|
false
|
|
end
|
|
end
|
|
|
|
#
|
|
# Provides some overrides for web-based consoles
|
|
#
|
|
module WebConsoleShell
|
|
|
|
def supports_color?
|
|
false
|
|
end
|
|
end
|
|
|
|
def initialize(framework, console_id)
|
|
# Configure the framework
|
|
self.framework = framework
|
|
|
|
# Configure the ID
|
|
self.console_id = console_id
|
|
|
|
# Create a new pipe
|
|
self.pipe = WebConsolePipe.new
|
|
|
|
# Create a read subscriber
|
|
self.pipe.create_subscriber('msfweb')
|
|
|
|
# Initialize the console with our pipe
|
|
self.console = Msf::Ui::Console::Driver.new(
|
|
'msf',
|
|
'>',
|
|
{
|
|
'Framework' => self.framework,
|
|
'LocalInput' => self.pipe,
|
|
'LocalOutput' => self.pipe,
|
|
'AllowCommandPassthru' => false,
|
|
'Resource' => [],
|
|
}
|
|
)
|
|
|
|
self.console.extend(WebConsoleShell)
|
|
self.console.block_command('irb')
|
|
|
|
self.thread = Thread.new { self.console.run }
|
|
|
|
update_access()
|
|
end
|
|
|
|
def update_access
|
|
self.last_access = Time.now
|
|
end
|
|
|
|
def read
|
|
update_access
|
|
self.pipe.read_subscriber('msfweb')
|
|
end
|
|
|
|
def write(buf)
|
|
update_access
|
|
self.pipe.write_input(buf)
|
|
end
|
|
|
|
def execute(cmd)
|
|
self.console.run_single(cmd)
|
|
end
|
|
|
|
def prompt
|
|
self.pipe.prompt
|
|
end
|
|
|
|
def tab_complete(cmd)
|
|
if(self.console.active_session)
|
|
return self.console.active_session.console.tab_complete(cmd)
|
|
end
|
|
self.console.tab_complete(cmd)
|
|
end
|
|
|
|
def shutdown
|
|
self.pipe.close
|
|
self.thread.kill
|
|
end
|
|
|
|
def busy
|
|
self.console.busy
|
|
end
|
|
|
|
def session_detach
|
|
if(self.console.active_session)
|
|
#background interactive meterpreter channel
|
|
if(self.console.active_session.respond_to?('channels'))
|
|
self.console.active_session.channels.each_value do |ch|
|
|
if(ch.respond_to?('interacting') && ch.interacting)
|
|
ch.detach()
|
|
return
|
|
end
|
|
end
|
|
end
|
|
#background session
|
|
self.console.active_session.completed = true
|
|
self.console.active_session.detach()
|
|
end
|
|
end
|
|
|
|
def session_kill
|
|
if(self.console.active_session)
|
|
#close interactive meterpreter channel
|
|
if(self.console.active_session.respond_to?('channels'))
|
|
self.console.active_session.channels.each_value do |ch|
|
|
if(ch.respond_to?('interacting') && ch.interacting)
|
|
ch._close
|
|
return
|
|
end
|
|
end
|
|
end
|
|
#kill session
|
|
self.console.active_session.kill()
|
|
end
|
|
end
|
|
|
|
def active_module
|
|
self.console.active_module
|
|
end
|
|
|
|
def active_module=(val)
|
|
self.console.active_module = val
|
|
end
|
|
|
|
end
|
|
|
|
|
|
end
|
|
end
|
|
end
|
|
|