risu 1.4.4 → 1.4.5

data/lib/risu/templates/graphs.rb

@@ -1,33 +1,60 @@
-text Report.classification, :align => :center
-text "\n"
+module Risu
+	module Modules
+		class Graphs < Risu::Base::TemplateBase
 
-font_size(24) { text Report.title, :align => :center }
-font_size(18) { 
-	text "This report was prepared by\n#{Report.author}", :align => :center
-}
+			#
+			#
+			def initialize ()
+				@template_info =
+				{
+					:name => "graphs",
+					:author => "hammackj",
+					:version => "0.0.1",
+					:description => "Generates a report with all the graphs in it",
+					:scanner => "Nessus"
+				}
+			end
 
-text "\n\n\n"
+			#
+			#
+			def render(output)
+				output.text Report.classification, :align => :center
+				output.text "\n"
 
-start_new_page
+				output.font_size(24) do
+					output.text Report.title, :align => :center
+				end
 
-image Item.risks_by_severity_graph, :width => 500, :height => 375, :position => :center
+				output.font_size(18) do
+					output.text "This report was prepared by\n#{Report.author}", :align => :center
+				end
 
-start_new_page
+				output.text "\n\n\n"
 
-image Item.risks_by_service_graph(10), :width => 500, :height => 375, :position => :center
+				output.start_new_page
 
-start_new_page
+				output.image Item.risks_by_severity_graph, :width => 500, :height => 375, :position => :center
 
-image Plugin.top_by_count_graph(10), :width => 500, :height => 375, :position => :center
+				output.start_new_page
 
-start_new_page
+				output.image Item.risks_by_service_graph(10), :width => 500, :height => 375, :position => :center
 
-image Host.top_vuln_graph(10), :width => 500, :height => 375, :position => :center
+				output.start_new_page
 
-start_new_page
+				output.image Plugin.top_by_count_graph(10), :width => 500, :height => 375, :position => :center
 
-image Host.other_os_graph, :width => 500, :height => 375, :position => :center
+				output.start_new_page
 
-start_new_page
+				output.image Host.top_vuln_graph(10), :width => 500, :height => 375, :position => :center
 
-image Host.windows_os_graph, :width => 500, :height => 375, :position => :center
+				output.start_new_page
+
+				output.image Host.other_os_graph, :width => 500, :height => 375, :position => :center
+
+				output.start_new_page
+
+				output.image Host.windows_os_graph, :width => 500, :height => 375, :position => :center
+			end
+		end
+	end
+end

data/lib/risu/templates/host_summary.rb

