API tools faq
paste
Advertisement
bejiitas_wrath

Windows XP Taskmanager source.

bejiitas_wrath
May 4th, 2024
929
0
Never
Add comment
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 43.34 KB | None | 0 0
raw download clone embed print report
  1. /*++
  2.  
  3. Copyright (c) 1992  Microsoft Corporation
  4.  
  5. Module Name:
  6.  
  7.     taskman.c
  8.  
  9. Abstract:
  10.  
  11.     This file contains the source for the windows Task Manager.
  12.     Taskman basically is a dialog box, which enumerates active windows
  13.     keep in the user window manager, then sets active focus to the selected
  14.     dialog box element(ie active window).
  15.  
  16. --*/
  17.  
  18. #define UNICODE
  19.  
  20. #include "taskman.h"
  21. #include <port1632.h>
  22. #include <shellapi.h>
  23. #include <shlapip.h>
  24.  
  25. //LATER find correct define for NT
  26. #if !defined(NTWIN) && !defined(DOSWIN32) && !defined(WIN16)
  27. #define NTWIN   1
  28. #endif
  29.  
  30. #define ARRAYSIZE(x)    (sizeof(x)/sizeof(x[0]))
  31.  
  32. #define MAXPATHFIELD 260
  33.  
  34. #define INIT_MAX_FILES 4
  35. #define FILES_KEY  L"Software\\Microsoft\\Windows NT\\CurrentVersion\\Program Manager\\Recent File List"
  36. #define MAXFILES_ENTRY L"Max Files"
  37. #define FILE_ENTRY L"File%lu"
  38.  
  39. #define WM_CONTENTSCHANGED (WM_USER+5)
  40.  
  41. TCHAR szPathField[MAXPATHFIELD];
  42. TCHAR szDirField[MAXPATHFIELD];
  43. TCHAR szTitle[MAXPATHFIELD];
  44. TCHAR szMessage[MAXMSGBOXLEN];
  45.  
  46. TCHAR szUserHomeDir[MAXPATHFIELD];
  47. TCHAR szWindowsDirectory[MAXPATHFIELD];
  48.  
  49. TCHAR szOOMExitMsg[64] = TEXT("Close an application and try again."); // 64
  50. TCHAR szOOMExitTitle[32] = TEXT("Extremely Low on Memory"); // 32
  51.  
  52. TCHAR szNoRun[] = TEXT("NoRun");
  53. // registry key for groups
  54.  
  55. HANDLE hInst;
  56.  
  57. BOOL bChangedDefaultButton;
  58.  
  59. INT_PTR APIENTRY TaskmanDlgProc(HWND hDlg, UINT wMsg, WPARAM wParam, LPARAM lParam);
  60.  
  61. WORD APIENTRY ExecProgram(LPTSTR lpszPath,LPTSTR lpDir,LPTSTR lpTitle);
  62. void APIENTRY SaveRecentFileList (HWND, LPTSTR);
  63.  
  64. VOID GetPathInfo(PTSTR szPath,PTSTR *pszFileName,PTSTR *pszExt,WORD *pich,BOOL *pfUnc);
  65. VOID GetFilenameFromPath(PTSTR szPath, PTSTR szFilename);
  66. VOID GetDirectoryFromPath(PTSTR szFilePath, PTSTR szDir);
  67. BOOL IsUserAdmin();
  68. BOOL OKToExec();
  69. BOOL TestTokenForAdmin(HANDLE Token);
  70. VOID SetDefButton(HWND hwndDlg, INT  idButton);
  71.  
  72. WINUSERAPI VOID SwitchToThisWindow(HWND, BOOL);
  73. INT MyMessageBox(HWND hWnd,WORD idTitle,WORD idMessage,PWSTR psz,WORD wStyle);
  74.  
  75. INT MyX = 0;
  76. INT MyY = 0;
  77. INT dxTaskman;
  78. INT dyTaskman;
  79. INT dxScreen;
  80. INT dyScreen;
  81.  
  82. HWND ghwndDialog;
  83. BOOL fExecOK = TRUE;
  84. BOOL fMsgBox = FALSE;
  85.  
  86.  
  87. PVOID Alloc(
  88.     DWORD   Bytes)
  89. {
  90.     HANDLE  hMem;
  91.     PVOID   Buffer;
  92.  
  93.     hMem = LocalAlloc(LMEM_MOVEABLE, Bytes + sizeof(hMem));
  94.  
  95.     if (hMem == NULL) {
  96.         return(NULL);
  97.     }
  98.  
  99.     Buffer = LocalLock(hMem);
  100.     if (Buffer == NULL) {
  101.         LocalFree(hMem);
  102.         return(NULL);
  103.     }
  104.  
  105.     *((PHANDLE)Buffer) = hMem;
  106.  
  107.     return (PVOID)(((PHANDLE)Buffer)+1);
  108. }
  109.  
  110.  
  111. BOOL Free(
  112.     PVOID   Buffer)
  113. {
  114.     HANDLE  hMem;
  115.  
  116.     hMem = *(((PHANDLE)Buffer) - 1);
  117.  
  118.     LocalUnlock(hMem);
  119.  
  120.     return(LocalFree(hMem) == NULL);
  121. }
  122.  
  123.  
  124. VOID
  125. HideWindow(HWND hwnd)
  126. {
  127.    if (!fMsgBox) {
  128.  
  129.       if (fExecOK) {
  130.          SetDlgItemText(ghwndDialog, IDD_PATH, TEXT(""));
  131.       }
  132.  
  133.       // redundant?  why do they do the reverse twice for show below?
  134.       ShowWindow(ghwndDialog, SW_HIDE);
  135.  
  136.       SetWindowPos(ghwndDialog, HWND_NOTOPMOST, 0, 0, 0, 0,
  137.          SWP_HIDEWINDOW | SWP_NOACTIVATE | SWP_NOMOVE | SWP_NOSIZE);
  138.    }
  139.  
  140.    // Erase dark border from depressed pushbuttons
  141.    SendMessage(GetDlgItem(hwnd, IDCANCEL), // IDCANCEL
  142.       BM_SETSTYLE, BS_PUSHBUTTON, MAKELPARAM(TRUE, 0));
  143.    SendMessage(GetDlgItem(hwnd, IDD_TERMINATE),
  144.       BM_SETSTYLE, BS_PUSHBUTTON, MAKELPARAM(TRUE, 0));
  145.    SendMessage(GetDlgItem(hwnd, IDD_CASCADE),
  146.       BM_SETSTYLE, BS_PUSHBUTTON, MAKELPARAM(TRUE, 0));
  147.    SendMessage(GetDlgItem(hwnd, IDD_TILE),
  148.       BM_SETSTYLE, BS_PUSHBUTTON, MAKELPARAM(TRUE, 0));
  149.    SendMessage(GetDlgItem(hwnd, IDD_ARRANGEICONS),
  150.       BM_SETSTYLE, BS_PUSHBUTTON, MAKELPARAM(TRUE, 0));
  151. }
  152.  
  153. /*
  154.  * We call HideTasklist() when we want to remove the tasklist window
  155.  * from the screen but not select another window (ie. when we're about
  156.  * to select another app.  We call ShowWindow(SW_HIDE) directly when
  157.  * we're doing something like tiling or cascading so a window other than
  158.  * the tasklist will become the foreground window.
  159.  */
  160. VOID HideTasklist(VOID)
  161. {
  162.     if (fExecOK) {
  163.        SetDlgItemText(ghwndDialog, IDD_PATH, TEXT(""));
  164.     }
  165.  
  166.     SetWindowPos(ghwndDialog, HWND_TOP, 0, 0, 0, 0, SWP_HIDEWINDOW |
  167.             SWP_NOACTIVATE | SWP_NOMOVE | SWP_NOSIZE | SWP_NOZORDER);
  168.  
  169. }
  170.  
  171.  
  172. VOID ShowTasklist(
  173.    POINT pt)
  174. {
  175.     /*
  176.      * Retract the drop down listbox.
  177.      */
  178.  
  179.      if (fExecOK) {
  180.         SendDlgItemMessage(ghwndDialog, IDD_PATH,
  181.                            CB_SHOWDROPDOWN,0,0);
  182.      }
  183.  
  184.     SetWindowPos(ghwndDialog, HWND_TOPMOST, pt.x, pt.y, 0, 0,
  185.        SWP_NOSIZE | SWP_NOACTIVATE );
  186.     SetForegroundWindow(ghwndDialog);
  187.  
  188.     ShowWindow(ghwndDialog, SW_NORMAL);
  189. }
  190.  
  191.  
  192. /*** ActivateSelectedWindow --         Calls user, to set active window, selected
  193.  *                                                                 by the user.
  194.  *
  195.  *
  196.  * ActivateSelectedWindow(HWND hwndLB)
  197.  *
  198.  * ENTRY -         HWND hwndLB - handle to window, which is to become the active
  199.  *                                                  window, with focus.
  200.  * EXIT  -
  201.  * SYNOPSIS -  This function takes the hwnd passed into it, calls user
  202.  *                                to set active focus to that window.
  203.  * WARNINGS -
  204.  * EFFECTS  -
  205.  *
  206.  */
  207.  
  208. VOID ActivateSelectedWindow(
  209.     HWND hwndLB)
  210. {
  211.    INT nIndex;
  212.    HWND hwndT;
  213.    HWND hwndLastActive;
  214.    DWORD lTemp;
  215.  
  216.     /*
  217.      * Get the hwnd of the item which was selected.
  218.      */
  219.     nIndex = (int)SendMessage(hwndLB, LB_GETCURSEL, 0, 0);
  220.     hwndT = (HWND)SendMessage(hwndLB, LB_GETITEMDATA, nIndex, 0);
  221.  
  222.     if (!IsWindow(hwndT)) {
  223.         /*
  224.          * We gotta make sure the window is valid before doing stuff with it.
  225.          * An app may terminate itself in the background rendering these
  226.          * window handles invalid.
  227.          */
  228.         goto Beep;
  229.     }
  230.  
  231.     /*
  232.      * Switch to that task.
  233.      * HACK! Activate the window in the hwndLastActive field of the WndStruct.
  234.      */
  235.     hwndLastActive = GetLastActivePopup(hwndT);
  236.  
  237.     if (!IsWindow(hwndLastActive)) {
  238.         goto Beep;
  239.     }
  240.  
  241.     /*
  242.      * But only if it isn't disabled.
  243.      */
  244.     lTemp = GetWindowLong(hwndLastActive, GWL_STYLE);
  245.     if (!(lTemp & WS_DISABLED)) {
  246.         /*
  247.          * HACK!! Use SwitchToThisWindow() to bring dialog parents as well.
  248.          */
  249.         SwitchToThisWindow(hwndLastActive, TRUE);
  250.  
  251.     } else {
  252. Beep:
  253.         MessageBeep(0);
  254.     }
  255. }
  256.  
  257. #ifdef NTWIN
  258. /*** DoEndTask --
  259.  *
  260.  * void DoEndTask( HWND hwnd )
  261.  */
  262. VOID DoEndTask(
  263.    HWND hwnd )
  264. {
  265.    TCHAR szMsgBoxText[MAXMSGBOXLEN];
  266.    TCHAR szTempField[MAXTASKNAMELEN];
  267.    INT nch;
  268.  
  269.  
  270.    if (!EndTask(hwnd, FALSE, FALSE)) {
  271.       /* App does not want to close, ask user if
  272.        * he wants to blow it away
  273.        */
  274.  
  275.        InternalGetWindowText(hwnd, (LPTSTR)szTempField, MAXTASKNAMELEN);
  276.  
  277.        /* Load the message box string, it is very long (greater than 255 chars
  278.         * which is why we load it in two pieces
  279.         */
  280.         nch = LoadString(NULL, IDS_MSGBOXSTR1, szMsgBoxText, MAXMSGBOXLEN);
  281.         LoadString(NULL, IDS_MSGBOXSTR2, &szMsgBoxText[nch], MAXMSGBOXLEN-nch);
  282.  
  283.         if( MessageBox( NULL, szMsgBoxText, szTempField,
  284.                 MB_SETFOREGROUND | MB_SYSTEMMODAL | MB_YESNO ) == IDYES) {
  285.             EndTask(hwnd, FALSE, TRUE);
  286.         }
  287.     }
  288. }
  289.  
  290.  
  291. /*** CallEndTask --                 A separate thread to instigate EndTask
  292.  *
  293.  * CallEndTask( HWND hwnd );
  294.  *
  295.  * ENTRY -      HWND hwnd - window handle for the task to be killed
  296.  * EXIT  -
  297.  * SYNOPSIS -  This function calls EndTask on the given window to kill the
  298.  *              task that owns that window.
  299.  *
  300.  * WARNINGS -
  301.  * EFFECTS  -   Kills the task that owns hwnd.
  302.  *
  303.  */
  304.  
  305. DWORD CallEndTask(
  306.     HWND hwnd)
  307. {
  308.     DoEndTask(hwnd);
  309.  
  310.     ExitThread(0);
  311.     return 0; /* placate compiler */
  312. }
  313. #endif
  314.  
  315.  
  316. /*** TaskmanDlgProc --         Dialog Procedure for Taskman Window
  317.  *
  318.  *
  319.  *
  320.  * TaskmanDlgProc(HWND hDlg, WORD wMSG, DWORD wParam, LPARAM lParam)
  321.  *
  322.  * ENTRY -         HWND hhDlg                 - handle to dialog box.
  323.  *                        WORD wMsg                  - message to be acted upon.
  324.  *                        DWORD wParam        - value specific to wMsg.
  325.  *                        LPARAM lParam                - value specific to wMsg.
  326.  *
  327.  * EXIT  -           True if success, False if not.
  328.  * SYNOPSIS -  Dialog box message processing function.
  329.  *
  330.  * WARNINGS -
  331.  * EFFECTS  -
  332.  *
  333.  */
  334.  
  335. INT_PTR TaskmanDlgProc(
  336.     HWND hwnd,
  337.     UINT wMsg,
  338.     WPARAM wParam,
  339.     LPARAM lParam)
  340. {
  341.     int nIndex;
  342.     RECT rc;
  343.     HWND hwndLB;
  344.     HWND hwndNext;
  345.     TCHAR szTempField[MAXTASKNAMELEN];
  346.     POINT pt;
  347.     HKEY  hKey;
  348.     DWORD dwDisp;
  349.     DWORD dwDataType, dwMaxFiles=INIT_MAX_FILES, dwMaxFilesSize, dwCount;
  350.     TCHAR szFileEntry[20];
  351.     TCHAR szFullPath[MAXPATHFIELD];
  352. #ifndef NTWIN
  353.     LONG lTemp;
  354. #endif
  355.  
  356.     hwndLB = GetDlgItem(hwnd, IDD_TASKLISTBOX);
  357.  
  358.     switch (wMsg) {
  359.  
  360.     case WM_INITDIALOG:
  361.         /*
  362.          * call private api to mark task man as a system app. This causes
  363.          * it to be killed after all other non-system apps during shutdown.
  364.          */
  365. //      MarkProcess(MP_SYSTEMAPP);
  366.         GetWindowRect(hwnd, &rc);
  367.         dxTaskman = rc.right - rc.left;
  368.         dyTaskman = rc.bottom - rc.top;
  369.         dxScreen = GetSystemMetrics(SM_CXSCREEN);
  370.         dyScreen = GetSystemMetrics(SM_CYSCREEN);
  371.  
  372.         pt.x = (dxScreen - dxTaskman) / 2;
  373.         pt.y = (dyScreen - dyTaskman) / 2;
  374.  
  375.         SetWindowPos(hwnd, HWND_NOTOPMOST, pt.x, pt.y, 0, 0,
  376.            SWP_NOSIZE | SWP_NOZORDER | SWP_NOACTIVATE);
  377.  
  378.         SendDlgItemMessage(hwnd, IDD_PATH, EM_LIMITTEXT, MAXPATHFIELD-4, 0L);
  379.         szPathField[0] = TEXT('\0');
  380.  
  381.         bChangedDefaultButton = FALSE;
  382.  
  383.         return FALSE;
  384.  
  385.     case WM_SHOWWINDOW:
  386.         /*
  387.          * If we're being shown fill in the listbox.  We do this here
  388.          * rather than in WM_ACTIVATE process so we can do it while the
  389.          * dialog is still invisible.
  390.          */
  391.         if (wParam != 0) {
  392.             DWORD pidTaskMan = GetCurrentProcessId();
  393.  
  394.             /*
  395.              * First delete any previous entries.
  396.              */
  397.             while ((int)SendMessage(hwndLB, LB_DELETESTRING, 0, 0) != LB_ERR);
  398.  
  399.             /*
  400.              * Search the window list for enabled top level windows.
  401.              */
  402.             hwndNext = GetWindow(hwnd, GW_HWNDFIRST);
  403.             while (hwndNext) {
  404.  
  405.                 /*
  406.                  * Only add non-owned, visible, non-Taskman, Top Level Windows.
  407.                  */
  408.                 if ((hwndNext != hwnd) && (IsWindowVisible(hwndNext)) &&
  409.                         (!GetWindow(hwndNext, GW_OWNER))) {
  410.                     DWORD pidNext;
  411.                     GetWindowThreadProcessId(hwndNext, &pidNext);
  412.                     if (pidNext != pidTaskMan) {
  413.                         if (InternalGetWindowText(hwndNext, (LPTSTR)szTempField,
  414.                                 MAXTASKNAMELEN)) {
  415.                             nIndex = (int)SendMessage(hwndLB, LB_ADDSTRING, 0,
  416.                                     (DWORD_PTR)(LPTSTR)szTempField);
  417.                             SendMessage(hwndLB, LB_SETITEMDATA, nIndex,
  418.                                     (DWORD_PTR)hwndNext);
  419.                         }
  420.                     }
  421.                 }
  422.  
  423.                 hwndNext = GetWindow(hwndNext, GW_HWNDNEXT);
  424.             }
  425.             SendMessage(hwndLB, LB_SETCURSEL, 0, 0);
  426.  
  427.             //
  428.             // Set the default button to "Switch To"
  429.             //
  430.  
  431.             SetDefButton(hwnd,IDD_SWITCH);
  432.  
  433.             //
  434.             // Load the combobox with the recently used files.
  435.             //
  436.  
  437.             if (GetDlgItem(hwnd, IDD_PATH)) {
  438.  
  439.                 //
  440.                 // FIrst empty the combo box from the last time.
  441.                 //
  442.  
  443.                 SendDlgItemMessage (hwnd, IDD_PATH,
  444.                                     CB_RESETCONTENT, 0, 0);
  445.  
  446.  
  447.                 //
  448.                 // Load the combobox with recently used files from the registry.
  449.                 //
  450.                 // Query the max number of files first.
  451.                 //
  452.  
  453.                 if (RegCreateKeyEx (HKEY_CURRENT_USER, FILES_KEY, 0, 0,
  454.                                     REG_OPTION_NON_VOLATILE, KEY_READ | KEY_WRITE,
  455.                                     NULL, &hKey, &dwDisp) == ERROR_SUCCESS) {
  456.  
  457.                     if (dwDisp == REG_OPENED_EXISTING_KEY) {
  458.  
  459.                         //
  460.                         //  Query the max number of entries
  461.                         //
  462.  
  463.                         dwMaxFilesSize = sizeof (DWORD);
  464.  
  465.                         if (RegQueryValueEx (hKey, MAXFILES_ENTRY, NULL, &dwDataType,
  466.                                        (LPBYTE)&dwMaxFiles, &dwMaxFilesSize) == ERROR_SUCCESS) {
  467.  
  468.                             //
  469.                             //  Now Query each entry and add it to the list box.
  470.                             //
  471.  
  472.                             for (dwCount=0; dwCount < dwMaxFiles; dwCount++) {
  473.  
  474.                                 wsprintf (szFileEntry, FILE_ENTRY, dwCount);
  475.                                 dwMaxFilesSize = MAXPATHFIELD+1;
  476.  
  477.                                 if (RegQueryValueEx (hKey, szFileEntry, NULL, &dwDataType,
  478.                                                  (LPBYTE) szFullPath, &dwMaxFilesSize) == ERROR_SUCCESS) {
  479.  
  480.                                     //
  481.                                     // Found an entry.  Add it to the combo box.
  482.                                     //
  483.  
  484.                                     SendDlgItemMessage (hwnd, IDD_PATH,
  485.                                                         CB_ADDSTRING, 0,
  486.                                                         (LPARAM)szFullPath);
  487.  
  488.                                 } else {
  489.                                     break;
  490.                                 }
  491.                             }
  492.                         }
  493.                     } else {
  494.                         //
  495.                         // We are working with a new key, so we need to
  496.                         // set the default number of files.
  497.                         //
  498.  
  499.                         RegSetValueEx (hKey, MAXFILES_ENTRY, 0, REG_DWORD,
  500.                                        (CONST BYTE *) &dwMaxFiles, sizeof (DWORD));
  501.                     }
  502.  
  503.                     //
  504.                     //  Close the registry key
  505.                     //
  506.  
  507.                     RegCloseKey (hKey);
  508.  
  509.                 }
  510.             }
  511.  
  512.             //
  513.             // Disable the Run button and set the focus to the
  514.             // listbox.
  515.             //
  516.  
  517.             EnableWindow(GetDlgItem(hwnd, IDD_RUN), FALSE);
  518.             SetFocus(hwndLB);
  519.         }
  520.         break;
  521.  
  522.     case WM_ACTIVATE:
  523.         /*
  524.          * If we're being deactivated clear the listbox so we
  525.          * can fill it in afresh when we're re-activated.
  526.          */
  527.         if (wParam == 0) {
  528.             /*
  529.              * If we're not already invisible, hide ourself.
  530.              */
  531.             if (IsWindowVisible(hwnd)) {
  532.                 HideWindow(hwnd);
  533.             }
  534.         }
  535.  
  536.         if (!bChangedDefaultButton) {
  537.            SetDefButton(hwnd,IDD_SWITCH);
  538.         }
  539.  
  540.         break;
  541.  
  542.     case WM_ACTIVATEAPP:
  543.         if (wParam)
  544.             return FALSE;
  545.  
  546.         /*
  547.          * If we are not visible when we get this message it is because
  548.          * we are already in the process of terminating.  If we don't
  549.          * ignore this we get into a weird race condition and the frame
  550.          * of the window being activated doesn't get fully drawn.  (BG)
  551.          */
  552.         if (IsWindowVisible(hwnd)) {
  553.             HideWindow(hwnd);
  554.         }
  555.         break;
  556.  
  557. #ifdef JAPAN // bug fix
  558.     //
  559.     // Do nothing. Let the progman main thread do the work.
  560.     //
  561. #else // not JAPAN
  562.     case WM_WININICHANGE:
  563.         //
  564.         // Check if the user's environment variables have changed, if so
  565.         // regenerate the environment, so that new apps started from
  566.         // taskman will have the latest environment.
  567.         //
  568.         if (lParam && (!lstrcmpi((LPTSTR)lParam, (LPTSTR) TEXT("Environment")))) {
  569.             PVOID pEnv;
  570.  
  571.             RegenerateUserEnvironment(&pEnv, TRUE);
  572.             break;
  573.         }
  574.         else {
  575.             return FALSE;
  576.         }
  577. #endif // JAPAN
  578.  
  579.     case WM_CONTENTSCHANGED:
  580.        if (fExecOK) {
  581.           if (GetDlgItemText(hwnd, IDD_PATH, (LPTSTR)szPathField, MAXPATHFIELD)) {
  582.              EnableWindow(GetDlgItem(hwnd, IDD_RUN), TRUE);
  583.              if (!bChangedDefaultButton) {
  584.                 SetDefButton(hwnd,IDD_RUN);
  585.                 bChangedDefaultButton = TRUE;
  586.              }
  587.           } else {
  588.              EnableWindow(GetDlgItem(hwnd, IDD_RUN), FALSE);
  589.              if (bChangedDefaultButton) {
  590.                 SetDefButton(hwnd,IDD_SWITCH);
  591.                 bChangedDefaultButton = FALSE;
  592.              }
  593.           }
  594.        }
  595.  
  596.        break;
  597.  
  598.  
  599.     case WM_COMMAND:
  600.         switch(LOWORD(wParam)) {
  601.  
  602.         case IDD_TASKLISTBOX:
  603.  
  604.             switch(HIWORD(wParam)) {
  605.  
  606.             case LBN_DBLCLK:
  607.                 HideTasklist();
  608.                 ActivateSelectedWindow(hwndLB);
  609.                 break;
  610.  
  611.             default:
  612.                 return FALSE;
  613.             }
  614.             break;
  615.  
  616.         case IDD_PATH:
  617.            PostMessage (hwnd, WM_CONTENTSCHANGED, 0, 0);
  618.            break;
  619.  
  620.  
  621.         case IDOK:
  622.            if (!bChangedDefaultButton) {
  623.               goto Switchem;
  624.            }
  625.  
  626.         case IDD_RUN:
  627.            if (fExecOK) {
  628.               TCHAR szFilename[MAXPATHFIELD+4];
  629.               WORD ret;
  630.  
  631.               GetDlgItemText(hwnd, IDD_PATH, szPathField, MAXPATHFIELD);
  632.               DoEnvironmentSubst(szPathField, MAXPATHFIELD);
  633.               GetDirectoryFromPath(szPathField, szDirField);
  634.               if (*szDirField) {
  635.                   // Convert path into a .\foo.exe style thing.
  636.                   lstrcpy(szFilename, L".\\");
  637.                   // Tag the filename and params on to the end of the dot slash.
  638.                   GetFilenameFromPath(szPathField, szFilename+2);
  639.                   if (*(szFilename+2) == L'"' ) {
  640.                       SheRemoveQuotes(szFilename+2);
  641.                       CheckEscapes(szFilename, ARRAYSIZE(szFilename));
  642.                   }
  643.               }
  644.               else {
  645.                   GetFilenameFromPath(szPathField, szFilename);
  646.               }
  647.  
  648.               ret = ExecProgram(szFilename, szDirField, szFilename);
  649.  
  650.               if (ret) {
  651.                  fMsgBox = TRUE;
  652.                  MyMessageBox( hwnd, IDS_EXECERRTITLE, ret, szPathField,
  653.                     MB_SYSTEMMODAL | MB_OK | MB_ICONEXCLAMATION );
  654.                  fMsgBox = FALSE;
  655.  
  656.                  SetFocus(GetDlgItem(hwnd, IDD_PATH));
  657.               } else {
  658.                  GetDlgItemText(hwnd, IDD_PATH, szPathField, MAXPATHFIELD);
  659.                  SaveRecentFileList (hwnd, szPathField);
  660.                  HideWindow(hwnd);
  661.               }
  662.  
  663.            }
  664.            break;
  665.  
  666. Switchem:
  667.  
  668.         case IDD_SWITCH:
  669.             HideTasklist();
  670.             ActivateSelectedWindow(hwndLB);
  671.             break;
  672.  
  673.         case IDCANCEL:
  674.            HideWindow(hwnd);
  675.            break;
  676.  
  677.         case IDD_TERMINATE:
  678.             /*
  679.              * Get the hwnd of the item which was selected.
  680.              */
  681.             nIndex = (int)SendMessage(hwndLB, LB_GETCURSEL, 0, 0);
  682.             hwndNext = (HWND)SendMessage(hwndLB, LB_GETITEMDATA, nIndex, 0);
  683.  
  684.             if (!IsWindow(hwndNext)) {
  685.                 HideWindow(hwnd);
  686.                 MessageBeep(0);
  687.                 break;
  688.             }
  689.  
  690. #ifndef NTWIN   /* Since NTWIN uses WM_ENDSESSION to kill the app,
  691.                  * It is OK to kill it when it is disabled
  692.                  */
  693.             /*
  694.              * Test if the toplevel window is disabled. If it is
  695.              * diabled, we don't want to send a WM_CLOSE message to
  696.              * the parent. This is because the app could have a dialog
  697.              * box up and it's not expecting a CLOSE message...
  698.              * Nasty rips can happen... Instead, active the window so
  699.              * that the user can dismis any dialog box or fix whatever
  700.              * is causing the top level window to be disabled...
  701.              */
  702.             lTemp = GetWindowLong(hwndNext, GWL_STYLE);
  703.             if (lTemp & WS_DISABLED) {
  704.                 HideTasklist();
  705.                 MessageBeep(0);
  706.                 ActivateSelectedWindow(hwndLB);
  707.  
  708.             } else
  709. #endif
  710.             {
  711.                 /* Always activate the window first.  This prevents
  712.                  * apps from going to Beep mode.  Failing to do this
  713.                  * can cause re-entrancy problems in the app if we
  714.                  * do this again before activating the app.
  715.                  *
  716.                  * However, don't do this if it is a old app task.
  717.                  */
  718.  
  719. #ifdef WIN16    /* if NTWIN, then always do this, as is no winoldapp */
  720.                 if (!IsWinoldapTask(GetTaskFromHwnd(hwndNext)))
  721. #endif
  722.                 HideWindow(hwnd);
  723.                 ActivateSelectedWindow(hwndLB);
  724. #ifdef NTWIN
  725.                 {
  726.                     DWORD idt;
  727.                     HANDLE hThread;
  728.  
  729.                     hThread = CreateThread(NULL, 0,
  730.                              (LPTHREAD_START_ROUTINE)CallEndTask,
  731.                              (LPVOID)hwndNext, 0,
  732.                              &idt);
  733.  
  734.                     if (hThread == NULL) {
  735.                         /*
  736.                          * Can not create thread, just call EndTask
  737.                          * syncronously
  738.                          */
  739.                         DoEndTask( hwndNext );
  740.                     } else {
  741.                         CloseHandle(hThread);
  742.                     }
  743.                 }
  744. #else
  745.                 EndTask(hwndNext, FALSE, FALSE);
  746. #endif
  747.             }
  748.  
  749.             break;
  750.  
  751.         case IDD_TILE:
  752.         case IDD_CASCADE:
  753.             {
  754.                 HWND hwndDesktop;
  755.  
  756.                 HideWindow(hwnd);
  757.  
  758.                 hwndDesktop = GetDesktopWindow();
  759.  
  760.                 if (wParam == IDD_CASCADE) {
  761.                     CascadeChildWindows(hwndDesktop, 0);
  762.  
  763.                 } else {
  764.                     /*
  765.                      * If shift is down, tile vertically, else horizontally.
  766.                      */
  767.                     TileChildWindows(hwndDesktop, ((GetKeyState(VK_SHIFT) &
  768.                             0x8000) ? MDITILE_HORIZONTAL : MDITILE_VERTICAL));
  769.                 }
  770.                 break;
  771.             }
  772.  
  773.         case IDD_ARRANGEICONS:
  774.             /*
  775.              * Let's restore the saved bits before ArrangeIcons
  776.              * FIX for Bug #4884; --SANKAR-- 10-02-89
  777.              */
  778.             HideWindow(hwnd);
  779.             ArrangeIconicWindows(GetDesktopWindow());
  780.             break;
  781.         }
  782.  
  783.         break;
  784.  
  785.  
  786.     case WM_CLOSE:
  787.         /*
  788.          * If wParam != 0, this is a shutdown request, so exit.
  789.          */
  790.         if (wParam != 0)
  791.             ExitProcess(0);
  792.         return FALSE;
  793.         break;
  794.  
  795.     case WM_HOTKEY:
  796.         if (wParam == 1) {
  797.             pt.x = (dxScreen - dxTaskman) / 2;
  798.             pt.y = (dyScreen - dyTaskman) / 2;
  799.             ShowTasklist(pt);
  800.         }
  801.         break;
  802.  
  803.     case WM_LOGOFF:
  804.         PostQuitMessage(0);
  805.         break;
  806.  
  807.     default:
  808.         return FALSE;
  809.     }
  810.  
  811.     return TRUE;
  812.  
  813.     lParam;
  814. }
  815.  
  816. //*************************************************************
  817. //
  818. //  SetDefButton()
  819. //
  820. //  Purpose:    Sets the default button
  821. //
  822. //  Parameters: HWND hDlg     - Window handle of dialog box
  823. //              INT  idButton - ID of button
  824. //
  825. //  Return:     void
  826. //
  827. //*************************************************************
  828.  
  829. VOID SetDefButton(HWND hwndDlg, INT  idButton)
  830. {
  831.     LRESULT lr;
  832.  
  833.     if (HIWORD(lr = SendMessage(hwndDlg, DM_GETDEFID, 0, 0)) == DC_HASDEFID)
  834.     {
  835.         HWND hwndOldDefButton = GetDlgItem(hwndDlg, LOWORD(lr));
  836.  
  837.         SendMessage (hwndOldDefButton,
  838.                      BM_SETSTYLE,
  839.                      MAKEWPARAM(BS_PUSHBUTTON, 0),
  840.                      MAKELPARAM(TRUE, 0));
  841.     }
  842.  
  843.     SendMessage( hwndDlg, DM_SETDEFID, idButton, 0L );
  844.     SendMessage( GetDlgItem(hwndDlg, idButton),
  845.                  BM_SETSTYLE,
  846.                  MAKEWPARAM( BS_DEFPUSHBUTTON, 0 ),
  847.                  MAKELPARAM( TRUE, 0 ));
  848.  
  849. }
  850.  
  851.  
  852. /*** SaveRecentFileList --    Save the list of recently used files
  853.  *
  854.  * void APIENTRY SaveRecentFileList (HWND hwnd, LPTSTR szCurrentFile);
  855.  *
  856.  *
  857.  *
  858.  * ENTRY - HWND   hwnd            - handle to dialog box.
  859.  *         LPTSTR szCurrentFile   - pointer to selected filename
  860.  *
  861.  * EXIT  -
  862.  * SYNOPSIS -
  863.  *
  864.  * WARNINGS -
  865.  * EFFECTS  -
  866.  *
  867.  */
  868.  
  869.  
  870. void APIENTRY SaveRecentFileList (HWND hwnd, LPTSTR szCurrentFile)
  871. {
  872.     HKEY  hKey;
  873.     DWORD dwDisp;
  874.     DWORD dwDataType, dwMaxFiles=INIT_MAX_FILES, dwMaxFilesSize, dwCount;
  875.     TCHAR szFileEntry[20];
  876.     DWORD dwEnd=0;
  877.     DWORD dwFileNum=0;
  878.     DWORD dwDup;
  879.     static TCHAR szRecentFilePath[MAXPATHFIELD+1];
  880.  
  881.     //
  882.     // Open registry key
  883.     //
  884.  
  885.     if ( RegCreateKeyEx (HKEY_CURRENT_USER, FILES_KEY, 0, 0,
  886.                              REG_OPTION_NON_VOLATILE, KEY_READ | KEY_WRITE,
  887.                              NULL, &hKey, &dwDisp) != ERROR_SUCCESS) {
  888.         return;
  889.     }
  890.  
  891.  
  892.     //
  893.     // Query the max number of files to save first.
  894.     //
  895.  
  896.     dwMaxFilesSize = sizeof (DWORD);
  897.  
  898.     RegQueryValueEx (hKey, MAXFILES_ENTRY, NULL, &dwDataType,
  899.                     (LPBYTE)&dwMaxFiles, &dwMaxFilesSize);
  900.  
  901.     //
  902.     // If the user request 0 entries, then exit now.
  903.     //
  904.     if (dwMaxFiles == 0) {
  905.         RegCloseKey (hKey);
  906.         return;
  907.     }
  908.  
  909.     //
  910.     // Find out how many items are in the list box.
  911.     //
  912.  
  913.     dwEnd = (DWORD)SendDlgItemMessage (hwnd, IDD_PATH, CB_GETCOUNT, 0, 0);
  914.  
  915.     //
  916.     // If the max number of items we want to save is less than the
  917.     // number of entries, then change the ending point.
  918.     //
  919.  
  920.     if (dwMaxFiles < dwEnd) {
  921.         dwEnd = dwMaxFiles;
  922.     }
  923.  
  924.     //
  925.     // Add the first entry (the current file)
  926.     //
  927.  
  928.     wsprintf (szFileEntry, FILE_ENTRY, dwFileNum++);
  929.     dwMaxFilesSize = MAXPATHFIELD+1;
  930.  
  931.     RegSetValueEx (hKey, szFileEntry, 0, REG_SZ, (CONST BYTE *)szCurrentFile,
  932.                    sizeof (TCHAR) * (lstrlen (szCurrentFile)+1));
  933.  
  934.  
  935.     //
  936.     // Check for a duplicate string.
  937.     //
  938.  
  939.     dwDup = (DWORD)SendDlgItemMessage (hwnd, IDD_PATH, CB_FINDSTRING,
  940.                                 (WPARAM) -1, (LPARAM) szCurrentFile);
  941.  
  942.     //
  943.     // If we already have dwMaxFiles in the list and we don't have any
  944.     // duplicates, then we only want to save dwMaxFiles - 1 entries
  945.     // (drop the last entry).
  946.     //
  947.     //
  948.  
  949.     if ( (dwEnd == dwMaxFiles) && (dwDup == CB_ERR) ) {
  950.         dwEnd--;
  951.     }
  952.  
  953.     //
  954.     // Now loop through the remaining entries
  955.     //
  956.  
  957.     for (dwCount=0; dwCount < dwEnd; dwCount++) {
  958.  
  959.         //
  960.         // Check to see if we are at the duplicate entry.  If
  961.         // so skip on to the next item.
  962.         //
  963.  
  964.         if ((dwDup != CB_ERR) && (dwCount == dwDup)) {
  965.             continue;
  966.         }
  967.  
  968.         //
  969.         // Get an entry out of the listbox.
  970.         //
  971.  
  972.         SendDlgItemMessage (hwnd, IDD_PATH, CB_GETLBTEXT, (WPARAM) dwCount,
  973.                             (LPARAM) szRecentFilePath);
  974.  
  975.         //
  976.         // If we get a NULL string, break out of the loop.
  977.         //
  978.  
  979.         if (!(*szRecentFilePath) || !szRecentFilePath) {
  980.             break;
  981.         }
  982.  
  983.         //
  984.         // Build the entry name
  985.         //
  986.  
  987.         wsprintf (szFileEntry, FILE_ENTRY, dwFileNum);
  988.         dwMaxFilesSize = MAXPATHFIELD+1;
  989.  
  990.         //
  991.         // Save the entry
  992.         //
  993.  
  994.         RegSetValueEx (hKey, szFileEntry, 0, REG_SZ,(CONST BYTE *) szRecentFilePath,
  995.                        sizeof (TCHAR) * (lstrlen (szRecentFilePath)+1));
  996.  
  997.         //
  998.         // Increment our current file number
  999.         //
  1000.  
  1001.         dwFileNum++;
  1002.     }
  1003.  
  1004.     //
  1005.     // Close the key
  1006.     //
  1007.  
  1008.     RegCloseKey (hKey);
  1009.  
  1010. }
  1011.  
  1012.  
  1013.  
  1014. /*** Main --         Program entry point (was WinMain).
  1015.  *
  1016.  *
  1017.  *
  1018.  * Main(int argc, char *argv[], char *envp[])
  1019.  *
  1020.  * ENTRY -         int argc                - argument count.
  1021.  *                        char *argv[]        - argument list.
  1022.  *                        char *envp[]        - environment.
  1023.  *
  1024.  * EXIT  -           TRUE if success, FALSE if not.
  1025.  * SYNOPSIS -  Parses command line, for position to place dialog box, if no
  1026.  *                                position (came from ctl/esc) then center on screen.
  1027.  *                                Also make sure only one instance of taskman.
  1028.  *
  1029.  * WARNINGS -
  1030.  * EFFECTS  -
  1031.  */
  1032.  
  1033. INT __cdecl main(
  1034.    INT argc,
  1035.    CHAR *argv[],
  1036.    CHAR *envp[])
  1037. {
  1038.    MSG msg;
  1039.  
  1040.    /*
  1041.     * First set the priority of taskman so it is higher than foreground apps
  1042.     * that spin in loops - this way it'll always come up when you hit
  1043.     * ctrl-esc.
  1044.     */
  1045.    hInst = GetModuleHandle(NULL);
  1046.  
  1047.    SetPriorityClass(GetCurrentProcess(), HIGH_PRIORITY_CLASS);
  1048.  
  1049.    {
  1050.        //
  1051.        // Set the working set size to 200k.
  1052.        //
  1053.  
  1054.        QUOTA_LIMITS QuotaLimits;
  1055.        NTSTATUS status;
  1056.  
  1057.        status = NtQueryInformationProcess( NtCurrentProcess(),
  1058.                                            ProcessQuotaLimits,
  1059.                                            &QuotaLimits,
  1060.                                            sizeof(QUOTA_LIMITS),
  1061.                                            NULL );
  1062.        if (NT_SUCCESS(status)) {
  1063.            QuotaLimits.MinimumWorkingSetSize = 300 * 1024;
  1064.            QuotaLimits.MaximumWorkingSetSize = 372 * 1024;
  1065.  
  1066.            NtSetInformationProcess( NtCurrentProcess(),
  1067.                                     ProcessQuotaLimits,
  1068.                                     &QuotaLimits,
  1069.                                     sizeof(QUOTA_LIMITS) );
  1070.        }
  1071.    }
  1072.  
  1073.    /*
  1074.     * Taskman will work in the windows directory, and switch to the
  1075.     * original directory (home directory) before execing programs.
  1076.     * This is to prevent weird popups if a UNC original directory is
  1077.     * disconnected.
  1078.     */
  1079.  
  1080.    GetCurrentDirectory(MAXPATHFIELD, szUserHomeDir);
  1081.    GetWindowsDirectory(szWindowsDirectory, MAXPATHFIELD);
  1082.    SetCurrentDirectory(szWindowsDirectory);
  1083.  
  1084.    fExecOK = OKToExec();
  1085.    if (!IsUserAdmin() && !fExecOK) {
  1086.       ghwndDialog = CreateDialog(hInst, MAKEINTRESOURCE(WMPTASKMANDLG), NULL,
  1087.          TaskmanDlgProc);
  1088.    } else {
  1089.       ghwndDialog = CreateDialog(hInst, MAKEINTRESOURCE(PWRTASKMANDLG), NULL,
  1090.          TaskmanDlgProc);
  1091.    }
  1092.  
  1093.    if (ghwndDialog == NULL)
  1094.        return 0;
  1095.  
  1096.    LoadString(hInst, IDS_OOMEXITTITLE, szOOMExitTitle, 32);
  1097.    LoadString(hInst, IDS_OOMEXITMSG, szOOMExitMsg, 64);
  1098.  
  1099.    if (!RegisterHotKey(ghwndDialog, 1, MOD_CONTROL, VK_ESCAPE) ||
  1100.            !RegisterTasklist(ghwndDialog)) {
  1101.        goto exit;
  1102.    }
  1103.  
  1104.    while (GetMessage(&msg, (HWND)NULL, (UINT)0, (UINT)0)) {
  1105.        if (!IsDialogMessage(ghwndDialog, &msg)) {
  1106.           if ((msg.message == WM_SYSCOMMAND) && (msg.wParam == SC_TASKLIST)) {
  1107.              POINT pt;
  1108.  
  1109.              GetCursorPos(&pt);
  1110.              pt.x = max(pt.x - (dyTaskman / 2), 0);
  1111.              pt.x = min(pt.x, dxScreen - dxTaskman);
  1112.              pt.y = max(pt.y - (GetSystemMetrics(SM_CYCAPTION) * 2), 0);
  1113.              pt.y = min(pt.y, dyScreen - dyTaskman);
  1114.  
  1115.              ShowTasklist(pt);
  1116.           } else {
  1117.  
  1118.               //
  1119.               //  We need to have a regular message loop in order
  1120.               //  to handle the DDE messages generated by spawning
  1121.               //  an application via an association.
  1122.               //
  1123.  
  1124.               TranslateMessage (&msg);
  1125.               DispatchMessage (&msg);
  1126.           }
  1127.  
  1128.        }
  1129.    }
  1130.  
  1131. exit:
  1132.     DestroyWindow(ghwndDialog);
  1133.     return 0;
  1134.  
  1135.     argc;
  1136.     argv;
  1137.     envp;
  1138. }
  1139.  
  1140.  
  1141. WORD APIENTRY
  1142. ExecProgram(
  1143.     LPTSTR lpszPath,
  1144.     LPTSTR lpDir,
  1145.     LPTSTR lpTitle
  1146.     )
  1147. {
  1148.   WORD      ret;
  1149.   HCURSOR   hCursor;
  1150.   LPTSTR     lpP;
  1151.   TCHAR cSeparator;
  1152.   TCHAR lpReservedFormat[] = TEXT("dde.%d,hotkey.%d");
  1153.   TCHAR lpReserved[100];  // used for DDE request of icons from console apps
  1154.                          // add for passing the hotkey associated with an item.
  1155.   HANDLE hProcess;
  1156.  
  1157.   ret = 0;
  1158.  
  1159.   /*
  1160.    * Set the current directory to the user's home directory if possible.
  1161.    */
  1162.   SetCurrentDirectory(szUserHomeDir);
  1163.  
  1164.   hCursor = SetCursor(LoadCursor(NULL, IDC_WAIT));
  1165.  
  1166.   /* Don't mess with the mouse state; unless we're on a mouseless system.
  1167.    */
  1168.   if (!GetSystemMetrics(SM_MOUSEPRESENT))
  1169.       ShowCursor(TRUE);
  1170.  
  1171.   /* skip leading spaces
  1172.    */
  1173.   while (*lpszPath == TEXT(' '))
  1174.       lpszPath++;
  1175.  
  1176.   /* skip past path
  1177.    */
  1178.   lpP = lpszPath;
  1179.   if (*lpszPath == TEXT('"')) {
  1180.      cSeparator = TEXT('"');
  1181.      lpP++;
  1182.   }
  1183.   else {
  1184.      cSeparator = TEXT(' ');
  1185.   }
  1186.  
  1187.   for (; *lpP && *lpP != cSeparator; lpP = CharNext(lpP))
  1188.       ;
  1189.   if (*lpP == TEXT('"')) {
  1190.      lpP++;
  1191.   }
  1192.  
  1193.   /* if stuff on end, separate it
  1194.    */
  1195.   if (*lpP)
  1196.       *lpP++ = 0;
  1197.  
  1198.  
  1199.   /* Try to exec 'szCommandLine'. */
  1200.  
  1201.   /*changed order, since wPendINstance is a 32b HANDLE, and ret is WORD*/
  1202.     if (!lpP)
  1203.         lpP = TEXT("");
  1204.  
  1205.     wsprintf(lpReserved, lpReservedFormat, 0, 0);
  1206.  
  1207.     ret = (WORD)RealShellExecute(ghwndDialog, NULL, lpszPath, lpP,
  1208.                             lpDir, NULL, lpTitle, lpReserved,
  1209.                             (WORD)SW_SHOWNORMAL, &hProcess);
  1210.  
  1211.   /*BUG BUG these are DOS exec function return codes, no map yet to NT return codes!*/
  1212.   switch (ret) {
  1213.       case 0:
  1214.       case SE_ERR_OOM:    // 8
  1215.           ret = IDS_NOMEMORYMSG;
  1216.           break;
  1217.  
  1218.       case SE_ERR_FNF:    // 2
  1219.           ret = IDS_FILENOTFOUNDMSG;
  1220.           break;
  1221.  
  1222.       case SE_ERR_PNF:    // 3
  1223.           ret = IDS_BADPATHMSG;
  1224.           break;
  1225.  
  1226.       case 4:
  1227.           ret = IDS_MANYOPENFILESMSG;
  1228.           break;
  1229.  
  1230.       case 5:
  1231.           ret = IDS_ACCESSDENIED;
  1232.           break;
  1233.  
  1234.       case 10:
  1235.           ret = IDS_NEWWINDOWSMSG;
  1236.           break;
  1237.  
  1238.       case 12:
  1239.           ret = IDS_OS2APPMSG;
  1240.           break;
  1241.  
  1242.       case 15:
  1243.           /* KERNEL has already put up a messagebox for this one. */
  1244.           ret = 0;
  1245.           break;
  1246.  
  1247.       case 16:
  1248.           ret = IDS_MULTIPLEDSMSG;
  1249.           break;
  1250.  
  1251.       case 18:
  1252.           ret = IDS_PMODEONLYMSG;
  1253.           break;
  1254.  
  1255.       case 19:
  1256.           ret = IDS_COMPRESSEDEXE;
  1257.           break;
  1258.  
  1259.       case 20:
  1260.           ret = IDS_INVALIDDLL;
  1261.           break;
  1262.  
  1263.       case SE_ERR_SHARE:
  1264.           ret = IDS_SHAREERROR;
  1265.           break;
  1266.  
  1267.       case SE_ERR_ASSOCINCOMPLETE:
  1268.           ret = IDS_ASSOCINCOMPLETE;
  1269.           break;
  1270.  
  1271.       case SE_ERR_DDETIMEOUT:
  1272.       case SE_ERR_DDEFAIL:
  1273.       case SE_ERR_DDEBUSY:
  1274.           ret = IDS_DDEFAIL;
  1275.           break;
  1276.  
  1277.       case SE_ERR_NOASSOC:
  1278.           ret = IDS_NOASSOCMSG;
  1279.           break;
  1280.  
  1281.       default:
  1282.           ret = 0;
  1283.           break;
  1284.   }
  1285.  
  1286.   if (!GetSystemMetrics(SM_MOUSEPRESENT)) {
  1287.       /*
  1288.        * We want to turn the mouse off here on mouseless systems, but
  1289.        * the mouse will already have been turned off by USER if the
  1290.        * app has GP'd so make sure everything's kosher.
  1291.        */
  1292.       if (ShowCursor(FALSE) != -1)
  1293.           ShowCursor(TRUE);
  1294.   }
  1295.  
  1296.   SetCursor(hCursor);
  1297.  
  1298.   /*
  1299.    * Reset the working directory to the windows directory.
  1300.    */
  1301.   SetCurrentDirectory(szWindowsDirectory);
  1302.  
  1303.   return(ret);
  1304. }
  1305.  
  1306.  
  1307. VOID
  1308. GetDirectoryFromPath(
  1309.    PTSTR szFilePath,
  1310.    PTSTR szDir)
  1311. {
  1312.    PTSTR pFileName;
  1313.    PTSTR pExt;
  1314.    WORD ich;
  1315.    BOOL fUnc;
  1316.  
  1317.    *szDir = TEXT('\0');
  1318.  
  1319.    /* Get info about file path. */
  1320.    GetPathInfo(szFilePath, &pFileName, &pExt, &ich, &fUnc);
  1321.  
  1322.    /* UNC paths don't (conceptually to Progman) have a directory component. */
  1323.    if (fUnc)
  1324.       return;
  1325.  
  1326.    /* Does it have a directory component ? */
  1327.    if (pFileName != szFilePath) { // Yep.
  1328.       /* copy path to temp. */
  1329.       if (*szFilePath == TEXT('"')) {
  1330.          szFilePath++;
  1331.       }
  1332.       lstrcpy(szDir, szFilePath);
  1333.       /* check path style. */
  1334.       if (ich <= 3 && *(szDir+1) == TEXT(':')) {
  1335.          /*
  1336.           * The path is "c:\foo.c" or "c:foo.c" style.
  1337.           * Don't remove the last slash/colon, just the filename.
  1338.           */
  1339.          szDir[pFileName-szFilePath] = TEXT('\0');
  1340.       }
  1341.  
  1342.       else if (ich == 1) {
  1343.          /*
  1344.           * something like "\foo.c"
  1345.           * Don't remove the last slash/colon, just the filename.
  1346.           */
  1347.           szDir[pFileName-szFilePath] = TEXT('\0');
  1348.        }
  1349.        else {
  1350.           /*
  1351.            * The filepath is a full normal path.
  1352.            * Could be something like "..\foo.c" or ".\foo.c" though.
  1353.            * Stomp on the last slash to get just the path.
  1354.            */
  1355.            szDir[pFileName-szFilePath-1] = TEXT('\0');
  1356.        }
  1357.     }
  1358.  
  1359.     /* else just a filename with no path. */
  1360. }
  1361.  
  1362. VOID
  1363. GetFilenameFromPath(
  1364.    PTSTR szPath,
  1365.    PTSTR szFilename)
  1366. {
  1367.    DWORD dummy;
  1368.    PTSTR pFileName;
  1369.    BOOL fUNC;
  1370.  
  1371.    GetPathInfo(szPath, &pFileName, (PTSTR*) &dummy, (WORD*) &dummy,
  1372.       &fUNC);
  1373.  
  1374.    /* If it's a UNC then the 'filename' part is the whole thing. */
  1375.    if (fUNC || (szPath == pFileName))
  1376.       lstrcpy(szFilename, szPath);
  1377.    else {
  1378.       if (*szPath == TEXT('"')) {
  1379.          *szFilename++ = TEXT('"');
  1380.       }
  1381.       lstrcpy(szFilename, pFileName);
  1382.    }
  1383. }
  1384.  
  1385.  
  1386. VOID
  1387. GetPathInfo(
  1388.    PTSTR szPath,
  1389.    PTSTR *pszFileName,
  1390.    PTSTR *pszExt,
  1391.    WORD *pich,
  1392.    BOOL *pfUnc)
  1393. {
  1394.    TCHAR *pch;          // Temp variable.
  1395.    WORD ich = 0;       // Temp.
  1396.    BOOL InQuotes;
  1397.  
  1398.    *pszExt = NULL;         // If no extension, return NULL.
  1399.    *pszFileName = szPath;  // If no seperate filename component, return path.
  1400.    *pich = 0;
  1401.    *pfUnc = FALSE;         // Default to not UNC style.
  1402.  
  1403.    //
  1404.    // Check if path is in quotes.
  1405.    //
  1406.    if (InQuotes = (*szPath == TEXT('"'))) {
  1407.       szPath++;
  1408.    }
  1409.  
  1410.    // Check for UNC style paths.
  1411.    if (*szPath == TEXT('\\') && *(szPath+1) == TEXT('\\'))
  1412.       *pfUnc = TRUE;
  1413.  
  1414.    // Search forward to find the last backslash or colon in the path.
  1415.    // While we're at it, look for the last dot.
  1416.    for (pch = szPath; *pch; pch = CharNext(pch)) {
  1417.  
  1418.       if ((*pch == TEXT(' ')) && (!InQuotes)) {
  1419.          // Found a space - stop here.
  1420.          break;
  1421.       }
  1422.       if (*pch == TEXT('"')) {
  1423.          // Found a the second quote - stop here.
  1424.          pch++;
  1425.          break;
  1426.       }
  1427.       if (*pch == TEXT('\\') || *pch == TEXT(':')) {
  1428.          // Found it, record ptr to it and it's index.
  1429.          *pszFileName = pch+1;
  1430.          *pich = ich + (WORD)1;
  1431.       }
  1432.       if (*pch == TEXT('.')) {
  1433.          // Found a dot.
  1434.          *pszExt = pch;
  1435.       }
  1436.       ich++;
  1437.    }
  1438.  
  1439.    /* Check that the last dot is part of the last filename. */
  1440.    if (*pszExt < *pszFileName)
  1441.       *pszExt = NULL;
  1442. }
  1443.  
  1444.  
  1445. BOOL
  1446. IsUserAdmin()
  1447. {
  1448.     BOOL UserIsAdmin = FALSE;
  1449.     HANDLE Token;
  1450.     PSID    AdminAliasSid = NULL;
  1451.     SID_IDENTIFIER_AUTHORITY NtAuthority = SECURITY_NT_AUTHORITY;
  1452.  
  1453.     //
  1454.     // Get the token of the current process.
  1455.     //
  1456.  
  1457.     if (!OpenProcessToken(
  1458.           GetCurrentProcess(),
  1459.           TOKEN_QUERY,
  1460.           &Token) ) {
  1461.        return(FALSE);
  1462.     }
  1463.  
  1464.     UserIsAdmin = TestTokenForAdmin(Token);
  1465.  
  1466.  
  1467.     CloseHandle(Token);
  1468.  
  1469.     return(UserIsAdmin);
  1470. }
  1471.  
  1472. BOOL
  1473. TestTokenForAdmin(
  1474.    HANDLE Token
  1475.    )
  1476. {
  1477.    NTSTATUS    Status;
  1478.    DWORD       InfoLength;
  1479.    PTOKEN_GROUPS TokenGroupList;
  1480.    DWORD       GroupIndex;
  1481.    PSID        AdminSid;
  1482.    BOOL        FoundAdmin;
  1483.    SID_IDENTIFIER_AUTHORITY SystemSidAuthority = SECURITY_NT_AUTHORITY;
  1484.  
  1485.    //
  1486.    // Get a list of groups in the token
  1487.    //
  1488.  
  1489.    Status = NtQueryInformationToken(
  1490.       Token,                    // Handle
  1491.       TokenGroups,              // TokenInformationClass
  1492.       NULL,                     // TokenInformation
  1493.       0,                        // TokenInformationLength
  1494.       &InfoLength               // ReturnLength
  1495.    );
  1496.  
  1497.    if ((Status != STATUS_SUCCESS) && (Status != STATUS_BUFFER_TOO_SMALL)) {
  1498.       return(FALSE);
  1499.    }
  1500.  
  1501.    TokenGroupList = Alloc(InfoLength);
  1502.  
  1503.    if (TokenGroupList == NULL) {
  1504.       return(FALSE);
  1505.    }
  1506.  
  1507.    Status = NtQueryInformationToken(
  1508.       Token,                    // Handle
  1509.       TokenGroups,              // TokenInformationClass
  1510.       TokenGroupList,           // TokenInformation
  1511.       InfoLength,               // TokenInformationLength
  1512.       &InfoLength               // ReturnLength
  1513.    );
  1514.  
  1515.    if (!NT_SUCCESS(Status)) {
  1516.       LocalFree(TokenGroupList);
  1517.       return(FALSE);
  1518.    }
  1519.  
  1520.    //
  1521.    // Create the admin sid
  1522.    //
  1523.    Status = RtlAllocateAndInitializeSid(
  1524.       &SystemSidAuthority, 2,
  1525.       SECURITY_BUILTIN_DOMAIN_RID,
  1526.       DOMAIN_ALIAS_RID_ADMINS,
  1527.       0, 0, 0, 0, 0, 0,
  1528.       &AdminSid);
  1529.  
  1530.    if (!NT_SUCCESS(Status)) {
  1531.       Free(TokenGroupList);
  1532.       return(FALSE);
  1533.    }
  1534.  
  1535.    //
  1536.    // Search group list for admin alias
  1537.    //
  1538.    FoundAdmin = FALSE;
  1539.  
  1540.    for (GroupIndex=0; GroupIndex < TokenGroupList->GroupCount; GroupIndex++ ) {
  1541.       if (RtlEqualSid(TokenGroupList->Groups[GroupIndex].Sid, AdminSid)) {
  1542.          FoundAdmin = TRUE;
  1543.          break;
  1544.       }
  1545.    }
  1546.  
  1547.    //
  1548.    // Tidy up
  1549.    //
  1550.  
  1551.    RtlFreeSid(AdminSid);
  1552.    Free(TokenGroupList);
  1553.  
  1554.    return(FoundAdmin);
  1555. }
  1556.  
  1557. BOOL
  1558. OKToExec()
  1559. {
  1560.    TCHAR szRestrict[]        = TEXT("Restrictions");
  1561.    TCHAR szProgramManager[]  = TEXT("Software\\Microsoft\\Windows NT\\CurrentVersion\\Program Manager");
  1562.    HKEY hkeyProgramManager  = NULL;  // progman.ini key
  1563.    HKEY hkeyPMRestrict      = NULL;
  1564.    DWORD cbData, dwType;
  1565.    BOOL fNoRun = FALSE;
  1566.  
  1567.   /*
  1568.    * Create/Open the registry keys corresponding to progman.ini sections.
  1569.    */
  1570.   if (!RegCreateKeyEx(HKEY_CURRENT_USER, szProgramManager, 0, szProgramManager, 0,
  1571.                          KEY_READ | KEY_WRITE, NULL, &hkeyProgramManager, NULL)) {
  1572.  
  1573.       RegCreateKeyEx(hkeyProgramManager, szRestrict, 0, szProgramManager, 0,
  1574.                          KEY_READ, NULL, &hkeyPMRestrict, NULL);
  1575.  
  1576.   } else {
  1577.     return(FALSE);
  1578.   }
  1579.  
  1580.   if (hkeyPMRestrict) {
  1581.       cbData = sizeof(fNoRun);
  1582.       RegQueryValueEx(hkeyPMRestrict, szNoRun, 0, &dwType, (LPBYTE)&fNoRun,
  1583.          &cbData);
  1584.   }
  1585.  
  1586.   if (hkeyPMRestrict) {
  1587.       RegCloseKey(hkeyPMRestrict);
  1588.       hkeyPMRestrict = NULL;
  1589.   }
  1590.  
  1591.   RegCloseKey(hkeyProgramManager);
  1592.  
  1593.   return(!fNoRun);
  1594.  
  1595. }
  1596.  
  1597. INT
  1598. MyMessageBox(
  1599.    HWND hWnd,
  1600.    WORD idTitle,
  1601.    WORD idMessage,
  1602.    PWSTR psz,
  1603.    WORD wStyle)
  1604. {
  1605.     WCHAR    szTempField[MAXMSGBOXLEN];
  1606.     INT     iMsgResult;
  1607.  
  1608.     if (!LoadString(hInst, idTitle, szTitle, ARRAYSIZE(szTitle))){
  1609.         goto MessageBoxOOM;
  1610.     }
  1611.     if (idMessage < 32){
  1612.         if (!LoadString(hInst, IDS_UNKNOWNMSG, szTempField, ARRAYSIZE(szTempField))){
  1613.             goto MessageBoxOOM;
  1614.         }
  1615.         wsprintf(szMessage, szTempField, idMessage);
  1616.     }
  1617.     else{
  1618.         if (!LoadString(hInst, idMessage, szTempField, ARRAYSIZE(szTempField)))
  1619.             goto MessageBoxOOM;
  1620.  
  1621.         if (psz) {
  1622.             wsprintf(szMessage, szTempField, (LPTSTR)psz);
  1623.         }
  1624.         else
  1625.             lstrcpy(szMessage, szTempField);
  1626.     }
  1627.  
  1628.     iMsgResult = MessageBox(hWnd, szMessage, szTitle, wStyle );
  1629.  
  1630.     if (iMsgResult == -1){
  1631.  
  1632. MessageBoxOOM:
  1633.         MessageBox(hWnd, szOOMExitMsg, szOOMExitTitle, MB_SYSTEMMODAL | MB_ICONHAND | MB_OK);
  1634.     }
  1635.  
  1636.     return(iMsgResult);
  1637. }
  1638.  
Tags: Code windows source taskman
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement
Public Pastes
Advertisement
We use cookies for various purposes including analytics. By continuing to use Pastebin, you agree to our use of cookies as described in the Cookies Policy.  OK, I Understand
Not a member of Pastebin yet?
Sign Up, it unlocks many cool features!