added ELF loading support to loader.cpp

1 files changed, 113 insertions, 69 deletions

diff --git a/src/core/src/loader.cpp b/src/core/src/loader.cpp
index 994ce85e9..7e7eb9161 100644
--- a/src/core/src/loader.cpp
+++ b/src/core/src/loader.cpp
@@ -26,35 +26,67 @@
 #include "loader.h"
 #include "system.h"
 #include "file_sys/directory_file_system.h"
+#include "elf/elf_reader.h"
 
 ////////////////////////////////////////////////////////////////////////////////////////////////////
 
 /// Loads an extracted CXI from a directory
 bool LoadDirectory_CXI(std::string &filename) {
-        std::string full_path = filename;
-        std::string path, file, extension;
-        SplitPath(ReplaceAll(full_path, "\\", "/"), &path, &file, &extension);
+    std::string full_path = filename;
+    std::string path, file, extension;
+    SplitPath(ReplaceAll(full_path, "\\", "/"), &path, &file, &extension);
 #if EMU_PLATFORM == PLATFORM_WINDOWS
-        path = ReplaceAll(path, "/", "\\");
+    path = ReplaceAll(path, "/", "\\");
 #endif
-        DirectoryFileSystem *fs = new DirectoryFileSystem(&System::g_ctr_file_system, path);
-        System::g_ctr_file_system.Mount("fs:", fs);
+    DirectoryFileSystem *fs = new DirectoryFileSystem(&System::g_ctr_file_system, path);
+    System::g_ctr_file_system.Mount("fs:", fs);
 
-        std::string final_name = "fs:/" + file + extension;
-        File::IOFile f(filename, "rb");
+    std::string final_name = "fs:/" + file + extension;
+    File::IOFile f(filename, "rb");
 
-        if (f.IsOpen()) {
-                // TODO(ShizZy): read here to memory....
-        }
-        ERROR_LOG(TIME, "Unimplemented function!");
-        return true;
+    if (f.IsOpen()) {
+        // TODO(ShizZy): read here to memory....
+    }
+    ERROR_LOG(TIME, "Unimplemented function!");
+    return true;
+}
+
+/// Loads a CTR ELF file
+bool Load_ELF(std::string &filename) {
+    std::string full_path = filename;
+    std::string path, file, extension;
+    SplitPath(ReplaceAll(full_path, "\\", "/"), &path, &file, &extension);
+#if EMU_PLATFORM == PLATFORM_WINDOWS
+    path = ReplaceAll(path, "/", "\\");
+#endif
+    File::IOFile f(filename, "rb");
+
+    if (f.IsOpen()) {
+        u64 size = f.GetSize();
+        u8* buffer = new u8[size];
+        ElfReader* elf_reader = NULL;
+
+        f.ReadBytes(buffer, size);
+
+        elf_reader = new ElfReader(buffer);
+        elf_reader->LoadInto(0x00100000);
+
+        delete[] buffer;
+        delete elf_reader;
+    }
+    else {
+        return false;
+    }
+    f.Close();
+
+    return true;
 }
 
 namespace Loader {
 
 bool IsBootableDirectory() {
-        ERROR_LOG(TIME, "Unimplemented function!");
-        return true;
+    ERROR_LOG(TIME, "Unimplemented function!");
+    return true;
 }
 
 /**
@@ -64,28 +96,36 @@ bool IsBootableDirectory() {
  * @return FileType of file
  */
 FileType IdentifyFile(std::string &filename) {
-        if (filename.size() == 0) {
-                ERROR_LOG(LOADER, "invalid filename %s", filename.c_str());
-                return FILETYPE_ERROR;
-        }
-        std::string extension = filename.size() >= 5 ? filename.substr(filename.size() - 4) : "";
-        
-        if (File::IsDirectory(filename)) {
-                if (IsBootableDirectory()) {
-                        return FILETYPE_DIRECTORY_CXI;
-                } else {
-                        return FILETYPE_NORMAL_DIRECTORY;
-                }
-        } else if (!strcasecmp(extension.c_str(),".zip")) {
-                return FILETYPE_ARCHIVE_ZIP;
-        } else if (!strcasecmp(extension.c_str(),".rar")) {
-                return FILETYPE_ARCHIVE_RAR;
-        } else if (!strcasecmp(extension.c_str(),".r00")) {
-                return FILETYPE_ARCHIVE_RAR;
-        } else if (!strcasecmp(extension.c_str(),".r01")) {
-                return FILETYPE_ARCHIVE_RAR;
-        }
-        return FILETYPE_UNKNOWN;
+    if (filename.size() == 0) {
+        ERROR_LOG(LOADER, "invalid filename %s", filename.c_str());
+        return FILETYPE_ERROR;
+    }
+    std::string extension = filename.size() >= 5 ? filename.substr(filename.size() - 4) : "";
+
+    if (File::IsDirectory(filename)) {
+        if (IsBootableDirectory()) {
+            return FILETYPE_DIRECTORY_CXI;
+        }
+        else {
+            return FILETYPE_NORMAL_DIRECTORY;
+        }
+    }
+    else if (!strcasecmp(extension.c_str(), ".elf")) {
+        return FILETYPE_CTR_ELF; // TODO(bunnei): Do some filetype checking :p
+    }
+    else if (!strcasecmp(extension.c_str(), ".zip")) {
+        return FILETYPE_ARCHIVE_ZIP;
+    }
+    else if (!strcasecmp(extension.c_str(), ".rar")) {
+        return FILETYPE_ARCHIVE_RAR;
+    }
+    else if (!strcasecmp(extension.c_str(), ".r00")) {
+        return FILETYPE_ARCHIVE_RAR;
+    }
+    else if (!strcasecmp(extension.c_str(), ".r01")) {
+        return FILETYPE_ARCHIVE_RAR;
+    }
+    return FILETYPE_UNKNOWN;
 }
 
 /**
@@ -95,48 +135,52 @@ FileType IdentifyFile(std::string &filename) {
  * @return True on success, otherwise false
  */
 bool LoadFile(std::string &filename, std::string *error_string) {
-        INFO_LOG(LOADER,"Identifying file...");
-        // Note that this can modify filename!
-        switch (IdentifyFile(filename)) {
+    INFO_LOG(LOADER, "Identifying file...");
+
+    // Note that this can modify filename!
+    switch (IdentifyFile(filename)) {
+
+    case FILETYPE_CTR_ELF:
+        return Load_ELF(filename);
 
-        case FILETYPE_DIRECTORY_CXI:
-                return LoadDirectory_CXI(filename);
+    case FILETYPE_DIRECTORY_CXI:
+        return LoadDirectory_CXI(filename);
 
-        case FILETYPE_ERROR:
-                ERROR_LOG(LOADER, "Could not read file");
-                *error_string = "Error reading file";
-                break;
+    case FILETYPE_ERROR:
+        ERROR_LOG(LOADER, "Could not read file");
+        *error_string = "Error reading file";
+        break;
 
-        case FILETYPE_ARCHIVE_RAR:
+    case FILETYPE_ARCHIVE_RAR:
 #ifdef WIN32
-                *error_string = "RAR file detected (Require WINRAR)";
+        *error_string = "RAR file detected (Require WINRAR)";
 #else
-                *error_string = "RAR file detected (Require UnRAR)";
+        *error_string = "RAR file detected (Require UnRAR)";
 #endif
-                break;
+        break;
 
-        case FILETYPE_ARCHIVE_ZIP:
+    case FILETYPE_ARCHIVE_ZIP:
 #ifdef WIN32
-                *error_string = "ZIP file detected (Require WINRAR)";
+        *error_string = "ZIP file detected (Require WINRAR)";
 #else
-                *error_string = "ZIP file detected (Require UnRAR)";
+        *error_string = "ZIP file detected (Require UnRAR)";
 #endif
-                break;
-
-        case FILETYPE_NORMAL_DIRECTORY:
-                ERROR_LOG(LOADER, "Just a directory.");
-                *error_string = "Just a directory.";
-                break;
-
-        case FILETYPE_UNKNOWN_BIN:
-        case FILETYPE_UNKNOWN_ELF:
-        case FILETYPE_UNKNOWN:
-        default:
-                ERROR_LOG(LOADER, "Failed to identify file");
-                *error_string = "Failed to identify file";
-                break;
-        }
-        return false;
+        break;
+
+    case FILETYPE_NORMAL_DIRECTORY:
+        ERROR_LOG(LOADER, "Just a directory.");
+        *error_string = "Just a directory.";
+        break;
+
+    case FILETYPE_UNKNOWN_BIN:
+    case FILETYPE_UNKNOWN_ELF:
+    case FILETYPE_UNKNOWN:
+    default:
+        ERROR_LOG(LOADER, "Failed to identify file");
+        *error_string = "Failed to identify file";
+        break;
+    }
+    return false;
 }
 
 } // namespace
\ No newline at end of file