1// Copyright 2014 PDFium Authors. All rights reserved.
2// Use of this source code is governed by a BSD-style license that can be
3// found in the LICENSE file.
4
5// Original code copyright 2014 Foxit Software Inc. http://www.foxitsoftware.com
6
7#include <stddef.h>  // For offsetof().
8
9#include <algorithm>
10#include <cctype>
11
12#include "core/include/fxcrt/fx_basic.h"
13#include "core/include/fxcrt/fx_ext.h"
14#include "third_party/base/numerics/safe_math.h"
15
16// static
17CFX_WideString::StringData* CFX_WideString::StringData::Create(int nLen) {
18  // TODO(palmer): |nLen| should really be declared as |size_t|, or
19  // at least unsigned.
20  if (nLen == 0 || nLen < 0) {
21    return NULL;
22  }
23
24  // Fixed portion of header plus a NUL wide char not in m_nAllocLength.
25  int overhead = offsetof(StringData, m_String) + sizeof(FX_WCHAR);
26  pdfium::base::CheckedNumeric<int> iSize = nLen;
27  iSize *= sizeof(FX_WCHAR);
28  iSize += overhead;
29
30  // Now round to an 8-byte boundary. We'd expect that this is the minimum
31  // granularity of any of the underlying allocators, so there may be cases
32  // where we can save a re-alloc when adding a few characters to a string
33  // by using this otherwise wasted space.
34  iSize += 7;
35  int totalSize = iSize.ValueOrDie() & ~7;
36  int usableLen = (totalSize - overhead) / sizeof(FX_WCHAR);
37  FXSYS_assert(usableLen >= nLen);
38
39  void* pData = FX_Alloc(uint8_t, totalSize);
40  return new (pData) StringData(nLen, usableLen);
41}
42CFX_WideString::~CFX_WideString() {
43  if (m_pData) {
44    m_pData->Release();
45  }
46}
47CFX_WideString::CFX_WideString(const CFX_WideString& stringSrc) {
48  if (!stringSrc.m_pData) {
49    m_pData = NULL;
50    return;
51  }
52  if (stringSrc.m_pData->m_nRefs >= 0) {
53    m_pData = stringSrc.m_pData;
54    m_pData->Retain();
55  } else {
56    m_pData = NULL;
57    *this = stringSrc;
58  }
59}
60CFX_WideString::CFX_WideString(const FX_WCHAR* lpsz, FX_STRSIZE nLen) {
61  if (nLen < 0) {
62    nLen = lpsz ? FXSYS_wcslen(lpsz) : 0;
63  }
64  if (nLen) {
65    m_pData = StringData::Create(nLen);
66    if (m_pData) {
67      FXSYS_memcpy(m_pData->m_String, lpsz, nLen * sizeof(FX_WCHAR));
68    }
69  } else {
70    m_pData = NULL;
71  }
72}
73CFX_WideString::CFX_WideString(FX_WCHAR ch) {
74  m_pData = StringData::Create(1);
75  if (m_pData) {
76    m_pData->m_String[0] = ch;
77  }
78}
79CFX_WideString::CFX_WideString(const CFX_WideStringC& str) {
80  if (str.IsEmpty()) {
81    m_pData = NULL;
82    return;
83  }
84  m_pData = StringData::Create(str.GetLength());
85  if (m_pData) {
86    FXSYS_memcpy(m_pData->m_String, str.GetPtr(),
87                 str.GetLength() * sizeof(FX_WCHAR));
88  }
89}
90CFX_WideString::CFX_WideString(const CFX_WideStringC& str1,
91                               const CFX_WideStringC& str2) {
92  m_pData = NULL;
93  int nNewLen = str1.GetLength() + str2.GetLength();
94  if (nNewLen == 0) {
95    return;
96  }
97  m_pData = StringData::Create(nNewLen);
98  if (m_pData) {
99    FXSYS_memcpy(m_pData->m_String, str1.GetPtr(),
100                 str1.GetLength() * sizeof(FX_WCHAR));
101    FXSYS_memcpy(m_pData->m_String + str1.GetLength(), str2.GetPtr(),
102                 str2.GetLength() * sizeof(FX_WCHAR));
103  }
104}
105void CFX_WideString::ReleaseBuffer(FX_STRSIZE nNewLength) {
106  if (!m_pData) {
107    return;
108  }
109  CopyBeforeWrite();
110  if (nNewLength == -1) {
111    nNewLength = m_pData ? FXSYS_wcslen(m_pData->m_String) : 0;
112  }
113  if (nNewLength == 0) {
114    Empty();
115    return;
116  }
117  FXSYS_assert(nNewLength <= m_pData->m_nAllocLength);
118  m_pData->m_nDataLength = nNewLength;
119  m_pData->m_String[nNewLength] = 0;
120}
121const CFX_WideString& CFX_WideString::operator=(const FX_WCHAR* lpsz) {
122  if (!lpsz || lpsz[0] == 0) {
123    Empty();
124  } else {
125    AssignCopy(FXSYS_wcslen(lpsz), lpsz);
126  }
127  return *this;
128}
129const CFX_WideString& CFX_WideString::operator=(
130    const CFX_WideStringC& stringSrc) {
131  if (stringSrc.IsEmpty()) {
132    Empty();
133  } else {
134    AssignCopy(stringSrc.GetLength(), stringSrc.GetPtr());
135  }
136  return *this;
137}
138const CFX_WideString& CFX_WideString::operator=(
139    const CFX_WideString& stringSrc) {
140  if (m_pData == stringSrc.m_pData) {
141    return *this;
142  }
143  if (stringSrc.IsEmpty()) {
144    Empty();
145  } else if ((m_pData && m_pData->m_nRefs < 0) ||
146             (stringSrc.m_pData && stringSrc.m_pData->m_nRefs < 0)) {
147    AssignCopy(stringSrc.m_pData->m_nDataLength, stringSrc.m_pData->m_String);
148  } else {
149    Empty();
150    m_pData = stringSrc.m_pData;
151    if (m_pData) {
152      m_pData->Retain();
153    }
154  }
155  return *this;
156}
157const CFX_WideString& CFX_WideString::operator+=(FX_WCHAR ch) {
158  ConcatInPlace(1, &ch);
159  return *this;
160}
161const CFX_WideString& CFX_WideString::operator+=(const FX_WCHAR* lpsz) {
162  if (lpsz) {
163    ConcatInPlace(FXSYS_wcslen(lpsz), lpsz);
164  }
165  return *this;
166}
167const CFX_WideString& CFX_WideString::operator+=(const CFX_WideString& string) {
168  if (!string.m_pData) {
169    return *this;
170  }
171  ConcatInPlace(string.m_pData->m_nDataLength, string.m_pData->m_String);
172  return *this;
173}
174const CFX_WideString& CFX_WideString::operator+=(
175    const CFX_WideStringC& string) {
176  if (string.IsEmpty()) {
177    return *this;
178  }
179  ConcatInPlace(string.GetLength(), string.GetPtr());
180  return *this;
181}
182bool CFX_WideString::Equal(const wchar_t* ptr) const {
183  if (!m_pData) {
184    return !ptr || ptr[0] == L'\0';
185  }
186  if (!ptr) {
187    return m_pData->m_nDataLength == 0;
188  }
189  return wcslen(ptr) == m_pData->m_nDataLength &&
190         wmemcmp(ptr, m_pData->m_String, m_pData->m_nDataLength) == 0;
191}
192bool CFX_WideString::Equal(const CFX_WideStringC& str) const {
193  if (!m_pData) {
194    return str.IsEmpty();
195  }
196  return str.GetLength() == m_pData->m_nDataLength &&
197         wmemcmp(str.GetPtr(), m_pData->m_String, m_pData->m_nDataLength) == 0;
198}
199bool CFX_WideString::Equal(const CFX_WideString& other) const {
200  if (IsEmpty()) {
201    return other.IsEmpty();
202  }
203  if (other.IsEmpty()) {
204    return false;
205  }
206  return other.m_pData->m_nDataLength == m_pData->m_nDataLength &&
207         wmemcmp(other.m_pData->m_String, m_pData->m_String,
208                 m_pData->m_nDataLength) == 0;
209}
210void CFX_WideString::Empty() {
211  if (m_pData) {
212    m_pData->Release();
213    m_pData = NULL;
214  }
215}
216void CFX_WideString::ConcatInPlace(FX_STRSIZE nSrcLen,
217                                   const FX_WCHAR* lpszSrcData) {
218  if (nSrcLen == 0 || !lpszSrcData) {
219    return;
220  }
221  if (!m_pData) {
222    m_pData = StringData::Create(nSrcLen);
223    if (m_pData) {
224      FXSYS_memcpy(m_pData->m_String, lpszSrcData, nSrcLen * sizeof(FX_WCHAR));
225    }
226    return;
227  }
228  if (m_pData->m_nRefs > 1 ||
229      m_pData->m_nDataLength + nSrcLen > m_pData->m_nAllocLength) {
230    ConcatCopy(m_pData->m_nDataLength, m_pData->m_String, nSrcLen, lpszSrcData);
231  } else {
232    FXSYS_memcpy(m_pData->m_String + m_pData->m_nDataLength, lpszSrcData,
233                 nSrcLen * sizeof(FX_WCHAR));
234    m_pData->m_nDataLength += nSrcLen;
235    m_pData->m_String[m_pData->m_nDataLength] = 0;
236  }
237}
238void CFX_WideString::ConcatCopy(FX_STRSIZE nSrc1Len,
239                                const FX_WCHAR* lpszSrc1Data,
240                                FX_STRSIZE nSrc2Len,
241                                const FX_WCHAR* lpszSrc2Data) {
242  FX_STRSIZE nNewLen = nSrc1Len + nSrc2Len;
243  if (nNewLen <= 0) {
244    return;
245  }
246  // Don't release until done copying, might be one of the arguments.
247  StringData* pOldData = m_pData;
248  m_pData = StringData::Create(nNewLen);
249  if (m_pData) {
250    wmemcpy(m_pData->m_String, lpszSrc1Data, nSrc1Len);
251    wmemcpy(m_pData->m_String + nSrc1Len, lpszSrc2Data, nSrc2Len);
252  }
253  pOldData->Release();
254}
255void CFX_WideString::CopyBeforeWrite() {
256  if (!m_pData || m_pData->m_nRefs <= 1) {
257    return;
258  }
259  StringData* pData = m_pData;
260  m_pData->Release();
261  FX_STRSIZE nDataLength = pData->m_nDataLength;
262  m_pData = StringData::Create(nDataLength);
263  if (m_pData) {
264    FXSYS_memcpy(m_pData->m_String, pData->m_String,
265                 (nDataLength + 1) * sizeof(FX_WCHAR));
266  }
267}
268void CFX_WideString::AllocBeforeWrite(FX_STRSIZE nLen) {
269  if (m_pData && m_pData->m_nRefs <= 1 && m_pData->m_nAllocLength >= nLen) {
270    return;
271  }
272  Empty();
273  m_pData = StringData::Create(nLen);
274}
275void CFX_WideString::AssignCopy(FX_STRSIZE nSrcLen,
276                                const FX_WCHAR* lpszSrcData) {
277  AllocBeforeWrite(nSrcLen);
278  FXSYS_memcpy(m_pData->m_String, lpszSrcData, nSrcLen * sizeof(FX_WCHAR));
279  m_pData->m_nDataLength = nSrcLen;
280  m_pData->m_String[nSrcLen] = 0;
281}
282int CFX_WideString::Compare(const FX_WCHAR* lpsz) const {
283  if (m_pData)
284    return FXSYS_wcscmp(m_pData->m_String, lpsz);
285  return (!lpsz || lpsz[0] == 0) ? 0 : -1;
286}
287CFX_ByteString CFX_WideString::UTF8Encode() const {
288  return FX_UTF8Encode(*this);
289}
290CFX_ByteString CFX_WideString::UTF16LE_Encode() const {
291  if (!m_pData) {
292    return CFX_ByteString("\0\0", 2);
293  }
294  int len = m_pData->m_nDataLength;
295  CFX_ByteString result;
296  FX_CHAR* buffer = result.GetBuffer(len * 2 + 2);
297  for (int i = 0; i < len; i++) {
298    buffer[i * 2] = m_pData->m_String[i] & 0xff;
299    buffer[i * 2 + 1] = m_pData->m_String[i] >> 8;
300  }
301  buffer[len * 2] = 0;
302  buffer[len * 2 + 1] = 0;
303  result.ReleaseBuffer(len * 2 + 2);
304  return result;
305}
306void CFX_WideString::ConvertFrom(const CFX_ByteString& str,
307                                 CFX_CharMap* pCharMap) {
308  if (!pCharMap) {
309    pCharMap = CFX_CharMap::GetDefaultMapper();
310  }
311  *this = pCharMap->m_GetWideString(pCharMap, str);
312}
313void CFX_WideString::Reserve(FX_STRSIZE len) {
314  GetBuffer(len);
315  ReleaseBuffer(GetLength());
316}
317FX_WCHAR* CFX_WideString::GetBuffer(FX_STRSIZE nMinBufLength) {
318  if (!m_pData && nMinBufLength == 0) {
319    return NULL;
320  }
321  if (m_pData && m_pData->m_nRefs <= 1 &&
322      m_pData->m_nAllocLength >= nMinBufLength) {
323    return m_pData->m_String;
324  }
325  if (!m_pData) {
326    m_pData = StringData::Create(nMinBufLength);
327    if (!m_pData) {
328      return NULL;
329    }
330    m_pData->m_nDataLength = 0;
331    m_pData->m_String[0] = 0;
332    return m_pData->m_String;
333  }
334  StringData* pOldData = m_pData;
335  FX_STRSIZE nOldLen = pOldData->m_nDataLength;
336  if (nMinBufLength < nOldLen) {
337    nMinBufLength = nOldLen;
338  }
339  m_pData = StringData::Create(nMinBufLength);
340  if (!m_pData) {
341    return NULL;
342  }
343  FXSYS_memcpy(m_pData->m_String, pOldData->m_String,
344               (nOldLen + 1) * sizeof(FX_WCHAR));
345  m_pData->m_nDataLength = nOldLen;
346  pOldData->Release();
347  return m_pData->m_String;
348}
349CFX_WideString CFX_WideString::FromLocal(const char* str, FX_STRSIZE len) {
350  CFX_WideString result;
351  result.ConvertFrom(CFX_ByteString(str, len));
352  return result;
353}
354CFX_WideString CFX_WideString::FromUTF8(const char* str, FX_STRSIZE len) {
355  if (!str || 0 == len) {
356    return CFX_WideString();
357  }
358
359  CFX_UTF8Decoder decoder;
360  for (FX_STRSIZE i = 0; i < len; i++) {
361    decoder.Input(str[i]);
362  }
363  return decoder.GetResult();
364}
365CFX_WideString CFX_WideString::FromUTF16LE(const unsigned short* wstr,
366                                           FX_STRSIZE wlen) {
367  if (!wstr || 0 == wlen) {
368    return CFX_WideString();
369  }
370
371  CFX_WideString result;
372  FX_WCHAR* buf = result.GetBuffer(wlen);
373  for (int i = 0; i < wlen; i++) {
374    buf[i] = wstr[i];
375  }
376  result.ReleaseBuffer(wlen);
377  return result;
378}
379FX_STRSIZE CFX_WideString::WStringLength(const unsigned short* str) {
380  FX_STRSIZE len = 0;
381  if (str)
382    while (str[len])
383      len++;
384  return len;
385}
386
387void CFX_WideString::AllocCopy(CFX_WideString& dest,
388                               FX_STRSIZE nCopyLen,
389                               FX_STRSIZE nCopyIndex) const {
390  // |FX_STRSIZE| is currently typedef'd as in |int|. TODO(palmer): It
391  // should be a |size_t|, or at least unsigned.
392  if (nCopyLen == 0 || nCopyLen < 0) {
393    return;
394  }
395  pdfium::base::CheckedNumeric<FX_STRSIZE> iSize =
396      static_cast<FX_STRSIZE>(sizeof(FX_WCHAR));
397  iSize *= nCopyLen;
398  ASSERT(!dest.m_pData);
399  dest.m_pData = StringData::Create(nCopyLen);
400  if (dest.m_pData) {
401    FXSYS_memcpy(dest.m_pData->m_String, m_pData->m_String + nCopyIndex,
402                 iSize.ValueOrDie());
403  }
404}
405CFX_WideString CFX_WideString::Left(FX_STRSIZE nCount) const {
406  if (!m_pData) {
407    return CFX_WideString();
408  }
409  if (nCount < 0) {
410    nCount = 0;
411  }
412  if (nCount >= m_pData->m_nDataLength) {
413    return *this;
414  }
415  CFX_WideString dest;
416  AllocCopy(dest, nCount, 0);
417  return dest;
418}
419CFX_WideString CFX_WideString::Mid(FX_STRSIZE nFirst) const {
420  return Mid(nFirst, m_pData->m_nDataLength - nFirst);
421}
422CFX_WideString CFX_WideString::Mid(FX_STRSIZE nFirst, FX_STRSIZE nCount) const {
423  if (!m_pData) {
424    return CFX_WideString();
425  }
426  if (nFirst < 0) {
427    nFirst = 0;
428  }
429  if (nCount < 0) {
430    nCount = 0;
431  }
432  if (nFirst + nCount > m_pData->m_nDataLength) {
433    nCount = m_pData->m_nDataLength - nFirst;
434  }
435  if (nFirst > m_pData->m_nDataLength) {
436    nCount = 0;
437  }
438  if (nFirst == 0 && nFirst + nCount == m_pData->m_nDataLength) {
439    return *this;
440  }
441  CFX_WideString dest;
442  AllocCopy(dest, nCount, nFirst);
443  return dest;
444}
445CFX_WideString CFX_WideString::Right(FX_STRSIZE nCount) const {
446  if (!m_pData) {
447    return CFX_WideString();
448  }
449  if (nCount < 0) {
450    nCount = 0;
451  }
452  if (nCount >= m_pData->m_nDataLength) {
453    return *this;
454  }
455  CFX_WideString dest;
456  AllocCopy(dest, nCount, m_pData->m_nDataLength - nCount);
457  return dest;
458}
459int CFX_WideString::CompareNoCase(const FX_WCHAR* lpsz) const {
460  if (!m_pData) {
461    return (!lpsz || lpsz[0] == 0) ? 0 : -1;
462  }
463  return FXSYS_wcsicmp(m_pData->m_String, lpsz);
464}
465int CFX_WideString::Compare(const CFX_WideString& str) const {
466  if (!m_pData) {
467    if (!str.m_pData) {
468      return 0;
469    }
470    return -1;
471  }
472  if (!str.m_pData) {
473    return 1;
474  }
475  int this_len = m_pData->m_nDataLength;
476  int that_len = str.m_pData->m_nDataLength;
477  int min_len = this_len < that_len ? this_len : that_len;
478  for (int i = 0; i < min_len; i++) {
479    if (m_pData->m_String[i] < str.m_pData->m_String[i]) {
480      return -1;
481    }
482    if (m_pData->m_String[i] > str.m_pData->m_String[i]) {
483      return 1;
484    }
485  }
486  if (this_len < that_len) {
487    return -1;
488  }
489  if (this_len > that_len) {
490    return 1;
491  }
492  return 0;
493}
494void CFX_WideString::SetAt(FX_STRSIZE nIndex, FX_WCHAR ch) {
495  if (!m_pData) {
496    return;
497  }
498  ASSERT(nIndex >= 0);
499  ASSERT(nIndex < m_pData->m_nDataLength);
500  CopyBeforeWrite();
501  m_pData->m_String[nIndex] = ch;
502}
503void CFX_WideString::MakeLower() {
504  if (!m_pData) {
505    return;
506  }
507  CopyBeforeWrite();
508  if (GetLength() < 1) {
509    return;
510  }
511  FXSYS_wcslwr(m_pData->m_String);
512}
513void CFX_WideString::MakeUpper() {
514  if (!m_pData) {
515    return;
516  }
517  CopyBeforeWrite();
518  if (GetLength() < 1) {
519    return;
520  }
521  FXSYS_wcsupr(m_pData->m_String);
522}
523FX_STRSIZE CFX_WideString::Find(const FX_WCHAR* lpszSub,
524                                FX_STRSIZE nStart) const {
525  FX_STRSIZE nLength = GetLength();
526  if (nLength < 1 || nStart > nLength) {
527    return -1;
528  }
529  const FX_WCHAR* lpsz = FXSYS_wcsstr(m_pData->m_String + nStart, lpszSub);
530  return lpsz ? (int)(lpsz - m_pData->m_String) : -1;
531}
532FX_STRSIZE CFX_WideString::Find(FX_WCHAR ch, FX_STRSIZE nStart) const {
533  if (!m_pData) {
534    return -1;
535  }
536  FX_STRSIZE nLength = m_pData->m_nDataLength;
537  if (nStart >= nLength) {
538    return -1;
539  }
540  const FX_WCHAR* lpsz = FXSYS_wcschr(m_pData->m_String + nStart, ch);
541  return (lpsz) ? (int)(lpsz - m_pData->m_String) : -1;
542}
543void CFX_WideString::TrimRight(const FX_WCHAR* lpszTargetList) {
544  FXSYS_assert(lpszTargetList);
545  if (!m_pData || *lpszTargetList == 0) {
546    return;
547  }
548  CopyBeforeWrite();
549  FX_STRSIZE len = GetLength();
550  if (len < 1) {
551    return;
552  }
553  FX_STRSIZE pos = len;
554  while (pos) {
555    if (!FXSYS_wcschr(lpszTargetList, m_pData->m_String[pos - 1])) {
556      break;
557    }
558    pos--;
559  }
560  if (pos < len) {
561    m_pData->m_String[pos] = 0;
562    m_pData->m_nDataLength = pos;
563  }
564}
565void CFX_WideString::TrimRight(FX_WCHAR chTarget) {
566  FX_WCHAR str[2] = {chTarget, 0};
567  TrimRight(str);
568}
569void CFX_WideString::TrimRight() {
570  TrimRight(L"\x09\x0a\x0b\x0c\x0d\x20");
571}
572void CFX_WideString::TrimLeft(const FX_WCHAR* lpszTargets) {
573  FXSYS_assert(lpszTargets);
574  if (!m_pData || *lpszTargets == 0) {
575    return;
576  }
577  CopyBeforeWrite();
578  if (GetLength() < 1) {
579    return;
580  }
581  const FX_WCHAR* lpsz = m_pData->m_String;
582  while (*lpsz != 0) {
583    if (!FXSYS_wcschr(lpszTargets, *lpsz)) {
584      break;
585    }
586    lpsz++;
587  }
588  if (lpsz != m_pData->m_String) {
589    int nDataLength =
590        m_pData->m_nDataLength - (FX_STRSIZE)(lpsz - m_pData->m_String);
591    FXSYS_memmove(m_pData->m_String, lpsz,
592                  (nDataLength + 1) * sizeof(FX_WCHAR));
593    m_pData->m_nDataLength = nDataLength;
594  }
595}
596void CFX_WideString::TrimLeft(FX_WCHAR chTarget) {
597  FX_WCHAR str[2] = {chTarget, 0};
598  TrimLeft(str);
599}
600void CFX_WideString::TrimLeft() {
601  TrimLeft(L"\x09\x0a\x0b\x0c\x0d\x20");
602}
603FX_STRSIZE CFX_WideString::Replace(const FX_WCHAR* lpszOld,
604                                   const FX_WCHAR* lpszNew) {
605  if (GetLength() < 1) {
606    return 0;
607  }
608  if (!lpszOld) {
609    return 0;
610  }
611  FX_STRSIZE nSourceLen = FXSYS_wcslen(lpszOld);
612  if (nSourceLen == 0) {
613    return 0;
614  }
615  FX_STRSIZE nReplacementLen = lpszNew ? FXSYS_wcslen(lpszNew) : 0;
616  FX_STRSIZE nCount = 0;
617  FX_WCHAR* lpszStart = m_pData->m_String;
618  FX_WCHAR* lpszEnd = m_pData->m_String + m_pData->m_nDataLength;
619  FX_WCHAR* lpszTarget;
620  {
621    while ((lpszTarget = (FX_WCHAR*)FXSYS_wcsstr(lpszStart, lpszOld)) &&
622           lpszStart < lpszEnd) {
623      nCount++;
624      lpszStart = lpszTarget + nSourceLen;
625    }
626  }
627  if (nCount > 0) {
628    CopyBeforeWrite();
629    FX_STRSIZE nOldLength = m_pData->m_nDataLength;
630    FX_STRSIZE nNewLength =
631        nOldLength + (nReplacementLen - nSourceLen) * nCount;
632    if (m_pData->m_nAllocLength < nNewLength || m_pData->m_nRefs > 1) {
633      StringData* pOldData = m_pData;
634      const FX_WCHAR* pstr = m_pData->m_String;
635      m_pData = StringData::Create(nNewLength);
636      if (!m_pData) {
637        return 0;
638      }
639      FXSYS_memcpy(m_pData->m_String, pstr,
640                   pOldData->m_nDataLength * sizeof(FX_WCHAR));
641      pOldData->Release();
642    }
643    lpszStart = m_pData->m_String;
644    lpszEnd = m_pData->m_String + std::max(m_pData->m_nDataLength, nNewLength);
645    {
646      while ((lpszTarget = (FX_WCHAR*)FXSYS_wcsstr(lpszStart, lpszOld)) !=
647                 NULL &&
648             lpszStart < lpszEnd) {
649        FX_STRSIZE nBalance =
650            nOldLength -
651            (FX_STRSIZE)(lpszTarget - m_pData->m_String + nSourceLen);
652        FXSYS_memmove(lpszTarget + nReplacementLen, lpszTarget + nSourceLen,
653                      nBalance * sizeof(FX_WCHAR));
654        FXSYS_memcpy(lpszTarget, lpszNew, nReplacementLen * sizeof(FX_WCHAR));
655        lpszStart = lpszTarget + nReplacementLen;
656        lpszStart[nBalance] = 0;
657        nOldLength += (nReplacementLen - nSourceLen);
658      }
659    }
660    ASSERT(m_pData->m_String[nNewLength] == 0);
661    m_pData->m_nDataLength = nNewLength;
662  }
663  return nCount;
664}
665FX_STRSIZE CFX_WideString::Insert(FX_STRSIZE nIndex, FX_WCHAR ch) {
666  CopyBeforeWrite();
667  if (nIndex < 0) {
668    nIndex = 0;
669  }
670  FX_STRSIZE nNewLength = GetLength();
671  if (nIndex > nNewLength) {
672    nIndex = nNewLength;
673  }
674  nNewLength++;
675  if (!m_pData || m_pData->m_nAllocLength < nNewLength) {
676    StringData* pOldData = m_pData;
677    const FX_WCHAR* pstr = m_pData->m_String;
678    m_pData = StringData::Create(nNewLength);
679    if (!m_pData) {
680      return 0;
681    }
682    if (pOldData) {
683      FXSYS_memmove(m_pData->m_String, pstr,
684                    (pOldData->m_nDataLength + 1) * sizeof(FX_WCHAR));
685      pOldData->Release();
686    } else {
687      m_pData->m_String[0] = 0;
688    }
689  }
690  FXSYS_memmove(m_pData->m_String + nIndex + 1, m_pData->m_String + nIndex,
691                (nNewLength - nIndex) * sizeof(FX_WCHAR));
692  m_pData->m_String[nIndex] = ch;
693  m_pData->m_nDataLength = nNewLength;
694  return nNewLength;
695}
696FX_STRSIZE CFX_WideString::Delete(FX_STRSIZE nIndex, FX_STRSIZE nCount) {
697  if (GetLength() < 1) {
698    return 0;
699  }
700  if (nIndex < 0) {
701    nIndex = 0;
702  }
703  FX_STRSIZE nOldLength = m_pData->m_nDataLength;
704  if (nCount > 0 && nIndex < nOldLength) {
705    CopyBeforeWrite();
706    int nBytesToCopy = nOldLength - (nIndex + nCount) + 1;
707    FXSYS_memmove(m_pData->m_String + nIndex,
708                  m_pData->m_String + nIndex + nCount,
709                  nBytesToCopy * sizeof(FX_WCHAR));
710    m_pData->m_nDataLength = nOldLength - nCount;
711  }
712  return m_pData->m_nDataLength;
713}
714FX_STRSIZE CFX_WideString::Remove(FX_WCHAR chRemove) {
715  if (!m_pData) {
716    return 0;
717  }
718  CopyBeforeWrite();
719  if (GetLength() < 1) {
720    return 0;
721  }
722  FX_WCHAR* pstrSource = m_pData->m_String;
723  FX_WCHAR* pstrDest = m_pData->m_String;
724  FX_WCHAR* pstrEnd = m_pData->m_String + m_pData->m_nDataLength;
725  while (pstrSource < pstrEnd) {
726    if (*pstrSource != chRemove) {
727      *pstrDest = *pstrSource;
728      pstrDest++;
729    }
730    pstrSource++;
731  }
732  *pstrDest = 0;
733  FX_STRSIZE nCount = (FX_STRSIZE)(pstrSource - pstrDest);
734  m_pData->m_nDataLength -= nCount;
735  return nCount;
736}
737#define FORCE_ANSI 0x10000
738#define FORCE_UNICODE 0x20000
739#define FORCE_INT64 0x40000
740void CFX_WideString::FormatV(const FX_WCHAR* lpszFormat, va_list argList) {
741  va_list argListSave;
742#if defined(__ARMCC_VERSION) ||                                              \
743    (!defined(_MSC_VER) && (_FX_CPU_ == _FX_X64_ || _FX_CPU_ == _FX_IA64_ || \
744                            _FX_CPU_ == _FX_ARM64_)) ||                      \
745    defined(__native_client__)
746  va_copy(argListSave, argList);
747#else
748  argListSave = argList;
749#endif
750  int nMaxLen = 0;
751  for (const FX_WCHAR* lpsz = lpszFormat; *lpsz != 0; lpsz++) {
752    if (*lpsz != '%' || *(lpsz = lpsz + 1) == '%') {
753      nMaxLen += FXSYS_wcslen(lpsz);
754      continue;
755    }
756    int nItemLen = 0;
757    int nWidth = 0;
758    for (; *lpsz != 0; lpsz++) {
759      if (*lpsz == '#') {
760        nMaxLen += 2;
761      } else if (*lpsz == '*') {
762        nWidth = va_arg(argList, int);
763      } else if (*lpsz != '-' && *lpsz != '+' && *lpsz != '0' && *lpsz != ' ') {
764        break;
765      }
766    }
767    if (nWidth == 0) {
768      nWidth = FXSYS_wtoi(lpsz);
769      while (std::iswdigit(*lpsz))
770        ++lpsz;
771    }
772    if (nWidth < 0 || nWidth > 128 * 1024) {
773      lpszFormat = L"Bad width";
774      nMaxLen = 10;
775      break;
776    }
777    int nPrecision = 0;
778    if (*lpsz == '.') {
779      lpsz++;
780      if (*lpsz == '*') {
781        nPrecision = va_arg(argList, int);
782        lpsz++;
783      } else {
784        nPrecision = FXSYS_wtoi(lpsz);
785        while (std::iswdigit(*lpsz))
786          ++lpsz;
787      }
788    }
789    if (nPrecision < 0 || nPrecision > 128 * 1024) {
790      lpszFormat = L"Bad precision";
791      nMaxLen = 14;
792      break;
793    }
794    int nModifier = 0;
795    if (*lpsz == L'I' && *(lpsz + 1) == L'6' && *(lpsz + 2) == L'4') {
796      lpsz += 3;
797      nModifier = FORCE_INT64;
798    } else {
799      switch (*lpsz) {
800        case 'h':
801          nModifier = FORCE_ANSI;
802          lpsz++;
803          break;
804        case 'l':
805          nModifier = FORCE_UNICODE;
806          lpsz++;
807          break;
808        case 'F':
809        case 'N':
810        case 'L':
811          lpsz++;
812          break;
813      }
814    }
815    switch (*lpsz | nModifier) {
816      case 'c':
817      case 'C':
818        nItemLen = 2;
819        va_arg(argList, int);
820        break;
821      case 'c' | FORCE_ANSI:
822      case 'C' | FORCE_ANSI:
823        nItemLen = 2;
824        va_arg(argList, int);
825        break;
826      case 'c' | FORCE_UNICODE:
827      case 'C' | FORCE_UNICODE:
828        nItemLen = 2;
829        va_arg(argList, int);
830        break;
831      case 's': {
832        const FX_WCHAR* pstrNextArg = va_arg(argList, const FX_WCHAR*);
833        if (pstrNextArg) {
834          nItemLen = FXSYS_wcslen(pstrNextArg);
835          if (nItemLen < 1) {
836            nItemLen = 1;
837          }
838        } else {
839          nItemLen = 6;
840        }
841      } break;
842      case 'S': {
843        const FX_CHAR* pstrNextArg = va_arg(argList, const FX_CHAR*);
844        if (pstrNextArg) {
845          nItemLen = FXSYS_strlen(pstrNextArg);
846          if (nItemLen < 1) {
847            nItemLen = 1;
848          }
849        } else {
850          nItemLen = 6;
851        }
852      } break;
853      case 's' | FORCE_ANSI:
854      case 'S' | FORCE_ANSI: {
855        const FX_CHAR* pstrNextArg = va_arg(argList, const FX_CHAR*);
856        if (pstrNextArg) {
857          nItemLen = FXSYS_strlen(pstrNextArg);
858          if (nItemLen < 1) {
859            nItemLen = 1;
860          }
861        } else {
862          nItemLen = 6;
863        }
864      } break;
865      case 's' | FORCE_UNICODE:
866      case 'S' | FORCE_UNICODE: {
867        FX_WCHAR* pstrNextArg = va_arg(argList, FX_WCHAR*);
868        if (pstrNextArg) {
869          nItemLen = FXSYS_wcslen(pstrNextArg);
870          if (nItemLen < 1) {
871            nItemLen = 1;
872          }
873        } else {
874          nItemLen = 6;
875        }
876      } break;
877    }
878    if (nItemLen != 0) {
879      if (nPrecision != 0 && nItemLen > nPrecision) {
880        nItemLen = nPrecision;
881      }
882      if (nItemLen < nWidth) {
883        nItemLen = nWidth;
884      }
885    } else {
886      switch (*lpsz) {
887        case 'd':
888        case 'i':
889        case 'u':
890        case 'x':
891        case 'X':
892        case 'o':
893          if (nModifier & FORCE_INT64) {
894            va_arg(argList, int64_t);
895          } else {
896            va_arg(argList, int);
897          }
898          nItemLen = 32;
899          if (nItemLen < nWidth + nPrecision) {
900            nItemLen = nWidth + nPrecision;
901          }
902          break;
903        case 'a':
904        case 'A':
905        case 'e':
906        case 'E':
907        case 'g':
908        case 'G':
909          va_arg(argList, double);
910          nItemLen = 128;
911          if (nItemLen < nWidth + nPrecision) {
912            nItemLen = nWidth + nPrecision;
913          }
914          break;
915        case 'f':
916          if (nWidth + nPrecision > 100) {
917            nItemLen = nPrecision + nWidth + 128;
918          } else {
919            double f;
920            char pszTemp[256];
921            f = va_arg(argList, double);
922            FXSYS_snprintf(pszTemp, sizeof(pszTemp), "%*.*f", nWidth,
923                           nPrecision + 6, f);
924            nItemLen = FXSYS_strlen(pszTemp);
925          }
926          break;
927        case 'p':
928          va_arg(argList, void*);
929          nItemLen = 32;
930          if (nItemLen < nWidth + nPrecision) {
931            nItemLen = nWidth + nPrecision;
932          }
933          break;
934        case 'n':
935          va_arg(argList, int*);
936          break;
937      }
938    }
939    nMaxLen += nItemLen;
940  }
941  GetBuffer(nMaxLen);
942  if (m_pData) {
943    FXSYS_vswprintf((wchar_t*)m_pData->m_String, nMaxLen + 1,
944                    (const wchar_t*)lpszFormat, argListSave);
945    ReleaseBuffer();
946  }
947  va_end(argListSave);
948}
949void CFX_WideString::Format(const FX_WCHAR* lpszFormat, ...) {
950  va_list argList;
951  va_start(argList, lpszFormat);
952  FormatV(lpszFormat, argList);
953  va_end(argList);
954}
955FX_FLOAT FX_wtof(const FX_WCHAR* str, int len) {
956  if (len == 0) {
957    return 0.0;
958  }
959  int cc = 0;
960  FX_BOOL bNegative = FALSE;
961  if (str[0] == '+') {
962    cc++;
963  } else if (str[0] == '-') {
964    bNegative = TRUE;
965    cc++;
966  }
967  int integer = 0;
968  while (cc < len) {
969    if (str[cc] == '.') {
970      break;
971    }
972    integer = integer * 10 + FXSYS_toDecimalDigitWide(str[cc]);
973    cc++;
974  }
975  FX_FLOAT fraction = 0;
976  if (str[cc] == '.') {
977    cc++;
978    FX_FLOAT scale = 0.1f;
979    while (cc < len) {
980      fraction += scale * FXSYS_toDecimalDigitWide(str[cc]);
981      scale *= 0.1f;
982      cc++;
983    }
984  }
985  fraction += (FX_FLOAT)integer;
986  return bNegative ? -fraction : fraction;
987}
988int CFX_WideString::GetInteger() const {
989  return m_pData ? FXSYS_wtoi(m_pData->m_String) : 0;
990}
991FX_FLOAT CFX_WideString::GetFloat() const {
992  return m_pData ? FX_wtof(m_pData->m_String, m_pData->m_nDataLength) : 0.0f;
993}
994static CFX_ByteString _DefMap_GetByteString(CFX_CharMap* pCharMap,
995                                            const CFX_WideString& widestr) {
996  int src_len = widestr.GetLength();
997  int codepage = pCharMap->m_GetCodePage ? pCharMap->m_GetCodePage() : 0;
998  int dest_len = FXSYS_WideCharToMultiByte(codepage, 0, widestr.c_str(),
999                                           src_len, NULL, 0, NULL, NULL);
1000  if (dest_len == 0) {
1001    return CFX_ByteString();
1002  }
1003  CFX_ByteString bytestr;
1004  FX_CHAR* dest_buf = bytestr.GetBuffer(dest_len);
1005  FXSYS_WideCharToMultiByte(codepage, 0, widestr.c_str(), src_len, dest_buf,
1006                            dest_len, NULL, NULL);
1007  bytestr.ReleaseBuffer(dest_len);
1008  return bytestr;
1009}
1010static CFX_WideString _DefMap_GetWideString(CFX_CharMap* pCharMap,
1011                                            const CFX_ByteString& bytestr) {
1012  int src_len = bytestr.GetLength();
1013  int codepage = pCharMap->m_GetCodePage ? pCharMap->m_GetCodePage() : 0;
1014  int dest_len =
1015      FXSYS_MultiByteToWideChar(codepage, 0, bytestr, src_len, NULL, 0);
1016  if (dest_len == 0) {
1017    return CFX_WideString();
1018  }
1019  CFX_WideString widestr;
1020  FX_WCHAR* dest_buf = widestr.GetBuffer(dest_len);
1021  FXSYS_MultiByteToWideChar(codepage, 0, bytestr, src_len, dest_buf, dest_len);
1022  widestr.ReleaseBuffer(dest_len);
1023  return widestr;
1024}
1025static int _DefMap_GetGBKCodePage() {
1026  return 936;
1027}
1028static int _DefMap_GetUHCCodePage() {
1029  return 949;
1030}
1031static int _DefMap_GetJISCodePage() {
1032  return 932;
1033}
1034static int _DefMap_GetBig5CodePage() {
1035  return 950;
1036}
1037static const CFX_CharMap g_DefaultMapper = {&_DefMap_GetWideString,
1038                                            &_DefMap_GetByteString, NULL};
1039static const CFX_CharMap g_DefaultGBKMapper = {
1040    &_DefMap_GetWideString, &_DefMap_GetByteString, &_DefMap_GetGBKCodePage};
1041static const CFX_CharMap g_DefaultJISMapper = {
1042    &_DefMap_GetWideString, &_DefMap_GetByteString, &_DefMap_GetJISCodePage};
1043static const CFX_CharMap g_DefaultUHCMapper = {
1044    &_DefMap_GetWideString, &_DefMap_GetByteString, &_DefMap_GetUHCCodePage};
1045static const CFX_CharMap g_DefaultBig5Mapper = {
1046    &_DefMap_GetWideString, &_DefMap_GetByteString, &_DefMap_GetBig5CodePage};
1047CFX_CharMap* CFX_CharMap::GetDefaultMapper(int32_t codepage) {
1048  switch (codepage) {
1049    case 0:
1050      return (CFX_CharMap*)&g_DefaultMapper;
1051    case 932:
1052      return (CFX_CharMap*)&g_DefaultJISMapper;
1053    case 936:
1054      return (CFX_CharMap*)&g_DefaultGBKMapper;
1055    case 949:
1056      return (CFX_CharMap*)&g_DefaultUHCMapper;
1057    case 950:
1058      return (CFX_CharMap*)&g_DefaultBig5Mapper;
1059  }
1060  return NULL;
1061}
1062