Category

Development Platform
Home > SourceCode/Document > Windows Develop > Windows Kernel > View Code
strret.cpp
Package: shell.rar　[view]
Upload User: xhy777
Upload Date: 2007-02-14
Package Size: 24088k
Code Size: 3k
Category:

Windows Kernel

Development Platform:

Visual C++

strret.cpp：Code Content
  1. //+-------------------------------------------------------------------------
  2. //
  3. //  Microsoft Windows
  4. //
  5. //  Copyright (C) Microsoft Corporation, 1997 - 1999
  6. //
  7. //  File:       strret.cpp
  8. //
  9. //--------------------------------------------------------------------------
  11. ///////////////////////////////////////////////////////////////////////////////
  12. /*  File: strret.cpp
  14.     Description: Class to handle STRRET objects used in returning strings
  15.         from the Windows Shell.  Main function is to properly clean up any
  16.         dynamic storage.
  19.     Revision History:
  21.     Date        Description                                          Programmer
  22.     --------    ---------------------------------------------------  ----------
  23.     07/01/97    Initial creation.                                    BrianAu
  24. */
  25. ///////////////////////////////////////////////////////////////////////////////
  26. #include "pch.h"
  27. #pragma hdrstop
  29. #include "strret.h"
  32. StrRet::StrRet(
  33.     LPITEMIDLIST pidl,
  34.     UINT type
  35.     ) : m_pMalloc(NULL)
  36. {
  37.     uType = type;
  38.     pOleStr = NULL;
  39.     SHGetMalloc(&m_pMalloc);
  40.     m_pidl = pidl;
  41. }
  44. StrRet::~StrRet(
  45.     VOID
  46.     )
  47. {
  48.     FreeOleStr();
  50.     if (NULL != m_pMalloc)
  51.     {
  52.         m_pMalloc->Release();
  53.     }
  54. }
  57. StrRet::StrRet(const StrRet& rhs)
  58. {
  59.     *this = rhs;
  60. }
  62.     
  64. StrRet& StrRet::operator = (const StrRet& rhs)
  65. {
  66.     if (this != &rhs)
  67.     {
  68.         if (NULL != m_pMalloc)
  69.             m_pMalloc->Release();
  71.         m_pMalloc = rhs.m_pMalloc;
  72.         if (NULL != m_pMalloc)
  73.             m_pMalloc->AddRef();
  74.         
  75.         m_pidl = rhs.m_pidl;
  76.         switch(rhs.uType)
  77.         {
  78.             case STRRET_WSTR:
  79.                 CopyOleStr(rhs.pOleStr);
  80.                 break;
  82.             case STRRET_CSTR:
  83.                 lstrcpynA(cStr, rhs.cStr, ARRAYSIZE(cStr));
  84.                 break;
  86.             case STRRET_OFFSET:
  87.                 uOffset = rhs.uOffset;
  88.                 break;
  89.         }
  90.     }
  91.     return *this;
  92. }
  94. VOID
  95. StrRet::FreeOleStr(
  96.     VOID
  97.     )
  98. {
  99.     if (NULL != m_pMalloc &&
  100.         STRRET_WSTR == uType && 
  101.         NULL != pOleStr)
  102.     {
  103.         m_pMalloc->Free(pOleStr);
  104.         pOleStr = NULL;
  105.     }
  106. }
  108. VOID
  109. StrRet::CopyOleStr(
  110.     LPCWSTR pszOle
  111.     )
  112. {
  113.     FreeOleStr();
  114.     pOleStr = new WCHAR[lstrlenW(pszOle) + 1];
  115.     if (NULL != pOleStr)
  116.     {
  117.         lstrcpyW(pOleStr, pszOle);
  118.     }
  119. }
  122. LPTSTR 
  123. StrRet::GetString(
  124.     LPTSTR pszStr, 
  125.     INT cchStr
  126.     ) const
  127. {
  128.     switch(uType)
  129.     {
  130.         case STRRET_WSTR:
  131. #ifndef UNICODE
  132.             WideCharToMultiByte(CP_ACP, 0, pOleStr, -1, pszStr, cchStr, NULL, NULL);
  133. #else
  134.             lstrcpyn(pszStr, pOleStr, cchStr);
  135. #endif                                    
  136.             break;
  138.         case STRRET_CSTR:
  139. #ifndef UNICODE
  140.             lstrcpyn(pszStr, cStr, cchStr);
  141. #else
  142.             MultiByteToWideChar(CP_ACP, 0, cStr, -1, pszStr, cchStr);
  143. #endif
  144.             break;
  146.         case STRRET_OFFSET:
  147.             lstrcpyn(pszStr, (LPCTSTR)((LPBYTE)m_pidl + uOffset), cchStr);
  148.             break;
  149.     }
  150.     return pszStr;
  151. }