87 lines
1.2 KiB
Ruby
87 lines
1.2 KiB
Ruby
require 'msf/base'
|
|
|
|
module Msf
|
|
module Sessions
|
|
|
|
###
|
|
#
|
|
# This class provides basic interaction with a command shell on the remote
|
|
# endpoint. This session is initialized with a stream that will be used
|
|
# as the pipe for reading and writing the command shell.
|
|
#
|
|
###
|
|
class CommandShell
|
|
|
|
#
|
|
# This interface supports basic interaction.
|
|
#
|
|
include Msf::Session::Basic
|
|
|
|
#
|
|
# This interface supports interacting with a single command shell.
|
|
#
|
|
include Msf::Session::Provider::SingleCommandShell
|
|
|
|
#
|
|
# Returns the type of session.
|
|
#
|
|
def self.type
|
|
"shell"
|
|
end
|
|
|
|
#
|
|
# Returns the session description.
|
|
#
|
|
def desc
|
|
"Command shell"
|
|
end
|
|
|
|
def run_cmd(cmd)
|
|
write_shell(cmd)
|
|
return rstream.get
|
|
end
|
|
#
|
|
# Calls the class method.
|
|
#
|
|
def type
|
|
self.class.type
|
|
end
|
|
|
|
#
|
|
# The shell will have been initialized by default.
|
|
#
|
|
def init_shell
|
|
return true
|
|
end
|
|
|
|
#
|
|
# Read from the command shell.
|
|
#
|
|
def read_shell(length = nil)
|
|
if length.nil?
|
|
rv = rstream.get
|
|
else
|
|
rv = rstream.read(length)
|
|
end
|
|
return rv
|
|
end
|
|
|
|
#
|
|
# Writes to the command shell.
|
|
#
|
|
def write_shell(buf)
|
|
rstream.write(buf)
|
|
end
|
|
|
|
#
|
|
# Closes the shell.
|
|
#
|
|
def close_shell()
|
|
rstream.close
|
|
end
|
|
|
|
end
|
|
|
|
end
|
|
end
|