334 lines
9.1 KiB
JavaScript
334 lines
9.1 KiB
JavaScript
'use strict';
|
|
|
|
var Buffer = require('safer-buffer').Buffer;
|
|
|
|
// == UTF32-LE/BE codec. ==========================================================
|
|
|
|
exports._utf32 = Utf32Codec;
|
|
|
|
function Utf32Codec(codecOptions, iconv) {
|
|
this.iconv = iconv;
|
|
this.bomAware = true;
|
|
this.isLE = codecOptions.isLE;
|
|
}
|
|
|
|
exports.utf32le = { type: '_utf32', isLE: true };
|
|
exports.utf32be = { type: '_utf32', isLE: false };
|
|
|
|
// Aliases
|
|
exports.ucs4le = 'utf32le';
|
|
exports.ucs4be = 'utf32be';
|
|
|
|
Utf32Codec.prototype.encoder = Utf32Encoder;
|
|
Utf32Codec.prototype.decoder = Utf32Decoder;
|
|
|
|
// -- Encoding
|
|
|
|
function Utf32Encoder(options, codec) {
|
|
this.isLE = codec.isLE;
|
|
this.highSurrogate = 0;
|
|
}
|
|
|
|
Utf32Encoder.prototype.write = function (str) {
|
|
var src = Buffer.from(str, 'ucs2');
|
|
var dst = Buffer.alloc(src.length * 2);
|
|
var write32 = this.isLE ? dst.writeUInt32LE : dst.writeUInt32BE;
|
|
var offset = 0;
|
|
|
|
for (var i = 0; i < src.length; i += 2) {
|
|
var code = src.readUInt16LE(i);
|
|
var isHighSurrogate = 0xd800 <= code && code < 0xdc00;
|
|
var isLowSurrogate = 0xdc00 <= code && code < 0xe000;
|
|
|
|
if (this.highSurrogate) {
|
|
if (isHighSurrogate || !isLowSurrogate) {
|
|
// There shouldn't be two high surrogates in a row, nor a high surrogate which isn't followed by a low
|
|
// surrogate. If this happens, keep the pending high surrogate as a stand-alone semi-invalid character
|
|
// (technically wrong, but expected by some applications, like Windows file names).
|
|
write32.call(dst, this.highSurrogate, offset);
|
|
offset += 4;
|
|
} else {
|
|
// Create 32-bit value from high and low surrogates;
|
|
var codepoint =
|
|
(((this.highSurrogate - 0xd800) << 10) | (code - 0xdc00)) + 0x10000;
|
|
|
|
write32.call(dst, codepoint, offset);
|
|
offset += 4;
|
|
this.highSurrogate = 0;
|
|
|
|
continue;
|
|
}
|
|
}
|
|
|
|
if (isHighSurrogate) this.highSurrogate = code;
|
|
else {
|
|
// Even if the current character is a low surrogate, with no previous high surrogate, we'll
|
|
// encode it as a semi-invalid stand-alone character for the same reasons expressed above for
|
|
// unpaired high surrogates.
|
|
write32.call(dst, code, offset);
|
|
offset += 4;
|
|
this.highSurrogate = 0;
|
|
}
|
|
}
|
|
|
|
if (offset < dst.length) dst = dst.slice(0, offset);
|
|
|
|
return dst;
|
|
};
|
|
|
|
Utf32Encoder.prototype.end = function () {
|
|
// Treat any leftover high surrogate as a semi-valid independent character.
|
|
if (!this.highSurrogate) return;
|
|
|
|
var buf = Buffer.alloc(4);
|
|
|
|
if (this.isLE) buf.writeUInt32LE(this.highSurrogate, 0);
|
|
else buf.writeUInt32BE(this.highSurrogate, 0);
|
|
|
|
this.highSurrogate = 0;
|
|
|
|
return buf;
|
|
};
|
|
|
|
// -- Decoding
|
|
|
|
function Utf32Decoder(options, codec) {
|
|
this.isLE = codec.isLE;
|
|
this.badChar = codec.iconv.defaultCharUnicode.charCodeAt(0);
|
|
this.overflow = [];
|
|
}
|
|
|
|
Utf32Decoder.prototype.write = function (src) {
|
|
if (src.length === 0) return '';
|
|
|
|
var i = 0;
|
|
var codepoint = 0;
|
|
var dst = Buffer.alloc(src.length + 4);
|
|
var offset = 0;
|
|
var isLE = this.isLE;
|
|
var overflow = this.overflow;
|
|
var badChar = this.badChar;
|
|
|
|
if (overflow.length > 0) {
|
|
for (; i < src.length && overflow.length < 4; i++) overflow.push(src[i]);
|
|
|
|
if (overflow.length === 4) {
|
|
// NOTE: codepoint is a signed int32 and can be negative.
|
|
// NOTE: We copied this block from below to help V8 optimize it (it works with array, not buffer).
|
|
if (isLE) {
|
|
codepoint =
|
|
overflow[i] |
|
|
(overflow[i + 1] << 8) |
|
|
(overflow[i + 2] << 16) |
|
|
(overflow[i + 3] << 24);
|
|
} else {
|
|
codepoint =
|
|
overflow[i + 3] |
|
|
(overflow[i + 2] << 8) |
|
|
(overflow[i + 1] << 16) |
|
|
(overflow[i] << 24);
|
|
}
|
|
overflow.length = 0;
|
|
|
|
offset = _writeCodepoint(dst, offset, codepoint, badChar);
|
|
}
|
|
}
|
|
|
|
// Main loop. Should be as optimized as possible.
|
|
for (; i < src.length - 3; i += 4) {
|
|
// NOTE: codepoint is a signed int32 and can be negative.
|
|
if (isLE) {
|
|
codepoint =
|
|
src[i] | (src[i + 1] << 8) | (src[i + 2] << 16) | (src[i + 3] << 24);
|
|
} else {
|
|
codepoint =
|
|
src[i + 3] | (src[i + 2] << 8) | (src[i + 1] << 16) | (src[i] << 24);
|
|
}
|
|
offset = _writeCodepoint(dst, offset, codepoint, badChar);
|
|
}
|
|
|
|
// Keep overflowing bytes.
|
|
for (; i < src.length; i++) {
|
|
overflow.push(src[i]);
|
|
}
|
|
|
|
return dst.slice(0, offset).toString('ucs2');
|
|
};
|
|
|
|
function _writeCodepoint(dst, offset, codepoint, badChar) {
|
|
// NOTE: codepoint is signed int32 and can be negative. We keep it that way to help V8 with optimizations.
|
|
if (codepoint < 0 || codepoint > 0x10ffff) {
|
|
// Not a valid Unicode codepoint
|
|
codepoint = badChar;
|
|
}
|
|
|
|
// Ephemeral Planes: Write high surrogate.
|
|
if (codepoint >= 0x10000) {
|
|
codepoint -= 0x10000;
|
|
|
|
var high = 0xd800 | (codepoint >> 10);
|
|
dst[offset++] = high & 0xff;
|
|
dst[offset++] = high >> 8;
|
|
|
|
// Low surrogate is written below.
|
|
var codepoint = 0xdc00 | (codepoint & 0x3ff);
|
|
}
|
|
|
|
// Write BMP char or low surrogate.
|
|
dst[offset++] = codepoint & 0xff;
|
|
dst[offset++] = codepoint >> 8;
|
|
|
|
return offset;
|
|
}
|
|
|
|
Utf32Decoder.prototype.end = function () {
|
|
this.overflow.length = 0;
|
|
};
|
|
|
|
// == UTF-32 Auto codec =============================================================
|
|
// Decoder chooses automatically from UTF-32LE and UTF-32BE using BOM and space-based heuristic.
|
|
// Defaults to UTF-32LE. http://en.wikipedia.org/wiki/UTF-32
|
|
// Encoder/decoder default can be changed: iconv.decode(buf, 'utf32', {defaultEncoding: 'utf-32be'});
|
|
|
|
// Encoder prepends BOM (which can be overridden with (addBOM: false}).
|
|
|
|
exports.utf32 = Utf32AutoCodec;
|
|
exports.ucs4 = 'utf32';
|
|
|
|
function Utf32AutoCodec(options, iconv) {
|
|
this.iconv = iconv;
|
|
}
|
|
|
|
Utf32AutoCodec.prototype.encoder = Utf32AutoEncoder;
|
|
Utf32AutoCodec.prototype.decoder = Utf32AutoDecoder;
|
|
|
|
// -- Encoding
|
|
|
|
function Utf32AutoEncoder(options, codec) {
|
|
options = options || {};
|
|
|
|
if (options.addBOM === undefined) options.addBOM = true;
|
|
|
|
this.encoder = codec.iconv.getEncoder(
|
|
options.defaultEncoding || 'utf-32le',
|
|
options
|
|
);
|
|
}
|
|
|
|
Utf32AutoEncoder.prototype.write = function (str) {
|
|
return this.encoder.write(str);
|
|
};
|
|
|
|
Utf32AutoEncoder.prototype.end = function () {
|
|
return this.encoder.end();
|
|
};
|
|
|
|
// -- Decoding
|
|
|
|
function Utf32AutoDecoder(options, codec) {
|
|
this.decoder = null;
|
|
this.initialBufs = [];
|
|
this.initialBufsLen = 0;
|
|
this.options = options || {};
|
|
this.iconv = codec.iconv;
|
|
}
|
|
|
|
Utf32AutoDecoder.prototype.write = function (buf) {
|
|
if (!this.decoder) {
|
|
// Codec is not chosen yet. Accumulate initial bytes.
|
|
this.initialBufs.push(buf);
|
|
this.initialBufsLen += buf.length;
|
|
|
|
if (this.initialBufsLen < 32)
|
|
// We need more bytes to use space heuristic (see below)
|
|
return '';
|
|
|
|
// We have enough bytes -> detect endianness.
|
|
var encoding = detectEncoding(
|
|
this.initialBufs,
|
|
this.options.defaultEncoding
|
|
);
|
|
this.decoder = this.iconv.getDecoder(encoding, this.options);
|
|
|
|
var resStr = '';
|
|
for (var i = 0; i < this.initialBufs.length; i++)
|
|
resStr += this.decoder.write(this.initialBufs[i]);
|
|
|
|
this.initialBufs.length = this.initialBufsLen = 0;
|
|
return resStr;
|
|
}
|
|
|
|
return this.decoder.write(buf);
|
|
};
|
|
|
|
Utf32AutoDecoder.prototype.end = function () {
|
|
if (!this.decoder) {
|
|
var encoding = detectEncoding(
|
|
this.initialBufs,
|
|
this.options.defaultEncoding
|
|
);
|
|
this.decoder = this.iconv.getDecoder(encoding, this.options);
|
|
|
|
var resStr = '';
|
|
for (var i = 0; i < this.initialBufs.length; i++)
|
|
resStr += this.decoder.write(this.initialBufs[i]);
|
|
|
|
var trail = this.decoder.end();
|
|
if (trail) resStr += trail;
|
|
|
|
this.initialBufs.length = this.initialBufsLen = 0;
|
|
return resStr;
|
|
}
|
|
|
|
return this.decoder.end();
|
|
};
|
|
|
|
function detectEncoding(bufs, defaultEncoding) {
|
|
var b = [];
|
|
var charsProcessed = 0;
|
|
var invalidLE = 0,
|
|
invalidBE = 0; // Number of invalid chars when decoded as LE or BE.
|
|
var bmpCharsLE = 0,
|
|
bmpCharsBE = 0; // Number of BMP chars when decoded as LE or BE.
|
|
|
|
outer_loop: for (var i = 0; i < bufs.length; i++) {
|
|
var buf = bufs[i];
|
|
for (var j = 0; j < buf.length; j++) {
|
|
b.push(buf[j]);
|
|
if (b.length === 4) {
|
|
if (charsProcessed === 0) {
|
|
// Check BOM first.
|
|
if (b[0] === 0xff && b[1] === 0xfe && b[2] === 0 && b[3] === 0) {
|
|
return 'utf-32le';
|
|
}
|
|
if (b[0] === 0 && b[1] === 0 && b[2] === 0xfe && b[3] === 0xff) {
|
|
return 'utf-32be';
|
|
}
|
|
}
|
|
|
|
if (b[0] !== 0 || b[1] > 0x10) invalidBE++;
|
|
if (b[3] !== 0 || b[2] > 0x10) invalidLE++;
|
|
|
|
if (b[0] === 0 && b[1] === 0 && (b[2] !== 0 || b[3] !== 0))
|
|
bmpCharsBE++;
|
|
if ((b[0] !== 0 || b[1] !== 0) && b[2] === 0 && b[3] === 0)
|
|
bmpCharsLE++;
|
|
|
|
b.length = 0;
|
|
charsProcessed++;
|
|
|
|
if (charsProcessed >= 100) {
|
|
break outer_loop;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
// Make decisions.
|
|
if (bmpCharsBE - invalidBE > bmpCharsLE - invalidLE) return 'utf-32be';
|
|
if (bmpCharsBE - invalidBE < bmpCharsLE - invalidLE) return 'utf-32le';
|
|
|
|
// Couldn't decide (likely all zeros or not enough data).
|
|
return defaultEncoding || 'utf-32le';
|
|
}
|