update

1 files changed, 7 insertions, 4 deletions

diff --git a/src/libjin/audio/source.cpp b/src/libjin/audio/source.cpp
index ec80a0f..a09bbab 100644
--- a/src/libjin/audio/source.cpp
+++ b/src/libjin/audio/source.cpp
@@ -6,14 +6,17 @@ namespace jin
 namespace audio
 {
 
+    static int check_header(const void *data, int size, char *str, int offset) {
+        int len = strlen(str);
+        return (size >= offset + len) && !memcmp((char*)data + offset, str, len);
+    }
+
     SourceType Source::getType(const void* mem, int size)
     {
-        const char* p = (const char* )mem;
-        if (memcmp(p, "RIFF", 4) == 0 && memcmp(p + 8, "WAVE", 4) == 0)
+        if(check_header(mem, size, "WAVE", 8))
             return SourceType::WAV;
-        if (memcmp(p, "OggS", 4) == 0)
+        if(check_header(mem, size, "OggS", 0))
             return SourceType::OGG;
-
         return SourceType::INVALID;
     }