| /* |
| * 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; |
| } |