add RSS feed plugin

Author: mubix

plugins/rssfeed.rb

@@ -0,0 +1,123 @@
+#
+# $Id$
+# $Revision$
+#
+
+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)
+    status = "created" ? select(nil, nil, nil, 25) : ""
+    title = "#{session.type} session - #{session.sid} #{status}."
+    content = ""
+    session.workspace ? content << "Workspace:\t#{session.workspace}\n" : ""
+    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 = Thread.new 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 = []
+  end
+
+
+  def initialize(framework, opts)
+    require 'rss'
+    super
+
+    @items = []
+    self.queue = []
+    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
+
+