97 lines
1.8 KiB
Ruby
97 lines
1.8 KiB
Ruby
require 'rex'
|
|
require 'rex/logging'
|
|
require 'msf/base'
|
|
|
|
module Msf
|
|
|
|
###
|
|
#
|
|
# This module provides an initialization interface for logging.
|
|
#
|
|
###
|
|
class Logging
|
|
|
|
@@initialized = false
|
|
@@session_logging = false
|
|
|
|
#
|
|
# Initialize logging.
|
|
#
|
|
def self.init
|
|
if (! @@initialized)
|
|
@@initialized = true
|
|
|
|
f = Rex::Logging::Sinks::Flatfile.new(
|
|
Msf::Config.log_directory + File::SEPARATOR + "framework.log")
|
|
|
|
# Register each known log source
|
|
[
|
|
Rex::LogSource,
|
|
Msf::LogSource,
|
|
'base',
|
|
].each { |src|
|
|
register_log_source(src, f)
|
|
}
|
|
end
|
|
end
|
|
|
|
#
|
|
# Enables a log source.
|
|
#
|
|
def self.enable_log_source(src, level = 0)
|
|
if (log_source_registered?(src) == false)
|
|
f = Rex::Logging::Sinks::Flatfile.new(
|
|
Msf::Config.log_directory + File::SEPARATOR + "#{src}.log")
|
|
|
|
register_log_source(src, f, level)
|
|
end
|
|
end
|
|
|
|
#
|
|
# Stops logging for a given log source.
|
|
#
|
|
def self.disable_log_source(src)
|
|
deregister_log_source(src)
|
|
end
|
|
|
|
#
|
|
# Sets whether or not session logging is to be enabled.
|
|
#
|
|
def self.enable_session_logging(tf)
|
|
@@session_logging = tf
|
|
end
|
|
|
|
#
|
|
# Returns whether or not session logging is enabled.
|
|
#
|
|
def self.session_logging_enabled?
|
|
@@session_logging || false
|
|
end
|
|
|
|
#
|
|
# Starts logging for a given session.
|
|
#
|
|
def self.start_session_log(session)
|
|
if (log_source_registered?(session.log_source) == false)
|
|
f = Rex::Logging::Sinks::Flatfile.new(
|
|
Msf::Config.session_log_directory + File::SEPARATOR + "#{session.log_file_name}.log")
|
|
|
|
register_log_source(session.log_source, f)
|
|
|
|
rlog("\n[*] Logging started: #{Time.now}\n\n", session.log_source)
|
|
end
|
|
end
|
|
|
|
#
|
|
# Stops logging for a given session.
|
|
#
|
|
def self.stop_session_log(session)
|
|
rlog("\n[*] Logging stopped: #{Time.now}\n\n", session.log_source)
|
|
|
|
deregister_log_source(session.log_source)
|
|
end
|
|
|
|
end
|
|
|
|
end
|