root/third_party/wtl/include/atlmisc.h

INCLUDED FROM

// Windows Template Library - WTL version 8.0
// Copyright (C) Microsoft Corporation. All rights reserved.
//
// This file is a part of the Windows Template Library.
// The use and distribution terms for this software are covered by the
// Microsoft Permissive License (Ms-PL) which can be found in the file
// Ms-PL.txt at the root of this distribution.

#ifndef __ATLMISC_H__
#define __ATLMISC_H__

#pragma once

#ifndef __cplusplus
        #error ATL requires C++ compilation (use a .cpp suffix)
#endif

#ifndef __ATLAPP_H__
        #error atlmisc.h requires atlapp.h to be included first
#endif


#ifdef _ATL_TMP_NO_CSTRING
  #define _WTL_NO_CSTRING
#endif

#if defined(_WTL_USE_CSTRING) && defined(_WTL_NO_CSTRING)
        #error Conflicting options - both _WTL_USE_CSTRING and _WTL_NO_CSTRING are defined
#endif // defined(_WTL_USE_CSTRING) && defined(_WTL_NO_CSTRING)

#if !defined(_WTL_USE_CSTRING) && !defined(_WTL_NO_CSTRING)
  #define _WTL_USE_CSTRING
#endif // !defined(_WTL_USE_CSTRING) && !defined(_WTL_NO_CSTRING)

#ifndef _WTL_NO_CSTRING
  #if defined(_ATL_USE_CSTRING_FLOAT) && defined(_ATL_MIN_CRT)
        #error Cannot use CString floating point formatting with _ATL_MIN_CRT defined
  #endif // defined(_ATL_USE_CSTRING_FLOAT) && defined(_ATL_MIN_CRT)
#endif // !_WTL_NO_CSTRING


///////////////////////////////////////////////////////////////////////////////
// Classes in this file:
//
// CSize
// CPoint
// CRect
// CString
//
// CRecentDocumentListBase<T, t_cchItemLen, t_nFirstID, t_nLastID>
// CRecentDocumentList
// CFindFile
//
// Global functions:
//   AtlLoadAccelerators()
//   AtlLoadMenu()
//   AtlLoadBitmap()
//   AtlLoadSysBitmap()
//   AtlLoadCursor()
//   AtlLoadSysCursor()
//   AtlLoadIcon()
//   AtlLoadSysIcon()
//   AtlLoadBitmapImage()
//   AtlLoadCursorImage()
//   AtlLoadIconImage()
//   AtlLoadSysBitmapImage()
//   AtlLoadSysCursorImage()
//   AtlLoadSysIconImage()
//   AtlLoadString()
//
//   AtlGetStockPen()
//   AtlGetStockBrush()
//   AtlGetStockFont()
//   AtlGetStockPalette()
//
//   AtlCompactPath()


namespace WTL
{

#ifndef _WTL_NO_WTYPES

// forward declarations
class CSize;
class CPoint;
class CRect;

///////////////////////////////////////////////////////////////////////////////
// CSize - Wrapper for Windows SIZE structure.

class CSize : public SIZE
{
public:
// Constructors
        CSize()
        {
                cx = 0;
                cy = 0;
        }

        CSize(int initCX, int initCY)
        {
                cx = initCX;
                cy = initCY;
        }

        CSize(SIZE initSize)
        {
                *(SIZE*)this = initSize;
        }

        CSize(POINT initPt)
        {
                *(POINT*)this = initPt;
        }

        CSize(DWORD dwSize)
        {
                cx = (short)LOWORD(dwSize);
                cy = (short)HIWORD(dwSize);
        }

// Operations
        BOOL operator ==(SIZE size) const
        {
                return (cx == size.cx && cy == size.cy);
        }

        BOOL operator !=(SIZE size) const
        {
                return (cx != size.cx || cy != size.cy);
        }

        void operator +=(SIZE size)
        {
                cx += size.cx;
                cy += size.cy;
        }

        void operator -=(SIZE size)
        {
                cx -= size.cx;
                cy -= size.cy;
        }

        void SetSize(int CX, int CY)
        {
                cx = CX;
                cy = CY;
        }

// Operators returning CSize values
        CSize operator +(SIZE size) const
        {
                return CSize(cx + size.cx, cy + size.cy);
        }

        CSize operator -(SIZE size) const
        {
                return CSize(cx - size.cx, cy - size.cy);
        }

        CSize operator -() const
        {
                return CSize(-cx, -cy);
        }

// Operators returning CPoint values
        CPoint operator +(POINT point) const;
        CPoint operator -(POINT point) const;

// Operators returning CRect values
        CRect operator +(const RECT* lpRect) const;
        CRect operator -(const RECT* lpRect) const;
};


///////////////////////////////////////////////////////////////////////////////
// CPoint - Wrapper for Windows POINT structure.

class CPoint : public POINT
{
public:
// Constructors
        CPoint()
        {
                x = 0;
                y = 0;
        }

        CPoint(int initX, int initY)
        {
                x = initX;
                y = initY;
        }

        CPoint(POINT initPt)
        {
                *(POINT*)this = initPt;
        }

        CPoint(SIZE initSize)
        {
                *(SIZE*)this = initSize;
        }

        CPoint(DWORD dwPoint)
        {
                x = (short)LOWORD(dwPoint);
                y = (short)HIWORD(dwPoint);
        }

// Operations
        void Offset(int xOffset, int yOffset)
        {
                x += xOffset;
                y += yOffset;
        }

        void Offset(POINT point)
        {
                x += point.x;
                y += point.y;
        }

        void Offset(SIZE size)
        {
                x += size.cx;
                y += size.cy;
        }

        BOOL operator ==(POINT point) const
        {
                return (x == point.x && y == point.y);
        }

        BOOL operator !=(POINT point) const
        {
                return (x != point.x || y != point.y);
        }

        void operator +=(SIZE size)
        {
                x += size.cx;
                y += size.cy;
        }

        void operator -=(SIZE size)
        {
                x -= size.cx;
                y -= size.cy;
        }

        void operator +=(POINT point)
        {
                x += point.x;
                y += point.y;
        }

        void operator -=(POINT point)
        {
                x -= point.x;
                y -= point.y;
        }

        void SetPoint(int X, int Y)
        {
                x = X;
                y = Y;
        }

// Operators returning CPoint values
        CPoint operator +(SIZE size) const
        {
                return CPoint(x + size.cx, y + size.cy);
        }

        CPoint operator -(SIZE size) const
        {
                return CPoint(x - size.cx, y - size.cy);
        }

        CPoint operator -() const
        {
                return CPoint(-x, -y);
        }

        CPoint operator +(POINT point) const
        {
                return CPoint(x + point.x, y + point.y);
        }

// Operators returning CSize values
        CSize operator -(POINT point) const
        {
                return CSize(x - point.x, y - point.y);
        }

// Operators returning CRect values
        CRect operator +(const RECT* lpRect) const;
        CRect operator -(const RECT* lpRect) const;
};


///////////////////////////////////////////////////////////////////////////////
// CRect - Wrapper for Windows RECT structure.

class CRect : public RECT
{
public:
// Constructors
        CRect()
        {
                left = 0;
                top = 0;
                right = 0;
                bottom = 0;
        }

        CRect(int l, int t, int r, int b)
        {
                left = l;
                top = t;
                right = r;
                bottom = b;
        }

        CRect(const RECT& srcRect)
        {
                ::CopyRect(this, &srcRect);
        }

        CRect(LPCRECT lpSrcRect)
        {
                ::CopyRect(this, lpSrcRect);
        }

        CRect(POINT point, SIZE size)
        {
                right = (left = point.x) + size.cx;
                bottom = (top = point.y) + size.cy;
        }

        CRect(POINT topLeft, POINT bottomRight)
        {
                left = topLeft.x;
                top = topLeft.y;
                right = bottomRight.x;
                bottom = bottomRight.y;
        }

// Attributes (in addition to RECT members)
        int Width() const
        {
                return right - left;
        }

        int Height() const
        {
                return bottom - top;
        }

        CSize Size() const
        {
                return CSize(right - left, bottom - top);
        }

        CPoint& TopLeft()
        {
                return *((CPoint*)this);
        }

        CPoint& BottomRight()
        {
                return *((CPoint*)this + 1);
        }

        const CPoint& TopLeft() const
        {
                return *((CPoint*)this);
        }

        const CPoint& BottomRight() const
        {
                return *((CPoint*)this + 1);
        }

        CPoint CenterPoint() const
        {
                return CPoint((left + right) / 2, (top + bottom) / 2);
        }

        // convert between CRect and LPRECT/LPCRECT (no need for &)
        operator LPRECT()
        {
                return this;
        }

        operator LPCRECT() const
        {
                return this;
        }

        BOOL IsRectEmpty() const
        {
                return ::IsRectEmpty(this);
        }

        BOOL IsRectNull() const
        {
                return (left == 0 && right == 0 && top == 0 && bottom == 0);
        }

        BOOL PtInRect(POINT point) const
        {
                return ::PtInRect(this, point);
        }

// Operations
        void SetRect(int x1, int y1, int x2, int y2)
        {
                ::SetRect(this, x1, y1, x2, y2);
        }

        void SetRect(POINT topLeft, POINT bottomRight)
        {
                ::SetRect(this, topLeft.x, topLeft.y, bottomRight.x, bottomRight.y);
        }

        void SetRectEmpty()
        {
                ::SetRectEmpty(this);
        }

        void CopyRect(LPCRECT lpSrcRect)
        {
                ::CopyRect(this, lpSrcRect);
        }

        BOOL EqualRect(LPCRECT lpRect) const
        {
                return ::EqualRect(this, lpRect);
        }

        void InflateRect(int x, int y)
        {
                ::InflateRect(this, x, y);
        }

        void InflateRect(SIZE size)
        {
                ::InflateRect(this, size.cx, size.cy);
        }

        void InflateRect(LPCRECT lpRect)
        {
                left -= lpRect->left;
                top -= lpRect->top;
                right += lpRect->right;
                bottom += lpRect->bottom;
        }

        void InflateRect(int l, int t, int r, int b)
        {
                left -= l;
                top -= t;
                right += r;
                bottom += b;
        }

        void DeflateRect(int x, int y)
        {
                ::InflateRect(this, -x, -y);
        }

        void DeflateRect(SIZE size)
        {
                ::InflateRect(this, -size.cx, -size.cy);
        }

        void DeflateRect(LPCRECT lpRect)
        {
                left += lpRect->left;
                top += lpRect->top;
                right -= lpRect->right;
                bottom -= lpRect->bottom;
        }

        void DeflateRect(int l, int t, int r, int b)
        {
                left += l;
                top += t;
                right -= r;
                bottom -= b;
        }

        void OffsetRect(int x, int y)
        {
                ::OffsetRect(this, x, y);
        }
        void OffsetRect(SIZE size)
        {
                ::OffsetRect(this, size.cx, size.cy);
        }

        void OffsetRect(POINT point)
        {
                ::OffsetRect(this, point.x, point.y);
        }

        void NormalizeRect()
        {
                int nTemp;
                if (left > right)
                {
                        nTemp = left;
                        left = right;
                        right = nTemp;
                }
                if (top > bottom)
                {
                        nTemp = top;
                        top = bottom;
                        bottom = nTemp;
                }
        }

        // absolute position of rectangle
        void MoveToY(int y)
        {
                bottom = Height() + y;
                top = y;
        }

        void MoveToX(int x)
        {
                right = Width() + x;
                left = x;
        }

        void MoveToXY(int x, int y)
        {
                MoveToX(x);
                MoveToY(y);
        }

        void MoveToXY(POINT pt)
        {
                MoveToX(pt.x);
                MoveToY(pt.y);
        }

        // operations that fill '*this' with result
        BOOL IntersectRect(LPCRECT lpRect1, LPCRECT lpRect2)
        {
                return ::IntersectRect(this, lpRect1, lpRect2);
        }

        BOOL UnionRect(LPCRECT lpRect1, LPCRECT lpRect2)
        {
                return ::UnionRect(this, lpRect1, lpRect2);
        }

        BOOL SubtractRect(LPCRECT lpRectSrc1, LPCRECT lpRectSrc2)
        {
                return ::SubtractRect(this, lpRectSrc1, lpRectSrc2);
        }

// Additional Operations
        void operator =(const RECT& srcRect)
        {
                ::CopyRect(this, &srcRect);
        }

        BOOL operator ==(const RECT& rect) const
        {
                return ::EqualRect(this, &rect);
        }

        BOOL operator !=(const RECT& rect) const
        {
                return !::EqualRect(this, &rect);
        }

        void operator +=(POINT point)
        {
                ::OffsetRect(this, point.x, point.y);
        }

        void operator +=(SIZE size)
        {
                ::OffsetRect(this, size.cx, size.cy);
        }

        void operator +=(LPCRECT lpRect)
        {
                InflateRect(lpRect);
        }

        void operator -=(POINT point)
        {
                ::OffsetRect(this, -point.x, -point.y);
        }

        void operator -=(SIZE size)
        {
                ::OffsetRect(this, -size.cx, -size.cy);
        }

        void operator -=(LPCRECT lpRect)
        {
                DeflateRect(lpRect);
        }

        void operator &=(const RECT& rect)
        {
                ::IntersectRect(this, this, &rect);
        }

        void operator |=(const RECT& rect)
        {
                ::UnionRect(this, this, &rect);
        }

// Operators returning CRect values
        CRect operator +(POINT pt) const
        {
                CRect rect(*this);
                ::OffsetRect(&rect, pt.x, pt.y);
                return rect;
        }

        CRect operator -(POINT pt) const
        {
                CRect rect(*this);
                ::OffsetRect(&rect, -pt.x, -pt.y);
                return rect;
        }

        CRect operator +(LPCRECT lpRect) const
        {
                CRect rect(this);
                rect.InflateRect(lpRect);
                return rect;
        }

        CRect operator +(SIZE size) const
        {
                CRect rect(*this);
                ::OffsetRect(&rect, size.cx, size.cy);
                return rect;
        }

        CRect operator -(SIZE size) const
        {
                CRect rect(*this);
                ::OffsetRect(&rect, -size.cx, -size.cy);
                return rect;
        }

        CRect operator -(LPCRECT lpRect) const
        {
                CRect rect(this);
                rect.DeflateRect(lpRect);
                return rect;
        }

        CRect operator &(const RECT& rect2) const
        {
                CRect rect;
                ::IntersectRect(&rect, this, &rect2);
                return rect;
        }

        CRect operator |(const RECT& rect2) const
        {
                CRect rect;
                ::UnionRect(&rect, this, &rect2);
                return rect;
        }

        CRect MulDiv(int nMultiplier, int nDivisor) const
        {
                return CRect(
                        ::MulDiv(left, nMultiplier, nDivisor),
                        ::MulDiv(top, nMultiplier, nDivisor),
                        ::MulDiv(right, nMultiplier, nDivisor),
                        ::MulDiv(bottom, nMultiplier, nDivisor));
        }
};


// CSize implementation

inline CPoint CSize::operator +(POINT point) const
{ return CPoint(cx + point.x, cy + point.y); }

inline CPoint CSize::operator -(POINT point) const
{ return CPoint(cx - point.x, cy - point.y); }

inline CRect CSize::operator +(const RECT* lpRect) const
{ return CRect(lpRect) + *this; }

inline CRect CSize::operator -(const RECT* lpRect) const
{ return CRect(lpRect) - *this; }


// CPoint implementation

inline CRect CPoint::operator +(const RECT* lpRect) const
{ return CRect(lpRect) + *this; }

inline CRect CPoint::operator -(const RECT* lpRect) const
{ return CRect(lpRect) - *this; }

#endif // !_WTL_NO_WTYPES


// WTL::CSize or ATL::CSize scalar operators 

#if !defined(_WTL_NO_SIZE_SCALAR) && (!defined(_WTL_NO_WTYPES) || defined(__ATLTYPES_H__))

template <class Num>
inline CSize operator *(SIZE s, Num n) 
{
        return CSize((int)(s.cx * n), (int)(s.cy * n));
};

template <class Num>
inline void operator *=(SIZE & s, Num n)
{
        s = s * n;
};      

template <class Num>
inline CSize operator /(SIZE s, Num n) 
{
        return CSize((int)(s.cx / n), (int)(s.cy / n));
};

template <class Num>
inline void operator /=(SIZE & s, Num n)
{
        s = s / n;
};      

#endif // !defined(_WTL_NO_SIZE_SCALAR) && (!defined(_WTL_NO_WTYPES) || defined(__ATLTYPES_H__))


///////////////////////////////////////////////////////////////////////////////
// CString - String class

#ifndef _WTL_NO_CSTRING

struct CStringData
{
        long nRefs;     // reference count
        int nDataLength;
        int nAllocLength;
        // TCHAR data[nAllocLength]

