From 033da6245a3f8b680bf2897b152843b1207ad9b7 Mon Sep 17 00:00:00 2001
From: HD Moore <hd_moore@rapid7.com>
Date: Sat, 16 Sep 2006 06:27:14 +0000
Subject: [PATCH] Display an ugly warning message if a broken module is found
 in the tree

git-svn-id: file:///home/svn/framework3/trunk@3896 4d416f70-5f16-0410-b530-b9f4589650da
---
 lib/msf/core/module_manager.rb | 19 +++++++++++++++++--
 lib/msf/ui/console/driver.rb   |  9 +++++++++
 2 files changed, 26 insertions(+), 2 deletions(-)

diff --git a/lib/msf/core/module_manager.rb b/lib/msf/core/module_manager.rb
index fbeeced037..5f0fc1ce3d 100644
--- a/lib/msf/core/module_manager.rb
+++ b/lib/msf/core/module_manager.rb
@@ -316,6 +316,7 @@ class ModuleManager < ModuleSet
 		self.module_history       = {}
 		self.module_history_mtime = {}
 		self.module_sets          = {}
+		self.module_failed        = {}
 		self.framework            = framework
 
 		MODULE_TYPES.each { |type|
@@ -391,6 +392,13 @@ class ModuleManager < ModuleSet
 		return module_sets[MODULE_AUX]
 	end
 
+	#
+	# Returns the set of modules that failed to load.
+	#
+	def failed
+		return module_failed
+	end
+	
 	##
 	#
 	# Module cache management to support demand loaded modules.
@@ -627,9 +635,11 @@ class ModuleManager < ModuleSet
 			begin
 				if (!load(mod.file_path))
 					elog("Failed to load module from #{mod.file_path}")
+					self.module_failed[mod.file_path] = "Failed to reload the module"
 					return nil
 				end
-
+			self.module_failed.delete(mod.file_path)
+			
 			rescue
 				elog("Failed to reload module #{mod} from #{mod.file_path}: #{$!}")
 				raise $!
@@ -776,6 +786,7 @@ protected
 					# Keep scanning since we just successfully loaded a delay load
 					# module.
 					ks = true
+
 				# Trap the name error and flag this file path as still needing to
 				# be delay loaded.
 				rescue NameError
@@ -787,6 +798,7 @@ protected
 		# Log all modules that failed to load due to problems
 		delay.each_pair { |file, err|
 			elog("Failed to load module from #{file}: #{err}")
+			self.module_failed[file] = err
 		}
 
 		# Perform any required recalculations for the individual module types
@@ -895,6 +907,7 @@ protected
 				added = module_history[file]
 			else
 				elog("Loaded #{file} but no classes were added.")
+				self.module_failed[file] = "Loaded file, but no classes were registered"
 				return false
 			end
 		else
@@ -921,7 +934,8 @@ protected
 		end
 
 		ilog("Loaded #{type} module #{added} from #{file}.", 'core', LEV_2)
-
+		self.module_failed.delete(file)
+		
 		# Do some processing on the loaded module to get it into the
 		# right associations
 		on_module_load(added, type, name, {
@@ -1038,6 +1052,7 @@ protected
 	attr_accessor :modules, :module_sets # :nodoc:
 	attr_accessor :module_paths # :nodoc:
 	attr_accessor :module_history, :module_history_mtime # :nodoc:
+	attr_accessor :module_failed # :nodoc:
 
 end
 
diff --git a/lib/msf/ui/console/driver.rb b/lib/msf/ui/console/driver.rb
index 0744a861f4..a1fa5ad634 100644
--- a/lib/msf/ui/console/driver.rb
+++ b/lib/msf/ui/console/driver.rb
@@ -192,6 +192,15 @@ class Driver < Msf::Ui::Driver
 	# displayed, scripts can be processed, and other fun can be had.
 	#
 	def on_startup
+		# Check for modules that failed to load
+		if (framework.modules.failed.length > 0)
+			print("[*] WARNING! The following modules could not be loaded!\n\n")
+			framework.modules.failed.each_pair do |file, err|
+				print("\t#{file}: #{err.to_s}\n\n")
+			end
+			print("\n")
+		end
+		
 		# Build the banner message
 		run_single("banner")
 	end