base/applications/taskmgr/endproc.c

   1 /*
   2  *  ReactOS Task Manager
   3  *
   4  *  endproc.c
   5  *
   6  *  Copyright (C) 1999 - 2001  Brian Palmer               <brianp@reactos.org>
   7  *                2005         Klemens Friedl             <frik85@reactos.at>
   8  *                2014         Ismael Ferreras Morezuelas <swyterzone+ros@gmail.com>
   9  *
  10  * This library is free software; you can redistribute it and/or
  11  * modify it under the terms of the GNU Lesser General Public
  12  * License as published by the Free Software Foundation; either
  13  * version 2.1 of the License, or (at your option) any later version.
  14  *
  15  * This library is distributed in the hope that it will be useful,
  16  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  17  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  18  * Lesser General Public License for more details.
  19  *
  20  * You should have received a copy of the GNU Lesser General Public
  21  * License along with this library; if not, write to the Free Software
  22  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA
  23  */
  24
  25 #include "precomp.h"
  26
  27 #define NTOS_MODE_USER
  28 #include <ndk/psfuncs.h>
  29
  30 void ProcessPage_OnEndProcess(void)
  31 {
  32     DWORD   dwProcessId;
  33     HANDLE  hProcess;
  34     WCHAR   szTitle[256];
  35     WCHAR   strErrorText[260];
  36
  37     dwProcessId = GetSelectedProcessId();
  38
  39     if (dwProcessId == 0)
  40         return;
  41
  42     hProcess = OpenProcess(PROCESS_TERMINATE | PROCESS_QUERY_INFORMATION, FALSE, dwProcessId);
  43
  44     /* forbid killing system processes even if we have privileges -- sigh, windows kludge! */
  45     if (hProcess && IsCriticalProcess(hProcess))
  46     {
  47         LoadStringW(hInst, IDS_MSG_UNABLETERMINATEPRO, szTitle, 256);
  48         LoadStringW(hInst, IDS_MSG_CLOSESYSTEMPROCESS, strErrorText, 256);
  49         MessageBoxW(hMainWnd, strErrorText, szTitle, MB_OK|MB_ICONWARNING|MB_TOPMOST);
  50         CloseHandle(hProcess);
  51         return;
  52     }
  53
  54     /* if this is a standard process just ask for confirmation before doing it */
  55     LoadStringW(hInst, IDS_MSG_WARNINGTERMINATING, strErrorText, 256);
  56     LoadStringW(hInst, IDS_MSG_TASKMGRWARNING, szTitle, 256);
  57     if (MessageBoxW(hMainWnd, strErrorText, szTitle, MB_YESNO|MB_ICONWARNING|MB_TOPMOST) != IDYES)
  58     {
  59         if (hProcess) CloseHandle(hProcess);
  60         return;
  61     }
  62
  63     /* no such process or not enough privileges to open its token */
  64     if (!hProcess)
  65     {
  66         GetLastErrorText(strErrorText, 260);
  67         LoadStringW(hInst, IDS_MSG_UNABLETERMINATEPRO, szTitle, 256);
  68         MessageBoxW(hMainWnd, strErrorText, szTitle, MB_OK|MB_ICONSTOP|MB_TOPMOST);
  69         return;
  70     }
  71
  72     /* try to kill it, and notify the user if didn't work */
  73     if (!TerminateProcess(hProcess, 1))
  74     {
  75         GetLastErrorText(strErrorText, 260);
  76         LoadStringW(hInst, IDS_MSG_UNABLETERMINATEPRO, szTitle, 256);
  77         MessageBoxW(hMainWnd, strErrorText, szTitle, MB_OK|MB_ICONSTOP|MB_TOPMOST);
  78     }
  79
  80     CloseHandle(hProcess);
  81 }
  82
  83 BOOL IsCriticalProcess(HANDLE hProcess)
  84 {
  85     NTSTATUS status;
  86     ULONG BreakOnTermination;
  87
  88     /* return early if the process handle does not exist */
  89     if (!hProcess)
  90         return FALSE;
  91
  92     /* the important system processes that we don't want to let the user
  93        kill come marked as critical, this simplifies the check greatly.
  94
  95        a critical process brings the system down when is terminated:
  96        <http://www.geoffchappell.com/studies/windows/win32/ntdll/api/rtl/peb/setprocessiscritical.htm> */
  97
  98     status = NtQueryInformationProcess(hProcess,
  99                                        ProcessBreakOnTermination,
 100                                        &BreakOnTermination,
 101                                        sizeof(ULONG),
 102                                        NULL);
 103
 104     if (NT_SUCCESS(status) && BreakOnTermination)
 105         return TRUE;
 106
 107     return FALSE;
 108 }
 109
 110 void ProcessPage_OnEndProcessTree(void)
 111 {
 112     DWORD   dwProcessId;
 113     HANDLE  hProcess;
 114     WCHAR   szTitle[256];
 115     WCHAR   strErrorText[260];
 116
 117     dwProcessId = GetSelectedProcessId();
 118
 119     if (dwProcessId == 0)
 120         return;
 121
 122     hProcess = OpenProcess(PROCESS_TERMINATE | PROCESS_QUERY_INFORMATION, FALSE, dwProcessId);
 123
 124     /* forbid killing system processes even if we have privileges -- sigh, windows kludge! */
 125     if (hProcess && IsCriticalProcess(hProcess))
 126     {
 127         LoadStringW(hInst, IDS_MSG_UNABLETERMINATEPRO, szTitle, 256);
 128         LoadStringW(hInst, IDS_MSG_CLOSESYSTEMPROCESS, strErrorText, 256);
 129         MessageBoxW(hMainWnd, strErrorText, szTitle, MB_OK|MB_ICONWARNING|MB_TOPMOST);
 130         CloseHandle(hProcess);
 131         return;
 132     }
 133
 134     LoadStringW(hInst, IDS_MSG_WARNINGTERMINATING, strErrorText, 256);
 135     LoadStringW(hInst, IDS_MSG_TASKMGRWARNING, szTitle, 256);
 136     if (MessageBoxW(hMainWnd, strErrorText, szTitle, MB_YESNO|MB_ICONWARNING) != IDYES)
 137     {
 138         if (hProcess) CloseHandle(hProcess);
 139         return;
 140     }
 141
 142     if (!hProcess)
 143     {
 144         GetLastErrorText(strErrorText, 260);
 145         LoadStringW(hInst, IDS_MSG_UNABLETERMINATEPRO, szTitle, 256);
 146         MessageBoxW(hMainWnd, strErrorText, szTitle, MB_OK|MB_ICONSTOP);
 147         return;
 148     }
 149
 150     if (!TerminateProcess(hProcess, 0))
 151     {
 152         GetLastErrorText(strErrorText, 260);
 153         LoadStringW(hInst, IDS_MSG_UNABLETERMINATEPRO, szTitle, 256);
 154         MessageBoxW(hMainWnd, strErrorText, szTitle, MB_OK|MB_ICONSTOP);
 155     }
 156
 157     CloseHandle(hProcess);
 158 }