树莓派使用树莓派使用python-librtmp实现实现rtmp推流推流h264的方法的方法
今天小编就为大家分享一篇树莓派使用python-librtmp实现rtmp推流h264的方法，具有很好的参考价值，希望对
大家有所帮助。一起跟随小编过来看看吧
目的是能使用Python进行rtmp推流，方便在h264帧里加入弹幕等操作。
librtmp使用的是使用的是0.3.0，使用树莓派，使用树莓派noir官方摄像头适配的。官方摄像头适配的。
通过wireshark抓ffmpeg的包一点点改动，最终可以在red5和斗鱼上推流了。
没怎么写过python，有不恰当的地方请包涵。
上代码：上代码：
# -- coding: utf-8 --
# http://blog.csdn.net/luhanglei
import picamera
import time
import traceback
import ctypes
from librtmp import *
global meta_packet
global start_time
class Writer(): # camera可以通过一个类文件的对象来输出，实现write方法即可
conn = None # rtmp连接
sps = None # 记录sps帧，发过以后就不需要再发了（抓包看到ffmpeg是这样的）
pps = None # 同上
sps_len = 0 # 同上
pps_len = 0 # 同上
time_stamp = 0
def __init__(self, conn):
self.conn = conn
def write(self, data):
try:
# 寻找h264帧间隔符
indexs = []
index = 0
data_len = len(data)
while index < data_len - 3:
if ord(data[index]) == 0x00 and ord(data[index + 1]) == 0x00 and ord(
data[index + 2]) == 0x00 and ord(data[index + 3]) == 0x01:
indexs.append(index)
index = index + 3
index = index + 1
# 寻找h264帧间隔符 完成
# 通过间隔符个数确定类型，树莓派摄像头的第一帧是sps+pps同时发的
if len(indexs) == 1: # 非sps pps帧
buf = data[4: len(data)] # 裁掉原来的头（00 00 00 01），把帧内容拿出来
buf_len = len(buf)
type = ord(buf[0]) & 0x1f
if type == 0x05: # 关键帧，根据wire shark抓包结果，需要拼装sps pps 帧内容 三部分，长度都用4个字节表示
body0 = 0x17
data_body_array = [bytes(bytearray(
[body0, 0x01, 0x00, 0x00, 0x00, (self.sps_len >> 24) & 0xff, (self.sps_len >> 16) & 0xff,
(self.sps_len >> 8) & 0xff,
self.sps_len & 0xff])), self.sps,
bytes(bytearray(
[(self.pps_len >> 24) & 0xff, (self.pps_len >> 16) & 0xff, (self.pps_len >> 8) & 0xff,
self.pps_len & 0xff])),
self.pps,
bytes(bytearray(
[(buf_len >> 24) & 0xff, (buf_len >> 16) & 0xff, (buf_len >> 8) & 0xff, (buf_len) & 0xff])),
buf
]
mbody = ''.join(data_body_array)
time_stamp = 0 # 第一次发出的时候，发时间戳0，此后发真时间戳
if self.time_stamp != 0:
time_stamp = int((time.time() - start_time) * 1000)
packet_body = RTMPPacket(type=PACKET_TYPE_VIDEO, format=PACKET_SIZE_LARGE, channel=0x06,
timestamp=time_stamp, body=mbody)
packet_body.packet.m_nInfoField2 = 1
评论0