mirror of
https://github.com/morgan9e/macos-stats
synced 2026-04-14 00:04:15 +09:00
288 lines
11 KiB
Swift
288 lines
11 KiB
Swift
//
|
|
// readers.swift
|
|
// Disk
|
|
//
|
|
// Created by Serhiy Mytrovtsiy on 07/05/2020.
|
|
// Using Swift 5.0.
|
|
// Running on macOS 10.15.
|
|
//
|
|
// Copyright © 2020 Serhiy Mytrovtsiy. All rights reserved.
|
|
//
|
|
|
|
import Cocoa
|
|
import Kit
|
|
import IOKit
|
|
import Darwin
|
|
|
|
internal class CapacityReader: Reader<Disks> {
|
|
internal var list: Disks = Disks()
|
|
|
|
public override func read() {
|
|
let keys: [URLResourceKey] = [.volumeNameKey]
|
|
let removableState = Store.shared.bool(key: "Disk_removable", defaultValue: false)
|
|
let paths = FileManager.default.mountedVolumeURLs(includingResourceValuesForKeys: keys)!
|
|
|
|
guard let session = DASessionCreate(kCFAllocatorDefault) else {
|
|
error("cannot create main DASessionCreate()", log: self.log)
|
|
return
|
|
}
|
|
|
|
var active: [String] = []
|
|
for url in paths {
|
|
if url.pathComponents.count == 1 || (url.pathComponents.count > 1 && url.pathComponents[1] == "Volumes") {
|
|
if let disk = DADiskCreateFromVolumePath(kCFAllocatorDefault, session, url as CFURL) {
|
|
if let diskName = DADiskGetBSDName(disk) {
|
|
let BSDName: String = String(cString: diskName)
|
|
active.append(BSDName)
|
|
|
|
if let d = self.list.first(where: { $0.BSDName == BSDName}), let idx = self.list.index(where: { $0.BSDName == BSDName}) {
|
|
if d.removable && !removableState {
|
|
self.list.remove(at: idx)
|
|
continue
|
|
}
|
|
|
|
if let path = d.path {
|
|
self.list.updateFreeSize(idx, newValue: self.freeDiskSpaceInBytes(path))
|
|
}
|
|
|
|
continue
|
|
}
|
|
|
|
if var d = driveDetails(disk, removableState: removableState) {
|
|
if let path = d.path {
|
|
d.free = self.freeDiskSpaceInBytes(path)
|
|
d.size = self.totalDiskSpaceInBytes(path)
|
|
}
|
|
self.list.append(d)
|
|
self.list.sort()
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
active.difference(from: self.list.map{ $0.BSDName }).forEach { (BSDName: String) in
|
|
if let idx = self.list.index(where: { $0.BSDName == BSDName }) {
|
|
self.list.remove(at: idx)
|
|
}
|
|
}
|
|
|
|
self.callback(self.list)
|
|
}
|
|
|
|
private func freeDiskSpaceInBytes(_ path: URL) -> Int64 {
|
|
do {
|
|
let systemAttributes = try FileManager.default.attributesOfFileSystem(forPath: path.path)
|
|
if let freeSpace = (systemAttributes[FileAttributeKey.systemFreeSize] as? NSNumber)?.int64Value {
|
|
return freeSpace
|
|
}
|
|
} catch let err {
|
|
error("error retrieving free space #2: \(err.localizedDescription)", log: self.log)
|
|
}
|
|
|
|
do {
|
|
if let url = URL(string: path.absoluteString) {
|
|
let values = try url.resourceValues(forKeys: [.volumeAvailableCapacityForImportantUsageKey])
|
|
if let capacity = values.volumeAvailableCapacityForImportantUsage, capacity != 0 {
|
|
return capacity
|
|
}
|
|
}
|
|
} catch let err {
|
|
error("error retrieving free space #1: \(err.localizedDescription)", log: self.log)
|
|
}
|
|
|
|
return 0
|
|
}
|
|
|
|
private func totalDiskSpaceInBytes(_ path: URL) -> Int64 {
|
|
do {
|
|
let systemAttributes = try FileManager.default.attributesOfFileSystem(forPath: path.path)
|
|
if let totalSpace = (systemAttributes[FileAttributeKey.systemSize] as? NSNumber)?.int64Value {
|
|
return totalSpace
|
|
}
|
|
} catch let err {
|
|
error("error retrieving total space #2: \(err.localizedDescription)", log: self.log)
|
|
}
|
|
|
|
do {
|
|
if let url = URL(string: path.absoluteString) {
|
|
let values = try url.resourceValues(forKeys: [.volumeTotalCapacityKey])
|
|
if let space = values.volumeAvailableCapacityForImportantUsage, space != 0 {
|
|
return space
|
|
}
|
|
}
|
|
} catch let err {
|
|
error("error retrieving total space #1: \(err.localizedDescription)", log: self.log)
|
|
}
|
|
|
|
return 0
|
|
}
|
|
}
|
|
|
|
internal class ActivityReader: Reader<Disks> {
|
|
internal var list: Disks = Disks()
|
|
|
|
init() {
|
|
super.init()
|
|
}
|
|
|
|
override func setup() {
|
|
setInterval(1)
|
|
}
|
|
|
|
public override func read() {
|
|
let keys: [URLResourceKey] = [.volumeNameKey]
|
|
let removableState = Store.shared.bool(key: "Disk_removable", defaultValue: false)
|
|
let paths = FileManager.default.mountedVolumeURLs(includingResourceValuesForKeys: keys)!
|
|
|
|
guard let session = DASessionCreate(kCFAllocatorDefault) else {
|
|
error("cannot create a DASessionCreate()", log: self.log)
|
|
return
|
|
}
|
|
|
|
var active: [String] = []
|
|
for url in paths {
|
|
if url.pathComponents.count == 1 || (url.pathComponents.count > 1 && url.pathComponents[1] == "Volumes") {
|
|
if let disk = DADiskCreateFromVolumePath(kCFAllocatorDefault, session, url as CFURL) {
|
|
if let diskName = DADiskGetBSDName(disk) {
|
|
let BSDName: String = String(cString: diskName)
|
|
active.append(BSDName)
|
|
|
|
if let d = self.list.first(where: { $0.BSDName == BSDName}), let idx = self.list.index(where: { $0.BSDName == BSDName}) {
|
|
if d.removable && !removableState {
|
|
self.list.remove(at: idx)
|
|
continue
|
|
}
|
|
|
|
self.driveStats(idx, d)
|
|
continue
|
|
}
|
|
|
|
if let d = driveDetails(disk, removableState: removableState) {
|
|
self.list.append(d)
|
|
self.list.sort()
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
active.difference(from: self.list.map{ $0.BSDName }).forEach { (BSDName: String) in
|
|
if let idx = self.list.index(where: { $0.BSDName == BSDName }) {
|
|
self.list.remove(at: idx)
|
|
}
|
|
}
|
|
|
|
self.callback(self.list)
|
|
}
|
|
|
|
private func driveStats(_ idx: Int, _ d: drive) {
|
|
guard let props = getIOProperties(d.parent) else {
|
|
return
|
|
}
|
|
|
|
if let statistics = props.object(forKey: "Statistics") as? NSDictionary {
|
|
let readBytes = statistics.object(forKey: "Bytes (Read)") as? Int64 ?? 0
|
|
let writeBytes = statistics.object(forKey: "Bytes (Write)") as? Int64 ?? 0
|
|
|
|
if d.activity.readBytes != 0 {
|
|
self.list.updateRead(idx, newValue: readBytes - d.activity.readBytes)
|
|
}
|
|
if d.activity.writeBytes != 0 {
|
|
self.list.updateWrite(idx, newValue: writeBytes - d.activity.writeBytes)
|
|
}
|
|
|
|
self.list.updateReadWrite(idx, read: readBytes, write: writeBytes)
|
|
}
|
|
|
|
return
|
|
}
|
|
}
|
|
|
|
private func driveDetails(_ disk: DADisk, removableState: Bool) -> drive? {
|
|
var d: drive = drive()
|
|
|
|
if let bsdName = DADiskGetBSDName(disk) {
|
|
d.BSDName = String(cString: bsdName)
|
|
}
|
|
|
|
if let diskDescription = DADiskCopyDescription(disk) {
|
|
if let dict = diskDescription as? [String: AnyObject] {
|
|
if let removable = dict[kDADiskDescriptionMediaRemovableKey as String] {
|
|
if removable as! Bool {
|
|
if !removableState {
|
|
return nil
|
|
}
|
|
d.removable = true
|
|
}
|
|
}
|
|
|
|
if let mediaName = dict[kDADiskDescriptionVolumeNameKey as String] {
|
|
d.mediaName = mediaName as! String
|
|
if d.mediaName == "Recovery" {
|
|
return nil
|
|
}
|
|
}
|
|
if d.mediaName == "" {
|
|
if let mediaName = dict[kDADiskDescriptionMediaNameKey as String] {
|
|
d.mediaName = mediaName as! String
|
|
if d.mediaName == "Recovery" {
|
|
return nil
|
|
}
|
|
}
|
|
}
|
|
if let deviceModel = dict[kDADiskDescriptionDeviceModelKey as String] {
|
|
d.model = (deviceModel as! String).trimmingCharacters(in: .whitespacesAndNewlines)
|
|
}
|
|
if let deviceProtocol = dict[kDADiskDescriptionDeviceProtocolKey as String] {
|
|
d.connectionType = deviceProtocol as! String
|
|
}
|
|
if let volumePath = dict[kDADiskDescriptionVolumePathKey as String] {
|
|
if let url = volumePath as? NSURL {
|
|
d.path = url as URL
|
|
|
|
if let components = url.pathComponents {
|
|
d.root = components.count == 1
|
|
|
|
if components.count > 1 && components[1] == "Volumes" {
|
|
if let name: String = url.lastPathComponent, name != "" {
|
|
d.mediaName = name
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
if let volumeKind = dict[kDADiskDescriptionVolumeKindKey as String] {
|
|
d.fileSystem = volumeKind as! String
|
|
}
|
|
}
|
|
}
|
|
|
|
if d.path == nil {
|
|
return nil
|
|
}
|
|
|
|
let partitionLevel = d.BSDName.filter { "0"..."9" ~= $0 }.count
|
|
if let parent = getDeviceIOParent(DADiskCopyIOMedia(disk), level: Int(partitionLevel)) {
|
|
d.parent = parent
|
|
}
|
|
|
|
return d
|
|
}
|
|
|
|
// https://opensource.apple.com/source/bless/bless-152/libbless/APFS/BLAPFSUtilities.c.auto.html
|
|
public func getDeviceIOParent(_ obj: io_registry_entry_t, level: Int) -> io_registry_entry_t? {
|
|
var parent: io_registry_entry_t = 0
|
|
|
|
if IORegistryEntryGetParentEntry(obj, kIOServicePlane, &parent) != KERN_SUCCESS {
|
|
return nil
|
|
}
|
|
|
|
for _ in 1...level where IORegistryEntryGetParentEntry(parent, kIOServicePlane, &parent) != KERN_SUCCESS {
|
|
IOObjectRelease(parent)
|
|
return nil
|
|
}
|
|
|
|
return parent
|
|
}
|