WpTarget modules reworked

spec/shared_examples/wp_target/wp_login_protection.rb

@@ -0,0 +1,89 @@
+# encoding: UTF-8
+
+shared_examples 'WpTarget::WpLoginProtection' do
+
+  let(:fixtures_dir) { SPEC_FIXTURES_WPSCAN_WP_TARGET_DIR + '/wp_login_protection' }
+
+  before { wp_target.stub(:wp_plugins_dir).and_return('wp-content/plugins') }
+
+  # It will test all protected methods has_.*_protection with each fixtures to be sure that
+  # there is not false positive : for example the login-lock must not be detected as login-lockdown
+  describe '#has_.*_protection?' do
+
+    pattern = WpTarget::WpLoginProtection::LOGIN_PROTECTION_METHOD_PATTERN
+    fixtures = %w{
+      wp-login-clean.php wp-login-login_lockdown.php wp-login-login_lock.php
+      wp-login-better_wp_security.php wp-login-simple_login_lockdown.php wp-login-login_security_solution.php
+      wp-login-limit_login_attempts.php wp-login-bluetrait_event_viewer.php
+    }
+
+    # For plugins which are detected from the existence of their directory into wp-content/plugins/ (or one of their file)
+    # and not from a regex into the login page
+    special_plugins = %w{better_wp_security simple_login_lockdown login_security_solution limit_login_attempts bluetrait_event_viewer}
+
+    after :each do
+      stub_request_to_fixture(url: login_url, fixture: @fixture)
+
+      # Stub all special plugins urls to a 404 except if it's the one we want
+      special_plugins.each do |special_plugin|
+        special_plugin_call_detection_symbol = :"has_#{special_plugin}_protection?"
+        special_plugin_call_url_symbol = :"#{special_plugin}_url"
+
+        status_code = (@symbol_to_call === special_plugin_call_detection_symbol and @expected === true) ? 200 : 404
+        stub_request(:get, wp_target.send(special_plugin_call_url_symbol).to_s).to_return(status: status_code)
+      end
+
+      wp_target.send(@symbol_to_call).should === @expected
+    end
+
+    self.protected_instance_methods.grep(pattern).each do |symbol_to_call|
+      plugin_name_from_symbol = symbol_to_call[pattern, 1].gsub('_', '-')
+
+      fixtures.each do |fixture|
+        plugin_name_from_fixture = fixture[/wp-login-(.*)\.php/i, 1].gsub('_', '-')
+        expected = plugin_name_from_fixture === plugin_name_from_symbol ? true : false
+
+        it "#{symbol_to_call} with #{fixture} returns #{expected}" do
+          @plugin_name    = plugin_name_from_fixture
+          @fixture        = fixtures_dir + '/' + fixture
+          @symbol_to_call = symbol_to_call
+          @expected       = expected
+        end
+      end
+    end
+  end
+
+  # Factorise this with the code above ? :D
+  describe '#login_protection_plugin' do
+    after :each do
+      stub_request_to_fixture(url: login_url, fixture: @fixture)
+      stub_request(:get, wp_target.send(:better_wp_security_url).to_s).to_return(status: 404)
+      stub_request(:get, wp_target.send(:simple_login_lockdown_url).to_s).to_return(status: 404)
+      stub_request(:get, wp_target.send(:login_security_solution_url).to_s).to_return(status: 404)
+      stub_request(:get, wp_target.send(:limit_login_attempts_url).to_s).to_return(status: 404)
+      stub_request(:get, wp_target.send(:bluetrait_event_viewer_url).to_s).to_return(status: 404)
+
+      wp_target.login_protection_plugin().should == @plugin_expected
+      wp_target.has_login_protection?.should === @has_protection_expected
+    end
+
+    it 'returns nil if no protection is present' do
+      @fixture                 = fixtures_dir + '/wp-login-clean.php'
+      @plugin_expected         = nil
+      @has_protection_expected = false
+    end
+
+    it 'returns a login-lockdown WpPlugin object' do
+      @fixture                 = fixtures_dir + '/wp-login-login_lockdown.php'
+      @plugin_expected         = WpPlugin.new(wp_target.uri, name: 'login-lockdown')
+      @has_protection_expected = true
+    end
+
+    it 'returns a login-lock WpPlugin object' do
+      @fixture                 = fixtures_dir + '/wp-login-login_lock.php'
+      @plugin_expected         = WpPlugin.new(wp_target.uri, name: 'login-lock')
+      @has_protection_expected = true
+    end
+  end
+
+end