管理人の独り言（プログラミング関連）

1070：７７４さん：2008/12/11(木) 20:45:24 ID:CENbthlg0: 同一フォルダ内のSleipnirが一つでも起動していると終了しないプログラム。
-----
#define STRICT 1

#if !defined(UNICODE) && !defined(_UNICODE)
#error UNICODE版でしかコンパイル出来んぞい。
#endif

#pragma comment(lib, "psapi.lib")

#include <iostream>
#include <cstdio>
#include <string>
#include <vector>
#include <windows.h>
#include <lmcons.h>
#include <tlhelp32.h>
#include <psapi.h>
#include <tchar.h>

using namespace std;

std::vector<DWORD> GetSleipnirProcessId(){
std::vector<DWORD> ret;

HANDLE hEnum = CreateToolhelp32Snapshot(TH32CS_SNAPPROCESS, 0);
if(hEnum != INVALID_HANDLE_VALUE){
PROCESSENTRY32W ps = {sizeof(ps)};

if(Process32FirstW(hEnum, &ps)){
WCHAR curpath[MAX_PATH + 1];
DWORD curProcessId = GetCurrentProcessId();
if(GetCurrentDirectoryW(sizeof(curpath) / sizeof(curpath[0]) - 1, curpath)){
CharLowerBuffW(curpath, lstrlenW(curpath));
do{
if(curProcessId != ps.th32ProcessID){
HANDLE hProcess = OpenProcess(PROCESS_QUERY_INFORMATION | PROCESS_VM_READ, FALSE, ps.th32ProcessID);
if(hProcess){
DWORD dummy;
HMODULE hMod;
if(EnumProcessModules(hProcess, &hMod, sizeof(hMod), &dummy)){
WCHAR path[MAX_PATH + 1];
if(GetModuleFileNameExW(hProcess, hMod, path, sizeof(path) / sizeof(path[0]) - 1)){
unsigned int len = lstrlenW(path);
CharLowerBuffW(path, len);
if(curpath[0] == path[0]){
while(path[--len] != L'\\');
path[len] = L'\0';
if(path[len + 1] && path[len + 2] && path[len + 3] && path[len + 4] && path[len + 5] && path[len + 6] && path[len + 7] && path[len + 8] && path[len + 1] == L's' && path[len + 2] == L'l' && path[len + 3] == L'e' && path[len + 4] == L'i' && path[len + 5] == L'p' && path[len + 6] == L'n' && path[len + 7] == L'i' && path[len + 8] == 'r'){
if(lstrcmpW(path, curpath) == 0){
ret.push_back(ps.th32ProcessID);
}
}
}
}
}
}
CloseHandle(hProcess);
}
}while(Process32NextW(hEnum, &ps));
}
}
}
return ret;
}

int main(int argc, char **argv){
//SetPriorityClass(GetCurrentProcess(), HIGH_PRIORITY_CLASS);
std::vector<DWORD> sleipnirs = GetSleipnirProcessId();
while(1){
unsigned int len = sleipnirs.size();
if(len == 0)
break;

HANDLE *p = (HANDLE *)std::malloc(len * sizeof(HANDLE));
for(unsigned int i = 0; i < len; ++i){
p[i] = OpenProcess(SYNCHRONIZE, FALSE, sleipnirs[i]);
}

WaitForMultipleObjects(len, p, TRUE, INFINITE);

for(unsigned int i = 0; i < len; ++i){
CloseHandle(p[i]);
}

std::free(p);
sleipnirs = GetSleipnirProcessId();
}

}

※書き込む際の注意事項はこちら

※画像アップローダーはこちら

（画像を表示できるのは「画像リンクのサムネイル表示」がオンの掲示板に限ります）