@@ -1,40 +1,63 @@
-text Report.classification, :align => :center
-text "\n"
-
-font_size(22) { text Report.title, :align => :center }
-font_size(18) {
-	text "Host Summary Report", :align => :center
-	text "\n"
-	text "This report was prepared by\n#{Report.author}", :align => :center
-}
-
-text "\n\n\n"
-
-results = Array.new
-
-headers = ["Hostname", "Total", "High", "Medium", "Low", "Info"]
-header_widths = {0 => 137, 1 => 75, 2 => 75, 3 => 75, 4 => 75, 5 => 75}
-
-Host.sorted.each do |host|
-	row = Array.new
-
-	total  = Item.risks.where(:host_id => host.id).count
-	high = Item.high_risks.where(:host_id => host.id).count
-	medium = Item.medium_risks.where(:host_id => host.id).count
-	low = Item.low_risks.where(:host_id => host.id).count
-	info = Item.info_risks.where(:host_id => host.id).count
-
-	row.push(host.name)
-	row.push(total)
-	row.push(high)
-	row.push(medium)
-	row.push(low)
-	row.push(info)
-
-	results.push(row)
-end
-
-table([headers] + results, :header => true, :column_widths => header_widths, :row_colors => ['ffffff', 'E5E5E5']) do
-	row(0).style(:font_style => :bold, :background_color => 'D0D0D0')
-	cells.borders = [:top, :bottom, :left, :right]
+module Risu
+	module Modules
+		class HostSummary < Risu::Base::TemplateBase
+			
+			#
+			#
+			def initialize ()
+				@template_info = 
+				{ 
+					:name => "host_summary", 
+					:author => "hammackj", 
+					:version => "0.0.1", 
+					:description => "Generates a Host Summary Report"
+				}
+			end
+			
+			#
+			#
+			def render(output)
+				output.text Report.classification, :align => :center
+				output.text "\n"
+
+				output.font_size(22) { output.text Report.title, :align => :center }
+				output.font_size(18) {
+					output.text "Host Summary Report", :align => :center
+					output.text "\n"
+					output.text "This report was prepared by\n#{Report.author}", :align => :center
+				}
+
+				output.text "\n\n\n"
+
+				results = Array.new
+
+				headers = ["Hostname", "Total", "High", "Medium", "Low", "Info"]
+				header_widths = {0 => 137, 1 => 75, 2 => 75, 3 => 75, 4 => 75, 5 => 75}
+
+				Host.sorted.each do |host|
+					row = Array.new
+
+					total  = Item.risks.where(:host_id => host.id).count
+					high = Item.high_risks.where(:host_id => host.id).count
+					medium = Item.medium_risks.where(:host_id => host.id).count
+					low = Item.low_risks.where(:host_id => host.id).count
+					info = Item.info_risks.where(:host_id => host.id).count
+
+					row.push(host.name)
+					row.push(total)
+					row.push(high)
+					row.push(medium)
+					row.push(low)
+					row.push(info)
+
+					results.push(row)
+				end
+
+				output.table([headers] + results, :header => true, :column_widths => header_widths, :row_colors => ['ffffff', 'E5E5E5']) do
+					row(0).style(:font_style => :bold, :background_color => 'D0D0D0')
+					cells.borders = [:top, :bottom, :left, :right]
+				end
+			end
+		end
+	end
 end

data/lib/risu/templates/ms_patch_summary.rb

@@ -1,37 +1,61 @@
-text Report.classification, :align => :center
-text "\n"
-
-font_size(22) { text Report.title, :align => :center }
-font_size(18) { 
-	text "Missing Microsoft Patch Summary", :align => :center
-	text "\n"
-	text "This report was prepared by\n#{Report.author}", :align => :center
-}
-
-text "\n\n\n"
-
-Item.ms_patches.each do |item|
-	host = Host.find_by_id(item.host_id)
-	
-	if host == nil
-		next
-	end
-	
-	if host.name != nil
-		text "Host:", :style => :bold 
-		text host.name
-	end
-	
-	if host.os != nil
-		text "OS:", :style => :bold
-		text host.os
-	end
-	
-	if host.mac != nil
-		text "Mac:", :style => :bold
-		text host.mac
+module Risu
+	module Modules
+		class MSPatchSummary < Risu::Base::TemplateBase
+			
+			#
+			#
+			def initialize ()
+				@template_info = 
+				{ 
+					:name => "ms_patch_summary", 
+					:author => "hammackj", 
+					:version => "0.0.1", 
+					:description => "Generates a Microsoft Patch Summary Report"
+				}
+			end
+			
+			#
+			#
+			def render(output)
+				output.text Report.classification, :align => :center
+				output.text "\n"
+
+				output.font_size(22) { output.text Report.title, :align => :center }
+				output.font_size(18) { 
+					output.text "Missing Microsoft Patch Summary", :align => :center
+					output.text "\n"
+					output.text "This report was prepared by\n#{Report.author}", :align => :center
+				}
+
+				output.text "\n\n\n"
+
+				Item.ms_patches.each do |item|
+					host = Host.find_by_id(item.host_id)
+
+					if host == nil
+						next
+					end
+
+					if host.name != nil
+						output.text "Host:", :style => :bold 
+						output.text host.name
+					end
+
+					if host.os != nil
+						output.text "OS:", :style => :bold
+						output.text host.os
+					end
+
+					if host.mac != nil
+						output.text "Mac:", :style => :bold
+						output.text host.mac
+					end
+					output.text "\n"
+					output.text item.plugin_output
+					output.text "\n"
+				end
+
+			end
+		end
 	end
