require_relative '../spec_helper'

ruby_version_is ''...'3.0' do
  require 'rexml/document'

  describe "REXML::Document.new" do

    it "resists CVE-2014-8080 by raising an exception when entity expansion has grown too large" do
      xml = <<XML
      <?xml version="1.0" encoding="UTF-8" ?>
        <!DOCTYPE x [
          <!ENTITY % x0 "xxxxxxxxxx">
          <!ENTITY % x1 "%x0;%x0;%x0;%x0;%x0;%x0;%x0;%x0;%x0;%x0;">
          <!ENTITY % x2 "%x1;%x1;%x1;%x1;%x1;%x1;%x1;%x1;%x1;%x1;">
          <!ENTITY % x3 "%x2;%x2;%x2;%x2;%x2;%x2;%x2;%x2;%x2;%x2;">
          <!ENTITY % x4 "%x3;%x3;%x3;%x3;%x3;%x3;%x3;%x3;%x3;%x3;">
          <!ENTITY % x5 "%x4;%x4;%x4;%x4;%x4;%x4;%x4;%x4;%x4;%x4;">
          <!ENTITY % x6 "%x5;%x5;%x5;%x5;%x5;%x5;%x5;%x5;%x5;%x5;">
          <!ENTITY % x7 "%x6;%x6;%x6;%x6;%x6;%x6;%x6;%x6;%x6;%x6;">
          <!ENTITY % x8 "%x7;%x7;%x7;%x7;%x7;%x7;%x7;%x7;%x7;%x7;">
          <!ENTITY % x9 "%x8;%x8;%x8;%x8;%x8;%x8;%x8;%x8;%x8;%x8;">
        ]>
        <x>
          %x9;%x9;%x9;%x9;%x9;%x9;%x9;%x9;%x9;%x9;
        </x>
XML

      -> {
        REXML::Document.new(xml).doctype.entities['x9'].value
      }.should raise_error(REXML::ParseException, /entity expansion has grown too large/)
    end

  end
end