mirror of
				https://github.com/chibicitiberiu/rainmeter-studio.git
				synced 2024-02-24 04:33:31 +00:00 
			
		
		
		
	Fixed line endings and applied gitignore
This commit is contained in:
		@@ -1,40 +1,40 @@
 | 
			
		||||
#include <VerRsrc.h>
 | 
			
		||||
#include "../../Version.h"
 | 
			
		||||
 | 
			
		||||
/////////////////////////////////////////////////////////////////////////////
 | 
			
		||||
//
 | 
			
		||||
// Version
 | 
			
		||||
//
 | 
			
		||||
 | 
			
		||||
VS_VERSION_INFO VERSIONINFO
 | 
			
		||||
 FILEVERSION 1,3,0,0
 | 
			
		||||
 PRODUCTVERSION PRODUCTVER
 | 
			
		||||
 FILEFLAGSMASK 0x17L
 | 
			
		||||
#ifdef _DEBUG
 | 
			
		||||
 FILEFLAGS VS_FF_DEBUG
 | 
			
		||||
#else
 | 
			
		||||
 FILEFLAGS 0x0L
 | 
			
		||||
#endif
 | 
			
		||||
 FILEOS VOS_NT_WINDOWS32
 | 
			
		||||
 FILETYPE VFT_DLL
 | 
			
		||||
 FILESUBTYPE VFT_UNKNOWN
 | 
			
		||||
{
 | 
			
		||||
    BLOCK "StringFileInfo"
 | 
			
		||||
    {
 | 
			
		||||
        BLOCK "040904E4"
 | 
			
		||||
        {
 | 
			
		||||
            VALUE "FileVersion", "1.3.0.0"
 | 
			
		||||
            VALUE "LegalCopyright", "<22> 2010 - David Negstad"
 | 
			
		||||
            VALUE "ProductName", "Rainmeter"
 | 
			
		||||
#ifdef _WIN64
 | 
			
		||||
            VALUE "ProductVersion", STRPRODUCTVER " (64-bit)"
 | 
			
		||||
#else
 | 
			
		||||
            VALUE "ProductVersion", STRPRODUCTVER " (32-bit)"
 | 
			
		||||
#endif //_WIN64
 | 
			
		||||
        }
 | 
			
		||||
    }
 | 
			
		||||
    BLOCK "VarFileInfo"
 | 
			
		||||
    {
 | 
			
		||||
        VALUE "Translation", 0x409, 1252
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
#include <VerRsrc.h>
 | 
			
		||||
#include "../../Version.h"
 | 
			
		||||
 | 
			
		||||
/////////////////////////////////////////////////////////////////////////////
 | 
			
		||||
//
 | 
			
		||||
// Version
 | 
			
		||||
//
 | 
			
		||||
 | 
			
		||||
VS_VERSION_INFO VERSIONINFO
 | 
			
		||||
 FILEVERSION 1,3,0,0
 | 
			
		||||
 PRODUCTVERSION PRODUCTVER
 | 
			
		||||
 FILEFLAGSMASK 0x17L
 | 
			
		||||
#ifdef _DEBUG
 | 
			
		||||
 FILEFLAGS VS_FF_DEBUG
 | 
			
		||||
#else
 | 
			
		||||
 FILEFLAGS 0x0L
 | 
			
		||||
#endif
 | 
			
		||||
 FILEOS VOS_NT_WINDOWS32
 | 
			
		||||
 FILETYPE VFT_DLL
 | 
			
		||||
 FILESUBTYPE VFT_UNKNOWN
 | 
			
		||||
{
 | 
			
		||||
    BLOCK "StringFileInfo"
 | 
			
		||||
    {
 | 
			
		||||
        BLOCK "040904E4"
 | 
			
		||||
        {
 | 
			
		||||
            VALUE "FileVersion", "1.3.0.0"
 | 
			
		||||
            VALUE "LegalCopyright", "<22> 2010 - David Negstad"
 | 
			
		||||
            VALUE "ProductName", "Rainmeter"
 | 
			
		||||
#ifdef _WIN64
 | 
			
		||||
            VALUE "ProductVersion", STRPRODUCTVER " (64-bit)"
 | 
			
		||||
#else
 | 
			
		||||
            VALUE "ProductVersion", STRPRODUCTVER " (32-bit)"
 | 
			
		||||
#endif //_WIN64
 | 
			
		||||
        }
 | 
			
		||||
    }
 | 
			
		||||
    BLOCK "VarFileInfo"
 | 
			
		||||
    {
 | 
			
		||||
        VALUE "Translation", 0x409, 1252
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 
 | 
			
		||||
@@ -1,36 +1,36 @@
 | 
			
		||||
<?xml version="1.0" encoding="utf-8"?>
 | 
			
		||||
<Project DefaultTargets="Build" ToolsVersion="4.0" xmlns="http://schemas.microsoft.com/developer/msbuild/2003">
 | 
			
		||||
  <Import Project="$(SolutionDir)Build\VS\Rainmeter.Cpp.Default.props" />
 | 
			
		||||
  <PropertyGroup Label="Globals">
 | 
			
		||||
    <ProjectGuid>{17D3BD92-6F5D-438C-A89B-88F4CE06DB94}</ProjectGuid>
 | 
			
		||||
    <ConfigurationType>DynamicLibrary</ConfigurationType>
 | 
			
		||||
    <TargetName>ResMon</TargetName>
 | 
			
		||||
  </PropertyGroup>
 | 
			
		||||
  <Import Project="$(VCTargetsPath)\Microsoft.Cpp.Default.props" />
 | 
			
		||||
  <Import Project="$(VCTargetsPath)\Microsoft.Cpp.props" />
 | 
			
		||||
  <Import Project="$(SolutionDir)Build\VS\Rainmeter.Cpp.props" />
 | 
			
		||||
  <Import Project="$(SolutionDir)Build\VS\RainmeterPlugin.Cpp.props" />
 | 
			
		||||
  <ImportGroup Label="ExtensionSettings">
 | 
			
		||||
  </ImportGroup>
 | 
			
		||||
  <ImportGroup Label="PropertySheets">
 | 
			
		||||
    <Import Project="$(UserRootDir)\Microsoft.Cpp.$(Platform).user.props" Condition="exists('$(UserRootDir)\Microsoft.Cpp.$(Platform).user.props')" Label="LocalAppDataPlatform" />
 | 
			
		||||
  </ImportGroup>
 | 
			
		||||
  <PropertyGroup Label="UserMacros" />
 | 
			
		||||
  <ItemDefinitionGroup>
 | 
			
		||||
    <ClCompile>
 | 
			
		||||
      <PreprocessorDefinitions>_USRDLL;%(PreprocessorDefinitions)</PreprocessorDefinitions>
 | 
			
		||||
    </ClCompile>
 | 
			
		||||
    <Link>
 | 
			
		||||
      <AdditionalDependencies>Psapi.lib;%(AdditionalDependencies)</AdditionalDependencies>
 | 
			
		||||
    </Link>
 | 
			
		||||
  </ItemDefinitionGroup>
 | 
			
		||||
  <ItemGroup>
 | 
			
		||||
    <ResourceCompile Include="PluginResMon.rc" />
 | 
			
		||||
  </ItemGroup>
 | 
			
		||||
  <ItemGroup>
 | 
			
		||||
    <ClCompile Include="ResMon.cpp" />
 | 
			
		||||
  </ItemGroup>
 | 
			
		||||
  <Import Project="$(VCTargetsPath)\Microsoft.Cpp.targets" />
 | 
			
		||||
  <ImportGroup Label="ExtensionTargets">
 | 
			
		||||
  </ImportGroup>
 | 
			
		||||
<?xml version="1.0" encoding="utf-8"?>
 | 
			
		||||
<Project DefaultTargets="Build" ToolsVersion="4.0" xmlns="http://schemas.microsoft.com/developer/msbuild/2003">
 | 
			
		||||
  <Import Project="$(SolutionDir)Build\VS\Rainmeter.Cpp.Default.props" />
 | 
			
		||||
  <PropertyGroup Label="Globals">
 | 
			
		||||
    <ProjectGuid>{17D3BD92-6F5D-438C-A89B-88F4CE06DB94}</ProjectGuid>
 | 
			
		||||
    <ConfigurationType>DynamicLibrary</ConfigurationType>
 | 
			
		||||
    <TargetName>ResMon</TargetName>
 | 
			
		||||
  </PropertyGroup>
 | 
			
		||||
  <Import Project="$(VCTargetsPath)\Microsoft.Cpp.Default.props" />
 | 
			
		||||
  <Import Project="$(VCTargetsPath)\Microsoft.Cpp.props" />
 | 
			
		||||
  <Import Project="$(SolutionDir)Build\VS\Rainmeter.Cpp.props" />
 | 
			
		||||
  <Import Project="$(SolutionDir)Build\VS\RainmeterPlugin.Cpp.props" />
 | 
			
		||||
  <ImportGroup Label="ExtensionSettings">
 | 
			
		||||
  </ImportGroup>
 | 
			
		||||
  <ImportGroup Label="PropertySheets">
 | 
			
		||||
    <Import Project="$(UserRootDir)\Microsoft.Cpp.$(Platform).user.props" Condition="exists('$(UserRootDir)\Microsoft.Cpp.$(Platform).user.props')" Label="LocalAppDataPlatform" />
 | 
			
		||||
  </ImportGroup>
 | 
			
		||||
  <PropertyGroup Label="UserMacros" />
 | 
			
		||||
  <ItemDefinitionGroup>
 | 
			
		||||
    <ClCompile>
 | 
			
		||||
      <PreprocessorDefinitions>_USRDLL;%(PreprocessorDefinitions)</PreprocessorDefinitions>
 | 
			
		||||
    </ClCompile>
 | 
			
		||||
    <Link>
 | 
			
		||||
      <AdditionalDependencies>Psapi.lib;%(AdditionalDependencies)</AdditionalDependencies>
 | 
			
		||||
    </Link>
 | 
			
		||||
  </ItemDefinitionGroup>
 | 
			
		||||
  <ItemGroup>
 | 
			
		||||
    <ResourceCompile Include="PluginResMon.rc" />
 | 
			
		||||
  </ItemGroup>
 | 
			
		||||
  <ItemGroup>
 | 
			
		||||
    <ClCompile Include="ResMon.cpp" />
 | 
			
		||||
  </ItemGroup>
 | 
			
		||||
  <Import Project="$(VCTargetsPath)\Microsoft.Cpp.targets" />
 | 
			
		||||
  <ImportGroup Label="ExtensionTargets">
 | 
			
		||||
  </ImportGroup>
 | 
			
		||||
</Project>
 | 
			
		||||
@@ -1,187 +1,187 @@
 | 
			
		||||
/*
 | 
			
		||||
  Copyright (C) 2004 David Negstad
 | 
			
		||||
 | 
			
		||||
  This program is free software; you can redistribute it and/or
 | 
			
		||||
  modify it under the terms of the GNU General Public License
 | 
			
		||||
  as published by the Free Software Foundation; either version 2
 | 
			
		||||
  of the License, or (at your option) any later version.
 | 
			
		||||
 | 
			
		||||
  This program is distributed in the hope that it will be useful,
 | 
			
		||||
  but WITHOUT ANY WARRANTY; without even the implied warranty of
 | 
			
		||||
  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 | 
			
		||||
  GNU General Public License for more details.
 | 
			
		||||
 | 
			
		||||
  You should have received a copy of the GNU General Public License
 | 
			
		||||
  along with this program; if not, write to the Free Software
 | 
			
		||||
  Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
 | 
			
		||||
*/
 | 
			
		||||
 | 
			
		||||
/*
 | 
			
		||||
  Requires Win2k or later.  Sorry to anyone still using 9x or ME.
 | 
			
		||||
  All my results seem to agree with Task Manager (it gets tedious adding up
 | 
			
		||||
  the different process values).  Already proved useful in identifying
 | 
			
		||||
  resource leaks in itself (all fixed).  There's irony for you!
 | 
			
		||||
*/
 | 
			
		||||
 | 
			
		||||
#include <windows.h>
 | 
			
		||||
#include <stdio.h>
 | 
			
		||||
#include <psapi.h>
 | 
			
		||||
#include "../../Common/RawString.h"
 | 
			
		||||
#include "../../Library/Export.h"	// Rainmeter's exported functions
 | 
			
		||||
 | 
			
		||||
// system resources that can be counted
 | 
			
		||||
enum MEASURETYPE
 | 
			
		||||
{
 | 
			
		||||
	GDI_COUNT,
 | 
			
		||||
	USER_COUNT,
 | 
			
		||||
	HANDLE_COUNT,
 | 
			
		||||
	WINDOW_COUNT
 | 
			
		||||
};
 | 
			
		||||
 | 
			
		||||
struct MeasureData
 | 
			
		||||
{
 | 
			
		||||
	MEASURETYPE type;
 | 
			
		||||
	RawString process;
 | 
			
		||||
 | 
			
		||||
	MeasureData() : type(GDI_COUNT) {}
 | 
			
		||||
};
 | 
			
		||||
 | 
			
		||||
// used to track the number of existing windows
 | 
			
		||||
UINT g_WindowCount = 0;
 | 
			
		||||
 | 
			
		||||
// count the windows
 | 
			
		||||
BOOL CALLBACK EnumWindowProc(HWND hWnd, LPARAM lParam)
 | 
			
		||||
{
 | 
			
		||||
	++g_WindowCount;
 | 
			
		||||
	return TRUE;
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
PLUGIN_EXPORT void Initialize(void** data, void* rm)
 | 
			
		||||
{
 | 
			
		||||
	MeasureData* measure = new MeasureData;
 | 
			
		||||
	*data = measure;
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
PLUGIN_EXPORT void Reload(void* data, void* rm, double* maxValue)
 | 
			
		||||
{
 | 
			
		||||
	MeasureData* measure = (MeasureData*)data;
 | 
			
		||||
 | 
			
		||||
	LPCWSTR type = RmReadString(rm, L"ResCountType", L"GDI");
 | 
			
		||||
	if (_wcsicmp(L"GDI", type) == 0)
 | 
			
		||||
	{
 | 
			
		||||
		measure->type = GDI_COUNT;
 | 
			
		||||
	}
 | 
			
		||||
	else if (_wcsicmp(L"USER", type) == 0)
 | 
			
		||||
	{
 | 
			
		||||
		measure->type = USER_COUNT;
 | 
			
		||||
	}
 | 
			
		||||
	else if (_wcsicmp(L"HANDLE", type) == 0)
 | 
			
		||||
	{
 | 
			
		||||
		measure->type = HANDLE_COUNT;
 | 
			
		||||
	}
 | 
			
		||||
	else if (_wcsicmp(L"WINDOW", type) == 0)
 | 
			
		||||
	{
 | 
			
		||||
		measure->type = WINDOW_COUNT;
 | 
			
		||||
	}
 | 
			
		||||
	else
 | 
			
		||||
	{
 | 
			
		||||
		WCHAR buffer[256];
 | 
			
		||||
		_snwprintf_s(buffer, _TRUNCATE, L"ResMon.dll: GDICountType=%s is not valid in [%s]", type, RmGetMeasureName(rm));
 | 
			
		||||
		RmLog(LOG_ERROR, buffer);
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	measure->process = RmReadString(rm, L"ProcessName", L"");
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
PLUGIN_EXPORT double Update(void* data)
 | 
			
		||||
{
 | 
			
		||||
	MeasureData* measure = (MeasureData*)data;
 | 
			
		||||
 | 
			
		||||
	// count the existing window objects
 | 
			
		||||
	if (measure->type == WINDOW_COUNT)
 | 
			
		||||
	{
 | 
			
		||||
		g_WindowCount = 0;
 | 
			
		||||
		EnumChildWindows(nullptr, EnumWindowProc, 0);
 | 
			
		||||
		return g_WindowCount;
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	const WCHAR* processName = measure->process.c_str();
 | 
			
		||||
	bool name = !measure->process.empty();
 | 
			
		||||
 | 
			
		||||
	DWORD aProcesses[1024];
 | 
			
		||||
	DWORD bytesNeeded;
 | 
			
		||||
	WCHAR buffer[1024];
 | 
			
		||||
	HMODULE hMod[1024];
 | 
			
		||||
	DWORD cbNeeded;
 | 
			
		||||
 | 
			
		||||
	if (!EnumProcesses(aProcesses, sizeof(aProcesses), &bytesNeeded))
 | 
			
		||||
	{
 | 
			
		||||
		return 0.0;
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// step through the running processes
 | 
			
		||||
	DWORD flags = PROCESS_QUERY_INFORMATION;
 | 
			
		||||
 | 
			
		||||
	if (name)
 | 
			
		||||
	{
 | 
			
		||||
		flags |= PROCESS_VM_READ;
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	UINT resourceCount = 0;
 | 
			
		||||
	for (UINT i = 0, isize = bytesNeeded / sizeof(DWORD); i < isize; ++i)
 | 
			
		||||
	{
 | 
			
		||||
		HANDLE hProcess = OpenProcess(flags, true, aProcesses[i]);
 | 
			
		||||
		if (hProcess != nullptr)
 | 
			
		||||
		{
 | 
			
		||||
			if (name)
 | 
			
		||||
			{
 | 
			
		||||
				if (EnumProcessModules(hProcess, hMod, sizeof(hMod), &cbNeeded))
 | 
			
		||||
				{
 | 
			
		||||
					if (GetModuleBaseName(hProcess, hMod[0], buffer, sizeof(buffer)))
 | 
			
		||||
					{
 | 
			
		||||
						if (_wcsicmp(buffer, processName) != 0)
 | 
			
		||||
						{
 | 
			
		||||
							CloseHandle(hProcess);
 | 
			
		||||
							continue;
 | 
			
		||||
						}
 | 
			
		||||
					}
 | 
			
		||||
					else
 | 
			
		||||
					{
 | 
			
		||||
						CloseHandle(hProcess);
 | 
			
		||||
						continue;
 | 
			
		||||
					}
 | 
			
		||||
				}
 | 
			
		||||
				else
 | 
			
		||||
				{
 | 
			
		||||
					CloseHandle(hProcess);
 | 
			
		||||
					continue;
 | 
			
		||||
				}
 | 
			
		||||
			}
 | 
			
		||||
 | 
			
		||||
			if (measure->type == GDI_COUNT)
 | 
			
		||||
			{
 | 
			
		||||
				resourceCount += GetGuiResources(hProcess, GR_GDIOBJECTS);
 | 
			
		||||
			}
 | 
			
		||||
			else if (measure->type == USER_COUNT)
 | 
			
		||||
			{
 | 
			
		||||
				resourceCount += GetGuiResources(hProcess, GR_USEROBJECTS);
 | 
			
		||||
			}
 | 
			
		||||
			else if (measure->type == HANDLE_COUNT)
 | 
			
		||||
			{
 | 
			
		||||
				DWORD tempHandleCount = 0;
 | 
			
		||||
				GetProcessHandleCount(hProcess, &tempHandleCount);
 | 
			
		||||
				resourceCount += tempHandleCount;
 | 
			
		||||
			}
 | 
			
		||||
		}
 | 
			
		||||
		CloseHandle(hProcess);
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	return resourceCount;
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
PLUGIN_EXPORT void Finalize(void* data)
 | 
			
		||||
{
 | 
			
		||||
	MeasureData* measure = (MeasureData*)data;
 | 
			
		||||
	delete measure;
 | 
			
		||||
}
 | 
			
		||||
/*
 | 
			
		||||
  Copyright (C) 2004 David Negstad
 | 
			
		||||
 | 
			
		||||
  This program is free software; you can redistribute it and/or
 | 
			
		||||
  modify it under the terms of the GNU General Public License
 | 
			
		||||
  as published by the Free Software Foundation; either version 2
 | 
			
		||||
  of the License, or (at your option) any later version.
 | 
			
		||||
 | 
			
		||||
  This program is distributed in the hope that it will be useful,
 | 
			
		||||
  but WITHOUT ANY WARRANTY; without even the implied warranty of
 | 
			
		||||
  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 | 
			
		||||
  GNU General Public License for more details.
 | 
			
		||||
 | 
			
		||||
  You should have received a copy of the GNU General Public License
 | 
			
		||||
  along with this program; if not, write to the Free Software
 | 
			
		||||
  Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
 | 
			
		||||
*/
 | 
			
		||||
 | 
			
		||||
/*
 | 
			
		||||
  Requires Win2k or later.  Sorry to anyone still using 9x or ME.
 | 
			
		||||
  All my results seem to agree with Task Manager (it gets tedious adding up
 | 
			
		||||
  the different process values).  Already proved useful in identifying
 | 
			
		||||
  resource leaks in itself (all fixed).  There's irony for you!
 | 
			
		||||
*/
 | 
			
		||||
 | 
			
		||||
#include <windows.h>
 | 
			
		||||
#include <stdio.h>
 | 
			
		||||
#include <psapi.h>
 | 
			
		||||
#include "../../Common/RawString.h"
 | 
			
		||||
#include "../../Library/Export.h"	// Rainmeter's exported functions
 | 
			
		||||
 | 
			
		||||
// system resources that can be counted
 | 
			
		||||
enum MEASURETYPE
 | 
			
		||||
{
 | 
			
		||||
	GDI_COUNT,
 | 
			
		||||
	USER_COUNT,
 | 
			
		||||
	HANDLE_COUNT,
 | 
			
		||||
	WINDOW_COUNT
 | 
			
		||||
};
 | 
			
		||||
 | 
			
		||||
struct MeasureData
 | 
			
		||||
{
 | 
			
		||||
	MEASURETYPE type;
 | 
			
		||||
	RawString process;
 | 
			
		||||
 | 
			
		||||
	MeasureData() : type(GDI_COUNT) {}
 | 
			
		||||
};
 | 
			
		||||
 | 
			
		||||
// used to track the number of existing windows
 | 
			
		||||
UINT g_WindowCount = 0;
 | 
			
		||||
 | 
			
		||||
// count the windows
 | 
			
		||||
BOOL CALLBACK EnumWindowProc(HWND hWnd, LPARAM lParam)
 | 
			
		||||
{
 | 
			
		||||
	++g_WindowCount;
 | 
			
		||||
	return TRUE;
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
PLUGIN_EXPORT void Initialize(void** data, void* rm)
 | 
			
		||||
{
 | 
			
		||||
	MeasureData* measure = new MeasureData;
 | 
			
		||||
	*data = measure;
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
PLUGIN_EXPORT void Reload(void* data, void* rm, double* maxValue)
 | 
			
		||||
{
 | 
			
		||||
	MeasureData* measure = (MeasureData*)data;
 | 
			
		||||
 | 
			
		||||
	LPCWSTR type = RmReadString(rm, L"ResCountType", L"GDI");
 | 
			
		||||
	if (_wcsicmp(L"GDI", type) == 0)
 | 
			
		||||
	{
 | 
			
		||||
		measure->type = GDI_COUNT;
 | 
			
		||||
	}
 | 
			
		||||
	else if (_wcsicmp(L"USER", type) == 0)
 | 
			
		||||
	{
 | 
			
		||||
		measure->type = USER_COUNT;
 | 
			
		||||
	}
 | 
			
		||||
	else if (_wcsicmp(L"HANDLE", type) == 0)
 | 
			
		||||
	{
 | 
			
		||||
		measure->type = HANDLE_COUNT;
 | 
			
		||||
	}
 | 
			
		||||
	else if (_wcsicmp(L"WINDOW", type) == 0)
 | 
			
		||||
	{
 | 
			
		||||
		measure->type = WINDOW_COUNT;
 | 
			
		||||
	}
 | 
			
		||||
	else
 | 
			
		||||
	{
 | 
			
		||||
		WCHAR buffer[256];
 | 
			
		||||
		_snwprintf_s(buffer, _TRUNCATE, L"ResMon.dll: GDICountType=%s is not valid in [%s]", type, RmGetMeasureName(rm));
 | 
			
		||||
		RmLog(LOG_ERROR, buffer);
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	measure->process = RmReadString(rm, L"ProcessName", L"");
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
PLUGIN_EXPORT double Update(void* data)
 | 
			
		||||
{
 | 
			
		||||
	MeasureData* measure = (MeasureData*)data;
 | 
			
		||||
 | 
			
		||||
	// count the existing window objects
 | 
			
		||||
	if (measure->type == WINDOW_COUNT)
 | 
			
		||||
	{
 | 
			
		||||
		g_WindowCount = 0;
 | 
			
		||||
		EnumChildWindows(nullptr, EnumWindowProc, 0);
 | 
			
		||||
		return g_WindowCount;
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	const WCHAR* processName = measure->process.c_str();
 | 
			
		||||
	bool name = !measure->process.empty();
 | 
			
		||||
 | 
			
		||||
	DWORD aProcesses[1024];
 | 
			
		||||
	DWORD bytesNeeded;
 | 
			
		||||
	WCHAR buffer[1024];
 | 
			
		||||
	HMODULE hMod[1024];
 | 
			
		||||
	DWORD cbNeeded;
 | 
			
		||||
 | 
			
		||||
	if (!EnumProcesses(aProcesses, sizeof(aProcesses), &bytesNeeded))
 | 
			
		||||
	{
 | 
			
		||||
		return 0.0;
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// step through the running processes
 | 
			
		||||
	DWORD flags = PROCESS_QUERY_INFORMATION;
 | 
			
		||||
 | 
			
		||||
	if (name)
 | 
			
		||||
	{
 | 
			
		||||
		flags |= PROCESS_VM_READ;
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	UINT resourceCount = 0;
 | 
			
		||||
	for (UINT i = 0, isize = bytesNeeded / sizeof(DWORD); i < isize; ++i)
 | 
			
		||||
	{
 | 
			
		||||
		HANDLE hProcess = OpenProcess(flags, true, aProcesses[i]);
 | 
			
		||||
		if (hProcess != nullptr)
 | 
			
		||||
		{
 | 
			
		||||
			if (name)
 | 
			
		||||
			{
 | 
			
		||||
				if (EnumProcessModules(hProcess, hMod, sizeof(hMod), &cbNeeded))
 | 
			
		||||
				{
 | 
			
		||||
					if (GetModuleBaseName(hProcess, hMod[0], buffer, sizeof(buffer)))
 | 
			
		||||
					{
 | 
			
		||||
						if (_wcsicmp(buffer, processName) != 0)
 | 
			
		||||
						{
 | 
			
		||||
							CloseHandle(hProcess);
 | 
			
		||||
							continue;
 | 
			
		||||
						}
 | 
			
		||||
					}
 | 
			
		||||
					else
 | 
			
		||||
					{
 | 
			
		||||
						CloseHandle(hProcess);
 | 
			
		||||
						continue;
 | 
			
		||||
					}
 | 
			
		||||
				}
 | 
			
		||||
				else
 | 
			
		||||
				{
 | 
			
		||||
					CloseHandle(hProcess);
 | 
			
		||||
					continue;
 | 
			
		||||
				}
 | 
			
		||||
			}
 | 
			
		||||
 | 
			
		||||
			if (measure->type == GDI_COUNT)
 | 
			
		||||
			{
 | 
			
		||||
				resourceCount += GetGuiResources(hProcess, GR_GDIOBJECTS);
 | 
			
		||||
			}
 | 
			
		||||
			else if (measure->type == USER_COUNT)
 | 
			
		||||
			{
 | 
			
		||||
				resourceCount += GetGuiResources(hProcess, GR_USEROBJECTS);
 | 
			
		||||
			}
 | 
			
		||||
			else if (measure->type == HANDLE_COUNT)
 | 
			
		||||
			{
 | 
			
		||||
				DWORD tempHandleCount = 0;
 | 
			
		||||
				GetProcessHandleCount(hProcess, &tempHandleCount);
 | 
			
		||||
				resourceCount += tempHandleCount;
 | 
			
		||||
			}
 | 
			
		||||
		}
 | 
			
		||||
		CloseHandle(hProcess);
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	return resourceCount;
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
PLUGIN_EXPORT void Finalize(void* data)
 | 
			
		||||
{
 | 
			
		||||
	MeasureData* measure = (MeasureData*)data;
 | 
			
		||||
	delete measure;
 | 
			
		||||
}
 | 
			
		||||
 
 | 
			
		||||
		Reference in New Issue
	
	Block a user