Merge pull request #3227 from evazion/fix-3225

Fix #3225: Exception in NoteSanitizer#relativize_links.
This commit is contained in:
Albert Yi
2017-07-20 15:44:15 -07:00
committed by GitHub
2 changed files with 10 additions and 4 deletions

View File

@@ -79,14 +79,15 @@ module NoteSanitizer
end
def self.relativize_links(node:, **env)
return unless node.name == "a" && node.attribute("href")
return unless node.name == "a" && node["href"].present?
href = node.attribute("href")
url = Addressable::URI.parse(href.value).normalize
url = Addressable::URI.heuristic_parse(node["href"]).normalize
if url.authority.in?(Danbooru.config.hostnames)
url.site = nil
href.value = url.to_s
node["href"] = url.to_s
end
rescue Addressable::URI::InvalidURIError
# do nothing for invalid urls
end
end

View File

@@ -28,5 +28,10 @@ class NoteSanitizerTest < ActiveSupport::TestCase
body = '<a href="http://sonohara.donmai.us/posts?tags=touhou#dtext-intro">touhou</a>'
assert_equal('<a href="/posts?tags=touhou#dtext-intro" rel="nofollow">touhou</a>', NoteSanitizer.sanitize(body))
end
should "not fail when rewriting bad links" do
body = %{<a href ="\nhttp!://www.google.com:12x3">google</a>}
assert_equal(%{<a rel="nofollow">google</a>}, NoteSanitizer.sanitize(body))
end
end
end