summaryrefslogtreecommitdiff
path: root/Source/Core/DiscIO/FileHandlerARC.cpp
diff options
context:
space:
mode:
authorJasper St. Pierre <jstpierre@mecheye.net>2013-12-07 15:14:29 -0500
committerJasper St. Pierre <jstpierre@mecheye.net>2013-12-31 14:03:19 -0500
commit34692ab826abc8f8faa61bdb2280b742424528f1 (patch)
treeb0eedc645badbea316ff59c125eaf808b28777ec /Source/Core/DiscIO/FileHandlerARC.cpp
parent43e618682e8093602fc83dfa902ee7ae3fd3a4f6 (diff)
Remove unnecessary Src/ folders
Diffstat (limited to 'Source/Core/DiscIO/FileHandlerARC.cpp')
-rw-r--r--Source/Core/DiscIO/FileHandlerARC.cpp245
1 files changed, 245 insertions, 0 deletions
diff --git a/Source/Core/DiscIO/FileHandlerARC.cpp b/Source/Core/DiscIO/FileHandlerARC.cpp
new file mode 100644
index 0000000000..26ec6ec9ec
--- /dev/null
+++ b/Source/Core/DiscIO/FileHandlerARC.cpp
@@ -0,0 +1,245 @@
+// Copyright 2013 Dolphin Emulator Project
+// Licensed under GPLv2
+// Refer to the license.txt file included.
+
+#include "FileHandlerARC.h"
+#include "StringUtil.h"
+#include "Blob.h"
+#include "FileUtil.h"
+
+#define ARC_ID 0x55aa382d
+
+namespace DiscIO
+{
+CARCFile::CARCFile(const std::string& _rFilename)
+ : m_pBuffer(NULL)
+ , m_Initialized(false)
+{
+ DiscIO::IBlobReader* pReader = DiscIO::CreateBlobReader(_rFilename.c_str());
+ if (pReader != NULL)
+ {
+ u64 FileSize = pReader->GetDataSize();
+ m_pBuffer = new u8[(u32)FileSize];
+ pReader->Read(0, FileSize, m_pBuffer);
+ delete pReader;
+
+ m_Initialized = ParseBuffer();
+ }
+}
+
+CARCFile::CARCFile(const std::string& _rFilename, u32 offset)
+ : m_pBuffer(NULL)
+ , m_Initialized(false)
+{
+ DiscIO::IBlobReader* pReader = DiscIO::CreateBlobReader(_rFilename.c_str());
+ if (pReader != NULL)
+ {
+ u64 FileSize = pReader->GetDataSize() - offset;
+ m_pBuffer = new u8[(u32)FileSize];
+ pReader->Read(offset, FileSize, m_pBuffer);
+ delete pReader;
+
+ m_Initialized = ParseBuffer();
+ }
+}
+
+CARCFile::CARCFile(const u8* _pBuffer, size_t _BufferSize)
+ : m_pBuffer(NULL)
+ , m_Initialized(false)
+{
+ m_pBuffer = new u8[_BufferSize];
+
+ if (m_pBuffer)
+ {
+ memcpy(m_pBuffer, _pBuffer, _BufferSize);
+ m_Initialized = ParseBuffer();
+ }
+}
+
+
+CARCFile::~CARCFile()
+{
+ delete [] m_pBuffer;
+}
+
+
+bool
+CARCFile::IsInitialized()
+{
+ return(m_Initialized);
+}
+
+
+size_t
+CARCFile::GetFileSize(const std::string& _rFullPath)
+{
+ if (!m_Initialized)
+ {
+ return(0);
+ }
+
+ const SFileInfo* pFileInfo = FindFileInfo(_rFullPath);
+
+ if (pFileInfo != NULL)
+ {
+ return((size_t) pFileInfo->m_FileSize);
+ }
+
+ return(0);
+}
+
+
+size_t
+CARCFile::ReadFile(const std::string& _rFullPath, u8* _pBuffer, size_t _MaxBufferSize)
+{
+ if (!m_Initialized)
+ {
+ return(0);
+ }
+
+ const SFileInfo* pFileInfo = FindFileInfo(_rFullPath);
+
+ if (pFileInfo == NULL)
+ {
+ return(0);
+ }
+
+ if (pFileInfo->m_FileSize > _MaxBufferSize)
+ {
+ return(0);
+ }
+
+ memcpy(_pBuffer, &m_pBuffer[pFileInfo->m_Offset], (size_t)pFileInfo->m_FileSize);
+ return((size_t) pFileInfo->m_FileSize);
+}
+
+
+bool
+CARCFile::ExportFile(const std::string& _rFullPath, const std::string& _rExportFilename)
+{
+ if (!m_Initialized)
+ {
+ return(false);
+ }
+
+ const SFileInfo* pFileInfo = FindFileInfo(_rFullPath);
+
+ if (pFileInfo == NULL)
+ {
+ return(false);
+ }
+
+ File::IOFile pFile(_rExportFilename, "wb");
+
+ return pFile.WriteBytes(&m_pBuffer[pFileInfo->m_Offset], (size_t) pFileInfo->m_FileSize);
+}
+
+
+bool
+CARCFile::ExportAllFiles(const std::string& _rFullPath)
+{
+ return(false);
+}
+
+
+bool
+CARCFile::ParseBuffer()
+{
+ // check ID
+ u32 ID = Common::swap32(*(u32*)(m_pBuffer));
+
+ if (ID != ARC_ID)
+ return false;
+
+ // read header
+ u32 FSTOffset = Common::swap32(*(u32*)(m_pBuffer + 0x4));
+ //u32 FSTSize = Common::swap32(*(u32*)(m_pBuffer + 0x8));
+ //u32 FileOffset = Common::swap32(*(u32*)(m_pBuffer + 0xC));
+
+ // read all file infos
+ SFileInfo Root;
+ Root.m_NameOffset = Common::swap32(*(u32*)(m_pBuffer + FSTOffset + 0x0));
+ Root.m_Offset = Common::swap32(*(u32*)(m_pBuffer + FSTOffset + 0x4));
+ Root.m_FileSize = Common::swap32(*(u32*)(m_pBuffer + FSTOffset + 0x8));
+
+ if (Root.IsDirectory())
+ {
+ m_FileInfoVector.resize((unsigned int)Root.m_FileSize);
+ const char* szNameTable = (char*)(m_pBuffer + FSTOffset);
+
+ for (size_t i = 0; i < m_FileInfoVector.size(); i++)
+ {
+ u8* Offset = m_pBuffer + FSTOffset + (i * 0xC);
+ m_FileInfoVector[i].m_NameOffset = Common::swap32(*(u32*)(Offset + 0x0));
+ m_FileInfoVector[i].m_Offset = Common::swap32(*(u32*)(Offset + 0x4));
+ m_FileInfoVector[i].m_FileSize = Common::swap32(*(u32*)(Offset + 0x8));
+
+ szNameTable += 0xC;
+ }
+
+ BuildFilenames(1, m_FileInfoVector.size(), NULL, szNameTable);
+ }
+
+ return(true);
+}
+
+
+size_t
+CARCFile::BuildFilenames(const size_t _FirstIndex, const size_t _LastIndex, const char* _szDirectory, const char* _szNameTable)
+{
+ size_t CurrentIndex = _FirstIndex;
+
+ while (CurrentIndex < _LastIndex)
+ {
+ SFileInfo& rFileInfo = m_FileInfoVector[CurrentIndex];
+ int uOffset = rFileInfo.m_NameOffset & 0xFFFFFF;
+
+ // check next index
+ if (rFileInfo.IsDirectory())
+ {
+ // this is a directory, build up the new szDirectory
+ if (_szDirectory != NULL)
+ {
+ sprintf(rFileInfo.m_FullPath, "%s%s/", _szDirectory, &_szNameTable[uOffset]);
+ }
+ else
+ {
+ sprintf(rFileInfo.m_FullPath, "%s/", &_szNameTable[uOffset]);
+ }
+
+ CurrentIndex = BuildFilenames(CurrentIndex + 1, (size_t) rFileInfo.m_FileSize, rFileInfo.m_FullPath, _szNameTable);
+ }
+ else
+ {
+ // this is a filename
+ if (_szDirectory != NULL)
+ {
+ sprintf(rFileInfo.m_FullPath, "%s%s", _szDirectory, &_szNameTable[uOffset]);
+ }
+ else
+ {
+ sprintf(rFileInfo.m_FullPath, "%s", &_szNameTable[uOffset]);
+ }
+
+ CurrentIndex++;
+ }
+ }
+
+ return(CurrentIndex);
+}
+
+
+const SFileInfo*
+CARCFile::FindFileInfo(std::string _rFullPath) const
+{
+ for (auto& fileInfo : m_FileInfoVector)
+ {
+ if (!strcasecmp(fileInfo.m_FullPath, _rFullPath.c_str()))
+ {
+ return(&fileInfo);
+ }
+ }
+
+ return(NULL);
+}
+} // namespace