Method GetEnvironment

WinSysCore/Processes.cpp:351–391

}

std::vector<std::pair<std::wstring, std::wstring>> Process::GetEnvironment(HANDLE hProcess) {
	std::vector<std::pair<std::wstring, std::wstring>> env;

	PEB peb;
	if (!GetProcessPeb(hProcess, &peb))
		return env;

	RTL_USER_PROCESS_PARAMETERS processParams;
	if (!::ReadProcessMemory(hProcess, peb.ProcessParameters, &processParams, sizeof(processParams), nullptr))
		return env;

	BYTE* buffer = nullptr;
	int size = 1 << 16;
	buffer = (BYTE*)malloc(size);
	if (!buffer) {
		return env;
	}
	for (;;) {
		if (::ReadProcessMemory(hProcess, processParams.Environment, buffer, size, nullptr))
			break;
		size -= 1 << 12;
	}

	env.reserve(64);

	for (auto p = (PWSTR)buffer; *p;) {
		std::pair<std::wstring, std::wstring> var;
		auto equal = wcschr(p, L'=');
		assert(equal);
		if (!equal)
			break;
		*equal = L'\0';
		var.first = p;
		p += ::wcslen(p) + 1;
		var.second = p;
		p += ::wcslen(p) + 1;
		env.push_back(std::move(var));
	}

	return env;
}