        TCHAR* data()
        { return (TCHAR*)(this + 1); }
};

// Globals

// For an empty string, m_pchData will point here
// (note: avoids special case of checking for NULL m_pchData)
// empty string data (and locked)
_declspec(selectany) int rgInitData[] = { -1, 0, 0, 0 };
_declspec(selectany) CStringData* _atltmpDataNil = (CStringData*)&rgInitData;
_declspec(selectany) LPCTSTR _atltmpPchNil = (LPCTSTR)(((BYTE*)&rgInitData) + sizeof(CStringData));


class CString
{
public:
// Constructors
        CString()
        {
                Init();
        }

        CString(const CString& stringSrc)
        {
                ATLASSERT(stringSrc.GetData()->nRefs != 0);
                if (stringSrc.GetData()->nRefs >= 0)
                {
                        ATLASSERT(stringSrc.GetData() != _atltmpDataNil);
                        m_pchData = stringSrc.m_pchData;
                        InterlockedIncrement(&GetData()->nRefs);
                }
                else
                {
                        Init();
                        *this = stringSrc.m_pchData;
                }
        }

        CString(TCHAR ch, int nRepeat = 1)
        {
                ATLASSERT(!_istlead(ch));   // can't create a lead byte string
                Init();
                if (nRepeat >= 1)
                {
                        if(AllocBuffer(nRepeat))
                        {
#ifdef _UNICODE
                                for (int i = 0; i < nRepeat; i++)
                                        m_pchData[i] = ch;
#else
                                memset(m_pchData, ch, nRepeat);
#endif
                        }
                }
        }

        CString(LPCTSTR lpsz)
        {
                Init();
                if (lpsz != NULL && HIWORD(lpsz) == NULL)
                {
                        UINT nID = LOWORD((DWORD_PTR)lpsz);
                        if (!LoadString(nID))
                                ATLTRACE2(atlTraceUI, 0, _T("Warning: implicit LoadString(%u) in CString failed\n"), nID);
                }
                else
                {
                        int nLen = SafeStrlen(lpsz);
                        if (nLen != 0)
                        {
                                if(AllocBuffer(nLen))
                                        SecureHelper::memcpy_x(m_pchData, (nLen + 1) * sizeof(TCHAR), lpsz, nLen * sizeof(TCHAR));
                        }
                }
        }

#ifdef _UNICODE
        CString(LPCSTR lpsz)
        {
                Init();
#if defined(_WIN32_WCE) && (_ATL_VER >= 0x0800)
                int nSrcLen = (lpsz != NULL) ? ATL::lstrlenA(lpsz) : 0;
#else
                int nSrcLen = (lpsz != NULL) ? lstrlenA(lpsz) : 0;
#endif
                if (nSrcLen != 0)
                {
                        if(AllocBuffer(nSrcLen))
                        {
                                _mbstowcsz(m_pchData, lpsz, nSrcLen + 1);
                                ReleaseBuffer();
                        }
                }
        }
#else // !_UNICODE
        CString(LPCWSTR lpsz)
        {
                Init();
                int nSrcLen = (lpsz != NULL) ? (int)wcslen(lpsz) : 0;
                if (nSrcLen != 0)
                {
                        if(AllocBuffer(nSrcLen * 2))
                        {
                                _wcstombsz(m_pchData, lpsz, (nSrcLen * 2) + 1);
                                ReleaseBuffer();
                        }
                }
        }
#endif // !_UNICODE

        CString(LPCTSTR lpch, int nLength)
        {
                Init();
                if (nLength != 0)
                {
                        if(AllocBuffer(nLength))
                                SecureHelper::memcpy_x(m_pchData, (nLength + 1) * sizeof(TCHAR), lpch, nLength * sizeof(TCHAR));
                }
        }

#ifdef _UNICODE
        CString(LPCSTR lpsz, int nLength)
        {
                Init();
                if (nLength != 0)
                {
                        if(AllocBuffer(nLength))
                        {
                                int n = ::MultiByteToWideChar(CP_ACP, 0, lpsz, nLength, m_pchData, nLength + 1);
                                ReleaseBuffer((n >= 0) ? n : -1);
                        }
                }
        }
#else // !_UNICODE
        CString(LPCWSTR lpsz, int nLength)
        {
                Init();
                if (nLength != 0)
                {
                        if(((nLength * 2) > nLength) && AllocBuffer(nLength * 2))
                        {
                                int n = ::WideCharToMultiByte(CP_ACP, 0, lpsz, nLength, m_pchData, (nLength * 2) + 1, NULL, NULL);
                                ReleaseBuffer((n >= 0) ? n : -1);
                        }
                }
        }
#endif // !_UNICODE

        CString(const unsigned char* lpsz)
        {
                Init();
                *this = (LPCSTR)lpsz;
        }

// Attributes & Operations
        int GetLength() const   // as an array of characters
        {
                return GetData()->nDataLength;
        }

        BOOL IsEmpty() const
        {
                return GetData()->nDataLength == 0;
        }

        void Empty()   // free up the data
        {
                if (GetData()->nDataLength == 0)
                        return;

                if (GetData()->nRefs >= 0)
                        Release();
                else
                        *this = _T("");

                ATLASSERT(GetData()->nDataLength == 0);
                ATLASSERT(GetData()->nRefs < 0 || GetData()->nAllocLength == 0);
        }

        TCHAR GetAt(int nIndex) const   // 0 based
        {
                ATLASSERT(nIndex >= 0);
                ATLASSERT(nIndex < GetData()->nDataLength);
                return m_pchData[nIndex];
        }

        TCHAR operator [](int nIndex) const   // same as GetAt
        {
                // same as GetAt
                ATLASSERT(nIndex >= 0);
                ATLASSERT(nIndex < GetData()->nDataLength);
                return m_pchData[nIndex];
        }

        void SetAt(int nIndex, TCHAR ch)
        {
                ATLASSERT(nIndex >= 0);
                ATLASSERT(nIndex < GetData()->nDataLength);

                CopyBeforeWrite();
                m_pchData[nIndex] = ch;
        }

        operator LPCTSTR() const   // as a C string
        {
                return m_pchData;
        }

        // overloaded assignment
        CString& operator =(const CString& stringSrc)
        {
                if (m_pchData != stringSrc.m_pchData)
                {
                        if ((GetData()->nRefs < 0 && GetData() != _atltmpDataNil) || stringSrc.GetData()->nRefs < 0)
                        {
                                // actual copy necessary since one of the strings is locked
                                AssignCopy(stringSrc.GetData()->nDataLength, stringSrc.m_pchData);
                        }
                        else
                        {
                                // can just copy references around
                                Release();
                                ATLASSERT(stringSrc.GetData() != _atltmpDataNil);
                                m_pchData = stringSrc.m_pchData;
                                InterlockedIncrement(&GetData()->nRefs);
                        }
                }
                return *this;
        }

        CString& operator =(TCHAR ch)
        {
                ATLASSERT(!_istlead(ch));   // can't set single lead byte
                AssignCopy(1, &ch);
                return *this;
        }

#ifdef _UNICODE
        CString& operator =(char ch)
        {
                *this = (TCHAR)ch;
                return *this;
        }
#endif

        CString& operator =(LPCTSTR lpsz)
        {
                ATLASSERT(lpsz == NULL || _IsValidString(lpsz));
                AssignCopy(SafeStrlen(lpsz), lpsz);
                return *this;
        }

#ifdef _UNICODE
        CString& operator =(LPCSTR lpsz)
        {
#if defined(_WIN32_WCE) && (_ATL_VER >= 0x0800)
                int nSrcLen = (lpsz != NULL) ? ATL::lstrlenA(lpsz) : 0;
#else
                int nSrcLen = (lpsz != NULL) ? lstrlenA(lpsz) : 0;
#endif
                if(AllocBeforeWrite(nSrcLen))
                {
                        _mbstowcsz(m_pchData, lpsz, nSrcLen + 1);
                        ReleaseBuffer();
                }
                return *this;
        }
#else // !_UNICODE
        CString& operator =(LPCWSTR lpsz)
        {
                int nSrcLen = (lpsz != NULL) ? (int)wcslen(lpsz) : 0;
                if(AllocBeforeWrite(nSrcLen * 2))
                {
                        _wcstombsz(m_pchData, lpsz, (nSrcLen * 2) + 1);
                        ReleaseBuffer();
                }
                return *this;
        }
#endif  // !_UNICODE

        CString& operator =(const unsigned char* lpsz)
        {
                *this = (LPCSTR)lpsz;
                return *this;
        }

        // string concatenation
        CString& operator +=(const CString& string)
        {
                ConcatInPlace(string.GetData()->nDataLength, string.m_pchData);
                return *this;
        }

        CString& operator +=(TCHAR ch)
        {
                ConcatInPlace(1, &ch);
                return *this;
        }

#ifdef _UNICODE
        CString& operator +=(char ch)
        {
                *this += (TCHAR)ch;
                return *this;
        }
#endif

        CString& operator +=(LPCTSTR lpsz)
        {
                ATLASSERT(lpsz == NULL || _IsValidString(lpsz));
                ConcatInPlace(SafeStrlen(lpsz), lpsz);
                return *this;
        }

        friend CString __stdcall operator +(const CString& string1, const CString& string2);
        friend CString __stdcall operator +(const CString& string, TCHAR ch);
        friend CString __stdcall operator +(TCHAR ch, const CString& string);
#ifdef _UNICODE
        friend CString __stdcall operator +(const CString& string, char ch);
        friend CString __stdcall operator +(char ch, const CString& string);
#endif
        friend CString __stdcall operator +(const CString& string, LPCTSTR lpsz);
        friend CString __stdcall operator +(LPCTSTR lpsz, const CString& string);

        // string comparison
        int Compare(LPCTSTR lpsz) const   // straight character (MBCS/Unicode aware)
        {
                return _cstrcmp(m_pchData, lpsz);
        }

        int CompareNoCase(LPCTSTR lpsz) const   // ignore case (MBCS/Unicode aware)
        {
                return _cstrcmpi(m_pchData, lpsz);
        }

#ifndef _WIN32_WCE
        // CString::Collate is often slower than Compare but is MBSC/Unicode
        //  aware as well as locale-sensitive with respect to sort order.
        int Collate(LPCTSTR lpsz) const   // NLS aware
        {
                return _cstrcoll(m_pchData, lpsz);
        }

        int CollateNoCase(LPCTSTR lpsz) const   // ignore case
        {
                return _cstrcolli(m_pchData, lpsz);
        }
#endif // !_WIN32_WCE

        // simple sub-string extraction
        CString Mid(int nFirst, int nCount) const
        {
                // out-of-bounds requests return sensible things
                if (nFirst < 0)
                        nFirst = 0;
                if (nCount < 0)
                        nCount = 0;

                if (nFirst + nCount > GetData()->nDataLength)
                        nCount = GetData()->nDataLength - nFirst;
                if (nFirst > GetData()->nDataLength)
                        nCount = 0;

                CString dest;
                AllocCopy(dest, nCount, nFirst, 0);
                return dest;
        }

        CString Mid(int nFirst) const
        {
                return Mid(nFirst, GetData()->nDataLength - nFirst);
        }

        CString Left(int nCount) const
        {
                if (nCount < 0)
                        nCount = 0;
                else if (nCount > GetData()->nDataLength)
                        nCount = GetData()->nDataLength;

                CString dest;
                AllocCopy(dest, nCount, 0, 0);
                return dest;
        }

        CString Right(int nCount) const
        {
                if (nCount < 0)
                        nCount = 0;
                else if (nCount > GetData()->nDataLength)
                        nCount = GetData()->nDataLength;

                CString dest;
                AllocCopy(dest, nCount, GetData()->nDataLength-nCount, 0);
                return dest;
        }

        CString SpanIncluding(LPCTSTR lpszCharSet) const   // strspn equivalent
        {
                ATLASSERT(_IsValidString(lpszCharSet));
                return Left(_cstrspn(m_pchData, lpszCharSet));
        }

        CString SpanExcluding(LPCTSTR lpszCharSet) const   // strcspn equivalent
        {
                ATLASSERT(_IsValidString(lpszCharSet));
                return Left(_cstrcspn(m_pchData, lpszCharSet));
        }

        // upper/lower/reverse conversion
        void MakeUpper()
        {
                CopyBeforeWrite();
                CharUpper(m_pchData);
        }

        void MakeLower()
        {
                CopyBeforeWrite();
                CharLower(m_pchData);
        }

        void MakeReverse()
        {
                CopyBeforeWrite();
                _cstrrev(m_pchData);
        }

        // trimming whitespace (either side)
        void TrimRight()
        {
                CopyBeforeWrite();

                // find beginning of trailing spaces by starting at beginning (DBCS aware)
                LPTSTR lpsz = m_pchData;
                LPTSTR lpszLast = NULL;
                while (*lpsz != _T('\0'))
                {
                        if (_cstrisspace(*lpsz))
                        {
                                if (lpszLast == NULL)
                                        lpszLast = lpsz;
                        }
                        else
                        {
                                lpszLast = NULL;
                        }
                        lpsz = ::CharNext(lpsz);
                }

                if (lpszLast != NULL)
                {
                        // truncate at trailing space start
                        *lpszLast = _T('\0');
                        GetData()->nDataLength = (int)(DWORD_PTR)(lpszLast - m_pchData);
                }
        }

        void TrimLeft()
        {
                CopyBeforeWrite();

                // find first non-space character
                LPCTSTR lpsz = m_pchData;
                while (_cstrisspace(*lpsz))
                        lpsz = ::CharNext(lpsz);

                // fix up data and length
                int nDataLength = GetData()->nDataLength - (int)(DWORD_PTR)(lpsz - m_pchData);
                SecureHelper::memmove_x(m_pchData, (GetData()->nAllocLength + 1) * sizeof(TCHAR), lpsz, (nDataLength + 1) * sizeof(TCHAR));
                GetData()->nDataLength = nDataLength;
        }

        // remove continuous occurrences of chTarget starting from right
        void TrimRight(TCHAR chTarget)
        {
                // find beginning of trailing matches
                // by starting at beginning (DBCS aware)

                CopyBeforeWrite();
                LPTSTR lpsz = m_pchData;
                LPTSTR lpszLast = NULL;

                while (*lpsz != _T('\0'))
                {
                        if (*lpsz == chTarget)
                        {
                                if (lpszLast == NULL)
                                        lpszLast = lpsz;
                        }
                        else
                                lpszLast = NULL;
                        lpsz = ::CharNext(lpsz);
                }

                if (lpszLast != NULL)
                {
                        // truncate at left-most matching character
                        *lpszLast = _T('\0');
                        GetData()->nDataLength = (int)(DWORD_PTR)(lpszLast - m_pchData);
                }
        }

