mirror of
https://github.com/CloverHackyColor/CloverBootloader.git
synced 2024-11-29 12:35:53 +01:00
521 lines
19 KiB
Swift
521 lines
19 KiB
Swift
//
|
||
// DriversCollection.swift
|
||
// Clover
|
||
//
|
||
// Created by vector sigma on 27/11/2019.
|
||
// Copyright © 2019 CloverHackyColor. All rights reserved.
|
||
//
|
||
|
||
import Cocoa
|
||
|
||
let collectionItemWith : Int = 155
|
||
|
||
let kUEFIRelativeOptionDir : String = /* CloverV2 + */ "EFI/CLOVER/drivers/off/UEFI"
|
||
let kBIOSRelativeOptionDir : String = /* CloverV2 + */ "EFI/CLOVER/drivers/off/BIOS"
|
||
|
||
let kBIOSRelativeDir : String = /* target volume + */ "EFI/CLOVER/drivers/BIOS"
|
||
let kUEFIRelativeDir : String = /* target volume + */ "EFI/CLOVER/drivers/UEFI"
|
||
|
||
let kUnknownUEFISection = "UEFI, but not from this installer"
|
||
let kUnknownBIOSSection = "BIOS, but not from this installer"
|
||
|
||
// MARK: ItemTextFieldCell (NSTextFieldCell sub class)
|
||
final class ItemTextFieldCell: NSTextFieldCell {
|
||
override func drawingRect(forBounds rect: NSRect) -> NSRect {
|
||
var nr = super.drawingRect(forBounds: rect)
|
||
let size = self.cellSize(forBounds: rect)
|
||
let diff = nr.size.height - size.height
|
||
if diff > 0 {
|
||
nr.size.height -= diff
|
||
nr.origin.y += (diff / 2)
|
||
}
|
||
return nr
|
||
}
|
||
}
|
||
|
||
// MARK: ItemTextField (NSTextfield sub class)
|
||
final class ItemTextField: NSTextField {
|
||
var trackingArea: NSTrackingArea? = nil
|
||
override init(frame frameRect: NSRect) {
|
||
super.init(frame: frameRect)
|
||
self.cell = ItemTextFieldCell()
|
||
|
||
self.trackingArea = NSTrackingArea(rect: self.bounds,
|
||
options: [NSTrackingArea.Options.activeAlways, NSTrackingArea.Options.mouseEnteredAndExited],
|
||
owner: self, userInfo: nil)
|
||
}
|
||
|
||
|
||
required init?(coder: NSCoder) {
|
||
fatalError("init(coder:) has not been implemented")
|
||
}
|
||
|
||
override func mouseDown(with event: NSEvent) {
|
||
self.showDescription()
|
||
}
|
||
|
||
override func updateTrackingAreas() {
|
||
if self.trackingArea != nil {
|
||
self.removeTrackingArea(self.trackingArea!)
|
||
}
|
||
|
||
self.trackingArea = NSTrackingArea(rect: self.bounds,
|
||
options: [NSTrackingArea.Options.activeAlways, NSTrackingArea.Options.mouseEnteredAndExited],
|
||
owner: self, userInfo: nil)
|
||
|
||
self.addTrackingArea(self.trackingArea!)
|
||
}
|
||
|
||
private func showDescription() {
|
||
if AppSD.isInstalling {
|
||
return
|
||
}
|
||
if let info = self.cell?.representedObject as? String {
|
||
if let ivc = self.target as? InstallerViewController {
|
||
ivc.post(text: info, add: false, color: nil, scroll: false)
|
||
}
|
||
}
|
||
}
|
||
}
|
||
|
||
// MARK: CollectionViewItem (NSCollectionViewItem sub class)
|
||
final class CollectionViewItem: NSCollectionViewItem {
|
||
var driver : EFIDriver? = nil
|
||
var installerController : InstallerViewController? = nil
|
||
var installerOutController : InstallerOutViewController? = nil
|
||
|
||
public func setState(_ state: NSControl.StateValue) {
|
||
self.driver?.state = state
|
||
self.checkBox.state = state
|
||
}
|
||
|
||
public let checkBox: NSButton = {
|
||
let butt = NSButton()
|
||
if #available(OSX 10.10, *) {
|
||
butt.controlSize = .regular
|
||
}
|
||
butt.imagePosition = .noImage
|
||
butt.frame = NSRect(x: 0, y: 0, width: 18, height: 18)
|
||
butt.setButtonType(NSButton.ButtonType.switch)
|
||
butt.state = .off
|
||
return butt
|
||
} ()
|
||
|
||
public let field: ItemTextField = {
|
||
let f = ItemTextField(frame: NSRect(x: 18, y: 0, width: collectionItemWith - 18, height: 18))
|
||
if #available(OSX 10.10, *) {
|
||
f.controlSize = .mini
|
||
}
|
||
f.drawsBackground = false
|
||
f.isEditable = false
|
||
f.isSelectable = false
|
||
return f
|
||
} ()
|
||
|
||
@objc func checkBoxPressed(_ sender: NSButton?) {
|
||
if (sender != nil) {
|
||
self.driver?.state = sender!.state
|
||
if let sectionName = self.driver?.sectionName, let installer = self.installerController {
|
||
let kind = self.driver!.kind
|
||
let currDriverName = self.driver!.src.lastPath
|
||
let sections = installer.sectionsUEFI
|
||
// MemoryFix drivers allow only one choice
|
||
if sectionName.hasSuffix("MemoryFix") && sections.contains(sectionName) {
|
||
allowOnly(driver: self.driver!, kind: kind, in: sectionName, installer: installer)
|
||
} else if sectionName.hasSuffix("FileSystem") && sections.contains(sectionName) {
|
||
if self.driver!.state == .on {
|
||
var exclude : [String] = [String]()
|
||
let currLower = currDriverName.deletingFileExtension.lowercased()
|
||
if currLower == "vboxhfs" {
|
||
exclude = ["hfsplus"]
|
||
} else if currLower == "hfsplus" {
|
||
exclude = ["vboxhfs"]
|
||
} else if currLower == "apfsdriverloader" {
|
||
exclude = ["apfs"]
|
||
} else if currLower == "apfs" {
|
||
exclude = ["apfsdriverloader"]
|
||
} else if currLower == "ntfs" {
|
||
exclude = ["grubntfs"]
|
||
} else if currLower == "grubntfs" {
|
||
exclude = ["ntfs"]
|
||
}
|
||
uncheck(list: exclude,
|
||
current: currDriverName,
|
||
kind: kind,
|
||
sectionName: sectionName,
|
||
installer: installer)
|
||
}
|
||
} else if sectionName.hasSuffix("mandatory") && sections.contains(sectionName) {
|
||
if self.driver!.state == .on {
|
||
var exclude : [String] = [String]()
|
||
let currLower = currDriverName.deletingFileExtension.lowercased()
|
||
if currLower == "smchelper" {
|
||
exclude = ["virtualsmc"]
|
||
} else if currLower == "virtualsmc" {
|
||
exclude = ["smchelper"]
|
||
}
|
||
uncheck(list: exclude,
|
||
current: currDriverName,
|
||
kind: kind,
|
||
sectionName: sectionName,
|
||
installer: installer)
|
||
}
|
||
} else if sectionName.hasSuffix("FileVault2") && sections.contains(sectionName) {
|
||
/*
|
||
AppleImageCodec.efi, AppleKeyAggregator.efi, AppleKeyMapAggregator.efi, AppleEvent.efi, AppleUITheme.efi, EnglishDxe-64.efi, FirmwareVolume.efi, HashServiceFix.efi
|
||
|
||
vs
|
||
|
||
AppleUiSupport
|
||
*/
|
||
if self.driver!.state == .on {
|
||
var exclude : [String] = [String]()
|
||
let currLower = currDriverName.deletingFileExtension.lowercased()
|
||
if currLower == "appleimagecodec" ||
|
||
currLower == "applekeyaggregator" ||
|
||
currLower == "applekeymapaggregator" ||
|
||
currLower == "appleevent" ||
|
||
currLower == "appleuitheme" ||
|
||
currLower == "englishdxe" ||
|
||
currLower == "firmwarevolume" ||
|
||
currLower == "hashservicefix" {
|
||
exclude = ["appleuisupport"]
|
||
} else if currLower == "appleuisupport" {
|
||
exclude = ["appleimagecodec",
|
||
"applekeyaggregator",
|
||
"applekeymapaggregator",
|
||
"appleevent",
|
||
"appleuitheme",
|
||
"englishdxe",
|
||
"firmwarevolume",
|
||
"hashservicefix"]
|
||
}
|
||
uncheck(list: exclude,
|
||
current: currDriverName,
|
||
kind: kind,
|
||
sectionName: sectionName,
|
||
installer: installer)
|
||
}
|
||
}
|
||
} else if let sectionName = self.driver?.sectionName, let installer = self.installerOutController {
|
||
let kind = self.driver!.kind
|
||
let currDriverName = self.driver!.src.lastPath
|
||
let sections = installer.sectionsUEFI
|
||
// MemoryFix drivers allow only one choice
|
||
if sectionName.hasSuffix("MemoryFix") && sections.contains(sectionName) {
|
||
allowOnly(driver: self.driver!, kind: kind, in: sectionName, installer: installer)
|
||
} else if sectionName.hasSuffix("FileSystem") && sections.contains(sectionName) {
|
||
if self.driver!.state == .on {
|
||
var exclude : [String] = [String]()
|
||
let currLower = currDriverName.deletingFileExtension.lowercased()
|
||
if currLower == "vboxhfs" {
|
||
exclude = ["hfsplus"]
|
||
} else if currLower == "hfsplus" {
|
||
exclude = ["vboxhfs"]
|
||
} else if currLower == "apfsdriverloader" {
|
||
exclude = ["apfs"]
|
||
} else if currLower == "apfs" {
|
||
exclude = ["apfsdriverloader"]
|
||
} else if currLower == "ntfs" {
|
||
exclude = ["grubntfs"]
|
||
} else if currLower == "grubntfs" {
|
||
exclude = ["ntfs"]
|
||
}
|
||
uncheck(list: exclude,
|
||
current: currDriverName,
|
||
kind: kind,
|
||
sectionName: sectionName,
|
||
installer: installer)
|
||
}
|
||
} else if sectionName.hasSuffix("mandatory") && sections.contains(sectionName) {
|
||
if self.driver!.state == .on {
|
||
var exclude : [String] = [String]()
|
||
let currLower = currDriverName.deletingFileExtension.lowercased()
|
||
if currLower == "smchelper" {
|
||
exclude = ["virtualsmc"]
|
||
} else if currLower == "virtualsmc" {
|
||
exclude = ["smchelper"]
|
||
}
|
||
uncheck(list: exclude,
|
||
current: currDriverName,
|
||
kind: kind,
|
||
sectionName: sectionName,
|
||
installer: installer)
|
||
}
|
||
} else if sectionName.hasSuffix("FileVault2") && sections.contains(sectionName) {
|
||
/*
|
||
AppleImageCodec.efi, AppleKeyAggregator.efi, AppleKeyMapAggregator.efi, AppleEvent.efi, AppleUITheme.efi, EnglishDxe-64.efi, FirmwareVolume.efi, HashServiceFix.efi
|
||
|
||
vs
|
||
|
||
AppleUiSupport
|
||
*/
|
||
if self.driver!.state == .on {
|
||
var exclude : [String] = [String]()
|
||
let currLower = currDriverName.deletingFileExtension.lowercased()
|
||
if currLower == "appleimagecodec" ||
|
||
currLower == "applekeyaggregator" ||
|
||
currLower == "applekeymapaggregator" ||
|
||
currLower == "appleevent" ||
|
||
currLower == "appleuitheme" ||
|
||
currLower == "englishdxe" ||
|
||
currLower == "firmwarevolume" ||
|
||
currLower == "hashservicefix" {
|
||
exclude = ["appleuisupport"]
|
||
} else if currLower == "appleuisupport" {
|
||
exclude = ["appleimagecodec",
|
||
"applekeyaggregator",
|
||
"applekeymapaggregator",
|
||
"appleevent",
|
||
"appleuitheme",
|
||
"englishdxe",
|
||
"firmwarevolume",
|
||
"hashservicefix"]
|
||
}
|
||
uncheck(list: exclude,
|
||
current: currDriverName,
|
||
kind: kind,
|
||
sectionName: sectionName,
|
||
installer: installer)
|
||
}
|
||
}
|
||
}
|
||
}
|
||
}
|
||
|
||
private func allowOnly(driver: EFIDriver,
|
||
kind: EFIkind,
|
||
in sectionName: String,
|
||
installer: NSViewController) {
|
||
if let vc = installer as? InstallerViewController {
|
||
let sections = vc.sectionsUEFI
|
||
let sectIndex : Int = sections.firstIndex(of: sectionName)!
|
||
let drivers = vc.driversUEFI[sectIndex]
|
||
let driverName = driver.src.lastPath
|
||
|
||
for (index, drv) in drivers.enumerated() {
|
||
if drv.src.lastPath != driverName {
|
||
vc.driversUEFI[sectIndex][index].state = .off
|
||
} else {
|
||
vc.driversUEFI[sectIndex][index].state = self.driver!.state
|
||
self.driver = vc.driversUEFI[sectIndex][index]
|
||
}
|
||
}
|
||
|
||
let unknownSection = kind == .uefi ? kUnknownUEFISection : kUnknownBIOSSection
|
||
if driver.state == .on && vc.sectionsUEFI.contains(unknownSection) {
|
||
if sectionName == "UEFI/MemoryFix" {
|
||
uncheck(list: ["aptiomemory", "osxlowmem", "osxaptiofix", "ocquirks", "openruntime"],
|
||
current: driverName,
|
||
kind: kind,
|
||
sectionName: unknownSection,
|
||
installer: installer)
|
||
}
|
||
}
|
||
|
||
// OcQuirks exception
|
||
if sectionName == "UEFI/MemoryFix" &&
|
||
kind == .uefi &&
|
||
(driverName.lowercased() == "ocquirks.efi" || driverName.lowercased() == "openruntime.efi") &&
|
||
driver.state == .on {
|
||
check(driver: "OcQuirks.efi", kind: kind, sectionName: sectionName, installer: vc)
|
||
check(driver: "OpenRuntime.efi", kind: kind, sectionName: sectionName, installer: vc)
|
||
}
|
||
} else if let vc = installer as? InstallerOutViewController {
|
||
let sections = vc.sectionsUEFI
|
||
let sectIndex : Int = sections.firstIndex(of: sectionName)!
|
||
let drivers = vc.driversUEFI[sectIndex]
|
||
let driverName = driver.src.lastPath
|
||
|
||
for (index, drv) in drivers.enumerated() {
|
||
if drv.src.lastPath != driverName {
|
||
vc.driversUEFI[sectIndex][index].state = .off
|
||
} else {
|
||
vc.driversUEFI[sectIndex][index].state = self.driver!.state
|
||
self.driver = vc.driversUEFI[sectIndex][index]
|
||
}
|
||
}
|
||
|
||
let unknownSection = kind == .uefi ? kUnknownUEFISection : kUnknownBIOSSection
|
||
if driver.state == .on && vc.sectionsUEFI.contains(unknownSection) {
|
||
if sectionName == "UEFI/MemoryFix" {
|
||
uncheck(list: ["aptiomemory", "osxlowmem", "osxaptiofix", "ocquirks", "openruntime"],
|
||
current: driverName,
|
||
kind: kind,
|
||
sectionName: unknownSection,
|
||
installer: installer)
|
||
}
|
||
}
|
||
|
||
// OcQuirks exception
|
||
if sectionName == "UEFI/MemoryFix" &&
|
||
kind == .uefi &&
|
||
(driverName.lowercased() == "ocquirks.efi" || driverName.lowercased() == "openruntime.efi") &&
|
||
driver.state == .on {
|
||
check(driver: "OcQuirks.efi", kind: kind, sectionName: sectionName, installer: vc)
|
||
check(driver: "OpenRuntime.efi", kind: kind, sectionName: sectionName, installer: vc)
|
||
}
|
||
}
|
||
}
|
||
|
||
private func check(driver name: String,
|
||
kind: EFIkind,
|
||
sectionName: String,
|
||
installer: NSViewController) {
|
||
if let vc = installer as? InstallerViewController {
|
||
let sections = vc.sectionsUEFI
|
||
let sectIndex : Int = sections.firstIndex(of: sectionName)!
|
||
let drivers = vc.driversUEFI[sectIndex]
|
||
|
||
for (index, drv) in drivers.enumerated() {
|
||
if drv.src.lastPath == name {
|
||
if vc.driversUEFI[sectIndex][index].state == .off {
|
||
vc.driversUEFI[sectIndex][index].state = .on
|
||
}
|
||
}
|
||
}
|
||
} else if let vc = installer as? InstallerOutViewController {
|
||
let sections = vc.sectionsUEFI
|
||
let sectIndex : Int = sections.firstIndex(of: sectionName)!
|
||
let drivers = vc.driversUEFI[sectIndex]
|
||
|
||
for (index, drv) in drivers.enumerated() {
|
||
if drv.src.lastPath == name {
|
||
if vc.driversUEFI[sectIndex][index].state == .off {
|
||
vc.driversUEFI[sectIndex][index].state = .on
|
||
}
|
||
}
|
||
}
|
||
}
|
||
}
|
||
|
||
private func uncheck(list: [String],
|
||
current: String,
|
||
kind: EFIkind,
|
||
sectionName: String,
|
||
installer: NSViewController) {
|
||
if let vc = installer as? InstallerViewController {
|
||
let sections = vc.sectionsUEFI
|
||
let sectIndex : Int = sections.firstIndex(of: sectionName)!
|
||
let drivers = vc.driversUEFI[sectIndex]
|
||
|
||
for (index, drv) in drivers.enumerated() {
|
||
if drv.src.lastPath != current {
|
||
for d in list {
|
||
if drv.src.lastPath.lowercased().hasPrefix(d) {
|
||
if vc.driversUEFI[sectIndex][index].state != .off {
|
||
vc.driversUEFI[sectIndex][index].state = .off
|
||
}
|
||
}
|
||
}
|
||
} else {
|
||
if vc.driversUEFI[sectIndex][index].state != self.driver!.state {
|
||
vc.driversUEFI[sectIndex][index].state = self.driver!.state
|
||
}
|
||
}
|
||
}
|
||
|
||
// Since we are excluding drivers by prefix.. then looks for the same in unknown drivers
|
||
let unknownSection = kind == .uefi ? kUnknownUEFISection : kUnknownBIOSSection
|
||
if vc.sectionsUEFI.contains(unknownSection) {
|
||
let usections = vc.sectionsUEFI
|
||
let usectIndex : Int = usections.firstIndex(of: unknownSection)!
|
||
let udrivers = vc.driversUEFI[usectIndex]
|
||
for (index, drv) in udrivers.enumerated() {
|
||
for d in list {
|
||
if drv.src.lastPath.lowercased().hasPrefix(d) {
|
||
vc.driversUEFI[usectIndex][index].state = .off
|
||
}
|
||
}
|
||
}
|
||
}
|
||
} else if let vc = installer as? InstallerOutViewController {
|
||
let sections = vc.sectionsUEFI
|
||
let sectIndex : Int = sections.firstIndex(of: sectionName)!
|
||
let drivers = vc.driversUEFI[sectIndex]
|
||
|
||
for (index, drv) in drivers.enumerated() {
|
||
if drv.src.lastPath != current {
|
||
for d in list {
|
||
if drv.src.lastPath.lowercased().hasPrefix(d) {
|
||
if vc.driversUEFI[sectIndex][index].state != .off {
|
||
vc.driversUEFI[sectIndex][index].state = .off
|
||
}
|
||
}
|
||
}
|
||
} else {
|
||
if vc.driversUEFI[sectIndex][index].state != self.driver!.state {
|
||
vc.driversUEFI[sectIndex][index].state = self.driver!.state
|
||
}
|
||
}
|
||
}
|
||
|
||
// Since we are excluding drivers by prefix.. then looks for the same in unknown drivers
|
||
let unknownSection = kind == .uefi ? kUnknownUEFISection : kUnknownBIOSSection
|
||
if vc.sectionsUEFI.contains(unknownSection) {
|
||
let usections = vc.sectionsUEFI
|
||
let usectIndex : Int = usections.firstIndex(of: unknownSection)!
|
||
let udrivers = vc.driversUEFI[usectIndex]
|
||
for (index, drv) in udrivers.enumerated() {
|
||
for d in list {
|
||
if drv.src.lastPath.lowercased().hasPrefix(d) {
|
||
vc.driversUEFI[usectIndex][index].state = .off
|
||
}
|
||
}
|
||
}
|
||
}
|
||
}
|
||
|
||
}
|
||
|
||
override func viewDidLoad() {
|
||
if #available(OSX 10.10, *) {
|
||
super.viewDidLoad()
|
||
}
|
||
self.view.wantsLayer = true
|
||
self.view.layer?.backgroundColor = NSColor.clear.cgColor
|
||
self.checkBox.target = self
|
||
self.checkBox.action = #selector(self.checkBoxPressed(_:))
|
||
}
|
||
|
||
override func loadView() {
|
||
self.view = NSView(frame: NSRect(x: 0, y: 0, width: collectionItemWith, height: 18))
|
||
self.view.addSubview(self.checkBox)
|
||
self.checkBox.target = self
|
||
self.view.addSubview(self.field)
|
||
|
||
if #available(OSX 10.10, *) {} else {
|
||
self.viewDidLoad()
|
||
}
|
||
}
|
||
}
|
||
|
||
// MARK: HeaderView (NSView sub class)
|
||
final class HeaderView: NSView {
|
||
public let field: NSTextField = {
|
||
let f = NSTextField()
|
||
if #available(OSX 10.10, *) {
|
||
f.controlSize = .regular
|
||
}
|
||
f.isEditable = false
|
||
f.isBordered = false
|
||
f.drawsBackground = false
|
||
f.textColor = .white
|
||
f.frame = NSRect(x: 0, y: 0, width: 250, height: 18)
|
||
return f
|
||
} ()
|
||
|
||
override init(frame frameRect: NSRect) {
|
||
super.init(frame: frameRect)
|
||
self.wantsLayer = true
|
||
self.layer?.backgroundColor = NSColor.gray.cgColor
|
||
self.addSubview(self.field)
|
||
self.field.stringValue = "header"
|
||
}
|
||
|
||
required init?(coder: NSCoder) {
|
||
fatalError("init(coder:) has not been implemented")
|
||
}
|
||
}
|
||
|