2018-10-22 07:00:50 +00:00
|
|
|
# frozen_string_literal: true
|
|
|
|
|
2017-06-01 21:21:14 +00:00
|
|
|
module Gitlab
|
|
|
|
module EncodingHelper
|
|
|
|
extend self
|
|
|
|
|
|
|
|
# This threshold is carefully tweaked to prevent usage of encodings detected
|
|
|
|
# by CharlockHolmes with low confidence. If CharlockHolmes confidence is low,
|
|
|
|
# we're better off sticking with utf8 encoding.
|
|
|
|
# Reason: git diff can return strings with invalid utf8 byte sequences if it
|
|
|
|
# truncates a diff in the middle of a multibyte character. In this case
|
|
|
|
# CharlockHolmes will try to guess the encoding and will likely suggest an
|
|
|
|
# obscure encoding with low confidence.
|
|
|
|
# There is a lot more info with this merge request:
|
|
|
|
# https://gitlab.com/gitlab-org/gitlab_git/merge_requests/77#note_4754193
|
2017-07-20 13:02:07 +00:00
|
|
|
ENCODING_CONFIDENCE_THRESHOLD = 50
|
2017-06-01 21:21:14 +00:00
|
|
|
|
|
|
|
def encode!(message)
|
2018-01-04 22:27:37 +00:00
|
|
|
message = force_encode_utf8(message)
|
2017-06-01 21:21:14 +00:00
|
|
|
return message if message.valid_encoding?
|
|
|
|
|
|
|
|
# return message if message type is binary
|
|
|
|
detect = CharlockHolmes::EncodingDetector.detect(message)
|
2017-09-05 17:16:08 +00:00
|
|
|
return message.force_encoding("BINARY") if detect_binary?(message, detect)
|
2017-06-01 21:21:14 +00:00
|
|
|
|
|
|
|
if detect && detect[:encoding] && detect[:confidence] > ENCODING_CONFIDENCE_THRESHOLD
|
2017-09-03 11:45:44 +00:00
|
|
|
# force detected encoding if we have sufficient confidence.
|
2017-06-01 21:21:14 +00:00
|
|
|
message.force_encoding(detect[:encoding])
|
|
|
|
end
|
|
|
|
|
|
|
|
# encode and clean the bad chars
|
|
|
|
message.replace clean(message)
|
2018-02-09 17:58:29 +00:00
|
|
|
rescue ArgumentError => e
|
|
|
|
return unless e.message.include?('unknown encoding name')
|
|
|
|
|
2017-06-01 21:21:14 +00:00
|
|
|
encoding = detect ? detect[:encoding] : "unknown"
|
|
|
|
"--broken encoding: #{encoding}"
|
|
|
|
end
|
|
|
|
|
2017-09-05 17:16:08 +00:00
|
|
|
def detect_binary?(data, detect = nil)
|
2017-09-04 17:34:15 +00:00
|
|
|
detect ||= CharlockHolmes::EncodingDetector.detect(data)
|
2017-09-05 17:16:08 +00:00
|
|
|
detect && detect[:type] == :binary && detect[:confidence] == 100
|
2017-09-04 17:34:15 +00:00
|
|
|
end
|
|
|
|
|
2017-09-05 17:16:08 +00:00
|
|
|
def detect_libgit2_binary?(data)
|
2017-09-04 17:34:15 +00:00
|
|
|
# EncodingDetector checks the first 1024 * 1024 bytes for NUL byte, libgit2 checks
|
|
|
|
# only the first 8000 (https://github.com/libgit2/libgit2/blob/2ed855a9e8f9af211e7274021c2264e600c0f86b/src/filter.h#L15),
|
|
|
|
# which is what we use below to keep a consistent behavior.
|
|
|
|
detect = CharlockHolmes::EncodingDetector.new(8000).detect(data)
|
2017-09-05 17:16:08 +00:00
|
|
|
detect && detect[:type] == :binary
|
2017-09-03 11:45:44 +00:00
|
|
|
end
|
|
|
|
|
2017-06-01 21:21:14 +00:00
|
|
|
def encode_utf8(message)
|
2018-01-04 22:27:37 +00:00
|
|
|
message = force_encode_utf8(message)
|
|
|
|
return message if message.valid_encoding?
|
2017-06-14 16:11:03 +00:00
|
|
|
|
2017-06-01 21:21:14 +00:00
|
|
|
detect = CharlockHolmes::EncodingDetector.detect(message)
|
2017-06-06 16:40:07 +00:00
|
|
|
if detect && detect[:encoding]
|
2017-06-01 21:21:14 +00:00
|
|
|
begin
|
|
|
|
CharlockHolmes::Converter.convert(message, detect[:encoding], 'UTF-8')
|
|
|
|
rescue ArgumentError => e
|
|
|
|
Rails.logger.warn("Ignoring error converting #{detect[:encoding]} into UTF8: #{e.message}")
|
|
|
|
|
|
|
|
''
|
|
|
|
end
|
|
|
|
else
|
|
|
|
clean(message)
|
|
|
|
end
|
2018-01-04 22:27:37 +00:00
|
|
|
rescue ArgumentError
|
2018-07-02 10:43:06 +00:00
|
|
|
nil
|
2017-06-01 21:21:14 +00:00
|
|
|
end
|
2017-09-04 19:32:57 +00:00
|
|
|
|
2018-07-04 14:02:01 +00:00
|
|
|
def encode_binary(str)
|
|
|
|
return "" if str.nil?
|
2017-12-26 18:53:31 +00:00
|
|
|
|
2018-07-04 14:02:01 +00:00
|
|
|
str.dup.force_encoding(Encoding::ASCII_8BIT)
|
2017-12-26 18:53:31 +00:00
|
|
|
end
|
|
|
|
|
2018-07-04 14:02:01 +00:00
|
|
|
def binary_stringio(str)
|
2018-08-29 16:38:17 +00:00
|
|
|
StringIO.new(str.freeze || '').tap { |io| io.set_encoding(Encoding::ASCII_8BIT) }
|
2017-12-26 18:53:31 +00:00
|
|
|
end
|
|
|
|
|
2017-06-01 21:21:14 +00:00
|
|
|
private
|
|
|
|
|
2018-01-04 22:27:37 +00:00
|
|
|
def force_encode_utf8(message)
|
|
|
|
raise ArgumentError unless message.respond_to?(:force_encoding)
|
|
|
|
return message if message.encoding == Encoding::UTF_8 && message.valid_encoding?
|
|
|
|
|
|
|
|
message = message.dup if message.respond_to?(:frozen?) && message.frozen?
|
|
|
|
|
|
|
|
message.force_encoding("UTF-8")
|
|
|
|
end
|
|
|
|
|
2017-06-01 21:21:14 +00:00
|
|
|
def clean(message)
|
2018-03-22 20:22:20 +00:00
|
|
|
message.encode("UTF-16BE", undef: :replace, invalid: :replace, replace: "".encode("UTF-16BE"))
|
2017-06-01 21:21:14 +00:00
|
|
|
.encode("UTF-8")
|
|
|
|
.gsub("\0".encode("UTF-8"), "")
|
|
|
|
end
|
|
|
|
end
|
|
|
|
end
|