树莓派使用python-librtmp实现rtmp推流h264的方法


Posted in Python onJuly 22, 2019

目的是能使用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
        else: # 非关键帧
          body0 = 0x27
          data_body_array = [bytes(bytearray(
            [body0, 0x01, 0x00, 0x00, 0x00, (buf_len >> 24) & 0xff, (buf_len >> 16) & 0xff,
             (buf_len >> 8) & 0xff,
             (buf_len) & 0xff])), buf]
          mbody = ''.join(data_body_array)
          # if (self.time_stamp == 0):
          self.time_stamp = int((time.time() - start_time) * 1000)
          packet_body = RTMPPacket(type=PACKET_TYPE_VIDEO, format=PACKET_SIZE_MEDIUM, channel=0x06,
                       timestamp=self.time_stamp, body=mbody)
        self.conn.send_packet(packet_body)
      elif len(indexs) == 2: # sps pps帧
        if self.sps is not None:
          return
        data_body_array = [bytes(bytearray([0x17, 0x00, 0x00, 0x00, 0x00, 0x01]))]
        sps = data[indexs[0] + 4: indexs[1]]
        sps_len = len(sps)
        pps = data[indexs[1] + 4: len(data)]
        pps_len = len(pps)
        self.sps = sps
        self.sps_len = sps_len
        self.pps = pps
        self.pps_len = pps_len
        data_body_array.append(sps[1:4])
        data_body_array.append(bytes(bytearray([0xff, 0xe1, (sps_len >> 8) & 0xff, sps_len & 0xff])))
        data_body_array.append(sps)
        data_body_array.append(bytes(bytearray([0x01, (pps_len >> 8) & 0xff, pps_len & 0xff])))
        data_body_array.append(pps)
        data_body = ''.join(data_body_array)
        body_packet = RTMPPacket(type=PACKET_TYPE_VIDEO, format=PACKET_SIZE_LARGE, channel=0x06,
                     timestamp=0, body=data_body)
        body_packet.packet.m_nInfoField2 = 1
 
        self.conn.send_packet(meta_packet, queue=True)
        self.conn.send_packet(body_packet, queue=True)
    except Exception, e:
      traceback.print_exc()
 
  def flush(self):
    pass
 
 
def get_property_string(string): # 返回两字节string长度及string
  length = len(string)
  return ''.join([chr((length >> 8) & 0xff), chr(length & 0xff), string])
 
 
def get_meta_string(string): # 按照meta packet要求格式返回bytes,带02前缀
  return ''.join([chr(0x02), get_property_string(string)])
 
 
def get_meta_double(db):
  nums = [0x00]
  fp = ctypes.pointer(ctypes.c_double(db))
  cp = ctypes.cast(fp, ctypes.POINTER(ctypes.c_longlong))
  for i in range(7, -1, -1):
    nums.append((cp.contents.value >> (i * 8)) & 0xff)
  return ''.join(bytes(bytearray(nums)))
 
 
def get_meta_boolean(isTrue):
  nums = [0x01]
  if (isTrue):
    nums.append(0x01)
  else:
    nums.append(0x00)
  return ''.join(bytes(bytearray(nums)))
 
 
conn = RTMP(
  'rtmp://192.168.199.154/oflaDemo/test', # 推流地址
  live=True)
librtmp.RTMP_EnableWrite(conn.rtmp)
conn.connect()
start_time = time.time()
# 拼装视频格式的数据包
meta_body_array = [get_meta_string('@setDataFrame'), get_meta_string('onMetaData'),
          bytes(bytearray([0x08, 0x00, 0x00, 0x00, 0x06])), # 两个字符串和ECMA array头,共计6个元素,注释掉了音频相关数据
          get_property_string('width'), get_meta_double(640.0),
          get_property_string('height'), get_meta_double(480.0),
          get_property_string('videodatarate'), get_meta_double(0.0),
          get_property_string('framerate'), get_meta_double(25.0),
          get_property_string('videocodecid'), get_meta_double(7.0),
          # get_property_string('audiodatarate'), get_meta_double(125.0),
          # get_property_string('audiosamplerate'), get_meta_double(44100.0),
          # get_property_string('audiosamplesize'), get_meta_double(16.0),
          # get_property_string('stereo'), get_meta_boolean(True),
          # get_property_string('audiocodecid'), get_meta_double(10.0),
          get_property_string('encoder'), get_meta_string('Lavf57.56.101'),
          bytes(bytearray([0x00, 0x00, 0x09]))
          ]
meta_body = ''.join(meta_body_array)
print meta_body.encode('hex')
meta_packet = RTMPPacket(type=PACKET_TYPE_INFO, format=PACKET_SIZE_LARGE, channel=0x04,
             timestamp=0, body=meta_body)
