safe_redirection 0.0.2 → 0.0.3

data/lib/safe_redirection/constants.rb

@@ -1,3 +1,6 @@
 module SafeRedirection
-  VERSION = '0.0.2'
+  VERSION = '0.0.3'
+
+  class Error < StandardError; end
+  class SanitizationCancelled < SafeRedirection::Error; end
 end

data/lib/safe_redirection/sanitizer.rb

@@ -12,11 +12,15 @@ module SafeRedirection
       uri = URI(redirect_url)
       path = relative_path(uri.path)
 
-      if %w{http https}.include? uri.scheme
-        resolver.recognize_path(path, :method => :get) rescue default_url
+      if %w{http https}.include?(uri.scheme) || uri.scheme.nil?
+        resolver.recognize_path(path, :method => :get)
       else
         default_url
       end
+    rescue SafeRedirection::SanitizationCancelled
+      redirect_url
+    rescue
+      default_url
     end
 
     def base_path

data/spec/lib/safe_redirection/sanitizer_spec.rb

@@ -37,6 +37,14 @@ describe SafeRedirection::Sanitizer do
         subject.safe_url_for('http://test.tld/some/path/subpath')
       end
     end
+
+    context "when the resolver raises SafeRedirection::SanitizationCancelled" do
+      it "returns the URL passed" do
+        resolver.stub(:recognize_path).and_raise(SafeRedirection::SanitizationCancelled)
+        url = '/some/path?param=value'
+        subject.safe_url_for(url).should == url
+      end
+    end
   end
 
   describe "#base_path" do

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: safe_redirection
 version: !ruby/object:Gem::Version
-  version: 0.0.2
+  version: 0.0.3
   prerelease: 
 platform: ruby
 authors:
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2012-07-13 00:00:00.000000000 Z
+date: 2012-07-18 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rake
@@ -104,5 +104,5 @@ rubyforge_project:
 rubygems_version: 1.8.24
 signing_key: 
 specification_version: 3
-summary: safe_redirection-0.0.2
+summary: safe_redirection-0.0.3
 test_files: []