H264通过RTMP发布 V2.0 (Red5 Wowza 测试通过）

雨点打透心脏的1/2处 2022-03-20 03:06 191阅读 0赞

直接上代码

**\[cpp\]** [view plain][] [copy][view plain]

1.  // demo.cpp : 定义控制台应用程序的入口点。
2.  //
3.  \#include "stdafx.h"
4.  \#include <MqOaI.h>
5.  **extern**"C"
6.  \{ 
7.  \#include "../../common/common.h"
8.  \#include "../../common/cpu.h""
9.  \#include "../../x264.h"
10. \#include "../../encoder/set.h"
11. \} 
12. \#include "librtmp/rtmp\_sys.h"
13. \#include "librtmp/log.h"
14. \#include "librtmp/amf.h"
15. \#include "CameraDS.h"
16. **void** ConvertYCbCr2BGR(unsigned **char** \*pYUV,unsigned **char** \*pBGR,**int** iWidth,**int** iHeight); 
17. **void** ConvertRGB2YUV(**int** w,**int** h,unsigned **char** \*bmp,unsigned **char** \*yuv); 
18. **int** InitSockets() 
19. \{ 
20. \#ifdef WIN32
21. **WORD** version; 
22.  WSADATA wsaData; 
23.  version = MAKEWORD(1, 1); 
24. **return** (WSAStartup(version, &wsaData) == 0); 
25. \#else
26. **return** TRUE; 
27. \#endif
28. \} 
29. **inline****void** CleanupSockets() 
30. \{ 
31. \#ifdef WIN32
32.  WSACleanup(); 
33. \#endif
34. \} 
35. \#define HEX2BIN(a) (((a)&0x40)?((a)&0xf)+9:((a)&0xf))
36. **int** hex2bin(**char** \*str, **char** \*\*hex) 
37. \{ 
38. **char** \*ptr; 
39. **int** i, l = strlen(str); 
40. **if** (l & 1) 
41. **return** 0; 
42.  \*hex = (**char** \*)malloc(l/2); 
43.  ptr = \*hex; 
44. **if** (!ptr) 
45. **return** 0; 
46. **for** (i=0; i<l; i+=2) 
47.  \*ptr++ = (HEX2BIN(str\[i\]) << 4) | HEX2BIN(str\[i+1\]); 
48. **return** l/2; 
49. \} 
50. **char** \* put\_byte( **char** \*output, uint8\_t nVal ) 
51. \{ 
52.  output\[0\] = nVal; 
53. **return** output+1; 
54. \} 
55. **char** \* put\_be16(**char** \*output, uint16\_t nVal ) 
56. \{ 
57.  output\[1\] = nVal & 0xff; 
58.  output\[0\] = nVal >> 8; 
59. **return** output+2; 
60. \} 
61. **char** \* put\_be24(**char** \*output,uint32\_t nVal ) 
62. \{ 
63.  output\[2\] = nVal & 0xff; 
64.  output\[1\] = nVal >> 8; 
65.  output\[0\] = nVal >> 16; 
66. **return** output+3; 
67. \} 
68. **char** \* put\_be32(**char** \*output, uint32\_t nVal ) 
69. \{ 
70.  output\[3\] = nVal & 0xff; 
71.  output\[2\] = nVal >> 8; 
72.  output\[1\] = nVal >> 16; 
73.  output\[0\] = nVal >> 24; 
74. **return** output+4; 
75. \} 
76. **char** \* put\_be64( **char** \*output, uint64\_t nVal ) 
77. \{ 
78.  output=put\_be32( output, nVal >> 32 ); 
79.  output=put\_be32( output, nVal ); 
80. **return** output; 
81. \} 
82. **char** \* put\_amf\_string( **char** \*c, **const****char** \*str ) 
83. \{ 
84.  uint16\_t len = strlen( str ); 
85.  c=put\_be16( c, len ); 
86.  memcpy(c,str,len); 
87. **return** c+len; 
88. \} 
89. **char** \* put\_amf\_double( **char** \*c, **double** d ) 
90. \{ 
91.  \*c++ = AMF\_NUMBER; /\* type: Number \*/
92.  \{ 
93.  unsigned **char** \*ci, \*co; 
94.  ci = (unsigned **char** \*)&d; 
95.  co = (unsigned **char** \*)c; 
96.  co\[0\] = ci\[7\]; 
97.  co\[1\] = ci\[6\]; 
98.  co\[2\] = ci\[5\]; 
99.  co\[3\] = ci\[4\]; 
100.  co\[4\] = ci\[3\]; 
101.  co\[5\] = ci\[2\]; 
102.  co\[6\] = ci\[1\]; 
103.  co\[7\] = ci\[0\]; 
104.  \} 
105. **return** c+8; 
106. \} 
107. **int** main(**int** argc, **char** \* argv\[\]) 
108. \{ 
109. **if** (argc<2) 
110.  \{ 
111.  RTMP\_LogPrintf("RTMP\_URL IS NULL!!!/n"); 
112. //return -1;
113.  \} 
114. **if** (!InitSockets()) 
115.  \{ 
116.  RTMP\_LogPrintf("InitSockets Error!/n"); 
117. **return** \-1; 
118.  \} 
119.  RTMP\_LogPrintf("InitSockets!/n"); 
120.  CoInitialize(NULL); 
121.  CCameraDS camera; 
122. **if** (!camera.OpenCamera(0,320,240,**false**)) 
123.  \{ 
124.  RTMP\_LogPrintf("Open Camera Error/n"); 
125. **return** \-1; 
126.  \} 
127. **int** nHeight=camera.GetHeight(); 
128. **int** nWidth=camera.GetWidth(); 
129.  unsigned **char** \* szRGBBuffer=**new** unsigned **char**\[nHeight\*nWidth \* 3\]; 
130.  RTMP\_LogPrintf("Camera Open Scuess！ Picture Size\[%2dx%d\]/n",nWidth,nHeight); 
131.  RTMP\_debuglevel = RTMP\_LOGINFO; 
132.  RTMP \*r; 
133. //char uri\[\]="rtmp://127.0.0.1/live/test";
134. //char uri\[\]="rtmp://192.199.15.223/live/test";
135. //char uri\[\]="rtmp://221.9.244.4/live/jltv";
136. //char uri\[\]="rtmp://192.199.15.223/oflaDemo/red5StreamDemo";
137. //char uri\[\]="rtmp://192.199.15.151/live/test";
138. **char** uri\[\]="rtmp://127.0.0.1/live/zzj"; 
139.  r= RTMP\_Alloc(); 
140.  RTMP\_Init(r); 
141.  RTMP\_SetupURL(r, (**char**\*)uri); 
142.  RTMP\_EnableWrite(r); 
143.  RTMP\_Connect(r, NULL); 
144.  RTMP\_ConnectStream(r,0); 
145.  unsigned **char** szNalBuffer\[1024\*32\]; 
146.  unsigned **char** szBodyBuffer\[1024\*32\]; 
147.  x264\_nal\_t \*p264Nal; 
148. **int** i264Nal; 
149.  x264\_param\_t \* p264Param; 
150.  x264\_picture\_t \* p264Pic; 
151.  x264\_t \*p264Handle; 
152.  p264Param = **new** x264\_param\_t(); 
153.  p264Pic = **new** x264\_picture\_t(); 
154.  memset(p264Pic,0,**sizeof**(x264\_picture\_t)); 
155.  x264\_param\_default(p264Param); //set default param
156.  p264Param->i\_threads=2; 
157.  p264Param->i\_width = nWidth; //set frame width
158.  p264Param->i\_height = nHeight; //set frame height
159. /\*baseline level 1.1\*/
160.  p264Param->b\_cabac =0; 
161.  p264Param->i\_bframe =0; 
162.  p264Param->b\_interlaced=0; 
163.  p264Param->rc.i\_rc\_method=X264\_RC\_ABR;//X264\_RC\_CQP
164.  p264Param->i\_level\_idc=21; 
165.  p264Param->rc.i\_bitrate=200; 
166.  p264Param->i\_fps\_num=30; 
167.  p264Param->i\_keyint\_max=p264Param->i\_fps\_num\*3; 
168. **if**((p264Handle = x264\_encoder\_open(p264Param)) == NULL) 
169.  \{ 
170.  fprintf( stderr, "x264\_encoder\_open failed/n" ); 
171. **return** \-2; 
172.  \} 
173.  bs\_t bs=\{0\}; 
174.  x264\_picture\_alloc(p264Pic, X264\_CSP\_YV12, p264Param->i\_width, p264Param->i\_height); 
175.  p264Pic->i\_type = X264\_TYPE\_AUTO; 
176.  x264\_picture\_t pic\_out; 
177.  RTMPPacket packet=\{0\}; 
178.  memset(&packet,0,**sizeof**(RTMPPacket)); 
179.  packet.m\_nChannel = 0x04; 
180.  packet.m\_headerType = RTMP\_PACKET\_SIZE\_LARGE; 
181.  packet.m\_nTimeStamp = 0; 
182.  packet.m\_nInfoField2 = r->m\_stream\_id; 
183.  packet.m\_hasAbsTimestamp = 0; 
184.  packet.m\_body =(**char** \*) szBodyBuffer; 
185. **char** \* szTmp=(**char** \*)szBodyBuffer; 
186.  packet.m\_packetType = RTMP\_PACKET\_TYPE\_INFO; 
187.  szTmp=put\_byte(szTmp, AMF\_STRING ); 
188.  szTmp=put\_amf\_string(szTmp, "@setDataFrame" ); 
189.  szTmp=put\_byte(szTmp, AMF\_STRING ); 
190.  szTmp=put\_amf\_string(szTmp, "onMetaData" ); 
191.  szTmp=put\_byte(szTmp, AMF\_OBJECT ); 
192.  szTmp=put\_amf\_string( szTmp, "author" ); 
193.  szTmp=put\_byte(szTmp, AMF\_STRING ); 
194.  szTmp=put\_amf\_string( szTmp, "" ); 
195.  szTmp=put\_amf\_string( szTmp, "copyright" ); 
196.  szTmp=put\_byte(szTmp, AMF\_STRING ); 
197.  szTmp=put\_amf\_string( szTmp, "" ); 
198.  szTmp=put\_amf\_string( szTmp, "description" ); 
199.  szTmp=put\_byte(szTmp, AMF\_STRING ); 
200.  szTmp=put\_amf\_string( szTmp, "" ); 
201.  szTmp=put\_amf\_string( szTmp, "keywords" ); 
202.  szTmp=put\_byte(szTmp, AMF\_STRING ); 
203.  szTmp=put\_amf\_string( szTmp, "" ); 
204.  szTmp=put\_amf\_string( szTmp, "rating" ); 
205.  szTmp=put\_byte(szTmp, AMF\_STRING ); 
206.  szTmp=put\_amf\_string( szTmp, "" ); 
207.  szTmp=put\_amf\_string( szTmp, "presetname" ); 
208.  szTmp=put\_byte(szTmp, AMF\_STRING ); 
209.  szTmp=put\_amf\_string( szTmp, "Custom" ); 
210.  szTmp=put\_amf\_string( szTmp, "width" ); 
211.  szTmp=put\_amf\_double( szTmp, p264Param->i\_width ); 
212.  szTmp=put\_amf\_string( szTmp, "width" ); 
213.  szTmp=put\_amf\_double( szTmp, p264Param->i\_width ); 
214.  szTmp=put\_amf\_string( szTmp, "height" ); 
215.  szTmp=put\_amf\_double( szTmp, p264Param->i\_height ); 
216.  szTmp=put\_amf\_string( szTmp, "framerate" ); 
217.  szTmp=put\_amf\_double( szTmp, (**double**)p264Param->i\_fps\_num / p264Param->i\_fps\_den ); 
218.  szTmp=put\_amf\_string( szTmp, "videocodecid" ); 
219.  szTmp=put\_byte(szTmp, AMF\_STRING ); 
220.  szTmp=put\_amf\_string( szTmp, "avc1" ); 
221.  szTmp=put\_amf\_string( szTmp, "videodatarate" ); 
222.  szTmp=put\_amf\_double( szTmp, p264Param->rc.i\_bitrate ); 
223.  szTmp=put\_amf\_string( szTmp, "avclevel" ); 
224.  szTmp=put\_amf\_double( szTmp, p264Param->i\_level\_idc ); 
225.  szTmp=put\_amf\_string( szTmp, "avcprofile" ); 
226.  szTmp=put\_amf\_double( szTmp, 0x42 ); 
227.  szTmp=put\_amf\_string( szTmp, "videokeyframe\_frequency" ); 
228.  szTmp=put\_amf\_double( szTmp, 3 ); 
229.  szTmp=put\_amf\_string( szTmp, "" ); 
230.  szTmp=put\_byte( szTmp, AMF\_OBJECT\_END ); 
231.  packet.m\_nBodySize=szTmp-(**char** \*)szBodyBuffer; 
232.  RTMP\_SendPacket(r,&packet,1); 
233.  packet.m\_packetType = RTMP\_PACKET\_TYPE\_VIDEO; /\* VIDEO \*/
234.  szBodyBuffer\[ 0\]=0x17; 
235.  szBodyBuffer\[ 1\]=0x00; 
236.  szBodyBuffer\[ 2\]=0x00; 
237.  szBodyBuffer\[ 3\]=0x00; 
238.  szBodyBuffer\[ 4\]=0x00; 
239.  szBodyBuffer\[ 5\]=0x01; 
240.  szBodyBuffer\[ 6\]=0x42; 
241.  szBodyBuffer\[ 7\]=0xC0; 
242.  szBodyBuffer\[ 8\]=0x15; 
243.  szBodyBuffer\[ 9\]=0x03; 
244.  szBodyBuffer\[10\]=0x01; 
245.  szTmp=(**char** \*)szBodyBuffer+11; 
246. **short** slen=0; 
247.  bs\_init(&bs,szNalBuffer,16);//初始话bs
248.  x264\_sps\_write(&bs, p264Handle->sps);//读取编码器的SPS
249.  slen=bs.p-bs.p\_start+1;//spslen（short）
250.  slen=htons(slen); 
251.  memcpy(szTmp,&slen,**sizeof**(**short**)); 
252.  szTmp+=**sizeof**(**short**); 
253.  \*szTmp=0x67; 
254.  szTmp+=1; 
255.  memcpy(szTmp,bs.p\_start,bs.p-bs.p\_start); 
256.  szTmp+=bs.p-bs.p\_start; 
257.  \*szTmp=0x01; 
258.  szTmp+=1; 
259.  bs\_init(&bs,szNalBuffer,16);//初始话bs
260.  x264\_pps\_write(&bs, p264Handle->pps);//读取编码器的PPS
261.  slen=bs.p-bs.p\_start+1;//spslen（short）
262.  slen=htons(slen); 
263.  memcpy(szTmp,&slen,**sizeof**(**short**)); 
264.  szTmp+=**sizeof**(**short**); 
265.  \*szTmp=0x68; 
266.  szTmp+=1; 
267.  memcpy(szTmp,bs.p\_start,bs.p-bs.p\_start); 
268.  szTmp+=bs.p-bs.p\_start; 
269.  packet.m\_nBodySize=szTmp-(**char** \*)szBodyBuffer; 
270.  RTMP\_SendPacket(r,&packet,0); 
271.  unsigned **int** nTimes=0; 
272.  unsigned **int** oldTick=GetTickCount(); 
273.  unsigned **int** newTick=0; 
274.  packet.m\_nTimeStamp=0; 
275. 
276. **while**(**true**) 
277.  \{ 
278.  szBodyBuffer\[ 0\]=0x17; 
279.  szBodyBuffer\[ 1\]=0x01; 
280.  szBodyBuffer\[ 2\]=0x00; 
281.  szBodyBuffer\[ 3\]=0x00; 
282.  szBodyBuffer\[ 4\]=0x42; 
283.  unsigned **char** \* szTmp=szBodyBuffer+5; 
284.  unsigned **char** \* pNal=szNalBuffer; 
285.  nTimes++; 
286. **int** nFramsInPack=0; 
287. **while**(**true**) 
288.  \{ 
289.  nFramsInPack++; 
290.  unsigned **char** \* pCameraBuf = camera.QueryFrame(); 
291. **if** (!pCameraBuf) 
292.  \{ 
293. **return** \-1; 
294.  \} 
295. **for**(**int** ii=0;ii<nHeight;ii++) 
296.  \{ 
297.  memcpy(szRGBBuffer+(nWidth\*3)\*(nHeight-ii-1),pCameraBuf+(nWidth\*3)\*ii,nWidth\*3); 
298. //memcpy(pCameraBuf+nWidth\*(nHeight-ii-1),pCameraBuf+nWidth\*ii,nWidth\*3);
299. //memcpy(szLineBuffer,pCameraBuf+nWidth\*(nHeight-ii-1),nWidth\*3);
300.  \} 
301.  ConvertRGB2YUV(nWidth,nHeight,szRGBBuffer,p264Pic->img.plane\[0\]); 
302. //memcpy(p264Pic->img.plane\[0\],szNalBuffer,nWidth\*nHeight);
303. //memcpy(p264Pic->img.plane\[1\],szNalBuffer+nWidth\*nHeight,nWidth\*nHeight/4);
304. //memcpy(p264Pic->img.plane\[2\],szNalBuffer+nWidth\*nHeight\*5/4,nWidth\*nHeight/4);
305. /\*
306.  int nCount;
307.  nCount=fread(p264Pic->img.plane\[0\],1,176\*144,yuv);
308.  if (nCount<176\*144)
309.  \{ 
310.  fseek(yuv,SEEK\_SET,0);
311.  continue;
312.  \}
313.  nCount=fread(p264Pic->img.plane\[1\],1,176\*144/4,yuv);
314.  if (nCount<176\*144/4)
315.  \{ 
316.  fseek(yuv,SEEK\_SET,0);
317.  continue;
318.  \}
319.  nCount=fread(p264Pic->img.plane\[2\],1,176\*144/4,yuv);
320.  if (nCount<176\*144/4)
321.  \{ 
322.  fseek(yuv,SEEK\_SET,0);
323.  continue;
324.  \}
325.  \*/
326. **if**( x264\_encoder\_encode( p264Handle, &p264Nal, &i264Nal, p264Pic ,&pic\_out) < 0 ) 
327.  \{ 
328.  fprintf( stderr, "x264\_encoder\_encode failed/n" ); 
329.  \} 
330. **for**( **int** i = 0; i < i264Nal; i++ ) 
331.  \{ 
332. **int** i\_size; 
333. **int** i\_data; 
334.  i\_data = 1024\*32; 
335. **if**( ( i\_size = x264\_nal\_encode( pNal, &i\_data, 1, &p264Nal\[i\] ) ) > 0 ) 
336.  \{ 
337. **if** ((pNal\[4\]&0x60)==0) 
338.  \{ 
339. **continue**; 
340.  \} 
341. **if** (pNal\[4\]==0x67) 
342.  \{ 
343. **continue**; 
344.  \} 
345. **if** (pNal\[4\]==0x68) 
346.  \{ 
347. **continue**; 
348.  \} 
349.  memmove(pNal,pNal+4,i\_size-4); 
350.  pNal+=i\_size-4; 
351.  \} 
352. **else****if**( i\_size < 0 ) 
353.  \{ 
354.  fprintf( stderr,"need to increase buffer size (size=%d)/n", -i\_size ); 
355.  \} 
356.  \} 
357.  unsigned **int** nSize=pNal-szNalBuffer; 
358.  packet.m\_nBodySize=nSize+9; 
359. **if** (i264Nal>1) 
360.  \{ 
361.  szBodyBuffer\[ 0\]=0x17; 
362.  \} 
363. **else**
364.  \{ 
365.  szBodyBuffer\[ 0\]=0x27; 
366.  \} 
367.  put\_be32((**char** \*)szBodyBuffer+5,nSize); 
368.  memcpy(szBodyBuffer+9,szNalBuffer,pNal-szNalBuffer); 
369.  RTMP\_SendPacket(r,&packet,0); 
370.  Sleep(20); 
371.  newTick=GetTickCount(); 
372. //RTMP\_LogStatus("/rInfo NAUL Type:0x%02x size: %5d Tick:%03d %03d",szNalBuffer\[0\], nSize,33-nSleep,GetTickCount()-oldTick+nSleep);
373.  packet.m\_headerType = RTMP\_PACKET\_SIZE\_MEDIUM; 
374.  packet.m\_nTimeStamp+=newTick-oldTick; 
375.  oldTick=newTick; 
376. **break**; 
377.  \} 
378.  \} 
379. **return** 0; 
380. \}

![0_1304928973VX75.gif][]

[view plain]: http://blog.csdn.net/zblue78/article/details/6406707#
[0_1304928973VX75.gif]: /images/20220320/d75802f1041441259cdaa3dd79acdc35.png