-	text "\n"
-	text item.plugin_output
-	text "\n"
 end

data/lib/risu/templates/ms_update_summary.rb

@@ -1,43 +1,67 @@
-text Report.classification, :align => :center
-text "\n"
+module Risu
+	module Modules
+		class MSUpdateSummary < Risu::Base::TemplateBase
+			
+			#
+			#
+			def initialize ()
+				@template_info = 
+				{ 
+					:name => "ms_update_summary", 
+					:author => "hammackj", 
+					:version => "0.0.1", 
+					:description => "Generates a Microsoft Update Summary Report"
+				}
+			end
+			
+			#
+			#
+			def render(output)
+				output.text Report.classification, :align => :center
+				output.text "\n"
 
-font_size(22) { text Report.title, :align => :center }
-font_size(18) { 
-	text "Microsoft Update Summary", :align => :center
-	text "\n"
-	text "This report was prepared by\n#{Report.author}", :align => :center
-}
+				output.font_size(22) { output.text Report.title, :align => :center }
+				output.font_size(18) { 
+					output.text "Microsoft Update Summary", :align => :center
+					output.text "\n"
+					output.text "This report was prepared by\n#{Report.author}", :align => :center
+				}
 
-text "\n\n\n"
+				output.text "\n\n\n"
 
-font_size(12)
+				output.font_size(12)
 
-results = Array.new
+				results = Array.new
 
-headers = ["Hostname","Operating System", "Windows Update Status"]
-header_widths = {0 => 108, 1 => 264, 2 => 140}
+				headers = ["Hostname","Operating System", "Windows Update Status"]
+				header_widths = {0 => 108, 1 => 264, 2 => 140}
 
-Item.ms_update.each do |item|
-	host = Host.find_by_id(item.host_id)
-	
-	if host == nil
-		next
-	end
-	
-	row = Array.new
-	row.push(host.name)
-	row.push(host.os)
-	
-	if item.plugin_output =~ /'Automatic Updates' are disabled/
-		row.push("Disabled")
-	else
-		row.push("Enabled")
-	end
-	
-	results.push(row)
-end
+				Item.ms_update.each do |item|
+					host = Host.find_by_id(item.host_id)
+
+					if host == nil
+						next
+					end
+
+					row = Array.new
+					row.push(host.name)
+					row.push(host.os)
 
-table([headers] + results, :header => true, :column_widths => header_widths, :row_colors => ['ffffff', '336699']) do
-	row(0).style(:font_style => :bold, :background_color => 'cccccc')
-	cells.borders = [:top, :bottom, :left, :right]
+					if item.plugin_output =~ /'Automatic Updates' are disabled/
+						row.push("Disabled")
+					else
+						row.push("Enabled")
+					end
+
+					results.push(row)
+				end
+
+				output.table([headers] + results, :header => true, :column_widths => header_widths, :row_colors => ['ffffff', '336699']) do
+					row(0).style(:font_style => :bold, :background_color => 'cccccc')
+					cells.borders = [:top, :bottom, :left, :right]
+				end
+
+			end
+		end
+	end
 end

data/lib/risu/templates/pci_compliance.rb

