CloverBootloader/CloverApp/Clover/NVRAM.swift

//
//  NVRAM.swift
//  Clover
//
//  Created by vector sigma on 30/10/2019.
//  Copyright © 2019 CloverHackyColor. All rights reserved.
//

import Cocoa

// MARK: "/usr/sbin/nvram"
let nvram_cmd = "/usr/sbin/nvram"

// MARK: Get NVRAM
func getNVRAM() -> NSMutableDictionary? {
  var ref: io_registry_entry_t
  var masterPort = mach_port_t()
  var oResult: kern_return_t
  var result: kern_return_t
  oResult = IOMasterPort(bootstrap_port, &masterPort)
  
  if oResult != KERN_SUCCESS {
    return nil
  }
  
  ref = IORegistryEntryFromPath(masterPort, "IODeviceTree:/options")
  if ref == 0 {
    return nil
  }
  var dict : Unmanaged<CFMutableDictionary>?
  result = IORegistryEntryCreateCFProperties(ref, &dict, kCFAllocatorDefault, 0)
  if result != KERN_SUCCESS {
    IOObjectRelease(ref)
    return nil
  }
  IOObjectRelease(ref)
  return dict?.takeRetainedValue()
}

// MARK: set NVRAM key
@available(OSX 10.10, *)
func setNVRAM(key: String, stringValue: String) {
  var cmd : String = "do shell script \""
  cmd += "sudo \(nvram_cmd) \(key)=\(stringValue)" // sudo required otherwise wont work!
  cmd += "\" with administrator privileges"
  DispatchQueue.global(qos: .background).async {
    let script: NSAppleScript? = NSAppleScript(source: cmd)
    var error : NSDictionary? = nil
    script?.executeAndReturnError(&error)
    if error != nil {
      NSSound.beep()
      print(error!.description)
    }
  }
}

// MARK: delete NVRAM key
@available(OSX 10.10, *)
func deleteNVRAM(key: String) {
  var cmd : String = "do shell script \""
  cmd += "sudo \(nvram_cmd) -d \(key)" // sudo required otherwise wont work!
  cmd += "\" with administrator privileges"
  DispatchQueue.global(qos: .background).async {
    var error : NSDictionary? = nil
    let script: NSAppleScript? = NSAppleScript(source: cmd)
    script?.executeAndReturnError(&error)
    if error != nil {
      NSSound.beep()
      print(error!.description)
    }
  }
}