127 lines
3.9 KiB
Swift
127 lines
3.9 KiB
Swift
//
|
|
// HelperTool.swift
|
|
// com.davidwernhart.Helper
|
|
//
|
|
// Created by David Wernhart on 13.02.20.
|
|
// Copyright © 2020 David Wernhart. All rights reserved.
|
|
//
|
|
|
|
import Foundation
|
|
import IOKit.pwr_mgt
|
|
|
|
final class HelperTool: NSObject, HelperToolProtocol {
|
|
|
|
static let instance = HelperTool()
|
|
|
|
var modifiedKeys: [String: UInt8] = [:]
|
|
var openAssertions: [IOPMAssertionID] = []
|
|
|
|
func getVersion(withReply reply: (String) -> Void) {
|
|
// let version = Bundle.main.object(forInfoDictionaryKey: "CFBundleShortVersionString" as String) as? String ?? "(unknown version)"
|
|
// let build = Bundle.main.object(forInfoDictionaryKey: kCFBundleVersionKey as String) as? String ?? "(unknown build)"
|
|
// reply("v\(version) (\(build))")
|
|
reply(helperVersion)
|
|
|
|
}
|
|
|
|
func setSMCByte(key: String, value: UInt8) {
|
|
do {
|
|
try SMCKit.open()
|
|
} catch {
|
|
print(error)
|
|
exit(EX_UNAVAILABLE)
|
|
}
|
|
let smcKey = SMCKit.getKey(key, type: DataTypes.UInt8)
|
|
let bytes: SMCBytes = (value, UInt8(0), UInt8(0), UInt8(0), UInt8(0), UInt8(0),
|
|
UInt8(0), UInt8(0), UInt8(0), UInt8(0), UInt8(0), UInt8(0),
|
|
UInt8(0), UInt8(0), UInt8(0), UInt8(0), UInt8(0), UInt8(0),
|
|
UInt8(0), UInt8(0), UInt8(0), UInt8(0), UInt8(0), UInt8(0),
|
|
UInt8(0), UInt8(0), UInt8(0), UInt8(0), UInt8(0), UInt8(0),
|
|
UInt8(0), UInt8(0))
|
|
|
|
if(self.modifiedKeys[key] == nil){
|
|
readSMCByte(key: key) { (originalValue) in
|
|
self.modifiedKeys[key] = originalValue
|
|
_ = try? SMCKit.writeData(smcKey, data: bytes)
|
|
}
|
|
}
|
|
else{
|
|
_ = try? SMCKit.writeData(smcKey, data: bytes)
|
|
}
|
|
|
|
|
|
}
|
|
|
|
func readSMCByte(key: String, withReply reply: @escaping (UInt8) -> Void) {
|
|
do {
|
|
try SMCKit.open()
|
|
} catch {
|
|
print(error)
|
|
exit(EX_UNAVAILABLE)
|
|
}
|
|
|
|
let smcKey = SMCKit.getKey(key, type: DataTypes.UInt8)
|
|
do {
|
|
let status = try SMCKit.readData(smcKey).0
|
|
reply(status)
|
|
} catch {
|
|
reply(0)
|
|
}
|
|
}
|
|
|
|
func readSMCUInt32(key: String, withReply reply: @escaping (UInt32) -> Void) {
|
|
do {
|
|
try SMCKit.open()
|
|
} catch {
|
|
print(error)
|
|
exit(EX_UNAVAILABLE)
|
|
}
|
|
|
|
let smcKey = SMCKit.getKey(key, type: DataTypes.UInt32)
|
|
do {
|
|
let data = try SMCKit.readData(smcKey)
|
|
reply(UInt32(fromBytes: (data.0, data.1, data.2, data.3)))
|
|
} catch {
|
|
reply(0)
|
|
}
|
|
}
|
|
|
|
func createAssertion(assertion:String, withReply reply: @escaping (IOPMAssertionID) -> Void){
|
|
var assertionID : IOPMAssertionID = IOPMAssertionID(0)
|
|
let reason:CFString = "AlDente" as NSString
|
|
let cfAssertion:CFString = assertion as NSString
|
|
let success = IOPMAssertionCreateWithName(cfAssertion,
|
|
IOPMAssertionLevel(kIOPMAssertionLevelOn),
|
|
reason,
|
|
&assertionID)
|
|
if success == kIOReturnSuccess {
|
|
openAssertions.append(assertionID)
|
|
reply(assertionID)
|
|
}
|
|
else{
|
|
reply (UInt32(kCFNumberNaN))
|
|
}
|
|
}
|
|
|
|
func releaseAssertion(assertionID:IOPMAssertionID){
|
|
IOPMAssertionRelease(assertionID)
|
|
openAssertions.remove(at: openAssertions.firstIndex(of: assertionID)!)
|
|
}
|
|
|
|
func setResetVal(key:String, value: UInt8){
|
|
modifiedKeys[key]=value
|
|
}
|
|
|
|
func reset(){
|
|
for (key, value) in modifiedKeys{
|
|
setSMCByte(key: key, value: value)
|
|
}
|
|
|
|
for assertionID in openAssertions{
|
|
releaseAssertion(assertionID: assertionID)
|
|
}
|
|
modifiedKeys.removeAll()
|
|
openAssertions.removeAll()
|
|
}
|
|
}
|