        // remove continuous occcurrences of characters in passed string, starting from right
        void TrimRight(LPCTSTR lpszTargetList)
        {
                // find beginning of trailing matches by starting at beginning (DBCS aware)

                CopyBeforeWrite();
                LPTSTR lpsz = m_pchData;
                LPTSTR lpszLast = NULL;

                while (*lpsz != _T('\0'))
                {
                        TCHAR* pNext = ::CharNext(lpsz);
                        if(pNext > lpsz + 1)
                        {
                                if (_cstrchr_db(lpszTargetList, *lpsz, *(lpsz + 1)) != NULL)
                                {
                                        if (lpszLast == NULL)
                                                lpszLast = lpsz;
                                }
                                else
                                {
                                        lpszLast = NULL;
                                }
                        }
                        else
                        {
                                if (_cstrchr(lpszTargetList, *lpsz) != NULL)
                                {
                                        if (lpszLast == NULL)
                                                lpszLast = lpsz;
                                }
                                else
                                {
                                        lpszLast = NULL;
                                }
                        }

                        lpsz = pNext;
                }

                if (lpszLast != NULL)
                {
                        // truncate at left-most matching character
                        *lpszLast = _T('\0');
                        GetData()->nDataLength = (int)(DWORD_PTR)(lpszLast - m_pchData);
                }
        }

        // remove continuous occurrences of chTarget starting from left
        void TrimLeft(TCHAR chTarget)
        {
                // find first non-matching character

                CopyBeforeWrite();
                LPCTSTR lpsz = m_pchData;

                while (chTarget == *lpsz)
                        lpsz = ::CharNext(lpsz);

                if (lpsz != m_pchData)
                {
                        // fix up data and length
                        int nDataLength = GetData()->nDataLength - (int)(DWORD_PTR)(lpsz - m_pchData);
                        SecureHelper::memmove_x(m_pchData, (GetData()->nAllocLength + 1) * sizeof(TCHAR), lpsz, (nDataLength + 1) * sizeof(TCHAR));
                        GetData()->nDataLength = nDataLength;
                }
        }

        // remove continuous occcurrences of characters in passed string, starting from left
        void TrimLeft(LPCTSTR lpszTargets)
        {
                // if we're not trimming anything, we're not doing any work
                if (SafeStrlen(lpszTargets) == 0)
                        return;

                CopyBeforeWrite();
                LPCTSTR lpsz = m_pchData;

                while (*lpsz != _T('\0'))
                {
                        TCHAR* pNext = ::CharNext(lpsz);
                        if(pNext > lpsz + 1)
                        {
                                if (_cstrchr_db(lpszTargets, *lpsz, *(lpsz + 1)) == NULL)
                                        break;
                        }
                        else
                        {
                                if (_cstrchr(lpszTargets, *lpsz) == NULL)
                                        break;
                        }
                        lpsz = pNext;
                }

                if (lpsz != m_pchData)
                {
                        // fix up data and length
                        int nDataLength = GetData()->nDataLength - (int)(DWORD_PTR)(lpsz - m_pchData);
                        SecureHelper::memmove_x(m_pchData, (GetData()->nAllocLength + 1) * sizeof(TCHAR), lpsz, (nDataLength + 1) * sizeof(TCHAR));
                        GetData()->nDataLength = nDataLength;
                }
        }

        // advanced manipulation
        // replace occurrences of chOld with chNew
        int Replace(TCHAR chOld, TCHAR chNew)
        {
                int nCount = 0;

                // short-circuit the nop case
                if (chOld != chNew)
                {
                        // otherwise modify each character that matches in the string
                        CopyBeforeWrite();
                        LPTSTR psz = m_pchData;
                        LPTSTR pszEnd = psz + GetData()->nDataLength;
                        while (psz < pszEnd)
                        {
                                // replace instances of the specified character only
                                if (*psz == chOld)
                                {
                                        *psz = chNew;
                                        nCount++;
                                }
                                psz = ::CharNext(psz);
                        }
                }
                return nCount;
        }

        // replace occurrences of substring lpszOld with lpszNew;
        // empty lpszNew removes instances of lpszOld
        int Replace(LPCTSTR lpszOld, LPCTSTR lpszNew)
        {
                // can't have empty or NULL lpszOld

                int nSourceLen = SafeStrlen(lpszOld);
                if (nSourceLen == 0)
                        return 0;
                int nReplacementLen = SafeStrlen(lpszNew);

                // loop once to figure out the size of the result string
                int nCount = 0;
                LPTSTR lpszStart = m_pchData;
                LPTSTR lpszEnd = m_pchData + GetData()->nDataLength;
                LPTSTR lpszTarget = NULL;
                while (lpszStart < lpszEnd)
                {
                        while ((lpszTarget = (TCHAR*)_cstrstr(lpszStart, lpszOld)) != NULL)
                        {
                                nCount++;
                                lpszStart = lpszTarget + nSourceLen;
                        }
                        lpszStart += lstrlen(lpszStart) + 1;
                }

                // if any changes were made, make them
                if (nCount > 0)
                {
                        CopyBeforeWrite();

                        // if the buffer is too small, just allocate a new buffer (slow but sure)
                        int nOldLength = GetData()->nDataLength;
                        int nNewLength =  nOldLength + (nReplacementLen - nSourceLen) * nCount;
                        if (GetData()->nAllocLength < nNewLength || GetData()->nRefs > 1)
                        {
                                CStringData* pOldData = GetData();
                                LPTSTR pstr = m_pchData;
                                if(!AllocBuffer(nNewLength))
                                        return -1;
                                SecureHelper::memcpy_x(m_pchData, (nNewLength + 1) * sizeof(TCHAR), pstr, pOldData->nDataLength * sizeof(TCHAR));
                                CString::Release(pOldData);
                        }
                        // else, we just do it in-place
                        lpszStart = m_pchData;
                        lpszEnd = m_pchData + GetData()->nDataLength;

                        // loop again to actually do the work
                        while (lpszStart < lpszEnd)
                        {
                                while ((lpszTarget = (TCHAR*)_cstrstr(lpszStart, lpszOld)) != NULL)
                                {
                                        int nBalance = nOldLength - ((int)(DWORD_PTR)(lpszTarget - m_pchData) + nSourceLen);
                                        int cchBuffLen = GetData()->nAllocLength - (int)(DWORD_PTR)(lpszTarget - m_pchData);
                                        SecureHelper::memmove_x(lpszTarget + nReplacementLen, (cchBuffLen - nReplacementLen + 1) * sizeof(TCHAR), lpszTarget + nSourceLen, nBalance * sizeof(TCHAR));
                                        SecureHelper::memcpy_x(lpszTarget, (cchBuffLen + 1) * sizeof(TCHAR), lpszNew, nReplacementLen * sizeof(TCHAR));
                                        lpszStart = lpszTarget + nReplacementLen;
                                        lpszStart[nBalance] = _T('\0');
                                        nOldLength += (nReplacementLen - nSourceLen);
                                }
                                lpszStart += lstrlen(lpszStart) + 1;
                        }
                        ATLASSERT(m_pchData[nNewLength] == _T('\0'));
                        GetData()->nDataLength = nNewLength;
                }

                return nCount;
        }

        // remove occurrences of chRemove
        int Remove(TCHAR chRemove)
        {
                CopyBeforeWrite();

                LPTSTR pstrSource = m_pchData;
                LPTSTR pstrDest = m_pchData;
                LPTSTR pstrEnd = m_pchData + GetData()->nDataLength;

                while (pstrSource < pstrEnd)
                {
                        if (*pstrSource != chRemove)
                        {
                                *pstrDest = *pstrSource;
                                pstrDest = ::CharNext(pstrDest);
                        }
                        pstrSource = ::CharNext(pstrSource);
                }
                *pstrDest = _T('\0');
                int nCount = (int)(DWORD_PTR)(pstrSource - pstrDest);
                GetData()->nDataLength -= nCount;

                return nCount;
        }

        // insert character at zero-based index; concatenates if index is past end of string
        int Insert(int nIndex, TCHAR ch)
        {
                CopyBeforeWrite();

                if (nIndex < 0)
                        nIndex = 0;

                int nNewLength = GetData()->nDataLength;
                if (nIndex > nNewLength)
                        nIndex = nNewLength;
                nNewLength++;

                if (GetData()->nAllocLength < nNewLength)
                {
                        CStringData* pOldData = GetData();
                        LPTSTR pstr = m_pchData;
                        if(!AllocBuffer(nNewLength))
                                return -1;
                        SecureHelper::memcpy_x(m_pchData, (nNewLength + 1) * sizeof(TCHAR), pstr, (pOldData->nDataLength + 1) * sizeof(TCHAR));
                        CString::Release(pOldData);
                }

                // move existing bytes down
                SecureHelper::memmove_x(m_pchData + nIndex + 1, (GetData()->nAllocLength - nIndex) * sizeof(TCHAR), m_pchData + nIndex, (nNewLength - nIndex) * sizeof(TCHAR));
                m_pchData[nIndex] = ch;
                GetData()->nDataLength = nNewLength;

                return nNewLength;
        }

        // insert substring at zero-based index; concatenates if index is past end of string
        int Insert(int nIndex, LPCTSTR pstr)
        {
                if (nIndex < 0)
                        nIndex = 0;

                int nInsertLength = SafeStrlen(pstr);
                int nNewLength = GetData()->nDataLength;
                if (nInsertLength > 0)
                {
                        CopyBeforeWrite();
                        if (nIndex > nNewLength)
                                nIndex = nNewLength;
                        nNewLength += nInsertLength;

                        if (GetData()->nAllocLength < nNewLength)
                        {
                                CStringData* pOldData = GetData();
                                LPTSTR pstr = m_pchData;
                                if(!AllocBuffer(nNewLength))
                                        return -1;
                                SecureHelper::memcpy_x(m_pchData, (nNewLength + 1) * sizeof(TCHAR), pstr, (pOldData->nDataLength + 1) * sizeof(TCHAR));
                                CString::Release(pOldData);
                        }

                        // move existing bytes down
                        SecureHelper::memmove_x(m_pchData + nIndex + nInsertLength, (GetData()->nAllocLength + 1 - nIndex - nInsertLength) * sizeof(TCHAR), m_pchData + nIndex, (nNewLength - nIndex - nInsertLength + 1) * sizeof(TCHAR));
                        SecureHelper::memcpy_x(m_pchData + nIndex, (GetData()->nAllocLength + 1 - nIndex) * sizeof(TCHAR), pstr, nInsertLength * sizeof(TCHAR));
                        GetData()->nDataLength = nNewLength;
                }

                return nNewLength;
        }

        // delete nCount characters starting at zero-based index
        int Delete(int nIndex, int nCount = 1)
        {
                if (nIndex < 0)
                        nIndex = 0;
                int nLength = GetData()->nDataLength;
                if (nCount > 0 && nIndex < nLength)
                {
                        if((nIndex + nCount) > nLength)
                                nCount = nLength - nIndex;
                        CopyBeforeWrite();
                        int nBytesToCopy = nLength - (nIndex + nCount) + 1;

                        SecureHelper::memmove_x(m_pchData + nIndex, (GetData()->nAllocLength + 1 - nIndex) * sizeof(TCHAR), m_pchData + nIndex + nCount, nBytesToCopy * sizeof(TCHAR));
                        nLength -= nCount;
                        GetData()->nDataLength = nLength;
                }

                return nLength;
        }

        // searching (return starting index, or -1 if not found)
        // look for a single character match
        int Find(TCHAR ch) const   // like "C" strchr
        {
                return Find(ch, 0);
        }

        int ReverseFind(TCHAR ch) const
        {
                // find last single character
                LPCTSTR lpsz = _cstrrchr(m_pchData, (_TUCHAR)ch);

                // return -1 if not found, distance from beginning otherwise
                return (lpsz == NULL) ? -1 : (int)(lpsz - m_pchData);
        }

        int Find(TCHAR ch, int nStart) const   // starting at index
        {
                int nLength = GetData()->nDataLength;
                if (nStart < 0 || nStart >= nLength)
                        return -1;

                // find first single character
                LPCTSTR lpsz = _cstrchr(m_pchData + nStart, (_TUCHAR)ch);

                // return -1 if not found and index otherwise
                return (lpsz == NULL) ? -1 : (int)(lpsz - m_pchData);
        }

        int FindOneOf(LPCTSTR lpszCharSet) const
        {
                ATLASSERT(_IsValidString(lpszCharSet));
                LPCTSTR lpsz = _cstrpbrk(m_pchData, lpszCharSet);
                return (lpsz == NULL) ? -1 : (int)(lpsz - m_pchData);
        }

        // look for a specific sub-string
        // find a sub-string (like strstr)
        int Find(LPCTSTR lpszSub) const   // like "C" strstr
        {
                return Find(lpszSub, 0);
        }

        int Find(LPCTSTR lpszSub, int nStart) const   // starting at index
        {
                ATLASSERT(_IsValidString(lpszSub));

                int nLength = GetData()->nDataLength;
                if (nStart < 0 || nStart > nLength)
                        return -1;

                // find first matching substring
                LPCTSTR lpsz = _cstrstr(m_pchData + nStart, lpszSub);

                // return -1 for not found, distance from beginning otherwise
                return (lpsz == NULL) ? -1 : (int)(lpsz - m_pchData);
        }

        // Concatentation for non strings
        CString& Append(int n)
        {
                const int cchBuff = 12;
                TCHAR szBuffer[cchBuff] = { 0 };
                SecureHelper::wsprintf_x(szBuffer, cchBuff, _T("%d"), n);
                ConcatInPlace(SafeStrlen(szBuffer), szBuffer);
                return *this;
        }

        // simple formatting
        // formatting (using wsprintf style formatting)
        BOOL __cdecl Format(LPCTSTR lpszFormat, ...)
        {
                ATLASSERT(_IsValidString(lpszFormat));

                va_list argList;
                va_start(argList, lpszFormat);
                BOOL bRet = FormatV(lpszFormat, argList);
                va_end(argList);
                return bRet;
        }

        BOOL __cdecl Format(UINT nFormatID, ...)
        {
                CString strFormat;
                BOOL bRet = strFormat.LoadString(nFormatID);
                ATLASSERT(bRet != 0);

                va_list argList;
                va_start(argList, nFormatID);
                bRet = FormatV(strFormat, argList);
                va_end(argList);
                return bRet;
        }

