73 lines
1.3 KiB
Ruby
73 lines
1.3 KiB
Ruby
|
# -*- coding: binary -*-
|
||
|
|
|
||
|
|
module Msf
|
||
|
|
module Sessions
|
||
|
|
|
||
|
|
###
|
||
|
|
#
|
||
|
|
# This class provides the ability to receive a custom stage callback
|
||
|
|
#
|
||
|
|
###
|
||
|
|
class Custom
|
||
|
|
|
||
|
|
#
|
||
|
|
# This interface supports basic interaction.
|
||
|
|
#
|
||
|
|
include Msf::Session
|
||
|
|
include Msf::Session::Basic
|
||
|
|
|
||
|
|
attr_accessor :arch
|
||
|
|
attr_accessor :platform
|
||
|
|
|
||
|
|
#
|
||
|
|
# Returns the type of session.
|
||
|
|
#
|
||
|
|
def self.type
|
||
|
|
"custom"
|
||
|
|
end
|
||
|
|
|
||
|
|
def initialize(rstream, opts = {})
|
||
|
|
super
|
||
|
|
self.platform ||= ""
|
||
|
|
self.arch ||= ""
|
||
|
|
datastore = opts[:datastore]
|
||
|
|
end
|
||
|
|
|
||
|
|
def self.create_session(rstream, opts = {})
|
||
|
|
Msf::Sessions::Custom.new(rstream, opts)
|
||
|
|
end
|
||
|
|
|
||
|
|
def process_autoruns(datastore)
|
||
|
|
cleanup
|
||
|
|
end
|
||
|
|
|
||
|
|
def cleanup
|
||
|
|
print_good("Custom stage sent; session has been closed")
|
||
|
|
if rstream
|
||
|
|
# this is also a best-effort
|
||
|
|
rstream.close rescue nil
|
||
|
|
rstream = nil
|
||
|
|
end
|
||
|
|
end
|
||
|
|
|
||
|
|
#
|
||
|
|
# Returns the session description.
|
||
|
|
#
|
||
|
|
def desc
|
||
|
|
"Custom"
|
||
|
|
end
|
||
|
|
|
||
|
|
def self.can_cleanup_files
|
||
|
|
false
|
||
|
|
end
|
||
|
|
|
||
|
|
#
|
||
|
|
# Calls the class method
|
||
|
|
#
|
||
|
|
def type
|
||
|
|
self.class.type
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|