mirror of
https://github.com/ruby/ruby.git
synced 2026-10-29 00:46:26 +08:00
The test assumed the length is narrower than size_t wherever unsigned long is, which stops holding once rb_len_t widens on mswin. It keeps running on mingw. https://bugs.ruby-lang.org/issues/22400 Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com>
389 lines
15 KiB
Ruby
389 lines
15 KiB
Ruby
# frozen_string_literal: true
|
||
require 'test/unit'
|
||
require 'cgi/escape'
|
||
require 'stringio'
|
||
require_relative 'update_env'
|
||
|
||
|
||
class CGIEscapeTest < Test::Unit::TestCase
|
||
include CGI::Escape
|
||
include UpdateEnv
|
||
|
||
def setup
|
||
@environ = {}
|
||
update_env(
|
||
'REQUEST_METHOD' => 'GET',
|
||
'SCRIPT_NAME' => nil,
|
||
)
|
||
@str1="&<>\" \xE3\x82\x86\xE3\x82\x93\xE3\x82\x86\xE3\x82\x93".dup
|
||
@str1.force_encoding("UTF-8") if defined?(::Encoding)
|
||
end
|
||
|
||
def teardown
|
||
ENV.update(@environ)
|
||
end
|
||
|
||
def test_cgi_escape
|
||
assert_equal('%26%3C%3E%22+%E3%82%86%E3%82%93%E3%82%86%E3%82%93', CGI.escape(@str1))
|
||
assert_equal('%26%3C%3E%22+%E3%82%86%E3%82%93%E3%82%86%E3%82%93'.ascii_only?, CGI.escape(@str1).ascii_only?) if defined?(::Encoding)
|
||
end
|
||
|
||
def test_cgi_escape_with_unreserved_characters
|
||
assert_equal("ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789-._~",
|
||
CGI.escape("ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789-._~"),
|
||
"should not escape any unreserved characters, as per RFC3986 Section 2.3")
|
||
end
|
||
|
||
def test_cgi_escape_with_invalid_byte_sequence
|
||
assert_equal('%C0%3C%3C', CGI.escape("\xC0\<\<".dup.force_encoding("UTF-8")))
|
||
end
|
||
|
||
def test_cgi_escape_preserve_encoding
|
||
assert_equal(Encoding::US_ASCII, CGI.escape("\xC0\<\<".dup.force_encoding("US-ASCII")).encoding)
|
||
assert_equal(Encoding::ASCII_8BIT, CGI.escape("\xC0\<\<".dup.force_encoding("ASCII-8BIT")).encoding)
|
||
assert_equal(Encoding::UTF_8, CGI.escape("\xC0\<\<".dup.force_encoding("UTF-8")).encoding)
|
||
end
|
||
|
||
def test_cgi_unescape
|
||
str = CGI.unescape('%26%3C%3E%22+%E3%82%86%E3%82%93%E3%82%86%E3%82%93')
|
||
assert_equal(@str1, str)
|
||
return unless defined?(::Encoding)
|
||
|
||
assert_equal(@str1.encoding, str.encoding)
|
||
assert_equal("\u{30E1 30E2 30EA 691C 7D22}", CGI.unescape("\u{30E1 30E2 30EA}%E6%A4%9C%E7%B4%A2"))
|
||
end
|
||
|
||
def test_cgi_unescape_preserve_encoding
|
||
assert_equal(Encoding::US_ASCII, CGI.unescape("%C0%3C%3C".dup.force_encoding("US-ASCII")).encoding)
|
||
assert_equal(Encoding::ASCII_8BIT, CGI.unescape("%C0%3C%3C".dup.force_encoding("ASCII-8BIT")).encoding)
|
||
assert_equal(Encoding::UTF_8, CGI.unescape("%C0%3C%3C".dup.force_encoding("UTF-8")).encoding)
|
||
end
|
||
|
||
def test_cgi_unescape_accept_charset
|
||
return unless defined?(::Encoding)
|
||
|
||
assert_raise(TypeError) {CGI.unescape('', nil)}
|
||
assert_separately(%w[-rcgi/escape], "#{<<-"begin;"}\n#{<<-"end;"}")
|
||
begin;
|
||
assert_equal("", CGI.unescape(''))
|
||
end;
|
||
end
|
||
|
||
def test_cgi_escapeURIComponent
|
||
assert_equal('%26%3C%3E%22%20%E3%82%86%E3%82%93%E3%82%86%E3%82%93', CGI.escapeURIComponent(@str1))
|
||
assert_equal('%26%3C%3E%22%20%E3%82%86%E3%82%93%E3%82%86%E3%82%93'.ascii_only?, CGI.escapeURIComponent(@str1).ascii_only?) if defined?(::Encoding)
|
||
end
|
||
|
||
def test_cgi_escape_uri_component
|
||
assert_equal('%26%3C%3E%22%20%E3%82%86%E3%82%93%E3%82%86%E3%82%93', CGI.escape_uri_component(@str1))
|
||
end
|
||
|
||
def test_cgi_escapeURIComponent_with_unreserved_characters
|
||
assert_equal("ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789-._~",
|
||
CGI.escapeURIComponent("ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789-._~"),
|
||
"should not encode any unreserved characters, as per RFC3986 Section 2.3")
|
||
end
|
||
|
||
def test_cgi_escapeURIComponent_with_invalid_byte_sequence
|
||
assert_equal('%C0%3C%3C', CGI.escapeURIComponent("\xC0\<\<".dup.force_encoding("UTF-8")))
|
||
end
|
||
|
||
def test_cgi_escapeURIComponent_preserve_encoding
|
||
assert_equal(Encoding::US_ASCII, CGI.escapeURIComponent("\xC0\<\<".dup.force_encoding("US-ASCII")).encoding)
|
||
assert_equal(Encoding::ASCII_8BIT, CGI.escapeURIComponent("\xC0\<\<".dup.force_encoding("ASCII-8BIT")).encoding)
|
||
assert_equal(Encoding::UTF_8, CGI.escapeURIComponent("\xC0\<\<".dup.force_encoding("UTF-8")).encoding)
|
||
end
|
||
|
||
def test_cgi_unescapeURIComponent
|
||
str = CGI.unescapeURIComponent('%26%3C%3E%22%20%E3%82%86%E3%82%93%E3%82%86%E3%82%93')
|
||
assert_equal(@str1, str)
|
||
return unless defined?(::Encoding)
|
||
|
||
assert_equal("foo+bar", CGI.unescapeURIComponent("foo+bar"))
|
||
|
||
assert_equal(@str1.encoding, str.encoding)
|
||
assert_equal("\u{30E1 30E2 30EA 691C 7D22}", CGI.unescapeURIComponent("\u{30E1 30E2 30EA}%E6%A4%9C%E7%B4%A2"))
|
||
end
|
||
|
||
def test_cgi_unescape_uri_component
|
||
str = CGI.unescape_uri_component('%26%3C%3E%22%20%E3%82%86%E3%82%93%E3%82%86%E3%82%93')
|
||
assert_equal(@str1, str)
|
||
end
|
||
|
||
def test_cgi_unescapeURIComponent_preserve_encoding
|
||
assert_equal(Encoding::US_ASCII, CGI.unescapeURIComponent("%C0%3C%3C".dup.force_encoding("US-ASCII")).encoding)
|
||
assert_equal(Encoding::ASCII_8BIT, CGI.unescapeURIComponent("%C0%3C%3C".dup.force_encoding("ASCII-8BIT")).encoding)
|
||
assert_equal(Encoding::UTF_8, CGI.unescapeURIComponent("%C0%3C%3C".dup.force_encoding("UTF-8")).encoding)
|
||
end
|
||
|
||
def test_cgi_unescapeURIComponent_accept_charset
|
||
return unless defined?(::Encoding)
|
||
|
||
assert_raise(TypeError) {CGI.unescapeURIComponent('', nil)}
|
||
assert_separately(%w[-rcgi/escape], "#{<<-"begin;"}\n#{<<-"end;"}")
|
||
begin;
|
||
assert_equal("", CGI.unescapeURIComponent(''))
|
||
end;
|
||
end
|
||
|
||
def test_cgi_escapeHTML
|
||
assert_equal("'&"><", CGI.escapeHTML("'&\"><"))
|
||
end
|
||
|
||
def test_cgi_escape_html_duplicated
|
||
orig = "Ruby".dup.force_encoding("US-ASCII")
|
||
str = CGI.escapeHTML(orig)
|
||
assert_equal(orig, str)
|
||
assert_not_same(orig, str)
|
||
end
|
||
|
||
def assert_cgi_escape_html_preserve_encoding(str, encoding)
|
||
assert_equal(encoding, CGI.escapeHTML(str.dup.force_encoding(encoding)).encoding)
|
||
end
|
||
|
||
def test_cgi_escape_html_preserve_encoding
|
||
Encoding.list do |enc|
|
||
assert_cgi_escape_html_preserve_encoding("'&\"><", enc)
|
||
assert_cgi_escape_html_preserve_encoding("Ruby", enc)
|
||
end
|
||
end
|
||
|
||
def test_cgi_escape_html_dont_freeze
|
||
assert_not_predicate CGI.escapeHTML("'&\"><".dup), :frozen?
|
||
assert_not_predicate CGI.escapeHTML("'&\"><".freeze), :frozen?
|
||
assert_not_predicate CGI.escapeHTML("Ruby".dup), :frozen?
|
||
assert_not_predicate CGI.escapeHTML("Ruby".freeze), :frozen?
|
||
end
|
||
|
||
def test_cgi_escape_html_large
|
||
return if RUBY_ENGINE == 'jruby'
|
||
ulen_max = 2 ** (RbConfig::SIZEOF.fetch("rb_len_t") { RbConfig::SIZEOF["long"] } * 8) - 1
|
||
size_max = RbConfig::LIMITS["SIZE_MAX"]
|
||
return unless ulen_max < size_max # Platforms not concerned
|
||
|
||
size = (ulen_max / 6 + 1)
|
||
begin
|
||
str = '"' * size
|
||
escaped = CGI.escapeHTML(str)
|
||
rescue NoMemoryError
|
||
omit "Not enough memory"
|
||
rescue => e
|
||
end
|
||
assert_raise_with_message(ArgumentError, /overflow/, ->{"length = #{escaped.length}"}) do
|
||
raise e if e
|
||
end
|
||
end
|
||
|
||
def test_cgi_unescapeHTML
|
||
assert_equal("'&\"><", CGI.unescapeHTML("'&"><"))
|
||
end
|
||
|
||
def test_cgi_unescapeHTML_invalid
|
||
assert_equal('&<&>"&abcdefghijklmn', CGI.unescapeHTML('&<&>"&abcdefghijklmn'))
|
||
end
|
||
|
||
module UnescapeHTMLTests
|
||
def test_cgi_unescapeHTML_following_known_first_letter
|
||
assert_equal('&a>&q>&l>&g>', CGI.unescapeHTML('&a>&q>&l>&g>'))
|
||
end
|
||
|
||
def test_cgi_unescapeHTML_following_number_sign
|
||
assert_equal('&#>&#x>', CGI.unescapeHTML('&#>&#x>'))
|
||
end
|
||
|
||
def test_cgi_unescapeHTML_following_invalid_numeric
|
||
assert_equal('�>�>', CGI.unescapeHTML('�>�>'))
|
||
end
|
||
|
||
# https://github.com/ruby/cgi/issues/103
|
||
# A numeric character reference must decode without raising even when the
|
||
# surrounding text already contains non-ASCII bytes.
|
||
def test_cgi_unescapeHTML_nonascii_with_numeric_charref
|
||
assert_equal("☆ α", CGI.unescapeHTML("☆ α"))
|
||
assert_equal("☆ α", CGI.unescapeHTML("☆ α"))
|
||
assert_equal("☆ α", CGI.unescapeHTML("☆ α"))
|
||
assert_equal("\u{1F984} α", CGI.unescapeHTML("\u{1F984} α"))
|
||
result = CGI.unescapeHTML("☆ α")
|
||
assert_equal(Encoding::UTF_8, result.encoding)
|
||
assert_predicate(result, :valid_encoding?)
|
||
end
|
||
|
||
def test_cgi_unescapeHTML_nonascii_with_named_charref
|
||
assert_equal("☆ & < > \" '", CGI.unescapeHTML("☆ & < > " '"))
|
||
end
|
||
|
||
def test_cgi_unescapeHTML_nonascii_charref_iso_8859_1
|
||
input = "\xE9 é é".dup.force_encoding("ISO-8859-1")
|
||
expected = "\xE9 \xE9 \xE9".dup.force_encoding("ISO-8859-1")
|
||
assert_equal(expected, CGI.unescapeHTML(input))
|
||
assert_equal(Encoding::ISO_8859_1, CGI.unescapeHTML(input).encoding)
|
||
# A code point outside Latin-1 exceeds the charlimit and is kept verbatim.
|
||
verbatim = "\xE9 α".dup.force_encoding("ISO-8859-1")
|
||
assert_equal(verbatim, CGI.unescapeHTML(verbatim))
|
||
end
|
||
|
||
def test_cgi_unescapeHTML_charref_out_of_range
|
||
assert_equal("�", CGI.unescapeHTML("�"))
|
||
assert_equal("�", CGI.unescapeHTML("�"))
|
||
# Leading zeros are preserved when kept verbatim, matching the C extension.
|
||
assert_equal("�", CGI.unescapeHTML("�"))
|
||
assert_equal("☆ �", CGI.unescapeHTML("☆ �"))
|
||
end
|
||
|
||
def test_cgi_unescapeHTML_surrogate_charref
|
||
# Surrogate code points are below the UTF-8 limit, so they are emitted as
|
||
# raw (intentionally invalid) UTF-8 bytes rather than raising, matching the
|
||
# C extension's rb_enc_mbcput. The result is not validated.
|
||
assert_equal([0xED, 0xA0, 0x80], CGI.unescapeHTML("�").bytes)
|
||
assert_equal([0xED, 0xA0, 0x80], CGI.unescapeHTML("�").bytes)
|
||
assert_equal([0xED, 0xBF, 0xBF], CGI.unescapeHTML("�").bytes)
|
||
end
|
||
|
||
def test_cgi_unescapeHTML_charref_preserve_encoding
|
||
["UTF-8", "EUC-JP", "Windows-31J", "ISO-8859-1", "US-ASCII"].each do |name|
|
||
enc = Encoding.find(name)
|
||
result = CGI.unescapeHTML("aAb".encode(enc))
|
||
assert_equal("aAb".encode(enc), result, name)
|
||
assert_equal(enc, result.encoding, name)
|
||
end
|
||
end
|
||
end
|
||
|
||
include UnescapeHTMLTests
|
||
|
||
Encoding.list.each do |enc|
|
||
begin
|
||
escaped = "'&"><".encode(enc)
|
||
unescaped = "'&\"><".encode(enc)
|
||
rescue Encoding::ConverterNotFoundError
|
||
next
|
||
else
|
||
define_method("test_cgi_escapeHTML:#{enc.name}") do
|
||
assert_equal(escaped, CGI.escapeHTML(unescaped))
|
||
end
|
||
define_method("test_cgi_unescapeHTML:#{enc.name}") do
|
||
assert_equal(unescaped, CGI.unescapeHTML(escaped))
|
||
end
|
||
end
|
||
end
|
||
|
||
Encoding.list.each do |enc|
|
||
next unless enc.ascii_compatible?
|
||
begin
|
||
escaped = "%25+%2B"
|
||
unescaped = "% +".encode(enc)
|
||
rescue Encoding::ConverterNotFoundError
|
||
next
|
||
else
|
||
define_method("test_cgi_escape:#{enc.name}") do
|
||
assert_equal(escaped, CGI.escape(unescaped))
|
||
end
|
||
define_method("test_cgi_unescape:#{enc.name}") do
|
||
assert_equal(unescaped, CGI.unescape(escaped, enc))
|
||
end
|
||
end
|
||
end
|
||
|
||
def test_cgi_unescapeHTML_uppercasecharacter
|
||
assert_equal("\xE3\x81\x82\xE3\x81\x84\xE3\x81\x86", CGI.unescapeHTML("あいう"))
|
||
end
|
||
|
||
def test_cgi_include_escape
|
||
assert_equal('%26%3C%3E%22+%E3%82%86%E3%82%93%E3%82%86%E3%82%93', escape(@str1))
|
||
end
|
||
|
||
def test_cgi_include_escapeHTML
|
||
assert_equal("'&"><", escapeHTML("'&\"><"))
|
||
end
|
||
|
||
def test_cgi_include_h
|
||
assert_equal("'&"><", h("'&\"><"))
|
||
end
|
||
|
||
def test_cgi_include_unescape
|
||
str = unescape('%26%3C%3E%22+%E3%82%86%E3%82%93%E3%82%86%E3%82%93')
|
||
assert_equal(@str1, str)
|
||
return unless defined?(::Encoding)
|
||
|
||
assert_equal(@str1.encoding, str.encoding)
|
||
assert_equal("\u{30E1 30E2 30EA 691C 7D22}", unescape("\u{30E1 30E2 30EA}%E6%A4%9C%E7%B4%A2"))
|
||
end
|
||
|
||
def test_cgi_include_unescapeHTML
|
||
assert_equal("'&\"><", unescapeHTML("'&"><"))
|
||
end
|
||
|
||
def test_cgi_escapeElement
|
||
assert_equal("<BR><A HREF="url"></A>", escapeElement('<BR><A HREF="url"></A>', "A", "IMG"))
|
||
assert_equal("<BR><A HREF="url"></A>", escapeElement('<BR><A HREF="url"></A>', ["A", "IMG"]))
|
||
assert_equal("<BR><A HREF="url"></A>", escape_element('<BR><A HREF="url"></A>', "A", "IMG"))
|
||
assert_equal("<BR><A HREF="url"></A>", escape_element('<BR><A HREF="url"></A>', ["A", "IMG"]))
|
||
|
||
assert_equal("<A <A HREF="url"></A>", escapeElement('<A <A HREF="url"></A>', "A", "IMG"))
|
||
assert_equal("<A <A HREF="url"></A>", escapeElement('<A <A HREF="url"></A>', ["A", "IMG"]))
|
||
assert_equal("<A <A HREF="url"></A>", escape_element('<A <A HREF="url"></A>', "A", "IMG"))
|
||
assert_equal("<A <A HREF="url"></A>", escape_element('<A <A HREF="url"></A>', ["A", "IMG"]))
|
||
|
||
assert_equal("<A <A ", escapeElement('<A <A ', "A", "IMG"))
|
||
assert_equal("<A <A ", escapeElement('<A <A ', ["A", "IMG"]))
|
||
end
|
||
|
||
|
||
def test_cgi_unescapeElement
|
||
assert_equal('<BR><A HREF="url"></A>', unescapeElement(escapeHTML('<BR><A HREF="url"></A>'), "A", "IMG"))
|
||
assert_equal('<BR><A HREF="url"></A>', unescapeElement(escapeHTML('<BR><A HREF="url"></A>'), ["A", "IMG"]))
|
||
assert_equal('<BR><A HREF="url"></A>', unescape_element(escapeHTML('<BR><A HREF="url"></A>'), "A", "IMG"))
|
||
assert_equal('<BR><A HREF="url"></A>', unescape_element(escapeHTML('<BR><A HREF="url"></A>'), ["A", "IMG"]))
|
||
|
||
assert_equal('<A <A HREF="url"></A>', unescapeElement(escapeHTML('<A <A HREF="url"></A>'), "A", "IMG"))
|
||
assert_equal('<A <A HREF="url"></A>', unescapeElement(escapeHTML('<A <A HREF="url"></A>'), ["A", "IMG"]))
|
||
assert_equal('<A <A HREF="url"></A>', unescape_element(escapeHTML('<A <A HREF="url"></A>'), "A", "IMG"))
|
||
assert_equal('<A <A HREF="url"></A>', unescape_element(escapeHTML('<A <A HREF="url"></A>'), ["A", "IMG"]))
|
||
|
||
assert_equal('<A <A ', unescapeElement(escapeHTML('<A <A '), "A", "IMG"))
|
||
assert_equal('<A <A ', unescapeElement(escapeHTML('<A <A '), ["A", "IMG"]))
|
||
assert_equal('<A <A ', unescape_element(escapeHTML('<A <A '), "A", "IMG"))
|
||
assert_equal('<A <A ', unescape_element(escapeHTML('<A <A '), ["A", "IMG"]))
|
||
end
|
||
end
|
||
|
||
class CGIEscapeNativeExtTest < Test::Unit::TestCase
|
||
def test_escape_html_uses_native_implementation
|
||
omit "C extension not available" unless defined?(CGI::EscapeExt) && CGI::EscapeExt.method_defined?(:escapeHTML)
|
||
assert_equal CGI::EscapeExt, CGI.method(:escape_html).owner
|
||
assert_equal CGI::EscapeExt, CGI.method(:h).owner
|
||
assert_equal CGI::EscapeExt, CGI.method(:unescape_html).owner
|
||
end
|
||
end
|
||
|
||
class CGIEscapePureRubyTest < Test::Unit::TestCase
|
||
def setup
|
||
CGI::EscapeExt.module_eval do
|
||
alias _escapeHTML escapeHTML
|
||
remove_method :escapeHTML
|
||
alias _unescapeHTML unescapeHTML
|
||
remove_method :unescapeHTML
|
||
end if defined?(CGI::EscapeExt) and CGI::EscapeExt.method_defined?(:escapeHTML)
|
||
end
|
||
|
||
def teardown
|
||
CGI::EscapeExt.module_eval do
|
||
alias escapeHTML _escapeHTML
|
||
remove_method :_escapeHTML
|
||
alias unescapeHTML _unescapeHTML
|
||
remove_method :_unescapeHTML
|
||
end if defined?(CGI::EscapeExt) and CGI::EscapeExt.method_defined?(:_escapeHTML)
|
||
end
|
||
|
||
include CGIEscapeTest::UnescapeHTMLTests
|
||
|
||
def test_cgi_escapeHTML_with_invalid_byte_sequence
|
||
assert_equal("<\xA4??>", CGI.escapeHTML(%[<\xA4??>]))
|
||
end
|
||
|
||
def test_cgi_unescapeHTML_with_invalid_byte_sequence
|
||
input = "\xFF&"
|
||
assert_equal(input, CGI.unescapeHTML(input))
|
||
end
|
||
end
|