mirror of
https://github.com/CloverHackyColor/CloverBootloader.git
synced 2024-11-12 09:54:36 +01:00
70988228c4
Enclose access to getOemDir... by if (selfOem.oemDirExists()). Enclose access to getKextsDir... by if (selfOem.isKextsDirFound()). closeDebugLog() in UninitRefitLib(). Prevent DBG loop.
224 lines
7.9 KiB
C++
224 lines
7.9 KiB
C++
/*
|
|
* Self.cpp
|
|
*
|
|
* Created on: Sep 28, 2020
|
|
* Author: jief
|
|
*/
|
|
|
|
#include <Platform.h>
|
|
#include "SelfOem.h"
|
|
#include "Self.h"
|
|
#include "../refit/lib.h"
|
|
|
|
#ifndef DEBUG_ALL
|
|
#define DEBUG_SELF 1
|
|
#else
|
|
#define DEBUG_SELF DEBUG_ALL
|
|
#endif
|
|
|
|
#if DEBUG_SELF == 0
|
|
#define DBG(...)
|
|
#else
|
|
#define DBG(...) DebugLog(DEBUG_SELF, __VA_ARGS__)
|
|
#endif
|
|
|
|
constexpr const LStringW KEXTS_DIRNAME(L"Kexts");
|
|
|
|
|
|
SelfOem selfOem;
|
|
|
|
|
|
//EFI_STATUS SelfOem::_openDir(const XStringW& path, bool* b, EFI_FILE** efiDir)
|
|
//{
|
|
// EFI_STATUS Status;
|
|
// Status = self.getCloverDir().Open(&self.getCloverDir(), efiDir, path.wc_str(), EFI_FILE_MODE_READ, 0);
|
|
// if ( EFI_ERROR(Status) ) {
|
|
// DBG("Error when opening dir '%ls\\%ls' : %s\n", self.getCloverDirFullPath().wc_str(), path.wc_str(), efiStrError(Status));
|
|
// *efiDir = NULL;
|
|
// *b = false;
|
|
// }else{
|
|
// *b = true;
|
|
// }
|
|
// return Status;
|
|
//}
|
|
|
|
|
|
bool SelfOem::_checkOEMPath()
|
|
{
|
|
EFI_STATUS Status;
|
|
|
|
// if ( !selfOem.oemDirExists() ) return false;
|
|
|
|
// EFI_FILE* efiDir;
|
|
Status = self.getCloverDir().Open(&self.getCloverDir(), &m_OemDir, m_OemPathRelToSelfDir.wc_str(), EFI_FILE_MODE_READ, 0);
|
|
if ( Status == EFI_NOT_FOUND ) {
|
|
DBG("_checkOEMPath Look for oem dir at path '%ls\\%ls'. Dir doesn't exist.\n", self.getCloverDirFullPath().wc_str(), m_OemPathRelToSelfDir.wc_str());
|
|
m_OemDir = NULL;
|
|
return false;
|
|
}
|
|
if ( EFI_ERROR(Status) != EFI_SUCCESS ) {
|
|
DBG("Cannot open dir at '%ls\\%ls' dir : %s\n", self.getCloverDirFullPath().wc_str(), m_OemPathRelToSelfDir.wc_str(), efiStrError(Status));
|
|
m_OemDir = NULL;
|
|
return false;
|
|
}
|
|
BOOLEAN res2 = FileExists(m_OemDir, SWPrintf("%s.plist", m_ConfName.c_str()));
|
|
if ( !res2 ) {
|
|
DBG("_checkOEMPath looked for config file at '%ls\\%ls\\%s.plist'. File doesn't exist.\n", self.getCloverDirFullPath().wc_str(), m_OemPathRelToSelfDir.wc_str(), m_ConfName.c_str());
|
|
m_OemDir->Close(m_OemDir);
|
|
m_OemDir = NULL;
|
|
return false;
|
|
}
|
|
DBG("_checkOEMPath: set OEMPath: '%ls\\%ls'\n", self.getCloverDirFullPath().wc_str(), m_OemPathRelToSelfDir.wc_str());
|
|
return true;
|
|
}
|
|
|
|
bool SelfOem::_setOemPathRelToSelfDir(bool isFirmwareClover, const XString8& OEMBoard, const XString8& OEMProduct, INT32 frequency, UINTN nLanCards, UINT8 gLanMac[4][6])
|
|
{
|
|
|
|
if ( nLanCards > 0 ) {
|
|
m_OemPathRelToSelfDir.SWPrintf("OEM\\%s--%02X-%02X-%02X-%02X-%02X-%02X", OEMProduct.c_str(), gLanMac[0][0], gLanMac[0][1], gLanMac[0][2], gLanMac[0][3], gLanMac[0][4], gLanMac[0][5]);
|
|
if ( _checkOEMPath() ) return true;
|
|
}
|
|
if ( nLanCards > 1 ) {
|
|
m_OemPathRelToSelfDir.SWPrintf("OEM\\%s--%02X-%02X-%02X-%02X-%02X-%02X", OEMProduct.c_str(), gLanMac[1][0], gLanMac[1][1], gLanMac[1][2], gLanMac[1][3], gLanMac[1][4], gLanMac[1][5]);
|
|
if ( _checkOEMPath() ) return true;
|
|
}
|
|
if ( nLanCards > 2 ) {
|
|
m_OemPathRelToSelfDir.SWPrintf("OEM\\%s--%02X-%02X-%02X-%02X-%02X-%02X", OEMProduct.c_str(), gLanMac[2][0], gLanMac[2][1], gLanMac[2][2], gLanMac[2][3], gLanMac[2][4], gLanMac[2][5]);
|
|
if ( _checkOEMPath() ) return true;
|
|
}
|
|
if ( nLanCards > 3 ) {
|
|
m_OemPathRelToSelfDir.SWPrintf("OEM\\%s--%02X-%02X-%02X-%02X-%02X-%02X", OEMProduct.c_str(), gLanMac[3][0], gLanMac[3][1], gLanMac[3][2], gLanMac[3][3], gLanMac[3][4], gLanMac[3][5]);
|
|
if ( _checkOEMPath() ) return true;
|
|
}
|
|
if ( !isFirmwareClover ) {
|
|
m_OemPathRelToSelfDir.SWPrintf("OEM\\%s\\UEFI", OEMBoard.c_str());
|
|
if ( _checkOEMPath() ) return true;
|
|
}
|
|
m_OemPathRelToSelfDir.SWPrintf("OEM\\%s", OEMProduct.c_str());
|
|
if ( _checkOEMPath() ) return true;
|
|
m_OemPathRelToSelfDir.SWPrintf("OEM\\%s-%d", OEMProduct.c_str(), frequency);
|
|
if ( _checkOEMPath() ) return true;
|
|
m_OemPathRelToSelfDir.SWPrintf("OEM\\%s", OEMBoard.c_str());
|
|
if ( _checkOEMPath() ) return true;
|
|
m_OemPathRelToSelfDir.SWPrintf("OEM\\%s-%d", OEMBoard.c_str(), frequency);
|
|
if ( _checkOEMPath() ) return true;
|
|
|
|
// m_OemPathRelToSelfDir.takeValueFrom(".");
|
|
// DBG("set OEMPath to \".\"\n");
|
|
m_OemPathRelToSelfDir.setEmpty();
|
|
|
|
return false;
|
|
}
|
|
|
|
|
|
EFI_STATUS SelfOem::_initialize()
|
|
{
|
|
EFI_STATUS Status;
|
|
|
|
if ( oemDirExists() ) {
|
|
m_OemFulPath = SWPrintf("%ls\\%ls", self.getCloverDirFullPath().wc_str(), m_OemPathRelToSelfDir.wc_str());
|
|
m_configDirPathRelToSelfDir = getOemPathRelToSelfDir();
|
|
m_configDirPathRelToSelfDirWithTrailingSlash.SWPrintf("%ls\\", m_configDirPathRelToSelfDir.wc_str());
|
|
}else{
|
|
m_OemFulPath.setEmpty();
|
|
m_configDirPathRelToSelfDir.setEmpty();
|
|
m_configDirPathRelToSelfDirWithTrailingSlash.setEmpty();
|
|
}
|
|
if ( m_OemDir == NULL ) {
|
|
assert( m_OemPathRelToSelfDir.isEmpty() );
|
|
assert( m_OemFulPath.isEmpty() );
|
|
}else{
|
|
assert( m_OemPathRelToSelfDir.notEmpty() );
|
|
assert( m_OemFulPath.notEmpty() );
|
|
}
|
|
|
|
|
|
|
|
if ( m_KextsDir != NULL ) panic("%s : m_KextsDir != NULL", __FUNCTION__);
|
|
if ( oemDirExists() ) {
|
|
Status = m_OemDir->Open(m_OemDir, &m_KextsDir, KEXTS_DIRNAME.wc_str(), EFI_FILE_MODE_READ, 0);
|
|
if ( !EFI_ERROR(Status) ) {
|
|
m_KextsPathRelToSelfDir.SWPrintf("%ls\\%ls", getOemPathRelToSelfDir().wc_str(), KEXTS_DIRNAME.wc_str());
|
|
m_KextsFullPath.SWPrintf("%ls\\%ls", getOemFullPath().wc_str(), KEXTS_DIRNAME.wc_str());
|
|
}else{
|
|
DBG("Cannot open %ls\\%ls : %s", getOemFullPath().wc_str(), KEXTS_DIRNAME.wc_str(), efiStrError(Status));
|
|
m_KextsDir = NULL;
|
|
}
|
|
}
|
|
// if ( Status != EFI_SUCCESS && Status != EFI_NOT_FOUND ) {
|
|
// panic("Cannot open kexts dir %ls\\%ls : %s", getOemFullPath().wc_str(), KEXTS_DIRNAME.wc_str(), efiStrError(Status));
|
|
// }
|
|
if ( m_KextsDir == NULL ) {
|
|
Status = self.getCloverDir().Open(&self.getCloverDir(), &m_KextsDir, KEXTS_DIRNAME.wc_str(), EFI_FILE_MODE_READ, 0);
|
|
if ( EFI_ERROR(Status) ) {
|
|
DBG("Cannot open %ls\\%ls : %s", self.getCloverDirFullPath().wc_str(), KEXTS_DIRNAME.wc_str(), efiStrError(Status));
|
|
//panic("Cannot open kexts dir at '%ls\\%ls'", self.getCloverDirFullPath().wc_str(), KEXTS_DIRNAME.wc_str());
|
|
m_KextsDir = NULL;
|
|
m_KextsPathRelToSelfDir.setEmpty();
|
|
m_KextsFullPath.setEmpty();
|
|
}else{
|
|
m_KextsPathRelToSelfDir = KEXTS_DIRNAME;
|
|
m_KextsFullPath.SWPrintf("%ls\\%ls", self.getCloverDirFullPath().wc_str(), KEXTS_DIRNAME.wc_str());
|
|
}
|
|
}else{
|
|
}
|
|
if ( m_KextsDir == NULL ) {
|
|
assert( m_KextsPathRelToSelfDir.isEmpty() );
|
|
assert( m_KextsFullPath.isEmpty() );
|
|
}else{
|
|
assert( m_KextsPathRelToSelfDir.notEmpty() );
|
|
assert( m_KextsFullPath.notEmpty() );
|
|
}
|
|
|
|
if ( isKextsDirFound() ) {
|
|
DBG("Kexts dir = '%ls'\n", getKextsFullPath().wc_str());
|
|
}else{
|
|
DBG("Kexts dir = none\n");
|
|
}
|
|
|
|
return EFI_SUCCESS;
|
|
}
|
|
|
|
EFI_STATUS SelfOem::initialize(const XString8& confName, bool isFirmwareClover, const XString8& OEMBoard, const XString8& OEMProduct, INT32 frequency, UINTN nLanCards, UINT8 gLanMac[4][6])
|
|
{
|
|
m_ConfName = confName;
|
|
|
|
// Initialise m_OemPathRelToSelfDir and leave m_OemDir opened.
|
|
_setOemPathRelToSelfDir(isFirmwareClover, OEMBoard, OEMProduct, frequency, nLanCards, gLanMac);
|
|
|
|
return _initialize();
|
|
}
|
|
|
|
EFI_STATUS SelfOem::reInitialize()
|
|
{
|
|
closeHandle();
|
|
|
|
// No need to call _setOemPathRelToSelfDir again, but need to open m_OemDir, if it exists
|
|
if ( oemDirExists() ) {
|
|
EFI_STATUS Status = self.getCloverDir().Open(&self.getCloverDir(), &m_OemDir, m_OemPathRelToSelfDir.wc_str(), EFI_FILE_MODE_READ, 0);
|
|
if ( EFI_ERROR(Status) ) {
|
|
panic("Impossible to reopen dir '%ls\\%ls', although it was opened the first time : %s", self.getCloverDirFullPath().wc_str(), m_OemPathRelToSelfDir.wc_str(), efiStrError(Status));
|
|
}
|
|
}
|
|
return _initialize();
|
|
}
|
|
|
|
|
|
void SelfOem::closeHandle(void)
|
|
{
|
|
if (m_KextsDir != NULL) {
|
|
m_KextsDir->Close(m_KextsDir);
|
|
m_KextsDir = NULL;
|
|
m_KextsPathRelToSelfDir.setEmpty();
|
|
m_KextsFullPath.setEmpty();
|
|
}
|
|
if (m_OemDir != NULL) {
|
|
m_OemDir->Close(m_OemDir);
|
|
m_OemDir = NULL;
|
|
// m_OemPathRelToSelfDir.setEmpty(); // do not empty m_OemPathRelToSelfDir, we need it in reInitialize()
|
|
// m_OemFulPath.setEmpty(); // doesn't matter, it'll be re-initialised in _initialize()
|
|
}
|
|
}
|
|
|