diff options
Diffstat (limited to 'test/rexml/test_entity.rb')
-rw-r--r-- | test/rexml/test_entity.rb | 18 |
1 files changed, 18 insertions, 0 deletions
diff --git a/test/rexml/test_entity.rb b/test/rexml/test_entity.rb index e6d6f29007..5900fac7a8 100644 --- a/test/rexml/test_entity.rb +++ b/test/rexml/test_entity.rb @@ -104,6 +104,24 @@ class EntityTester < Test::Unit::TestCase assert_equal source, out end + def test_entity_string_limit + template = '<!DOCTYPE bomb [ <!ENTITY a "^" > ]> <bomb>$</bomb>' + len = 5120 # 5k per entity + template.sub!(/\^/, "B" * len) + + # 10k is OK + entities = '&a;' * 2 # 5k entity * 2 = 10k + xmldoc = REXML::Document.new(template.sub(/\$/, entities)) + assert_equal(len * 2, xmldoc.root.text.bytesize) + + # above 10k explodes + entities = '&a;' * 3 # 5k entity * 2 = 15k + xmldoc = REXML::Document.new(template.sub(/\$/, entities)) + assert_raises(RuntimeError) do + xmldoc.root.text + end + end + def test_raw source = '<!DOCTYPE foo [ <!ENTITY ent "replace"> |