Land #8612, RSS feed plugin

<3 @mubix

plugins/rssfeed.rb

@@ -0,0 +1,120 @@
+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)
+    self.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
+      begin
+      while(true)
+        while(event = self.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
+  end
+
+  def stop_event_queue
+    self.queue_thread.kill if self.queue_thread
+    self.queue_thread = nil
+    self.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
+
+    self.on_plugin_load
+  end
+
+  def cleanup
+    self.on_plugin_unload
+    self.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