From 322c043a513a72e7e0dae4936d0103fdd5dd795e Mon Sep 17 00:00:00 2001 From: jvyden Date: Sat, 13 Nov 2021 06:24:17 -0500 Subject: [PATCH] Add PNG file type --- ProjectLighthouse/Helpers/FileHelper.cs | 4 +++- ProjectLighthouse/Types/Files/LbpFileType.cs | 3 ++- 2 files changed, 5 insertions(+), 2 deletions(-) diff --git a/ProjectLighthouse/Helpers/FileHelper.cs b/ProjectLighthouse/Helpers/FileHelper.cs index 62d9f06b..f519effa 100644 --- a/ProjectLighthouse/Helpers/FileHelper.cs +++ b/ProjectLighthouse/Helpers/FileHelper.cs @@ -27,6 +27,7 @@ namespace LBPUnion.ProjectLighthouse.Helpers LbpFileType.Voice => true, LbpFileType.Plan => true, LbpFileType.Jpeg => true, + LbpFileType.Png => true, #if DEBUG _ => throw new ArgumentOutOfRangeException(nameof(file), $"Unhandled file type ({file.FileType}) in FileHelper.IsFileSafe()"), #else @@ -66,10 +67,11 @@ namespace LBPUnion.ProjectLighthouse.Helpers { reader.BaseStream.Position = 0; - // Determine if file is JPEG + // Determine if file is JPEG/PNG byte[] header = reader.ReadBytes(9); if (header[0] == 0xFF && header[1] == 0xD8 && header[2] == 0xFF && header[3] == 0xE0) return LbpFileType.Jpeg; + if (header[0] == 0x89 && header[1] == 0x50 && header[2] == 0x4E && header[3] == 0x47) return LbpFileType.Png; return LbpFileType.Unknown; // Still unknown. } diff --git a/ProjectLighthouse/Types/Files/LbpFileType.cs b/ProjectLighthouse/Types/Files/LbpFileType.cs index abf4de65..69ba34e7 100644 --- a/ProjectLighthouse/Types/Files/LbpFileType.cs +++ b/ProjectLighthouse/Types/Files/LbpFileType.cs @@ -9,7 +9,8 @@ namespace LBPUnion.ProjectLighthouse.Types.Files Plan, // PLN, uploaded with levels Voice, // VOP, voice data Painting, // PTG, paintings - Jpeg, // JFIF / FIF, used in sticker switches + Jpeg, // JFIF / FIF, used in sticker switches, + Png, // used in LBP Vita Unknown, } } \ No newline at end of file