Category

Development Platform
Home > SourceCode/Document > Windows Develop > Windows Kernel > View Code
shinit.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++

shinit.c：Code Content
  1. #include "shprv.h"
  3. #ifdef DEBUG
  4. char ach[20];
  5. void GetTaskName(HTASK hTask, LPSTR pname, int size)
  6. {   _asm {
  7.         push    ds
  8.         les     di,pname
  9.         mov     ds,hTask
  10.         lea     si,ds:[0xF2]
  11.         movsw
  12.         movsw
  13.         movsw
  14.         movsw
  15.         xor     al,al
  16.         stosb
  17.         pop     ds
  18. }}
  19. #endif
  21. typedef struct _TASKINFO {
  22.     struct _TASKINFO   *next;           // next link
  23.     HTASK               hTask;          // task that owns this link
  24.     HMODULE             hShell16;
  25. }   TASKINFO;
  27. TASKINFO *TaskInfoList;
  28. HMODULE hShell16;
  30. //
  31. //  FindTask
  32. //
  33. TASKINFO * NEAR PASCAL FindTask()
  34. {
  35.     TASKINFO *pti;
  36.     HTASK hTask = GetCurrentTask();
  38.     for (pti=TaskInfoList; pti; pti=pti->next)
  39.     {
  40.         if (pti->hTask == hTask)
  41.             return pti;
  42.     }
  44.     pti = (TASKINFO *)LocalAlloc(LPTR, sizeof(TASKINFO));
  46.     if (pti == NULL)
  47.         return NULL;
  49.     pti->next  = TaskInfoList;
  50.     pti->hTask = hTask;
  51.     TaskInfoList = pti;
  52.     return pti;
  53. }
  55. //
  56. //  FreeTask
  57. //
  58. void NEAR PASCAL FreeTask()
  59. {
  60.     TASKINFO *pti, *p;
  61.     HTASK hTask = GetCurrentTask();
  63.     if (TaskInfoList == NULL)
  64.         return;
  66.     if (TaskInfoList->hTask == hTask)
  67.     {
  68.         pti = TaskInfoList;
  69.         TaskInfoList = pti->next;
  70.     }
  71.     else
  72.     {
  73.         for (pti=TaskInfoList; pti; p=pti,pti=pti->next)
  74.         {
  75.             if (pti->hTask == hTask)
  76.                 break;
  77.         }
  79.         if (pti == NULL)
  80.             return;
  82.         p->next = pti->next;
  83.     }
  85.     LocalFree((HLOCAL)pti);
  86. }
  88. //
  89. // load the passed library in the context of the calling app, only
  90. // once.
  91. //
  92. HMODULE _loadds NEAR PASCAL LoadShell16()
  93. {
  94.     TASKINFO *pti;
  96.     pti = FindTask();
  98.     if (pti->hShell16 == NULL)
  99.     {
  100. #ifdef DEBUG
  101.         GetTaskName(GetCurrentTask(), ach, sizeof(ach));
  102.         DebugOutput(DBF_WARNING, "SHELL: loading SHELL16 for %ls", (LPSTR)ach);
  103. #endif
  104.         pti->hShell16 = LoadLibrary("SHELL16.DLL");
  106.         //
  107.         // because we do a GetProcAddress() only once per entry point
  108.         // we cant ever *realy* free SHELL16, so we
  109.         // load it a extra time to keep it in memory until our WEP
  110.         //
  111.         if (hShell16 == NULL)
  112.             hShell16 = LoadLibrary("SHELL16.DLL");
  113.     }
  115.     return pti->hShell16;
  116. }
  118. //
  119. //  FreeShell16
  120. //
  121. void NEAR PASCAL FreeShell16()
  122. {
  123.     TASKINFO *pti;
  125.     pti = FindTask();
  127.     if (pti->hShell16)
  128.     {
  129. #ifdef DEBUG
  130.         GetTaskName(GetCurrentTask(), ach, sizeof(ach));
  131.         DebugOutput(DBF_WARNING, "SHELL: freeing SHELL16 for %ls", (LPSTR)ach);
  132. #endif
  133.         FreeLibrary(pti->hShell16);
  134.         pti->hShell16 = NULL;
  135.     }
  136. }
  138. BOOL CALLBACK LibMain(HINSTANCE hinst, UINT wDS, DWORD unused)
  139. {
  140.     return TRUE;
  141. }
  143. BOOL CALLBACK _loadds WEP(BOOL fSystemExit)
  144. {
  145.     if (hShell16)
  146.         FreeLibrary(hShell16);
  147.     hShell16 = NULL;
  148.     return TRUE;
  149. }
  151. #define DLL_PROCESS_ATTACH 1    
  152. #define DLL_PROCESS_DETACH 0    
  154. BOOL FAR PASCAL _loadds DllEntryPoint(DWORD dwReason, WORD  hInst, WORD  wDS, WORD wHeapSize, DWORD dwReserved1, WORD wReserved2)
  155. {
  156.     switch(dwReason)
  157.     {
  158.         case DLL_PROCESS_ATTACH:
  159. #ifdef DEBUG
  160.             GetTaskName(GetCurrentTask(), ach, sizeof(ach));
  161.             DebugOutput(DBF_WARNING, "SHELL: process attach for %ls", (LPSTR)ach);
  162. #endif
  163.             break;
  165.         case DLL_PROCESS_DETACH:
  166. #ifdef DEBUG
  167.             GetTaskName(GetCurrentTask(), ach, sizeof(ach));
  168.             DebugOutput(DBF_WARNING, "SHELL: process detach for %ls", (LPSTR)ach);
  169. #endif
  170.             FreeShell16();
  171.             FreeTask();
  172.             break;
  173.     }
  175.     return TRUE;
  176. }