@@ -1,66 +1,90 @@
-text Report.classification, :align => :center
-text "\n"
-
-font_size(22) { text Report.title, :align => :center }
-font_size(18) { 
-	text "PCI /DSS Complience Overview", :align => :center
-	text "\n"
-	text "This report was prepared by\n#{Report.author}", :align => :center
-}
-
-text "\n\n\n"
-
-@hosts_count = Host.find(:all, :conditions => ["pci_dss_compliance is not null"]).count
-@hosts_passed = Host.find(:all, :conditions => ["pci_dss_compliance like 'passed'"])
-@hosts_failed = Host.find(:all, :conditions => ["pci_dss_compliance like 'failed'"])
-
-font_size(20) {
-	text "Summary\n", :style => :bold
-}
-
-text "Of #{@hosts_count} total hosts, #{@hosts_passed.count} passed and #{@hosts_failed.count} failed."
-
-text "\n\n"
-
-if @hosts_passed.length > 0
-	font_size(20) { 
-		fill_color "00FF00"
-		text "PCI / DSS Compliant Hosts", :style => :bold
-		fill_color "000000"
-		}
-	
-	text "\n"
-	
-	@hosts_passed.each do |host|	
-		text "#{host.ip} / #{host.fqdn} - passed\n"
-	end unless @hosts_passed == nil
-	
-	start_new_page
-end
+module Risu
+	module Modules
+		class PCICompliance < Risu::Base::TemplateBase
+			
+			#
+			#
+			def initialize ()
+				@template_info = 
+				{ 
+					:name => "pci_compliance", 
+					:author => "hammackj", 
+					:version => "0.0.1", 
+					:description => "Generates a PCI Compliance Overview Report"
+				}
+			end
+			
+			#
+			#
+			def render(output)
+				output.text Report.classification, :align => :center
+				output.text "\n"
+
+				output.font_size(22) { output.text Report.title, :align => :center }
+				output.font_size(18) { 
+					output.text "PCI /DSS Complience Overview", :align => :center
+					output.text "\n"
+					output.text "This report was prepared by\n#{Report.author}", :align => :center
+				}
+
+				output.text "\n\n\n"
+
+				@hosts_count = Host.find(:all, :conditions => ["pci_dss_compliance is not null"]).count
+				@hosts_passed = Host.find(:all, :conditions => ["pci_dss_compliance like 'passed'"])
+				@hosts_failed = Host.find(:all, :conditions => ["pci_dss_compliance like 'failed'"])
+
+				output.font_size(20) {
+					output.text "Summary\n", :style => :bold
+				}
+
+				output.text "Of #{@hosts_count} total hosts, #{@hosts_passed.count} passed and #{@hosts_failed.count} failed."
+
+				output.text "\n\n"
+
+				if @hosts_passed.length > 0
+					output.font_size(20) { 
+						output.fill_color "00FF00"
+						output.text "PCI / DSS Compliant Hosts", :style => :bold
+						output.fill_color "000000"
+						}
+
+					output.text "\n"
+
+					@hosts_passed.each do |host|	
+						output.text "#{host.ip} / #{host.fqdn} - passed\n"
+					end unless @hosts_passed == nil
+
+					output.start_new_page
+				end
+
+				if @hosts_failed.length > 0
+					output.font_size(20) { 
+						output.fill_color "FF0000"
+						output.text "Non PCI / DSS Compliant Hosts", :style => :bold 
+						output.fill_color "000000"
+						}
+
+					output.text "\n"
+
+					@hosts_failed.each do |host|	
+							host_id = host.id
+							plugin = Plugin.find(:first, :conditions => { :id => 33929 })
+							item = Item.find(:first, :conditions => { :host_id => host_id, :plugin_id => plugin.id })
+
+							output.text "#{host.ip} / #{host.fqdn} - failed\n", :style => :bold
+							output.text "Description:\n", :style => :bold
+							output.text "#{plugin.description}\n"
+							output.text "Plugin Output:\n", :style => :bold
+							output.text "#{item.plugin_output}\n"
+
+							output.text "\n"
+
+					end unless @hosts_failed == nil
+
+					output.start_new_page
+				end
 
-if @hosts_failed.length > 0
-	font_size(20) { 
-		fill_color "FF0000"
-		text "Non PCI / DSS Compliant Hosts", :style => :bold 
-		fill_color "000000"
-		}
-	
-	text "\n"
-	
-	@hosts_failed.each do |host|	
-			host_id = host.id
-			plugin = Plugin.find(:first, :conditions => { :id => 33929 })
-			item = Item.find(:first, :conditions => { :host_id => host_id, :plugin_id => plugin.id })
-
-			text "#{host.ip} / #{host.fqdn} - failed\n", :style => :bold
-			text "Description:\n", :style => :bold
-			text "#{plugin.description}\n"
-			text "Plugin Output:\n", :style => :bold
-			text "#{item.plugin_output}\n"
-
-			text "\n"
-
-	end unless @hosts_failed == nil
-	
-	start_new_page
+			end
+		end
+	end
 end