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