openprocess2.cpp

linux下的终端管理工具源码

// OpenProcess2.cpp : Defines the entry point for the application.
//


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

static char processName[500]={"notepad.exe"};

bool processIsToKill(int , char **);
int  processIsRunning(const char *, bool = false);

// PSAPI Function Pointers.
   BOOL  (WINAPI *lpfEnumProcesses)( DWORD *, DWORD cb, DWORD * );
   BOOL  (WINAPI *lpfEnumProcessModules)( HANDLE, HMODULE *,DWORD, LPDWORD );
   DWORD (WINAPI *lpfGetModuleBaseName)( HANDLE, HMODULE,LPTSTR, DWORD );


// ToolHelp Function Pointers.
   HANDLE (WINAPI *lpfCreateToolhelp32Snapshot)(DWORD,DWORD);
   BOOL   (WINAPI *lpfProcess32First)(HANDLE,LPPROCESSENTRY32);
   BOOL   (WINAPI *lpfProcess32Next)(HANDLE,LPPROCESSENTRY32);
   BOOL   (WINAPI *lpfModule32First)(HANDLE,LPMODULEENTRY32);
   BOOL   (WINAPI *lpfModule32Next)(HANDLE,LPMODULEENTRY32);


// The entry point returns the follow codes:
// 0 process not running.
// 101 process running
// 102 process was running and not succesly terminated
// 103 process was running and succesly terminated
// 104 generic error

bool processIsToKill(int argc, char **argv)
{
   bool ret = false;
   
   if(argc == 3 && strcmp(argv[2],"kill") == 0)
        ret = true;
      
   strcpy(processName , argv[1]);
   return ret;
}

int processIsRunning(const char * procName , bool procTokill)
{
  int result = 0;
    
  return result;
}



int main(int argc , char **argv)
{
 	// TODO: Place code here.

	 int result = 0;
   
   //processIsToKill(argc,argv);
   
   BOOL bResult,bResultm;
	 DWORD aiPID[1000],iCb=1000,iNumProc,iV2000=0;
	 DWORD iCbneeded,i,iFound=0;
	 
   char szName[MAX_PATH],
        szToTermUpper[MAX_PATH];
	 
   HANDLE hProc,
          hSnapShot,
          hSnapShotm;
	 
   OSVERSIONINFO osvi;
   HINSTANCE hInstLib;
	 
   int iLen,
       iLenP,
       indx;
   
   HMODULE hMod;
	 PROCESSENTRY32 procentry;      
	 MODULEENTRY32  modentry;
	 
   int Cntr=0;

	 iLenP=strlen(processName);
	 
   if(iLenP<1 || iLenP>MAX_PATH) return 104;
	 
   for(indx=0;indx<iLenP;indx++)
		szToTermUpper[indx]=toupper(processName[indx]);
	 szToTermUpper[iLenP]=0;

   //First check what version of Windows we're in
	 osvi.dwOSVersionInfoSize = sizeof(OSVERSIONINFO);
     bResult=GetVersionEx(&osvi);
	 if(!bResult)     // Unable to identify system version
	    return 104;

 	 //At Present we only support Win/NT/2000 or Win/9x
	 if((osvi.dwPlatformId != VER_PLATFORM_WIN32_NT) &&
		 (osvi.dwPlatformId != VER_PLATFORM_WIN32_WINDOWS)) return 104;
   
   if(osvi.dwPlatformId==VER_PLATFORM_WIN32_NT)
   {
     hInstLib = LoadLibraryA("PSAPI.DLL");
  
     if(hInstLib == NULL) return 104;

     // Get procedure addresses.
     lpfEnumProcesses = (BOOL(WINAPI *)(DWORD *,DWORD,DWORD*)) GetProcAddress( hInstLib, "EnumProcesses");
     lpfEnumProcessModules = (BOOL(WINAPI *)(HANDLE, HMODULE *,DWORD, LPDWORD)) GetProcAddress( hInstLib,"EnumProcessModules");
     lpfGetModuleBaseName =(DWORD (WINAPI *)(HANDLE, HMODULE,LPTSTR, DWORD )) GetProcAddress( hInstLib,"GetModuleBaseNameA");

     if( lpfEnumProcesses == NULL ||
         lpfEnumProcessModules == NULL ||
         lpfGetModuleBaseName == NULL)
         {
           FreeLibrary(hInstLib);
           return 104;
         }

      
		 bResult=lpfEnumProcesses(aiPID,iCb,&iCbneeded);
		 if(!bResult)
     {
			 FreeLibrary(hInstLib);
			 return 104;
     }
    
    // How many processes are there?
		iNumProc=iCbneeded/sizeof(DWORD);

		// Get and match the name of each process
		Cntr=0;
		for(i=0;i<iNumProc;i++)
		{
			// Get the (module) name for this process
      strcpy(szName,"Unknown");
			
      // First, get a handle to the process
	    hProc=OpenProcess(PROCESS_QUERY_INFORMATION|PROCESS_VM_READ,FALSE,aiPID[i]);
	    
      // Now, get the process name
	    if(hProc)
			{
         if(lpfEnumProcessModules(hProc,&hMod,sizeof(hMod),&iCbneeded) )
			   {
           iLen=lpfGetModuleBaseName(hProc,hMod,szName,MAX_PATH);
			   }
			}
	    CloseHandle(hProc);
			

			if(strcmp(_strupr(szName),szToTermUpper)==0)
			{
				// Process found, now terminate it
				iFound=1;
				
        if(processIsToKill(argc, argv))
        {
          hProc=OpenProcess(PROCESS_TERMINATE,FALSE,aiPID[i]);
				  TerminateProcess(hProc,0);
          CloseHandle(hProc);
        }

        if(hProc)
				   Cntr++;

				if(Cntr==2)
				{
					return 103;
        }
			}
		}
		return 0;
   }
  return result;
}