        BOOL FormatV(LPCTSTR lpszFormat, va_list argList)
        {
                ATLASSERT(_IsValidString(lpszFormat));

                enum _FormatModifiers
                {
                        FORCE_ANSI =    0x10000,
                        FORCE_UNICODE = 0x20000,
                        FORCE_INT64 =   0x40000
                };

                va_list argListSave = argList;

                // make a guess at the maximum length of the resulting string
                int nMaxLen = 0;
                for (LPCTSTR lpsz = lpszFormat; *lpsz != _T('\0'); lpsz = ::CharNext(lpsz))
                {
                        // handle '%' character, but watch out for '%%'
                        if (*lpsz != _T('%') || *(lpsz = ::CharNext(lpsz)) == _T('%'))
                        {
                                nMaxLen += (int)(::CharNext(lpsz) - lpsz);
                                continue;
                        }

                        int nItemLen = 0;

                        // handle '%' character with format
                        int nWidth = 0;
                        for (; *lpsz != _T('\0'); lpsz = ::CharNext(lpsz))
                        {
                                // check for valid flags
                                if (*lpsz == _T('#'))
                                        nMaxLen += 2;   // for '0x'
                                else if (*lpsz == _T('*'))
                                        nWidth = va_arg(argList, int);
                                else if (*lpsz == _T('-') || *lpsz == _T('+') || *lpsz == _T('0') || *lpsz == _T(' '))
                                        ;
                                else // hit non-flag character
                                        break;
                        }
                        // get width and skip it
                        if (nWidth == 0)
                        {
                                // width indicated by
                                nWidth = _cstrtoi(lpsz);
                                for (; *lpsz != _T('\0') && _cstrisdigit(*lpsz); lpsz = ::CharNext(lpsz))
                                        ;
                        }
                        ATLASSERT(nWidth >= 0);

                        int nPrecision = 0;
                        if (*lpsz == _T('.'))
                        {
                                // skip past '.' separator (width.precision)
                                lpsz = ::CharNext(lpsz);

                                // get precision and skip it
                                if (*lpsz == _T('*'))
                                {
                                        nPrecision = va_arg(argList, int);
                                        lpsz = ::CharNext(lpsz);
                                }
                                else
                                {
                                        nPrecision = _cstrtoi(lpsz);
                                        for (; *lpsz != _T('\0') && _cstrisdigit(*lpsz); lpsz = ::CharNext(lpsz))
                                                ;
                                }
                                ATLASSERT(nPrecision >= 0);
                        }

                        // should be on type modifier or specifier
                        int nModifier = 0;
                        if(lpsz[0] == _T('I') && lpsz[1] == _T('6') && lpsz[2] == _T('4'))
                        {
                                lpsz += 3;
                                nModifier = FORCE_INT64;
                        }
                        else
                        {
                                switch (*lpsz)
                                {
                                // modifiers that affect size
                                case _T('h'):
                                        nModifier = FORCE_ANSI;
                                        lpsz = ::CharNext(lpsz);
                                        break;
                                case _T('l'):
                                        nModifier = FORCE_UNICODE;
                                        lpsz = ::CharNext(lpsz);
                                        break;

                                // modifiers that do not affect size
                                case _T('F'):
                                case _T('N'):
                                case _T('L'):
                                        lpsz = ::CharNext(lpsz);
                                        break;
                                }
                        }

                        // now should be on specifier
                        switch (*lpsz | nModifier)
                        {
                        // single characters
                        case _T('c'):
                        case _T('C'):
                                nItemLen = 2;
                                va_arg(argList, TCHAR);
                                break;
                        case _T('c') | FORCE_ANSI:
                        case _T('C') | FORCE_ANSI:
                                nItemLen = 2;
                                va_arg(argList, char);
                                break;
                        case _T('c') | FORCE_UNICODE:
                        case _T('C') | FORCE_UNICODE:
                                nItemLen = 2;
                                va_arg(argList, WCHAR);
                                break;

                        // strings
                        case _T('s'):
                        {
                                LPCTSTR pstrNextArg = va_arg(argList, LPCTSTR);
                                if (pstrNextArg == NULL)
                                {
                                        nItemLen = 6;  // "(null)"
                                }
                                else
                                {
                                        nItemLen = lstrlen(pstrNextArg);
                                        nItemLen = __max(1, nItemLen);
                                }
                                break;
                        }

                        case _T('S'):
                        {
#ifndef _UNICODE
                                LPWSTR pstrNextArg = va_arg(argList, LPWSTR);
                                if (pstrNextArg == NULL)
                                {
                                        nItemLen = 6;  // "(null)"
                                }
                                else
                                {
                                        nItemLen = (int)wcslen(pstrNextArg);
                                        nItemLen = __max(1, nItemLen);
                                }
#else // _UNICODE
                                LPCSTR pstrNextArg = va_arg(argList, LPCSTR);
                                if (pstrNextArg == NULL)
                                {
                                        nItemLen = 6; // "(null)"
                                }
                                else
                                {
#if defined(_WIN32_WCE) && (_ATL_VER >= 0x0800)
                                        nItemLen = ATL::lstrlenA(pstrNextArg);
#else
                                        nItemLen = lstrlenA(pstrNextArg);
#endif
                                        nItemLen = __max(1, nItemLen);
                                }
#endif // _UNICODE
                                break;
                        }

                        case _T('s') | FORCE_ANSI:
                        case _T('S') | FORCE_ANSI:
                        {
                                LPCSTR pstrNextArg = va_arg(argList, LPCSTR);
                                if (pstrNextArg == NULL)
                                {
                                        nItemLen = 6; // "(null)"
                                }
                                else
                                {
#if defined(_WIN32_WCE) && (_ATL_VER >= 0x0800)
                                        nItemLen = ATL::lstrlenA(pstrNextArg);
#else
                                        nItemLen = lstrlenA(pstrNextArg);
#endif
                                        nItemLen = __max(1, nItemLen);
                                }
                                break;
                        }

                        case _T('s') | FORCE_UNICODE:
                        case _T('S') | FORCE_UNICODE:
                        {
                                LPWSTR pstrNextArg = va_arg(argList, LPWSTR);
                                if (pstrNextArg == NULL)
                                {
                                        nItemLen = 6; // "(null)"
                                }
                                else
                                {
                                        nItemLen = (int)wcslen(pstrNextArg);
                                        nItemLen = __max(1, nItemLen);
                                }
                                break;
                        }
                        }

                        // adjust nItemLen for strings
                        if (nItemLen != 0)
                        {
                                nItemLen = __max(nItemLen, nWidth);
                                if (nPrecision != 0)
                                        nItemLen = __min(nItemLen, nPrecision);
                        }
                        else
                        {
                                switch (*lpsz)
                                {
                                // integers
                                case _T('d'):
                                case _T('i'):
                                case _T('u'):
                                case _T('x'):
                                case _T('X'):
                                case _T('o'):
                                        if (nModifier & FORCE_INT64)
                                                va_arg(argList, __int64);
                                        else
                                                va_arg(argList, int);
                                        nItemLen = 32;
                                        nItemLen = __max(nItemLen, nWidth + nPrecision);
                                        break;

#ifndef _ATL_USE_CSTRING_FLOAT
                                case _T('e'):
                                case _T('E'):
                                case _T('f'):
                                case _T('g'):
                                case _T('G'):
                                        ATLASSERT(!"Floating point (%%e, %%E, %%f, %%g, and %%G) is not supported by the WTL::CString class.");
#ifndef _DEBUG
                                        ::OutputDebugString(_T("Floating point (%%e, %%f, %%g, and %%G) is not supported by the WTL::CString class."));
#ifndef _WIN32_WCE
                                        ::DebugBreak();
#else // CE specific
                                        DebugBreak();
#endif // _WIN32_WCE
#endif // !_DEBUG
                                        break;
#else // _ATL_USE_CSTRING_FLOAT
                                case _T('e'):
                                case _T('E'):
                                case _T('g'):
                                case _T('G'):
                                        va_arg(argList, double);
                                        nItemLen = 128;
                                        nItemLen = __max(nItemLen, nWidth + nPrecision);
                                        break;
                                case _T('f'):
                                        {
                                                double f = va_arg(argList, double);
                                                // 312 == strlen("-1+(309 zeroes).")
                                                // 309 zeroes == max precision of a double
                                                // 6 == adjustment in case precision is not specified,
                                                //   which means that the precision defaults to 6
                                                int cchLen = __max(nWidth, 312 + nPrecision + 6);
                                                CTempBuffer<TCHAR, _WTL_STACK_ALLOC_THRESHOLD> buff;
                                                LPTSTR pszTemp = buff.Allocate(cchLen);
                                                if(pszTemp != NULL)
                                                {
                                                        SecureHelper::sprintf_x(pszTemp, cchLen, _T("%*.*f"), nWidth, nPrecision + 6, f);
                                                        nItemLen = (int)_tcslen(pszTemp);
                                                }
                                                else
                                                {
                                                        nItemLen = cchLen;
                                                }
                                        }
                                        break;
#endif // _ATL_USE_CSTRING_FLOAT

                                case _T('p'):
                                        va_arg(argList, void*);
                                        nItemLen = 32;
                                        nItemLen = __max(nItemLen, nWidth + nPrecision);
                                        break;

                                // no output
                                case _T('n'):
                                        va_arg(argList, int*);
                                        break;

                                default:
                                        ATLASSERT(FALSE);  // unknown formatting option
                                }
                        }

                        // adjust nMaxLen for output nItemLen
                        nMaxLen += nItemLen;
                }

                if(GetBuffer(nMaxLen) == NULL)
                        return FALSE;
#ifndef _ATL_USE_CSTRING_FLOAT
                int nRet = SecureHelper::wvsprintf_x(m_pchData, GetAllocLength() + 1, lpszFormat, argListSave);
#else // _ATL_USE_CSTRING_FLOAT
                int nRet = SecureHelper::vsprintf_x(m_pchData, GetAllocLength() + 1, lpszFormat, argListSave);
#endif // _ATL_USE_CSTRING_FLOAT
                nRet;   // ref
                ATLASSERT(nRet <= GetAllocLength());
                ReleaseBuffer();

                va_end(argListSave);
                return TRUE;
        }

        // formatting for localization (uses FormatMessage API)
        // formatting (using FormatMessage style formatting)
        BOOL __cdecl FormatMessage(LPCTSTR lpszFormat, ...)
        {
                // format message into temporary buffer lpszTemp
                va_list argList;
                va_start(argList, lpszFormat);
                LPTSTR lpszTemp;
                BOOL bRet = TRUE;

                if (::FormatMessage(FORMAT_MESSAGE_FROM_STRING | FORMAT_MESSAGE_ALLOCATE_BUFFER,
                                lpszFormat, 0, 0, (LPTSTR)&lpszTemp, 0, &argList) == 0 || lpszTemp == NULL)
                        bRet = FALSE;

                // assign lpszTemp into the resulting string and free the temporary
                *this = lpszTemp;
                LocalFree(lpszTemp);
                va_end(argList);
                return bRet;
        }

        BOOL __cdecl FormatMessage(UINT nFormatID, ...)
        {
                // get format string from string table
                CString strFormat;
                BOOL bRetTmp = strFormat.LoadString(nFormatID);
                bRetTmp;   // ref
                ATLASSERT(bRetTmp != 0);

                // format message into temporary buffer lpszTemp
                va_list argList;
                va_start(argList, nFormatID);
                LPTSTR lpszTemp;
                BOOL bRet = TRUE;

                if (::FormatMessage(FORMAT_MESSAGE_FROM_STRING | FORMAT_MESSAGE_ALLOCATE_BUFFER,
                                strFormat, 0, 0, (LPTSTR)&lpszTemp, 0, &argList) == 0 || lpszTemp == NULL)
                        bRet = FALSE;

                // assign lpszTemp into the resulting string and free lpszTemp
                *this = lpszTemp;
                LocalFree(lpszTemp);
                va_end(argList);
                return bRet;
        }

        // Windows support
        BOOL LoadString(UINT nID)   // load from string resource (255 chars max.)
        {
#ifdef _UNICODE
                const int CHAR_FUDGE = 1;   // one TCHAR unused is good enough
#else
                const int CHAR_FUDGE = 2;   // two BYTES unused for case of DBC last char
#endif

                // try fixed buffer first (to avoid wasting space in the heap)
                TCHAR szTemp[256];
                int nCount =  sizeof(szTemp) / sizeof(szTemp[0]);
                int nLen = _LoadString(nID, szTemp, nCount);
                if (nCount - nLen > CHAR_FUDGE)
                {
                        *this = szTemp;
                        return (nLen > 0);
                }

                // try buffer size of 512, then larger size until entire string is retrieved
                int nSize = 256;
                do
                {
                        nSize += 256;
                        LPTSTR lpstr = GetBuffer(nSize - 1);
                        if(lpstr == NULL)
                        {
                                nLen = 0;
                                break;
                        }
                        nLen = _LoadString(nID, lpstr, nSize);
                } while (nSize - nLen <= CHAR_FUDGE);
                ReleaseBuffer();

                return (nLen > 0);
        }

#ifndef _UNICODE
        // ANSI <-> OEM support (convert string in place)
        void AnsiToOem()
        {
                CopyBeforeWrite();
                ::AnsiToOem(m_pchData, m_pchData);
        }

        void OemToAnsi()
        {
                CopyBeforeWrite();
                ::OemToAnsi(m_pchData, m_pchData);
        }
#endif

#ifndef _ATL_NO_COM
        // OLE BSTR support (use for OLE automation)
        BSTR AllocSysString() const
        {
#if defined(_UNICODE) || defined(OLE2ANSI)
                BSTR bstr = ::SysAllocStringLen(m_pchData, GetData()->nDataLength);
#else
                int nLen = MultiByteToWideChar(CP_ACP, 0, m_pchData,
                        GetData()->nDataLength, NULL, NULL);
                BSTR bstr = ::SysAllocStringLen(NULL, nLen);
                if(bstr != NULL)
                        MultiByteToWideChar(CP_ACP, 0, m_pchData, GetData()->nDataLength, bstr, nLen);
#endif
                return bstr;
        }

        BSTR SetSysString(BSTR* pbstr) const
        {
#if defined(_UNICODE) || defined(OLE2ANSI)
                ::SysReAllocStringLen(pbstr, m_pchData, GetData()->nDataLength);
#else
                int nLen = MultiByteToWideChar(CP_ACP, 0, m_pchData,
                        GetData()->nDataLength, NULL, NULL);
                if(::SysReAllocStringLen(pbstr, NULL, nLen))
                        MultiByteToWideChar(CP_ACP, 0, m_pchData, GetData()->nDataLength, *pbstr, nLen);
#endif
                ATLASSERT(*pbstr != NULL);
                return *pbstr;
        }
#endif // !_ATL_NO_COM

        // Access to string implementation buffer as "C" character array
        LPTSTR GetBuffer(int nMinBufLength)
        {
                ATLASSERT(nMinBufLength >= 0);

                if (GetData()->nRefs > 1 || nMinBufLength > GetData()->nAllocLength)
                {
                        // we have to grow the buffer
                        CStringData* pOldData = GetData();
                        int nOldLen = GetData()->nDataLength;   // AllocBuffer will tromp it
                        if (nMinBufLength < nOldLen)
                                nMinBufLength = nOldLen;

                        if(!AllocBuffer(nMinBufLength))
                                return NULL;

                        SecureHelper::memcpy_x(m_pchData, (nMinBufLength + 1) * sizeof(TCHAR), pOldData->data(), (nOldLen + 1) * sizeof(TCHAR));
                        GetData()->nDataLength = nOldLen;
                        CString::Release(pOldData);
                }
                ATLASSERT(GetData()->nRefs <= 1);

                // return a pointer to the character storage for this string
                ATLASSERT(m_pchData != NULL);
                return m_pchData;
        }

        void ReleaseBuffer(int nNewLength = -1)
        {
                CopyBeforeWrite();   // just in case GetBuffer was not called

                if (nNewLength == -1)
                        nNewLength = lstrlen(m_pchData);   // zero terminated

                ATLASSERT(nNewLength <= GetData()->nAllocLength);
                GetData()->nDataLength = nNewLength;
                m_pchData[nNewLength] = _T('\0');
        }

        LPTSTR GetBufferSetLength(int nNewLength)
        {
                ATLASSERT(nNewLength >= 0);

                if(GetBuffer(nNewLength) == NULL)
                        return NULL;

                GetData()->nDataLength = nNewLength;
                m_pchData[nNewLength] = _T('\0');
                return m_pchData;
        }

        void FreeExtra()
        {
                ATLASSERT(GetData()->nDataLength <= GetData()->nAllocLength);
                if (GetData()->nDataLength != GetData()->nAllocLength)
                {
                        CStringData* pOldData = GetData();
                        if(AllocBuffer(GetData()->nDataLength))
                        {
                                SecureHelper::memcpy_x(m_pchData, (GetData()->nAllocLength + 1) * sizeof(TCHAR), pOldData->data(), pOldData->nDataLength * sizeof(TCHAR));
                                ATLASSERT(m_pchData[GetData()->nDataLength] == _T('\0'));
                                CString::Release(pOldData);
                        }
                }
                ATLASSERT(GetData() != NULL);
        }

        // Use LockBuffer/UnlockBuffer to turn refcounting off
        LPTSTR LockBuffer()
        {
                LPTSTR lpsz = GetBuffer(0);
                if(lpsz != NULL)
                        GetData()->nRefs = -1;
                return lpsz;
        }

        void UnlockBuffer()
        {
                ATLASSERT(GetData()->nRefs == -1);
                if (GetData() != _atltmpDataNil)
                        GetData()->nRefs = 1;
        }

// Implementation
public:
        ~CString()   //  free any attached data
        {
                if (GetData() != _atltmpDataNil)
                {
                        if (InterlockedDecrement(&GetData()->nRefs) <= 0)
                                delete[] (BYTE*)GetData();
                }
        }

