Преглед изворни кода

Merge pull request #718 from lioncash/read

loader/nso: Check if read succeeded in IdentifyFile() before checking magic value
bunnei пре 8 година
родитељ
комит
dd0446ff43
1 измењених фајлова са 6 додато и 4 уклоњено
  1. 6 4
      src/core/loader/nso.cpp

+ 6 - 4
src/core/loader/nso.cpp

@@ -55,13 +55,15 @@ AppLoader_NSO::AppLoader_NSO(FileSys::VirtualFile file) : AppLoader(std::move(fi
 
 FileType AppLoader_NSO::IdentifyType(const FileSys::VirtualFile& file) {
     u32 magic = 0;
-    file->ReadObject(&magic);
+    if (file->ReadObject(&magic) != sizeof(magic)) {
+        return FileType::Error;
+    }
 
-    if (Common::MakeMagic('N', 'S', 'O', '0') == magic) {
-        return FileType::NSO;
+    if (Common::MakeMagic('N', 'S', 'O', '0') != magic) {
+        return FileType::Error;
     }
 
-    return FileType::Error;
+    return FileType::NSO;
 }
 
 static std::vector<u8> DecompressSegment(const std::vector<u8>& compressed_data,