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.
package oshi.hardware.platform.mac;
import oshi.hardware.UsbDevice;
import oshi.hardware.common.AbstractUsbDevice;
import oshi.jna.platform.mac.CoreFoundation;
import oshi.jna.platform.mac.IOKit;
import oshi.util.platform.mac.CfUtil;
import oshi.util.platform.mac.IOKitUtil;
import com.sun.jna.Memory;
import com.sun.jna.Pointer;
import com.sun.jna.ptr.IntByReference;
import com.sun.jna.ptr.LongByReference;
import java.util.*;
/**
* @author angju
*
*/
public class MacUsbDevice extends AbstractUsbDevice {
private static final long serialVersionUID = 2L;
/*
* Maps to store information using RegistryEntryID as the key
*/
private static Map nameMap = new HashMap();
private static Map vendorMap = new HashMap();
private static Map vendorIdMap = new HashMap();
private static Map productIdMap = new HashMap();
private static Map serialMap = new HashMap();
private static Map> hubMap = new HashMap>();
public MacUsbDevice(String name, String vendor, String vendorId, String productId, String serialNumber,
UsbDevice[] connectedDevices) {
super(name, vendor, vendorId, productId, serialNumber, connectedDevices);
}
/**
* {@inheritDoc}
*/
public static UsbDevice[] getUsbDevices(boolean tree) {
UsbDevice[] devices = getUsbDevices();
if (tree) {
return devices;
}
List deviceList = new ArrayList();
// Top level is controllers; they won't be added to the list, but all
// their connected devices will be
for (UsbDevice device : devices) {
addDevicesToList(deviceList, device.getConnectedDevices());
}
return deviceList.toArray(new UsbDevice[deviceList.size()]);
}
private static void addDevicesToList(List deviceList, UsbDevice[] connectedDevices) {
for (UsbDevice device : connectedDevices) {
deviceList.add(new MacUsbDevice(device.getName(), device.getVendor(), device.getVendorId(),
device.getProductId(), device.getSerialNumber(), new MacUsbDevice[0]));
addDevicesToList(deviceList, device.getConnectedDevices());
}
}
private static UsbDevice[] getUsbDevices() {
// Reusable buffer for getting IO name strings
Pointer buffer = new Memory(128); // io_name_t is char[128]
// Empty out maps
nameMap.clear();
vendorMap.clear();
vendorIdMap.clear();
productIdMap.clear();
serialMap.clear();
hubMap.clear();
// Iterate over USB Controllers. All devices are children of one of
// these controllers in the "IOService" plane
List usbControllers = new ArrayList();
IntByReference iter = new IntByReference();
IOKitUtil.getMatchingServices("IOUSBController", iter);
int device = IOKit.INSTANCE.IOIteratorNext(iter.getValue());
while (device != 0) {
// Unique global identifier for this device
LongByReference id = new LongByReference();
IOKit.INSTANCE.IORegistryEntryGetRegistryEntryID(device, id);
usbControllers.add(id.getValue());
// Get device name and store in map
IOKit.INSTANCE.IORegistryEntryGetName(device, buffer);
nameMap.put(id.getValue(), buffer.getString(0));
// The only information we have in registry for this device is the
// locationID. Use that to search for matching PCI device to obtain
// more information.
CoreFoundation.CFTypeRef ref = IOKit.INSTANCE.IORegistryEntryCreateCFProperty(device, CfUtil.getCFString("locationID"),
CfUtil.ALLOCATOR, 0);
if (ref != null && ref.getPointer() != null) {
getControllerIdByLocation(id.getValue(), ref);
}
CfUtil.release(ref);
// Now iterate the children of this device in the "IOService" plane.
// If device parent is root, link to the controller
IntByReference childIter = new IntByReference();
IOKit.INSTANCE.IORegistryEntryGetChildIterator(device, "IOService", childIter);
int childDevice = IOKit.INSTANCE.IOIteratorNext(childIter.getValue());
while (childDevice != 0) {
// Unique global identifier for this device
LongByReference childId = new LongByReference();
IOKit.INSTANCE.IORegistryEntryGetRegistryEntryID(childDevice, childId);
// Get this device's parent in the "IOUSB" plane
IntByReference parent = new IntByReference();
IOKit.INSTANCE.IORegistryEntryGetParentEntry(childDevice, "IOUSB", parent);
// If the parent is not an IOUSBDevice (will be root), set the
// parentId to the controller
LongByReference parentId = new LongByReference();
if (!IOKit.INSTANCE.IOObjectConformsTo(parent.getValue(), "IOUSBDevice")) {
parentId = id;
} else {
// Unique global identifier for the parent
IOKit.INSTANCE.IORegistryEntryGetRegistryEntryID(parent.getValue(), parentId);
}
// Store parent in map
computeIfAbsent(hubMap,parentId.getValue()).add(childId.getValue());
// Get device name and store in map
IOKit.INSTANCE.IORegistryEntryGetName(childDevice, buffer);
nameMap.put(childId.getValue(), buffer.getString(0));
// Get vendor and store in map
String vendor = IOKitUtil.getIORegistryStringProperty(childDevice, "USB Vendor Name");
if (vendor != null) {
vendorMap.put(childId.getValue(), vendor);
}
// Get vendorId and store in map
long vendorId = IOKitUtil.getIORegistryLongProperty(childDevice, "idVendor");
if (vendorId != 0) {
vendorIdMap.put(childId.getValue(), String.format("%04x", 0xffff & vendorId));
}
// Get productId and store in map
long productId = IOKitUtil.getIORegistryLongProperty(childDevice, "idProduct");
if (productId != 0) {
productIdMap.put(childId.getValue(), String.format("%04x", 0xffff & productId));
}
// Get serial and store in map
String serial = IOKitUtil.getIORegistryStringProperty(childDevice, "USB Serial Number");
if (serial != null) {
serialMap.put(childId.getValue(), serial);
}
IOKit.INSTANCE.IOObjectRelease(childDevice);
childDevice = IOKit.INSTANCE.IOIteratorNext(childIter.getValue());
}
IOKit.INSTANCE.IOObjectRelease(childIter.getValue());
IOKit.INSTANCE.IOObjectRelease(device);
device = IOKit.INSTANCE.IOIteratorNext(iter.getValue());
}
IOKit.INSTANCE.IOObjectRelease(iter.getValue());
// Build tree and return
List controllerDevices = new ArrayList();
for (Long controller : usbControllers) {
controllerDevices.add(getDeviceAndChildren(controller, "0000", "0000"));
}
return controllerDevices.toArray(new UsbDevice[controllerDevices.size()]);
}
/**
* Looks up vendor and product id information for a USB Host Controller by
* cross-referencing the location
*
* @param id
* The global unique ID for the host controller used as a key for
* maps
* @param locationId
* The locationID of this controller returned from the registry
*/
private static void getControllerIdByLocation(long id, CoreFoundation.CFTypeRef locationId) {
// Create a matching property dictionary from the locationId
CoreFoundation.CFMutableDictionaryRef propertyDict = CoreFoundation.INSTANCE.CFDictionaryCreateMutable(CfUtil.ALLOCATOR, 0,
null, null);
CoreFoundation.INSTANCE.CFDictionarySetValue(propertyDict, CfUtil.getCFString("locationID"), locationId);
CoreFoundation.CFMutableDictionaryRef matchingDict = CoreFoundation.INSTANCE.CFDictionaryCreateMutable(CfUtil.ALLOCATOR, 0,
null, null);
CoreFoundation.INSTANCE.CFDictionarySetValue(matchingDict, CfUtil.getCFString("IOPropertyMatch"), propertyDict);
// search for all IOservices that match the locationID
IntByReference serviceIterator = new IntByReference();
IOKitUtil.getMatchingServices(matchingDict, serviceIterator);
// getMatchingServices releases matchingDict
CfUtil.release(propertyDict);
// Iterate matching services looking for devices whose parents have the
// vendor and device ids
boolean found = false;
int matchingService = IOKit.INSTANCE.IOIteratorNext(serviceIterator.getValue());
while (matchingService != 0 && !found) {
// Get the parent, which contains the keys we need
IntByReference parent = new IntByReference();
IOKit.INSTANCE.IORegistryEntryGetParentEntry(matchingService, "IOService", parent);
// look up the vendor-id by key
// vendor-id is a byte array of 4 bytes
byte[] vid = IOKitUtil.getIORegistryByteArrayProperty(parent.getValue(), "vendor-id");
if (vid != null && vid.length >= 2) {
vendorIdMap.put(id, String.format("%02x%02x", vid[1], vid[0]));
found = true;
}
// look up the device-id by key
// device-id is a byte array of 4 bytes
byte[] pid = IOKitUtil.getIORegistryByteArrayProperty(parent.getValue(), "device-id");
if (pid != null && pid.length >= 2) {
productIdMap.put(id, String.format("%02x%02x", pid[1], pid[0]));
found = true;
}
// iterate
matchingService = IOKit.INSTANCE.IOIteratorNext(matchingService);
}
}
/**
* Recursively creates MacUsbDevices by fetching information from maps to
* populate fields
*
* @param registryEntryId
* The device unique registry id.
* @param vid
* The default (parent) vendor ID
* @param pid
* The default (parent) product ID
* @return A MacUsbDevice corresponding to this device
*/
private static MacUsbDevice getDeviceAndChildren(Long registryEntryId, String vid, String pid) {
String vendorId = getOrDefault_1(vendorIdMap,registryEntryId, vid);
String productId = getOrDefault_1(productIdMap,registryEntryId, pid);
List childIds = getOrDefault_2(hubMap,registryEntryId, new ArrayList());
List usbDevices = new ArrayList();
for (Long id : childIds) {
usbDevices.add(getDeviceAndChildren(id, vendorId, productId));
}
Collections.sort(usbDevices);
return new MacUsbDevice(getOrDefault_1(nameMap,registryEntryId, vendorId + ":" + productId),
getOrDefault_1(vendorMap,registryEntryId, ""), vendorId, productId,
getOrDefault_1(serialMap,registryEntryId, ""), usbDevices.toArray(new UsbDevice[usbDevices.size()]));
}
private static List getOrDefault_2(Map> map, Long name, ArrayList defaultValue){
if (map.containsKey(name)){
return defaultValue;
}
List result = map.get(name);
if (result == null){
return defaultValue;
} else {
return result;
}
}
private static String getOrDefault_1(Map map, Long name, String defaultValue){
if (map.containsKey(name)){
return defaultValue;
}
String result = map.get(name);
if (result == null){
return defaultValue;
} else {
return result;
}
}
private static List computeIfAbsent(Map> map, Long key){
List result = map.get(key);
if (result == null){
result = new ArrayList();
map.put(key, result);
}
return result;
}
}
