(2013-06-26 14:32:10)
[置頂] 使用librtmp發(fā)布flash視頻直播流（H264、AAC）的C++類(lèi)代碼分類(lèi)：
音視頻編碼 2013-03-11 20:09 353人閱讀
評論(5)
舉報rtmpflashc++h264aac最近在研究使用開(kāi)源librtmp庫發(fā)布flash視頻直播流，經(jīng)過(guò)幾天的折騰后，寫(xiě)了一個(gè)封裝調用librtmp的類(lèi)，時(shí)間緊迫沒(méi)好好整理，僅供需要的朋友參考
頭文件：RtmpStream.h
[cpp]
view plaincopy#pragma once
/// 對開(kāi)源庫rtmpdump的封裝
#include
#include "common.h"
struct RTMP;
struct RTMPMetadata;
struct RTMPPacket;
class CRtmpStream
{
public:
CRtmpStream(void);
virtual ~CRtmpStream(void);
public:
/// 打開(kāi)
BOOL Open(const char * apUrl);
/// 重新打開(kāi)最近的連接
BOOL Reopen();
/// 關(guān)閉
void Close();
/// 是否打開(kāi)
inline BOOL IsOpened() const
{
return m_pHandle != NULL;
}
/// 發(fā)送元數據
BOOL SendMetadata(const RTMPMetadata & astruMetadata);
/// 發(fā)送AAC數據
BOOL SendAACData(const void * apBuf, const unsigned int auBufLen, const unsigned int auTimestamp);
/// 發(fā)送H264數據
BOOL SendH264Data(const void * apBuf, const unsigned int auBufLen, const unsigned int auTimestamp,
const BOOL abIsKey, const int aiOffset);
private:
void CloseStream();
private:
RTMP * m_pHandle;
unsigned int m_uStartAudioTime;
unsigned int m_uStartVideoTime;
std::string m_strLastUrl;
CCriticalSectionEx m_oLock;
BOOL m_bIsLiving;
};
實(shí)現：RtmpStream.cpp
[cpp]
view plaincopy#include "stdafx.h"
#include "RtmpStream.h"
#include "rtmp/rtmp.h"
#include "rtmp/flvpush.h"
#include
#include
static class CInitSocket
{
public:
CInitSocket()
{
#ifdef WIN32
WSADATA wsa_data;
WORD w_version_requested = MAKEWORD(2, 2);
if ( WSAStartup(w_version_requested, &wsa_data) != 0 )
{
m_bResult = false;
return;
}
//else if ( LOBYTE(wsa_data.wVersion) != 2 || HIBYTE(wsa_data.wVersion) != 2 )
//{
// WSACleanup();
// return false;
//}
#endif
m_bResult = true;
}
~CInitSocket()
{
#ifdef WIN32
if ( m_bResult )
{
WSACleanup();
}
#endif
}
bool m_bResult;
} s_oInitSocket;
CRtmpStream::CRtmpStream(void)
: m_pHandle(NULL)
, m_bIsLiving(FALSE)
, m_uStartAudioTime(0)
, m_uStartVideoTime(0)
{
}
CRtmpStream::~CRtmpStream(void)
{
Close();
}
BOOL CRtmpStream::Open(const char * apUrl)
{
CCriticalSectionHelperEx oDoLock(m_oLock);
if ( m_bIsLiving )
{
return FALSE;
}
if ( IsOpened() )
{
return FALSE;
}
m_pHandle = RTMP_Alloc();
if ( m_pHandle == NULL )
{
assert(FALSE);
return FALSE;
}
RTMP_Init(m_pHandle);
if ( !RTMP_SetupURL(m_pHandle, apUrl) )
{
Close();
assert(FALSE);
return FALSE;
}
RTMP_EnableWrite(m_pHandle);
if ( !RTMP_Connect(m_pHandle, NULL) )
{
Close();
assert(FALSE);
return FALSE;
}
if ( !RTMP_ConnectStream(m_pHandle, 0) )
{
Close();
assert(FALSE);
return FALSE;
}
m_bIsLiving = TRUE;
m_uStartAudioTime = 0;
m_uStartVideoTime = 0;
if ( m_strLastUrl.c_str() != apUrl )
{
m_strLastUrl = apUrl;
}
return TRUE;
}
/// 重新打開(kāi)最近的連接
BOOL CRtmpStream::Reopen()
{
return Open(m_strLastUrl.c_str());
}
void CRtmpStream::CloseStream()
{
if ( m_pHandle != NULL )
{
RTMP_Close(m_pHandle);
::Sleep(20);
RTMP_Free(m_pHandle);
m_pHandle = NULL;
}
}
void CRtmpStream::Close()
{
CCriticalSectionHelperEx oDoLock(m_oLock);
CloseStream();
m_bIsLiving = FALSE;
}
/// 發(fā)送元數據
BOOL CRtmpStream::SendMetadata(const RTMPMetadata & astruMetadata)
{
if ( !m_bIsLiving )
{
return FALSE;
}
CCriticalSectionHelperEx oDoLock(m_oLock);
if ( !IsOpened() )
{
assert(FALSE);
return FALSE;
}
if ( !RTMP_SendMetadata(m_pHandle, &astruMetadata) )
{
assert(FALSE);
return FALSE;
}
return TRUE;
}
/// 發(fā)送AAC數據
BOOL CRtmpStream::SendAACData(const void * apBuf, const unsigned int auBufLen, const unsigned int auTimestamp)
{
if ( !m_bIsLiving )
{
return FALSE;
}
if ( auBufLen == 0 )
{
return TRUE;
}
CCriticalSectionHelperEx oDoLock(m_oLock);
if ( !IsOpened() )
{
return TRUE;
}
unsigned int uCurTimestamp;
if ( m_uStartAudioTime )
{
// 有值
uCurTimestamp = auTimestamp - m_uStartAudioTime;
}
else
{
// is 0
uCurTimestamp = 1;
m_uStartAudioTime = auTimestamp;
}
if ( !RTMP_SendAACData(m_pHandle, apBuf, auBufLen, uCurTimestamp) )
{
Close();
Reopen();
assert(FALSE);
return FALSE;
}
return TRUE;
}
/// 發(fā)送H264數據
BOOL CRtmpStream::SendH264Data(const void * apBuf, const unsigned int auBufLen, const unsigned int auTimestamp,
const BOOL abIsKey, const int aiOffset)
{
if ( !m_bIsLiving )
{
return FALSE;
}
if ( auBufLen == 0 )
{
return TRUE;
}
CCriticalSectionHelperEx oDoLock(m_oLock);
if ( !IsOpened() )
{
return TRUE;
}
unsigned int uCurTimestamp;
if ( m_uStartVideoTime )
{
// 有值
uCurTimestamp = auTimestamp - m_uStartVideoTime;
}
else
{
// is 0
uCurTimestamp = 1;
m_uStartVideoTime = auTimestamp;
}
if ( !RTMP_SendH264Data(m_pHandle, apBuf, auBufLen, uCurTimestamp, abIsKey, aiOffset) )
{
Close();
Reopen();
assert(FALSE);
return FALSE;
}
return TRUE;
}
調用順序：
1、打開(kāi)Open
2、發(fā)送元數據SendMetadata
3、發(fā)送音頻AAC數據SendAACData或視頻H264數據SendH264Data
4、Close我的資源里有編譯好的靜態(tài)庫和代碼（使用mingw gcc編譯，VC里也可直接使用）
