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.Disks;
import oshi.hardware.HWDiskStore;
import oshi.hardware.HWPartition;
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.ExecutingCommand;
import oshi.util.ParseUtil;
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 implements Disks {
private static final long serialVersionUID = 1L;
private static final Logger LOG = LoggerFactory.getLogger(MacDisks.class);
private static final Map mountPointMap = new HashMap<>();
private static final Map logicalVolumeMap = new HashMap<>();
@Override
public HWDiskStore[] getDisks() {
mountPointMap.clear();
logicalVolumeMap.clear();
List result = new ArrayList<>();
// Use statfs to populate mount point map
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);
// Iterate all mounted file systems
for (Statfs f : fs) {
String mntFrom = new String(f.f_mntfromname).trim();
mountPointMap.put(mntFrom.replace("/dev/", ""), new String(f.f_mntonname).trim());
}
// Parse `diskutil cs list` to populate logical volume map
Set physicalVolumes = new HashSet<>();
boolean logicalVolume = false;
for (String line : ExecutingCommand.runNative("diskutil cs list")) {
if (line.contains("Logical Volume Group")) {
// Logical Volume Group defines beginning of grouping which will
// list multiple physical volumes followed by the logical volume
// they are associated with. Each physical volume will be a key
// with the logical volume as its value, but since the value
// doesn't appear until the end we collect the keys in a list
physicalVolumes.clear();
logicalVolume = false;
continue;
} else if (line.contains("Logical Volume Family")) {
// Done collecting physical volumes, prepare to store logical
// volume
logicalVolume = true;
continue;
} else if (line.contains("Disk:")) {
String volume = ParseUtil.parseLastString(line);
if (logicalVolume) {
// Store this disk as the logical volume value for all the
// physical volume keys
for (String pv : physicalVolumes) {
logicalVolumeMap.put(pv, volume);
}
physicalVolumes.clear();
} else {
physicalVolumes.add(ParseUtil.parseLastString(line));
}
}
}
// 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];
}
// Get IOMedia objects representing whole drives
List bsdNames = new ArrayList<>();
IntByReference iter = new IntByReference();
IOKitUtil.getMatchingServices("IOMedia", iter);
int media = IOKit.INSTANCE.IOIteratorNext(iter.getValue());
while (media != 0) {
if (IOKitUtil.getIORegistryBooleanProperty(media, "Whole")) {
DADiskRef disk = DiskArbitration.INSTANCE.DADiskCreateFromIOMedia(CfUtil.ALLOCATOR, session, media);
bsdNames.add(DiskArbitration.INSTANCE.DADiskGetBSDName(disk));
}
IOKit.INSTANCE.IOObjectRelease(media);
media = IOKit.INSTANCE.IOIteratorNext(iter.getValue());
}
// Now iterate the bsdNames
for (String bsdName : bsdNames) {
String model = "";
String serial = "";
long size = 0L;
long xferTime;
// 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 (!"Disk Image".equals(model)) {
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], 0L);
// 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"));
diskStore.setTimeStamp(System.currentTimeMillis());
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 {
// This is normal for FileVault drives, Fusion
// drives, and other virtual bsd names
LOG.debug("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);
// We need a CFBoolean that's false.
// Whole disk has 'true' for Whole and 'false'
// for leaf; store the boolean false
p = CoreFoundation.INSTANCE.CFDictionaryGetValue(properties,
CfUtil.getCFString("Leaf"));
CFBooleanRef cfFalse = new CFBooleanRef();
cfFalse.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("Whole"),
cfFalse);
matchingDict = CoreFoundation.INSTANCE.CFDictionaryCreateMutable(CfUtil.ALLOCATOR, 0,
null, null);
CoreFoundation.INSTANCE.CFDictionarySetValue(matchingDict,
CfUtil.getCFString("IOPropertyMatch"), propertyDict);
// search for IOservices that match the BSD Unit
// with whole=false; 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"));
if (volumePtr == null) {
name = type;
} else {
name = CfUtil.cfPointerToString(volumePtr);
}
CfUtil.release(diskInfo);
}
CfUtil.release(disk);
}
String mountPoint;
if (logicalVolumeMap.containsKey(partBsdName)) {
mountPoint = "Logical Volume: " + logicalVolumeMap.get(partBsdName);
} else {
mountPoint = mountPointMap.getOrDefault(partBsdName, "");
}
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"), mountPoint));
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()]);
}
}
