--- /dev/null 2019-02-13 16:01:02.292000000 +0300 +++ new/src/os/windows/vm/pdh_interface.cpp 2019-02-15 19:02:55.802191770 +0300 @@ -0,0 +1,165 @@ +/* + * Copyright (c) 2012, 2018, Oracle and/or its affiliates. All rights reserved. + * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER. + * + * This code is free software; you can redistribute it and/or modify it + * under the terms of the GNU General Public License version 2 only, as + * published by the Free Software Foundation. + * + * This code 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 + * version 2 for more details (a copy is included in the LICENSE file that + * accompanied this code). + * + * You should have received a copy of the GNU General Public License version + * 2 along with this work; if not, write to the Free Software Foundation, + * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA. + * + * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA + * or visit www.oracle.com if you need additional information or have any + * questions. + * + */ + +#include "precompiled.hpp" +#include "pdh_interface.hpp" +#include "runtime/os.hpp" +#include "utilities/macros.hpp" + +// PDH API +typedef PDH_STATUS (WINAPI *PdhAddCounter_Fn)(HQUERY, LPCSTR, DWORD, HCOUNTER*); +typedef PDH_STATUS (WINAPI *PdhOpenQuery_Fn)(LPCWSTR, DWORD, HQUERY*); +typedef DWORD (WINAPI *PdhCloseQuery_Fn)(HQUERY); +typedef PDH_STATUS (WINAPI *PdhCollectQueryData_Fn)(HQUERY); +typedef DWORD (WINAPI *PdhGetFormattedCounterValue_Fn)(HCOUNTER, DWORD, LPDWORD, PPDH_FMT_COUNTERVALUE); +typedef PDH_STATUS (WINAPI *PdhEnumObjectItems_Fn)(LPCTSTR, LPCTSTR, LPCTSTR, LPTSTR, LPDWORD, LPTSTR, LPDWORD, DWORD, DWORD); +typedef PDH_STATUS (WINAPI *PdhRemoveCounter_Fn)(HCOUNTER); +typedef PDH_STATUS (WINAPI *PdhLookupPerfNameByIndex_Fn)(LPCSTR, DWORD, LPSTR, LPDWORD); +typedef PDH_STATUS (WINAPI *PdhMakeCounterPath_Fn)(PDH_COUNTER_PATH_ELEMENTS*, LPTSTR, LPDWORD, DWORD); + +PdhAddCounter_Fn PdhDll::_PdhAddCounter = NULL; +PdhOpenQuery_Fn PdhDll::_PdhOpenQuery = NULL; +PdhCloseQuery_Fn PdhDll::_PdhCloseQuery = NULL; +PdhCollectQueryData_Fn PdhDll::_PdhCollectQueryData = NULL; +PdhGetFormattedCounterValue_Fn PdhDll::_PdhGetFormattedCounterValue = NULL; +PdhEnumObjectItems_Fn PdhDll::_PdhEnumObjectItems = NULL; +PdhRemoveCounter_Fn PdhDll::_PdhRemoveCounter = NULL; +PdhLookupPerfNameByIndex_Fn PdhDll::_PdhLookupPerfNameByIndex = NULL; +PdhMakeCounterPath_Fn PdhDll::_PdhMakeCounterPath = NULL; + +LONG PdhDll::_critical_section = 0; +LONG PdhDll::_initialized = 0; +LONG PdhDll::_pdh_reference_count = 0; +HMODULE PdhDll::_hModule = NULL; + +void PdhDll::initialize(void) { + _hModule = os::win32::load_Windows_dll("pdh.dll", NULL, 0); + if (NULL == _hModule) { + return; + } + // The 'A' at the end means the ANSI (not the UNICODE) vesions of the methods + _PdhAddCounter = (PdhAddCounter_Fn)::GetProcAddress(_hModule, "PdhAddCounterA"); + _PdhOpenQuery = (PdhOpenQuery_Fn)::GetProcAddress(_hModule, "PdhOpenQueryA"); + _PdhCloseQuery = (PdhCloseQuery_Fn)::GetProcAddress(_hModule, "PdhCloseQuery"); + _PdhCollectQueryData = (PdhCollectQueryData_Fn)::GetProcAddress(_hModule, "PdhCollectQueryData"); + _PdhGetFormattedCounterValue = (PdhGetFormattedCounterValue_Fn)::GetProcAddress(_hModule, "PdhGetFormattedCounterValue"); + _PdhEnumObjectItems = (PdhEnumObjectItems_Fn)::GetProcAddress(_hModule, "PdhEnumObjectItemsA"); + _PdhRemoveCounter = (PdhRemoveCounter_Fn)::GetProcAddress(_hModule, "PdhRemoveCounter"); + _PdhLookupPerfNameByIndex = (PdhLookupPerfNameByIndex_Fn)::GetProcAddress(_hModule, "PdhLookupPerfNameByIndexA"); + _PdhMakeCounterPath = (PdhMakeCounterPath_Fn)::GetProcAddress(_hModule, "PdhMakeCounterPathA"); + InterlockedExchange(&_initialized, 1); +} + +bool PdhDll::PdhDetach(void) { + LONG prev_ref_count = InterlockedExchangeAdd(&_pdh_reference_count, -1); + BOOL ret = false; + if (1 == prev_ref_count) { + if (_initialized && _hModule != NULL) { + ret = FreeLibrary(_hModule); + if (ret) { + _hModule = NULL; + _PdhAddCounter = NULL; + _PdhOpenQuery = NULL; + _PdhCloseQuery = NULL; + _PdhCollectQueryData = NULL; + _PdhGetFormattedCounterValue = NULL; + _PdhEnumObjectItems = NULL; + _PdhRemoveCounter = NULL; + _PdhLookupPerfNameByIndex = NULL; + _PdhMakeCounterPath = NULL; + InterlockedExchange(&_initialized, 0); + } + } + } + return ret != 0; +} + +bool PdhDll::PdhAttach(void) { + InterlockedExchangeAdd(&_pdh_reference_count, 1); + if (1 == _initialized) { + return true; + } + while (InterlockedCompareExchange(&_critical_section, 1, 0) == 1); + if (0 == _initialized) { + initialize(); + } + while (InterlockedCompareExchange(&_critical_section, 0, 1) == 0); + return (_PdhAddCounter != NULL && _PdhOpenQuery != NULL + && _PdhCloseQuery != NULL && PdhCollectQueryData != NULL + && _PdhGetFormattedCounterValue != NULL && _PdhEnumObjectItems != NULL + && _PdhRemoveCounter != NULL && PdhLookupPerfNameByIndex != NULL + && _PdhMakeCounterPath != NULL); +} + +PDH_STATUS PdhDll::PdhAddCounter(HQUERY hQuery, LPCSTR szFullCounterPath, DWORD dwUserData, HCOUNTER* phCounter) { + assert(_initialized && _PdhAddCounter != NULL, "PdhAvailable() not yet called"); + return _PdhAddCounter(hQuery, szFullCounterPath, dwUserData, phCounter); +} + +PDH_STATUS PdhDll::PdhOpenQuery(LPCWSTR szDataSource, DWORD dwUserData, HQUERY* phQuery) { + assert(_initialized && _PdhOpenQuery != NULL, "PdhAvailable() not yet called"); + return _PdhOpenQuery(szDataSource, dwUserData, phQuery); +} + +DWORD PdhDll::PdhCloseQuery(HQUERY hQuery) { + assert(_initialized && _PdhCloseQuery != NULL, "PdhAvailable() not yet called"); + return _PdhCloseQuery(hQuery); +} + +PDH_STATUS PdhDll::PdhCollectQueryData(HQUERY hQuery) { + assert(_initialized && _PdhCollectQueryData != NULL, "PdhAvailable() not yet called"); + return _PdhCollectQueryData(hQuery); +} + +DWORD PdhDll::PdhGetFormattedCounterValue(HCOUNTER hCounter, DWORD dwFormat, LPDWORD lpdwType, PPDH_FMT_COUNTERVALUE pValue) { + assert(_initialized && _PdhGetFormattedCounterValue != NULL, "PdhAvailable() not yet called"); + return _PdhGetFormattedCounterValue(hCounter, dwFormat, lpdwType, pValue); +} + +PDH_STATUS PdhDll::PdhEnumObjectItems(LPCTSTR szDataSource, LPCTSTR szMachineName, LPCTSTR szObjectName, + LPTSTR mszCounterList, LPDWORD pcchCounterListLength, LPTSTR mszInstanceList, + LPDWORD pcchInstanceListLength, DWORD dwDetailLevel, DWORD dwFlags) { + assert(_initialized && _PdhEnumObjectItems != NULL, "PdhAvailable() not yet called"); + return _PdhEnumObjectItems(szDataSource, szMachineName, szObjectName, mszCounterList, pcchCounterListLength, + mszInstanceList, pcchInstanceListLength, dwDetailLevel, dwFlags); +} + +PDH_STATUS PdhDll::PdhRemoveCounter(HCOUNTER hCounter) { + assert(_initialized && _PdhRemoveCounter != NULL, "PdhAvailable() not yet called"); + return _PdhRemoveCounter(hCounter); +} + +PDH_STATUS PdhDll::PdhLookupPerfNameByIndex(LPCSTR szMachineName, DWORD dwNameIndex, LPSTR szNameBuffer, LPDWORD pcchNameBufferSize) { + assert(_initialized && _PdhLookupPerfNameByIndex != NULL, "PdhAvailable() not yet called"); + return _PdhLookupPerfNameByIndex(szMachineName, dwNameIndex, szNameBuffer, pcchNameBufferSize); +} + +PDH_STATUS PdhDll::PdhMakeCounterPath(PDH_COUNTER_PATH_ELEMENTS* pCounterPathElements, LPTSTR szFullPathBuffer, LPDWORD pcchBufferSize, DWORD dwFlags) { + assert(_initialized && _PdhMakeCounterPath != NULL, "PdhAvailable() not yet called"); + return _PdhMakeCounterPath(pCounterPathElements, szFullPathBuffer, pcchBufferSize, dwFlags); +} + +bool PdhDll::PdhStatusFail(PDH_STATUS pdhStat) { + return pdhStat != ERROR_SUCCESS && pdhStat != PDH_MORE_DATA; +}