Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance. Project price only 1 $
You can buy this project and download/modify it how often you want.
/**
* MIT License
*
* Copyright (c) 2010 - 2020 The OSHI Project Contributors: https://github.com/oshi/oshi/graphs/contributors
*
* Permission is hereby granted, free of charge, to any person obtaining a copy
* of this software and associated documentation files (the "Software"), to deal
* in the Software without restriction, including without limitation the rights
* to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
* copies of the Software, and to permit persons to whom the Software is
* furnished to do so, subject to the following conditions:
*
* The above copyright notice and this permission notice shall be included in all
* copies or substantial portions of the Software.
*
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
* SOFTWARE.
*/
package oshi.driver.windows.registry;
import java.util.Collection;
import java.util.HashMap;
import java.util.Map;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import com.sun.jna.Pointer; // NOSONAR squid:s1191
import com.sun.jna.platform.win32.Kernel32;
import com.sun.jna.platform.win32.VersionHelpers;
import com.sun.jna.platform.win32.Wtsapi32;
import com.sun.jna.platform.win32.Wtsapi32.WTS_PROCESS_INFO_EX;
import com.sun.jna.platform.win32.COM.WbemcliUtil.WmiResult;
import com.sun.jna.ptr.IntByReference;
import com.sun.jna.ptr.PointerByReference;
import oshi.annotation.concurrent.Immutable;
import oshi.annotation.concurrent.ThreadSafe;
import oshi.driver.windows.wmi.Win32Process;
import oshi.driver.windows.wmi.Win32Process.ProcessXPProperty;
import oshi.util.platform.windows.WmiUtil;
/**
* Utility to read process data from HKEY_PERFORMANCE_DATA information with
* backup from Performance Counters or WMI
*/
@ThreadSafe
public final class ProcessWtsData {
private static final Logger LOG = LoggerFactory.getLogger(ProcessWtsData.class);
private static final boolean IS_WINDOWS7_OR_GREATER = VersionHelpers.IsWindows7OrGreater();
private ProcessWtsData() {
}
/**
* Query the registry for process performance counters
*
* @param pids
* An optional collection of process IDs to filter the list to. May
* be null for no filtering.
* @return A map with Process ID as the key and a {@link WtsInfo} object
* populated with data.
*/
public static Map queryProcessWtsMap(Collection pids) {
if (IS_WINDOWS7_OR_GREATER) {
// Get processes from WTS
return queryProcessWtsMapFromWTS(pids);
}
// Pre-Win7 we can't use WTSEnumerateProcessesEx so we'll grab the
// same info from WMI and fake the array
return queryProcessWtsMapFromPerfMon(pids);
}
private static Map queryProcessWtsMapFromWTS(Collection pids) {
Map wtsMap = new HashMap<>();
IntByReference pCount = new IntByReference(0);
final PointerByReference ppProcessInfo = new PointerByReference();
if (!Wtsapi32.INSTANCE.WTSEnumerateProcessesEx(Wtsapi32.WTS_CURRENT_SERVER_HANDLE,
new IntByReference(Wtsapi32.WTS_PROCESS_INFO_LEVEL_1), Wtsapi32.WTS_ANY_SESSION, ppProcessInfo,
pCount)) {
LOG.error("Failed to enumerate Processes. Error code: {}", Kernel32.INSTANCE.GetLastError());
return wtsMap;
}
// extract the pointed-to pointer and create array
Pointer pProcessInfo = ppProcessInfo.getValue();
final WTS_PROCESS_INFO_EX processInfoRef = new WTS_PROCESS_INFO_EX(pProcessInfo);
WTS_PROCESS_INFO_EX[] processInfo = (WTS_PROCESS_INFO_EX[]) processInfoRef.toArray(pCount.getValue());
for (int i = 0; i < processInfo.length; i++) {
if (pids == null || pids.contains(processInfo[i].ProcessId)) {
wtsMap.put(processInfo[i].ProcessId,
new WtsInfo(processInfo[i].pProcessName, "", processInfo[i].NumberOfThreads,
processInfo[i].PagefileUsage & 0xffff_ffffL,
processInfo[i].KernelTime.getValue() / 10_000L,
processInfo[i].UserTime.getValue() / 10_000, processInfo[i].HandleCount));
}
}
// Clean up memory
if (!Wtsapi32.INSTANCE.WTSFreeMemoryEx(Wtsapi32.WTS_PROCESS_INFO_LEVEL_1, pProcessInfo, pCount.getValue())) {
LOG.warn("Failed to Free Memory for Processes. Error code: {}", Kernel32.INSTANCE.GetLastError());
}
return wtsMap;
}
private static Map queryProcessWtsMapFromPerfMon(Collection pids) {
Map wtsMap = new HashMap<>();
WmiResult processWmiResult = Win32Process.queryProcesses(pids);
for (int i = 0; i < processWmiResult.getResultCount(); i++) {
wtsMap.put(WmiUtil.getUint32(processWmiResult, ProcessXPProperty.PROCESSID, i), new WtsInfo(
WmiUtil.getString(processWmiResult, ProcessXPProperty.NAME, i),
WmiUtil.getString(processWmiResult, ProcessXPProperty.EXECUTABLEPATH, i),
WmiUtil.getUint32(processWmiResult, ProcessXPProperty.THREADCOUNT, i),
// WMI Pagefile usage is in KB
1024 * (WmiUtil.getUint32(processWmiResult, ProcessXPProperty.PAGEFILEUSAGE, i) & 0xffff_ffffL),
WmiUtil.getUint64(processWmiResult, ProcessXPProperty.KERNELMODETIME, i) / 10_000L,
WmiUtil.getUint64(processWmiResult, ProcessXPProperty.USERMODETIME, i) / 10_000L,
WmiUtil.getUint32(processWmiResult, ProcessXPProperty.HANDLECOUNT, i)));
}
return wtsMap;
}
/**
* Class to encapsulate data from WTS Process Info
*/
@Immutable
public static class WtsInfo {
private final String name;
private final String path;
private final int threadCount;
private final long virtualSize;
private final long kernelTime;
private final long userTime;
private final long openFiles;
public WtsInfo(String name, String path, int threadCount, long virtualSize, long kernelTime, long userTime,
long openFiles) {
this.name = name;
this.path = path;
this.threadCount = threadCount;
this.virtualSize = virtualSize;
this.kernelTime = kernelTime;
this.userTime = userTime;
this.openFiles = openFiles;
}
/**
* @return the name
*/
public String getName() {
return name;
}
/**
* @return the path
*/
public String getPath() {
return path;
}
/**
* @return the threadCount
*/
public int getThreadCount() {
return threadCount;
}
/**
* @return the virtualSize
*/
public long getVirtualSize() {
return virtualSize;
}
/**
* @return the kernelTime
*/
public long getKernelTime() {
return kernelTime;
}
/**
* @return the userTime
*/
public long getUserTime() {
return userTime;
}
/**
* @return the openFiles
*/
public long getOpenFiles() {
return openFiles;
}
}
}
