138 lines
3.3 KiB
Ruby
138 lines
3.3 KiB
Ruby
#
|
|
# $Id$
|
|
# $Revision$
|
|
#
|
|
|
|
module Msf
|
|
|
|
###
|
|
#
|
|
# This class illustrates a sample plugin. Plugins can change the behavior of
|
|
# the framework by adding new features, new user interface commands, or
|
|
# through any other arbitrary means. They are designed to have a very loose
|
|
# definition in order to make them as useful as possible.
|
|
#
|
|
###
|
|
class Plugin::ThreadTest < Msf::Plugin
|
|
|
|
###
|
|
#
|
|
# This class implements a sample console command dispatcher.
|
|
#
|
|
###
|
|
class ConsoleCommandDispatcher
|
|
include Msf::Ui::Console::CommandDispatcher
|
|
|
|
#
|
|
# The dispatcher's name.
|
|
#
|
|
def name
|
|
"ThreadTest"
|
|
end
|
|
|
|
#
|
|
# Returns the hash of commands supported by this dispatcher.
|
|
#
|
|
def commands
|
|
{
|
|
"start_thread" => "Start a background thread that writes to the console",
|
|
"stop_thread" => "Stop a background thread",
|
|
"list_thread" => "List running threads"
|
|
}
|
|
end
|
|
|
|
def cmd_start_thread(*args)
|
|
if (@mythread)
|
|
print_line("Test thread is already running")
|
|
return
|
|
end
|
|
|
|
@mythread = ::Thread.new {
|
|
while(true)
|
|
print_line("--- test thread ---")
|
|
select(nil, nil, nil, 5)
|
|
end
|
|
}
|
|
print_line("Test thread created")
|
|
end
|
|
|
|
def cmd_stop_thread(*args)
|
|
if (! @mythread)
|
|
print_line("No test thread is running")
|
|
return
|
|
end
|
|
|
|
@mythread.kill
|
|
@mythread = nil
|
|
print_line("Test thread stopped")
|
|
end
|
|
|
|
def cmd_list_thread(*args)
|
|
Thread.list.each do |t|
|
|
print_line(sprintf("Thread: 0x%.8x (%s/%d) (%s)", t.object_id, t.status, t.priority, t.tsource))
|
|
print_line("")
|
|
end
|
|
end
|
|
end
|
|
|
|
#
|
|
# The constructor is called when an instance of the plugin is created. The
|
|
# framework instance that the plugin is being associated with is passed in
|
|
# the framework parameter. Plugins should call the parent constructor when
|
|
# inheriting from Msf::Plugin to ensure that the framework attribute on
|
|
# their instance gets set.
|
|
#
|
|
def initialize(framework, opts)
|
|
super
|
|
|
|
# If this plugin is being loaded in the context of a console application
|
|
# that uses the framework's console user interface driver, register
|
|
# console dispatcher commands.
|
|
add_console_dispatcher(ConsoleCommandDispatcher)
|
|
|
|
# Extend the thread to track the calling source
|
|
Thread.class_eval("
|
|
attr_accessor :tsource
|
|
|
|
alias initialize_old initialize
|
|
|
|
def initialize(&block)
|
|
self.tsource = caller(1)
|
|
initialize_old(&block)
|
|
end
|
|
")
|
|
|
|
print_status("ThreadTest plugin loaded.")
|
|
end
|
|
|
|
#
|
|
# The cleanup routine for plugins gives them a chance to undo any actions
|
|
# they may have done to the framework. For instance, if a console
|
|
# dispatcher was added, then it should be removed in the cleanup routine.
|
|
#
|
|
def cleanup
|
|
# If we had previously registered a console dispatcher with the console,
|
|
# deregister it now.
|
|
remove_console_dispatcher('ThreadTest')
|
|
end
|
|
|
|
#
|
|
# This method returns a short, friendly name for the plugin.
|
|
#
|
|
def name
|
|
"threadtest"
|
|
end
|
|
|
|
#
|
|
# This method returns a brief description of the plugin. It should be no
|
|
# more than 60 characters, but there are no hard limits.
|
|
#
|
|
def desc
|
|
"Thread testing plugin"
|
|
end
|
|
|
|
protected
|
|
end
|
|
|
|
end
|