risu 1.4.4 → 1.4.5

data/lib/risu/nessusdocument.rb

@@ -1,66 +0,0 @@
-# encoding: utf-8
-
-module Risu
-
-	# A Object to represet the Nessus xml file in memory
-	#
-	# @author Jacob Hammack <jacob.hammack@hammackj.com>
-	class NessusDocument
-
-		# Creates a instance of the NessusDocument class
-		#
-		def initialize document
-			@document = document
-		end
-
-		# Checks the validness of a NessusDocument
-		#
-		# @return [Boolean] True if valid, False if invalid
-		def valid?
-			if File.exist?(@document)
-				@parser = LibXML::XML::Parser.file @document
-				doc = @parser.parse
-
-				if doc.root.name == nil
-					return false
-				end
-				
-				if doc.root.name == "NessusClientData_v2"
-					return true
-				elsif doc.root.name == "NessusClientData"
-					return false
-				else
-					return false
-				end
-			else
-				return false
-			end
-		end
-
-		# Invokes the SAX parser on the XML document
-		#
-		def parse
-			@parser = LibXML::XML::SaxParser.file @document
-			@parser.callbacks = NessusSaxListener.new
-			@parser.parse
-		end
-
-		# Fixes the ip field if nil and replaces it with the name if its an ip
-		#
-		def fix_ips
-			@hosts = Host.all
-
-			@hosts.each do |host|
-				if host.ip == nil
-					begin
-						ip = IPAddr.new host.name
-						host.ip = ip.to_string
-						host.save
-					rescue ArgumentError => ae
-						next
-					end
-				end
-			end
-		end
-	end
-end

data/lib/risu/prawn_templater.rb

@@ -1,38 +0,0 @@
-# encoding: utf-8
-
-module Risu
-
-	# Templater class for generating a report from a erb template
-	#
-	# @author Jacob Hammack
-	class PrawnTemplater
-		attr_accessor :template, :template_source, :findings, :output_file
-		
-		# Setups of the Templater class initalizing all of the variables
-		#
-		# @return [PrawnTemplater] New Instance
-		def initialize(template, findings, output)
-			@template = template
-			@findings = findings
-			@output_file = output
-
-			@template_source = File.new(@template).read
-		end
-		
-		# Generates a report based on the erb template
-		#
-		# @return [String] html output of the erb template
-		def generate
-			begin				
-				source = @template_source
-				template = @template
-				Prawn::Document.generate(@output_file, :margin => [75, 50, 75, 50]) do
-					font_size 12
-					eval source
-				end
-			rescue => e
-				puts "Error: #{e.message} \n #{e.backtrace.join("\n\t")}\n"
-			end
-		end
-	end
-end