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