        int GetAllocLength() const
        {
                return GetData()->nAllocLength;
        }

        static BOOL __stdcall _IsValidString(LPCTSTR lpsz, int /*nLength*/ = -1)
        {
                return (lpsz != NULL) ? TRUE : FALSE;
        }

protected:
        LPTSTR m_pchData;   // pointer to ref counted string data

        // implementation helpers
        CStringData* GetData() const
        {
                ATLASSERT(m_pchData != NULL);
                return ((CStringData*)m_pchData) - 1;
        }

        void Init()
        {
                m_pchData = _GetEmptyString().m_pchData;
        }

        BOOL AllocCopy(CString& dest, int nCopyLen, int nCopyIndex, int nExtraLen) const
        {
                // will clone the data attached to this string
                // allocating 'nExtraLen' characters
                // Places results in uninitialized string 'dest'
                // Will copy the part or all of original data to start of new string

                BOOL bRet = FALSE;
                int nNewLen = nCopyLen + nExtraLen;
                if (nNewLen == 0)
                {
                        dest.Init();
                        bRet = TRUE;
                }
                else if(nNewLen >= nCopyLen)
                {
                        if(dest.AllocBuffer(nNewLen))
                        {
                                SecureHelper::memcpy_x(dest.m_pchData, (nNewLen + 1) * sizeof(TCHAR), m_pchData + nCopyIndex, nCopyLen * sizeof(TCHAR));
                                bRet = TRUE;
                        }
                }

                return bRet;
        }

        // always allocate one extra character for '\0' termination
        // assumes [optimistically] that data length will equal allocation length
        BOOL AllocBuffer(int nLen)
        {
                ATLASSERT(nLen >= 0);
                ATLASSERT(nLen <= INT_MAX - 1);   // max size (enough room for 1 extra)

                if (nLen == 0)
                {
                        Init();
                }
                else
                {
                        CStringData* pData = NULL;
                        ATLTRY(pData = (CStringData*)new BYTE[sizeof(CStringData) + (nLen + 1) * sizeof(TCHAR)]);
                        if(pData == NULL)
                                return FALSE;

                        pData->nRefs = 1;
                        pData->data()[nLen] = _T('\0');
                        pData->nDataLength = nLen;
                        pData->nAllocLength = nLen;
                        m_pchData = pData->data();
                }

                return TRUE;
        }

        // Assignment operators
        //  All assign a new value to the string
        //      (a) first see if the buffer is big enough
        //      (b) if enough room, copy on top of old buffer, set size and type
        //      (c) otherwise free old string data, and create a new one
        //
        //  All routines return the new string (but as a 'const CString&' so that
        //      assigning it again will cause a copy, eg: s1 = s2 = "hi there".
        //
        void AssignCopy(int nSrcLen, LPCTSTR lpszSrcData)
        {
                if(AllocBeforeWrite(nSrcLen))
                {
                        SecureHelper::memcpy_x(m_pchData, (nSrcLen + 1) * sizeof(TCHAR), lpszSrcData, nSrcLen * sizeof(TCHAR));
                        GetData()->nDataLength = nSrcLen;
                        m_pchData[nSrcLen] = _T('\0');
                }
        }

        // Concatenation
        // NOTE: "operator +" is done as friend functions for simplicity
        //      There are three variants:
        //          CString + CString
        // and for ? = TCHAR, LPCTSTR
        //          CString + ?
        //          ? + CString
        BOOL ConcatCopy(int nSrc1Len, LPCTSTR lpszSrc1Data, int nSrc2Len, LPCTSTR lpszSrc2Data)
        {
                // -- master concatenation routine
                // Concatenate two sources
                // -- assume that 'this' is a new CString object

                BOOL bRet = TRUE;
                int nNewLen = nSrc1Len + nSrc2Len;
                if(nNewLen < nSrc1Len || nNewLen < nSrc2Len)
                {
                        bRet = FALSE;
                }
                else if(nNewLen != 0)
                {
                        bRet = AllocBuffer(nNewLen);
                        if (bRet)
                        {
                                SecureHelper::memcpy_x(m_pchData, (nNewLen + 1) * sizeof(TCHAR), lpszSrc1Data, nSrc1Len * sizeof(TCHAR));
                                SecureHelper::memcpy_x(m_pchData + nSrc1Len, (nNewLen + 1 - nSrc1Len) * sizeof(TCHAR), lpszSrc2Data, nSrc2Len * sizeof(TCHAR));
                        }
                }
                return bRet;
        }

        void ConcatInPlace(int nSrcLen, LPCTSTR lpszSrcData)
        {
                //  -- the main routine for += operators

                // concatenating an empty string is a no-op!
                if (nSrcLen == 0)
                        return;

                // if the buffer is too small, or we have a width mis-match, just
                //   allocate a new buffer (slow but sure)
                if (GetData()->nRefs > 1 || GetData()->nDataLength + nSrcLen > GetData()->nAllocLength)
                {
                        // we have to grow the buffer, use the ConcatCopy routine
                        CStringData* pOldData = GetData();
                        if (ConcatCopy(GetData()->nDataLength, m_pchData, nSrcLen, lpszSrcData))
                        {
                                ATLASSERT(pOldData != NULL);
                                CString::Release(pOldData);
                        }
                }
                else
                {
                        // fast concatenation when buffer big enough
                        SecureHelper::memcpy_x(m_pchData + GetData()->nDataLength, (GetData()->nAllocLength + 1) * sizeof(TCHAR), lpszSrcData, nSrcLen * sizeof(TCHAR));
                        GetData()->nDataLength += nSrcLen;
                        ATLASSERT(GetData()->nDataLength <= GetData()->nAllocLength);
                        m_pchData[GetData()->nDataLength] = _T('\0');
                }
        }

        void CopyBeforeWrite()
        {
                if (GetData()->nRefs > 1)
                {
                        CStringData* pData = GetData();
                        Release();
                        if(AllocBuffer(pData->nDataLength))
                                SecureHelper::memcpy_x(m_pchData, (GetData()->nAllocLength + 1) * sizeof(TCHAR), pData->data(), (pData->nDataLength + 1) * sizeof(TCHAR));
                }
                ATLASSERT(GetData()->nRefs <= 1);
        }

        BOOL AllocBeforeWrite(int nLen)
        {
                BOOL bRet = TRUE;
                if (GetData()->nRefs > 1 || nLen > GetData()->nAllocLength)
                {
                        Release();
                        bRet = AllocBuffer(nLen);
                }
                ATLASSERT(GetData()->nRefs <= 1);
                return bRet;
        }

        void Release()
        {
                if (GetData() != _atltmpDataNil)
                {
                        ATLASSERT(GetData()->nRefs != 0);
                        if (InterlockedDecrement(&GetData()->nRefs) <= 0)
                                delete[] (BYTE*)GetData();
                        Init();
                }
        }

        static void PASCAL Release(CStringData* pData)
        {
                if (pData != _atltmpDataNil)
                {
                        ATLASSERT(pData->nRefs != 0);
                        if (InterlockedDecrement(&pData->nRefs) <= 0)
                                delete[] (BYTE*)pData;
                }
        }

        static int PASCAL SafeStrlen(LPCTSTR lpsz)
        {
                return (lpsz == NULL) ? 0 : lstrlen(lpsz);
        }

        static int __stdcall _LoadString(UINT nID, LPTSTR lpszBuf, UINT nMaxBuf)
        {
#ifdef _DEBUG
                // LoadString without annoying warning from the Debug kernel if the
                //  segment containing the string is not present
                if (::FindResource(ModuleHelper::GetResourceInstance(), MAKEINTRESOURCE((nID >> 4) + 1), RT_STRING) == NULL)
                {
                        lpszBuf[0] = _T('\0');
                        return 0;   // not found
                }
#endif // _DEBUG

                int nLen = ::LoadString(ModuleHelper::GetResourceInstance(), nID, lpszBuf, nMaxBuf);
                if (nLen == 0)
                        lpszBuf[0] = _T('\0');

                return nLen;
        }

        static const CString& __stdcall _GetEmptyString()
        {
                return *(CString*)&_atltmpPchNil;
        }

// CString conversion helpers
        static int __cdecl _wcstombsz(char* mbstr, const wchar_t* wcstr, size_t count)
        {
                if (count == 0 && mbstr != NULL)
                        return 0;

                int result = ::WideCharToMultiByte(CP_ACP, 0, wcstr, -1, mbstr, (int)count, NULL, NULL);
                ATLASSERT(mbstr == NULL || result <= (int)count);
                if (result > 0)
                        mbstr[result - 1] = 0;
                return result;
        }

        static int __cdecl _mbstowcsz(wchar_t* wcstr, const char* mbstr, size_t count)
        {
                if (count == 0 && wcstr != NULL)
                        return 0;

                int result = ::MultiByteToWideChar(CP_ACP, 0, mbstr, -1, wcstr, (int)count);
                ATLASSERT(wcstr == NULL || result <= (int)count);
                if (result > 0)
                        wcstr[result - 1] = 0;
                return result;
        }

// Helpers to avoid CRT startup code
#ifdef _ATL_MIN_CRT
        static const TCHAR* _cstrchr(const TCHAR* p, TCHAR ch)
        {
                // strchr for '\0' should succeed
                while (*p != 0)
                {
                        if (*p == ch)
                                break;
                        p = ::CharNext(p);
                }
                return (*p == ch) ? p : NULL;
        }

        static const TCHAR* _cstrrchr(const TCHAR* p, TCHAR ch)
        {
                const TCHAR* lpsz = NULL;
                while (*p != 0)
                {
                        if (*p == ch)
                                lpsz = p;
                        p = ::CharNext(p);
                }
                return lpsz;
        }

        static TCHAR* _cstrrev(TCHAR* pStr)
        {
                // optimize NULL, zero-length, and single-char case
                if ((pStr == NULL) || (pStr[0] == _T('\0')) || (pStr[1] == _T('\0')))
                        return pStr;

                TCHAR* p = pStr;

                while (*p != 0) 
                {
                        TCHAR* pNext = ::CharNext(p);
                        if(pNext > p + 1)
                        {
                                char p1 = *(char*)p;
                                *(char*)p = *(char*)(p + 1);
                                *(char*)(p + 1) = p1;
                        }
                        p = pNext;
                }

                p--;
                TCHAR* q = pStr;

                while (q < p)
                {
                        TCHAR t = *q;
                        *q = *p;
                        *p = t;
                        q++;
                        p--;
                }
                return pStr;
        }

        static const TCHAR* _cstrstr(const TCHAR* pStr, const TCHAR* pCharSet)
        {
                int nLen = lstrlen(pCharSet);
                if (nLen == 0)
                        return (TCHAR*)pStr;

                const TCHAR* pRet = NULL;
                const TCHAR* pCur = pStr;
                while((pCur = _cstrchr(pCur, *pCharSet)) != NULL)
                {
                        if(memcmp(pCur, pCharSet, nLen * sizeof(TCHAR)) == 0)
                        {
                                pRet = pCur;
                                break;
                        }
                        pCur = ::CharNext(pCur);
                }
                return pRet;
        }

        static int _cstrspn(const TCHAR* pStr, const TCHAR* pCharSet)
        {
                int nRet = 0;
                const TCHAR* p = pStr;
                while (*p != 0)
                {
                        const TCHAR* pNext = ::CharNext(p);
                        if(pNext > p + 1)
                        {
                                if(_cstrchr_db(pCharSet, *p, *(p + 1)) == NULL)
                                        break;
                                nRet += 2;
                        }
                        else
                        {
                                if(_cstrchr(pCharSet, *p) == NULL)
                                        break;
                                nRet++;
                        }
                        p = pNext;
                }
                return nRet;
        }

        static int _cstrcspn(const TCHAR* pStr, const TCHAR* pCharSet)
        {
                int nRet = 0;
                TCHAR* p = (TCHAR*)pStr;
                while (*p != 0)
                {
                        TCHAR* pNext = ::CharNext(p);
                        if(pNext > p + 1)
                        {
                                if(_cstrchr_db(pCharSet, *p, *(p + 1)) != NULL)
                                        break;
                                nRet += 2;
                        }
                        else
                        {
                                if(_cstrchr(pCharSet, *p) != NULL)
                                        break;
                                nRet++;
                        }
                        p = pNext;
                }
                return nRet;
        }

        static const TCHAR* _cstrpbrk(const TCHAR* p, const TCHAR* lpszCharSet)
        {
                int n = _cstrcspn(p, lpszCharSet);
                return (p[n] != 0) ? &p[n] : NULL;
        }

        static int _cstrisdigit(TCHAR ch)
        {
                WORD type;
                GetStringTypeEx(GetThreadLocale(), CT_CTYPE1, &ch, 1, &type);
                return (type & C1_DIGIT) == C1_DIGIT;
        }

        static int _cstrisspace(TCHAR ch)
        {
                WORD type;
                GetStringTypeEx(GetThreadLocale(), CT_CTYPE1, &ch, 1, &type);
                return (type & C1_SPACE) == C1_SPACE;
        }

        static int _cstrcmp(const TCHAR* pstrOne, const TCHAR* pstrOther)
        {
                return lstrcmp(pstrOne, pstrOther);
        }

        static int _cstrcmpi(const TCHAR* pstrOne, const TCHAR* pstrOther)
        {
                return lstrcmpi(pstrOne, pstrOther);
        }

        static int _cstrcoll(const TCHAR* pstrOne, const TCHAR* pstrOther)
        {
                int nRet = CompareString(GetThreadLocale(), 0, pstrOne, -1, pstrOther, -1);
                ATLASSERT(nRet != 0);
                return nRet - 2;   // convert to strcmp convention
        }

        static int _cstrcolli(const TCHAR* pstrOne, const TCHAR* pstrOther)
        {
                int nRet = CompareString(GetThreadLocale(), NORM_IGNORECASE, pstrOne, -1, pstrOther, -1);
                ATLASSERT(nRet != 0);
                return nRet - 2;   // convert to strcmp convention
        }

        static int _cstrtoi(const TCHAR* nptr)
        {
                int c;       // current char
                int total;   // current total
                int sign;    // if '-', then negative, otherwise positive

                while (_cstrisspace(*nptr))
                        ++nptr;

                c = (int)(_TUCHAR)*nptr++;
                sign = c;   // save sign indication
                if (c == _T('-') || c == _T('+'))
                        c = (int)(_TUCHAR)*nptr++;   // skip sign

                total = 0;

                while (_cstrisdigit((TCHAR)c))
                {
                        total = 10 * total + (c - '0');   // accumulate digit
                        c = (int)(_TUCHAR)*nptr++;        // get next char
                }

                if (sign == '-')
                        return -total;
                else
                        return total;   // return result, negated if necessary
        }
#else // !_ATL_MIN_CRT
        static const TCHAR* _cstrchr(const TCHAR* p, TCHAR ch)
        {
                return _tcschr(p, ch);
        }

        static const TCHAR* _cstrrchr(const TCHAR* p, TCHAR ch)
        {
                return _tcsrchr(p, ch);
        }

        static TCHAR* _cstrrev(TCHAR* pStr)
        {
                return _tcsrev(pStr);
        }

        static const TCHAR* _cstrstr(const TCHAR* pStr, const TCHAR* pCharSet)
        {
                return _tcsstr(pStr, pCharSet);
        }

        static int _cstrspn(const TCHAR* pStr, const TCHAR* pCharSet)
        {
                return (int)_tcsspn(pStr, pCharSet);
        }

        static int _cstrcspn(const TCHAR* pStr, const TCHAR* pCharSet)
        {
                return (int)_tcscspn(pStr, pCharSet);
        }

        static const TCHAR* _cstrpbrk(const TCHAR* p, const TCHAR* lpszCharSet)
        {
                return _tcspbrk(p, lpszCharSet);
        }

        static int _cstrisdigit(TCHAR ch)
        {
                return _istdigit(ch);
        }

        static int _cstrisspace(TCHAR ch)
        {
                return _istspace((_TUCHAR)ch);
        }

