基于librtmp的推流与拉流，x86上的简单示例

Myth丶恋晨 2023-01-17 12:54 60阅读 0赞

先将RTMPDUMP、OPENSSL下载解压并编译到对应目录下，  
再将以下源文件与Makefile放到相对的目录下，进行编译与运行即可。

[RTMPDUMP源码下载：https://download.csdn.net/download/weixin\_43793181/20324389][RTMPDUMP_https_download.csdn.net_download_weixin_43793181_20324389]  
[OPENSSL源码下载：https://download.csdn.net/download/weixin\_43793181/20324368][OPENSSL_https_download.csdn.net_download_weixin_43793181_20324368]  
[推流与拉流源码下载：https://download.csdn.net/download/weixin\_43793181/20324130][https_download.csdn.net_download_weixin_43793181_20324130]

rtmp\_send.c

#include <stdio.h>
    #include <stdlib.h>
    #include <string.h>
    #include <stdint.h>
    #ifndef WIN32
    #include <unistd.h>
    #endif
    
    #include "../rtmpdump-2.3/librtmp/rtmp_sys.h"
    #include "../rtmpdump-2.3/librtmp/log.h"
     
    #define HTON16(x) ((x>>8&0xff)|(x<<8&0xff00))
    #define HTON24(x) ((x>>16&0xff)|(x<<16&0xff0000)|(x&0xff00))
    #define HTON32(x) ((x>>24&0xff)|(x>>8&0xff00)|\ (x<<8&0xff0000)|(x<<24&0xff000000))
    #define HTONTIME(x) ((x>>16&0xff)|(x<<16&0xff0000)|(x&0xff00)|(x&0xff000000))
    
    #define RTMP_SERVER_URL "rtmp://192.168.1.183:1935/live/test0001" //推流地址
    #define LOCAL_FILE "2019_01_04_16_01_40_377579012.flv" //本地文件
    /*read 1 byte*/
    int ReadU8(uint32_t *u8,FILE*fp){ 
             if(fread(u8,1,1,fp)!=1)
                       return 0;
             return 1;
    }
    /*read 2 byte*/
    int ReadU16(uint32_t *u16,FILE*fp){ 
             if(fread(u16,2,1,fp)!=1)
                       return 0;
             *u16=HTON16(*u16);
             return 1;
    }
    /*read 3 byte*/
    int ReadU24(uint32_t *u24,FILE*fp){ 
             if(fread(u24,3,1,fp)!=1)
                       return 0;
             *u24=HTON24(*u24);
             return 1;
    }
    /*read 4 byte*/
    int ReadU32(uint32_t *u32,FILE*fp){ 
             if(fread(u32,4,1,fp)!=1)
                       return 0;
             *u32=HTON32(*u32);
             return 1;
    }
    /*read 1 byte,and loopback 1 byte at once*/
    int PeekU8(uint32_t *u8,FILE*fp){ 
             if(fread(u8,1,1,fp)!=1)
                       return 0;
             fseek(fp,-1,SEEK_CUR);
             return 1;
    }
    /*read 4 byte and convert to time format*/
    int ReadTime(uint32_t *utime,FILE*fp){ 
             if(fread(utime,4,1,fp)!=1)
                       return 0;
             *utime=HTONTIME(*utime);
             return 1;
    }
     
    
    int InitSockets()
    { 
        /* WORD version; WSADATA wsaData; version=MAKEWORD(2,2); return (WSAStartup(version, &wsaData) == 0); */
            return 1;
    }
     
    void CleanupSockets()
    { 
            // WSACleanup();
            return;
    }
     
    //Publish using RTMP_SendPacket()
    int publish_using_packet()
    { 
             RTMP *rtmp=NULL;                           
             RTMPPacket *packet=NULL;
             uint32_t start_time=0;
             uint32_t now_time=0;
             uint32_t add_time_times=0;
             //the timestamp of the previous frame
             long pre_frame_time=0;
             long lasttime=0;
             //int bNextIsKey=1;
             uint32_t preTagsize=0;
            
             //packet attributes
             uint32_t type=0;                       
             uint32_t datalength=0;           
             uint32_t timestamp=0;           
             uint32_t streamid=0;                        
     
             FILE*fp=NULL;
             fp=fopen(LOCAL_FILE,"rb");
             if (!fp)
    		 { 
                       RTMP_LogPrintf("Open File Error.\n");
                       CleanupSockets();
                       return -1;
             }
    		 
             /* set log level */
             //RTMP_LogLevel loglvl=RTMP_LOGDEBUG;
             //RTMP_LogSetLevel(loglvl);
    
             if (!InitSockets())
    		 { 
                       RTMP_LogPrintf("Init Socket Err\n");
                       return -1;
             }
             rtmp=RTMP_Alloc();
             RTMP_Init(rtmp);
             //set connection timeout,default 30s
             rtmp->Link.timeout=5;                      
             if(!RTMP_SetupURL(rtmp,RTMP_SERVER_URL))
             { 
                       RTMP_Log(RTMP_LOGERROR,"SetupURL Err\n");
                       RTMP_Free(rtmp);
                       CleanupSockets();
                       return -1;
             }
            
             //if unable,the AMF command would be 'play' instead of 'publish'
             RTMP_EnableWrite(rtmp);     
    
             if (!RTMP_Connect(rtmp,NULL))
    		 { 
                       RTMP_Log(RTMP_LOGERROR,"Connect Err\n");
                       RTMP_Free(rtmp);
                       CleanupSockets();
                       return -1;
             }
    
             if (!RTMP_ConnectStream(rtmp,10)){ 
                       RTMP_Log(RTMP_LOGERROR,"ConnectStream Err\n");
                       RTMP_Close(rtmp);
                       RTMP_Free(rtmp);
                       CleanupSockets();
                       return -1;
             }
     
             packet=(RTMPPacket*)malloc(sizeof(RTMPPacket));
             RTMPPacket_Alloc(packet,1024*1500);
             RTMPPacket_Reset(packet);
     
             packet->m_hasAbsTimestamp = 0;        
             packet->m_nChannel = 0x04;
             packet->m_nInfoField2 = rtmp->m_stream_id;
            
            while(1) { 
     
             RTMP_LogPrintf("Start to send data ...\n");
             
             fp=fopen(LOCAL_FILE,"rb");
             if (!fp){ 
                       RTMP_LogPrintf("Open File Error.\n");
                       CleanupSockets();
                       return -1;
             }
            
             //jump over FLV Header
             fseek(fp,9,SEEK_SET);
             //jump over previousTagSizen
             fseek(fp,4,SEEK_CUR);   
             
             start_time=RTMP_GetTime();
             lasttime = 0;
             pre_frame_time = 0;
             RTMP_LogPrintf("add_time_times:%8u ms\n",add_time_times);
    
             while(1)
             {      
                       if(((now_time=RTMP_GetTime())-start_time) < (pre_frame_time))
    				   {    
    				   		if(pre_frame_time>lasttime)
    						{ 
                                 RTMP_LogPrintf("TimeStamp:%8lu ms\n",pre_frame_time);
                                 lasttime=pre_frame_time;
                            }
                            usleep(10000); // 10ms
                            continue;
                       }
    				   
                       //not quite the same as FLV spec
                       if(!ReadU8(&type,fp))
                       { 
                            break;
                            printf("line:%d\n",__LINE__);
                       }                       
    
                       if(!ReadU24(&datalength,fp))
                       { 
                            break;
                            printf("line:%d\n",__LINE__);
                       }    
                       if(!ReadTime(&timestamp,fp))
                       { 
                            break;
                            printf("line:%d\n",__LINE__);
                       }    
    				   	
                       if(!ReadU24(&streamid,fp))
                       { 
                            break;
                            printf("line:%d\n",__LINE__);
                       }    
     
                       if (type!=0x08&&type!=0x09)
    				   { 
                            fseek(fp,datalength+4,SEEK_CUR);
                            continue;
                       }
                      
                       if(fread(packet->m_body,1,datalength,fp)!=datalength)
                       { 
                            break;
                            printf("line:%d\n",__LINE__);
                       }    
    
                       packet->m_headerType = RTMP_PACKET_SIZE_LARGE;
                       packet->m_nTimeStamp = timestamp + add_time_times;
                       packet->m_packetType = type;
                       packet->m_nBodySize  = datalength;
                       pre_frame_time=timestamp;
     
                       if (!RTMP_IsConnected(rtmp))
    				   { 
                            RTMP_Log(RTMP_LOGERROR,"rtmp is not connect\n");
                            printf("line:%d\n",__LINE__);
                            break;
                       }
                       if (!RTMP_SendPacket(rtmp,packet,0))
    				   { 
                            RTMP_Log(RTMP_LOGERROR,"Send Error\n");
                            printf("line:%d\n",__LINE__);
                            break;
                       }
     
                       if(!ReadU32(&preTagsize,fp))
                       { 
                            break;
                            printf("line:%d\n",__LINE__);
                       }    
                               
                       if(!PeekU8(&type,fp))
                       { 
                            break;
                            printf("line:%d\n",__LINE__);
                       }    
                       if(type==0x09)
    				   { 
                            if(fseek(fp,11,SEEK_CUR)!=0)
                                     break;
                            if(!PeekU8(&type,fp)){ 
                                     break;
                            }
                            /* if(type==0x17) bNextIsKey=1; else bNextIsKey=0; */
                            fseek(fp,-11,SEEK_CUR);
                       }
             } 
     
             RTMP_LogPrintf("\nSend Data Over\n");
             
             add_time_times += 237080;
            
             if(fp)
                fclose(fp);
            }
                   
    		if (rtmp!=NULL){ 
    		       RTMP_Close(rtmp);        
    		       RTMP_Free(rtmp); 
    		       rtmp=NULL;
    		}
    		if (packet!=NULL){ 
    		       RTMPPacket_Free(packet);    
    		       free(packet);
    		       packet=NULL;
    		}
     
             CleanupSockets();
             return 0;
    }
    	
    int main(int argc, char* argv[])
    { 
             publish_using_packet();
             return 0;
    }

Makefile

#!/bin/sh
    INCLUDE = /opt/work/rtmpdump-2.3/
    LIB_DIR = /opt/work/rtmpdump-2.3/librtmp
    LDFLAGS = -lrtmp
    
    SRC=rtmp_send.c
    
    all:$(SRC)
    	gcc -g -Wall $(SRC) -o send -I $(INCLUDE) -L $(LIB_DIR)  $(LDFLAGS)

rtmp\_recv.c

#include <stdio.h>
    #include <stdlib.h>
    #include <string.h>
    #include "../rtmpdump-2.3/librtmp/rtmp_sys.h"
    #include "../rtmpdump-2.3/librtmp/log.h"
     
    #define SAVE_FILE_NAME "receive.flv"
    #define RTMP_SERVER_URL "rtmp://192.168.1.183:1935/live/test0001" //推流地址
    //#define RTMP_SERVER_URL "rtmp://58.200.131.2:1935/livetv/cctv15" 
    int InitSockets()
    { 
        /* WORD version; WSADATA wsaData; version = MAKEWORD(1, 1); return (WSAStartup(version, &wsaData) == 0); */
        return 0;
    }
     
     
    void CleanupSockets()
    { 
    	//WSACleanup();
    }
     
    int main(int argc, char* argv[])
    { 
    	InitSockets();
    	
    	int nRead;
    	int bLiveStream=1;
    	
    	int bufsize=1024*1024*10;			
    	char *buf=(char*)malloc(bufsize);
    	memset(buf,0,bufsize);
    	long countbufsize=0;
    	
    	FILE *fp=fopen(SAVE_FILE_NAME,"wb");
    	if (!fp)
    	{ 
    		RTMP_LogPrintf("Open File Error.\n");
    		CleanupSockets();
    		return -1;
    	}
    	RTMP *rtmp=RTMP_Alloc();
    	RTMP_Init(rtmp);
    	rtmp->Link.timeout=60;										//设置连接超时时间上限
     
    	if(!RTMP_SetupURL(rtmp,RTMP_SERVER_URL))
    	{ 
    		RTMP_Log(RTMP_LOGERROR,"SetupURL Err\n");
    		RTMP_Free(rtmp);
    		CleanupSockets();
    		return -1;
    	}
    	if (bLiveStream)
    	{ 
    		rtmp->Link.lFlags|=RTMP_LF_LIVE;
    	}
    
    	RTMP_SetBufferMS(rtmp, 3600*1000);							//录制总时长1h
    	
    	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_Close(rtmp);
    		RTMP_Free(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;
    }

Makefile

#!/bin/sh
    INCLUDE = /opt/work/rtmpdump-2.3/
    LIB_DIR = /opt/work/rtmpdump-2.3/librtmp/	
    LDFLAGS = -lrtmp
    
    SRC=rtmp_recv.c
    
    all:$(SRC)
    	gcc -g -Wall $(SRC) -o recv -I $(INCLUDE) -L $(LIB_DIR)  $(LDFLAGS)

[RTMPDUMP_https_download.csdn.net_download_weixin_43793181_20324389]: https://download.csdn.net/download/weixin_43793181/20324389
[OPENSSL_https_download.csdn.net_download_weixin_43793181_20324368]: https://download.csdn.net/download/weixin_43793181/20324368
[https_download.csdn.net_download_weixin_43793181_20324130]: https://download.csdn.net/download/weixin_43793181/20324130