* lib/rexml/source.rb (REXML::IOSource#initialize): Reduce

  @line_break initialize code. It should be done only in #encoding=.
* lib/rexml/parsers/baseparser.rb: Don't set UTF-16 encoding to
  source by encoding="UTF-16" in XML declaration because UTF-16XX
  source encoding should be set in Source#initialize or
  IOSource#intialize. They should handle BOM. Parser should not
  consider about it.


git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/trunk@37361 b2dd03c8-39d4-4d8f-98ff-823fe69b080e

1 files changed, 30 insertions, 13 deletions

diff --git a/lib/rexml/source.rb b/lib/rexml/source.rb
index 112393cfd4..c15f63dcc8 100644
--- a/lib/rexml/source.rb
+++ b/lib/rexml/source.rb
@@ -144,22 +144,39 @@ module REXML
       # if there is one.  If there isn't one, the file MUST be UTF-8, as per
       # the XML spec.  If there is one, we can determine the encoding from
       # it.
-      @buffer = ""
-      str = @source.read( 2 ) || ''
       if encoding
-        self.encoding = encoding
-      elsif str[0,2] == "\xfe\xff"
-        @line_break = "\000>"
-      elsif str[0,2] == "\xff\xfe"
-        @line_break = ">\000"
-      elsif str[0,2] == "\xef\xbb"
-        str += @source.read(1)
-        str = '' if (str[2,1] == "\xBF")
-        @line_break = ">"
+        super("", encoding)
       else
-        @line_break = ">"
+        need_super_with_line = false
+        str = @source.read( 2 ) || ''
+        str.force_encoding("ASCII-8BIT")
+        if str[0, 2] == "\xfe\xff"
+          @source.binmode
+          @source.set_encoding("UTF-16BE")
+          super("", "UTF-16BE")
+        elsif str[0, 2] == "\xff\xfe"
+          @source.binmode
+          @source.set_encoding("UTF-16LE")
+          super("", "UTF-16LE")
+        elsif str[0, 2] == "\xef\xbb"
+          str += @source.read(1)
+          if str[2, 1] == "\xBF"
+            @source.set_encoding("UTF-8")
+            super("", "UTF-8")
+          else
+            need_super_with_line = true
+          end
+        else
+          need_super_with_line = true
+        end
+        if need_super_with_line
+          if @source.eof?
+            super(str)
+          else
+            super(str + @source.readline(">"))
+          end
+        end
       end
-      super( @source.eof? ? str : [email protected]( @line_break ) )
 
       if !@to_utf and
           @buffer.respond_to?(:force_encoding) and