mirror of https://github.com/AxioDL/nod.git
91 lines
2.0 KiB
C++
91 lines
2.0 KiB
C++
#include <stdio.h>
|
|
#include "NOD/NOD.hpp"
|
|
#include "NOD/DiscBase.hpp"
|
|
|
|
namespace NOD
|
|
{
|
|
|
|
LogVisor::LogModule LogModule("NODLib");
|
|
|
|
std::unique_ptr<IDiscIO> NewDiscIOISO(const SystemChar* path);
|
|
std::unique_ptr<IDiscIO> NewDiscIOWBFS(const SystemChar* path);
|
|
|
|
std::unique_ptr<DiscBase> OpenDiscFromImage(const SystemChar* path, bool& isWii)
|
|
{
|
|
/* Temporary file handle to determine image type */
|
|
#if NOD_UCS2
|
|
FILE* fp = _wfopen(path, L"rb");
|
|
#else
|
|
FILE* fp = fopen(path, "rb");
|
|
#endif
|
|
if (!fp)
|
|
{
|
|
#if NOD_UCS2
|
|
LogModule.report(LogVisor::Error, L"Unable to open '%s'", path);
|
|
#else
|
|
LogModule.report(LogVisor::Error, "Unable to open '%s'", path);
|
|
#endif
|
|
return std::unique_ptr<DiscBase>();
|
|
}
|
|
|
|
isWii = false;
|
|
std::unique_ptr<IDiscIO> discIO;
|
|
uint32_t magic;
|
|
fread(&magic, 1, 4, fp);
|
|
if (magic == NOD::SBig((uint32_t)'WBFS'))
|
|
{
|
|
fclose(fp);
|
|
discIO = NewDiscIOWBFS(path);
|
|
isWii = true;
|
|
}
|
|
else
|
|
{
|
|
fseeko(fp, 0x18, SEEK_SET);
|
|
fread(&magic, 1, 4, fp);
|
|
magic = NOD::SBig(magic);
|
|
if (magic == 0x5D1C9EA3)
|
|
{
|
|
fclose(fp);
|
|
discIO = NewDiscIOISO(path);
|
|
isWii = true;
|
|
}
|
|
else
|
|
{
|
|
fread(&magic, 1, 4, fp);
|
|
magic = NOD::SBig(magic);
|
|
if (magic == 0xC2339F3D)
|
|
{
|
|
fclose(fp);
|
|
discIO = NewDiscIOISO(path);
|
|
}
|
|
else
|
|
fclose(fp);
|
|
}
|
|
}
|
|
|
|
if (!discIO)
|
|
{
|
|
#if NOD_UCS2
|
|
LogModule.report(LogVisor::Error, L"'%s' is not a valid image", path);
|
|
#else
|
|
LogModule.report(LogVisor::Error, "'%s' is not a valid image", path);
|
|
#endif
|
|
return std::unique_ptr<DiscBase>();
|
|
}
|
|
|
|
if (isWii)
|
|
return std::unique_ptr<DiscBase>(new DiscWii(std::move(discIO)));
|
|
|
|
return std::unique_ptr<DiscBase>(new DiscGCN(std::move(discIO)));
|
|
|
|
}
|
|
|
|
std::unique_ptr<DiscBase> OpenDiscFromImage(const SystemChar* path)
|
|
{
|
|
bool isWii;
|
|
return OpenDiscFromImage(path, isWii);
|
|
}
|
|
|
|
}
|
|
|