Forum Algerie213

#include <windows.h>
#include <tlhelp32.h>
#include <commctrl.h>

char usage[]= "process killer:\\n"
              "------------------------------------------\\n"
              "help             this data\\n"
              "refresh          show the running processes\\n"
              "kill [proc]      kill the chosen process\\n"
              "kill all         kill all processes\\n";
                  

void disp_error(char * message)
{
    printf("%s\\n\\n", message);
    ExitProcess(0);
    return;
}

void refresh() 
{
    HANDLE sys;
    int ProcessCnt = 0;
  
    if (!(sys = CreateToolhelp32Snapshot(TH32CS_SNAPPROCESS, 0))) return;
   
    PROCESSENTRY32 proc;
    ZeroMemory(&proc, sizeof(proc));
    proc.dwSize = sizeof(proc);
  
    if (!Process32First(sys, &proc)) return;
   
    do {
    
      printf("%s\\n", proc.szExeFile);
      ProcessCnt++;
     } while (Process32Next(sys, &proc));
  
    CloseHandle(sys);
}

BOOL set_processes_access() 
{
   DWORD dwPID;
   HANDLE hProcess, hToken;
   LUID Luid;
   TOKEN_PRIVILEGES tpDebug;
  
   dwPID = GetCurrentProcessId();
   if ((hProcess = OpenProcess(PROCESS_ALL_ACCESS, FALSE, dwPID)) == NULL) return FALSE;
   if (OpenProcessToken(hProcess, TOKEN_ALL_ACCESS, &hToken) == 0) return FALSE;
   if (LookupPrivilegeValue(NULL, SE_DEBUG_NAME, &Luid) == 0) return FALSE;
  
   tpDebug.PrivilegeCount = 1;
   tpDebug.Privileges[0].Luid = Luid;
   tpDebug.Privileges[0].Attributes = SE_PRIVILEGE_ENABLED;
  
   if ((AdjustTokenPrivileges(hToken, FALSE, &tpDebug, sizeof(tpDebug), NULL, NULL)) == 0) return FALSE;
  
   CloseHandle(hToken);
   CloseHandle(hProcess);
  
   return TRUE;                                                                                                                                                                                                                                                                                                                                
}    
 
BOOL kill(char* itemSelected) 
{
   if (!set_processes_access()) return FALSE;
  
   HANDLE sys;
  
   if (!(sys = CreateToolhelp32Snapshot(TH32CS_SNAPPROCESS, 0))) return FALSE;
  
   PROCESSENTRY32 proc;
   ZeroMemory(&proc, sizeof(proc));
   proc.dwSize = sizeof(proc);
  
   if (!Process32First(sys, &proc)) return FALSE;
  
   DWORD pID;
  
   do {
   
      if (!strcmp(proc.szExeFile, itemSelected)) 
      {
         
        pID = proc.th32ProcessID;
        break;
      }       
         
    } while (Process32Next(sys, &proc));
    
   CloseHandle(sys);
  
   HANDLE hOpenProc;
  
   if (!(hOpenProc = OpenProcess(PROCESS_TERMINATE, FALSE, pID))) return FALSE;
   TerminateProcess(hOpenProc, 0);
   CloseHandle(hOpenProc);
    
   return TRUE;
}

void kill_all() 
{
    HANDLE sys;
    int ProcessCnt = 0;
  
    if (!(sys = CreateToolhelp32Snapshot(TH32CS_SNAPPROCESS, 0))) return;
   
    PROCESSENTRY32 proc;
    ZeroMemory(&proc, sizeof(proc));
    proc.dwSize = sizeof(proc);
  
    if (!Process32First(sys, &proc)) return;
   
    do {
    
      kill(proc.szExeFile);
      ProcessCnt++;
     } while (Process32Next(sys, &proc));
  
    CloseHandle(sys);
}

int main(int argc, char *argv[])
{
    if(argc < 2) disp_error(usage);

    if(!strcmp(argv[1], "help"))
    {
        disp_error(usage);
    }
    
    else if(!strcmp(argv[1], "refresh"))
    {
        refresh();
    }
    
    else if(!strcmp(argv[1], "kill"))
    {
        if(!strcmp(argv[2], "all"))
        {
            kill_all();
        }
        
        else kill(argv[2]);
    }
}