check that uploads are WebM files

2 files changed, 2 insertions, 0 deletions

diff --git a/posthandler.php b/posthandler.php
index f2f60291..639254e1 100644
--- a/posthandler.php
+++ b/posthandler.php
@@ -6,6 +6,7 @@ function postHandler($post) {
     if ($post->has_file && $post->extension == 'webm') {
         require_once dirname(__FILE__) . '/videodata.php';
         $videoDetails = videoData($post->file_path);
+        if (!isset($videoDetails['container']) || $videoDetails['container'] != 'webm') return "not a WebM file";
 
         // Set thumbnail
         $thumbName = $board['dir'] . $config['dir']['thumb'] . $post->file_id . '.webm';
diff --git a/videodata.php b/videodata.php
index 1b1e7eb6..9bf26396 100644
--- a/videodata.php
+++ b/videodata.php
@@ -107,6 +107,7 @@ function videoData($filename) {
 
     try {
         $root = readMatroska($fileHandle);
+        $data['container'] = $root->get('EBML')->get('DocType');
 
         // Locate segment information and tracks
         $segment = $root->get('Segment');