重写MP4点播

src/Record/MP4.cpp

@@ -0,0 +1,140 @@
+/*
+ * MIT License
+ *
+ * Copyright (c) 2016-2020 xiongziliang <771730766@qq.com>
+ *
+ * This file is part of ZLMediaKit(https://github.com/xiongziliang/ZLMediaKit).
+ *
+ * Permission is hereby granted, free of charge, to any person obtaining a copy
+ * of this software and associated documentation files (the "Software"), to deal
+ * in the Software without restriction, including without limitation the rights
+ * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+ * copies of the Software, and to permit persons to whom the Software is
+ * furnished to do so, subject to the following conditions:
+ *
+ * The above copyright notice and this permission notice shall be included in all
+ * copies or substantial portions of the Software.
+ *
+ * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+ * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+ * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+ * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+ * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+ * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+ * SOFTWARE.
+ */
+
+#ifdef ENABLE_MP4
+#include "MP4.h"
+#include "Util/File.h"
+#include "Util/logger.h"
+#include "Common/config.h"
+using namespace toolkit;
+namespace mediakit {
+
+static struct mov_buffer_t s_io = {
+        [](void* ctx, void* data, uint64_t bytes) {
+            MP4File *thiz = (MP4File *)ctx;
+            return thiz->onRead(data,bytes);
+        },
+        [](void* ctx, const void* data, uint64_t bytes){
+            MP4File *thiz = (MP4File *)ctx;
+            return thiz->onWrite(data,bytes);
+        },
+        [](void* ctx, uint64_t offset) {
+            MP4File *thiz = (MP4File *)ctx;
+            return thiz->onSeek(offset);
+        },
+        [](void* ctx){
+            MP4File *thiz = (MP4File *)ctx;
+            return thiz->onTell();
+        }
+};
+
+MP4File::Writer MP4File::createWriter(){
+    GET_CONFIG(bool, mp4FastStart, Record::kFastStart);
+    Writer writer;
+    writer.reset(mov_writer_create(&s_io,this,mp4FastStart ? MOV_FLAG_FASTSTART : 0),[](mov_writer_t *ptr){
+        if(ptr){
+            mov_writer_destroy(ptr);
+        }
+    });
+    if(!writer){
+        throw std::runtime_error("写入mp4文件失败!");
+    }
+    return writer;
+}
+
+MP4File::Reader MP4File::createReader(){
+    Reader reader;
+    reader.reset(mov_reader_create(&s_io,this),[](mov_reader_t *ptr){
+        if(ptr){
+            mov_reader_destroy(ptr);
+        }
+    });
+    if(!reader){
+        throw std::runtime_error("读取mp4文件失败!");
+    }
+    return reader;
+}
+
+#if defined(_WIN32) || defined(_WIN64)
+    #define fseek64 _fseeki64
+#define ftell64 _ftelli64
+#else
+#define fseek64 fseek
+#define ftell64 ftell
+#endif
+
+void MP4File::openFile(const char *file,const char *mode) {
+    //创建文件
+    auto fp = File::createfile_file(file,mode);
+    if(!fp){
+        throw std::runtime_error(string("打开文件失败:") + file);
+    }
+
+    GET_CONFIG(uint32_t,mp4BufSize,Record::kFileBufSize);
+
+    //新建文件io缓存
+    std::shared_ptr<char> file_buf(new char[mp4BufSize],[](char *ptr){
+        if(ptr){
+            delete [] ptr;
+        }
+    });
+
+    if(file_buf){
+        //设置文件io缓存
+        setvbuf(fp, file_buf.get(), _IOFBF, mp4BufSize);
+    }
+
+    //创建智能指针
+    _file.reset(fp,[file_buf](FILE *fp) {
+        fclose(fp);
+    });
+}
+
+void MP4File::closeFile() {
+    _file = nullptr;
+}
+
+int MP4File::onRead(void *data, uint64_t bytes) {
+    if (bytes == fread(data, 1, bytes, _file.get())){
+        return 0;
+    }
+    return 0 != ferror(_file.get()) ? ferror(_file.get()) : -1 /*EOF*/;
+}
+
+int MP4File::onWrite(const void *data, uint64_t bytes) {
+    return bytes == fwrite(data, 1, bytes, _file.get()) ? 0 : ferror(_file.get());
+}
+
+int MP4File::onSeek(uint64_t offset) {
+    return fseek64(_file.get(), offset, SEEK_SET);
+}
+
+uint64_t MP4File::onTell() {
+    return ftell64(_file.get());
+}
+
+}//namespace mediakit
+#endif //NABLE_MP4RECORD