        static int _cstrcmp(const TCHAR* pstrOne, const TCHAR* pstrOther)
        {
                return _tcscmp(pstrOne, pstrOther);
        }

        static int _cstrcmpi(const TCHAR* pstrOne, const TCHAR* pstrOther)
        {
                return _tcsicmp(pstrOne, pstrOther);
        }

#ifndef _WIN32_WCE
        static int _cstrcoll(const TCHAR* pstrOne, const TCHAR* pstrOther)
        {
                return _tcscoll(pstrOne, pstrOther);
        }

        static int _cstrcolli(const TCHAR* pstrOne, const TCHAR* pstrOther)
        {
                return _tcsicoll(pstrOne, pstrOther);
        }
#endif // !_WIN32_WCE

        static int _cstrtoi(const TCHAR* nptr)
        {
                return _ttoi(nptr);
        }
#endif // !_ATL_MIN_CRT

        static const TCHAR* _cstrchr_db(const TCHAR* p, TCHAR ch1, TCHAR ch2)
        {
                const TCHAR* lpsz = NULL;
                while (*p != 0)
                {
                        if (*p == ch1 && *(p + 1) == ch2)
                        {
                                lpsz = p;
                                break;
                        }
                        p = ::CharNext(p);
                }
                return lpsz;
        }
};


// Compare helpers

inline bool __stdcall operator ==(const CString& s1, const CString& s2)
{ return s1.Compare(s2) == 0; }

inline bool __stdcall operator ==(const CString& s1, LPCTSTR s2)
{ return s1.Compare(s2) == 0; }

inline bool __stdcall operator ==(LPCTSTR s1, const CString& s2)
{ return s2.Compare(s1) == 0; }

inline bool __stdcall operator !=(const CString& s1, const CString& s2)
{ return s1.Compare(s2) != 0; }

inline bool __stdcall operator !=(const CString& s1, LPCTSTR s2)
{ return s1.Compare(s2) != 0; }

inline bool __stdcall operator !=(LPCTSTR s1, const CString& s2)
{ return s2.Compare(s1) != 0; }

inline bool __stdcall operator <(const CString& s1, const CString& s2)
{ return s1.Compare(s2) < 0; }

inline bool __stdcall operator <(const CString& s1, LPCTSTR s2)
{ return s1.Compare(s2) < 0; }

inline bool __stdcall operator <(LPCTSTR s1, const CString& s2)
{ return s2.Compare(s1) > 0; }

inline bool __stdcall operator >(const CString& s1, const CString& s2)
{ return s1.Compare(s2) > 0; }

inline bool __stdcall operator >(const CString& s1, LPCTSTR s2)
{ return s1.Compare(s2) > 0; }

inline bool __stdcall operator >(LPCTSTR s1, const CString& s2)
{ return s2.Compare(s1) < 0; }

inline bool __stdcall operator <=(const CString& s1, const CString& s2)
{ return s1.Compare(s2) <= 0; }

inline bool __stdcall operator <=(const CString& s1, LPCTSTR s2)
{ return s1.Compare(s2) <= 0; }

inline bool __stdcall operator <=(LPCTSTR s1, const CString& s2)
{ return s2.Compare(s1) >= 0; }

inline bool __stdcall operator >=(const CString& s1, const CString& s2)
{ return s1.Compare(s2) >= 0; }

inline bool __stdcall operator >=(const CString& s1, LPCTSTR s2)
{ return s1.Compare(s2) >= 0; }

inline bool __stdcall operator >=(LPCTSTR s1, const CString& s2)
{ return s2.Compare(s1) <= 0; }


// CString "operator +" functions

inline CString __stdcall operator +(const CString& string1, const CString& string2)
{
        CString s;
        s.ConcatCopy(string1.GetData()->nDataLength, string1.m_pchData, string2.GetData()->nDataLength, string2.m_pchData);
        return s;
}

inline CString __stdcall operator +(const CString& string, TCHAR ch)
{
        CString s;
        s.ConcatCopy(string.GetData()->nDataLength, string.m_pchData, 1, &ch);
        return s;
}

inline CString __stdcall operator +(TCHAR ch, const CString& string)
{
        CString s;
        s.ConcatCopy(1, &ch, string.GetData()->nDataLength, string.m_pchData);
        return s;
}

#ifdef _UNICODE
inline CString __stdcall operator +(const CString& string, char ch)
{
        return string + (TCHAR)ch;
}

inline CString __stdcall operator +(char ch, const CString& string)
{
        return (TCHAR)ch + string;
}
#endif // _UNICODE

inline CString __stdcall operator +(const CString& string, LPCTSTR lpsz)
{
        ATLASSERT(lpsz == NULL || CString::_IsValidString(lpsz));
        CString s;
        s.ConcatCopy(string.GetData()->nDataLength, string.m_pchData, CString::SafeStrlen(lpsz), lpsz);
        return s;
}

inline CString __stdcall operator +(LPCTSTR lpsz, const CString& string)
{
        ATLASSERT(lpsz == NULL || CString::_IsValidString(lpsz));
        CString s;
        s.ConcatCopy(CString::SafeStrlen(lpsz), lpsz, string.GetData()->nDataLength, string.m_pchData);
        return s;
}

#endif // !_WTL_NO_CSTRING


///////////////////////////////////////////////////////////////////////////////
// CRecentDocumentList - MRU List Support

#ifndef _WIN32_WCE

#ifndef _WTL_MRUEMPTY_TEXT
  #define _WTL_MRUEMPTY_TEXT    _T("(empty)")
#endif

// forward declaration
inline bool AtlCompactPath(LPTSTR lpstrOut, LPCTSTR lpstrIn, int cchLen);

template <class T, int t_cchItemLen = MAX_PATH, int t_nFirstID = ID_FILE_MRU_FIRST, int t_nLastID = ID_FILE_MRU_LAST>
class CRecentDocumentListBase
{
public:
// Declarations
        struct _DocEntry
        {
                TCHAR szDocName[t_cchItemLen];
                bool operator ==(const _DocEntry& de) const
                { return (lstrcmpi(szDocName, de.szDocName) == 0); }
        };

        enum
        {
                m_nMaxEntries_Min = 2,
                m_nMaxEntries_Max = t_nLastID - t_nFirstID + 1,
                m_cchMaxItemLen_Min = 6,
                m_cchMaxItemLen_Max = t_cchItemLen,
                m_cchItemNameLen = 11
        };

// Data members
        ATL::CSimpleArray<_DocEntry> m_arrDocs;
        int m_nMaxEntries;   // default is 4
        HMENU m_hMenu;

        TCHAR m_szNoEntries[t_cchItemLen];

        int m_cchMaxItemLen;

// Constructor
        CRecentDocumentListBase() : m_hMenu(NULL), m_nMaxEntries(4), m_cchMaxItemLen(-1)
        {
                // These ASSERTs verify values of the template arguments
                ATLASSERT(t_cchItemLen > m_cchMaxItemLen_Min);
                ATLASSERT(m_nMaxEntries_Max > m_nMaxEntries_Min);
        }

// Attributes
        HMENU GetMenuHandle() const
        {
                return m_hMenu;
        }

        void SetMenuHandle(HMENU hMenu)
        {
                ATLASSERT(hMenu == NULL || ::IsMenu(hMenu));
                m_hMenu = hMenu;
                if(m_hMenu == NULL || (::GetMenuString(m_hMenu, t_nFirstID, m_szNoEntries, t_cchItemLen, MF_BYCOMMAND) == 0))
                {
                        T* pT = static_cast<T*>(this);
                        pT;   // avoid level 4 warning
                        SecureHelper::strncpy_x(m_szNoEntries, _countof(m_szNoEntries), pT->GetMRUEmptyText(), _TRUNCATE);
                }
        }

        int GetMaxEntries() const
        {
                return m_nMaxEntries;
        }

        void SetMaxEntries(int nMaxEntries)
        {
                ATLASSERT(nMaxEntries >= m_nMaxEntries_Min && nMaxEntries <= m_nMaxEntries_Max);
                if(nMaxEntries < m_nMaxEntries_Min)
                        nMaxEntries = m_nMaxEntries_Min;
                else if(nMaxEntries > m_nMaxEntries_Max)
                        nMaxEntries = m_nMaxEntries_Max;
                m_nMaxEntries = nMaxEntries;
        }

        int GetMaxItemLength() const
        {
                return m_cchMaxItemLen;
        }

        void SetMaxItemLength(int cchMaxLen)
        {
                ATLASSERT((cchMaxLen >= m_cchMaxItemLen_Min && cchMaxLen <= m_cchMaxItemLen_Max) || cchMaxLen == -1);
                if(cchMaxLen != -1)
                {
                        if(cchMaxLen < m_cchMaxItemLen_Min)
                                cchMaxLen = m_cchMaxItemLen_Min;
                        else if(cchMaxLen > m_cchMaxItemLen_Max)
                                cchMaxLen = m_cchMaxItemLen_Max;
                }
                m_cchMaxItemLen = cchMaxLen;
                T* pT = static_cast<T*>(this);
                pT->UpdateMenu();
        }

// Operations
        BOOL AddToList(LPCTSTR lpstrDocName)
        {
                _DocEntry de;
                errno_t nRet = SecureHelper::strncpy_x(de.szDocName, _countof(de.szDocName), lpstrDocName, _TRUNCATE);
                if(nRet != 0 && nRet != STRUNCATE)
                        return FALSE;

                for(int i = 0; i < m_arrDocs.GetSize(); i++)
                {
                        if(lstrcmpi(m_arrDocs[i].szDocName, lpstrDocName) == 0)
                        {
                                m_arrDocs.RemoveAt(i);
                                break;
                        }
                }

                if(m_arrDocs.GetSize() == m_nMaxEntries)
                        m_arrDocs.RemoveAt(0);

                BOOL bRet = m_arrDocs.Add(de);
                if(bRet)
                {
                        T* pT = static_cast<T*>(this);
                        bRet = pT->UpdateMenu();
                }
                return bRet;
        }

        // This function is deprecated because it is not safe. 
        // Use the version below that accepts the buffer length.
#if (_MSC_VER >= 1300)
        __declspec(deprecated)
#endif
        BOOL GetFromList(int /*nItemID*/, LPTSTR /*lpstrDocName*/)
        {
                ATLASSERT(FALSE);
                return FALSE;
        }

        BOOL GetFromList(int nItemID, LPTSTR lpstrDocName, int cchLength)
        {
                int nIndex = m_arrDocs.GetSize() - (nItemID - t_nFirstID) - 1;
                if(nIndex < 0 || nIndex >= m_arrDocs.GetSize())
                        return FALSE;
                if(lstrlen(m_arrDocs[nIndex].szDocName) >= cchLength)
                        return FALSE;
                SecureHelper::strcpy_x(lpstrDocName, cchLength, m_arrDocs[nIndex].szDocName);

                return TRUE;
        }

#if defined(_WTL_USE_CSTRING) || defined(__ATLSTR_H__)
        BOOL GetFromList(int nItemID, _CSTRING_NS::CString& strDocName)
        {
                int nIndex = m_arrDocs.GetSize() - (nItemID - t_nFirstID) - 1;
                if(nIndex < 0 || nIndex >= m_arrDocs.GetSize())
                        return FALSE;
                strDocName = m_arrDocs[nIndex].szDocName;
                return TRUE;
        }
#endif // defined(_WTL_USE_CSTRING) || defined(__ATLSTR_H__)

        BOOL RemoveFromList(int nItemID)
        {
                int nIndex = m_arrDocs.GetSize() - (nItemID - t_nFirstID) - 1;
                BOOL bRet = m_arrDocs.RemoveAt(nIndex);
                if(bRet)
                {
                        T* pT = static_cast<T*>(this);
                        bRet = pT->UpdateMenu();
                }
                return bRet;
        }

        BOOL MoveToTop(int nItemID)
        {
                int nIndex = m_arrDocs.GetSize() - (nItemID - t_nFirstID) - 1;
                if(nIndex < 0 || nIndex >= m_arrDocs.GetSize())
                        return FALSE;
                _DocEntry de;
                de = m_arrDocs[nIndex];
                m_arrDocs.RemoveAt(nIndex);
                BOOL bRet = m_arrDocs.Add(de);
                if(bRet)
                {
                        T* pT = static_cast<T*>(this);
                        bRet = pT->UpdateMenu();
                }
                return bRet;
        }

        BOOL ReadFromRegistry(LPCTSTR lpstrRegKey)
        {
                T* pT = static_cast<T*>(this);
                ATL::CRegKey rkParent;
                ATL::CRegKey rk;

                LONG lRet = rkParent.Open(HKEY_CURRENT_USER, lpstrRegKey);
                if(lRet != ERROR_SUCCESS)
                        return FALSE;
                lRet = rk.Open(rkParent, pT->GetRegKeyName());
                if(lRet != ERROR_SUCCESS)
                        return FALSE;

                DWORD dwRet = 0;
#if (_ATL_VER >= 0x0700)
                lRet = rk.QueryDWORDValue(pT->GetRegCountName(), dwRet);
#else
                lRet = rk.QueryValue(dwRet, pT->GetRegCountName());
#endif
                if(lRet != ERROR_SUCCESS)
                        return FALSE;
                SetMaxEntries(dwRet);

                m_arrDocs.RemoveAll();

                TCHAR szRetString[t_cchItemLen] = { 0 };
                _DocEntry de;

                for(int nItem = m_nMaxEntries; nItem > 0; nItem--)
                {
                        TCHAR szBuff[m_cchItemNameLen] = { 0 };
                        SecureHelper::wsprintf_x(szBuff, m_cchItemNameLen, pT->GetRegItemName(), nItem);
#if (_ATL_VER >= 0x0700)
                        ULONG ulCount = t_cchItemLen;
                        lRet = rk.QueryStringValue(szBuff, szRetString, &ulCount);
#else
                        DWORD dwCount = t_cchItemLen * sizeof(TCHAR);
                        lRet = rk.QueryValue(szRetString, szBuff, &dwCount);
#endif
                        if(lRet == ERROR_SUCCESS)
                        {
                                SecureHelper::strcpy_x(de.szDocName, _countof(de.szDocName), szRetString);
                                m_arrDocs.Add(de);
                        }
                }

                rk.Close();
                rkParent.Close();

                return pT->UpdateMenu();
        }

