cgranleese-r7
89 lines
2.3 KiB
Ruby
89 lines
2.3 KiB
Ruby
# -*- coding: binary -*-
|
|
|
|
require 'rex/post/sql/ui/console/command_dispatcher/client'
|
|
require 'rex/post/sql/ui/console/command_dispatcher/core'
|
|
|
|
module Rex
|
|
module Post
|
|
module Sql
|
|
module Ui
|
|
module Console
|
|
|
|
###
|
|
#
|
|
# Base class for all command dispatchers within the Generic SQL console user interface.
|
|
#
|
|
###
|
|
module CommandDispatcher
|
|
|
|
include Msf::Ui::Console::CommandDispatcher::Session
|
|
|
|
#
|
|
# Initializes an instance of the core command set using the supplied session and client
|
|
# for interactivity.
|
|
#
|
|
# @param [Rex::Post::PostgreSQL::Ui::Console] console
|
|
def initialize(console)
|
|
super
|
|
@msf_loaded = nil
|
|
@filtered_commands = []
|
|
end
|
|
|
|
#
|
|
# Returns the SQL client context.
|
|
# @return [Object]
|
|
def client
|
|
console = shell
|
|
console.client
|
|
end
|
|
|
|
#
|
|
# Returns the PostgreSQL session context.
|
|
#
|
|
# @return [Msf::Sessions::PostgreSQL]
|
|
def session
|
|
console = shell
|
|
console.session
|
|
end
|
|
|
|
#
|
|
# Returns the commands that meet the requirements
|
|
#
|
|
def filter_commands(all, reqs)
|
|
all.delete_if do |cmd, _desc|
|
|
if reqs[cmd]&.any? { |req| !client.commands.include?(req) }
|
|
@filtered_commands << cmd
|
|
true
|
|
end
|
|
end
|
|
end
|
|
|
|
def unknown_command(cmd, line)
|
|
if @filtered_commands.include?(cmd)
|
|
print_error("The \"#{cmd}\" command is not supported by this session type (#{session.session_type})")
|
|
return :handled
|
|
end
|
|
|
|
super
|
|
end
|
|
|
|
#
|
|
# Returns true if the client has a framework object.
|
|
#
|
|
# Used for firing framework session events
|
|
#
|
|
def msf_loaded?
|
|
return @msf_loaded unless @msf_loaded.nil?
|
|
|
|
# if we get here we must not have initialized yet
|
|
|
|
@msf_loaded = !session.framework.nil?
|
|
@msf_loaded
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|