Spaces:
No application file
No application file
| ; | |
| 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'; | |
| } | |