#include <iostream>
#include <windows.h>
#include <tlhelp32.h>
#include <vector>
#include <fstream>
#include <map>
using namespace std;

// 需要關(guān)閉進(jìn)程名字所在文件路徑
#define CLOSE_FILE_PATH "names.txt"
// 每CHECK_INTERVAL時(shí)間(毫秒)后檢查并結(jié)束一次進(jìn)程
#define CHECK_INTERVAL 3000

class Controller {

private:
 
 // 進(jìn)程信息結(jié)構(gòu)體，包含進(jìn)程id和進(jìn)程名
 struct PInfo {
  long pId;
  wchar_t *pName;

  PInfo(long pId, wchar_t *pName) : pId(pId), pName(pName) {}
 };

 // 根據(jù)pid關(guān)閉進(jìn)程
 static int closeProcess(unsigned long pid) {

  HANDLE hProcess = OpenProcess(PROCESS_TERMINATE, FALSE, pid);
 
  if (hProcess) {
   TerminateProcess(hProcess, -1);
  }
  
  return 0;
 }

 // 獲取所有進(jìn)程信息
 static map<wstring, long> getProcessInfo() {
  HANDLE hProcess = CreateToolhelp32Snapshot(TH32CS_SNAPPROCESS, 0);
  PROCESSENTRY32 currentProcess;
  currentProcess.dwSize = sizeof(currentProcess);

  map<wstring, long> pInfosMap = map<wstring, long>();

  Process32First(hProcess, &currentProcess);

  wchar_t *pNameStr;
  bool flag = true;

  while (flag) {
   pNameStr = new WCHAR[MAX_PATH];

   // wcscpy(pNameStr, currentProcess.szExeFile);
   wcscpy_s(pNameStr, wcslen(currentProcess.szExeFile) + 1, currentProcess.szExeFile);

   pInfosMap.insert(pair<wstring, long>(pNameStr, currentProcess.th32ProcessID));
   flag = Process32Next(hProcess, &currentProcess);
  }

  return pInfosMap;
 }

 // char*轉(zhuǎn)wchar*
 static wchar_t * charToWchar(const char* cchar)
 {
  wchar_t *m_wchar;
  int len = MultiByteToWideChar(CP_ACP, 0, cchar, strlen(cchar), NULL, 0);
  m_wchar = new wchar_t[len + 1];
  MultiByteToWideChar(CP_ACP, 0, cchar, strlen(cchar), m_wchar, len);
  m_wchar[len] = '\0';
  return m_wchar;
 }

 // 循環(huán)獲取需要關(guān)閉的進(jìn)程名
 template <typename Callback>
 static void forClosePName(Callback callback) {
  ifstream examplefile(CLOSE_FILE_PATH);
  if (!examplefile.is_open()) {
   cout << "Error opening file";
   exit(1);
  }

  char buffer[260];
  
  while (!examplefile.eof()) {

   examplefile.getline(buffer, 260);

   callback(buffer);
  }
 }

 // 將需要關(guān)閉的進(jìn)程名保存到vector<wchar_t*>中返回
 vector<wchar_t*> getClosePName() {
  vector<wchar_t*> closePNames =  vector<wchar_t*>();
  forClosePName([&](auto pName) {
   closePNames.push_back(charToWchar(pName));
  });
  return closePNames;
 }

public:
 Controller() {

  // 需要結(jié)束的進(jìn)程名
  vector<wchar_t*> closePNames = getClosePName();
  // 所有進(jìn)程信息map
  map<wstring, long> processMap;

  while (true) {

   processMap = getProcessInfo();

   for (auto pName : closePNames) {
    
    //printf("%ls \n", pName);

    long closePId = processMap[pName];

    if (closePId == 0 || wstring(pName) == L"系統(tǒng)空閑進(jìn)程")
     continue;

    closeProcess(closePId);
   }

   Sleep(CHECK_INTERVAL);
  }
 }
};

int main() {

 Controller();

 return 0;
}