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.
/**
* Oshi (https://github.com/dblock/oshi)
*
* Copyright (c) 2010 - 2016 The Oshi Project Team
*
* All rights reserved. This program and the accompanying materials
* are made available under the terms of the Eclipse Public License v1.0
* which accompanies this distribution, and is available at
* http://www.eclipse.org/legal/epl-v10.html
*
* Maintainers:
* dblock[at]dblock[dot]org
* widdis[at]gmail[dot]com
* enrico.bianchi[at]gmail[dot]com
*
* Contributors:
* https://github.com/dblock/oshi/graphs/contributors
*/
package oshi.hardware.platform.mac;
import java.util.ArrayList;
import java.util.Collections;
import java.util.HashMap;
import java.util.HashSet;
import java.util.List;
import java.util.Map;
import java.util.Set;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import com.sun.jna.Pointer;
import com.sun.jna.ptr.IntByReference;
import com.sun.jna.ptr.PointerByReference;
import oshi.hardware.HWDiskStore;
import oshi.hardware.HWPartition;
import oshi.hardware.common.AbstractDisks;
import oshi.jna.platform.mac.CoreFoundation;
import oshi.jna.platform.mac.CoreFoundation.CFBooleanRef;
import oshi.jna.platform.mac.CoreFoundation.CFDictionaryRef;
import oshi.jna.platform.mac.CoreFoundation.CFMutableDictionaryRef;
import oshi.jna.platform.mac.CoreFoundation.CFNumberRef;
import oshi.jna.platform.mac.CoreFoundation.CFStringRef;
import oshi.jna.platform.mac.DiskArbitration;
import oshi.jna.platform.mac.DiskArbitration.DADiskRef;
import oshi.jna.platform.mac.DiskArbitration.DASessionRef;
import oshi.jna.platform.mac.IOKit;
import oshi.jna.platform.mac.SystemB;
import oshi.jna.platform.mac.SystemB.Statfs;
import oshi.util.platform.mac.CfUtil;
import oshi.util.platform.mac.IOKitUtil;
/**
* Mac hard disk implementation.
*
* @author enrico[dot]bianchi[at]gmail[dot]com
*/
public class MacDisks extends AbstractDisks {
private static final long serialVersionUID = 1L;
private static final Logger LOG = LoggerFactory.getLogger(MacDisks.class);
private static final Map mountPointMap = new HashMap<>();
@Override
public HWDiskStore[] getDisks() {
mountPointMap.clear();
List result = new ArrayList<>();
// Use statfs to find all drives
int numfs = SystemB.INSTANCE.getfsstat64(null, 0, 0);
// Create array to hold results
Statfs[] fs = new Statfs[numfs];
// Fill array with results
SystemB.INSTANCE.getfsstat64(fs, numfs * new Statfs().size(), SystemB.MNT_NOWAIT);
// Open a DiskArbitration session
DASessionRef session = DiskArbitration.INSTANCE.DASessionCreate(CfUtil.ALLOCATOR);
if (session == null) {
LOG.error("Unable to open session to DiskArbitration framework.");
return new HWDiskStore[0];
}
// Create a set in case we have multiple partitions mounted
Set bsdNames = new HashSet();
// Iterate all mounted file systems
for (Statfs f : fs) {
String mntFrom = new String(f.f_mntfromname).trim();
// OS X registry uses the BSD Name, e.g. disk0, disk1, etc.
// Strip off the partition # to get base disk name
String[] split = mntFrom.split("/dev/|s\\d+");
if (split.length > 1) {
bsdNames.add(split[1]);
}
mountPointMap.put(mntFrom.replace("/dev/", ""), new String(f.f_mntonname).trim());
}
// Now iterate the bsdNames
for (String bsdName : bsdNames) {
String model = "";
String serial = "";
long size = 0L;
long xferTime = 0L;
// Get a reference to the disk - only matching /dev/disk*
String path = "/dev/" + bsdName;
// Get the DiskArbitration dictionary for this disk, which has model
// and size (capacity)
DADiskRef disk = DiskArbitration.INSTANCE.DADiskCreateFromBSDName(CfUtil.ALLOCATOR, session, path);
if (disk != null) {
CFDictionaryRef diskInfo = DiskArbitration.INSTANCE.DADiskCopyDescription(disk);
if (diskInfo != null) {
// Parse out model and size from their respective keys
Pointer modelPtr = CoreFoundation.INSTANCE.CFDictionaryGetValue(diskInfo,
CfUtil.getCFString("DADeviceModel"));
model = CfUtil.cfPointerToString(modelPtr);
Pointer sizePtr = CoreFoundation.INSTANCE.CFDictionaryGetValue(diskInfo,
CfUtil.getCFString("DAMediaSize"));
size = CfUtil.cfPointerToLong(sizePtr);
CfUtil.release(diskInfo);
// Use the model as a key to get serial from IOKit
if (!model.equals("Disk Image")) {
CFStringRef modelNameRef = CFStringRef.toCFString(model);
CFMutableDictionaryRef propertyDict = CoreFoundation.INSTANCE
.CFDictionaryCreateMutable(CfUtil.ALLOCATOR, 0, null, null);
CoreFoundation.INSTANCE.CFDictionarySetValue(propertyDict, CfUtil.getCFString("Model"),
modelNameRef);
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 model
IntByReference serviceIterator = new IntByReference();
IOKitUtil.getMatchingServices(matchingDict, serviceIterator);
// getMatchingServices releases matchingDict
CfUtil.release(modelNameRef);
CfUtil.release(propertyDict);
int sdService = IOKit.INSTANCE.IOIteratorNext(serviceIterator.getValue());
while (sdService != 0) {
// look up the serial number
serial = IOKitUtil.getIORegistryStringProperty(sdService, "Serial Number");
IOKit.INSTANCE.IOObjectRelease(sdService);
if (serial != null) {
break;
}
// iterate
sdService = IOKit.INSTANCE.IOIteratorNext(serviceIterator.getValue());
}
if (serial == null) {
serial = "";
}
IOKit.INSTANCE.IOObjectRelease(serviceIterator.getValue());
}
}
CfUtil.release(disk);
// If empty, ignore
if (size <= 0) {
continue;
}
HWDiskStore diskStore = new HWDiskStore(bsdName, model.trim(), serial.trim(), size, 0L, 0L, 0L, 0L, 0L,
new HWPartition[0]);
// Now look up the device using the BSD Name to get its
// statistics
CFMutableDictionaryRef matchingDict = IOKitUtil.getBSDNameMatchingDict(bsdName);
if (matchingDict != null) {
// search for all IOservices that match the bsd name
IntByReference driveList = new IntByReference();
IOKitUtil.getMatchingServices(matchingDict, driveList);
// getMatchingServices releases matchingDict
int drive = IOKit.INSTANCE.IOIteratorNext(driveList.getValue());
// Should only match one drive
if (drive != 0) {
// Should be an IOMedia object with a parent
// IOBlockStorageDriver object
// Get the properties from the parent
IntByReference parent = new IntByReference();
if (IOKit.INSTANCE.IOObjectConformsTo(drive, "IOMedia")
&& IOKit.INSTANCE.IORegistryEntryGetParentEntry(drive, "IOService", parent) == 0) {
PointerByReference propsPtr = new PointerByReference();
if (IOKit.INSTANCE.IOObjectConformsTo(parent.getValue(), "IOBlockStorageDriver")
&& IOKit.INSTANCE.IORegistryEntryCreateCFProperties(parent.getValue(), propsPtr,
CfUtil.ALLOCATOR, 0) == 0) {
CFMutableDictionaryRef properties = new CFMutableDictionaryRef();
properties.setPointer(propsPtr.getValue());
// We now have a properties object with the
// statistics we need on it. Fetch them
Pointer statsPtr = CoreFoundation.INSTANCE.CFDictionaryGetValue(properties,
CfUtil.getCFString("Statistics"));
CFDictionaryRef statistics = new CFDictionaryRef();
statistics.setPointer(statsPtr);
// Now get the stats we want
Pointer stat = CoreFoundation.INSTANCE.CFDictionaryGetValue(statistics,
CfUtil.getCFString("Operations (Read)"));
diskStore.setReads(CfUtil.cfPointerToLong(stat));
stat = CoreFoundation.INSTANCE.CFDictionaryGetValue(statistics,
CfUtil.getCFString("Bytes (Read)"));
diskStore.setReadBytes(CfUtil.cfPointerToLong(stat));
stat = CoreFoundation.INSTANCE.CFDictionaryGetValue(statistics,
CfUtil.getCFString("Operations (Write)"));
diskStore.setWrites(CfUtil.cfPointerToLong(stat));
stat = CoreFoundation.INSTANCE.CFDictionaryGetValue(statistics,
CfUtil.getCFString("Bytes (Write)"));
diskStore.setWriteBytes(CfUtil.cfPointerToLong(stat));
// Total time is in nanoseconds. Add read+write
// and convert total to ms
stat = CoreFoundation.INSTANCE.CFDictionaryGetValue(statistics,
CfUtil.getCFString("Total Time (Read)"));
xferTime = CfUtil.cfPointerToLong(stat);
stat = CoreFoundation.INSTANCE.CFDictionaryGetValue(statistics,
CfUtil.getCFString("Total Time (Write)"));
xferTime += CfUtil.cfPointerToLong(stat);
diskStore.setTransferTime(xferTime / 10000L);
CfUtil.release(properties);
} else {
LOG.error("Unable to find block storage driver properties for {}", bsdName);
}
// Now get partitions for this disk.
List partitions = new ArrayList<>();
if (IOKit.INSTANCE.IORegistryEntryCreateCFProperties(drive, propsPtr, CfUtil.ALLOCATOR,
0) == 0) {
CFMutableDictionaryRef properties = new CFMutableDictionaryRef();
properties.setPointer(propsPtr.getValue());
// Partitions will match BSD Unit property
Pointer p = CoreFoundation.INSTANCE.CFDictionaryGetValue(properties,
CfUtil.getCFString("BSD Unit"));
CFNumberRef bsdUnit = new CFNumberRef();
bsdUnit.setPointer(p);
// Whole disk has 'true' for Whole and 'false'
// for leaf; store the boolean true
p = CoreFoundation.INSTANCE.CFDictionaryGetValue(properties,
CfUtil.getCFString("Whole"));
CFBooleanRef leaf = new CFBooleanRef();
leaf.setPointer(p);
// create a matching dict for BSD Unit
CFMutableDictionaryRef propertyDict = CoreFoundation.INSTANCE
.CFDictionaryCreateMutable(CfUtil.ALLOCATOR, 0, null, null);
CoreFoundation.INSTANCE.CFDictionarySetValue(propertyDict,
CfUtil.getCFString("BSD Unit"), bsdUnit);
CoreFoundation.INSTANCE.CFDictionarySetValue(propertyDict, CfUtil.getCFString("Leaf"),
leaf);
matchingDict = CoreFoundation.INSTANCE.CFDictionaryCreateMutable(CfUtil.ALLOCATOR, 0,
null, null);
CoreFoundation.INSTANCE.CFDictionarySetValue(matchingDict,
CfUtil.getCFString("IOPropertyMatch"), propertyDict);
// search for all IOservices that match the
// BSD Unit with leaf=true; these are partitions
IntByReference serviceIterator = new IntByReference();
IOKitUtil.getMatchingServices(matchingDict, serviceIterator);
// getMatchingServices releases matchingDict
CfUtil.release(properties);
CfUtil.release(propertyDict);
int sdService = IOKit.INSTANCE.IOIteratorNext(serviceIterator.getValue());
while (sdService != 0) {
// look up the BSD Name
String partBsdName = IOKitUtil.getIORegistryStringProperty(sdService, "BSD Name");
String name = partBsdName;
String type = "";
// Get the DiskArbitration dictionary for
// this partition
disk = DiskArbitration.INSTANCE.DADiskCreateFromBSDName(CfUtil.ALLOCATOR, session,
partBsdName);
if (disk != null) {
diskInfo = DiskArbitration.INSTANCE.DADiskCopyDescription(disk);
if (diskInfo != null) {
// get volume name from its key
Pointer volumePtr = CoreFoundation.INSTANCE.CFDictionaryGetValue(diskInfo,
CfUtil.getCFString("DAMediaName"));
type = CfUtil.cfPointerToString(volumePtr);
volumePtr = CoreFoundation.INSTANCE.CFDictionaryGetValue(diskInfo,
CfUtil.getCFString("DAVolumeName"));
name = CfUtil.cfPointerToString(volumePtr);
CfUtil.release(diskInfo);
}
CfUtil.release(disk);
}
partitions.add(new HWPartition(partBsdName, name, type,
IOKitUtil.getIORegistryStringProperty(sdService, "UUID"),
IOKitUtil.getIORegistryLongProperty(sdService, "Size"),
IOKitUtil.getIORegistryIntProperty(sdService, "BSD Major"),
IOKitUtil.getIORegistryIntProperty(sdService, "BSD Minor"),
mountPointMap.getOrDefault(partBsdName, "")));
IOKit.INSTANCE.IOObjectRelease(sdService);
// iterate
sdService = IOKit.INSTANCE.IOIteratorNext(serviceIterator.getValue());
}
IOKit.INSTANCE.IOObjectRelease(serviceIterator.getValue());
} else {
LOG.error("Unable to find properties for {}", bsdName);
}
Collections.sort(partitions);
diskStore.setPartitions(partitions.toArray(new HWPartition[partitions.size()]));
IOKit.INSTANCE.IOObjectRelease(parent.getValue());
} else {
LOG.error("Unable to find IOMedia device or parent for ", bsdName);
}
IOKit.INSTANCE.IOObjectRelease(drive);
}
IOKit.INSTANCE.IOObjectRelease(driveList.getValue());
}
result.add(diskStore);
}
}
// Close DA session
CfUtil.release(session);
Collections.sort(result);
return result.toArray(new HWDiskStore[result.size()]);
}
}