        BOOL WriteToRegistry(LPCTSTR lpstrRegKey)
        {
                T* pT = static_cast<T*>(this);
                pT;   // avoid level 4 warning
                ATL::CRegKey rkParent;
                ATL::CRegKey rk;

                LONG lRet = rkParent.Create(HKEY_CURRENT_USER, lpstrRegKey);
                if(lRet != ERROR_SUCCESS)
                        return FALSE;
                lRet = rk.Create(rkParent, pT->GetRegKeyName());
                if(lRet != ERROR_SUCCESS)
                        return FALSE;

#if (_ATL_VER >= 0x0700)
                lRet = rk.SetDWORDValue(pT->GetRegCountName(), m_nMaxEntries);
#else
                lRet = rk.SetValue(m_nMaxEntries, pT->GetRegCountName());
#endif
                ATLASSERT(lRet == ERROR_SUCCESS);

                // set new values
                int nItem;
                for(nItem = m_arrDocs.GetSize(); nItem > 0; nItem--)
                {
                        TCHAR szBuff[m_cchItemNameLen] = { 0 };
                        SecureHelper::wsprintf_x(szBuff, m_cchItemNameLen, pT->GetRegItemName(), nItem);
                        TCHAR szDocName[t_cchItemLen] = { 0 };
                        GetFromList(t_nFirstID + nItem - 1, szDocName, t_cchItemLen);
#if (_ATL_VER >= 0x0700)
                        lRet = rk.SetStringValue(szBuff, szDocName);
#else
                        lRet = rk.SetValue(szDocName, szBuff);
#endif
                        ATLASSERT(lRet == ERROR_SUCCESS);
                }

                // delete unused keys
                for(nItem = m_arrDocs.GetSize() + 1; nItem < m_nMaxEntries_Max; nItem++)
                {
                        TCHAR szBuff[m_cchItemNameLen] = { 0 };
                        SecureHelper::wsprintf_x(szBuff, m_cchItemNameLen, pT->GetRegItemName(), nItem);
                        rk.DeleteValue(szBuff);
                }

                rk.Close();
                rkParent.Close();

                return TRUE;
        }

// Implementation
        BOOL UpdateMenu()
        {
                if(m_hMenu == NULL)
                        return FALSE;
                ATLASSERT(::IsMenu(m_hMenu));

                int nItems = ::GetMenuItemCount(m_hMenu);
                int nInsertPoint;
                for(nInsertPoint = 0; nInsertPoint < nItems; nInsertPoint++)
                {
                        CMenuItemInfo mi;
                        mi.fMask = MIIM_ID;
                        ::GetMenuItemInfo(m_hMenu, nInsertPoint, TRUE, &mi);
                        if (mi.wID == t_nFirstID)
                                break;
                }
                ATLASSERT(nInsertPoint < nItems && "You need a menu item with an ID = t_nFirstID");

                int nItem;
                for(nItem = t_nFirstID; nItem < t_nFirstID + m_nMaxEntries; nItem++)
                {
                        // keep the first one as an insertion point
                        if (nItem != t_nFirstID)
                                ::DeleteMenu(m_hMenu, nItem, MF_BYCOMMAND);
                }

                TCHAR szItemText[t_cchItemLen + 6] = { 0 };   // add space for &, 2 digits, and a space
                int nSize = m_arrDocs.GetSize();
                nItem = 0;
                if(nSize > 0)
                {
                        for(nItem = 0; nItem < nSize; nItem++)
                        {
                                if(m_cchMaxItemLen == -1)
                                {
                                        SecureHelper::wsprintf_x(szItemText, t_cchItemLen + 6, _T("&%i %s"), nItem + 1, m_arrDocs[nSize - 1 - nItem].szDocName);
                                }
                                else
                                {
                                        TCHAR szBuff[t_cchItemLen] = { 0 };
                                        T* pT = static_cast<T*>(this);
                                        pT;   // avoid level 4 warning
                                        bool bRet = pT->CompactDocumentName(szBuff, m_arrDocs[nSize - 1 - nItem].szDocName, m_cchMaxItemLen);
                                        bRet;   // avoid level 4 warning
                                        ATLASSERT(bRet);
                                        SecureHelper::wsprintf_x(szItemText, t_cchItemLen + 6, _T("&%i %s"), nItem + 1, szBuff);
                                }
                                ::InsertMenu(m_hMenu, nInsertPoint + nItem, MF_BYPOSITION | MF_STRING, t_nFirstID + nItem, szItemText);
                        }
                }
                else    // empty
                {
                        ::InsertMenu(m_hMenu, nInsertPoint, MF_BYPOSITION | MF_STRING, t_nFirstID, m_szNoEntries);
                        ::EnableMenuItem(m_hMenu, t_nFirstID, MF_GRAYED);
                        nItem++;
                }
                ::DeleteMenu(m_hMenu, nInsertPoint + nItem, MF_BYPOSITION);

                return TRUE;
        }

// Overrideables
        // override to provide a different method of compacting document names
        static bool CompactDocumentName(LPTSTR lpstrOut, LPCTSTR lpstrIn, int cchLen)
        {
                return AtlCompactPath(lpstrOut, lpstrIn, cchLen);
        }

        static LPCTSTR GetRegKeyName()
        {
                return _T("Recent Document List");
        }

        static LPCTSTR GetRegCountName()
        {
                return _T("DocumentCount");
        }

        static LPCTSTR GetRegItemName()
        {
                // Note: This string is a format string used with wsprintf().
                // Resulting formatted string must be m_cchItemNameLen or less 
                // characters long, including the terminating null character.
                return _T("Document%i");
        }

        static LPCTSTR GetMRUEmptyText()
        {
                return _WTL_MRUEMPTY_TEXT;
        }
};

class CRecentDocumentList : public CRecentDocumentListBase<CRecentDocumentList>
{
public:
// nothing here
};

#endif // _WIN32_WCE


///////////////////////////////////////////////////////////////////////////////
// CFindFile - file search helper class

class CFindFile
{
public:
// Data members
        WIN32_FIND_DATA m_fd;
        TCHAR m_lpszRoot[MAX_PATH];
        TCHAR m_chDirSeparator;
        HANDLE m_hFind;
        BOOL m_bFound;

// Constructor/destructor
        CFindFile() : m_hFind(NULL), m_chDirSeparator(_T('\\')), m_bFound(FALSE)
        { }

        ~CFindFile()
        {
                Close();
        }

// Attributes
        ULONGLONG GetFileSize() const
        {
                ATLASSERT(m_hFind != NULL);

                ULARGE_INTEGER nFileSize = { 0 };

                if(m_bFound)
                {
                        nFileSize.LowPart = m_fd.nFileSizeLow;
                        nFileSize.HighPart = m_fd.nFileSizeHigh;
                }
                else
                {
                        nFileSize.QuadPart = 0;
                }

                return nFileSize.QuadPart;
        }

        BOOL GetFileName(LPTSTR lpstrFileName, int cchLength) const
        {
                ATLASSERT(m_hFind != NULL);
                if(lstrlen(m_fd.cFileName) >= cchLength)
                        return FALSE;

                if(m_bFound)
                        SecureHelper::strcpy_x(lpstrFileName, cchLength, m_fd.cFileName);

                return m_bFound;
        }

        BOOL GetFilePath(LPTSTR lpstrFilePath, int cchLength) const
        {
                ATLASSERT(m_hFind != NULL);

                int nLen = lstrlen(m_lpszRoot);
#ifndef _WIN32_WCE
                ATLASSERT(nLen > 0);
                if(nLen == 0)
                        return FALSE;

                bool bAddSep = (m_lpszRoot[nLen - 1] != _T('\\') && m_lpszRoot[nLen - 1] !=_T('/'));
#else // CE specific
                // allow diskless devices (nLen == 0)
                bool bAddSep = ((nLen == 0) || (m_lpszRoot[nLen - 1] != _T('\\') && m_lpszRoot[nLen - 1] !=_T('/')));
#endif // _WIN32_WCE

                if((lstrlen(m_lpszRoot) + (bAddSep ?  1 : 0)) >= cchLength)
                        return FALSE;

                SecureHelper::strcpy_x(lpstrFilePath, cchLength, m_lpszRoot);

                if(bAddSep)
                {
                        TCHAR szSeparator[2] = { m_chDirSeparator, 0 };
                        SecureHelper::strcat_x(lpstrFilePath, cchLength, szSeparator);
                }

                SecureHelper::strcat_x(lpstrFilePath, cchLength, m_fd.cFileName);

                return TRUE;
        }

#ifndef _WIN32_WCE
        BOOL GetFileTitle(LPTSTR lpstrFileTitle, int cchLength) const
        {
                ATLASSERT(m_hFind != NULL);

                TCHAR szBuff[MAX_PATH] = { 0 };
                if(!GetFileName(szBuff, MAX_PATH))
                        return FALSE;

                if(lstrlen(szBuff) >= cchLength || cchLength < 1)
                        return FALSE;

                // find the last dot
                LPTSTR pstrDot  = (LPTSTR)_cstrrchr(szBuff, _T('.'));
                if(pstrDot != NULL)
                        *pstrDot = 0;

                SecureHelper::strcpy_x(lpstrFileTitle, cchLength, szBuff);

                return TRUE;
        }
#endif // !_WIN32_WCE

        BOOL GetFileURL(LPTSTR lpstrFileURL, int cchLength) const
        {
                ATLASSERT(m_hFind != NULL);

                TCHAR szBuff[MAX_PATH] = { 0 };
                if(!GetFilePath(szBuff, MAX_PATH))
                        return FALSE;
                LPCTSTR lpstrFileURLPrefix = _T("file://");
                if(lstrlen(szBuff) + lstrlen(lpstrFileURLPrefix) >= cchLength)
                        return FALSE;
                SecureHelper::strcpy_x(lpstrFileURL, cchLength, lpstrFileURLPrefix);
                SecureHelper::strcat_x(lpstrFileURL, cchLength, szBuff);

                return TRUE;
        }

        BOOL GetRoot(LPTSTR lpstrRoot, int cchLength) const
        {
                ATLASSERT(m_hFind != NULL);
                if(lstrlen(m_lpszRoot) >= cchLength)
                        return FALSE;

                SecureHelper::strcpy_x(lpstrRoot, cchLength, m_lpszRoot);

                return TRUE;
        }

#if defined(_WTL_USE_CSTRING) || defined(__ATLSTR_H__)
        _CSTRING_NS::CString GetFileName() const
        {
                ATLASSERT(m_hFind != NULL);

                _CSTRING_NS::CString ret;

                if(m_bFound)
                        ret = m_fd.cFileName;
                return ret;
        }

        _CSTRING_NS::CString GetFilePath() const
        {
                ATLASSERT(m_hFind != NULL);

                _CSTRING_NS::CString strResult = m_lpszRoot;
                int nLen = strResult.GetLength();
#ifndef _WIN32_WCE
                ATLASSERT(nLen > 0);
                if(nLen == 0)
                        return strResult;

                if((strResult[nLen - 1] != _T('\\')) && (strResult[nLen - 1] != _T('/')))
#else // CE specific
                // allow diskless devices (nLen == 0)
                if((nLen == 0) || ((strResult[nLen - 1] != _T('\\')) && (strResult[nLen - 1] != _T('/'))))
#endif // _WIN32_WCE
                        strResult += m_chDirSeparator;
                strResult += GetFileName();
                return strResult;
        }

#ifndef _WIN32_WCE
        _CSTRING_NS::CString GetFileTitle() const
        {
                ATLASSERT(m_hFind != NULL);

                _CSTRING_NS::CString strResult;
                GetFileTitle(strResult.GetBuffer(MAX_PATH), MAX_PATH);
                strResult.ReleaseBuffer();

                return strResult;
        }
#endif // !_WIN32_WCE

        _CSTRING_NS::CString GetFileURL() const
        {
                ATLASSERT(m_hFind != NULL);

                _CSTRING_NS::CString strResult("file://");
                strResult += GetFilePath();
                return strResult;
        }

        _CSTRING_NS::CString GetRoot() const
        {
                ATLASSERT(m_hFind != NULL);

                _CSTRING_NS::CString str = m_lpszRoot;
                return str;
        }
#endif // defined(_WTL_USE_CSTRING) || defined(__ATLSTR_H__)

        BOOL GetLastWriteTime(FILETIME* pTimeStamp) const
        {
                ATLASSERT(m_hFind != NULL);
                ATLASSERT(pTimeStamp != NULL);

                if(m_bFound && pTimeStamp != NULL)
                {
                        *pTimeStamp = m_fd.ftLastWriteTime;
                        return TRUE;
                }

                return FALSE;
        }

        BOOL GetLastAccessTime(FILETIME* pTimeStamp) const
        {
                ATLASSERT(m_hFind != NULL);
                ATLASSERT(pTimeStamp != NULL);

                if(m_bFound && pTimeStamp != NULL)
                {
                        *pTimeStamp = m_fd.ftLastAccessTime;
                        return TRUE;
                }

                return FALSE;
        }

        BOOL GetCreationTime(FILETIME* pTimeStamp) const
        {
                ATLASSERT(m_hFind != NULL);

                if(m_bFound && pTimeStamp != NULL)
                {
                        *pTimeStamp = m_fd.ftCreationTime;
                        return TRUE;
                }

                return FALSE;
        }

        BOOL MatchesMask(DWORD dwMask) const
        {
                ATLASSERT(m_hFind != NULL);

                if(m_bFound)
                        return ((m_fd.dwFileAttributes & dwMask) != 0);

                return FALSE;
        }

        BOOL IsDots() const
        {
                ATLASSERT(m_hFind != NULL);

                // return TRUE if the file name is "." or ".." and
                // the file is a directory

                BOOL bResult = FALSE;
                if(m_bFound && IsDirectory())
                {
                        if(m_fd.cFileName[0] == _T('.') && (m_fd.cFileName[1] == _T('\0') || (m_fd.cFileName[1] == _T('.') && m_fd.cFileName[2] == _T('\0'))))
                                bResult = TRUE;
                }

                return bResult;
        }

        BOOL IsReadOnly() const
        {
                return MatchesMask(FILE_ATTRIBUTE_READONLY);
        }

        BOOL IsDirectory() const
        {
                return MatchesMask(FILE_ATTRIBUTE_DIRECTORY);
        }

        BOOL IsCompressed() const
        {
                return MatchesMask(FILE_ATTRIBUTE_COMPRESSED);
        }

        BOOL IsSystem() const
        {
                return MatchesMask(FILE_ATTRIBUTE_SYSTEM);
        }

        BOOL IsHidden() const
        {
                return MatchesMask(FILE_ATTRIBUTE_HIDDEN);
        }

        BOOL IsTemporary() const
        {
                return MatchesMask(FILE_ATTRIBUTE_TEMPORARY);
        }

        BOOL IsNormal() const
        {
                return MatchesMask(FILE_ATTRIBUTE_NORMAL);
        }

        BOOL IsArchived() const
        {
                return MatchesMask(FILE_ATTRIBUTE_ARCHIVE);
        }

// Operations
        BOOL FindFile(LPCTSTR pstrName = NULL)
        {
                Close();

                if(pstrName == NULL)
                {
                        pstrName = _T("*.*");
                }
                else if(lstrlen(pstrName) >= MAX_PATH)
                {
                        ATLASSERT(FALSE);
                        return FALSE;
                }

                SecureHelper::strcpy_x(m_fd.cFileName, _countof(m_fd.cFileName), pstrName);

                m_hFind = ::FindFirstFile(pstrName, &m_fd);

                if(m_hFind == INVALID_HANDLE_VALUE)
                        return FALSE;

#ifndef _WIN32_WCE
                bool bFullPath = (::GetFullPathName(pstrName, MAX_PATH, m_lpszRoot, NULL) != 0);
#else // CE specific
                errno_t nRet = SecureHelper::strncpy_x(m_lpszRoot, _countof(m_lpszRoot), pstrName, _TRUNCATE);
                bool bFullPath = (nRet == 0 || nRet == STRUNCATE);
#endif // _WIN32_WCE

                // passed name isn't a valid path but was found by the API
                ATLASSERT(bFullPath);
                if(!bFullPath)
                {
                        Close();
                        ::SetLastError(ERROR_INVALID_NAME);
                        return FALSE;
                }
                else
                {
                        // find the last forward or backward whack
                        LPTSTR pstrBack  = (LPTSTR)_cstrrchr(m_lpszRoot, _T('\\'));
                        LPTSTR pstrFront = (LPTSTR)_cstrrchr(m_lpszRoot, _T('/'));

                        if(pstrFront != NULL || pstrBack != NULL)
                        {
                                if(pstrFront == NULL)
                                        pstrFront = m_lpszRoot;
                                if(pstrBack == NULL)
                                        pstrBack = m_lpszRoot;

                                // from the start to the last whack is the root

                                if(pstrFront >= pstrBack)
                                        *pstrFront = _T('\0');
                                else
                                        *pstrBack = _T('\0');
                        }
                }

                m_bFound = TRUE;

                return TRUE;
        }

        BOOL FindNextFile()
        {
                ATLASSERT(m_hFind != NULL);

                if(m_hFind == NULL)
                        return FALSE;

                if(!m_bFound)
                        return FALSE;

                m_bFound = ::FindNextFile(m_hFind, &m_fd);

                return m_bFound;
        }

