From 1055efbeaa8aea2a578f8dea141628fdeeed51ce Mon Sep 17 00:00:00 2001
From: Luke Imhoff <luke_imhoff@rapid7.com>
Date: Mon, 2 Jun 2014 12:32:54 -0500
Subject: [PATCH] Add module paths from paths['modules'] from Rails app and
 engines

MSP-9653

Allow rails engines (and other applications, like
Metasploit::Pro::Engine::Application) to define their own module paths
using the paths['modules'] entry for Rails Applications/Engines.
---
 lib/msf/base/simple/framework/module_paths.rb | 26 ++++++++++++++++---
 1 file changed, 23 insertions(+), 3 deletions(-)

diff --git a/lib/msf/base/simple/framework/module_paths.rb b/lib/msf/base/simple/framework/module_paths.rb
index 481068cf3c..e0977be756 100644
--- a/lib/msf/base/simple/framework/module_paths.rb
+++ b/lib/msf/base/simple/framework/module_paths.rb
@@ -9,9 +9,10 @@ module Msf
           # Ensure the module cache is accurate
           self.modules.refresh_cache_from_database
 
-          # Initialize the default module search paths
-          if (Msf::Config.module_directory)
-            self.modules.add_module_path(Msf::Config.module_directory, opts)
+          add_engine_module_paths(Rails.application, opts)
+
+          Rails.application.railties.engines.each do |engine|
+            add_engine_module_paths(engine, opts)
           end
 
           # Initialize the user module search path
@@ -27,6 +28,25 @@ module Msf
             }
           end
         end
+
+        private
+
+        # Add directories `engine.paths['modules']` from `engine`.
+        #
+        # @param engine [Rails::Engine] a rails engine or application
+        # @param options [Hash] options for {Msf::ModuleManager::ModulePaths#add_module_paths}
+        # @return [void]
+        def add_engine_module_paths(engine, options={})
+          modules_paths = engine.paths['modules']
+
+          if modules_paths
+            modules_directories = modules_paths.existent_directories
+
+            modules_directories.each do |modules_directory|
+              modules.add_module_path(modules_directory, options)
+            end
+          end
+        end
       end
     end
   end