GCC Code Coverage Report
Directory: ./ Exec Total Coverage
File: node_i18n.cc Lines: 366 416 88.0 %
Date: 2022-06-06 04:15:48 Branches: 140 216 64.8 %

Line Branch Exec Source
1
// Copyright Joyent, Inc. and other Node contributors.
2
//
3
// Permission is hereby granted, free of charge, to any person obtaining a
4
// copy of this software and associated documentation files (the
5
// "Software"), to deal in the Software without restriction, including
6
// without limitation the rights to use, copy, modify, merge, publish,
7
// distribute, sublicense, and/or sell copies of the Software, and to permit
8
// persons to whom the Software is furnished to do so, subject to the
9
// following conditions:
10
//
11
// The above copyright notice and this permission notice shall be included
12
// in all copies or substantial portions of the Software.
13
//
14
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
15
// OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
16
// MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN
17
// NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM,
18
// DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR
19
// OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE
20
// USE OR OTHER DEALINGS IN THE SOFTWARE.
21
22
/*
23
 * notes: by srl295
24
 *  - When in NODE_HAVE_SMALL_ICU mode, ICU is linked against "stub" (null) data
25
 *     ( stubdata/libicudata.a ) containing nothing, no data, and it's also
26
 *    linked against a "small" data file which the SMALL_ICUDATA_ENTRY_POINT
27
 *    macro names. That's the "english+root" data.
28
 *
29
 *    If icu_data_path is non-null, the user has provided a path and we assume
30
 *    it goes somewhere useful. We set that path in ICU, and exit.
31
 *    If icu_data_path is null, they haven't set a path and we want the
32
 *    "english+root" data.  We call
33
 *       udata_setCommonData(SMALL_ICUDATA_ENTRY_POINT,...)
34
 *    to load up the english+root data.
35
 *
36
 *  - when NOT in NODE_HAVE_SMALL_ICU mode, ICU is linked directly with its full
37
 *    data. All of the variables and command line options for changing data at
38
 *    runtime are disabled, as they wouldn't fully override the internal data.
39
 *    See:  http://bugs.icu-project.org/trac/ticket/10924
40
 */
41
42
43
#include "node_i18n.h"
44
#include "node_external_reference.h"
45
46
#if defined(NODE_HAVE_I18N_SUPPORT)
47
48
#include "base_object-inl.h"
49
#include "node.h"
50
#include "node_buffer.h"
51
#include "node_errors.h"
52
#include "node_internals.h"
53
#include "util-inl.h"
54
#include "v8.h"
55
56
#include <unicode/utypes.h>
57
#include <unicode/putil.h>
58
#include <unicode/uchar.h>
59
#include <unicode/uclean.h>
60
#include <unicode/udata.h>
61
#include <unicode/uidna.h>
62
#include <unicode/ucnv.h>
63
#include <unicode/utf8.h>
64
#include <unicode/utf16.h>
65
#include <unicode/timezone.h>
66
#include <unicode/ulocdata.h>
67
#include <unicode/uvernum.h>
68
#include <unicode/uversion.h>
69
#include <unicode/ustring.h>
70
71
#ifdef NODE_HAVE_SMALL_ICU
72
/* if this is defined, we have a 'secondary' entry point.
73
   compare following to utypes.h defs for U_ICUDATA_ENTRY_POINT */
74
#define SMALL_ICUDATA_ENTRY_POINT \
75
  SMALL_DEF2(U_ICU_VERSION_MAJOR_NUM, U_LIB_SUFFIX_C_NAME)
