Firefly1126移植rtmp推流器到RKMedia源码库中

Firefly1126移植rtmp推流器到RKMedia源码库中#

1，桌面上新建一个librtmp文件夹，在其中放入srs-librtmp.a和srs-librtmp.h两个文件及libssl.a,libcrypto.a两个文件 ，文件分别来自

/home/kxq/Desktop/srs.oschina/trunk/objs/lib

/home/kxq/Desktop/srs.oschina/trunk/objs/include

/home/kxq/Desktop/srs.oschina/trunk/objs/openssl/lib

2,将librtmp文件夹放入RkMedia文件夹下,文件夹位置

/home/kxq/Desktop/rv1126_rv1109_linux_release_20210306/external/rkmedia/examples

3，新建一个rkmedia_vi_venc_rtmp_test.c文件，在其中编入srs-librtmp.a库，并将srs-librtmp.a文件改名为librtmp.a,使用rtmp推流

首先在同一目录下的cmakelist.txt文件中添加c文件的信息如下 ：

Copy
#--------------------------
# rkmedia_vi_venc_rtmp_test
#--------------------------
link_directories("/home/kxq/Desktop/rv1126_rv1109_linux_release_20210306/external/rkmedia/examples/librtmp/")
add_executable(rkmedia_vi_venc_rtmp_test rkmedia_vi_venc_rtmp_test.c ${COMMON_SRC})
add_dependencies(rkmedia_vi_venc_rtmp_test easymedia)
target_link_libraries(rkmedia_vi_venc_rtmp_test easymedia 
rtmp -ldl -lstdc++ -lm 
ssl -ldl -lstdc++ -lm 
crypto -ldl -lstdc++ -lm)
target_include_directories(rkmedia_vi_venc_rtmp_test PRIVATE ${CMAKE_SOURCE_DIR}/include)
install(TARGETS rkmedia_vi_venc_rtmp_test RUNTIME DESTINATION "bin")

4，参考rkmedia_vi_venc_rtmp_test.c和srs_h264_raw_publish.c加入rtmp的推流程序，rkmedia_vi_venc_rtmp_test.c文件中主要做了以下改动

首先在头文件中加入rtmp库引用位置

Copy
#include "librtmp/srs_librtmp.h"

再加入rtmp推流所用全局变量

Copy
static srs_rtmp_t rtmp = NULL;
static int dts,pts = 0;

加入推流代码

Copy
void push_rtmp_stream(srs_rtmp_t rtmp_input, MEDIA_BUFFER mb, int dts_input, int pts_input){

char* h264_raw = (char*)RK_MPI_MB_GetPtr(mb);

if(h264_raw != NULL){
        // send out the h264 packet over RTMP
        int ret = srs_h264_write_raw_frames(rtmp_input, h264_raw, RK_MPI_MB_GetSize(mb), dts_input, pts_input);
        //printf("ret = %d\n",ret);
        if (ret != 0) {
            if (srs_h264_is_dvbsp_error(ret)) {
                srs_human_trace("ignore drop video error, code=%d", ret);
            } else if (srs_h264_is_duplicated_sps_error(ret)) {
                srs_human_trace("ignore duplicated sps, code=%d", ret);
            } else if (srs_h264_is_duplicated_pps_error(ret)) {
                srs_human_trace("ignore duplicated pps, code=%d", ret);
            } else {
                srs_human_trace("send h264 raw data failed. ret=%d", ret);
                srs_rtmp_destroy(rtmp);
                return ;
            }
        }
    }

    usleep(10 * 1000);
    
}

将回调函数中改为

Copy
void video_packet_cb(MEDIA_BUFFER mb) {
  static RK_S32 packet_cnt = 0;
  double fps_rtmp = 25;
  if (quit)
    return;

  printf("#Get packet-%d, size %zu\n", packet_cnt, RK_MPI_MB_GetSize(mb));

  push_rtmp_stream(rtmp, mb, dts, pts);

  RK_MPI_MB_ReleaseBuffer(mb);

  dts += 1000 / fps_rtmp;
  pts = dts;
  packet_cnt++;
}

在Main函数中加入rtmp握手初始化相关代码

Copy
  // init rtsp
  //g_rtsplive = create_rtsp_demo(554);
  //g_rtsp_session = rtsp_new_session(g_rtsplive, "/live/main_stream");
  //init rtmp
  printf("Example for srs-librtmp\n");
  printf("SRS(ossrs) client librtmp library.\n");
  printf("version: %d.%d.%d\n", srs_version_major(), srs_version_minor(), srs_version_revision());
        
  // connect rtmp context
  const char* rtmp_url = "rtmp://公网rtmp服务器IP：端口/live/livestream";
  rtmp = srs_rtmp_create(rtmp_url);
    
  if (srs_rtmp_handshake(rtmp) != 0) {
      srs_human_trace("simple handshake failed.");
      srs_rtmp_destroy(rtmp);
      return 0;
  }
  srs_human_trace("simple handshake success");
    
  if (srs_rtmp_connect_app(rtmp) != 0) {
      srs_human_trace("connect vhost/app failed.");
      srs_rtmp_destroy(rtmp);
      return 0;
  }
  srs_human_trace("connect vhost/app success");
    
  if (srs_rtmp_publish_stream(rtmp) != 0) {
      srs_human_trace("publish stream failed.");
      srs_rtmp_destroy(rtmp);
      return 0;
  }
  srs_human_trace("publish stream success");

在主函数退出时注销rtmp推流服务

Copy
srs_rtmp_destroy(rtmp);

5，编译程序指令如下

Copy
# SDK根目录,选择环境
source envsetup.sh firefly_rv1126_rv1109
# 重编rkmedia源码
make rkmedia-dirclean && make rkmedia
# rkmedia库/程序打包到文件系统(oem.img)
./build.sh rootfs
# 重新烧写oem.img,若有其他基础包配置更新(如ffmpeg),则需要重烧rootfs.img

或者执行完前两步后，在SDK/buildroot/output/firefly_rv1126_rv1109/oem/usr/bin目录下找到rkmedia_vi_venc_rtmp_test二进制可执行程序，移动到板子某一个目录下，再进行测试。

测试代码

./rkmedia_vi_venc_rtmp_test -w 640 -h 480 -d /dev/video25

拉流代码

vlc rtmp://公网rtmp服务器IP：端口/live/livestream

注：不要用vlc获取rtmp流，在window下下载OBS软件点击媒体源，设置中将本地文件勾选去掉，在输入中输入rtmp://公网rtmp服务器IP：端口/live/livestream，之后点击确认开始拉流。