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 #ifndef CORE_INCLUDE_FXCRT_FX_STRING_H_
8 #define CORE_INCLUDE_FXCRT_FX_STRING_H_
9 
10 #include <stdint.h>  // For intptr_t.
11 #include <algorithm>
12 
13 #include "fx_memory.h"
14 #include "fx_system.h"
15 
16 class CFX_BinaryBuf;
17 class CFX_ByteString;
18 class CFX_WideString;
19 struct CFX_CharMap;
20 
21 // An immutable string with caller-provided storage which must outlive the
22 // string itself.
23 class CFX_ByteStringC {
24  public:
25   typedef FX_CHAR value_type;
26 
CFX_ByteStringC()27   CFX_ByteStringC() {
28     m_Ptr = NULL;
29     m_Length = 0;
30   }
31 
CFX_ByteStringC(const uint8_t * ptr,FX_STRSIZE size)32   CFX_ByteStringC(const uint8_t* ptr, FX_STRSIZE size) {
33     m_Ptr = ptr;
34     m_Length = size;
35   }
36 
CFX_ByteStringC(const FX_CHAR * ptr)37   CFX_ByteStringC(const FX_CHAR* ptr) {
38     m_Ptr = (const uint8_t*)ptr;
39     m_Length = ptr ? FXSYS_strlen(ptr) : 0;
40   }
41 
42   // |ch| must be an lvalue that outlives the the CFX_ByteStringC. However,
43   // the use of char rvalues are not caught at compile time.  They are
44   // implicitly promoted to CFX_ByteString (see below) and then the
45   // CFX_ByteStringC is constructed from the CFX_ByteString via the alternate
46   // constructor below. The CFX_ByteString then typically goes out of scope
47   // and |m_Ptr| may be left pointing to invalid memory. Beware.
48   // TODO(tsepez): Mark single-argument string constructors as explicit.
CFX_ByteStringC(FX_CHAR & ch)49   CFX_ByteStringC(FX_CHAR& ch) {
50     m_Ptr = (const uint8_t*)&ch;
51     m_Length = 1;
52   }
53 
CFX_ByteStringC(const FX_CHAR * ptr,FX_STRSIZE len)54   CFX_ByteStringC(const FX_CHAR* ptr, FX_STRSIZE len) {
55     m_Ptr = (const uint8_t*)ptr;
56     m_Length = (len == -1) ? FXSYS_strlen(ptr) : len;
57   }
58 
CFX_ByteStringC(const CFX_ByteStringC & src)59   CFX_ByteStringC(const CFX_ByteStringC& src) {
60     m_Ptr = src.m_Ptr;
61     m_Length = src.m_Length;
62   }
63 
64   CFX_ByteStringC(const CFX_ByteString& src);
65 
66   CFX_ByteStringC& operator=(const FX_CHAR* src) {
67     m_Ptr = (const uint8_t*)src;
68     m_Length = m_Ptr ? FXSYS_strlen(src) : 0;
69     return *this;
70   }
71 
72   CFX_ByteStringC& operator=(const CFX_ByteStringC& src) {
73     m_Ptr = src.m_Ptr;
74     m_Length = src.m_Length;
75     return *this;
76   }
77 
78   CFX_ByteStringC& operator=(const CFX_ByteString& src);
79 
80   bool operator==(const char* ptr) const {
81     return FXSYS_strlen(ptr) == m_Length &&
82            FXSYS_memcmp(ptr, m_Ptr, m_Length) == 0;
83   }
84   bool operator==(const CFX_ByteStringC& other) const {
85     return other.m_Length == m_Length &&
86            FXSYS_memcmp(other.m_Ptr, m_Ptr, m_Length) == 0;
87   }
88   bool operator!=(const char* ptr) const { return !(*this == ptr); }
89   bool operator!=(const CFX_ByteStringC& other) const {
90     return !(*this == other);
91   }
92 
93   FX_DWORD GetID(FX_STRSIZE start_pos = 0) const;
94 
GetPtr()95   const uint8_t* GetPtr() const { return m_Ptr; }
96 
GetCStr()97   const FX_CHAR* GetCStr() const { return (const FX_CHAR*)m_Ptr; }
98 
GetLength()99   FX_STRSIZE GetLength() const { return m_Length; }
100 
IsEmpty()101   bool IsEmpty() const { return m_Length == 0; }
102 
GetAt(FX_STRSIZE index)103   uint8_t GetAt(FX_STRSIZE index) const { return m_Ptr[index]; }
104 
105   CFX_ByteStringC Mid(FX_STRSIZE index, FX_STRSIZE count = -1) const {
106     if (index < 0) {
107       index = 0;
108     }
109     if (index > m_Length) {
110       return CFX_ByteStringC();
111     }
112     if (count < 0 || count > m_Length - index) {
113       count = m_Length - index;
114     }
115     return CFX_ByteStringC(m_Ptr + index, count);
116   }
117 
118   const uint8_t& operator[](size_t index) const { return m_Ptr[index]; }
119 
120   bool operator<(const CFX_ByteStringC& that) const {
121     int result = memcmp(m_Ptr, that.m_Ptr, std::min(m_Length, that.m_Length));
122     return result < 0 || (result == 0 && m_Length < that.m_Length);
123   }
124 
125  protected:
126   const uint8_t* m_Ptr;
127   FX_STRSIZE m_Length;
128 
129  private:
new(size_t)130   void* operator new(size_t) throw() { return NULL; }
131 };
132 inline bool operator==(const char* lhs, const CFX_ByteStringC& rhs) {
133   return rhs == lhs;
134 }
135 inline bool operator!=(const char* lhs, const CFX_ByteStringC& rhs) {
136   return rhs != lhs;
137 }
138 #define FXBSTR_ID(c1, c2, c3, c4) ((c1 << 24) | (c2 << 16) | (c3 << 8) | (c4))
139 
140 // A mutable string with shared buffers using copy-on-write semantics that
141 // avoids the cost of std::string's iterator stability guarantees.
142 class CFX_ByteString {
143  public:
144   typedef FX_CHAR value_type;
145 
CFX_ByteString()146   CFX_ByteString() : m_pData(nullptr) {}
147 
148   // Copy constructor.
149   CFX_ByteString(const CFX_ByteString& str);
150 
151   // Move constructor.
CFX_ByteString(CFX_ByteString && other)152   inline CFX_ByteString(CFX_ByteString&& other) {
153     m_pData = other.m_pData;
154     other.m_pData = nullptr;
155   }
156 
157   CFX_ByteString(char ch);
CFX_ByteString(const FX_CHAR * ptr)158   CFX_ByteString(const FX_CHAR* ptr)
159       : CFX_ByteString(ptr, ptr ? FXSYS_strlen(ptr) : 0) {}
160 
161   CFX_ByteString(const FX_CHAR* ptr, FX_STRSIZE len);
162   CFX_ByteString(const uint8_t* ptr, FX_STRSIZE len);
163 
164   CFX_ByteString(const CFX_ByteStringC& bstrc);
165   CFX_ByteString(const CFX_ByteStringC& bstrc1, const CFX_ByteStringC& bstrc2);
166 
167   ~CFX_ByteString();
168 
169   static CFX_ByteString FromUnicode(const FX_WCHAR* ptr, FX_STRSIZE len = -1);
170 
171   static CFX_ByteString FromUnicode(const CFX_WideString& str);
172 
173   // Explicit conversion to raw string
c_str()174   const FX_CHAR* c_str() const { return m_pData ? m_pData->m_String : ""; }
175 
176   // Implicit conversion to C-style string -- deprecated
177   operator const FX_CHAR*() const { return m_pData ? m_pData->m_String : ""; }
178 
179   operator const uint8_t*() const {
180     return m_pData ? (const uint8_t*)m_pData->m_String : NULL;
181   }
182 
GetLength()183   FX_STRSIZE GetLength() const { return m_pData ? m_pData->m_nDataLength : 0; }
184 
IsEmpty()185   bool IsEmpty() const { return !GetLength(); }
186 
187   int Compare(const CFX_ByteStringC& str) const;
188 
189   bool Equal(const char* ptr) const;
190   bool Equal(const CFX_ByteStringC& str) const;
191   bool Equal(const CFX_ByteString& other) const;
192 
193   bool EqualNoCase(const CFX_ByteStringC& str) const;
194 
195   bool operator==(const char* ptr) const { return Equal(ptr); }
196   bool operator==(const CFX_ByteStringC& str) const { return Equal(str); }
197   bool operator==(const CFX_ByteString& other) const { return Equal(other); }
198 
199   bool operator!=(const char* ptr) const { return !(*this == ptr); }
200   bool operator!=(const CFX_ByteStringC& str) const { return !(*this == str); }
201   bool operator!=(const CFX_ByteString& other) const {
202     return !(*this == other);
203   }
204 
205   bool operator<(const CFX_ByteString& str) const {
206     int result = FXSYS_memcmp(c_str(), str.c_str(),
207                               std::min(GetLength(), str.GetLength()));
208     return result < 0 || (result == 0 && GetLength() < str.GetLength());
209   }
210 
211   void Empty();
212 
213   const CFX_ByteString& operator=(const FX_CHAR* str);
214 
215   const CFX_ByteString& operator=(const CFX_ByteStringC& bstrc);
216 
217   const CFX_ByteString& operator=(const CFX_ByteString& stringSrc);
218 
219   const CFX_ByteString& operator=(const CFX_BinaryBuf& buf);
220 
221   void Load(const uint8_t* str, FX_STRSIZE len);
222 
223   const CFX_ByteString& operator+=(FX_CHAR ch);
224 
225   const CFX_ByteString& operator+=(const FX_CHAR* str);
226 
227   const CFX_ByteString& operator+=(const CFX_ByteString& str);
228 
229   const CFX_ByteString& operator+=(const CFX_ByteStringC& bstrc);
230 
GetAt(FX_STRSIZE nIndex)231   uint8_t GetAt(FX_STRSIZE nIndex) const {
232     return m_pData ? m_pData->m_String[nIndex] : 0;
233   }
234 
235   uint8_t operator[](FX_STRSIZE nIndex) const {
236     return m_pData ? m_pData->m_String[nIndex] : 0;
237   }
238 
239   void SetAt(FX_STRSIZE nIndex, FX_CHAR ch);
240 
241   FX_STRSIZE Insert(FX_STRSIZE index, FX_CHAR ch);
242 
243   FX_STRSIZE Delete(FX_STRSIZE index, FX_STRSIZE count = 1);
244 
245   void Format(const FX_CHAR* lpszFormat, ...);
246 
247   void FormatV(const FX_CHAR* lpszFormat, va_list argList);
248 
249   void Reserve(FX_STRSIZE len);
250 
251   FX_CHAR* GetBuffer(FX_STRSIZE len);
252 
253   void ReleaseBuffer(FX_STRSIZE len = -1);
254 
255   CFX_ByteString Mid(FX_STRSIZE first) const;
256 
257   CFX_ByteString Mid(FX_STRSIZE first, FX_STRSIZE count) const;
258 
259   CFX_ByteString Left(FX_STRSIZE count) const;
260 
261   CFX_ByteString Right(FX_STRSIZE count) const;
262 
263   FX_STRSIZE Find(const CFX_ByteStringC& lpszSub, FX_STRSIZE start = 0) const;
264 
265   FX_STRSIZE Find(FX_CHAR ch, FX_STRSIZE start = 0) const;
266 
267   FX_STRSIZE ReverseFind(FX_CHAR ch) const;
268 
269   void MakeLower();
270 
271   void MakeUpper();
272 
273   void TrimRight();
274 
275   void TrimRight(FX_CHAR chTarget);
276 
277   void TrimRight(const CFX_ByteStringC& lpszTargets);
278 
279   void TrimLeft();
280 
281   void TrimLeft(FX_CHAR chTarget);
282 
283   void TrimLeft(const CFX_ByteStringC& lpszTargets);
284 
285   FX_STRSIZE Replace(const CFX_ByteStringC& lpszOld,
286                      const CFX_ByteStringC& lpszNew);
287 
288   FX_STRSIZE Remove(FX_CHAR ch);
289 
290   CFX_WideString UTF8Decode() const;
291 
292   void ConvertFrom(const CFX_WideString& str, CFX_CharMap* pCharMap = NULL);
293 
294   FX_DWORD GetID(FX_STRSIZE start_pos = 0) const;
295 
296 #define FXFORMAT_SIGNED 1
297 #define FXFORMAT_HEX 2
298 #define FXFORMAT_CAPITAL 4
299 
300   static CFX_ByteString FormatInteger(int i, FX_DWORD flags = 0);
301   static CFX_ByteString FormatFloat(FX_FLOAT f, int precision = 0);
302 
303  protected:
304   // To ensure ref counts do not overflow, consider the worst possible case:
305   // the entire address space contains nothing but pointers to this object.
306   // Since the count increments with each new pointer, the largest value is
307   // the number of pointers that can fit into the address space. The size of
308   // the address space itself is a good upper bound on it; we need not go
309   // larger.
310   class StringData {
311    public:
312     static StringData* Create(int nLen);
Retain()313     void Retain() { ++m_nRefs; }
Release()314     void Release() {
315       if (--m_nRefs <= 0)
316         FX_Free(this);
317     }
318 
319     intptr_t m_nRefs;  // Would prefer ssize_t, but no windows support.
320     FX_STRSIZE m_nDataLength;
321     FX_STRSIZE m_nAllocLength;
322     FX_CHAR m_String[1];
323 
324    private:
StringData(FX_STRSIZE dataLen,FX_STRSIZE allocLen)325     StringData(FX_STRSIZE dataLen, FX_STRSIZE allocLen)
326         : m_nRefs(1), m_nDataLength(dataLen), m_nAllocLength(allocLen) {
327       FXSYS_assert(dataLen >= 0);
328       FXSYS_assert(allocLen >= 0);
329       FXSYS_assert(dataLen <= allocLen);
330       m_String[dataLen] = 0;
331     }
332     ~StringData() = delete;
333   };
334 
335   void AllocCopy(CFX_ByteString& dest,
336                  FX_STRSIZE nCopyLen,
337                  FX_STRSIZE nCopyIndex) const;
338   void AssignCopy(FX_STRSIZE nSrcLen, const FX_CHAR* lpszSrcData);
339   void ConcatCopy(FX_STRSIZE nSrc1Len,
340                   const FX_CHAR* lpszSrc1Data,
341                   FX_STRSIZE nSrc2Len,
342                   const FX_CHAR* lpszSrc2Data);
343   void ConcatInPlace(FX_STRSIZE nSrcLen, const FX_CHAR* lpszSrcData);
344   void CopyBeforeWrite();
345   void AllocBeforeWrite(FX_STRSIZE nLen);
346 
347   StringData* m_pData;
348   friend class fxcrt_ByteStringConcatInPlace_Test;
349 };
CFX_ByteStringC(const CFX_ByteString & src)350 inline CFX_ByteStringC::CFX_ByteStringC(const CFX_ByteString& src) {
351   m_Ptr = (const uint8_t*)src;
352   m_Length = src.GetLength();
353 }
354 inline CFX_ByteStringC& CFX_ByteStringC::operator=(const CFX_ByteString& src) {
355   m_Ptr = (const uint8_t*)src;
356   m_Length = src.GetLength();
357   return *this;
358 }
359 
360 inline bool operator==(const char* lhs, const CFX_ByteString& rhs) {
361   return rhs == lhs;
362 }
363 inline bool operator==(const CFX_ByteStringC& lhs, const CFX_ByteString& rhs) {
364   return rhs == lhs;
365 }
366 inline bool operator!=(const char* lhs, const CFX_ByteString& rhs) {
367   return rhs != lhs;
368 }
369 inline bool operator!=(const CFX_ByteStringC& lhs, const CFX_ByteString& rhs) {
370   return rhs != lhs;
371 }
372 
373 inline CFX_ByteString operator+(const CFX_ByteStringC& str1,
374                                 const CFX_ByteStringC& str2) {
375   return CFX_ByteString(str1, str2);
376 }
377 inline CFX_ByteString operator+(const CFX_ByteStringC& str1,
378                                 const FX_CHAR* str2) {
379   return CFX_ByteString(str1, str2);
380 }
381 inline CFX_ByteString operator+(const FX_CHAR* str1,
382                                 const CFX_ByteStringC& str2) {
383   return CFX_ByteString(str1, str2);
384 }
385 inline CFX_ByteString operator+(const CFX_ByteStringC& str1, FX_CHAR ch) {
386   return CFX_ByteString(str1, CFX_ByteStringC(ch));
387 }
388 inline CFX_ByteString operator+(FX_CHAR ch, const CFX_ByteStringC& str2) {
389   return CFX_ByteString(ch, str2);
390 }
391 inline CFX_ByteString operator+(const CFX_ByteString& str1,
392                                 const CFX_ByteString& str2) {
393   return CFX_ByteString(str1, str2);
394 }
395 inline CFX_ByteString operator+(const CFX_ByteString& str1, FX_CHAR ch) {
396   return CFX_ByteString(str1, CFX_ByteStringC(ch));
397 }
398 inline CFX_ByteString operator+(FX_CHAR ch, const CFX_ByteString& str2) {
399   return CFX_ByteString(ch, str2);
400 }
401 inline CFX_ByteString operator+(const CFX_ByteString& str1,
402                                 const FX_CHAR* str2) {
403   return CFX_ByteString(str1, str2);
404 }
405 inline CFX_ByteString operator+(const FX_CHAR* str1,
406                                 const CFX_ByteString& str2) {
407   return CFX_ByteString(str1, str2);
408 }
409 inline CFX_ByteString operator+(const CFX_ByteString& str1,
410                                 const CFX_ByteStringC& str2) {
411   return CFX_ByteString(str1, str2);
412 }
413 inline CFX_ByteString operator+(const CFX_ByteStringC& str1,
414                                 const CFX_ByteString& str2) {
415   return CFX_ByteString(str1, str2);
416 }
417 class CFX_WideStringC {
418  public:
419   typedef FX_WCHAR value_type;
420 
CFX_WideStringC()421   CFX_WideStringC() {
422     m_Ptr = NULL;
423     m_Length = 0;
424   }
425 
CFX_WideStringC(const FX_WCHAR * ptr)426   CFX_WideStringC(const FX_WCHAR* ptr) {
427     m_Ptr = ptr;
428     m_Length = ptr ? FXSYS_wcslen(ptr) : 0;
429   }
430 
CFX_WideStringC(FX_WCHAR & ch)431   CFX_WideStringC(FX_WCHAR& ch) {
432     m_Ptr = &ch;
433     m_Length = 1;
434   }
435 
CFX_WideStringC(const FX_WCHAR * ptr,FX_STRSIZE len)436   CFX_WideStringC(const FX_WCHAR* ptr, FX_STRSIZE len) {
437     m_Ptr = ptr;
438     m_Length = (len == -1) ? FXSYS_wcslen(ptr) : len;
439   }
440 
CFX_WideStringC(const CFX_WideStringC & src)441   CFX_WideStringC(const CFX_WideStringC& src) {
442     m_Ptr = src.m_Ptr;
443     m_Length = src.m_Length;
444   }
445 
446   CFX_WideStringC(const CFX_WideString& src);
447 
448   CFX_WideStringC& operator=(const FX_WCHAR* src) {
449     m_Ptr = src;
450     m_Length = FXSYS_wcslen(src);
451     return *this;
452   }
453 
454   CFX_WideStringC& operator=(const CFX_WideStringC& src) {
455     m_Ptr = src.m_Ptr;
456     m_Length = src.m_Length;
457     return *this;
458   }
459 
460   CFX_WideStringC& operator=(const CFX_WideString& src);
461 
462   bool operator==(const wchar_t* ptr) const {
463     return FXSYS_wcslen(ptr) == m_Length && wmemcmp(ptr, m_Ptr, m_Length) == 0;
464   }
465   bool operator==(const CFX_WideStringC& str) const {
466     return str.m_Length == m_Length && wmemcmp(str.m_Ptr, m_Ptr, m_Length) == 0;
467   }
468   bool operator!=(const wchar_t* ptr) const { return !(*this == ptr); }
469   bool operator!=(const CFX_WideStringC& str) const { return !(*this == str); }
470 
GetPtr()471   const FX_WCHAR* GetPtr() const { return m_Ptr; }
472 
GetLength()473   FX_STRSIZE GetLength() const { return m_Length; }
474 
IsEmpty()475   bool IsEmpty() const { return m_Length == 0; }
476 
GetAt(FX_STRSIZE index)477   FX_WCHAR GetAt(FX_STRSIZE index) const { return m_Ptr[index]; }
478 
Left(FX_STRSIZE count)479   CFX_WideStringC Left(FX_STRSIZE count) const {
480     if (count < 1) {
481       return CFX_WideStringC();
482     }
483     if (count > m_Length) {
484       count = m_Length;
485     }
486     return CFX_WideStringC(m_Ptr, count);
487   }
488 
489   CFX_WideStringC Mid(FX_STRSIZE index, FX_STRSIZE count = -1) const {
490     if (index < 0) {
491       index = 0;
492     }
493     if (index > m_Length) {
494       return CFX_WideStringC();
495     }
496     if (count < 0 || count > m_Length - index) {
497       count = m_Length - index;
498     }
499     return CFX_WideStringC(m_Ptr + index, count);
500   }
501 
Right(FX_STRSIZE count)502   CFX_WideStringC Right(FX_STRSIZE count) const {
503     if (count < 1) {
504       return CFX_WideStringC();
505     }
506     if (count > m_Length) {
507       count = m_Length;
508     }
509     return CFX_WideStringC(m_Ptr + m_Length - count, count);
510   }
511 
512   const FX_WCHAR& operator[](size_t index) const { return m_Ptr[index]; }
513 
514   bool operator<(const CFX_WideStringC& that) const {
515     int result = wmemcmp(m_Ptr, that.m_Ptr, std::min(m_Length, that.m_Length));
516     return result < 0 || (result == 0 && m_Length < that.m_Length);
517   }
518 
519  protected:
520   const FX_WCHAR* m_Ptr;
521   FX_STRSIZE m_Length;
522 
523  private:
new(size_t)524   void* operator new(size_t) throw() { return NULL; }
525 };
526 inline bool operator==(const wchar_t* lhs, const CFX_WideStringC& rhs) {
527   return rhs == lhs;
528 }
529 inline bool operator!=(const wchar_t* lhs, const CFX_WideStringC& rhs) {
530   return rhs != lhs;
531 }
532 #define FX_WSTRC(wstr) CFX_WideStringC(wstr, FX_ArraySize(wstr) - 1)
533 
534 // A mutable string with shared buffers using copy-on-write semantics that
535 // avoids the cost of std::string's iterator stability guarantees.
536 class CFX_WideString {
537  public:
538   typedef FX_WCHAR value_type;
539 
CFX_WideString()540   CFX_WideString() : m_pData(nullptr) {}
541 
542   // Copy constructor.
543   CFX_WideString(const CFX_WideString& str);
544 
545   // Move constructor.
CFX_WideString(CFX_WideString && other)546   inline CFX_WideString(CFX_WideString&& other) {
547     m_pData = other.m_pData;
548     other.m_pData = nullptr;
549   }
550 
CFX_WideString(const FX_WCHAR * ptr)551   CFX_WideString(const FX_WCHAR* ptr)
552       : CFX_WideString(ptr, ptr ? FXSYS_wcslen(ptr) : 0) {}
553 
554   CFX_WideString(const FX_WCHAR* ptr, FX_STRSIZE len);
555 
556   CFX_WideString(FX_WCHAR ch);
557 
558   CFX_WideString(const CFX_WideStringC& str);
559 
560   CFX_WideString(const CFX_WideStringC& str1, const CFX_WideStringC& str2);
561 
562   ~CFX_WideString();
563 
564   static CFX_WideString FromLocal(const char* str, FX_STRSIZE len = -1);
565 
566   static CFX_WideString FromUTF8(const char* str, FX_STRSIZE len);
567 
568   static CFX_WideString FromUTF16LE(const unsigned short* str, FX_STRSIZE len);
569 
570   static FX_STRSIZE WStringLength(const unsigned short* str);
571 
572   // Explicit conversion to raw string
c_str()573   const FX_WCHAR* c_str() const { return m_pData ? m_pData->m_String : L""; }
574 
575   // Implicit conversion to C-style wide string -- deprecated
576   operator const FX_WCHAR*() const { return m_pData ? m_pData->m_String : L""; }
577 
578   void Empty();
579 
IsEmpty()580   FX_BOOL IsEmpty() const { return !GetLength(); }
581 
GetLength()582   FX_STRSIZE GetLength() const { return m_pData ? m_pData->m_nDataLength : 0; }
583 
584   const CFX_WideString& operator=(const FX_WCHAR* str);
585 
586   const CFX_WideString& operator=(const CFX_WideString& stringSrc);
587 
588   const CFX_WideString& operator=(const CFX_WideStringC& stringSrc);
589 
590   const CFX_WideString& operator+=(const FX_WCHAR* str);
591 
592   const CFX_WideString& operator+=(FX_WCHAR ch);
593 
594   const CFX_WideString& operator+=(const CFX_WideString& str);
595 
596   const CFX_WideString& operator+=(const CFX_WideStringC& str);
597 
598   bool operator==(const wchar_t* ptr) const { return Equal(ptr); }
599   bool operator==(const CFX_WideStringC& str) const { return Equal(str); }
600   bool operator==(const CFX_WideString& other) const { return Equal(other); }
601 
602   bool operator!=(const wchar_t* ptr) const { return !(*this == ptr); }
603   bool operator!=(const CFX_WideStringC& str) const { return !(*this == str); }
604   bool operator!=(const CFX_WideString& other) const {
605     return !(*this == other);
606   }
607 
608   bool operator<(const CFX_WideString& str) const {
609     int result =
610         wmemcmp(c_str(), str.c_str(), std::min(GetLength(), str.GetLength()));
611     return result < 0 || (result == 0 && GetLength() < str.GetLength());
612   }
613 
GetAt(FX_STRSIZE nIndex)614   FX_WCHAR GetAt(FX_STRSIZE nIndex) const {
615     return m_pData ? m_pData->m_String[nIndex] : 0;
616   }
617 
618   FX_WCHAR operator[](FX_STRSIZE nIndex) const {
619     return m_pData ? m_pData->m_String[nIndex] : 0;
620   }
621 
622   void SetAt(FX_STRSIZE nIndex, FX_WCHAR ch);
623 
624   int Compare(const FX_WCHAR* str) const;
625 
626   int Compare(const CFX_WideString& str) const;
627 
628   int CompareNoCase(const FX_WCHAR* str) const;
629 
630   bool Equal(const wchar_t* ptr) const;
631   bool Equal(const CFX_WideStringC& str) const;
632   bool Equal(const CFX_WideString& other) const;
633 
634   CFX_WideString Mid(FX_STRSIZE first) const;
635 
636   CFX_WideString Mid(FX_STRSIZE first, FX_STRSIZE count) const;
637 
638   CFX_WideString Left(FX_STRSIZE count) const;
639 
640   CFX_WideString Right(FX_STRSIZE count) const;
641 
642   FX_STRSIZE Insert(FX_STRSIZE index, FX_WCHAR ch);
643 
644   FX_STRSIZE Delete(FX_STRSIZE index, FX_STRSIZE count = 1);
645 
646   void Format(const FX_WCHAR* lpszFormat, ...);
647 
648   void FormatV(const FX_WCHAR* lpszFormat, va_list argList);
649 
650   void MakeLower();
651 
652   void MakeUpper();
653 
654   void TrimRight();
655 
656   void TrimRight(FX_WCHAR chTarget);
657 
658   void TrimRight(const FX_WCHAR* lpszTargets);
659 
660   void TrimLeft();
661 
662   void TrimLeft(FX_WCHAR chTarget);
663 
664   void TrimLeft(const FX_WCHAR* lpszTargets);
665 
666   void Reserve(FX_STRSIZE len);
667 
668   FX_WCHAR* GetBuffer(FX_STRSIZE len);
669 
670   void ReleaseBuffer(FX_STRSIZE len = -1);
671 
672   int GetInteger() const;
673 
674   FX_FLOAT GetFloat() const;
675 
676   FX_STRSIZE Find(const FX_WCHAR* lpszSub, FX_STRSIZE start = 0) const;
677 
678   FX_STRSIZE Find(FX_WCHAR ch, FX_STRSIZE start = 0) const;
679 
680   FX_STRSIZE Replace(const FX_WCHAR* lpszOld, const FX_WCHAR* lpszNew);
681 
682   FX_STRSIZE Remove(FX_WCHAR ch);
683 
684   CFX_ByteString UTF8Encode() const;
685 
686   CFX_ByteString UTF16LE_Encode() const;
687 
688   void ConvertFrom(const CFX_ByteString& str, CFX_CharMap* pCharMap = NULL);
689 
690  protected:
691   class StringData {
692    public:
693     static StringData* Create(int nLen);
Retain()694     void Retain() { ++m_nRefs; }
Release()695     void Release() {
696       if (--m_nRefs <= 0)
697         FX_Free(this);
698     }
699 
700     intptr_t m_nRefs;  // Would prefer ssize_t, but no windows support.
701     FX_STRSIZE m_nDataLength;
702     FX_STRSIZE m_nAllocLength;
703     FX_WCHAR m_String[1];
704 
705    private:
StringData(FX_STRSIZE dataLen,FX_STRSIZE allocLen)706     StringData(FX_STRSIZE dataLen, FX_STRSIZE allocLen)
707         : m_nRefs(1), m_nDataLength(dataLen), m_nAllocLength(allocLen) {
708       FXSYS_assert(dataLen >= 0);
709       FXSYS_assert(allocLen >= 0);
710       FXSYS_assert(dataLen <= allocLen);
711       m_String[dataLen] = 0;
712     }
713     ~StringData() = delete;
714   };
715 
716   void CopyBeforeWrite();
717   void AllocBeforeWrite(FX_STRSIZE nLen);
718   void ConcatInPlace(FX_STRSIZE nSrcLen, const FX_WCHAR* lpszSrcData);
719   void ConcatCopy(FX_STRSIZE nSrc1Len,
720                   const FX_WCHAR* lpszSrc1Data,
721                   FX_STRSIZE nSrc2Len,
722                   const FX_WCHAR* lpszSrc2Data);
723   void AssignCopy(FX_STRSIZE nSrcLen, const FX_WCHAR* lpszSrcData);
724   void AllocCopy(CFX_WideString& dest,
725                  FX_STRSIZE nCopyLen,
726                  FX_STRSIZE nCopyIndex) const;
727 
728   StringData* m_pData;
729   friend class fxcrt_WideStringConcatInPlace_Test;
730 };
CFX_WideStringC(const CFX_WideString & src)731 inline CFX_WideStringC::CFX_WideStringC(const CFX_WideString& src) {
732   m_Ptr = src.c_str();
733   m_Length = src.GetLength();
734 }
735 inline CFX_WideStringC& CFX_WideStringC::operator=(const CFX_WideString& src) {
736   m_Ptr = src.c_str();
737   m_Length = src.GetLength();
738   return *this;
739 }
740 
741 inline CFX_WideString operator+(const CFX_WideStringC& str1,
742                                 const CFX_WideStringC& str2) {
743   return CFX_WideString(str1, str2);
744 }
745 inline CFX_WideString operator+(const CFX_WideStringC& str1,
746                                 const FX_WCHAR* str2) {
747   return CFX_WideString(str1, str2);
748 }
749 inline CFX_WideString operator+(const FX_WCHAR* str1,
750                                 const CFX_WideStringC& str2) {
751   return CFX_WideString(str1, str2);
752 }
753 inline CFX_WideString operator+(const CFX_WideStringC& str1, FX_WCHAR ch) {
754   return CFX_WideString(str1, CFX_WideStringC(ch));
755 }
756 inline CFX_WideString operator+(FX_WCHAR ch, const CFX_WideStringC& str2) {
757   return CFX_WideString(ch, str2);
758 }
759 inline CFX_WideString operator+(const CFX_WideString& str1,
760                                 const CFX_WideString& str2) {
761   return CFX_WideString(str1, str2);
762 }
763 inline CFX_WideString operator+(const CFX_WideString& str1, FX_WCHAR ch) {
764   return CFX_WideString(str1, CFX_WideStringC(ch));
765 }
766 inline CFX_WideString operator+(FX_WCHAR ch, const CFX_WideString& str2) {
767   return CFX_WideString(ch, str2);
768 }
769 inline CFX_WideString operator+(const CFX_WideString& str1,
770                                 const FX_WCHAR* str2) {
771   return CFX_WideString(str1, str2);
772 }
773 inline CFX_WideString operator+(const FX_WCHAR* str1,
774                                 const CFX_WideString& str2) {
775   return CFX_WideString(str1, str2);
776 }
777 inline CFX_WideString operator+(const CFX_WideString& str1,
778                                 const CFX_WideStringC& str2) {
779   return CFX_WideString(str1, str2);
780 }
781 inline CFX_WideString operator+(const CFX_WideStringC& str1,
782                                 const CFX_WideString& str2) {
783   return CFX_WideString(str1, str2);
784 }
785 inline bool operator==(const wchar_t* lhs, const CFX_WideString& rhs) {
786   return rhs == lhs;
787 }
788 inline bool operator==(const CFX_WideStringC& lhs, const CFX_WideString& rhs) {
789   return rhs == lhs;
790 }
791 inline bool operator!=(const wchar_t* lhs, const CFX_WideString& rhs) {
792   return rhs != lhs;
793 }
794 inline bool operator!=(const CFX_WideStringC& lhs, const CFX_WideString& rhs) {
795   return rhs != lhs;
796 }
797 FX_FLOAT FX_atof(const CFX_ByteStringC& str);
798 void FX_atonum(const CFX_ByteStringC& str, FX_BOOL& bInteger, void* pData);
799 FX_STRSIZE FX_ftoa(FX_FLOAT f, FX_CHAR* buf);
800 CFX_ByteString FX_UTF8Encode(const FX_WCHAR* pwsStr, FX_STRSIZE len);
FX_UTF8Encode(const CFX_WideStringC & wsStr)801 inline CFX_ByteString FX_UTF8Encode(const CFX_WideStringC& wsStr) {
802   return FX_UTF8Encode(wsStr.GetPtr(), wsStr.GetLength());
803 }
FX_UTF8Encode(const CFX_WideString & wsStr)804 inline CFX_ByteString FX_UTF8Encode(const CFX_WideString& wsStr) {
805   return FX_UTF8Encode(wsStr.c_str(), wsStr.GetLength());
806 }
807 
808 #endif  // CORE_INCLUDE_FXCRT_FX_STRING_H_
809