跳到主要內容

WINAPI - Process

前言

會開始研究C++去控制Process的原因，是因為專案中用到的Selenium會無預警的Hang住，接著工作都無法繼續執行下去。在Linux的部分，我寫了一個Script去kill它。但Windows的部分該怎麼辦呢? 其實可以透過taskkill的指令。Selenium是透過Java去啟動執行的，因此它依賴於Java Process(java.exe)。但如果我們輸入taskkill /im java.exe，相信會發生你不想看到的事情。因此我們該如何取得Selenium的Java Process呢?其實只要能夠取得執行參數就能做到。小弟所購買的Windows程式設計，僅教導如何取得到Process列表。經過Google後，看到一套叫ProcessExplorer的軟體。

讓我篤信:一定可以辦到的。(OS: Google大神提供的方法很多都是C#的…) 最後終於讓我找到歪外國人寫的API，透過fetch process heap去取得command line。

分析

GetNtProcessInfo所提供的方法BOOL sm_GetNtProcessInfo(const DWORD dwPID, smPROCESSINFO *ppi)，我們必須提供一個pid給它。因此我們需要實作以下內容：

  1. 從所有process中取得java process list。
  2. 將每一個java process透過sm_GetNtProcessInfo去取得各別的cmd line。
  3. 取得cmd line中有出現selenium字眼的java process 。
  4. 刪除掉它!

以上步驟，我們可以學習到：

  1. 取得Process list。
  2. 如何使用sm_GetNtProcessInfo API。
  3. 刪除Process list。

看起來雖然才三個項目，但這之中蘊含著無數的精華!

實作

ProcessUtil

首先我實作了一個ProcessUtil類別，它提供取得ProcEntry列表(getProcEntryList)、根據Process名稱取得ProcEntry列表(findProcEntryList)、根據pid取得ProcEntry(findProcEntry)與終結行程(terminateProcess)。以前在寫作業的時候，傳入字串的參數不是用string就是用char*。在看了Charles(這名字好眼熟?)的Windows程式設計後，得知Windows使用wchar去解決Unicode的問題。使用TCHAR是比較好的寫法，TCHAR的宣告可透過前處理器去決定用wchar還是char。像跨平台的程式碼，型態的宣告也會透過前處理器去決定。

#include "stdafx.h"
#include <list>
#include <map>
using namespace std;
 
class ProcessUtil {
public:
	static list<PROCESSENTRY32> getProcEntryList();
	static list<PROCESSENTRY32> findProcEntryList(_TCHAR* aProcName);
	static PPROCESSENTRY32 findProcEntry(DWORD aPid);
        static BOOL terminateProcess(DWORD aPid);
private:
	static map<DWORD,PROCESSENTRY32> getProcEntryWithPidMap();
};

getProcEntryWithPidMap將會回傳一個以pid為key，procEntry為value的map。實作原理是透過CreateToolhelp32Snapshot建立Process的Snapshot，接著透過Process32First找尋第一個Process，最後透過Process32Next將所有的Process資訊塞入map中。基本上書上或網路上找到的code都大同小異，這裡我把它做成map為了便於之後的search。

map<DWORD,PROCESSENTRY32> ProcessUtil::getProcEntryWithPidMap(){
	map<DWORD,PROCESSENTRY32> pidPeMap;
	HANDLE hProcessSnap = CreateToolhelp32Snapshot( TH32CS_SNAPPROCESS, 0 );
	if( hProcessSnap == INVALID_HANDLE_VALUE ) {
		return pidPeMap;
	}
 
	PROCESSENTRY32 pe32;
	pe32.dwSize = sizeof( PROCESSENTRY32 );
	if( !Process32First( hProcessSnap, &pe32 ) ){
		CloseHandle( hProcessSnap );          
		return pidPeMap;
	}
 
	do {
		pidPeMap.insert( pair<DWORD,PROCESSENTRY32>(pe32.th32ProcessID, pe32) );
	} while( Process32Next( hProcessSnap, &pe32 ) );
 
	CloseHandle( hProcessSnap );
	return pidPeMap;
}

剩下的三個method沒什麼特別的。getProcEntryList是透過getProcEntryWithPidMap實作；findProcEntryList透過wcscmp比對名稱；findProcEntry透過map去找到對應的value。這裡就和吃飯一樣容易。

list<PROCESSENTRY32> ProcessUtil::getProcEntryList(){
	list<PROCESSENTRY32> porcEntryList;
 
	map<DWORD,PROCESSENTRY32> pidPeMap = getProcEntryWithPidMap();
	map<DWORD,PROCESSENTRY32>::iterator it;
	for ( it=pidPeMap.begin() ; it != pidPeMap.end(); it++ ){
		porcEntryList.push_back((*it).second);
	}
 
	return porcEntryList;
}
 
list<PROCESSENTRY32> ProcessUtil::findProcEntryList(_TCHAR* aProcName){
	list<PROCESSENTRY32> porcEntryList;
 
	map<DWORD,PROCESSENTRY32> pidPeMap = getProcEntryWithPidMap();
	map<DWORD,PROCESSENTRY32>::iterator it;
	for ( it=pidPeMap.begin() ; it != pidPeMap.end(); it++ ){
		if(!wcscmp(aProcName, (*it).second.szExeFile)){
			porcEntryList.push_back((*it).second);
		}
	}
	return porcEntryList;
}
 
PPROCESSENTRY32 ProcessUtil::findProcEntry(DWORD aPid){
	map<DWORD,PROCESSENTRY32> pidPeMap = getProcEntryWithPidMap();
 
	map<DWORD,PROCESSENTRY32>::iterator it= pidPeMap.find(aPid);
	if( it == pidPeMap.end() )
		return NULL;
	return &it->second;
}
 
BOOL ProcessUtil::terminateProcess(DWORD aPid){
	HANDLE hProc = OpenProcess(PROCESS_TERMINATE, BOOL(0), aPid);
	if( hProc ) {
		TerminateProcess( hProc,0);
		return CloseHandle(hProc);
	}
	return false;
}

有了以上的code，我們就可以輕易的取到process的pid和終結行程。

Main

這裡實作以下幾個function：確認行程的Command line是否吻合(isMatchProcess)、終結所有名稱吻合的行程(terminalAllProcess)與終結所有名稱吻合且command line也吻合的行程(terminalMatchProcess)。基本上就是先透過ProcessUtil::findProcEntryList找到名稱吻合的行程列表，接著再透過參考文獻二的API去取得command line做比對。

#include "stdafx.h"
 
#include "stdafx.h"
#include "NTProcessInfo.h"
#include <vector>
#include <iostream>
#include <wchar.h>
using namespace std;
 
#include "ProcessUtil.h"
 
#ifdef _UNICODE
#define tcout wcout
#else
#define tcout cout
#endif
 
enum Action{
	EXIT, DEL_ALL, DEL_CHK
};
 
BOOL isMatchProcess(DWORD aPid, _TCHAR* aCmdLine){
	smPROCESSINFO procInfo;
	sm_GetNtProcessInfo( aPid, &procInfo);
	return wcswcs(procInfo.szCmdLine, aCmdLine) != 0;
}
 
void terminalAllProcess(list<PROCESSENTRY32> aProcList){
	list<PROCESSENTRY32>::iterator it;
	for ( it=aProcList.begin() ; it != aProcList.end(); it++ ){
		ProcessUtil::terminateProcess(it->th32ProcessID);
	}		
}
 
void terminalMatchProcess(list<PROCESSENTRY32> aProcList, _TCHAR* aCmdLine){
	list<PROCESSENTRY32>::iterator it;
	for ( it=aProcList.begin() ; it != aProcList.end(); it++ ){
		if( isMatchProcess(it->th32ProcessID, aCmdLine)) {
			tcout << "Terminal processes: " << it->th32ProcessID << endl;
			ProcessUtil::terminateProcess(it->th32ProcessID);
		}
	}		
}
 
int _tmain(int argc, _TCHAR* argv[])
{
	// check action.
	Action action;
	switch(argc){
	case 2:
		action = DEL_ALL;
		break;
	case 3:
		action = DEL_CHK;
		break;
	default:
		action = EXIT;
		break;
	}
 
	// exit
	if( action == EXIT ){
		return EXIT_SUCCESS;
	}
 
	// probe and delete process
	list<PROCESSENTRY32> foundProcList = ProcessUtil::findProcEntryList(argv[1]);
	int foundProcNumber = foundProcList.size();
	if( !foundProcNumber ){
		tcout << "Not found any process!" << endl;
		return EXIT_SUCCESS;
	} else {
		tcout << "Found " << foundProcNumber << " " << argv[1] << " processes." << endl;
	}
 
	if( action == DEL_ALL ) {
		tcout << "Terminal all processes." << endl;
		terminalAllProcess(foundProcList);
	} else if( action == DEL_CHK ){
		HMODULE hNtDll = sm_LoadNTDLLFunctions();
		if(!hNtDll){
			cout << "Load ntdll failed!" << endl;
			return EXIT_FAILURE;
		}
		sm_EnableTokenPrivilege(SE_DEBUG_NAME);
		terminalMatchProcess(foundProcList, argv[2]);
		sm_FreeNTDLLFunctions(hNtDll);
	}
 
	return EXIT_SUCCESS;
}
 
#include "NTProcessInfo.cpp"

這裡需要注意的是:在使用sm_GetNtProcessInfoLL之前，必須先透過sm_LoadNTDLLFunctions去Load NTDLL，接著再去EnableTokenPrivilege，最後用完要去清掉它。

	HMODULE hNtDll = sm_LoadNTDLLFunctions();
        if(!hNtDll){
		cout << "Load ntdll failed!" << endl;
		return EXIT_FAILURE;
	}
	sm_EnableTokenPrivilege(SE_DEBUG_NAME);
	terminalMatchProcess(foundProcList, argv[2]);
	sm_FreeNTDLLFunctions(hNtDll);

執行結果

如下圖所示，輸入參數為java.exe selenium。四個java processes中，只會刪除與selenium有關的，執行結束後剩下三個java processes。


參考文獻

標籤：

留言

張貼留言

這個網誌中的熱門文章

解決RobotFramework從3.1.2升級到3.2.2之後，Choose File突然會整個Hand住的問題

考慮到自動測試環境的維護，我們很久以前就使用java去執行robot framework。前陣子開始處理從3.1.2升級到3.2.2的事情，主要先把明確的runtime語法錯誤與deprecate item處理好，這部分內容可以參考: link 。 直到最近才發現，透過SeleniumLibrary執行Choose File去上傳檔案的動作，會導致測試案例timeout。本篇文章主要分享心路歷程與解決方法，我也送了一條issue給robot framework: link 。 我的環境如下: RobotFramework: 3.2.2 Selenium: 3.141.0 SeleniumLibrary: 3.3.1 Remote Selenium Version: selenium-server-standalone-3.141.59 首先並非所有Choose File的動作都會hang住，有些測試案例是可以執行的，但是上傳一個作業系統ISO檔案一定會發生問題。後來我透過wireshark去比對新舊版本的上傳動作，因為我使用 Remote Selenium ，所以Selenium會先把檔案透過REST API發送到Remote Selenium Server上。從下圖我們可以發現，在3.2.2的最後一個TCP封包，比3.1.2大概少了500個bytes。 於是就開始了我trace code之路。包含SeleniumLibrary產生要送給Remote Selenium Server的request內容，還有HTTP Content-Length的計算，我都確認過沒有問題。 最後發現問題是出在socket API的使用上，就是下圖的這支code: 最後發現可能因為開始使用nio的方式送資料，但沒處理到尚未送完的資料內容，而導致發生問題。加一個loop去做計算就可以解決了。 最後我有把解法提供給robot framework官方，在他們出新的版本之前，我是將改完的_socket.py放在我們自己的Lib底下，好讓我們測試可以正常進行。(shutil.py應該也是為了解某個bug而產生的樣子..)
張貼留言
閱讀完整內容

第一次寫MIB就上手

SNMP(Simple Network Management Protocol)是用來管理網路設備的一種Protocol，我對它的認識也是從工作接觸開始。雖說是管理網路設備，但是主機、電源供應器、RAID等也都可以透過它來做管理。如果你做了一個應用程式，當然所有的操作也都可以透過SNMP來完成，不過可能會很痛苦。前陣子遇到一個學弟，它告訴我說:「我可能不會想寫程式。」為什麼? 因為這是他痛苦的根源。 在這篇文章中，不是要告訴你SNMP是什麼，會看這篇文章的大哥們，應該已經對SNMP有些認識了。 是的!主題是MIB(Management information base)! 對於一個3th-party的SNMP oid，有MIB可以幫助你去了解它所提供的資訊是什麼，且可以對它做什麼操作。最近我運氣很好剛好做到關於修改MIB的工作，也讓我順便了解一下它的語法，接下來我要交給大家MIB的基礎認識。 smidump 我並非使用什麼高強的Editor去編寫MIB，我僅透過Nodepad++編輯和smidump編譯而已。smidump是Kay教我使用的一個將MIB module轉成樹狀結構或oid列表的工具，唯一的缺點是不會告訴你哪一行打錯。當然有錢直接買編輯樹狀結構的工具就可以不需要了解語法了! 安裝 在Ubuntu上可先輸入smidump確認是否安裝，如果沒安裝可透過apt-get install libsmi2ldbl安裝。(CentOS可以透過yum install libsmi) root@tonylin:~/multi-boot-server# smidump The program 'smidump' is currently not installed. You can install it by typing: apt-get install libsmi2ldbl 使用 透過下面兩行指令，就可以將mib file產生出對應的tree與oid列表的檔案。也可以透過這個結果確認MIB是不是你想要的。 smidump -f tree example1.mib > xtree.txt smidump -f identifiers example1.mib > xiden.txt 如果有參考其它檔案要加上p的參數: smidum...
張貼留言
閱讀完整內容

Windows DLL - 32-bit dll with 64-bit driver

前言 在64-bit作業系統上所使用的驅動程式，一定是64-bit，然而應用程式卻可能是32或64-bit。當32-bit應用程式傳值給驅動程式時是有可能會發生溢位的。主要原因是32與64-bit指標所佔用記憶體長度的不同。接下來我將透過Reference 1中的程式PhyMem，來告訴大家問題在哪與如何修改。(最後改完的程式碼恕我不提供) PhyMem介紹 這是一個中國人寫的程式。作用與WinIO相同，可以存取windows的io port與physical memory。這個程式包含pmdll、driver與test三個專案，分別產生dll、sys與exe。作者僅提供32-bit的版本，但只要修改編譯設定就可以讓它產生出64-bit的artifact。但如同我前言所說，如果你是32-bit的dll要存取64的sys該怎麼辦? 可以規定User在64-bit的OS用64-bit的應用程式就好了阿! 但是將一個32-bit應用程式改為64-bit的有這麼簡單嗎? 除此之外，中間傳遞的資料型態也會影響到正常功能。 設定driver專案 我開發環境是VC2008與DDK6000，下載Reference1的專案是無法直接編譯的，可以參考我的設定去修改編譯、連結參數。輸出檔可以看個人需求，根據32或64命名，或者是用同一個檔案名稱。 32-bit C/C++ > 一般 > 其它Include目錄($(DDKROOT)為設定於環境變數的DDK安裝目錄): $(DDKROOT)\inc\ddk";"$(DDKROOT)\inc\api";"$(DDKROOT)\inc\crt";"$(DDKROOT)\inc\crt\gl";"$(DDKROOT)\inc\crt\sys" 連結器 > 一般 > 其它程式庫目錄: $(DDKROOT)\lib\wnet\i386。 連結器 > 資訊清單檔: 將產生資訊清單與UAC選擇否，因為這不適用於driver。 連結器 > 進階 > 隨機化的基底位置: 選擇預設。 64-bit Reference1僅提供32-bit設定。64-bit可在建置>組態管理員中，新增x64平台，而設定檔可從win32複製過來修...
張貼留言
閱讀完整內容