156 lines
2.4 KiB
Ruby
156 lines
2.4 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
|
|
|
|
class WebConsolePipe < Rex::IO::BidirectionalPipe
|
|
|
|
attr_accessor :input
|
|
attr_accessor :output
|
|
attr_accessor :prompt
|
|
attr_accessor :killed
|
|
|
|
def intrinsic_shell?
|
|
true
|
|
end
|
|
|
|
def supports_readline
|
|
false
|
|
end
|
|
|
|
def _print_prompt
|
|
end
|
|
|
|
|
|
#
|
|
# Wrapper methods around input pipe
|
|
#
|
|
|
|
|
|
def close
|
|
self.pipe_input.close
|
|
end
|
|
|
|
def put(*args)
|
|
self.pipe_input.put(*args)
|
|
end
|
|
|
|
def gets
|
|
self.pipe_input.gets
|
|
end
|
|
|
|
def pgets
|
|
self.pipe_input.gets
|
|
end
|
|
|
|
def eof?
|
|
self.pipe_input.eof?
|
|
end
|
|
|
|
def fd(*args)
|
|
# Remove the following line to enable full sessions via the console
|
|
# We really should just hook the on_session() instead...
|
|
raise ::RuntimeError, "Session interaction should be performed via the Sessions tab"
|
|
self.pipe_input.fd(*args)
|
|
end
|
|
|
|
def sysread(*args)
|
|
self.pipe_input.sysread(*args)
|
|
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
|
|
self.pipe.input = self.pipe.pipe_input
|
|
|
|
# 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,
|
|
}
|
|
)
|
|
|
|
self.console.extend(WebConsoleShell)
|
|
|
|
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)
|
|
self.console.tab_complete(cmd)
|
|
end
|
|
|
|
def shutdown
|
|
self.pipe.killed = true
|
|
self.pipe.close
|
|
self.thread.kill
|
|
end
|
|
end
|
|
|
|
|
|
end
|
|
end
|
|
end
|