require 'rex' require 'msf/base' module Msf ### # # This module provides an initialization interface for logging. # ### class Logging # # Initialize logging. # def self.init if (@initialized != true) @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