H264通过RTMP发布

// demo.cpp : 定义控制台应用程序的入口点。 // #include "stdafx.h" #include <MqOaI.h> extern "C" { #include "../../common/common.h" #include "../../common/cpu.h"" #include "../../x264.h" #include "../../encoder/set.h" } #include "librtmp/rtmp_sys.h" #include "librtmp/log.h" #include "librtmp/amf.h" #include "CameraDS.h" void ConvertYCbCr2BGR(unsigned char *pYUV,unsigned char *pBGR,int iWidth,int iHeight); void ConvertRGB2YUV(int w,int h,unsigned char *bmp,unsigned char *yuv); int InitSockets() { #ifdef WIN32 WORD version; WSADATA wsaData; version = MAKEWORD(1, 1); return (WSAStartup(version, &wsaData) == 0); #else return TRUE; #endif } inline void CleanupSockets() { #ifdef WIN32 WSACleanup(); #endif } #define HEX2BIN(a) (((a)&0x40)?((a)&0xf)+9:((a)&0xf)) int hex2bin(char *str, char **hex) { char *ptr; int i, l = strlen(str); if (l & 1) return 0; *hex = (char *)malloc(l/2); ptr = *hex; if (!ptr) return 0; for (i=0; i<l; i+=2) *ptr++ = (HEX2BIN(str[i]) << 4) | HEX2BIN(str[i+1]); return l/2; } char * put_byte( char *output, uint8_t nVal ) { output[0] = nVal; return output+1; } char * put_be16(char *output, uint16_t nVal ) { output[1] = nVal & 0xff; output[0] = nVal >> 8; return output+2; } char * put_be24(char *output,uint32_t nVal ) { output[2] = nVal & 0xff; output[1] = nVal >> 8; output[0] = nVal >> 16; return output+3; } char * put_be32(char *output, uint32_t nVal ) { output[3] = nVal & 0xff; output[2] = nVal >> 8; output[1] = nVal >> 16; output[0] = nVal >> 24; return output+4; } char * put_be64( char *output, uint64_t nVal ) { output=put_be32( output, nVal >> 32 ); output=put_be32( output, nVal ); return output; } char * put_amf_string( char *c, const char *str ) { uint16_t len = strlen( str ); c=put_be16( c, len ); memcpy(c,str,len); return c+len; } char * put_amf_double( char *c, double d ) { *c++ = AMF_NUMBER; /* type: Number */ { unsigned char *ci, *co; ci = (unsigned char *)&d; co = (unsigned char *)c; co[0] = ci[7]; co[1] = ci[6]; co[2] = ci[5]; co[3] = ci[4]; co[4] = ci[3]; co[5] = ci[2]; co[6] = ci[1]; co[7] = ci[0]; } return c+8; } int main(int argc, char * argv[]) { if (argc<2) { RTMP_LogPrintf("RTMP_URL IS NULL!!!/n"); //return -1; } if (!InitSockets()) { RTMP_LogPrintf("InitSockets Error!/n"); return -1; } RTMP_LogPrintf("InitSockets!/n"); CoInitialize(NULL); CCameraDS camera; if (!camera.OpenCamera(0,320,240,false)) { RTMP_LogPrintf("Open Camera Error/n"); return -1; } int nHeight=camera.GetHeight(); int nWidth=camera.GetWidth(); unsigned char * szRGBBuffer=new unsigned char[nHeight*nWidth * 3]; RTMP_LogPrintf("Camera Open Scuess！ Picture Size[%2dx%d]/n",nWidth,nHeight); RTMP_debuglevel = RTMP_LOGINFO; RTMP *r; //char uri[]="rtmp://127.0.0.1/live/test"; //char uri[]="rtmp://192.199.15.223/live/test"; //char uri[]="rtmp://221.9.244.4/live/jltv"; //char uri[]="rtmp://192.199.15.223/oflaDemo/red5StreamDemo"; //char uri[]="rtmp://192.199.15.151/live/test"; char uri[]="rtmp://127.0.0.1/live/zzj"; r= RTMP_Alloc(); RTMP_Init(r); RTMP_SetupURL(r, (char*)uri); RTMP_EnableWrite(r); RTMP_Connect(r, NULL); RTMP_ConnectStream(r,0); unsigned char szNalBuffer[1024*32]; unsigned char szBodyBuffer[1024*32]; x264_nal_t *p264Nal; int i264Nal; x264_param_t * p264Param; x264_picture_t * p264Pic; x264_t *p264Handle; p264Param = new x264_param_t(); p264Pic = new x264_picture_t(); memset(p264Pic,0,sizeof(x264_picture_t)); x264_param_default(p264Param); //set default param p264Param->i_threads=2; p264Param->i_width = nWidth; //set frame width p264Param->i_height = nHeight; //set frame height /*baseline level 1.1*/ p264Param->b_cabac =0; p264Param->i_bframe =0; p264Param->b_interlaced=0; p264Param->rc.i_rc_method=X264_RC_ABR;//X264_RC_CQP p264Param->i_level_idc=21; p264Param->rc.i_bitrate=200; p264Param->i_fps_num=30; p264Param->i_keyint_max=p264Param->i_fps_num*3; if((p264Handle = x264_encoder_open(p264Param)) == NULL) { fprintf( stderr, "x264_encoder_open failed/n" ); return -2; } bs_t bs={0}; x264_picture_alloc(p264Pic, X264_CSP_YV12, p264Param->i_width, p264Param->i_height); p264Pic->i_type = X264_TYPE_AUTO; x264_picture_t pic_out; RTMPPacket packet={0}; memset(&packet,0,sizeof(RTMPPacket)); packet.m_nChannel = 0x04; packet.m_headerType = RTMP_PACKET_SIZE_LARGE; packet.m_nTimeStamp = 0; packet.m_nInfoField2 = r->m_stream_id; packet.m_hasAbsTimestamp = 0; packet.m_body =(char *) szBodyBuffer; char * szTmp=(char *)szBodyBuffer; packet.m_packetType = RTMP_PACKET_TYPE_INFO; szTmp=put_byte(szTmp, AMF_STRING ); szTmp=put_amf_string(szTmp, "@setDataFrame" ); szTmp=put_byte(szTmp, AMF_STRING ); szTmp=put_amf_string(szTmp, "onMetaData" ); szTmp=put_byte(szTmp, AMF_OBJECT ); szTmp=put_amf_string( szTmp, "author" ); szTmp=put_byte(szTmp, AMF_STRING ); szTmp=put_amf_string( szTmp, "" ); szTmp=put_amf_string( szTmp, "copyright" ); szTmp=put_byte(szTmp, AMF_STRING ); szTmp=put_amf_string( szTmp, "" ); szTmp=put_amf_string( szTmp, "description" ); szTmp=put_byte(szTmp, AMF_STRING ); szTmp=put_amf_string( szTmp, "" ); szTmp=put_amf_string( szTmp, "keywords" ); szTmp=put_byte(szTmp, AMF_STRING ); szTmp=put_amf_string( szTmp, "" ); szTmp=put_amf_string( szTmp, "rating" ); szTmp=put_byte(szTmp, AMF_STRING ); szTmp=put_amf_string( szTmp, "" ); szTmp=put_amf_string( szTmp, "presetname" ); szTmp=put_byte(szTmp, AMF_STRING ); szTmp=put_amf_string( szTmp, "Custom" ); szTmp=put_amf_string( szTmp, "width" ); szTmp=put_amf_double( szTmp, p264Param->i_width ); szTmp=put_amf_string( szTmp, "width" ); szTmp=put_amf_double( szTmp, p264Param->i_width ); szTmp=put_amf_string( szTmp, "height" ); szTmp=put_amf_double( szTmp, p264Param->i_height ); szTmp=put_amf_string( szTmp, "framerate" ); szTmp=put_amf_double( szTmp, (double)p264Param->i_fps_num / p264Param->i_fps_den ); szTmp=put_amf_string( szTmp, "videocodecid" ); szTmp=put_byte(szTmp, AMF_STRING ); szTmp=put_amf_string( szTmp, "avc1" ); szTmp=put_amf_string( szTmp, "videodatarate" ); szTmp=put_amf_double( szTmp, p264Param->rc.i_bitrate ); szTmp=put_amf_string( szTmp, "avclevel" ); szTmp=put_amf_double( szTmp, p264Param->i_level_idc ); szTmp=put_amf_string( szTmp, "avcprofile" ); szTmp=put_amf_double( szTmp, 0x42 ); szTmp=put_amf_string( szTmp, "videokeyframe_frequency" ); szTmp=put_amf_double( szTmp, 3 ); szTmp=put_amf_string( szTmp, "" ); szTmp=put_byte( szTmp, AMF_OBJECT_END ); packet.m_nBodySize=szTmp-(char *)szBodyBuffer; RTMP_SendPacket(r,&packet,1); packet.m_packetType = RTMP_PACKET_TYPE_VIDEO; /* V