ExtMfcSafeNativeTCHAR.h
上传用户:sesekoo
上传日期:2020-07-18
资源大小:21543k
文件大小:143k
源码类别:

界面编程

开发平台:

Visual C++

  1. // This is part of the Professional User Interface Suite library.
  2. // Copyright (C) 2001-2009 FOSS Software, Inc.
  3. // All rights reserved.
  4. //
  5. // http://www.prof-uis.com
  6. // mailto:support@prof-uis.com
  7. //
  8. // This source code can be used, modified and redistributed
  9. // under the terms of the license agreement that is included
  10. // in the Professional User Interface Suite package.
  11. //
  12. // Warranties and Disclaimers:
  13. // THIS SOFTWARE IS PROVIDED "AS IS" WITHOUT WARRANTY OF ANY KIND
  14. // INCLUDING, BUT NOT LIMITED TO, WARRANTIES OF MERCHANTABILITY,
  15. // FITNESS FOR A PARTICULAR PURPOSE AND NON-INFRINGEMENT.
  16. // IN NO EVENT WILL FOSS SOFTWARE INC. BE LIABLE FOR ANY DIRECT,
  17. // INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY OR CONSEQUENTIAL DAMAGES,
  18. // INCLUDING DAMAGES FOR LOSS OF PROFITS, LOSS OR INACCURACY OF DATA,
  19. // INCURRED BY ANY PERSON FROM SUCH PERSON'S USAGE OF THIS SOFTWARE
  20. // EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGES.
  21. #if (!defined __ExtMfcSafeNativeTCHAR_H)
  22. #define __ExtMfcSafeNativeTCHAR_H
  23. #if _MFC_VER >= 0x800
  24. // disable warning 4201
  25. #pragma warning( push )
  26. #pragma warning ( disable : 4996 )
  27. #endif // _MFC_VER >= 0x800
  28. //
  29. //
  30. //
  31. // Prof-UIS native unicode character type support for VC++ .NET 2002 or later
  32. //
  33. //
  34. //
  35. // enable template-based chars for unicode builds under Visual C++ .NET
  36. // (if commented then automatic conversion between "unsigned short" and
  37. // native "__wchar_t" is off)
  38. //
  39. // #define __EXT_MFC_ENABLE_TEMPLATED_CHARS
  40. //
  41. #if _MFC_VER >= 0x700
  42. #if (defined _UNICODE) && (defined _WCHAR_T_DEFINED)
  43. #if (defined __EXT_MFC_ENABLE_TEMPLATED_CHARS)
  44. // this should be configured by user
  45. // (determines whether Prof-UIS/UNICODE uses
  46. // native "__wchar_t" or not)
  47. // #define __EXT_MFC_COMPILED_WITH_NATIVE_WCHAR_T
  48. // signal we are using extra-safe and compatible chars
  49. #define __EXT_MFC_TEMPLATED_CHARS_IMPLEMENTED
  50. #if (defined _NATIVE_WCHAR_T_DEFINED)
  51. #pragma message("   Prof-UIS native unicode character type support:")
  52. #pragma message("      adding automatic conversion from unsigned short to __wchar_t")
  53. // if we compiling the Prof-UIS library or client project
  54. // withing native unicode character type,
  55. // then we need to include additonal exports for the clients
  56. // without _NATIVE_WCHAR_T_DEFINED
  57. #if (defined __PROF_UIS_IMPL__)
  58. #if (!defined __EXT_MFC_COMPILED_WITH_NATIVE_WCHAR_T)
  59. #error Prof-UIS native unicode character type support: the __EXT_MFC_COMPILED_WITH_NATIVE_WCHAR_T symbol should be definded to provide client programs about valid Prof-UIS types
  60. #endif // (!defined __EXT_MFC_COMPILED_WITH_NATIVE_WCHAR_T)
  61. #pragma message("   Prof-UIS real wchar_t type is:")
  62. #pragma message("      native __wchar_t")
  63. #endif // (defined __PROF_UIS_IMPL__)
  64. #else  // if - (defined _NATIVE_WCHAR_T_DEFINED)
  65. #pragma message("   Prof-UIS native unicode character type support:")
  66. #pragma message("      adding automatic conversion from __wchar_t to unsigned short")
  67. // if we compiling the Prof-UIS library  or client project
  68. // without native unicode character type, then we need
  69. // to include additonal exports for the clients
  70. // which need _NATIVE_WCHAR_T_DEFINED
  71. #if (defined __PROF_UIS_IMPL__)
  72. #if (defined __EXT_MFC_COMPILED_WITH_NATIVE_WCHAR_T)
  73. #error Prof-UIS native unicode character type support: the __EXT_MFC_COMPILED_WITH_NATIVE_WCHAR_T symbol should NOT be definded to provide client programs about valid Prof-UIS types
  74. #endif // (defined __EXT_MFC_COMPILED_WITH_NATIVE_WCHAR_T)
  75. #pragma message("   Prof-UIS real wchar_t type is:")
  76. #pragma message("      unsigned short")
  77. #endif // (defined __PROF_UIS_IMPL__)
  78. #endif // if not - (defined _NATIVE_WCHAR_T_DEFINED)
  79. #if (!defined __PROF_UIS_IMPL__)
  80. #if (defined __EXT_MFC_COMPILED_WITH_NATIVE_WCHAR_T)
  81. #pragma message("   Prof-UIS real wchar_t type is:")
  82. #pragma message("      native __wchar_t")
  83. #else // if - (defined __EXT_MFC_COMPILED_WITH_NATIVE_WCHAR_T)
  84. #pragma message("   Prof-UIS real wchar_t type is:")
  85. #pragma message("      unsigned short")
  86. #endif // if not - (defined __EXT_MFC_COMPILED_WITH_NATIVE_WCHAR_T)
  87. #if (defined _NATIVE_WCHAR_T_DEFINED)
  88. #pragma message("   Client project real wchar_t type is:")
  89. #pragma message("      native __wchar_t")
  90. #if (defined __EXT_MFC_COMPILED_WITH_NATIVE_WCHAR_T)
  91. #pragma message("      (Prof-UIS native wchar_t conversion system is not needed)")
  92. #endif // (defined __EXT_MFC_COMPILED_WITH_NATIVE_WCHAR_T)
  93. #else  // if - (defined _NATIVE_WCHAR_T_DEFINED)
  94. #pragma message("   Client project real wchar_t type is:")
  95. #pragma message("      unsigned short")
  96. #if (!defined __EXT_MFC_COMPILED_WITH_NATIVE_WCHAR_T)
  97. #pragma message("      (Prof-UIS native wchar_t conversion system is not needed)")
  98. #endif // (!defined __EXT_MFC_COMPILED_WITH_NATIVE_WCHAR_T)
  99. #endif // if not - (defined _NATIVE_WCHAR_T_DEFINED)
  100. #endif // (!defined __PROF_UIS_IMPL__)
  101. #if (defined __EXT_MFC_COMPILED_WITH_NATIVE_WCHAR_T)
  102. typedef __wchar_t __prof_uis_used_wchar_t;
  103. typedef unsigned short __prof_uis_converted_wchar_t;
  104. #else // if - (defined __EXT_MFC_COMPILED_WITH_NATIVE_WCHAR_T)
  105. typedef unsigned short __prof_uis_used_wchar_t;
  106. typedef __wchar_t __prof_uis_converted_wchar_t;
  107. #endif // if - not (defined __EXT_MFC_COMPILED_WITH_NATIVE_WCHAR_T)
  108. template < typename BaseType = char >
  109. class CExtSafeChTraitsBase
  110. {
  111. public:
  112. typedef char XCHAR;
  113. typedef char * PXSTR;
  114. typedef const char * PCXSTR;
  115. typedef __wchar_t YCHAR;
  116. typedef __wchar_t * PYSTR;
  117. typedef const __wchar_t * PCYSTR;
  118. }; // class CExtSafeChTraitsBase
  119. template<>
  120. class CExtSafeChTraitsBase< __wchar_t >
  121. {
  122. public:
  123. typedef __wchar_t XCHAR;
  124. typedef __wchar_t * PXSTR;
  125. typedef const __wchar_t * PCXSTR;
  126. typedef char YCHAR;
  127. typedef char * PYSTR;
  128. typedef const char * PCYSTR;
  129. }; // class CExtSafeChTraitsBase
  130. template<>
  131. class CExtSafeChTraitsBase< unsigned short >
  132. {
  133. public:
  134. typedef unsigned short XCHAR;
  135. typedef unsigned short * PXSTR;
  136. typedef const unsigned short * PCXSTR;
  137. typedef char YCHAR;
  138. typedef char * PYSTR;
  139. typedef const char * PCYSTR;
  140. }; // class CExtSafeChTraitsBase
  141. template < typename _CharType = char >
  142. class CExtSafeChTraitsCRT : public CExtSafeChTraitsBase < _CharType >
  143. {
  144. public:
  145. static char * CharNext( const char * p ) throw()
  146. {
  147. return
  148. reinterpret_cast
  149. < char * >
  150. (
  151. _mbsinc(
  152. reinterpret_cast
  153. < const unsigned char * >
  154. ( p )
  155. )
  156. );
  157. }
  158. static int IsDigit( char ch ) throw()
  159. {
  160. return _ismbcdigit( ch );
  161. }
  162. static int IsSpace( char ch ) throw()
  163. {
  164. return _ismbcspace( ch );
  165. }
  166. static int StringCompare( LPCSTR pszA, LPCSTR pszB ) throw()
  167. {
  168. return
  169. _mbscmp(
  170. reinterpret_cast < const unsigned char * > ( pszA ),
  171. reinterpret_cast < const unsigned char * > ( pszB )
  172. );
  173. }
  174. static int StringCompareIgnore( LPCSTR pszA, LPCSTR pszB ) throw()
  175. {
  176. return
  177. _mbsicmp(
  178. reinterpret_cast < const unsigned char * > ( pszA ),
  179. reinterpret_cast < const unsigned char * > ( pszB )
  180. );
  181. }
  182. static int StringCollate( LPCSTR pszA, LPCSTR pszB ) throw()
  183. {
  184. return
  185. _mbscoll(
  186. reinterpret_cast < const unsigned char * > ( pszA ),
  187. reinterpret_cast < const unsigned char * > ( pszB )
  188. );
  189. }
  190. static int StringCollateIgnore( LPCSTR pszA, LPCSTR pszB ) throw()
  191. {
  192. return
  193. _mbsicoll(
  194. reinterpret_cast < const unsigned char * > ( pszA ),
  195. reinterpret_cast < const unsigned char * > ( pszB )
  196. );
  197. }
  198. static LPCSTR StringFindString( LPCSTR pszBlock, LPCSTR pszMatch ) throw()
  199. {
  200. return
  201. reinterpret_cast < LPCSTR >
  202. (
  203. _mbsstr(
  204. reinterpret_cast < const unsigned char * > ( pszBlock ),
  205. reinterpret_cast < const unsigned char * > ( pszMatch )
  206. )
  207. );
  208. }
  209. static LPSTR StringFindString( LPSTR pszBlock, LPCSTR pszMatch ) throw()
  210. {
  211. return
  212. (
  213. const_cast < LPSTR >
  214. (
  215. StringFindString(
  216. const_cast < LPCSTR > ( pszBlock ),
  217. pszMatch
  218. )
  219. )
  220. );
  221. }
  222. static LPCSTR StringFindChar( LPCSTR pszBlock, char chMatch ) throw()
  223. {
  224. return
  225. reinterpret_cast < LPCSTR >
  226. (
  227. _mbschr(
  228. reinterpret_cast < const unsigned char * > ( pszBlock ),
  229. chMatch
  230. )
  231. );
  232. }
  233. static LPCSTR StringFindCharRev( LPCSTR psz, char ch ) throw()
  234. {
  235. return
  236. reinterpret_cast < LPSTR >
  237. (
  238. _mbsrchr(
  239. reinterpret_cast < const unsigned char * > ( psz ),
  240. ch
  241. )
  242. );
  243. }
  244. static LPCSTR StringScanSet( LPCSTR pszBlock, LPCSTR pszMatch ) throw()
  245. {
  246. return
  247. reinterpret_cast < LPCSTR >
  248. (
  249. _mbspbrk(
  250. reinterpret_cast < const unsigned char * > ( pszBlock ),
  251. reinterpret_cast < const unsigned char * > ( pszMatch )
  252. )
  253. );
  254. }
  255. static int StringSpanIncluding( LPCSTR pszBlock, LPCSTR pszSet ) throw()
  256. {
  257. return
  258. (int) _mbsspn(
  259. reinterpret_cast < const unsigned char * > ( pszBlock ),
  260. reinterpret_cast < const unsigned char * > ( pszSet )
  261. );
  262. }
  263. static int StringSpanExcluding( LPCSTR pszBlock, LPCSTR pszSet ) throw()
  264. {
  265. return
  266. (int) _mbscspn(
  267. reinterpret_cast < const unsigned char * > ( pszBlock ),
  268. reinterpret_cast < const unsigned char * > ( pszSet )
  269. );
  270. }
  271. static LPSTR StringUppercase( LPSTR psz ) throw()
  272. {
  273. return
  274. reinterpret_cast < LPSTR >
  275. (
  276. _mbsupr(
  277. reinterpret_cast < unsigned char * > ( psz )
  278. )
  279. );
  280. }
  281. static LPSTR StringLowercase( LPSTR psz ) throw()
  282. {
  283. return
  284. reinterpret_cast < LPSTR >
  285. (
  286. _mbslwr(
  287. reinterpret_cast < unsigned char * > ( psz )
  288. )
  289. );
  290. }
  291. static LPSTR StringReverse( LPSTR psz ) throw()
  292. {
  293. return
  294. reinterpret_cast < LPSTR >
  295. (
  296. _mbsrev(
  297. reinterpret_cast < unsigned char * > ( psz )
  298. )
  299. );
  300. }
  301. static int GetFormattedLength( LPCSTR pszFormat, va_list args ) throw()
  302. {
  303. return _vscprintf( pszFormat, args );
  304. }
  305. static int Format( LPSTR pszBuffer, LPCSTR pszFormat, va_list args ) throw()
  306. {
  307. return vsprintf( pszBuffer, pszFormat, args );
  308. }
  309. static int GetBaseTypeLength( LPCSTR pszSrc ) throw()
  310. {
  311. return int( strlen( pszSrc ) );
  312. }
  313. static int GetBaseTypeLength( LPCSTR pszSrc, int nLength ) throw()
  314. {
  315. (void)pszSrc;
  316. return nLength;
  317. }
  318. static int GetBaseTypeLength( LPCWSTR pszSource ) throw()
  319. {
  320. return
  321. ::WideCharToMultiByte(
  322. _AtlGetConversionACP(),
  323. 0,
  324. pszSource,
  325. -1,
  326. NULL,
  327. 0,
  328. NULL,
  329. NULL
  330. ) - 1;
  331. }
  332. static int GetBaseTypeLength( LPCWSTR pszSource, int nLength ) throw()
  333. {
  334. return
  335. ::WideCharToMultiByte(
  336. _AtlGetConversionACP(),
  337. 0,
  338. pszSource,
  339. nLength,
  340. NULL,
  341. 0,
  342. NULL,
  343. NULL
  344. );
  345. }
  346. static void ConvertToBaseType(
  347. LPSTR pszDest,
  348. int nDestLength,
  349. LPCSTR pszSrc,
  350. int nSrcLength = -1
  351. ) throw()
  352. {
  353. (void)nSrcLength;
  354. __EXT_MFC_MEMCPY(
  355. pszDest,
  356. nDestLength * sizeof(char),
  357. pszSrc,
  358. nDestLength * sizeof(char) 
  359. );
  360. }
  361. static void ConvertToBaseType(
  362. LPSTR pszDest,
  363. int nDestLength,
  364. LPCWSTR pszSrc,
  365. int nSrcLength = -1
  366. ) throw()
  367. {
  368. ::WideCharToMultiByte(
  369. _AtlGetConversionACP(),
  370. 0,
  371. pszSrc,
  372. nSrcLength,
  373. pszDest,
  374. nDestLength,
  375. NULL,
  376. NULL
  377. );
  378. }
  379. static void ConvertToOem( LPSTR psz ) throw()
  380. {
  381. ::AnsiToOem( psz, psz );
  382. }
  383. static void ConvertToAnsi( LPSTR psz ) throw()
  384. {
  385. ::OemToAnsi( psz, psz );
  386. }
  387. static void FloodCharacters( char ch, int nLength, char * pch ) throw()
  388. {
  389. memset( pch, ch, nLength );
  390. }
  391. static BSTR AllocSysString( const char * pchData, int nDataLength ) throw()
  392. {
  393. int nLen =
  394. ::MultiByteToWideChar(
  395. _AtlGetConversionACP(),
  396. 0,
  397. pchData,
  398. nDataLength,
  399. NULL,
  400. NULL
  401. );
  402. BSTR bstr = ::SysAllocStringLen( NULL, nLen );
  403. if( bstr != NULL )
  404. {
  405. ::MultiByteToWideChar(
  406. _AtlGetConversionACP(),
  407. 0,
  408. pchData,
  409. nDataLength,
  410. bstr,
  411. nLen
  412. );
  413. }
  414. return bstr;
  415. }
  416. static BOOL ReAllocSysString( const char * pchData, BSTR* pbstr, int nDataLength ) throw()
  417. {
  418. int nLen =
  419. ::MultiByteToWideChar(
  420. _AtlGetConversionACP(),
  421. 0,
  422. pchData,
  423. nDataLength,
  424. NULL,
  425. NULL
  426. );
  427. BOOL bSuccess = ::SysReAllocStringLen( pbstr, NULL, nLen );
  428. if( bSuccess )
  429. {
  430. ::MultiByteToWideChar(
  431. _AtlGetConversionACP(),
  432. 0,
  433. pchData,
  434. nDataLength,
  435. *pbstr,
  436. nLen
  437. );
  438. }
  439. return bSuccess;
  440. }
  441. static DWORD FormatMessage(
  442. DWORD dwFlags,
  443. LPCVOID pSource,
  444. DWORD dwMessageID,
  445. DWORD dwLanguageID,
  446. LPSTR pszBuffer,
  447. DWORD nSize,
  448. va_list * pArguments
  449. ) throw()
  450. {
  451. return
  452. ::FormatMessageA(
  453. dwFlags,
  454. pSource,
  455. dwMessageID,
  456. dwLanguageID,
  457. pszBuffer,
  458. nSize,
  459. pArguments
  460. );
  461. }
  462. static int SafeStringLen( LPCSTR psz ) throw()
  463. {
  464. return (psz != NULL) ? int( strlen( psz ) ) : 0;
  465. }
  466. static int SafeStringLen( LPCWSTR psz ) throw()
  467. {
  468. return (psz != NULL) ? int( wcslen( psz ) ) : 0;
  469. }
  470. static int GetCharLen( const __wchar_t * pch ) throw()
  471. {
  472. (void)pch;
  473. return 1;
  474. }
  475. static int GetCharLen( const char* pch ) throw()
  476. {
  477. return int(
  478. _mbclen(
  479. reinterpret_cast < const unsigned char * > ( pch )
  480. )
  481. );
  482. }
  483. static DWORD GetEnvironmentVariable(
  484. LPCSTR pszVar,
  485. LPSTR pszBuffer,
  486. DWORD dwSize
  487. ) throw()
  488. {
  489. return ::GetEnvironmentVariableA( pszVar, pszBuffer, dwSize );
  490. }
  491. }; // class CExtSafeChTraitsCRT
  492. template<>
  493. class CExtSafeChTraitsCRT < __wchar_t > : public ChTraitsBase < __wchar_t >
  494. {
  495. #if defined(_UNICODE) && !defined(_CSTRING_ALWAYS_THUNK)
  496. static DWORD _GetEnvironmentVariableW(
  497. const __wchar_t * pszName,
  498. __wchar_t * pszBuffer,
  499. DWORD nSize
  500. ) throw()
  501. {
  502. return
  503. ::GetEnvironmentVariableW(
  504. (LPCWSTR)pszName,
  505. (LPWSTR)pszBuffer,
  506. nSize
  507. );
  508. }
  509. #else
  510. static DWORD WINAPI _GetEnvironmentVariableW(
  511. const __wchar_t * pszName, 
  512. __wchar_t * pszBuffer,
  513. DWORD nSize
  514. ) throw()
  515. {
  516. return
  517. _strthunks.pfnGetEnvironmentVariableW(
  518. (LPCWSTR)pszName,
  519. (LPWSTR)pszBuffer,
  520. nSize
  521. );
  522. }
  523. #endif
  524. public:
  525. static __wchar_t * CharNext( const __wchar_t * psz ) throw()
  526. {
  527. return const_cast < __wchar_t * > ( psz+1 );
  528. }
  529. static int IsDigit( __wchar_t ch ) throw()
  530. {
  531. return iswdigit( ch );
  532. }
  533. static int IsSpace( __wchar_t ch ) throw()
  534. {
  535. return iswspace( ch );
  536. }
  537. static int StringCompare( const __wchar_t * pszA, const __wchar_t * pszB ) throw()
  538. {
  539. return wcscmp( (wchar_t *) pszA, (wchar_t *) pszB );
  540. }
  541. static int StringCompareIgnore( const __wchar_t * pszA, const __wchar_t * pszB ) throw()
  542. {
  543. return _wcsicmp( (wchar_t *)pszA, (wchar_t *)pszB );
  544. }
  545. static int StringCollate( const __wchar_t * pszA, const __wchar_t * pszB ) throw()
  546. {
  547. return wcscoll( (wchar_t *)pszA, (wchar_t *)pszB );
  548. }
  549. static int StringCollateIgnore( const __wchar_t * pszA, const __wchar_t * pszB ) throw()
  550. {
  551. return _wcsicoll( (wchar_t *)pszA, (wchar_t *)pszB );
  552. }
  553. static const __wchar_t * StringFindString( const __wchar_t * pszBlock, const __wchar_t * pszMatch ) throw()
  554. {
  555. return (__wchar_t *)wcsstr( (wchar_t *)pszBlock, (wchar_t *)pszMatch );
  556. }
  557. static __wchar_t * StringFindString( __wchar_t * pszBlock, const __wchar_t * pszMatch ) throw()
  558. {
  559. return ( const_cast < __wchar_t * > ( StringFindString( const_cast < const __wchar_t * > ( pszBlock ), pszMatch ) ) );
  560. }
  561. static const __wchar_t * StringFindChar( const __wchar_t * pszBlock, __wchar_t chMatch ) throw()
  562. {
  563. return (__wchar_t *)wcschr( (wchar_t *)pszBlock, (wchar_t)chMatch );
  564. }
  565. static const __wchar_t * StringFindCharRev( const __wchar_t * psz, __wchar_t ch ) throw()
  566. {
  567. return (__wchar_t *)wcsrchr( (wchar_t *)psz, (wchar_t)ch );
  568. }
  569. static const __wchar_t * StringScanSet( const __wchar_t * pszBlock, const __wchar_t * pszMatch ) throw()
  570. {
  571. return (__wchar_t *)wcspbrk( (wchar_t *)pszBlock, (wchar_t *)pszMatch );
  572. }
  573. static int StringSpanIncluding( const __wchar_t * pszBlock, const __wchar_t * pszSet ) throw()
  574. {
  575. return (int)wcsspn( (wchar_t *)pszBlock, (wchar_t *)pszSet );
  576. }
  577. static int StringSpanExcluding( const __wchar_t * pszBlock, const __wchar_t * pszSet ) throw()
  578. {
  579. return (int)wcscspn( (wchar_t *)pszBlock, (wchar_t *)pszSet );
  580. }
  581. static __wchar_t * StringUppercase( __wchar_t * psz ) throw()
  582. {
  583. return (__wchar_t *)_wcsupr( (wchar_t *)psz );
  584. }
  585. static __wchar_t * StringLowercase( __wchar_t * psz ) throw()
  586. {
  587. return (__wchar_t *)_wcslwr( (wchar_t *)psz );
  588. }
  589. static __wchar_t * StringReverse( __wchar_t * psz ) throw()
  590. {
  591. return (__wchar_t *)_wcsrev( (wchar_t *)psz );
  592. }
  593. static int GetFormattedLength( const __wchar_t * pszFormat, va_list args) throw()
  594. {
  595. return _vscwprintf( (wchar_t *)pszFormat, args );
  596. }
  597. static int Format( __wchar_t * pszBuffer, const __wchar_t * pszFormat, va_list args) throw()
  598. {
  599. return vswprintf( (wchar_t *)pszBuffer, (wchar_t *)pszFormat, args );
  600. }
  601. static int GetBaseTypeLength( LPCSTR pszSrc ) throw()
  602. {
  603. return
  604. ::MultiByteToWideChar(
  605. _AtlGetConversionACP(),
  606. 0,
  607. pszSrc,
  608. -1,
  609. NULL,
  610. 0
  611. ) - 1;
  612. }
  613. static int GetBaseTypeLength( LPCSTR pszSrc, int nLength ) throw()
  614. {
  615. return
  616. ::MultiByteToWideChar(
  617. _AtlGetConversionACP(),
  618. 0,
  619. pszSrc,
  620. nLength,
  621. NULL,
  622. 0
  623. );
  624. }
  625. static int GetBaseTypeLength( const __wchar_t * pszSrc ) throw()
  626. {
  627. return (int)wcslen( (wchar_t *)pszSrc );
  628. }
  629. static int GetBaseTypeLength( const __wchar_t * pszSrc, int nLength ) throw()
  630. {
  631. (void)pszSrc;
  632. return nLength;
  633. }
  634. static void ConvertToBaseType(
  635. __wchar_t * pszDest,
  636. int nDestLength,
  637. LPCSTR pszSrc,
  638. int nSrcLength = -1
  639. ) throw()
  640. {
  641. ::MultiByteToWideChar(
  642. _AtlGetConversionACP(),
  643. 0,
  644. pszSrc,
  645. nSrcLength,
  646. (LPWSTR)pszDest,
  647. nDestLength
  648. );
  649. }
  650. static void ConvertToBaseType(
  651. __wchar_t * pszDest,
  652. int nDestLength,
  653. const __wchar_t * pszSrc,
  654. int nSrcLength = -1
  655. ) throw()
  656. {
  657. (void)nSrcLength;
  658. __EXT_MFC_MEMCPY(
  659. pszDest, 
  660. nDestLength * sizeof(__wchar_t),
  661. pszSrc, 
  662. nDestLength * sizeof(__wchar_t) 
  663. );
  664. }
  665. static void FloodCharacters( __wchar_t ch, int nLength, __wchar_t * psz ) throw()
  666. {
  667. for( int i = 0; i < nLength; i++ )
  668. {
  669. psz[i] = ch;
  670. }
  671. }
  672. static BSTR AllocSysString( const __wchar_t * pchData, int nDataLength ) throw()
  673. {
  674. return ::SysAllocStringLen( (const OLECHAR *)pchData, nDataLength );
  675. }
  676. static BOOL ReAllocSysString( const __wchar_t * pchData, BSTR* pbstr, int nDataLength ) throw()
  677. {
  678. return ::SysReAllocStringLen( pbstr, (const OLECHAR *)pchData, nDataLength );
  679. }
  680. #ifdef _UNICODE
  681. static DWORD FormatMessage(
  682. DWORD dwFlags,
  683. LPCVOID pSource,
  684. DWORD dwMessageID,
  685. DWORD dwLanguageID,
  686. __wchar_t * pszBuffer,
  687. DWORD nSize,
  688. va_list * pArguments
  689. ) throw()
  690. {
  691. return
  692. ::FormatMessageW(
  693. dwFlags,
  694. pSource,
  695. dwMessageID,
  696. dwLanguageID,
  697. (LPWSTR)pszBuffer,
  698. nSize,
  699. pArguments
  700. );
  701. }
  702. #endif
  703. static int SafeStringLen( LPCSTR psz ) throw()
  704. {
  705. return (psz != NULL) ? (int)strlen( psz ) : 0;
  706. }
  707. static int SafeStringLen( const __wchar_t * psz ) throw()
  708. {
  709. return (psz != NULL) ? (int)wcslen( (wchar_t *)psz ) : 0;
  710. }
  711. static int GetCharLen( const __wchar_t * pch ) throw()
  712. {
  713. (void)pch;
  714. return 1;
  715. }
  716. static int GetCharLen( const char * pch ) throw()
  717. {
  718. return (int)(
  719. _mbclen(
  720. reinterpret_cast < const unsigned char * > ( pch )
  721. )
  722. );
  723. }
  724. static DWORD GetEnvironmentVariable( const __wchar_t * pszVar, __wchar_t * pszBuffer, DWORD dwSize ) throw()
  725. {
  726. return _GetEnvironmentVariableW( pszVar, pszBuffer, dwSize );
  727. }
  728. }; // class CExtSafeChTraitsCRT
  729. template<>
  730. class CExtSafeChTraitsCRT < unsigned short > : public ChTraitsBase < unsigned short >
  731. {
  732. #if defined(_UNICODE) && !defined(_CSTRING_ALWAYS_THUNK)
  733. static DWORD _GetEnvironmentVariableW(
  734. const unsigned short * pszName,
  735. unsigned short * pszBuffer,
  736. DWORD nSize
  737. ) throw()
  738. {
  739. return
  740. ::GetEnvironmentVariableW(
  741. (LPCWSTR)pszName,
  742. (LPWSTR)pszBuffer,
  743. nSize
  744. );
  745. }
  746. #else
  747. static DWORD WINAPI _GetEnvironmentVariableW(
  748. const unsigned short * pszName, 
  749. unsigned short * pszBuffer,
  750. DWORD nSize
  751. ) throw()
  752. {
  753. return
  754. _strthunks.pfnGetEnvironmentVariableW(
  755. (LPCWSTR)pszName,
  756. (LPWSTR)pszBuffer,
  757. nSize
  758. );
  759. }
  760. #endif
  761. public:
  762. static unsigned short * CharNext( const unsigned short * psz ) throw()
  763. {
  764. return const_cast < unsigned short * > ( psz+1 );
  765. }
  766. static int IsDigit( unsigned short ch ) throw()
  767. {
  768. return iswdigit( ch );
  769. }
  770. static int IsSpace( unsigned short ch ) throw()
  771. {
  772. return iswspace( ch );
  773. }
  774. static int StringCompare( const unsigned short * pszA, const unsigned short * pszB ) throw()
  775. {
  776. return wcscmp( (wchar_t *)pszA, (wchar_t *)pszB );
  777. }
  778. static int StringCompareIgnore( const unsigned short * pszA, const unsigned short * pszB ) throw()
  779. {
  780. return _wcsicmp( (wchar_t *)pszA, (wchar_t *)pszB );
  781. }
  782. static int StringCollate( const unsigned short * pszA, const unsigned short * pszB ) throw()
  783. {
  784. return wcscoll( (wchar_t *)pszA, (wchar_t *)pszB );
  785. }
  786. static int StringCollateIgnore( const unsigned short * pszA, const unsigned short * pszB ) throw()
  787. {
  788. return _wcsicoll( (wchar_t *)pszA, (wchar_t *)pszB );
  789. }
  790. static const unsigned short * StringFindString( const unsigned short * pszBlock, const unsigned short * pszMatch ) throw()
  791. {
  792. return (unsigned short *)wcsstr( (wchar_t *)pszBlock, (wchar_t *)pszMatch );
  793. }
  794. static unsigned short * StringFindString( unsigned short * pszBlock, const unsigned short * pszMatch ) throw()
  795. {
  796. return ( const_cast < unsigned short * > ( StringFindString( const_cast < const unsigned short * > ( pszBlock ), pszMatch ) ) );
  797. }
  798. static const unsigned short * StringFindChar( const unsigned short * pszBlock, unsigned short chMatch ) throw()
  799. {
  800. return (unsigned short *)wcschr( (wchar_t *)pszBlock, (wchar_t)chMatch );
  801. }
  802. static const unsigned short * StringFindCharRev( const unsigned short * psz, unsigned short ch ) throw()
  803. {
  804. return (unsigned short *)wcsrchr( (wchar_t *)psz, (wchar_t)ch );
  805. }
  806. static const unsigned short * StringScanSet( const unsigned short * pszBlock, const unsigned short * pszMatch ) throw()
  807. {
  808. return (unsigned short *)wcspbrk( (wchar_t *)pszBlock, (wchar_t *)pszMatch );
  809. }
  810. static int StringSpanIncluding( const unsigned short * pszBlock, const unsigned short * pszSet ) throw()
  811. {
  812. return (int)wcsspn( (wchar_t *)pszBlock, (wchar_t *)pszSet );
  813. }
  814. static int StringSpanExcluding( const unsigned short * pszBlock, const unsigned short * pszSet ) throw()
  815. {
  816. return (int)wcscspn( (wchar_t *)pszBlock, (wchar_t *)pszSet );
  817. }
  818. static unsigned short * StringUppercase( unsigned short * psz ) throw()
  819. {
  820. return (unsigned short *)_wcsupr( (wchar_t *)psz );
  821. }
  822. static unsigned short * StringLowercase( unsigned short * psz ) throw()
  823. {
  824. return (unsigned short *)_wcslwr( (wchar_t *)psz );
  825. }
  826. static unsigned short * StringReverse( unsigned short * psz ) throw()
  827. {
  828. return (unsigned short *)_wcsrev( (wchar_t *)psz );
  829. }
  830. static int GetFormattedLength( const unsigned short * pszFormat, va_list args) throw()
  831. {
  832. return _vscwprintf( (wchar_t *)pszFormat, args );
  833. }
  834. static int Format( unsigned short * pszBuffer, const unsigned short * pszFormat, va_list args) throw()
  835. {
  836. return vswprintf( (wchar_t *)pszBuffer, (wchar_t *)pszFormat, args );
  837. }
  838. static int GetBaseTypeLength( LPCSTR pszSrc ) throw()
  839. {
  840. return
  841. ::MultiByteToWideChar(
  842. _AtlGetConversionACP(),
  843. 0,
  844. pszSrc,
  845. -1,
  846. NULL,
  847. 0
  848. ) - 1;
  849. }
  850. static int GetBaseTypeLength( LPCSTR pszSrc, int nLength ) throw()
  851. {
  852. return
  853. ::MultiByteToWideChar(
  854. _AtlGetConversionACP(),
  855. 0,
  856. pszSrc,
  857. nLength,
  858. NULL,
  859. 0
  860. );
  861. }
  862. static int GetBaseTypeLength( const unsigned short * pszSrc ) throw()
  863. {
  864. return (int)wcslen( (wchar_t *)pszSrc );
  865. }
  866. static int GetBaseTypeLength( const unsigned short * pszSrc, int nLength ) throw()
  867. {
  868. (void)pszSrc;
  869. return nLength;
  870. }
  871. static void ConvertToBaseType(
  872. unsigned short * pszDest,
  873. int nDestLength,
  874. LPCSTR pszSrc,
  875. int nSrcLength = -1
  876. ) throw()
  877. {
  878. ::MultiByteToWideChar(
  879. _AtlGetConversionACP(),
  880. 0,
  881. pszSrc,
  882. nSrcLength,
  883. (LPWSTR)pszDest,
  884. nDestLength
  885. );
  886. }
  887. static void ConvertToBaseType(
  888. unsigned short * pszDest,
  889. int nDestLength,
  890. const unsigned short * pszSrc,
  891. int nSrcLength = -1
  892. ) throw()
  893. {
  894. (void)nSrcLength;
  895. __EXT_MFC_MEMCPY(
  896. pszDest,
  897. nDestLength * sizeof(unsigned short),
  898. pszSrc, 
  899. nDestLength * sizeof(unsigned short) 
  900. );
  901. }
  902. static void FloodCharacters( unsigned short ch, int nLength, unsigned short * psz ) throw()
  903. {
  904. for( int i = 0; i < nLength; i++ )
  905. {
  906. psz[i] = ch;
  907. }
  908. }
  909. static BSTR AllocSysString( const unsigned short* pchData, int nDataLength ) throw()
  910. {
  911. return ::SysAllocStringLen( (const OLECHAR *)pchData, nDataLength );
  912. }
  913. static BOOL ReAllocSysString( const unsigned short* pchData, BSTR* pbstr, int nDataLength ) throw()
  914. {
  915. return ::SysReAllocStringLen( pbstr, (const OLECHAR *)pchData, nDataLength );
  916. }
  917. #ifdef _UNICODE
  918. static DWORD FormatMessage(
  919. DWORD dwFlags,
  920. LPCVOID pSource,
  921. DWORD dwMessageID,
  922. DWORD dwLanguageID,
  923. unsigned short * pszBuffer,
  924. DWORD nSize,
  925. va_list * pArguments
  926. ) throw()
  927. {
  928. return
  929. ::FormatMessageW(
  930. dwFlags,
  931. pSource,
  932. dwMessageID,
  933. dwLanguageID,
  934. (LPWSTR)pszBuffer,
  935. nSize,
  936. pArguments
  937. );
  938. }
  939. #endif
  940. static int SafeStringLen( LPCSTR psz ) throw()
  941. {
  942. return (psz != NULL) ? (int)strlen( psz ) : 0;
  943. }
  944. static int SafeStringLen( const unsigned short * psz ) throw()
  945. {
  946. return (psz != NULL) ? (int)wcslen( (wchar_t *)psz ) : 0;
  947. }
  948. static int GetCharLen( const unsigned short * pch ) throw()
  949. {
  950. (void)pch;
  951. return 1;
  952. }
  953. static int GetCharLen( const char * pch ) throw()
  954. {
  955. return (int)(
  956. _mbclen(
  957. reinterpret_cast < const unsigned char * > ( pch )
  958. )
  959. );
  960. }
  961. static DWORD GetEnvironmentVariable(
  962. const unsigned short * pszVar,
  963. unsigned short * pszBuffer,
  964. DWORD dwSize
  965. ) throw()
  966. {
  967. return _GetEnvironmentVariableW( pszVar, pszBuffer, dwSize );
  968. }
  969. }; // class CExtSafeChTraitsCRT
  970. template <
  971. typename _CharType = char,
  972. class StringIterator = CExtSafeChTraitsCRT < char >
  973. >
  974. class CExtSafeStrTraitMFC : public StringIterator
  975. {
  976. public:
  977. static HINSTANCE FindStringResourceInstance( UINT nID ) throw()
  978. {
  979. return ( AfxFindStringResourceHandle( nID ) );
  980. }
  981. static ATL::IAtlStringMgr * GetDefaultManager() throw()
  982. {
  983. return ( AfxGetStringManager() );
  984. }
  985. static char* CharNext( const char * p ) throw()
  986. {
  987. return reinterpret_cast < char* > ( _mbsinc( reinterpret_cast < const unsigned char* > ( p ) ) );
  988. }
  989. static int IsDigit( char ch ) throw()
  990. {
  991. return _ismbcdigit( ch );
  992. }
  993. static int IsSpace( char ch ) throw()
  994. {
  995. return _ismbcspace( ch );
  996. }
  997. static int StringCompare( LPCSTR pszA, LPCSTR pszB ) throw()
  998. {
  999. return
  1000. _mbscmp(
  1001. reinterpret_cast < const unsigned char * > ( pszA ),
  1002. reinterpret_cast < const unsigned char * > ( pszB )
  1003. );
  1004. }
  1005. static int StringCompareIgnore( LPCSTR pszA, LPCSTR pszB ) throw()
  1006. {
  1007. return
  1008. _mbsicmp(
  1009. reinterpret_cast < const unsigned char * > ( pszA ),
  1010. reinterpret_cast < const unsigned char * > ( pszB )
  1011. );
  1012. }
  1013. static int StringCollate( LPCSTR pszA, LPCSTR pszB ) throw()
  1014. {
  1015. return
  1016. _mbscoll(
  1017. reinterpret_cast < const unsigned char * > ( pszA ),
  1018. reinterpret_cast < const unsigned char * > ( pszB )
  1019. );
  1020. }
  1021. static int StringCollateIgnore( LPCSTR pszA, LPCSTR pszB ) throw()
  1022. {
  1023. return
  1024. _mbsicoll(
  1025. reinterpret_cast < const unsigned char * > ( pszA ),
  1026. reinterpret_cast < const unsigned char * > ( pszB )
  1027. );
  1028. }
  1029. static LPCSTR StringFindString( LPCSTR pszBlock, LPCSTR pszMatch ) throw()
  1030. {
  1031. return reinterpret_cast < LPCSTR > (
  1032. _mbsstr(
  1033. reinterpret_cast < const unsigned char * > ( pszBlock ),
  1034. reinterpret_cast < const unsigned char * > ( pszMatch )
  1035. )
  1036. );
  1037. }
  1038. static LPSTR StringFindString( LPSTR pszBlock, LPCSTR pszMatch ) throw()
  1039. {
  1040. return ( const_cast < LPSTR >
  1041. (
  1042. StringFindString(
  1043. const_cast < LPCSTR > ( pszBlock ),
  1044. pszMatch
  1045. )
  1046. )
  1047. );
  1048. }
  1049. static LPCSTR StringFindChar( LPCSTR pszBlock, char chMatch ) throw()
  1050. {
  1051. return reinterpret_cast < LPCSTR > (
  1052. _mbschr(
  1053. reinterpret_cast < const unsigned char* > ( pszBlock ),
  1054. chMatch
  1055. )
  1056. );
  1057. }
  1058. static LPCSTR StringFindCharRev( LPCSTR psz, char ch ) throw()
  1059. {
  1060. return reinterpret_cast < LPSTR > (
  1061. _mbsrchr(
  1062. reinterpret_cast < const unsigned char* > ( psz ),
  1063. ch
  1064. )
  1065. );
  1066. }
  1067. static LPCSTR StringScanSet( LPCSTR pszBlock, LPCSTR pszMatch ) throw()
  1068. {
  1069. return reinterpret_cast < LPCSTR > (
  1070. _mbspbrk(
  1071. reinterpret_cast < const unsigned char * > ( pszBlock ),
  1072. reinterpret_cast < const unsigned char * > ( pszMatch )
  1073. )
  1074. );
  1075. }
  1076. static int StringSpanIncluding( LPCSTR pszBlock, LPCSTR pszSet ) throw()
  1077. {
  1078. return
  1079. (int) _mbsspn(
  1080. reinterpret_cast < const unsigned char * > ( pszBlock ),
  1081. reinterpret_cast < const unsigned char * > ( pszSet )
  1082. );
  1083. }
  1084. static int StringSpanExcluding( LPCSTR pszBlock, LPCSTR pszSet ) throw()
  1085. {
  1086. return
  1087. (int) _mbscspn(
  1088. reinterpret_cast < const unsigned char * > ( pszBlock ),
  1089. reinterpret_cast < const unsigned char * > ( pszSet )
  1090. );
  1091. }
  1092. static LPSTR StringUppercase( LPSTR psz ) throw()
  1093. {
  1094. return reinterpret_cast < LPSTR > (
  1095. _mbsupr(
  1096. reinterpret_cast < unsigned char * > ( psz )
  1097. )
  1098. );
  1099. }
  1100. static LPSTR StringLowercase( LPSTR psz ) throw()
  1101. {
  1102. return reinterpret_cast < LPSTR > (
  1103. _mbslwr(
  1104. reinterpret_cast < unsigned char * > ( psz )
  1105. )
  1106. );
  1107. }
  1108. static LPSTR StringReverse( LPSTR psz ) throw()
  1109. {
  1110. return reinterpret_cast < LPSTR > (
  1111. _mbsrev(
  1112. reinterpret_cast < unsigned char * > ( psz )
  1113. )
  1114. );
  1115. }
  1116. static int GetFormattedLength( LPCSTR pszFormat, va_list args ) throw()
  1117. {
  1118. return _vscprintf( pszFormat, args );
  1119. }
  1120. static int Format( LPSTR pszBuffer, LPCSTR pszFormat, va_list args ) throw()
  1121. {
  1122. return vsprintf( pszBuffer, pszFormat, args );
  1123. }
  1124. static int GetBaseTypeLength( LPCSTR pszSrc ) throw()
  1125. {
  1126. return int( strlen( pszSrc ) );
  1127. }
  1128. static int GetBaseTypeLength( LPCSTR pszSrc, int nLength ) throw()
  1129. {
  1130. (void)pszSrc;
  1131. return nLength;
  1132. }
  1133. static int GetBaseTypeLength( LPCWSTR pszSource ) throw()
  1134. {
  1135. return ::WideCharToMultiByte( _AtlGetConversionACP(), 0, pszSource, -1, NULL, 0, NULL, NULL )-1;
  1136. }
  1137. static int GetBaseTypeLength( LPCWSTR pszSource, int nLength ) throw()
  1138. {
  1139. return
  1140. ::WideCharToMultiByte(
  1141. _AtlGetConversionACP(),
  1142. 0,
  1143. pszSource,
  1144. nLength,
  1145. NULL,
  1146. 0,
  1147. NULL,
  1148. NULL
  1149. );
  1150. }
  1151. static void
  1152. ConvertToBaseType(
  1153. LPSTR pszDest,
  1154. int nDestLength,
  1155. LPCSTR pszSrc,
  1156. int nSrcLength = -1
  1157. ) throw()
  1158. {
  1159. (void)nSrcLength;
  1160. __EXT_MFC_MEMCPY(
  1161. pszDest, 
  1162. nDestLength * sizeof(char),
  1163. pszSrc, 
  1164. nDestLength * sizeof(char) 
  1165. );
  1166. }
  1167. static void ConvertToBaseType(
  1168. LPSTR pszDest,
  1169. int nDestLength,
  1170. LPCWSTR pszSrc,
  1171. int nSrcLength = -1
  1172. ) throw()
  1173. {
  1174. ::WideCharToMultiByte(
  1175. _AtlGetConversionACP(),
  1176. 0,
  1177. pszSrc,
  1178. nSrcLength,
  1179. pszDest,
  1180. nDestLength,
  1181. NULL,
  1182. NULL
  1183. );
  1184. }
  1185. static void ConvertToOem( LPSTR psz ) throw()
  1186. {
  1187. ::AnsiToOem( psz, psz );
  1188. }
  1189. static void ConvertToAnsi( LPSTR psz ) throw()
  1190. {
  1191. ::OemToAnsi( psz, psz );
  1192. }
  1193. static void FloodCharacters( char ch, int nLength, char* pch ) throw()
  1194. {
  1195. memset( pch, ch, nLength );
  1196. }
  1197. static BSTR AllocSysString( const char* pchData, int nDataLength ) throw()
  1198. {
  1199. int nLen =
  1200. ::MultiByteToWideChar(
  1201. _AtlGetConversionACP(),
  1202. 0,
  1203. pchData,
  1204. nDataLength,
  1205. NULL,
  1206. NULL
  1207. );
  1208. BSTR bstr = ::SysAllocStringLen( NULL, nLen );
  1209. if( bstr != NULL )
  1210. {
  1211. ::MultiByteToWideChar(
  1212. _AtlGetConversionACP(),
  1213. 0,
  1214. pchData,
  1215. nDataLength,
  1216. bstr,
  1217. nLen
  1218. );
  1219. }
  1220. return bstr;
  1221. }
  1222. static BOOL ReAllocSysString( const char* pchData, BSTR* pbstr, int nDataLength ) throw()
  1223. {
  1224. int nLen =
  1225. ::MultiByteToWideChar(
  1226. _AtlGetConversionACP(),
  1227. 0,
  1228. pchData,
  1229. nDataLength,
  1230. NULL,
  1231. NULL
  1232. );
  1233. BOOL bSuccess = ::SysReAllocStringLen( pbstr, NULL, nLen );
  1234. if( bSuccess )
  1235. {
  1236. ::MultiByteToWideChar(
  1237. _AtlGetConversionACP(),
  1238. 0,
  1239. pchData,
  1240. nDataLength,
  1241. *pbstr,
  1242. nLen
  1243. );
  1244. }
  1245. return bSuccess;
  1246. }
  1247. static DWORD FormatMessage(
  1248. DWORD dwFlags,
  1249. LPCVOID pSource,
  1250. DWORD dwMessageID,
  1251. DWORD dwLanguageID,
  1252. LPSTR pszBuffer,
  1253. DWORD nSize,
  1254. va_list * pArguments
  1255. ) throw()
  1256. {
  1257. return
  1258. ::FormatMessageA(
  1259. dwFlags,
  1260. pSource,
  1261. dwMessageID,
  1262. dwLanguageID,
  1263. pszBuffer,
  1264. nSize,
  1265. pArguments
  1266. );
  1267. }
  1268. static int SafeStringLen( LPCSTR psz ) throw()
  1269. {
  1270. return (psz != NULL) ? int( strlen( psz ) ) : 0;
  1271. }
  1272. static int SafeStringLen( LPCWSTR psz ) throw()
  1273. {
  1274. return (psz != NULL) ? int( wcslen( psz ) ) : 0;
  1275. }
  1276. static int GetCharLen( const __wchar_t* pch ) throw()
  1277. {
  1278. (void)pch;
  1279. return 1;
  1280. }
  1281. static int GetCharLen( const char* pch ) throw()
  1282. {
  1283. return int(
  1284. _mbclen(
  1285. reinterpret_cast < const unsigned char * > ( pch )
  1286. )
  1287. );
  1288. }
  1289. static DWORD GetEnvironmentVariable(
  1290. LPCSTR pszVar,
  1291. LPSTR pszBuffer,
  1292. DWORD dwSize
  1293. ) throw()
  1294. {
  1295. return ::GetEnvironmentVariableA( pszVar, pszBuffer, dwSize );
  1296. }
  1297. }; // class CExtSafeStrTraitMFC
  1298. template<>
  1299. class CExtSafeStrTraitMFC<
  1300. __wchar_t,
  1301. CExtSafeChTraitsCRT < __wchar_t >
  1302. >
  1303. {
  1304. #if defined(_UNICODE) && !defined(_CSTRING_ALWAYS_THUNK)
  1305. static DWORD _GetEnvironmentVariableW(
  1306. const __wchar_t * pszName,
  1307. __wchar_t * pszBuffer,
  1308. DWORD nSize
  1309. ) throw()
  1310. {
  1311. return
  1312. ::GetEnvironmentVariableW(
  1313. (LPCWSTR)pszName,
  1314. (LPWSTR)pszBuffer,
  1315. nSize
  1316. );
  1317. }
  1318. #else
  1319. static DWORD WINAPI _GetEnvironmentVariableW(
  1320. const __wchar_t * pszName, 
  1321. __wchar_t * pszBuffer,
  1322. DWORD nSize
  1323. ) throw()
  1324. {
  1325. return
  1326. _strthunks.pfnGetEnvironmentVariableW(
  1327. (LPCWSTR)pszName,
  1328. (LPWSTR)pszBuffer,
  1329. nSize
  1330. );
  1331. }
  1332. #endif
  1333. public:
  1334. static HINSTANCE FindStringResourceInstance( UINT nID ) throw()
  1335. {
  1336. return ( AfxFindStringResourceHandle( nID ) );
  1337. }
  1338. static ATL::IAtlStringMgr * GetDefaultManager() throw()
  1339. {
  1340. return ( AfxGetStringManager() );
  1341. }
  1342. static __wchar_t * CharNext( const __wchar_t * psz ) throw()
  1343. {
  1344. return const_cast < __wchar_t * > ( psz+1 );
  1345. }
  1346. static int IsDigit( __wchar_t ch ) throw()
  1347. {
  1348. return iswdigit( ch );
  1349. }
  1350. static int IsSpace( __wchar_t ch ) throw()
  1351. {
  1352. return iswspace( ch );
  1353. }
  1354. static int StringCompare( const __wchar_t * pszA, const __wchar_t * pszB ) throw()
  1355. {
  1356. return wcscmp( (wchar_t *)pszA, (wchar_t *)pszB );
  1357. }
  1358. static int StringCompareIgnore( const __wchar_t * pszA, const __wchar_t * pszB ) throw()
  1359. {
  1360. return _wcsicmp( (wchar_t *)pszA, (wchar_t *)pszB );
  1361. }
  1362. static int StringCollate( const __wchar_t * pszA, const __wchar_t * pszB ) throw()
  1363. {
  1364. return wcscoll( (wchar_t *)pszA, (wchar_t *)pszB );
  1365. }
  1366. static int StringCollateIgnore( const __wchar_t * pszA, const __wchar_t * pszB ) throw()
  1367. {
  1368. return _wcsicoll( (wchar_t *)pszA, (wchar_t *)pszB );
  1369. }
  1370. static const __wchar_t * StringFindString( const __wchar_t * pszBlock, const __wchar_t * pszMatch ) throw()
  1371. {
  1372. return (__wchar_t *)wcsstr( (wchar_t *)pszBlock, (wchar_t *)pszMatch );
  1373. }
  1374. static __wchar_t * StringFindString( __wchar_t * pszBlock, const __wchar_t * pszMatch ) throw()
  1375. {
  1376. return ( const_cast < __wchar_t * > ( StringFindString( const_cast < const __wchar_t * > ( pszBlock ), pszMatch ) ) );
  1377. }
  1378. static const __wchar_t * StringFindChar( const __wchar_t * pszBlock, __wchar_t chMatch ) throw()
  1379. {
  1380. return (__wchar_t *)wcschr( (wchar_t *)pszBlock, (wchar_t)chMatch );
  1381. }
  1382. static const __wchar_t * StringFindCharRev( const __wchar_t * psz, __wchar_t ch ) throw()
  1383. {
  1384. return (__wchar_t *)wcsrchr( (wchar_t *)psz, (wchar_t)ch );
  1385. }
  1386. static const __wchar_t * StringScanSet( const __wchar_t * pszBlock, const __wchar_t * pszMatch ) throw()
  1387. {
  1388. return (__wchar_t *)wcspbrk( (wchar_t *)pszBlock, (wchar_t *)pszMatch );
  1389. }
  1390. static int StringSpanIncluding( const __wchar_t * pszBlock, const __wchar_t * pszSet ) throw()
  1391. {
  1392. return (int)wcsspn( (wchar_t *)pszBlock, (wchar_t *)pszSet );
  1393. }
  1394. static int StringSpanExcluding( const __wchar_t * pszBlock, const __wchar_t * pszSet ) throw()
  1395. {
  1396. return (int)wcscspn( (wchar_t *)pszBlock, (wchar_t *)pszSet );
  1397. }
  1398. static __wchar_t * StringUppercase( __wchar_t * psz ) throw()
  1399. {
  1400. return (__wchar_t *)_wcsupr( (wchar_t *)psz );
  1401. }
  1402. static __wchar_t * StringLowercase( __wchar_t * psz ) throw()
  1403. {
  1404. return (__wchar_t *)_wcslwr( (wchar_t *)psz );
  1405. }
  1406. static __wchar_t * StringReverse( __wchar_t * psz ) throw()
  1407. {
  1408. return (__wchar_t *)_wcsrev( (wchar_t *)psz );
  1409. }
  1410. static int GetFormattedLength( const __wchar_t * pszFormat, va_list args) throw()
  1411. {
  1412. return _vscwprintf( (wchar_t *)pszFormat, args );
  1413. }
  1414. static int Format( __wchar_t * pszBuffer, const __wchar_t * pszFormat, va_list args) throw()
  1415. {
  1416. return vswprintf( (wchar_t *)pszBuffer, (wchar_t *)pszFormat, args );
  1417. }
  1418. static int GetBaseTypeLength( LPCSTR pszSrc ) throw()
  1419. {
  1420. return
  1421. ::MultiByteToWideChar(
  1422. _AtlGetConversionACP(),
  1423. 0,
  1424. pszSrc,
  1425. -1,
  1426. NULL,
  1427. 0
  1428. ) - 1;
  1429. }
  1430. static int GetBaseTypeLength( LPCSTR pszSrc, int nLength ) throw()
  1431. {
  1432. return
  1433. ::MultiByteToWideChar(
  1434. _AtlGetConversionACP(),
  1435. 0,
  1436. pszSrc,
  1437. nLength,
  1438. NULL,
  1439. 0
  1440. );
  1441. }
  1442. static int GetBaseTypeLength( const __wchar_t * pszSrc ) throw()
  1443. {
  1444. return (int)wcslen( (wchar_t *)pszSrc );
  1445. }
  1446. static int GetBaseTypeLength( const __wchar_t * pszSrc, int nLength ) throw()
  1447. {
  1448. (void)pszSrc;
  1449. return nLength;
  1450. }
  1451. static void ConvertToBaseType(
  1452. __wchar_t * pszDest,
  1453. int nDestLength,
  1454. LPCSTR pszSrc,
  1455. int nSrcLength = -1
  1456. ) throw()
  1457. {
  1458. ::MultiByteToWideChar(
  1459. _AtlGetConversionACP(),
  1460. 0,
  1461. pszSrc,
  1462. nSrcLength,
  1463. (LPWSTR)pszDest,
  1464. nDestLength
  1465. );
  1466. }
  1467. static void ConvertToBaseType(
  1468. __wchar_t * pszDest,
  1469. int nDestLength,
  1470. const __wchar_t * pszSrc,
  1471. int nSrcLength = -1
  1472. ) throw()
  1473. {
  1474. (void)nSrcLength;
  1475. __EXT_MFC_MEMCPY(
  1476. pszDest, 
  1477. nDestLength * sizeof(__wchar_t),
  1478. pszSrc, 
  1479. nDestLength * sizeof(__wchar_t) 
  1480. );
  1481. }
  1482. static void FloodCharacters( __wchar_t ch, int nLength, __wchar_t * psz ) throw()
  1483. {
  1484. for( int i = 0; i < nLength; i++ )
  1485. {
  1486. psz[i] = ch;
  1487. }
  1488. }
  1489. static BSTR AllocSysString( const __wchar_t * pchData, int nDataLength ) throw()
  1490. {
  1491. return ::SysAllocStringLen( (const OLECHAR *)pchData, nDataLength );
  1492. }
  1493. static BOOL ReAllocSysString( const __wchar_t * pchData, BSTR* pbstr, int nDataLength ) throw()
  1494. {
  1495. return ::SysReAllocStringLen( pbstr, (const OLECHAR *)pchData, nDataLength );
  1496. }
  1497. #ifdef _UNICODE
  1498. static DWORD FormatMessage(
  1499. DWORD dwFlags,
  1500. LPCVOID pSource,
  1501. DWORD dwMessageID,
  1502. DWORD dwLanguageID,
  1503. __wchar_t * pszBuffer,
  1504. DWORD nSize,
  1505. va_list * pArguments
  1506. ) throw()
  1507. {
  1508. return
  1509. ::FormatMessageW(
  1510. dwFlags,
  1511. pSource,
  1512. dwMessageID,
  1513. dwLanguageID,
  1514. (LPWSTR)pszBuffer,
  1515. nSize,
  1516. pArguments
  1517. );
  1518. }
  1519. #endif
  1520. static int SafeStringLen( LPCSTR psz ) throw()
  1521. {
  1522. return (psz != NULL) ? (int)strlen( psz ) : 0;
  1523. }
  1524. static int SafeStringLen( const __wchar_t * psz ) throw()
  1525. {
  1526. return (psz != NULL) ? (int)wcslen((wchar_t *) psz ) : 0;
  1527. }
  1528. static int GetCharLen( const __wchar_t* pch ) throw()
  1529. {
  1530. (void)pch;
  1531. return 1;
  1532. }
  1533. static int GetCharLen( const char* pch ) throw()
  1534. {
  1535. return (int)(
  1536. _mbclen(
  1537. reinterpret_cast < const unsigned char * > ( pch )
  1538. )
  1539. );
  1540. }
  1541. static DWORD GetEnvironmentVariable(
  1542. const __wchar_t * pszVar,
  1543. __wchar_t * pszBuffer,
  1544. DWORD dwSize
  1545. ) throw()
  1546. {
  1547. return _GetEnvironmentVariableW( pszVar, pszBuffer, dwSize );
  1548. }
  1549. }; // class CExtSafeStrTraitMFC
  1550. template<>
  1551. class CExtSafeStrTraitMFC<
  1552. unsigned short,
  1553. CExtSafeChTraitsCRT < unsigned short >
  1554. >
  1555. {
  1556. #if defined(_UNICODE) && !defined(_CSTRING_ALWAYS_THUNK)
  1557. static DWORD _GetEnvironmentVariableW(
  1558. const unsigned short * pszName,
  1559. unsigned short * pszBuffer,
  1560. DWORD nSize
  1561. ) throw()
  1562. {
  1563. return
  1564. ::GetEnvironmentVariableW(
  1565. (LPCWSTR)pszName,
  1566. (LPWSTR)pszBuffer,
  1567. nSize
  1568. );
  1569. }
  1570. #else
  1571. static DWORD WINAPI _GetEnvironmentVariableW(
  1572. const unsigned short * pszName, 
  1573. unsigned short * pszBuffer,
  1574. DWORD nSize
  1575. ) throw()
  1576. {
  1577. return
  1578. _strthunks.pfnGetEnvironmentVariableW(
  1579. (LPCWSTR)pszName,
  1580. (LPWSTR)pszBuffer,
  1581. nSize
  1582. );
  1583. }
  1584. #endif
  1585. public:
  1586. static HINSTANCE FindStringResourceInstance( UINT nID ) throw()
  1587. {
  1588. return ( AfxFindStringResourceHandle( nID ) );
  1589. }
  1590. static ATL::IAtlStringMgr * GetDefaultManager() throw()
  1591. {
  1592. return ( AfxGetStringManager() );
  1593. }
  1594. static unsigned short * CharNext( const unsigned short * psz ) throw()
  1595. {
  1596. return const_cast < unsigned short * > ( psz+1 );
  1597. }
  1598. static int IsDigit( unsigned short ch ) throw()
  1599. {
  1600. return iswdigit( ch );
  1601. }
  1602. static int IsSpace( unsigned short ch ) throw()
  1603. {
  1604. return iswspace( ch );
  1605. }
  1606. static int StringCompare( const unsigned short * pszA, const unsigned short * pszB ) throw()
  1607. {
  1608. return wcscmp( (wchar_t *)pszA, (wchar_t *)pszB );
  1609. }
  1610. static int StringCompareIgnore( const unsigned short * pszA, const unsigned short * pszB ) throw()
  1611. {
  1612. return _wcsicmp( (wchar_t *)pszA, (wchar_t *)pszB );
  1613. }
  1614. static int StringCollate( const unsigned short * pszA, const unsigned short * pszB ) throw()
  1615. {
  1616. return wcscoll( (wchar_t *)pszA, (wchar_t *)pszB );
  1617. }
  1618. static int StringCollateIgnore( const unsigned short * pszA, const unsigned short * pszB ) throw()
  1619. {
  1620. return _wcsicoll( (wchar_t *)pszA, (wchar_t *)pszB );
  1621. }
  1622. static const unsigned short * StringFindString( const unsigned short * pszBlock, const unsigned short * pszMatch ) throw()
  1623. {
  1624. return (unsigned short *)wcsstr( (wchar_t *)pszBlock, (wchar_t *)pszMatch );
  1625. }
  1626. static unsigned short * StringFindString( unsigned short * pszBlock, const unsigned short * pszMatch ) throw()
  1627. {
  1628. return ( const_cast < unsigned short * > ( StringFindString( const_cast < const unsigned short * > ( pszBlock ), pszMatch ) ) );
  1629. }
  1630. static const unsigned short * StringFindChar( const unsigned short * pszBlock, unsigned short chMatch ) throw()
  1631. {
  1632. return (unsigned short *)wcschr( (wchar_t *)pszBlock, (wchar_t)chMatch );
  1633. }
  1634. static const unsigned short * StringFindCharRev( const unsigned short * psz, unsigned short ch ) throw()
  1635. {
  1636. return (unsigned short *)wcsrchr( (wchar_t *)psz, (wchar_t)ch );
  1637. }
  1638. static const unsigned short * StringScanSet( const unsigned short * pszBlock, const unsigned short * pszMatch ) throw()
  1639. {
  1640. return (unsigned short *)wcspbrk( (wchar_t *)pszBlock, (wchar_t *)pszMatch );
  1641. }
  1642. static int StringSpanIncluding( const unsigned short * pszBlock, const unsigned short * pszSet ) throw()
  1643. {
  1644. return (int)wcsspn( (wchar_t *)pszBlock, (wchar_t *)pszSet );
  1645. }
  1646. static int StringSpanExcluding( const unsigned short * pszBlock, const unsigned short * pszSet ) throw()
  1647. {
  1648. return (int)wcscspn( (wchar_t *)pszBlock, (wchar_t *)pszSet );
  1649. }
  1650. static unsigned short * StringUppercase( unsigned short * psz ) throw()
  1651. {
  1652. return (unsigned short *)_wcsupr( (wchar_t *)psz );
  1653. }
  1654. static unsigned short * StringLowercase( unsigned short * psz ) throw()
  1655. {
  1656. return (unsigned short *)_wcslwr( (wchar_t *)psz );
  1657. }
  1658. static unsigned short * StringReverse( unsigned short * psz ) throw()
  1659. {
  1660. return (unsigned short *)_wcsrev( (wchar_t *)psz );
  1661. }
  1662. static int GetFormattedLength( const unsigned short * pszFormat, va_list args) throw()
  1663. {
  1664. return _vscwprintf( (wchar_t *)pszFormat, args );
  1665. }
  1666. static int Format( unsigned short * pszBuffer, const unsigned short * pszFormat, va_list args) throw()
  1667. {
  1668. return vswprintf( (wchar_t *)pszBuffer, (wchar_t *)pszFormat, args );
  1669. }
  1670. static int GetBaseTypeLength( LPCSTR pszSrc ) throw()
  1671. {
  1672. return
  1673. ::MultiByteToWideChar(
  1674. _AtlGetConversionACP(),
  1675. 0,
  1676. pszSrc,
  1677. -1,
  1678. NULL,
  1679. 0
  1680. ) - 1;
  1681. }
  1682. static int GetBaseTypeLength( LPCSTR pszSrc, int nLength ) throw()
  1683. {
  1684. return
  1685. ::MultiByteToWideChar(
  1686. _AtlGetConversionACP(),
  1687. 0,
  1688. pszSrc,
  1689. nLength,
  1690. NULL,
  1691. 0
  1692. );
  1693. }
  1694. static int GetBaseTypeLength( const unsigned short * pszSrc ) throw()
  1695. {
  1696. return (int)wcslen( (wchar_t *)pszSrc );
  1697. }
  1698. static int GetBaseTypeLength( const unsigned short * pszSrc, int nLength ) throw()
  1699. {
  1700. (void)pszSrc;
  1701. return nLength;
  1702. }
  1703. static void ConvertToBaseType(
  1704. unsigned short * pszDest,
  1705. int nDestLength,
  1706. LPCSTR pszSrc,
  1707. int nSrcLength = -1
  1708. ) throw()
  1709. {
  1710. ::MultiByteToWideChar(
  1711. _AtlGetConversionACP(),
  1712. 0,
  1713. pszSrc,
  1714. nSrcLength,
  1715. (LPWSTR)pszDest,
  1716. nDestLength
  1717. );
  1718. }
  1719. static void ConvertToBaseType(
  1720. unsigned short * pszDest,
  1721. int nDestLength,
  1722. const unsigned short * pszSrc,
  1723. int nSrcLength = -1
  1724. ) throw()
  1725. {
  1726. (void)nSrcLength;
  1727. __EXT_MFC_MEMCPY(
  1728. pszDest, 
  1729. nDestLength * sizeof(unsigned short),
  1730. pszSrc, 
  1731. nDestLength * sizeof(unsigned short) 
  1732. );
  1733. }
  1734. static void FloodCharacters( unsigned short ch, int nLength, unsigned short * psz ) throw()
  1735. {
  1736. for( int i = 0; i < nLength; i++ )
  1737. {
  1738. psz[i] = ch;
  1739. }
  1740. }
  1741. static BSTR AllocSysString( const unsigned short * pchData, int nDataLength ) throw()
  1742. {
  1743. return ::SysAllocStringLen( (const OLECHAR *)pchData, nDataLength );
  1744. }
  1745. static BOOL ReAllocSysString( const unsigned short * pchData, BSTR * pbstr, int nDataLength ) throw()
  1746. {
  1747. return ::SysReAllocStringLen( pbstr, (const OLECHAR *)pchData, nDataLength );
  1748. }
  1749. #ifdef _UNICODE
  1750. static DWORD FormatMessage(
  1751. DWORD dwFlags,
  1752. LPCVOID pSource,
  1753. DWORD dwMessageID,
  1754. DWORD dwLanguageID,
  1755. unsigned short * pszBuffer,
  1756. DWORD nSize,
  1757. va_list * pArguments
  1758. ) throw()
  1759. {
  1760. return
  1761. ::FormatMessageW(
  1762. dwFlags,
  1763. pSource,
  1764. dwMessageID,
  1765. dwLanguageID,
  1766. (LPWSTR)pszBuffer,
  1767. nSize,
  1768. pArguments
  1769. );
  1770. }
  1771. #endif
  1772. static int SafeStringLen( LPCSTR psz ) throw()
  1773. {
  1774. return (psz != NULL) ? (int)strlen( psz ) : 0;
  1775. }
  1776. static int SafeStringLen( const unsigned short * psz ) throw()
  1777. {
  1778. return (psz != NULL) ? (int)wcslen( (wchar_t *)psz ) : 0;
  1779. }
  1780. static int GetCharLen( const unsigned short* pch ) throw()
  1781. {
  1782. (void)pch;
  1783. return 1;
  1784. }
  1785. static int GetCharLen( const char* pch ) throw()
  1786. {
  1787. return (int)(
  1788. _mbclen(
  1789. reinterpret_cast < const unsigned char * > ( pch )
  1790. )
  1791. );
  1792. }
  1793. static DWORD GetEnvironmentVariable(
  1794. const unsigned short * pszVar,
  1795. unsigned short * pszBuffer,
  1796. DWORD dwSize
  1797. ) throw()
  1798. {
  1799. return _GetEnvironmentVariableW( pszVar, pszBuffer, dwSize );
  1800. }
  1801. }; // class CExtSafeStrTraitMFC
  1802. template < typename BaseType >
  1803. class CExtSafeSimpleStringT
  1804. {
  1805. public:
  1806. typedef typename CExtSafeChTraitsBase< BaseType >::XCHAR XCHAR;
  1807. typedef typename CExtSafeChTraitsBase< BaseType >::PXSTR PXSTR;
  1808. typedef typename CExtSafeChTraitsBase< BaseType >::PCXSTR PCXSTR;
  1809. typedef typename CExtSafeChTraitsBase< BaseType >::YCHAR YCHAR;
  1810. typedef typename CExtSafeChTraitsBase< BaseType >::PYSTR PYSTR;
  1811. typedef typename CExtSafeChTraitsBase< BaseType >::PCYSTR PCYSTR;
  1812. public:
  1813. explicit CExtSafeSimpleStringT( IAtlStringMgr * pStringMgr ) throw()
  1814. {
  1815. ATLASSERT( pStringMgr != NULL );
  1816. CStringData * pData = pStringMgr->GetNilString();
  1817. Attach( pData );
  1818. }
  1819. CExtSafeSimpleStringT( const CExtSafeSimpleStringT & strSrc )
  1820. {
  1821. CStringData * pSrcData = strSrc.GetData();
  1822. CStringData * pNewData = CloneData( pSrcData );
  1823. Attach( pNewData );
  1824. }
  1825. CExtSafeSimpleStringT( PCXSTR pszSrc, IAtlStringMgr * pStringMgr )
  1826. {
  1827. ATLASSERT( pStringMgr != NULL );
  1828. int nLength = StringLength( pszSrc );
  1829. CStringData * pData = pStringMgr->Allocate( nLength, sizeof(XCHAR) );
  1830. if( pData == NULL )
  1831. {
  1832. ThrowMemoryException();
  1833. }
  1834. Attach( pData );
  1835. SetLength( nLength );
  1836. CopyChars( m_pszData, pszSrc, nLength );
  1837. }
  1838. CExtSafeSimpleStringT( const XCHAR * pchSrc, int nLength, IAtlStringMgr * pStringMgr )
  1839. {
  1840. ATLASSERT( pStringMgr != NULL );
  1841. CStringData * pData = pStringMgr->Allocate( nLength, sizeof(XCHAR) );
  1842. if( pData == NULL )
  1843. {
  1844. ThrowMemoryException();
  1845. }
  1846. Attach( pData );
  1847. SetLength( nLength );
  1848. CopyChars( m_pszData, pchSrc, nLength );
  1849. }
  1850. ~CExtSafeSimpleStringT() throw()
  1851. {
  1852. CStringData * pData = GetData();
  1853. pData->Release();
  1854. }
  1855. CExtSafeSimpleStringT & operator = ( const CExtSafeSimpleStringT & strSrc )
  1856. {
  1857. CStringData * pSrcData = strSrc.GetData();
  1858. CStringData * pOldData = GetData();
  1859. if( pSrcData != pOldData )
  1860. {
  1861. if( pOldData->IsLocked() )
  1862. {
  1863. SetString( strSrc.GetString(), strSrc.GetLength() );
  1864. }
  1865. else
  1866. {
  1867. CStringData * pNewData = CloneData( pSrcData );
  1868. pOldData->Release();
  1869. Attach( pNewData );
  1870. }
  1871. }
  1872. return ( *this );
  1873. }
  1874. CExtSafeSimpleStringT & operator = ( PCXSTR pszSrc )
  1875. {
  1876. SetString( pszSrc );
  1877. return ( *this );
  1878. }
  1879. CExtSafeSimpleStringT & operator += ( const CExtSafeSimpleStringT & strSrc )
  1880. {
  1881. Append( strSrc );
  1882. return ( *this );
  1883. }
  1884. CExtSafeSimpleStringT & operator += ( PCXSTR pszSrc )
  1885. {
  1886. Append( pszSrc );
  1887. return ( *this );
  1888. }
  1889. template < int t_nSize >
  1890. CExtSafeSimpleStringT & operator += ( const CStaticString < XCHAR, t_nSize > & strSrc )
  1891. {
  1892. Append( strSrc.m_psz, strSrc.GetLength() );
  1893. return ( *this );
  1894. }
  1895. CExtSafeSimpleStringT & operator += ( char ch )
  1896. {
  1897. AppendChar( XCHAR( ch ) );
  1898. return ( *this );
  1899. }
  1900. CExtSafeSimpleStringT & operator += ( unsigned char ch )
  1901. {
  1902. AppendChar( XCHAR( ch ) );
  1903. return ( *this );
  1904. }
  1905. CExtSafeSimpleStringT & operator += ( wchar_t ch )
  1906. {
  1907. AppendChar( XCHAR( ch ) );
  1908. return ( *this );
  1909. }
  1910. XCHAR operator [] ( int iChar ) const throw()
  1911. {
  1912. ATLASSERT( (iChar >= 0) && (iChar <= GetLength()) );
  1913. return ( m_pszData[iChar] );
  1914. }
  1915. operator PCXSTR() const throw()
  1916. {
  1917. return ( m_pszData );
  1918. }
  1919. void Append( PCXSTR pszSrc )
  1920. {
  1921. Append( pszSrc, StringLength( pszSrc ) );
  1922. }
  1923. void Append( PCXSTR pszSrc, int nLength )
  1924. {
  1925. UINT_PTR nOffset = pszSrc-GetString();
  1926. UINT nOldLength = GetLength();
  1927. int nNewLength = nOldLength+nLength;
  1928. PXSTR pszBuffer = GetBuffer( nNewLength );
  1929. if( nOffset <= nOldLength )
  1930. {
  1931. pszSrc = pszBuffer+nOffset;
  1932. }
  1933. CopyChars( pszBuffer+nOldLength, pszSrc, nLength );
  1934. ReleaseBufferSetLength( nNewLength );
  1935. }
  1936. void AppendChar( XCHAR ch )
  1937. {
  1938. UINT nOldLength = GetLength();
  1939. int nNewLength = nOldLength+1;
  1940. PXSTR pszBuffer = GetBuffer( nNewLength );
  1941. pszBuffer[nOldLength] = ch;
  1942. ReleaseBufferSetLength( nNewLength );
  1943. }
  1944. void Append( const CExtSafeSimpleStringT & strSrc )
  1945. {
  1946. Append( strSrc.GetString(), strSrc.GetLength() );
  1947. }
  1948. void Empty() throw()
  1949. {
  1950. CStringData * pOldData = GetData();
  1951. IAtlStringMgr * pStringMgr = pOldData->pStringMgr;
  1952. if( pOldData->nDataLength == 0 )
  1953. {
  1954. return;
  1955. }
  1956. if( pOldData->IsLocked() )
  1957. {
  1958. SetLength( 0 );
  1959. }
  1960. else
  1961. {
  1962. pOldData->Release();
  1963. CStringData * pNewData = pStringMgr->GetNilString();
  1964. Attach( pNewData );
  1965. }
  1966. }
  1967. void FreeExtra() throw()
  1968. {
  1969. CStringData * pOldData = GetData();
  1970. int nLength = pOldData->nDataLength;
  1971. IAtlStringMgr * pStringMgr = pOldData->pStringMgr;
  1972. if( pOldData->nAllocLength == nLength )
  1973. {
  1974. return;
  1975. }
  1976. if( !pOldData->IsLocked() )
  1977. {
  1978. CStringData * pNewData = pStringMgr->Allocate( nLength, sizeof(XCHAR) );
  1979. if( pNewData == NULL )
  1980. {
  1981. SetLength( nLength );
  1982. return;
  1983. }
  1984. CopyChars( PXSTR( pNewData->data() ), PCXSTR( pOldData->data() ), nLength );
  1985. pOldData->Release();
  1986. Attach( pNewData );
  1987. SetLength( nLength );
  1988. }
  1989. }
  1990. int GetAllocLength() const throw()
  1991. {
  1992. return ( GetData()->nAllocLength );
  1993. }
  1994. XCHAR GetAt( int iChar ) const throw()
  1995. {
  1996. ATLASSERT( (iChar >= 0) && (iChar <= GetLength()) );
  1997. return ( m_pszData[iChar] );
  1998. }
  1999. PXSTR GetBuffer()
  2000. {
  2001. CStringData * pData = GetData();
  2002. if( pData->IsShared() )
  2003. {
  2004. Fork( pData->nDataLength );
  2005. }
  2006. return ( m_pszData );
  2007. }
  2008. PXSTR GetBuffer( int nMinBufferLength )
  2009. {
  2010. return ( PrepareWrite( nMinBufferLength ) );
  2011. }
  2012. PXSTR GetBufferSetLength( int nLength )
  2013. {
  2014. PXSTR pszBuffer = GetBuffer( nLength );
  2015. SetLength( nLength );
  2016. return ( pszBuffer );
  2017. }
  2018. int GetLength() const throw()
  2019. {
  2020. return ( GetData()->nDataLength );
  2021. }
  2022. IAtlStringMgr * GetManager() const throw()
  2023. {
  2024. return ( GetData()->pStringMgr->Clone() );
  2025. }
  2026. PCXSTR GetString() const throw()
  2027. {
  2028. return ( m_pszData );
  2029. }
  2030. bool IsEmpty() const throw()
  2031. {
  2032. return ( GetLength() == 0 );
  2033. }
  2034. PXSTR LockBuffer()
  2035. {
  2036. CStringData * pData = GetData();
  2037. if( pData->IsShared() )
  2038. {
  2039. Fork( pData->nDataLength );
  2040. pData = GetData();
  2041. }
  2042. pData->Lock();
  2043. return ( m_pszData );
  2044. }
  2045. void UnlockBuffer() throw()
  2046. {
  2047. CStringData * pData = GetData();
  2048. pData->Unlock();
  2049. }
  2050. void Preallocate( int nLength )
  2051. {
  2052. PrepareWrite( nLength );
  2053. }
  2054. void ReleaseBuffer( int nNewLength = -1 ) throw()
  2055. {
  2056. if( nNewLength == -1 )
  2057. {
  2058. nNewLength = StringLength( m_pszData );
  2059. }
  2060. SetLength( nNewLength );
  2061. }
  2062. void ReleaseBufferSetLength( int nNewLength ) throw()
  2063. {
  2064. ATLASSERT( nNewLength >= 0 );
  2065. SetLength( nNewLength );
  2066. }
  2067. void Truncate( int nNewLength )
  2068. {
  2069. ATLASSERT( nNewLength <= GetLength() );
  2070. GetBuffer( nNewLength );
  2071. ReleaseBufferSetLength( nNewLength );
  2072. }
  2073. void SetAt( int iChar, XCHAR ch )
  2074. {
  2075. ATLASSERT( (iChar >= 0) && (iChar < GetLength()) );
  2076. int nLength = GetLength();
  2077. PXSTR pszBuffer = GetBuffer();
  2078. pszBuffer[iChar] = ch;
  2079. ReleaseBufferSetLength( nLength );
  2080. }
  2081. void SetManager( IAtlStringMgr * pStringMgr )
  2082. {
  2083. ATLASSERT( IsEmpty() );
  2084. CStringData * pData = GetData();
  2085. pData->Release();
  2086. pData = pStringMgr->GetNilString();
  2087. Attach( pData );
  2088. }
  2089. void SetString( PCXSTR pszSrc )
  2090. {
  2091. SetString( pszSrc, StringLength( pszSrc ) );
  2092. }
  2093. void SetString( PCXSTR pszSrc, int nLength )
  2094. {
  2095. if( nLength == 0 )
  2096. {
  2097. Empty();
  2098. }
  2099. else
  2100. {
  2101. UINT nOldLength = GetLength();
  2102. UINT_PTR nOffset = pszSrc-GetString();
  2103. PXSTR pszBuffer = GetBuffer( nLength );
  2104. if( nOffset <= nOldLength )
  2105. {
  2106. CopyCharsOverlapped( pszBuffer, pszBuffer+nOffset, nLength );
  2107. }
  2108. else
  2109. {
  2110. CopyChars( pszBuffer, pszSrc, nLength );
  2111. }
  2112. ReleaseBufferSetLength( nLength );
  2113. }
  2114. }
  2115. public:
  2116. friend CExtSafeSimpleStringT operator + (
  2117. const CExtSafeSimpleStringT & str1,
  2118. const CExtSafeSimpleStringT & str2
  2119. )
  2120. {
  2121. CExtSafeSimpleStringT s( str1.GetManager() );
  2122. Concatenate( s, str1, str1.GetLength(), str2, str2.GetLength() );
  2123. return ( s );
  2124. }
  2125. friend CExtSafeSimpleStringT operator + (
  2126. const CExtSafeSimpleStringT & str1,
  2127. PCXSTR psz2
  2128. )
  2129. {
  2130. CExtSafeSimpleStringT s( str1.GetManager() );
  2131. Concatenate( s, str1, str1.GetLength(), psz2, StringLength( psz2 ) );
  2132. return ( s );
  2133. }
  2134. friend CExtSafeSimpleStringT operator + (
  2135. PCXSTR psz1,
  2136. const CExtSafeSimpleStringT & str2
  2137. )
  2138. {
  2139. CExtSafeSimpleStringT s( str2.GetManager() );
  2140. Concatenate( s, psz1, StringLength( psz1 ), str2, str2.GetLength() );
  2141. return ( s );
  2142. }
  2143. static void CopyChars( XCHAR * pchDest, const XCHAR * pchSrc, int nChars ) throw()
  2144. {
  2145. __EXT_MFC_MEMCPY(
  2146. pchDest, 
  2147. nChars * sizeof(XCHAR),
  2148. pchSrc, 
  2149. nChars * sizeof(XCHAR) 
  2150. );
  2151. }
  2152. static void CopyCharsOverlapped( XCHAR * pchDest, const XCHAR * pchSrc, int nChars ) throw()
  2153. {
  2154. UINT nSizeInBytes = 
  2155. nChars * sizeof(XCHAR);
  2156. __EXT_MFC_MEMMOVE(
  2157. pchDest, 
  2158. nSizeInBytes,
  2159. pchSrc, 
  2160. nSizeInBytes 
  2161. );
  2162. }
  2163. #ifdef _ATL_MIN_CRT
  2164. ATL_NOINLINE static int StringLength( PCXSTR psz ) throw()
  2165. {
  2166. int nLength = 0;
  2167. if( psz != NULL )
  2168. {
  2169. const XCHAR * pch = psz;
  2170. while( *pch != 0 )
  2171. {
  2172. nLength++;
  2173. pch++;
  2174. }
  2175. }
  2176. return ( nLength );
  2177. }
  2178. #else
  2179. static int StringLength( const char* psz ) throw()
  2180. {
  2181. if( psz == NULL )
  2182. {
  2183. return ( 0 );
  2184. }
  2185. return ( int( strlen( psz ) ) );
  2186. }
  2187. static int StringLength( const __wchar_t * psz ) throw()
  2188. {
  2189. if( psz == NULL )
  2190. {
  2191. return ( 0 );
  2192. }
  2193. return ( int( wcslen( (wchar_t *)psz ) ) );
  2194. }
  2195. static int StringLength( const unsigned short * psz ) throw()
  2196. {
  2197. if( psz == NULL )
  2198. {
  2199. return ( 0 );
  2200. }
  2201. return ( int( wcslen( (wchar_t *)psz ) ) );
  2202. }
  2203. #endif
  2204. protected:
  2205. static void Concatenate( CExtSafeSimpleStringT & strResult, PCXSTR psz1, int nLength1, PCXSTR psz2, int nLength2 )
  2206. {
  2207. int nNewLength = nLength1+nLength2;
  2208. PXSTR pszBuffer = strResult.GetBuffer( nNewLength );
  2209. CopyChars( pszBuffer, psz1, nLength1 );
  2210. CopyChars( pszBuffer+nLength1, psz2, nLength2 );
  2211. strResult.ReleaseBufferSetLength( nNewLength );
  2212. }
  2213. ATL_NOINLINE __declspec( noreturn ) static void ThrowMemoryException()
  2214. {
  2215. AtlThrow( E_OUTOFMEMORY );
  2216. }
  2217. private:
  2218. void Attach( CStringData * pData ) throw()
  2219. {
  2220. m_pszData = static_cast < PXSTR > ( pData->data() );
  2221. }
  2222. ATL_NOINLINE void Fork( int nLength )
  2223. {
  2224. CStringData * pOldData = GetData();
  2225. int nOldLength = pOldData->nDataLength;
  2226. CStringData * pNewData = pOldData->pStringMgr->Clone()->Allocate( nLength, sizeof(XCHAR) );
  2227. if( pNewData == NULL )
  2228. {
  2229. ThrowMemoryException();
  2230. }
  2231. int nCharsToCopy = ((nOldLength < nLength) ? nOldLength : nLength)+1;
  2232. CopyChars( PXSTR( pNewData->data() ), PCXSTR( pOldData->data() ), nCharsToCopy );
  2233. pNewData->nDataLength = nOldLength;
  2234. pOldData->Release();
  2235. Attach( pNewData );
  2236. }
  2237. CStringData * GetData() const throw()
  2238. {
  2239. return ( reinterpret_cast < CStringData * > ( m_pszData ) - 1 );
  2240. }
  2241. PXSTR PrepareWrite( int nLength )
  2242. {
  2243. CStringData * pOldData = GetData();
  2244. int nShared = 1-pOldData->nRefs;
  2245. int nTooShort = pOldData->nAllocLength-nLength;
  2246. if( (nShared|nTooShort) < 0 )
  2247. {
  2248. PrepareWrite2( nLength );
  2249. }
  2250. return ( m_pszData );
  2251. }
  2252. ATL_NOINLINE void PrepareWrite2( int nLength )
  2253. {
  2254. CStringData * pOldData = GetData();
  2255. if( pOldData->nDataLength > nLength )
  2256. {
  2257. nLength = pOldData->nDataLength;
  2258. }
  2259. if( pOldData->IsShared() )
  2260. {
  2261. Fork( nLength );
  2262. }
  2263. else if( pOldData->nAllocLength < nLength )
  2264. {
  2265. int nNewLength = pOldData->nAllocLength;
  2266. if( nNewLength > 1024 )
  2267. {
  2268. nNewLength += 1024;
  2269. }
  2270. else
  2271. {
  2272. nNewLength *= 2;
  2273. }
  2274. if( nNewLength < nLength )
  2275. {
  2276. nNewLength = nLength;
  2277. }
  2278. Reallocate( nNewLength );
  2279. }
  2280. }
  2281. ATL_NOINLINE void Reallocate( int nLength )
  2282. {
  2283. CStringData * pOldData = GetData();
  2284. ATLASSERT( pOldData->nAllocLength < nLength );
  2285. IAtlStringMgr * pStringMgr = pOldData->pStringMgr;
  2286. CStringData * pNewData = pStringMgr->Reallocate( pOldData, nLength, sizeof(XCHAR) );
  2287. if( pNewData == NULL )
  2288. {
  2289. ThrowMemoryException();
  2290. }
  2291. Attach( pNewData );
  2292. }
  2293. void SetLength( int nLength ) throw()
  2294. {
  2295. ATLASSERT( nLength >= 0 );
  2296. ATLASSERT( nLength <= GetData()->nAllocLength );
  2297. GetData()->nDataLength = nLength;
  2298. m_pszData[nLength] = 0;
  2299. }
  2300. static CStringData * CloneData( CStringData * pData )
  2301. {
  2302. CStringData * pNewData = NULL;
  2303. IAtlStringMgr * pNewStringMgr = pData->pStringMgr->Clone();
  2304. if( !pData->IsLocked() && (pNewStringMgr == pData->pStringMgr) )
  2305. {
  2306. pNewData = pData;
  2307. pNewData->AddRef();
  2308. }
  2309. else
  2310. {
  2311. pNewData = pNewStringMgr->Allocate( pData->nDataLength, sizeof(XCHAR) );
  2312. if( pNewData == NULL )
  2313. {
  2314. ThrowMemoryException();
  2315. }
  2316. pNewData->nDataLength = pData->nDataLength;
  2317. CopyChars( PXSTR( pNewData->data() ), PCXSTR( pData->data() ), pData->nDataLength+1 );
  2318. }
  2319. return ( pNewData );
  2320. }
  2321. private:
  2322. PXSTR m_pszData;
  2323. }; // class CExtSafeSimpleStringT
  2324. template < typename BaseType, class StringTraits >
  2325. class CExtSafeStringT : public CExtSafeSimpleStringT < BaseType >
  2326. {
  2327. public:
  2328. typedef CExtSafeSimpleStringT< BaseType > CThisSimpleString;
  2329. typedef StringTraits StrTraits;
  2330. typedef BaseType WCHAR;
  2331. typedef BaseType * LPWSTR;
  2332. typedef const BaseType * LCPWSTR;
  2333. public:
  2334. CExtSafeStringT() throw()
  2335. : CThisSimpleString( StringTraits::GetDefaultManager() )
  2336. {
  2337. }
  2338. explicit CExtSafeStringT( IAtlStringMgr * pStringMgr ) throw()
  2339. : CThisSimpleString( pStringMgr )
  2340. }
  2341. // CExtSafeStringT( const VARIANT & varSrc );
  2342. // CExtSafeStringT( const VARIANT & varSrc, IAtlStringMgr * pStringMgr );
  2343. static void Construct( CExtSafeStringT * pString )
  2344. {
  2345. new( pString ) CExtSafeStringT;
  2346. }
  2347. CExtSafeStringT( const CExtSafeStringT & strSrc )
  2348. : CThisSimpleString( strSrc )
  2349. {
  2350. }
  2351. CExtSafeStringT( const CThisSimpleString & strSrc )
  2352. : CThisSimpleString( strSrc )
  2353. {
  2354. }
  2355. CExtSafeStringT( const XCHAR * pszSrc )
  2356. : CThisSimpleString( StringTraits::GetDefaultManager() )
  2357. {
  2358. if( !CheckImplicitLoad( pszSrc ) )
  2359. {
  2360. *this = pszSrc;
  2361. }
  2362. }
  2363. CExtSafeStringT( LPCSTR pszSrc, IAtlStringMgr * pStringMgr )
  2364. : CThisSimpleString( pStringMgr )
  2365. {
  2366. if( !CheckImplicitLoad( pszSrc ) )
  2367. {
  2368. *this = pszSrc;
  2369. }
  2370. }
  2371. CSTRING_EXPLICIT CExtSafeStringT( const YCHAR * pszSrc )
  2372. : CThisSimpleString( StringTraits::GetDefaultManager() )
  2373. {
  2374. if( !CheckImplicitLoad( pszSrc ) )
  2375. {
  2376. *this = pszSrc;
  2377. }
  2378. }
  2379. CExtSafeStringT( LPCWSTR pszSrc, IAtlStringMgr * pStringMgr )
  2380. : CThisSimpleString( pStringMgr )
  2381. {
  2382. if( !CheckImplicitLoad( pszSrc ) )
  2383. {
  2384. *this = (PCXSTR)pszSrc;
  2385. }
  2386. }
  2387. //#ifdef _MANAGED
  2388. // CExtSafeStringT( System::String* pString )
  2389. // : CThisSimpleString( StringTraits::GetDefaultManager() )
  2390. // {
  2391. // const wchar_t __pin* psz = PtrToStringChars( pString );
  2392. // *this = psz;
  2393. // }
  2394. //#endif
  2395. CSTRING_EXPLICIT CExtSafeStringT( const unsigned char * pszSrc )
  2396. : CThisSimpleString( StringTraits::GetDefaultManager() )
  2397. {
  2398. *this = reinterpret_cast < const char* > ( pszSrc );
  2399. }
  2400. CExtSafeStringT(
  2401. const unsigned char * pszSrc,
  2402. IAtlStringMgr * pStringMgr
  2403. )
  2404. : CThisSimpleString( pStringMgr )
  2405. {
  2406. *this = reinterpret_cast < const char * > ( pszSrc );
  2407. }
  2408. CSTRING_EXPLICIT CExtSafeStringT( char ch, int nLength = 1 )
  2409. : CThisSimpleString( StringTraits::GetDefaultManager() )
  2410. {
  2411. ATLASSERT( nLength >= 0 );
  2412. if( nLength > 0 )
  2413. {
  2414. PXSTR pszBuffer = GetBuffer( nLength );
  2415. StringTraits::FloodCharacters( XCHAR( ch ), nLength, pszBuffer );
  2416. ReleaseBufferSetLength( nLength );
  2417. }
  2418. }
  2419. CSTRING_EXPLICIT CExtSafeStringT( wchar_t ch, int nLength = 1 )
  2420. : CThisSimpleString( StringTraits::GetDefaultManager() )
  2421. {
  2422. ATLASSERT( nLength >= 0 );
  2423. if( nLength > 0 )
  2424. {
  2425. PXSTR pszBuffer = GetBuffer( nLength );
  2426. StringTraits::FloodCharacters( XCHAR( ch ), nLength, pszBuffer );
  2427. ReleaseBufferSetLength( nLength );
  2428. }
  2429. }
  2430. CExtSafeStringT( const XCHAR * pch, int nLength )
  2431. : CThisSimpleString( pch, nLength, StringTraits::GetDefaultManager() )
  2432. {
  2433. }
  2434. CExtSafeStringT( const XCHAR * pch, int nLength, IAtlStringMgr * pStringMgr )
  2435. : CThisSimpleString( pch, nLength, pStringMgr )
  2436. {
  2437. }
  2438. CExtSafeStringT( const YCHAR * pch, int nLength )
  2439. : CThisSimpleString( StringTraits::GetDefaultManager() )
  2440. {
  2441. ATLASSERT( nLength >= 0 );
  2442. if( nLength > 0 )
  2443. {
  2444. ATLASSERT( AtlIsValidAddress( pch, nLength * sizeof(YCHAR), FALSE ) );
  2445. int nDestLength = StringTraits::GetBaseTypeLength( pch, nLength );
  2446. PXSTR pszBuffer = GetBuffer( nDestLength );
  2447. StringTraits::ConvertToBaseType( pszBuffer, nDestLength, pch, nLength );
  2448. ReleaseBufferSetLength( nDestLength );
  2449. }
  2450. }
  2451. CExtSafeStringT( const YCHAR * pch, int nLength, IAtlStringMgr * pStringMgr )
  2452. : CThisSimpleString( pStringMgr )
  2453. {
  2454. ATLASSERT( nLength >= 0 );
  2455. if( nLength > 0 )
  2456. {
  2457. ATLASSERT( AtlIsValidAddress( pch, nLength * sizeof(YCHAR), FALSE ) );
  2458. int nDestLength = StringTraits::GetBaseTypeLength( pch, nLength );
  2459. PXSTR pszBuffer = GetBuffer( nDestLength );
  2460. StringTraits::ConvertToBaseType( pszBuffer, nDestLength, pch, nLength );
  2461. ReleaseBufferSetLength( nDestLength );
  2462. }
  2463. }
  2464. ~CExtSafeStringT() throw()
  2465. {