新增读取h264文件转流媒体范例

2026-07-05 10:58:11 +08:00 · 2021-08-02 15:54:51 +08:00
parent 91f370e925
commit f5842f8c1f
4 changed files with 221 additions and 0 deletions
--- a/api/tests/h264_media_server.c
+++ b/api/tests/h264_media_server.c
@@ -0,0 +1,83 @@
+/*
+ * Copyright (c) 2016 The ZLMediaKit project authors. All Rights Reserved.
+ *
+ * This file is part of ZLMediaKit(https://github.com/xia-chu/ZLMediaKit).
+ *
+ * Use of this source code is governed by MIT license that can be found in the
+ * LICENSE file in the root of the source tree. All contributing project authors
+ * may be found in the AUTHORS file in the root of the source tree.
+ */
+
+#ifdef _WIN32
+#include "windows.h"
+#else
+#include "unistd.h"
+#endif
+#include "mk_mediakit.h"
+
+static int exit_flag = 0;
+static void s_on_exit(int sig) {
+    exit_flag = 1;
+}
+
+static void on_h264_frame(void *user_data, mk_h264_splitter splitter, const char *frame, int size) {
+#ifdef _WIN32
+    Sleep(40);
+#else
+    usleep(40 * 1000);
+#endif
+    mk_media media = (mk_media) user_data;
+    mk_media_input_h264(media, frame, size, 0, 0);
+}
+
+int main(int argc, char *argv[]) {
+    mk_config config = {
+            .ini = NULL,
+            .ini_is_path = 1,
+            .log_level = 0,
+            .log_file_path = NULL,
+            .log_file_days = 0,
+            .ssl = NULL,
+            .ssl_is_path = 1,
+            .ssl_pwd = NULL,
+            .thread_num = 0
+    };
+    mk_env_init(&config);
+    mk_http_server_start(80, 0);
+    mk_rtsp_server_start(554, 0);
+    mk_rtmp_server_start(1935, 0);
+
+    signal(SIGINT, s_on_exit);// 设置退出信号
+
+    FILE *fp = fopen(argv[1], "rb");
+    if (!fp) {
+        log_error("打开文件失败!");
+        return -1;
+    }
+
+    mk_media media = mk_media_create("__defaultVhost__", "live", "test", 0, 0, 0);
+    //h264的codec
+    mk_media_init_video(media, 0, 0, 0, 0);
+    mk_media_init_complete(media);
+
+    //创建h264分帧器
+    mk_h264_splitter splitter = mk_h264_splitter_create(on_h264_frame, media);
+
+    char buf[1024];
+    while (!exit_flag) {
+        int size = fread(buf, 1, sizeof(buf) - 1, fp);
+        if (size > 0) {
+            mk_h264_splitter_input_data(splitter, buf, size);
+        } else {
+            //文件读完了，重新开始
+            fseek(fp, 0, SEEK_SET);
+        }
+    }
+
+    log_info("文件读取完毕");
+    mk_h264_splitter_release(splitter);
+    mk_media_release(media);
+    fclose(fp);
+    mk_stop_all_server();
+    return 0;
+}