dwelch-r7
64 lines
937 B
Ruby
64 lines
937 B
Ruby
# -*- coding: binary -*-
|
|
|
|
|
|
module Rex
|
|
module Logging
|
|
module Sinks
|
|
|
|
###
|
|
#
|
|
# This class implements the LogSink interface and backs it against a stream
|
|
###
|
|
class Stream
|
|
|
|
include Rex::Logging::LogSink
|
|
|
|
def initialize(stream)
|
|
@stream = stream
|
|
end
|
|
|
|
#
|
|
# Writes log data to a stream
|
|
#
|
|
def log(sev, src, level, msg) # :nodoc:
|
|
if sev == LOG_RAW
|
|
stream.write(msg)
|
|
else
|
|
stream.write("[#{get_current_timestamp}] [#{log_code_for(sev)}(#{level})] #{src}: #{msg}\n")
|
|
end
|
|
|
|
stream.flush
|
|
end
|
|
|
|
def cleanup # :nodoc:
|
|
stream.close
|
|
end
|
|
|
|
protected
|
|
|
|
attr_accessor :stream # :nodoc:
|
|
|
|
#
|
|
# This method returns the corresponding log code for the given severity
|
|
#
|
|
def log_code_for(sev)
|
|
code = 'i'
|
|
|
|
case sev
|
|
when LOG_DEBUG
|
|
code = 'd'
|
|
when LOG_ERROR
|
|
code = 'e'
|
|
when LOG_INFO
|
|
code = 'i'
|
|
when LOG_WARN
|
|
code = 'w'
|
|
end
|
|
|
|
code
|
|
end
|
|
|
|
end
|
|
|
|
end end end
|