From b2c7bb0b283dd2a80f345e26c042d6ffaf05209c Mon Sep 17 00:00:00 2001
From: chai <chaifix@163.com>
Date: Sun, 22 Jul 2018 12:28:03 +0800
Subject: update

---
 src/libjin/audio/source.cpp | 11 +++++++----
 1 file changed, 7 insertions(+), 4 deletions(-)

(limited to 'src/libjin/audio/source.cpp')

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;
     }
 
-- 
cgit v1.1-26-g67d0