116 lines
2.8 KiB
Ruby
116 lines
2.8 KiB
Ruby
module Msf
|
|
###
|
|
#
|
|
# This class hooks all session events and puts it into an RSS feed
|
|
#
|
|
###
|
|
|
|
class Plugin::EventRSS < Msf::Plugin
|
|
|
|
attr_accessor :items, :queue, :queue_thread
|
|
|
|
include Msf::SessionEvent
|
|
|
|
def add_event(event)
|
|
queue.push(event)
|
|
end
|
|
|
|
def generate_feed(newitem)
|
|
items.unshift(newitem)
|
|
feed = RSS::Maker.make('atom') do |maker|
|
|
maker.channel.author = 'msfconsole'
|
|
maker.channel.updated = Time.new.to_s
|
|
maker.channel.about = 'https://metasploit.com'
|
|
maker.channel.title = 'msfconsole rss feed'
|
|
|
|
items.each do |rssitem|
|
|
maker.items.new_item do |item|
|
|
item.link = rssitem[:link]
|
|
item.title = rssitem[:title]
|
|
item.updated = rssitem[:date]
|
|
item.summary = rssitem[:content]
|
|
end
|
|
end
|
|
end
|
|
File.open('feed.rss', 'w') { |f| f.write(feed) }
|
|
end
|
|
|
|
def create_session_item(session, status)
|
|
if status == 'created'
|
|
select(nil, nil, nil, 25)
|
|
end
|
|
title = "#{session.type} session - #{session.sid} #{status}."
|
|
content = ''
|
|
if session.workspace
|
|
content << "Workspace:\t#{session.workspace}\n"
|
|
end
|
|
content << "Session Information: #{session.info}"
|
|
add_event({ title: title, date: Time.now.to_s, link: 'https://metasploit.com', content: content })
|
|
end
|
|
|
|
def on_session_open(session)
|
|
create_session_item(session, 'created')
|
|
end
|
|
|
|
def on_session_close(session, _reason = '')
|
|
create_session_item(session, 'closed')
|
|
end
|
|
|
|
def on_session_fail(reason = ''); end
|
|
|
|
def on_plugin_load
|
|
add_event({ title: 'RSS Plugin Loaded', date: Time.now.to_s, link: 'https://metasploit.com/', content: 'N/A' })
|
|
end
|
|
|
|
def on_plugin_unload
|
|
generate_feed({ title: 'RSS Plugin Unloaded', date: Time.now.to_s, link: 'https:/metasploit.com/', content: 'N/A' })
|
|
end
|
|
|
|
def start_event_queue
|
|
self.queue_thread = Rex::ThreadFactory.spawn('rss_plugin', false) do
|
|
loop do
|
|
while (event = queue.shift)
|
|
generate_feed(event)
|
|
end
|
|
select(nil, nil, nil, 0.25)
|
|
end
|
|
rescue ::Exception => e
|
|
print_status("RSS plugin: fatal error #{e} #{e.backtrace}")
|
|
end
|
|
end
|
|
|
|
def stop_event_queue
|
|
queue_thread.kill if queue_thread
|
|
self.queue_thread = nil
|
|
queue.clear
|
|
end
|
|
|
|
def initialize(framework, opts)
|
|
require 'rss'
|
|
super
|
|
|
|
@items = []
|
|
self.queue = Queue.new
|
|
self.framework.events.add_session_subscriber(self)
|
|
start_event_queue
|
|
|
|
on_plugin_load
|
|
end
|
|
|
|
def cleanup
|
|
on_plugin_unload
|
|
framework.events.remove_session_subscriber(self)
|
|
stop_event_queue
|
|
end
|
|
|
|
def name
|
|
'rss'
|
|
end
|
|
|
|
def desc
|
|
'Create an RSS feed of events'
|
|
end
|
|
|
|
end
|
|
end
|