        void Close()
        {
                m_bFound = FALSE;

                if(m_hFind != NULL && m_hFind != INVALID_HANDLE_VALUE)
                {
                        ::FindClose(m_hFind);
                        m_hFind = NULL;
                }
        }

// Helper
        static const TCHAR* _cstrrchr(const TCHAR* p, TCHAR ch)
        {
#ifdef _ATL_MIN_CRT
                const TCHAR* lpsz = NULL;
                while (*p != 0)
                {
                        if (*p == ch)
                                lpsz = p;
                        p = ::CharNext(p);
                }
                return lpsz;
#else // !_ATL_MIN_CRT
                return _tcsrchr(p, ch);
#endif // !_ATL_MIN_CRT
        }
};


///////////////////////////////////////////////////////////////////////////////
// Global functions for loading resources

inline HACCEL AtlLoadAccelerators(ATL::_U_STRINGorID table)
{
        return ::LoadAccelerators(ModuleHelper::GetResourceInstance(), table.m_lpstr);
}

inline HMENU AtlLoadMenu(ATL::_U_STRINGorID menu)
{
        return ::LoadMenu(ModuleHelper::GetResourceInstance(), menu.m_lpstr);
}

inline HBITMAP AtlLoadBitmap(ATL::_U_STRINGorID bitmap)
{
        return ::LoadBitmap(ModuleHelper::GetResourceInstance(), bitmap.m_lpstr);
}

#ifdef OEMRESOURCE
inline HBITMAP AtlLoadSysBitmap(ATL::_U_STRINGorID bitmap)
{
#ifdef _DEBUG
        WORD wID = (WORD)bitmap.m_lpstr;
        ATLASSERT(wID >= 32734 && wID <= 32767);
#endif // _DEBUG
        return ::LoadBitmap(NULL, bitmap.m_lpstr);
}
#endif // OEMRESOURCE

inline HCURSOR AtlLoadCursor(ATL::_U_STRINGorID cursor)
{
        return ::LoadCursor(ModuleHelper::GetResourceInstance(), cursor.m_lpstr);
}

inline HCURSOR AtlLoadSysCursor(LPCTSTR lpCursorName)
{
#if (WINVER >= 0x0500)
        ATLASSERT(lpCursorName == IDC_ARROW || lpCursorName == IDC_IBEAM || lpCursorName == IDC_WAIT ||
                lpCursorName == IDC_CROSS || lpCursorName == IDC_UPARROW || lpCursorName == IDC_SIZE ||
                lpCursorName == IDC_ICON || lpCursorName == IDC_SIZENWSE || lpCursorName == IDC_SIZENESW ||
                lpCursorName == IDC_SIZEWE || lpCursorName == IDC_SIZENS || lpCursorName == IDC_SIZEALL ||
                lpCursorName == IDC_NO || lpCursorName == IDC_APPSTARTING || lpCursorName == IDC_HELP ||
                lpCursorName == IDC_HAND);
#else // !(WINVER >= 0x0500)
        ATLASSERT(lpCursorName == IDC_ARROW || lpCursorName == IDC_IBEAM || lpCursorName == IDC_WAIT ||
                lpCursorName == IDC_CROSS || lpCursorName == IDC_UPARROW || lpCursorName == IDC_SIZE ||
                lpCursorName == IDC_ICON || lpCursorName == IDC_SIZENWSE || lpCursorName == IDC_SIZENESW ||
                lpCursorName == IDC_SIZEWE || lpCursorName == IDC_SIZENS || lpCursorName == IDC_SIZEALL ||
                lpCursorName == IDC_NO || lpCursorName == IDC_APPSTARTING || lpCursorName == IDC_HELP);
#endif // !(WINVER >= 0x0500)
        return ::LoadCursor(NULL, lpCursorName);
}

inline HICON AtlLoadIcon(ATL::_U_STRINGorID icon)
{
        return ::LoadIcon(ModuleHelper::GetResourceInstance(), icon.m_lpstr);
}

#ifndef _WIN32_WCE
inline HICON AtlLoadSysIcon(LPCTSTR lpIconName)
{
#if (WINVER >= 0x0600)
        ATLASSERT(lpIconName == IDI_APPLICATION || lpIconName == IDI_ASTERISK || lpIconName == IDI_EXCLAMATION ||
                  lpIconName == IDI_HAND || lpIconName == IDI_QUESTION || lpIconName == IDI_WINLOGO ||
                  lpIconName == IDI_SHIELD);
#else // !(WINVER >= 0x0600)
        ATLASSERT(lpIconName == IDI_APPLICATION || lpIconName == IDI_ASTERISK || lpIconName == IDI_EXCLAMATION ||
                  lpIconName == IDI_HAND || lpIconName == IDI_QUESTION || lpIconName == IDI_WINLOGO);
#endif // !(WINVER >= 0x0600)
        return ::LoadIcon(NULL, lpIconName);
}
#endif // !_WIN32_WCE

inline HBITMAP AtlLoadBitmapImage(ATL::_U_STRINGorID bitmap, UINT fuLoad = LR_DEFAULTCOLOR)
{
        return (HBITMAP)::LoadImage(ModuleHelper::GetResourceInstance(), bitmap.m_lpstr, IMAGE_BITMAP, 0, 0, fuLoad);
}

inline HCURSOR AtlLoadCursorImage(ATL::_U_STRINGorID cursor, UINT fuLoad = LR_DEFAULTCOLOR | LR_DEFAULTSIZE, int cxDesired = 0, int cyDesired = 0)
{
        return (HCURSOR)::LoadImage(ModuleHelper::GetResourceInstance(), cursor.m_lpstr, IMAGE_CURSOR, cxDesired, cyDesired, fuLoad);
}

inline HICON AtlLoadIconImage(ATL::_U_STRINGorID icon, UINT fuLoad = LR_DEFAULTCOLOR | LR_DEFAULTSIZE, int cxDesired = 0, int cyDesired = 0)
{
        return (HICON)::LoadImage(ModuleHelper::GetResourceInstance(), icon.m_lpstr, IMAGE_ICON, cxDesired, cyDesired, fuLoad);
}

#ifdef OEMRESOURCE
inline HBITMAP AtlLoadSysBitmapImage(WORD wBitmapID, UINT fuLoad = LR_DEFAULTCOLOR)
{
        ATLASSERT(wBitmapID >= 32734 && wBitmapID <= 32767);
        ATLASSERT((fuLoad & LR_LOADFROMFILE) == 0);   // this one doesn't load from a file
        return (HBITMAP)::LoadImage(NULL, MAKEINTRESOURCE(wBitmapID), IMAGE_BITMAP, 0, 0, fuLoad);
}
#endif // OEMRESOURCE

inline HCURSOR AtlLoadSysCursorImage(ATL::_U_STRINGorID cursor, UINT fuLoad = LR_DEFAULTCOLOR | LR_DEFAULTSIZE, int cxDesired = 0, int cyDesired = 0)
{
#ifdef _DEBUG
        WORD wID = (WORD)cursor.m_lpstr;
        ATLASSERT((wID >= 32512 && wID <= 32516) || (wID >= 32640 && wID <= 32648) || (wID == 32650) || (wID == 32651));
        ATLASSERT((fuLoad & LR_LOADFROMFILE) == 0);   // this one doesn't load from a file
#endif // _DEBUG
        return (HCURSOR)::LoadImage(NULL, cursor.m_lpstr, IMAGE_CURSOR, cxDesired, cyDesired, fuLoad);
}

inline HICON AtlLoadSysIconImage(ATL::_U_STRINGorID icon, UINT fuLoad = LR_DEFAULTCOLOR | LR_DEFAULTSIZE, int cxDesired = 0, int cyDesired = 0)
{
#ifdef _DEBUG
        WORD wID = (WORD)icon.m_lpstr;
        ATLASSERT(wID >= 32512 && wID <= 32517);
        ATLASSERT((fuLoad & LR_LOADFROMFILE) == 0);   // this one doesn't load from a file
#endif // _DEBUG
        return (HICON)::LoadImage(NULL, icon.m_lpstr, IMAGE_ICON, cxDesired, cyDesired, fuLoad);
}

#if (_ATL_VER < 0x0700)
inline int AtlLoadString(UINT uID, LPTSTR lpBuffer, int nBufferMax)
{
        return ::LoadString(_Module.GetResourceInstance(), uID, lpBuffer, nBufferMax);
}
#endif // (_ATL_VER < 0x0700)

#ifdef _WIN32_WCE // CE only direct access to the resource
inline LPCTSTR AtlLoadString(UINT uID)
{
        LPCTSTR s = (LPCTSTR)::LoadString(ModuleHelper::GetResourceInstance(), uID, NULL, 0);
#ifdef DEBUG // Check for null-termination
        if(s != NULL)
                // Note: RC -n <file.rc> compiles null-terminated resource strings
                ATLASSERT(s[*((WORD*)s -1) - 1] == L'\0');
#endif
        return s;
}
#endif // _WIN32_WCE

inline bool AtlLoadString(UINT uID, BSTR& bstrText)
{
        USES_CONVERSION;
        ATLASSERT(bstrText == NULL);

        LPTSTR lpstrText = NULL;
        int nRes = 0;
        for(int nLen = 256; ; nLen *= 2)
        {
                ATLTRY(lpstrText = new TCHAR[nLen]);
                if(lpstrText == NULL)
                        break;
                nRes = ::LoadString(ModuleHelper::GetResourceInstance(), uID, lpstrText, nLen);
                if(nRes < nLen - 1)
                        break;
                delete [] lpstrText;
                lpstrText = NULL;
        }

        if(lpstrText != NULL)
        {
                if(nRes != 0)
                        bstrText = ::SysAllocString(T2OLE(lpstrText));
                delete [] lpstrText;
        }

        return (bstrText != NULL) ? true : false;
}


///////////////////////////////////////////////////////////////////////////////
// Global functions for stock GDI objects

inline HPEN AtlGetStockPen(int nPen)
{
#if (_WIN32_WINNT >= 0x0500) && !defined(_WIN32_WCE)
        ATLASSERT(nPen == WHITE_PEN || nPen == BLACK_PEN || nPen == NULL_PEN || nPen == DC_PEN);
#else
        ATLASSERT(nPen == WHITE_PEN || nPen == BLACK_PEN || nPen == NULL_PEN);
#endif
        return (HPEN)::GetStockObject(nPen);
}

inline HBRUSH AtlGetStockBrush(int nBrush)
{
#if (_WIN32_WINNT >= 0x0500) && !defined(_WIN32_WCE)
        ATLASSERT((nBrush >= WHITE_BRUSH && nBrush <= HOLLOW_BRUSH) || nBrush == DC_BRUSH);
#else
        ATLASSERT(nBrush >= WHITE_BRUSH && nBrush <= HOLLOW_BRUSH);
#endif
        return (HBRUSH)::GetStockObject(nBrush);
}

inline HFONT AtlGetStockFont(int nFont)
{
#ifndef _WIN32_WCE
        ATLASSERT((nFont >= OEM_FIXED_FONT && nFont <= SYSTEM_FIXED_FONT) || nFont == DEFAULT_GUI_FONT);
#else // CE specific
        ATLASSERT(nFont == SYSTEM_FONT);
#endif // _WIN32_WCE
        return (HFONT)::GetStockObject(nFont);
}

inline HPALETTE AtlGetStockPalette(int nPalette)
{
        ATLASSERT(nPalette == DEFAULT_PALETTE); // the only one supported
        return (HPALETTE)::GetStockObject(nPalette);
}


///////////////////////////////////////////////////////////////////////////////
// Global function for compacting a path by replacing parts with ellipsis

// helper for multi-byte character sets
inline bool _IsDBCSTrailByte(LPCTSTR lpstr, int nChar)
{
#ifndef _UNICODE
        int i = nChar;
        for( ; i > 0; i--)
        {
                if(!::IsDBCSLeadByte(lpstr[i - 1]))
                        break;
        }
        return ((nChar > 0) && (((nChar - i) & 1) != 0));
#else // _UNICODE
        lpstr; nChar;
        return false;
#endif // _UNICODE
}

inline bool AtlCompactPath(LPTSTR lpstrOut, LPCTSTR lpstrIn, int cchLen)
{
        ATLASSERT(lpstrOut != NULL);
        ATLASSERT(lpstrIn != NULL);
        ATLASSERT(cchLen > 0);

        LPCTSTR szEllipsis = _T("...");
        const int cchEndEllipsis = 3;
        const int cchMidEllipsis = 4;

        if(lstrlen(lpstrIn) < cchLen)
        {
                SecureHelper::strcpy_x(lpstrOut, cchLen, lpstrIn);
                return true;
        }

        lpstrOut[0] = 0;

        // check if the separator is a slash or a backslash
        TCHAR chSlash = _T('\\');
        for(LPTSTR lpstr = (LPTSTR)lpstrIn; *lpstr != 0; lpstr = ::CharNext(lpstr))
        {
                if((*lpstr == _T('/')) || (*lpstr == _T('\\')))
                        chSlash = *lpstr;
        }

        // find the filename portion of the path
        LPCTSTR lpstrFileName = lpstrIn;
        for(LPCTSTR pPath = lpstrIn; *pPath; pPath = ::CharNext(pPath))
        {
                if((pPath[0] == _T('\\') || pPath[0] == _T(':') || pPath[0] == _T('/'))
                                && pPath[1] && pPath[1] != _T('\\') && pPath[1] != _T('/'))
                        lpstrFileName = pPath + 1;
        }
        int cchFileName = lstrlen(lpstrFileName);

        // handle just the filename without a path
        if(lpstrFileName == lpstrIn && cchLen > cchEndEllipsis)
        {
                bool bRet = (SecureHelper::strncpy_x(lpstrOut, cchLen, lpstrIn, cchLen - cchEndEllipsis - 1) == 0);
                if(bRet)
                {
#ifndef _UNICODE
                        if(_IsDBCSTrailByte(lpstrIn, cchLen - cchEndEllipsis))
                                lpstrOut[cchLen - cchEndEllipsis - 1] = 0;
#endif // _UNICODE
                        SecureHelper::strcat_x(lpstrOut, cchLen, szEllipsis);
                }
                return bRet;
        }

        // handle just ellipsis
        if((cchLen < (cchMidEllipsis + cchEndEllipsis)))
        {
                for(int i = 0; i < cchLen - 1; i++)
                        lpstrOut[i] = ((i + 1) == cchMidEllipsis) ? chSlash : _T('.');
                lpstrOut[cchLen - 1] = 0;
                return true;
        }

        // calc how much we have to copy
        int cchToCopy = cchLen - (cchMidEllipsis + cchFileName) - 1;

        if(cchToCopy < 0)
                cchToCopy = 0;

#ifndef _UNICODE
        if(cchToCopy > 0 && _IsDBCSTrailByte(lpstrIn, cchToCopy))
                cchToCopy--;
#endif // _UNICODE

        bool bRet = (SecureHelper::strncpy_x(lpstrOut, cchLen, lpstrIn, cchToCopy) == 0);
        if(!bRet)
                return false;

        // add ellipsis
        SecureHelper::strcat_x(lpstrOut, cchLen, szEllipsis);
        if(!bRet)
                return false;
        TCHAR szSlash[2] = { chSlash, 0 };
        SecureHelper::strcat_x(lpstrOut, cchLen, szSlash);
        if(!bRet)
                return false;

        // add filename (and ellipsis, if needed)
        if(cchLen > (cchMidEllipsis + cchFileName))
        {
                SecureHelper::strcat_x(lpstrOut, cchLen, lpstrFileName);
        }
        else
        {
                cchToCopy = cchLen - cchMidEllipsis - cchEndEllipsis - 1;
#ifndef _UNICODE
                if(cchToCopy > 0 && _IsDBCSTrailByte(lpstrFileName, cchToCopy))
                        cchToCopy--;
#endif // _UNICODE
                bRet = (SecureHelper::strncpy_x(&lpstrOut[cchMidEllipsis], cchLen - cchMidEllipsis, lpstrFileName, cchToCopy) == 0);
                if(bRet)
                        SecureHelper::strcat_x(lpstrOut, cchLen, szEllipsis);
        }

        return bRet;
}

}; // namespace WTL

#endif // __ATLMISC_H__