最简单的基于librtmp的示例：接收（RTMP保存为FLV）

标签：librtmp   rtmp   flv   流媒体   

流程图

源代码

/**
 * Simplest Librtmp Receive
 *
 * 雷霄骅，张晖
 * leixiaohua1020@126.com
 * zhanghuicuc@gmail.com
 * 中国传媒大学/数字电视技术
 * Communication University of China / Digital TV Technology
 * http://blog.csdn.net/leixiaohua1020
 *
 * 本程序用于接收RTMP流媒体并在本地保存成FLV格式的文件。
 * This program can receive rtmp live stream and save it as local flv file.
 */
#include <stdio.h>
#include "librtmp/rtmp_sys.h"
#include "librtmp/log.h"
 
int InitSockets()
{
         WORD version;
         WSADATA wsaData;
         version = MAKEWORD(1, 1);
         return (WSAStartup(version, &wsaData) == 0);
}
 
void CleanupSockets()
{
         WSACleanup();
}
 
int main(int argc, char* argv[])
{
         InitSockets();
        
         double duration=-1;
         int nRead;
         //is live stream ?
         bool bLiveStream=true;                              
        
        
         int bufsize=1024*1024*10;                       
         char *buf=(char*)malloc(bufsize);
         memset(buf,0,bufsize);
         long countbufsize=0;
        
         FILE *fp=fopen("receive.flv","wb");
         if (!fp){
                   RTMP_LogPrintf("Open File Error.\n");
                   CleanupSockets();
                   return -1;
         }
        
         /* set log level */
         //RTMP_LogLevel loglvl=RTMP_LOGDEBUG;
         //RTMP_LogSetLevel(loglvl);
 
         RTMP *rtmp=RTMP_Alloc();
         RTMP_Init(rtmp);
         //set connection timeout,default 30s
         rtmp->Link.timeout=10; 
         // HKS‘s live URL
         if(!RTMP_SetupURL(rtmp,"rtmp://live.hkstv.hk.lxdns.com/live/hks"))
         {
                   RTMP_Log(RTMP_LOGERROR,"SetupURL Err\n");
                   RTMP_Free(rtmp);
                   CleanupSockets();
                   return -1;
         }
         if (bLiveStream){
                   rtmp->Link.lFlags|=RTMP_LF_LIVE;
         }
        
         //1hour
         RTMP_SetBufferMS(rtmp, 3600*1000);         
        
         if(!RTMP_Connect(rtmp,NULL)){
                   RTMP_Log(RTMP_LOGERROR,"Connect Err\n");
                   RTMP_Free(rtmp);
                   CleanupSockets();
                   return -1;
         }
 
         if(!RTMP_ConnectStream(rtmp,0)){
                   RTMP_Log(RTMP_LOGERROR,"ConnectStream Err\n");
                   RTMP_Free(rtmp);
                   RTMP_Close(rtmp);
                   CleanupSockets();
                   return -1;
         }
 
         while(nRead=RTMP_Read(rtmp,buf,bufsize)){
                   fwrite(buf,1,nRead,fp);
 
                   countbufsize+=nRead;
                   RTMP_LogPrintf("Receive: %5dByte, Total: %5.2fkB\n",nRead,countbufsize*1.0/1024);
         }
 
         if(fp)
                   fclose(fp);
 
         if(buf){
                   free(buf);
         }
 
         if(rtmp){
                   RTMP_Close(rtmp);
                   RTMP_Free(rtmp);
                   CleanupSockets();
                   rtmp=NULL;
         }       
         return 0;
}

下载

最简单的基于librtmp的示例：接收（RTMP保存为FLV）

标签：librtmp   rtmp   flv   流媒体   

原文地址：http://blog.csdn.net/leixiaohua1020/article/details/42104893