Category

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

Windows Kernel

Development Platform:

Visual C++

util.c：Code Content
  1. //----------------------------------------------------------------------------
  2. //
  3. //----------------------------------------------------------------------------
  4. #include "smtidy.h"
  5. #include "util.h"
  7. #define INITGUID
  8. #include <initguid.h>
  10. #pragma data_seg(".text")
  11. #include <coguid.h>
  12. #include <oleguid.h>
  13. #pragma data_seg()
  15. //----------------------------------------------------------------------------
  16. BOOL Cursor_Wait(HCURSOR *phcur)
  17. {
  18.     Assert(phcur);
  20.     ShowCursor(TRUE);
  21.     *phcur = SetCursor(LoadCursor(NULL, IDC_WAIT));
  22.     
  23.     return TRUE;
  24. }
  26. //----------------------------------------------------------------------------
  27. BOOL Cursor_UnWait(HCURSOR hCur)
  28. {
  29.     Assert(hCur);
  31.     SetCursor(hCur);
  32.     ShowCursor(FALSE);
  33.     
  34.     return TRUE;
  35. }
  37. //----------------------------------------------------------------------------
  38. HRESULT PASCAL ICoCreateInstance(REFCLSID rclsid, REFIID riid, LPVOID FAR* ppv)
  39. {
  40.     LPCLASSFACTORY pcf;
  41.     HRESULT hres = SHDllGetClassObject(rclsid, &IID_IClassFactory, &pcf);
  42.     if (SUCCEEDED(hres))
  43.     {
  44. hres = pcf->lpVtbl->CreateInstance(pcf, NULL, riid, ppv);
  45. pcf->lpVtbl->Release(pcf);
  46.     }
  47.     return hres;
  48. }
  50. //----------------------------------------------------------------------------
  51. // Alloc a copy of the given string.
  52. BOOL WINAPI Sz_AllocCopy(LPCTSTR pszSrc, LPTSTR *ppszDst)
  53. {
  54.     BOOL fRet = FALSE;
  55.     
  56.     if (pszSrc && ppszDst)
  57.     {
  58.         if (LAlloc(Sz_Cb(pszSrc), ppszDst))
  59.         {
  60.             lstrcpy(*ppszDst, pszSrc);
  61.             fRet = TRUE;
  62.         }
  63.     }
  64.         
  65.     return fRet;
  66. }
  68. #ifdef DEBUG
  70. //----------------------------------------------------------------------------
  71. void WINAPI AssertFailed(LPCTSTR pszFile, int line)
  72. {
  73.     LPCTSTR psz;
  75.     for (psz = pszFile+lstrlen(pszFile); psz!=pszFile; psz=AnsiPrev(pszFile, psz))
  76.     {
  77.         if ((AnsiPrev(pszFile, psz)!= (psz-2)) && *(psz - 1) == '\')
  78.         break;
  79.     }
  81.     Dbg(TEXT("Assertion failed in %s on line %d."), psz, line);
  83.     DebugBreak();
  84. }
  86. //----------------------------------------------------------------------------
  87. void __cdecl _Dbg(LPCSTR pszMsg, ...)
  88. {
  89.     char sz[1024];
  90.     va_list ArgList;
  92.     va_start (ArgList, pszMsg);    
  93.     wvsprintf(sz, pszMsg, ArgList);
  94.     va_end (ArgList);
  95.     OutputDebugString(sz);
  96.     OutputDebugString("rn");
  97. }
  99. #else
  101. #ifndef NO_LOGGING
  102. //----------------------------------------------------------------------------
  103. HANDLE g_hFile = INVALID_HANDLE_VALUE;
  105. //----------------------------------------------------------------------------
  106. BOOL WINAPI _Dbg_OpenLog(void)
  107. {
  108.     BOOL fRet = FALSE;
  109.     
  110.     if (g_hFile == INVALID_HANDLE_VALUE)
  111.     {   
  112.         TCHAR sz[MAX_PATH];
  113.         GetWindowsDirectory(sz, ARRAYSIZE(sz));
  114.         PathAppend(sz, TEXT("smwiz.log"));
  115.         g_hFile = CreateFile(sz, GENERIC_WRITE, FILE_SHARE_READ|FILE_SHARE_WRITE, NULL, CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);
  116.         if (g_hFile != INVALID_HANDLE_VALUE)
  117.             fRet = TRUE;
  118.     }
  120.     return fRet;
  121. }
  123. //----------------------------------------------------------------------------
  124. BOOL WINAPI _Dbg_CloseLog(void)
  125. {
  126.     BOOL fRet = FALSE;
  127.     
  128.     if (g_hFile != INVALID_HANDLE_VALUE)
  129.     {   
  130.         CloseHandle(g_hFile);
  131.         g_hFile = INVALID_HANDLE_VALUE;
  132.         fRet = TRUE;
  133.     }
  135.     return fRet;
  136. }
  138. //----------------------------------------------------------------------------
  139. void __cdecl _Dbg(LPCTSTR pszMsg, ...)
  140. {
  141.     TCHAR sz[1024];
  142.     va_list ArgList;
  144.     va_start (ArgList, pszMsg);    
  145.     wvsprintf(sz, pszMsg, ArgList);
  146.     va_end (ArgList);
  147.     lstrcat(sz, TEXT("rn"));
  149.     if (g_hFile != INVALID_HANDLE_VALUE)
  150.     {
  151.         DWORD dwWritten;
  152.         WriteFile(g_hFile, sz, lstrlen(sz), &dwWritten, NULL);
  153.     }
  154. }
  155. #endif
  156. #endif