76
#define SMALL_DEF2(major, suff) SMALL_DEF(major, suff)
77
#ifndef U_LIB_SUFFIX_C_NAME
78
#define SMALL_DEF(major, suff) icusmdt##major##_dat
79
#else
80
#define SMALL_DEF(major, suff) icusmdt##suff##major##_dat
81
#endif
82
83
extern "C" const char U_DATA_API SMALL_ICUDATA_ENTRY_POINT[];
84
#endif
85
86
namespace node {
87
88
using v8::Context;
89
using v8::FunctionCallbackInfo;
90
using v8::FunctionTemplate;
91
using v8::Int32;
92
using v8::Isolate;
93
using v8::Local;
94
using v8::MaybeLocal;
95
using v8::NewStringType;
96
using v8::Object;
97
using v8::ObjectTemplate;
98
using v8::String;
99
using v8::Uint8Array;
100
using v8::Value;
101
102
namespace i18n {
103
namespace {
104
105
template <typename T>
106
4126
MaybeLocal<Object> ToBufferEndian(Environment* env, MaybeStackBuffer<T>* buf) {
107
4126
  MaybeLocal<Object> ret = Buffer::New(env, buf);
108
4126
  if (ret.IsEmpty())
109
    return ret;
110
111
  static_assert(sizeof(T) == 1 || sizeof(T) == 2,
112
                "Currently only one- or two-byte buffers are supported");
113
4116
  if (sizeof(T) > 1 && IsBigEndian()) {
114
    SPREAD_BUFFER_ARG(ret.ToLocalChecked(), retbuf);
115
    SwapBytes16(retbuf_data, retbuf_length);
116
  }
117
118
4126
  return ret;
119
}
120
121
// One-Shot Converters
122
123
2
void CopySourceBuffer(MaybeStackBuffer<UChar>* dest,
124
                      const char* data,
125
                      const size_t length,
126
                      const size_t length_in_chars) {
127
2
  dest->AllocateSufficientStorage(length_in_chars);
128
2
  char* dst = reinterpret_cast<char*>(**dest);
129
2
  memcpy(dst, data, length);
130
2
  if (IsBigEndian()) {
131
    SwapBytes16(dst, length);
132
  }
133
2
}
134
135
typedef MaybeLocal<Object> (*TranscodeFunc)(Environment* env,
136
                                            const char* fromEncoding,
137
                                            const char* toEncoding,
138
                                            const char* source,
139
                                            const size_t source_length,
140
                                            UErrorCode* status);
141
142
3
MaybeLocal<Object> Transcode(Environment* env,
143
                             const char* fromEncoding,
144
                             const char* toEncoding,
145
                             const char* source,
146
                             const size_t source_length,
147
                             UErrorCode* status) {
148
3
  *status = U_ZERO_ERROR;
149
  MaybeLocal<Object> ret;
150
6
  MaybeStackBuffer<char> result;
151
6
  Converter to(toEncoding);
152
6
  Converter from(fromEncoding);
153
154
3
  size_t sublen = ucnv_getMinCharSize(to.conv());
155
3
  std::string sub(sublen, '?');
156
3
  to.set_subst_chars(sub.c_str());
157
158
3
  const uint32_t limit = source_length * to.max_char_size();
159
3
  result.AllocateSufficientStorage(limit);
160
3
  char* target = *result;
161
3
  ucnv_convertEx(to.conv(), from.conv(), &target, target + limit,
162
                 &source, source + source_length, nullptr, nullptr,
163
                 nullptr, nullptr, true, true, status);
164
3
  if (U_SUCCESS(*status)) {
165
3
    result.SetLength(target - &result[0]);
166
3
    ret = ToBufferEndian(env, &result);
167
  }
168
3
  return ret;
169
}
170
171
4
MaybeLocal<Object> TranscodeToUcs2(Environment* env,
172
                                   const char* fromEncoding,
173
                                   const char* toEncoding,
174
                                   const char* source,
175
                                   const size_t source_length,
176
                                   UErrorCode* status) {
177
4
  *status = U_ZERO_ERROR;
178
  MaybeLocal<Object> ret;
179
8
  MaybeStackBuffer<UChar> destbuf(source_length);
180
4
  Converter from(fromEncoding);
181
4
  const size_t length_in_chars = source_length * sizeof(UChar);
182
4
  ucnv_toUChars(from.conv(), *destbuf, length_in_chars,
183
                source, source_length, status);
184
4
  if (U_SUCCESS(*status))
185
4
    ret = ToBufferEndian(env, &destbuf);
186
4
  return ret;
187
}
188
189
MaybeLocal<Object> TranscodeFromUcs2(Environment* env,
190
                                     const char* fromEncoding,
191
                                     const char* toEncoding,
192
                                     const char* source,
193
                                     const size_t source_length,
194
                                     UErrorCode* status) {
195
  *status = U_ZERO_ERROR;
196
  MaybeStackBuffer<UChar> sourcebuf;
197
  MaybeLocal<Object> ret;
198
  Converter to(toEncoding);
199
200
  size_t sublen = ucnv_getMinCharSize(to.conv());
201
  std::string sub(sublen, '?');
202
  to.set_subst_chars(sub.c_str());
203
204
  const size_t length_in_chars = source_length / sizeof(UChar);
205
  CopySourceBuffer(&sourcebuf, source, source_length, length_in_chars);
206
  MaybeStackBuffer<char> destbuf(length_in_chars);
207
  const uint32_t len = ucnv_fromUChars(to.conv(), *destbuf, length_in_chars,
208
                                       *sourcebuf, length_in_chars, status);
209
  if (U_SUCCESS(*status)) {
210
    destbuf.SetLength(len);
211
    ret = ToBufferEndian(env, &destbuf);
212
  }
213
  return ret;
214
}
215
216
2
MaybeLocal<Object> TranscodeUcs2FromUtf8(Environment* env,
217
                                         const char* fromEncoding,
218
                                         const char* toEncoding,
219
                                         const char* source,
220
                                         const size_t source_length,
221
                                         UErrorCode* status) {
222
2
  *status = U_ZERO_ERROR;
223
2
  MaybeStackBuffer<UChar> destbuf;
224
  int32_t result_length;
225
2
  u_strFromUTF8(*destbuf, destbuf.capacity(), &result_length,
226
                source, source_length, status);
227
  MaybeLocal<Object> ret;
228
2
  if (U_SUCCESS(*status)) {
229
1
    destbuf.SetLength(result_length);
230
1
    ret = ToBufferEndian(env, &destbuf);
231
1
  } else if (*status == U_BUFFER_OVERFLOW_ERROR) {
232
1
    *status = U_ZERO_ERROR;
233
1
    destbuf.AllocateSufficientStorage(result_length);
234
1
    u_strFromUTF8(*destbuf, result_length, &result_length,
235
                  source, source_length, status);
236
1
    if (U_SUCCESS(*status)) {
237
1
      destbuf.SetLength(result_length);
238
1
      ret = ToBufferEndian(env, &destbuf);
239
    }
240
  }
241
2
  return ret;
242
}
243
244
2
MaybeLocal<Object> TranscodeUtf8FromUcs2(Environment* env,
245
                                         const char* fromEncoding,
246
                                         const char* toEncoding,
247
                                         const char* source,
248
                                         const size_t source_length,
249
                                         UErrorCode* status) {
250
2
  *status = U_ZERO_ERROR;
251
  MaybeLocal<Object> ret;
252
2
  const size_t length_in_chars = source_length / sizeof(UChar);
253
  int32_t result_length;
254
4
  MaybeStackBuffer<UChar> sourcebuf;
255
2
  MaybeStackBuffer<char> destbuf;
256
2
  CopySourceBuffer(&sourcebuf, source, source_length, length_in_chars);
257
2
  u_strToUTF8(*destbuf, destbuf.capacity(), &result_length,
258
2
              *sourcebuf, length_in_chars, status);
259
2
  if (U_SUCCESS(*status)) {
260
1
    destbuf.SetLength(result_length);
261
1
    ret = ToBufferEndian(env, &destbuf);
262
1
  } else if (*status == U_BUFFER_OVERFLOW_ERROR) {
263
1
    *status = U_ZERO_ERROR;
264
1
    destbuf.AllocateSufficientStorage(result_length);
265
1
    u_strToUTF8(*destbuf, result_length, &result_length, *sourcebuf,
266
                length_in_chars, status);
267
1
    if (U_SUCCESS(*status)) {
268
1
      destbuf.SetLength(result_length);
269
1
      ret = ToBufferEndian(env, &destbuf);
270
    }
271
  }
272
2
  return ret;
273
}
274
275
22
const char* EncodingName(const enum encoding encoding) {
276

22
  switch (encoding) {
277
2
    case ASCII: return "us-ascii";
278
4
    case LATIN1: return "iso8859-1";
279
10
    case UCS2: return "utf16le";
280
6
    case UTF8: return "utf-8";
281
    default: return nullptr;
282
  }
283
}
284
285
24
bool SupportedEncoding(const enum encoding encoding) {
286
24
  switch (encoding) {
287
22
    case ASCII:
288
    case LATIN1:
289
    case UCS2:
290
22
    case UTF8: return true;
291
2
    default: return false;
292
  }
293
}
294
295
13
void Transcode(const FunctionCallbackInfo<Value>&args) {
296
13
  Environment* env = Environment::GetCurrent(args);
297
13
  Isolate* isolate = env->isolate();
298
13
  UErrorCode status = U_ZERO_ERROR;
299
  MaybeLocal<Object> result;
300
301
13
  ArrayBufferViewContents<char> input(args[0]);
302
13
  const enum encoding fromEncoding = ParseEncoding(isolate, args[1], BUFFER);
303
13
  const enum encoding toEncoding = ParseEncoding(isolate, args[2], BUFFER);
304
305

13
  if (SupportedEncoding(fromEncoding) && SupportedEncoding(toEncoding)) {
306
11
    TranscodeFunc tfn = &Transcode;
307

11
    switch (fromEncoding) {
308
4
      case ASCII:
309
      case LATIN1:
310
4
        if (toEncoding == UCS2)
311
4
          tfn = &TranscodeToUcs2;
312
4
        break;
313
4
      case UTF8:
314
4
        if (toEncoding == UCS2)
315
2
          tfn = &TranscodeUcs2FromUtf8;
316
4
        break;
317
3
      case UCS2:
318
3
        switch (toEncoding) {
319
1
          case UCS2:
320
1
            tfn = &Transcode;
321
1
            break;
322
2
          case UTF8:
323
2
            tfn = &TranscodeUtf8FromUcs2;
324
2
            break;
325
          default:
326
            tfn = &TranscodeFromUcs2;
327
        }
328
3
        break;
329
      default:
330
        // This should not happen because of the SupportedEncoding checks
331
        ABORT();
332
    }
333
334
    result = tfn(env, EncodingName(fromEncoding), EncodingName(toEncoding),
335
11
                 input.data(), input.length(), &status);
336
  } else {
337
2
    status = U_ILLEGAL_ARGUMENT_ERROR;
338
  }
339
340
13
  if (result.IsEmpty())
341
4
    return args.GetReturnValue().Set(status);
342
343
22
  return args.GetReturnValue().Set(result.ToLocalChecked());
344
}
345
346
2
void ICUErrorName(const FunctionCallbackInfo<Value>& args) {
347
2
  Environment* env = Environment::GetCurrent(args);
348
2
  CHECK(args[0]->IsInt32());
349
4
  UErrorCode status = static_cast<UErrorCode>(args[0].As<Int32>()->Value());
350
6
  args.GetReturnValue().Set(
351
2
      String::NewFromUtf8(env->isolate(),
352
2
                          u_errorName(status)).ToLocalChecked());
353
2
}
354
355
}  // anonymous namespace
356
357
10
Converter::Converter(const char* name, const char* sub) {
358
10
  UErrorCode status = U_ZERO_ERROR;
359
10
  UConverter* conv = ucnv_open(name, &status);
360
10
  CHECK(U_SUCCESS(status));
361
10
  conv_.reset(conv);
362
10
  set_subst_chars(sub);
363
10
}
364
365
4533
Converter::Converter(UConverter* converter, const char* sub)
366
4533
    : conv_(converter) {
367
4533
  set_subst_chars(sub);
368
4533
}
369
370
4546
void Converter::set_subst_chars(const char* sub) {
371
4546
  CHECK(conv_);
372
4546
  UErrorCode status = U_ZERO_ERROR;
373
4546
  if (sub != nullptr) {
374
3
    ucnv_setSubstChars(conv_.get(), sub, strlen(sub), &status);
375
3
    CHECK(U_SUCCESS(status));
376
  }
377
4546
}
378
379
1359
void Converter::reset() {
380
1359
  ucnv_reset(conv_.get());
381
1359
}
382
383
2149
size_t Converter::min_char_size() const {
384
2149
  CHECK(conv_);
385
2149
  return ucnv_getMinCharSize(conv_.get());
386
}
387
388
3
size_t Converter::max_char_size() const {
389
3
  CHECK(conv_);
390
3
  return ucnv_getMaxCharSize(conv_.get());
391
}
392
393
2
void ConverterObject::Has(const FunctionCallbackInfo<Value>& args) {
394
2
  Environment* env = Environment::GetCurrent(args);
395
396
2
  CHECK_GE(args.Length(), 1);
397
4
  Utf8Value label(env->isolate(), args[0]);
398
399
2
  UErrorCode status = U_ZERO_ERROR;
400
2
  ConverterPointer conv(ucnv_open(*label, &status));
401
4
  args.GetReturnValue().Set(!!U_SUCCESS(status));
402
2
}
403
404
4533
void ConverterObject::Create(const FunctionCallbackInfo<Value>& args) {
405
4533
  Environment* env = Environment::GetCurrent(args);
406
407
4533
  Local<ObjectTemplate> t = env->i18n_converter_template();
408
  Local<Object> obj;
409
9066
  if (!t->NewInstance(env->context()).ToLocal(&obj)) return;
410
411
4533
  CHECK_GE(args.Length(), 2);
412
4533
  Utf8Value label(env->isolate(), args[0]);
413
4533
  int flags = args[1]->Uint32Value(env->context()).ToChecked();
414
4533
  bool fatal =
415
4533
      (flags & CONVERTER_FLAGS_FATAL) == CONVERTER_FLAGS_FATAL;
416
417
4533
  UErrorCode status = U_ZERO_ERROR;
418
4533
  UConverter* conv = ucnv_open(*label, &status);
419
4533
  if (U_FAILURE(status))
420
    return;
421
422
4533
  if (fatal) {
423
337
    status = U_ZERO_ERROR;
424
337
    ucnv_setToUCallBack(conv, UCNV_TO_U_CALLBACK_STOP,
425
                        nullptr, nullptr, nullptr, &status);
426
  }
427
428
4533
  new ConverterObject(env, obj, conv, flags);
429
9066
  args.GetReturnValue().Set(obj);
430
}
431
432
2149
void ConverterObject::Decode(const FunctionCallbackInfo<Value>& args) {
433
2149
  Environment* env = Environment::GetCurrent(args);
434
435
2149
  CHECK_GE(args.Length(), 3);  // Converter, Buffer, Flags
436
437
  ConverterObject* converter;
438
6350
  ASSIGN_OR_RETURN_UNWRAP(&converter, args[0].As<Object>());
439
2149
  ArrayBufferViewContents<char> input(args[1]);
440
2149
  int flags = args[2]->Uint32Value(env->context()).ToChecked();
441
442
2149
  UErrorCode status = U_ZERO_ERROR;
443
2149
  MaybeStackBuffer<UChar> result;
444
  MaybeLocal<Object> ret;
445
446
2149
  UBool flush = (flags & CONVERTER_FLAGS_FLUSH) == CONVERTER_FLAGS_FLUSH;
447
448
  // When flushing the final chunk, the limit is the maximum
449
  // of either the input buffer length or the number of pending
450
  // characters times the min char size, multiplied by 2 as unicode may
451
  // take up to 2 UChars to encode a character
452
2149
  size_t limit = 2 * converter->min_char_size() *
453
3508
      (!flush ?
454
790
          input.length() :
455
          std::max(
456
4867
              input.length(),
457
1359
              static_cast<size_t>(
458
2718
                  ucnv_toUCountPending(converter->conv(), &status))));
459
2149
  status = U_ZERO_ERROR;
460
461
2149
  if (limit > 0)
462
2007
    result.AllocateSufficientStorage(limit);
463
464
2149
  auto cleanup = OnScopeLeave([&]() {
465
2149
    if (flush) {
466
      // Reset the converter state.
467
1359
      converter->set_bom_seen(false);
468
1359
      converter->reset();
469
    }
470
2149
  });
471
472
2149
  const char* source = input.data();
473
2149
  size_t source_length = input.length();
474
475
2149
  UChar* target = *result;
476
2149
  ucnv_toUnicode(converter->conv(),
477
                 &target,
478
2149
                 target + limit,
479
                 &source,
480
                 source + source_length,
481
                 nullptr,
482
                 flush,
483
                 &status);
484
485
2149
  if (U_SUCCESS(status)) {
486
2052
    bool omit_initial_bom = false;
487
2052
    if (limit > 0) {
488
1910
      result.SetLength(target - &result[0]);
489
3626
      if (result.length() > 0 &&
490
1716
          converter->unicode() &&
491

5269
          !converter->ignore_bom() &&
492
1643
          !converter->bom_seen()) {
493
        // If the very first result in the stream is a BOM, and we are not
494
        // explicitly told to ignore it, then we mark it for discarding.
495
1178
        if (result[0] == 0xFEFF)
496
44
          omit_initial_bom = true;
497
1178
        converter->set_bom_seen(true);
498
      }
499
    }
500
2052
    ret = ToBufferEndian(env, &result);
501

2096
    if (omit_initial_bom && !ret.IsEmpty()) {
502
      // Perform `ret = ret.slice(2)`.
503
44
      CHECK(ret.ToLocalChecked()->IsUint8Array());
504
88
      Local<Uint8Array> orig_ret = ret.ToLocalChecked().As<Uint8Array>();
505
44
      ret = Buffer::New(env,
506
44
                        orig_ret->Buffer(),
507
44
                        orig_ret->ByteOffset() + 2,
508
88
                        orig_ret->ByteLength() - 2)
509
                            .FromMaybe(Local<Uint8Array>());
510
    }
511
2052
    if (!ret.IsEmpty())
512
4104
      args.GetReturnValue().Set(ret.ToLocalChecked());
513
2052
    return;
514
  }
515
516
194
  args.GetReturnValue().Set(status);
517
}
518
519
4533
ConverterObject::ConverterObject(
520
    Environment* env,
521
    Local<Object> wrap,
522
    UConverter* converter,
523
    int flags,
524
4533
    const char* sub)
525
    : BaseObject(env, wrap),
526
      Converter(converter, sub),
527
4533
      flags_(flags) {
528
4533
  MakeWeak();
529
530
4533
  switch (ucnv_getType(converter)) {
531
971
    case UCNV_UTF8:
532
    case UCNV_UTF16_BigEndian:
533
    case UCNV_UTF16_LittleEndian:
534
971
      flags_ |= CONVERTER_FLAGS_UNICODE;
535
971
      break;
536
4533
    default: {
537
      // Fall through
538
    }
539
  }
540
4533
}
541
542
543
5237
bool InitializeICUDirectory(const std::string& path) {
544
5237
  UErrorCode status = U_ZERO_ERROR;
545
5237
  if (path.empty()) {
546
#ifdef NODE_HAVE_SMALL_ICU
547
    // install the 'small' data.
548
    udata_setCommonData(&SMALL_ICUDATA_ENTRY_POINT, &status);
549
#else  // !NODE_HAVE_SMALL_ICU
550
    // no small data, so nothing to do.
551
#endif  // !NODE_HAVE_SMALL_ICU
552
  } else {
553
    u_setDataDirectory(path.c_str());
554
    u_init(&status);
555
  }
556
5237
  return status == U_ZERO_ERROR;
557
}
558
559
void SetDefaultTimeZone(const char* tzid) {
560
  size_t tzidlen = strlen(tzid) + 1;
561
  UErrorCode status = U_ZERO_ERROR;
562
  MaybeStackBuffer<UChar, 256> id(tzidlen);
563
  u_charsToUChars(tzid, id.out(), tzidlen);
564
  // This is threadsafe:
565
  ucal_setDefaultTimeZone(id.out(), &status);
566
  CHECK(U_SUCCESS(status));
567
}
568
569
384
int32_t ToUnicode(MaybeStackBuffer<char>* buf,
570
                  const char* input,
571
                  size_t length) {
572
384
  UErrorCode status = U_ZERO_ERROR;
573
384
  uint32_t options = UIDNA_NONTRANSITIONAL_TO_UNICODE;
574
384
  UIDNA* uidna = uidna_openUTS46(options, &status);
575
384
  if (U_FAILURE(status))
576
    return -1;
577
384
  UIDNAInfo info = UIDNA_INFO_INITIALIZER;
578
579
384
  int32_t len = uidna_nameToUnicodeUTF8(uidna,
580
                                        input, length,
581
384
                                        **buf, buf->capacity(),
582
                                        &info,
583
                                        &status);
584
585
  // Do not check info.errors like we do with ToASCII since ToUnicode always
586
  // returns a string, despite any possible errors that may have occurred.
587
588
384
  if (status == U_BUFFER_OVERFLOW_ERROR) {
589
    status = U_ZERO_ERROR;
590
    buf->AllocateSufficientStorage(len);
591
    len = uidna_nameToUnicodeUTF8(uidna,
592
                                  input, length,
593
                                  **buf, buf->capacity(),
594
                                  &info,
595
                                  &status);
596
  }
597
598
  // info.errors is ignored as UTS #46 ToUnicode always produces a Unicode
599
  // string, regardless of whether an error occurred.
600
601
384
  if (U_FAILURE(status)) {
602
    len = -1;
603
    buf->SetLength(0);
604
  } else {
605
384
    buf->SetLength(len);
606
  }
607
608
384
  uidna_close(uidna);
609
384
  return len;
610
}
611
612
15060
int32_t ToASCII(MaybeStackBuffer<char>* buf,
613
                const char* input,
614
                size_t length,
615
                enum idna_mode mode) {
616
15060
  UErrorCode status = U_ZERO_ERROR;
617
15060
  uint32_t options =                  // CheckHyphens = false; handled later
618
    UIDNA_CHECK_BIDI |                // CheckBidi = true
619
    UIDNA_CHECK_CONTEXTJ |            // CheckJoiners = true
620
    UIDNA_NONTRANSITIONAL_TO_ASCII;   // Nontransitional_Processing
621
15060
  if (mode == IDNA_STRICT) {
622
    options |= UIDNA_USE_STD3_RULES;  // UseSTD3ASCIIRules = beStrict
623
                                      // VerifyDnsLength = beStrict;
624
                                      //   handled later
625
  }
626
627
15060
  UIDNA* uidna = uidna_openUTS46(options, &status);
628
15060
  if (U_FAILURE(status))
629
    return -1;
630
15060
  UIDNAInfo info = UIDNA_INFO_INITIALIZER;
631
632
15060
  int32_t len = uidna_nameToASCII_UTF8(uidna,
633
                                       input, length,
634
15060
                                       **buf, buf->capacity(),
635
                                       &info,
636
                                       &status);
637
638
15060
  if (status == U_BUFFER_OVERFLOW_ERROR) {
639
2
    status = U_ZERO_ERROR;
640
2
    buf->AllocateSufficientStorage(len);
641
2
    len = uidna_nameToASCII_UTF8(uidna,
642
                                 input, length,
643
2
                                 **buf, buf->capacity(),
644
                                 &info,
645
                                 &status);
646
  }
647
648
  // In UTS #46 which specifies ToASCII, certain error conditions are
649
  // configurable through options, and the WHATWG URL Standard promptly elects
650
  // to disable some of them to accommodate for real-world use cases.
651
  // Unfortunately, ICU4C's IDNA module does not support disabling some of
652
  // these options through `options` above, and thus continues throwing
653
  // unnecessary errors. To counter this situation, we just filter out the
654
  // errors that may have happened afterwards, before deciding whether to
655
  // return an error from this function.
656
657
  // CheckHyphens = false
658
  // (Specified in the current UTS #46 draft rev. 18.)
659
  // Refs:
660
  // - https://github.com/whatwg/url/issues/53
661
  // - https://github.com/whatwg/url/pull/309
662
  // - http://www.unicode.org/review/pri317/
663
  // - http://www.unicode.org/reports/tr46/tr46-18.html
664
  // - https://www.icann.org/news/announcement-2000-01-07-en
665
15060
  info.errors &= ~UIDNA_ERROR_HYPHEN_3_4;
666
15060
  info.errors &= ~UIDNA_ERROR_LEADING_HYPHEN;
667
15060
  info.errors &= ~UIDNA_ERROR_TRAILING_HYPHEN;
668
669
15060
  if (mode != IDNA_STRICT) {
670
    // VerifyDnsLength = beStrict
671
15060
    info.errors &= ~UIDNA_ERROR_EMPTY_LABEL;
672
15060
    info.errors &= ~UIDNA_ERROR_LABEL_TOO_LONG;
673
15060
    info.errors &= ~UIDNA_ERROR_DOMAIN_NAME_TOO_LONG;
674
  }
675
676


15060
  if (U_FAILURE(status) || (mode != IDNA_LENIENT && info.errors != 0)) {
677
133
    len = -1;
678
133
    buf->SetLength(0);
679
  } else {
680
14927
    buf->SetLength(len);
681
  }
682
683
15060
  uidna_close(uidna);
684
15060
  return len;
685
}
686
687
189
static void ToUnicode(const FunctionCallbackInfo<Value>& args) {
688
189
  Environment* env = Environment::GetCurrent(args);
689
189
  CHECK_GE(args.Length(), 1);
690
378
  CHECK(args[0]->IsString());
691
189
  Utf8Value val(env->isolate(), args[0]);
692
693
189
  MaybeStackBuffer<char> buf;
694
189
  int32_t len = ToUnicode(&buf, *val, val.length());
695
696
189
  if (len < 0) {
697
    return THROW_ERR_INVALID_ARG_VALUE(env, "Cannot convert name to Unicode");
698
  }
699
700
567
  args.GetReturnValue().Set(
701
189
      String::NewFromUtf8(env->isolate(),
702
189
                          *buf,
703
                          NewStringType::kNormal,
704
189
                          len).ToLocalChecked());
705
}
706
707
10129
static void ToASCII(const FunctionCallbackInfo<Value>& args) {
708
10129
  Environment* env = Environment::GetCurrent(args);
709
10129
  CHECK_GE(args.Length(), 1);
710
20258
  CHECK(args[0]->IsString());
711
10129
  Utf8Value val(env->isolate(), args[0]);
712
  // optional arg
713
10129
  bool lenient = args[1]->BooleanValue(env->isolate());
714
10129
  enum idna_mode mode = lenient ? IDNA_LENIENT : IDNA_DEFAULT;
715
716
10129
  MaybeStackBuffer<char> buf;
717
10129
  int32_t len = ToASCII(&buf, *val, val.length(), mode);
718
719
10129
  if (len < 0) {
720
9
    return THROW_ERR_INVALID_ARG_VALUE(env, "Cannot convert name to ASCII");
721
  }
722
723
30360
  args.GetReturnValue().Set(
724
10120
      String::NewFromUtf8(env->isolate(),
725
10120
                          *buf,
726
                          NewStringType::kNormal,
727
10120
                          len).ToLocalChecked());
728
}
729
730
// This is similar to wcwidth except that it takes the current unicode
731
// character properties database into consideration, allowing it to
732
// correctly calculate the column widths of things like emoji's and
733
// newer wide characters. wcwidth, on the other hand, uses a fixed
734
// algorithm that does not take things like emoji into proper
735
// consideration.
736
//
737
// TODO(TimothyGu): Investigate Cc (C0/C1 control codes). Both VTE (used by
738
// GNOME Terminal) and Konsole don't consider them to be zero-width (see refs
739
// below), and when printed in VTE it is Narrow. However GNOME Terminal doesn't
740
// allow it to be input. Linux's PTY terminal prints control characters as
741
// Narrow rhombi.
742
//
743
// TODO(TimothyGu): Investigate Hangul jamo characters. Medial vowels and final
744
// consonants are 0-width when combined with initial consonants; otherwise they
745
// are technically Wide. But many terminals (including Konsole and
746
// VTE/GLib-based) implement all medials and finals as 0-width.
747
//
748
// Refs: https://eev.ee/blog/2015/09/12/dark-corners-of-unicode/#combining-characters-and-character-width
749
// Refs: https://github.com/GNOME/glib/blob/79e4d4c6be/glib/guniprop.c#L388-L420
750
// Refs: https://github.com/KDE/konsole/blob/8c6a5d13c0/src/konsole_wcwidth.cpp#L101-L223
751
1235236
static int GetColumnWidth(UChar32 codepoint,
752
                          bool ambiguous_as_full_width = false) {
753
  // UCHAR_EAST_ASIAN_WIDTH is the Unicode property that identifies a
754
  // codepoint as being full width, wide, ambiguous, neutral, narrow,
755
  // or halfwidth.
756
1235236
  const int eaw = u_getIntPropertyValue(codepoint, UCHAR_EAST_ASIAN_WIDTH);
757

1235236
  switch (eaw) {
758
68370
    case U_EA_FULLWIDTH:
759
    case U_EA_WIDE:
760
68370
      return 2;
761
1012668
    case U_EA_AMBIGUOUS:
762
      // See: http://www.unicode.org/reports/tr11/#Ambiguous for details
763
1012668
      if (ambiguous_as_full_width) {
764
        return 2;
765
      }
766
      // If ambiguous_as_full_width is false:
767
      // Fall through
768
    case U_EA_NEUTRAL:
769
1142412
      if (u_hasBinaryProperty(codepoint, UCHAR_EMOJI_PRESENTATION)) {
770
        return 2;
771
      }
772
      // Fall through
773
    case U_EA_HALFWIDTH:
774
    case U_EA_NARROW:
775
    default:
776
1166866
      const auto zero_width_mask = U_GC_CC_MASK |  // C0/C1 control code
777
                                  U_GC_CF_MASK |  // Format control character
778
                                  U_GC_ME_MASK |  // Enclosing mark
779
                                  U_GC_MN_MASK;   // Nonspacing mark
780

2333686
      if (codepoint != 0x00AD &&  // SOFT HYPHEN is Cf but not zero-width
781
1328383
          ((U_MASK(u_charType(codepoint)) & zero_width_mask) ||
782
1166820
          u_hasBinaryProperty(codepoint, UCHAR_EMOJI_MODIFIER))) {
783
1005257
        return 0;
784
      }
785
161609
      return 1;
786
  }
787
}
788
789
// Returns the column width for the given String.
790
1196035
static void GetStringWidth(const FunctionCallbackInfo<Value>& args) {
791
1196035
  Environment* env = Environment::GetCurrent(args);
792
2392070
  CHECK(args[0]->IsString());
793
794
1196035
  bool ambiguous_as_full_width = args[1]->IsTrue();
795

1196035
  bool expand_emoji_sequence = !args[2]->IsBoolean() || args[2]->IsTrue();
796
797
1196035
  TwoByteValue value(env->isolate(), args[0]);
798
  // reinterpret_cast is required by windows to compile
799
1196035
  UChar* str = reinterpret_cast<UChar*>(*value);
800
  static_assert(sizeof(*str) == sizeof(**value),
801
                "sizeof(*str) == sizeof(**value)");
802
1196035
  UChar32 c = 0;
803
  UChar32 p;
804
1196035
  size_t n = 0;
805
1196035
  uint32_t width = 0;
806
807
2431271
  while (n < value.length()) {
808
1235236
    p = c;
809


1235236
    U16_NEXT(str, n, value.length(), c);
810
    // Don't count individual emoji codepoints that occur within an
811
    // emoji sequence. This is not necessarily foolproof. Some
812
    // environments display emoji sequences in the appropriate
813
    // condensed form (as a single emoji glyph), other environments
814
    // may not understand an emoji sequence and will display each
815
    // individual emoji separately. When this happens, the width
816
    // calculated will be off, and there's no reliable way of knowing
817
    // in advance if a particular sequence is going to be supported.
818
    // The expand_emoji_sequence option allows the caller to skip this
819
    // check and count each code within an emoji sequence separately.
820
    // https://www.unicode.org/reports/tr51/tr51-16.html#Emoji_ZWJ_Sequences
821
2470472
    if (!expand_emoji_sequence &&
822


1235236
        n > 0 && p == 0x200d &&  // 0x200d == ZWJ (zero width joiner)
823
        (u_hasBinaryProperty(c, UCHAR_EMOJI_PRESENTATION) ||
824
         u_hasBinaryProperty(c, UCHAR_EMOJI_MODIFIER))) {
825
      continue;
826
    }
827
1235236
    width += GetColumnWidth(c, ambiguous_as_full_width);
828
  }
829
2392070
  args.GetReturnValue().Set(width);
830
1196035
}
831
832
854
void Initialize(Local<Object> target,
833
                Local<Value> unused,
834
                Local<Context> context,
835
                void* priv) {
836
854
  Environment* env = Environment::GetCurrent(context);
837
854
  env->SetMethod(target, "toUnicode", ToUnicode);
838
854
  env->SetMethod(target, "toASCII", ToASCII);
839
854
  env->SetMethod(target, "getStringWidth", GetStringWidth);
840
841
  // One-shot converters
842
854
  env->SetMethod(target, "icuErrName", ICUErrorName);
843
854
  env->SetMethod(target, "transcode", Transcode);
844
845
  // ConverterObject
846
  {
847
854
    Local<FunctionTemplate> t = FunctionTemplate::New(env->isolate());
848
854
    t->Inherit(BaseObject::GetConstructorTemplate(env));
849
1708
    t->InstanceTemplate()->SetInternalFieldCount(
850
        ConverterObject::kInternalFieldCount);
851
    Local<String> converter_string =
852
854
        FIXED_ONE_BYTE_STRING(env->isolate(), "Converter");
853
854
    t->SetClassName(converter_string);
854
854
    env->set_i18n_converter_template(t->InstanceTemplate());
855
  }
856
857
854
  env->SetMethod(target, "getConverter", ConverterObject::Create);
858
854
  env->SetMethod(target, "decode", ConverterObject::Decode);
859
854
  env->SetMethod(target, "hasConverter", ConverterObject::Has);
860
854
}
861
862
5206
void RegisterExternalReferences(ExternalReferenceRegistry* registry) {
863
5206
  registry->Register(ToUnicode);
864
5206
  registry->Register(ToASCII);
865
5206
  registry->Register(GetStringWidth);
866
5206
  registry->Register(ICUErrorName);
867
5206
  registry->Register(Transcode);
868
5206
  registry->Register(ConverterObject::Create);
869
5206
  registry->Register(ConverterObject::Decode);
870
5206
  registry->Register(ConverterObject::Has);
871
5206
}
872
873
}  // namespace i18n
874
}  // namespace node
875
876
5274
NODE_MODULE_CONTEXT_AWARE_INTERNAL(icu, node::i18n::Initialize)
877
5206
NODE_MODULE_EXTERNAL_REFERENCE(icu, node::i18n::RegisterExternalReferences)
878
879
#endif  // NODE_HAVE_I18N_SUPPORT