+ | Process class that will be used as base for Threat class

Author: kaveOO

Threat/Process.cpp

@@ -0,0 +1,42 @@
+#include "Process.hpp"
+
+Process::Process(const wchar_t *processName) {
+	HANDLE			hSnapshot;
+	PROCESSENTRY32	processEntry = {};
+	BOOL			hResult;
+
+	hSnapshot = CreateToolhelp32Snapshot(TH32CS_SNAPPROCESS, 0);
+	if (hSnapshot == INVALID_HANDLE_VALUE) {
+		std::cerr << "Failed to create snapshot of processes" << std::endl;
+		return;
+	}
+
+	processEntry.dwSize = sizeof(PROCESSENTRY32);
+	hResult = Process32First(hSnapshot, &processEntry);
+
+	while (hResult) {
+		if (wcscmp(processName, processEntry.szExeFile) == 0) {
+			processId = processEntry.th32ProcessID;
+			break;
+		}
+		hResult = Process32Next(hSnapshot, &processEntry);
+	}
+
+	if (processId == 0) {
+		std::wcerr << "Failed to find " << processName << std::endl;
+	}
+
+	CloseHandle(hSnapshot);
+}
+
+Process::~Process() {
+	CloseHandle(hProcess);
+}
+
+DWORD Process::GetProcessId() const {
+	return processId;
+}
+
+HANDLE Process::GetProcessHandle() const {
+	return hProcess;
+}

Threat/Process.hpp

@@ -0,0 +1,18 @@
+#pragma once
+
+#include <Windows.h>
+#include <iostream>
+#include <TlHelp32.h>
+
+class Process {
+	private:
+		DWORD	processId = 0;
+		HANDLE	hProcess = NULL;
+	public:
+		Process() = delete;
+		Process(const wchar_t *processName);
+		~Process();
+
+		DWORD	GetProcessId()		const;
+		HANDLE	GetProcessHandle()	const;
+};