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