meta_packet.packet.m_nInfoField2 = 1 # 修改stream id
stream = conn.create_stream(writeable=True)
with picamera.PiCamera() as camera:
  camera.start_preview()
  time.sleep(2)
  camera.start_recording(Writer(conn), format='h264', resize=(640, 480), intra_period=25,
              quality=25) # 开始录制,数据输出到Writer的对象里
  while True:#永远不停止
    time.sleep(60)
  camera.stop_recording()
  camera.stop_preview()

以上这篇树莓派使用python-librtmp实现rtmp推流h264的方法就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
python正则匹配抓取豆瓣电影链接和评论代码分享
Dec 27 Python
python的keyword模块用法实例分析
Jun 30 Python
Python实现模拟时钟代码推荐
Nov 08 Python
python笔记:mysql、redis操作方法
Jun 28 Python
Python浅复制中对象生存周期实例分析
Apr 02 Python
Python基于property实现类的特性操作示例
Jun 15 Python
python获取指定字符串中重复模式最高的字符串方法
Jun 29 Python
python matlibplot绘制3D图形
Jul 02 Python
Python for循环中的陷阱详解
Jul 13 Python
python递归法解决棋盘分割问题
Jul 17 Python
python plotly画柱状图代码实例
Dec 13 Python
详解Python中的GIL(全局解释器锁)详解及解决GIL的几种方案
Jan 29 Python
python实现大文件分割与合并
Jul 22 #Python
cProfile Python性能分析工具使用详解
Jul 22 #Python
python实现大文本文件分割
Jul 22 #Python
python plotly绘制直方图实例详解
Jul 22 #Python
python分割一个文本为多个文本的方法
Jul 22 #Python
在linux系统下安装python librtmp包的实现方法
Jul 22 #Python
django搭建项目配置环境和创建表过程详解
Jul 22 #Python
You might like
PHP默认安装产生系统漏洞
2006/10/09 PHP
简单实用的网站PHP缓存类实例
2014/07/18 PHP
PHP中生成UUID自定义函数分享
2015/06/10 PHP
PHPStrom 新建FTP项目以及在线操作教程
2016/10/16 PHP
php基于单例模式封装mysql类完整实例
2016/10/18 PHP
php获取flash尺寸详细数据的方法
2016/11/12 PHP
解决Laravel blade模板转义html标签的问题
2019/09/03 PHP
云网广告中的代码,提示出错,大家找找
2006/11/21 Javascript
优化网页之快速的呈现我们的网页
2007/06/29 Javascript
js+css使DIV始终居于屏幕中间 左下 左上 右上 右下的代码集合
2011/03/10 Javascript
jQuery 全选/反选以及单击行改变背景色实例
2013/07/02 Javascript
jquery用data方法获取某个元素上的事件
2014/06/23 Javascript
json定义及jquery操作json的方法
2016/09/29 Javascript
微信小程序 swiper组件详解及实例代码
2016/10/25 Javascript
AngularJS实用基础知识_入门必备篇(推荐)
2017/07/10 Javascript
JS实现的走迷宫小游戏完整实例
2017/07/19 Javascript
Vue学习笔记进阶篇之vue-router安装及使用方法
2017/07/19 Javascript
手把手教你使用TypeScript开发Node.js应用
2019/05/06 Javascript
JavaScript的查询机制LHS和RHS解析
2019/08/16 Javascript
python登录QQ邮箱发信的实现代码
2013/02/10 Python
numpy自动生成数组详解
2017/12/15 Python
python实现Virginia无密钥解密
2019/03/20 Python
Python中的Socket 与 ScoketServer 通信及遇到问题解决方法
2019/04/01 Python
使用tensorflow实现VGG网络,训练mnist数据集方式
2020/05/26 Python
python如何输出反斜杠
2020/06/18 Python
CSS3教程(9):设置RGB颜色
2009/04/02 HTML / CSS
美体小铺法国官方网站:The Body Shop法国
2020/06/04 全球购物
校园自助餐厅的创业计划书
2013/12/26 职场文书
机关门卫制度
2014/02/01 职场文书
护理专业学生职业生涯规划范文
2014/03/11 职场文书
保证书范文大全
2014/04/28 职场文书
生物工程专业求职信
2014/09/03 职场文书
幼儿园法制宣传日活动总结
2014/11/01 职场文书
经验交流材料格式
2014/12/30 职场文书
创业计划书之香辣虾火锅
2019/09/23 职场文书
vue Element-ui表格实现树形结构表格
2021/06/07 Vue.js