RTMP 直播推流 Demo(二)—— 音频推流与视频推流

音视频编解码系列目录:

Android 音视频基础知识
Android 音视频播放器 Demo(一)—— 视频解码与渲染
Android 音视频播放器 Demo(二)—— 音频解码与音视频同步
RTMP 直播推流 Demo(一)—— 项目配置与视频预览
RTMP 直播推流 Demo(二)—— 音频推流与视频推流

上一节我们对项目进行了配置,并且实现了摄像头预览,摄像头采集到的图像数据已经可以通过 LivePusher 传递到 Native 层,接下来就可以开始音视频编码与推流了。

1、视频编码

通过 CameraHelper 可以获取视频的帧数据,我们需要对其进行编码再将数据“塞”到 RTMPPacket 中,然后才能将 RTMPPacket 发送给服务器。

编码之前,需要先对编码器等元素进行初始化。

1.1 初始化

Native 层的初始化是由 LivePusher 触发的:

class LivePusher(activity: Activity,cameraId: Int,previewWidth: Int,previewHeight: Int,fps: Int,bitrate: Int
) {private val mVideoChannel: VideoChannelprivate val mAudioChannel: AudioChannelinit {// 这个 nativeInit() 要在 AudioChannel 之前初始化,因为后者初始化时会用到 Native 方法nativeInit()mVideoChannel =VideoChannel(this, activity, cameraId, previewWidth, previewHeight, fps, bitrate)mAudioChannel = AudioChannel(this, 44100, 2)}
}

Native 入口为 native-lib,初始化时要创建 Native 层的 VideoChannel 和 AudioChannel:

// 视频通道,处理视频编码等工作
VideoChannel *videoChannel = nullptr;
// 音频通道,处理音频编码
AudioChannel *audioChannel = nullptr;extern "C"
JNIEXPORT void JNICALL
Java_com_rtmp_pusher_LivePusher_nativeInit(JNIEnv *env, jobject thiz) {videoChannel = new VideoChannel;videoChannel->setVideoCallback(callback);audioChannel = new AudioChannel;audioChannel->setAudioCallback(callback);// releasePacket 是丢弃 AVPacket 队列中元素的回调函数packets.setReleaseCallback(releasePacket);
}

我们使用 x264 进行视频编码,在编码前需要对 x264 编码器进行初始化,初始化的时机是 SurfaceView 的尺寸发生变化时(因为编码器初始化需要宽高数据)。在 CameraHelper 的 setPreviewOrientation() 中我们通过 OnSurfaceSizeChangedListener 将宽高变化发送出去了,在 VideoChannel 中接收时做编码器的初始化:

	// CameraHelper.OnSurfaceSizeChangedListeneroverride fun onSizeChanged(width: Int, height: Int) {mLivePusher.nativeInitVideoEncoder(width, height, mFps, mBitrate)}

在 native-lib 的对应函数中,将初始化的具体工作交给 Native 的 VideoChannel:

extern "C"
JNIEXPORT void JNICALL
Java_com_rtmp_pusher_LivePusher_nativeInitVideoEncoder(JNIEnv *env, jobject thiz, jint width,jint height, jint fps, jint bitrate) {if (videoChannel) {videoChannel->initVideoEncoder(width, height, fps, bitrate);}
}

initVideoEncoder() 主要是设置 x264 编码器的参数并打开编码器,此外还需根据宽高数据创建接收视频一帧数据(理解成一张图片)的对象 x264_picture_t:

/*** 初始化 x264 编码器,为了防止被多次初始化,需要加互斥锁保证线程安全*/
void VideoChannel::initVideoEncoder(int width, int height, int fps, int bitrate) {pthread_mutex_lock(&mutex);// 初始化编码器所需的参数mWidth = width;mHeight = height;mFps = fps;mBitrate = bitrate;// y 的个数为宽高之积,uv 的数量分别都是 y 的 1/4y_len = width * height;uv_len = y_len / 4;// 防止重复初始化编码器if (videoEncoder) {x264_encoder_close(videoEncoder);videoEncoder = nullptr;}// 防止重复初始化一帧图片if (pic_in) {x264_picture_clean(pic_in);DELETE(pic_in)}// 初始化 x264 编码器x264_param_t param;// 超快、零延迟,因为直播要求时效性x264_param_default_preset(&param, "ultrafast", "zerolatency");// 编码级别,代表了编码器能够处理的视频参数和编码限制,如视频分辨率、帧率、比特率等,32 为中等偏上,// 对应的最大编码码率为 2000,最大清晰度为 1280 * 720,最高帧率为 60 帧param.i_level_idc = 32;// 输出格式为 YUV420P,P 是 Planar 表示 UV 平面格式,例如 VVVVUUUU,没有 P 就是交错模式 VUVUVUVUparam.i_csp = X264_CSP_I420;param.i_width = width;param.i_height = height;// 设置两个 picture 之间的 B 帧数量为 0。直播不能有 B 帧,因为影响编解码效率param.i_bframe = 0;// 码率控制方式:CQP(恒定质量)、CRF(恒定码率)、ABR(平均码率)param.rc.i_rc_method = X264_RC_CRF;// 码率、瞬时最大码率、码率控制区大小(单位 Kb/s,设置了瞬时最大码率就必须设置它)param.rc.i_bitrate = bitrate / 1000;param.rc.i_vbv_max_bitrate = bitrate / 1000 * 1.2;param.rc.i_vbv_buffer_size = bitrate / 1000;// 0 表示只使用 fps 进行码率控制,1 表示使用时间基和时间戳param.b_vfr_input = 0;// 帧率分子与分母param.i_fps_num = fps;param.i_fps_den = 1;// 时间基的分子与分母,二者做除法可以算出帧之间的时间间隔,音视频同步时有用param.i_timebase_num = param.i_fps_den;param.i_timebase_den = param.i_fps_num;// 设置编码器的最大关键帧间隔,编码器在编码视频时会尽量保持每隔 fps * 2 帧生成一个关键帧,即约 2 秒一个关键帧param.i_keyint_max = fps * 2;// 是否重复头部,设置为 1,编码时会在每个关键帧之前加入 SPS/PPSparam.b_repeat_headers = 1;// 指定并行编码线程数,如果设置为 0 则会由 x264 编码库自动决定线程数param.i_threads = 1;// 将 param 参数应用到 H.264/AVC 规范定义的编码配置文件上,这些文件包括 baseline、main、high 等x264_param_apply_profile(&param, "baseline");// 为 pic_in 分配内存空间并初始化内部成员pic_in = new x264_picture_t;x264_picture_alloc(pic_in, param.i_csp, param.i_width, param.i_height);// 打开编码器videoEncoder = x264_encoder_open(&param);if (videoEncoder) {LOGD("成功打开x264编码器");}pthread_mutex_unlock(&mutex);
}

说明:

  1. 参数设置中用到的方法 x264_param_default_preset() 和 x264_param_apply_profile() 都有可选参数,这些参数在源码的函数上面以数组的形式呈现。比如 x264_param_default_preset():

    static const char * const x264_preset_names[] = { "ultrafast", "superfast", "veryfast", "faster", "fast", "medium", "slow", "slower", "veryslow", "placebo", 0 };static const char * const x264_tune_names[] = { "film", "animation", "grain", "stillimage", "psnr", "ssim", "fastdecode", "zerolatency", 0 };/*  returns 0 on success, negative on failure (e.g. invalid preset/tune name). */
    X264_API int x264_param_default_preset( x264_param_t *, const char *preset, const char *tune );
    

    x264_preset_names 就是 preset 的可选参数数组,x264_tune_names 就是 tune 的可选参数数组,如果选 0 表示使用默认配置。其他函数也是类似的套路。

  2. 参数全部设置完毕后通过 x264_encoder_open() 打开编码器会得到一个 x264_t 类型的对象,也就是编码器对象,我们将其设置为成员变量:

    class VideoChannel {
    ...
    private:...// x264 解码器x264_t *videoEncoder;// 表示 x264 的一帧图片x264_picture_t *pic_in;
    };
    

    此外 x264_picture_t 表示 x264 的一帧图片,也保存为成员变量,在通过 x264_picture_alloc() 为其分配了内存之后,编码器的初始化工作就完成了

1.2 将数据传递到 Native 层

初始化完毕可以进行编码了,上层的 VideoChannel 在 CameraHelper.OnPreviewListener 接口的 onPreviewFrame() 中会接收到 CameraHelper 传来的视频数据:

	// CameraHelper.OnPreviewListeneroverride fun onPreviewFrame(data: ByteArray) {if (mIsLiving) {mLivePusher.nativePushVideo(data)}}

mIsLiving 表示是否开始直播推流了,通过如下方法进行控制:

	fun startLive() {mIsLiving = true}fun stopLive() {mIsLiving = false}

具体的编码工作还是交给 Native 的 VideoChannel 执行:

extern "C"
JNIEXPORT void JNICALL
Java_com_rtmp_pusher_LivePusher_nativePushVideo(JNIEnv *env, jobject thiz, jbyteArray data_) {if (!videoChannel || !readyPushing) {return;}jbyte *data = env->GetByteArrayElements(data_, nullptr);videoChannel->encodeData(data);env->ReleaseByteArrayElements(data_, data, 0);
}

encodeData() 的内容较多,我们放在下一小节中单独讲解。

1.3 编码

编码的工作包括如下内容:

  1. 将摄像头采集到的 NV21 格式数据转换为 I420 格式数据
  2. 使用 x264 对一帧图片进行编码
  3. 将编码后的数据发送到队列中

格式转换

由于安卓摄像头采集的是 NV21 格式,但 Windows 或 IOS 都是 I420 格式,因此需要进行转换。二者的不同之处在于 UV 的排列方式:

    Y Y Y Y Y Y      Y Y Y Y Y Y      Y Y Y Y Y Y      Y Y Y Y Y Y      Y Y Y Y Y Y      Y Y Y Y Y Y      Y Y Y Y Y Y      Y Y Y Y Y Y      V U V U V U      U U U U U UV U V U V U      V V V V V V            - NV21 -          - I420 - 

因此我们要把 YUV 按照 I420 的排列方式存入 pic_in 中:

void VideoChannel::encodeData(signed char *data) {pthread_mutex_lock(&mutex);/** 1.将摄像头采集到的 NV21 格式转换为 I420 格式*/// 先将 y 拷贝到 pic_in->img.plane[0]memcpy(pic_in->img.plane[0], data, y_len);// 再拷贝 vu,u、v 分别存放在 pic_in->img.plane[1],pic_in->img.plane[2]// 在 NV21 中,vu 成对出现,v 在前for (int i = 0; i < uv_len; ++i) {*(pic_in->img.plane[1] + i) = *(data + y_len + i * 2 + 1);*(pic_in->img.plane[2] + i) = *(data + y_len + i * 2);}...
}

用 x264 编码

使用 x264_encoder_encode() 对刚刚得到的 pic_in 进行编码,得到编码后的 NAL 数组 nal 以及 NALU 的数量 pi_nal,当然还有编码后的图片 pic_out:

void VideoChannel::encodeData(signed char *data) {.../** 2.编码*/// 编码后得到的 NAL 数组x264_nal_t *nal = nullptr;// NAL 中 NALU 的数量int pi_nal;// 编码后输出的图片x264_picture_t pic_out;// 编码一张图片,如果失败会返回负值if (x264_encoder_encode(videoEncoder, &nal, &pi_nal, pic_in, &pic_out) < 0) {LOGE("x264编码失败");pthread_mutex_unlock(&mutex);return;}...
}

发送到队列中

上一步我们得到了编码后的 NALU 数组,接下来要做的就是将这些 NALU 封装进 RTMPPacket 并发送到队列中。在转换过程中,对于不同的 NALU 类型有不同的操作。这里我们先简单复习一下 NALU 的相关知识。

H.264 原始码流(又称为裸流),是由一个接一个的 NALU 组成的:

3.2.5-NALU结构

每个 NALU 之间都有一个起始码 00 00 00 01(此 NALU 有多片)或 00 00 01(此 NALU 只有一片),而 NALU 本身由 NAL 头和挂载数据 RBSP 组成,其中 NAL 头的低五位表示帧类型,我们需要关注的类型有四种:

  1. NAL 头的数据为 0x67,低五位为 0x67 & 0x1F = 7,类型是 NAL_SPS,表示这是一个 SPS 帧
  2. NAL 头的数据为 0x68,低五位为 0x68 & 0x1F = 8,类型是 NAL_PPS,表示这是一个 PPS 帧
  3. NAL 头的数据为 0x65,低五位为 0x65 & 0x1F = 5,类型是 NAL_SLICE_IDR,表示这是一个 IDR 帧,即关键帧
  4. NAL 头的数据为 0x61,低五位为 0x61 & 0x1F = 1,类型是 NAL_SLICE,表示这是一个非关键帧

这四种类型的格式如下,其中 SPS 与 PPS 是放在一起的:

2024-3-9.H264视频包格式

这里要解释一下为什么第一个字节是 0x17 或 0x27,下表是第一个字节的含义:

字段占位描述
FrameType4帧类型。
1: key frame(for AVC, a seekable frame)
2: inter frame(for AVC, a non-seekable frame)
3: disposable inter frame(H.263 only)
4: generated keyframe(reserved for server use only)
5: video info/command frame
CodecID4编码类型。
1: JPEG(目前未用到)
2: Sorenson H.263
3: Screen video
4: On2 VP6
5: On2 VP6 with alpha channel
6: Screen video version 2
7: AVC(高级视频编码)

关键帧的 FrameType 是 1,非关键帧的 FrameType 是 2,CodecID 都是 AVC 即 7,因此关键帧的第一个字节是 0x17,非关键帧第一个字节是 0x27。

此外,SPS 与 PPS 的数据需要注意每个字节的内容:

2024-3-9.SPS与PPS格式

接下来进入代码:

void VideoChannel::encodeData(signed char *data) {/** 3.将编码后的数据发送到队列中*/// SPS 与 PPS 的长度int sps_len, pps_len;// SPS 与 PPS 的内容uint8_t sps[100], pps[100];// 视频帧的展示时间戳要进行累加pic_in->i_pts += 1;for (int i = 0; i < pi_nal; ++i) {if (nal[i].i_type == NAL_SPS) {// SPS 的长度和内容都要刨除起始码的 4 个字节sps_len = nal[i].i_payload - 4;memcpy(sps, nal[i].p_payload + 4, sps_len);} else if (nal[i].i_type == NAL_PPS) {// PPS 的长度和内容也要刨除起始码的 4 个字节pps_len = nal[i].i_payload - 4;memcpy(pps, nal[i].p_payload + 4, pps_len);// PPS 在 SPS 后面,这里既然拿到了 PPS,说明 SPS 也已经拿到了,可以发送二者了sendSpsPps(sps, pps, sps_len, pps_len);} else {// 发送非 SPS、PPS 的数据帧,可能是 I 帧或 P 帧sendFrame(nal[i].i_type, nal[i].i_payload, nal[i].p_payload);}}pthread_mutex_unlock(&mutex);
}

sendSpsPps() 参照上面的表格填写每个字节的数据:

/*** 将 SPS 与 PPS 发送到队列中*/
void VideoChannel::sendSpsPps(uint8_t *sps, uint8_t *pps, int sps_len, int pps_len) {/** 1.创建一个 RTMPPacket 并分配空间*/// RTMPPacket 的数据长度int body_size = 5 + 8 + sps_len + 3 + pps_len;auto *packet = new RTMPPacket;RTMPPacket_Alloc(packet, body_size);/** 2.逐个字节填充 RTMPPacket 的内容*/int index = 0;// SPS 与 PPS 的第一个字节都是 0x17packet->m_body[index++] = 0x17;// 第 2 ~ 5 个字节都是 0packet->m_body[index++] = 0x00;packet->m_body[index++] = 0x00;packet->m_body[index++] = 0x00;packet->m_body[index++] = 0x00;// 第 6 个字节表示配置版本,这里我们写 0x01packet->m_body[index++] = 0x01;// 第 7 ~ 9 个字节分别是 sps[1]、sps[2]、sps[3]packet->m_body[index++] = sps[1];packet->m_body[index++] = sps[2];packet->m_body[index++] = sps[3];// 第 10 个字节是包长数据所使用的字节数,通常为 0xFFpacket->m_body[index++] = 0xFF;// 第 11 个字节表示 SPS 个数,通常为 0xE1packet->m_body[index++] = 0xE1;// 接下来是 SPS 长度,用两个字节表示,先存高 8 位packet->m_body[index++] = (sps_len >> 8) & 0xFF;packet->m_body[index++] = sps_len & 0xFF;// SPS 数据内容memcpy(&packet->m_body[index], sps, sps_len);index += sps_len;// 最后是 PPS 的个数、长度和内容,与 SPS 类型packet->m_body[index++] = 0x01;packet->m_body[index++] = (pps_len >> 8) & 0xFF;packet->m_body[index++] = pps_len & 0xFF;memcpy(&packet->m_body[index], pps, pps_len);/** 3.封包处理*/// 包的类型是视频包packet->m_packetType = RTMP_PACKET_TYPE_VIDEO;// 包的大小packet->m_nBodySize = body_size;// 通道 ID,可随意赋值但不要与 rtmp.c 内的 m_nChannel = 0x04 冲突packet->m_nChannel = 10;// SPS 与 PPS 没有时间戳packet->m_nTimeStamp = 0;// 不使用绝对时间戳,而是使用相对时间戳packet->m_hasAbsTimestamp = 0;// 包的头部类型,像 SPS 与 PPS 这种不是很大的可以设置为中等大小packet->m_headerType = RTMP_PACKET_SIZE_MEDIUM;/** 4.通过回调将 RTMPPacket 存入队列,因为队列是在 native-lib 中控制的*/videoCallback(packet);
}

sendFrame() 就是发送关键帧或非关键帧的数据,要比发送 SPS 和 PPS 简单些:

/*** 将数据帧(I 帧或 P 帧)发送到队列中* @param type 帧的类型* @param payload_size 帧的数据长度* @param payload 帧数据*/
void VideoChannel::sendFrame(int type, int payload_size, uint8_t *payload) {/** 1.先计算 RTMPPacket 的数据长度,该长度由固定的前 5 个字节,加上数据长度的 4 个* 字节,最后加上 H.264 裸数据构成。真正的数据长度是 payload_size 减去起始码的长度*/if (payload[2] == 0x00) {// 起始码是 00 00 00 01,那么有效的数据大小要减去 4 字节,// payload 的指针也要相应的跳过起始码的 4 字节payload_size -= 4;payload += 4;} else if (payload[2] == 0x01) {// 起始码是 00 00 01payload_size -= 3;payload += 3;}int body_size = 5 + 4 + payload_size;/** 2.创建 RTMPPacket 并填入数据*/auto packet = new RTMPPacket;RTMPPacket_Alloc(packet, body_size);// 关键帧的第 1 个字节是 0x17,非关键帧是 0x27if (type == NAL_SLICE_IDR) {packet->m_body[0] = 0x17;} else {packet->m_body[0] = 0x27;}// 第 2 个字节为 1 表示是关键帧或非关键帧,为 0 就是 PPS、SPS 包packet->m_body[1] = 0x01;packet->m_body[2] = 0x00;packet->m_body[3] = 0x00;packet->m_body[4] = 0x00;// 接下来的 4 个字节是帧长度,先保存最高的 8 位,以此类推packet->m_body[5] = (payload_size >> 24) & 0xFF;packet->m_body[6] = (payload_size >> 16) & 0xFF;packet->m_body[7] = (payload_size >> 8) & 0xFF;packet->m_body[8] = payload_size & 0xFF;// 从第 10 个字节开始拷贝帧数据memcpy(&packet->m_body[9], payload, payload_size);/** 3.封包操作,与 sendSpsPps() 类似,不同之处在于帧数据有时间戳,* 并且 m_headerType 为 LARGE 包的类型是视频包*/packet->m_packetType = RTMP_PACKET_TYPE_VIDEO;// 包的大小packet->m_nBodySize = body_size;// 通道 ID,可随意赋值但不要与 rtmp.c 内的 m_nChannel 冲突packet->m_nChannel = 10;// 帧数据有时间戳packet->m_nTimeStamp = -1;// 不需要绝对或相对时间戳packet->m_hasAbsTimestamp = 0;// 包的类型,关键帧数据量大,要设置为大包packet->m_headerType = RTMP_PACKET_SIZE_LARGE;/** 4.把 RTMPPacket 回调给 native-lib 存入队列中*/videoCallback(packet);
}

两个发送函数最后都是通过 videoCallback 这个回调接口将 RTMPPacket 存入队列的:

class VideoChannel {
public:typedef void (*VideoCallback)(RTMPPacket *packet);void setVideoCallback(VideoCallback videoCallback);
private:VideoCallback videoCallback;
}

native-lib 实现它:

SafeQueue<RTMPPacket *> packets;void callback(RTMPPacket *packet) {if (packet) {if (packet->m_nTimeStamp == -1) {// I/P/B帧需要时间戳,但是 SPS 和 PPS 本身没有时间戳也就不需要packet->m_nTimeStamp = RTMP_GetTime() - start_time;}packets.put(packet);}
}

如果 packet 的 m_nTimeStamp 之前设置为 -1,表示它需要时间戳,那么就计算出时间戳赋值给它,然后放进线程安全的队列 packets 中。

2、视频推流

用户点击界面的开始直播按钮,会触发 LivePusher 的 startLive(),进而调用 nativeStart(),执行对应的 Native 函数,需要开启一个子线程 pid_start 来连接 RTMP 服务器并发送 RTMPPacket 的工作:

/*** 在子线程中开启 task_start 任务,通过 RTMP 连接服务器并推送数据*/
extern "C"
JNIEXPORT void JNICALL
Java_com_rtmp_pusher_LivePusher_nativeStart(JNIEnv *env, jobject thiz, jstring path_) {// 避免多次开启if (isStart) {return;}isStart = true;const char *path = env->GetStringUTFChars(path_, nullptr);// 深拷贝,以免在子线程中使用 path 时,path 已经在本函数末尾被回收char *url = new char[strlen(path) + 1];strcpy(url, path);// 子线程开启 task_start 任务pthread_create(&pid_start, nullptr, task_start, url);// 回收 pathenv->ReleaseStringUTFChars(path_, path);
}

线程任务 task_start 会连接 RTMP 服务器,并且在一个循环中不断取出 RTMPPacket 发送给服务器实现推流。RTMP 的使用步骤如下:

3.2.5.2-RtmpDump代码流程

task_start 的内容就是实现如上步骤:

/*** 链接 RTMP 服务器,并不断从队列中取出 RTMPPacket 发送至服务器* 这里有两个问题需要再精进一些:* 1.考虑是否要将 1 ~ 7 步放在 do-while(false) 循环中?* 答:如果添加重试功能,可能需要这么做,以更好地进行流程控制* 2.如果推流速度远远慢于 RTMPPacket 入队的速度怎么办?* 答:可能会造成内存溢出,需要添加入队的速度控制,当前 Demo 代码可以正常运行*/
void *task_start(void *args) {char *url = static_cast<char *>(args);// 1.申请内存RTMP *rtmp = RTMP_Alloc();if (!rtmp) {LOGE("RTMP_Alloc failed");return nullptr;}// 2.初始化RTMP_Init(rtmp);// 设置连接的超时时间,单位为秒rtmp->Link.timeout = 5;// 3.设置地址if (!RTMP_SetupURL(rtmp, url)) {LOGE("RTMP_SetupURL failed");return nullptr;}// 4.开启输出模式RTMP_EnableWrite(rtmp);// 5.连接服务器if (!RTMP_Connect(rtmp, nullptr)) {LOGE("RTMP_Connect failed");return nullptr;}// 6.连接流if (!RTMP_ConnectStream(rtmp, 0)) {LOGE("RTMP_ConnectStream failed");return nullptr;}// 记录开始时间,准备向服务器推流start_time = RTMP_GetTime();readyPushing = true;packets.setEnable(true);RTMPPacket *packet = nullptr;// 获取并将音频头存入队列,实测不用放这个头也是可以成功将音频推流的callback(audioChannel->getAudioSeqHeader());// 在循环中将数据发送至服务器while (readyPushing) {// 从队列中取出 packet,队列为空时会阻塞packets.get(packet);if (!readyPushing) {break;}if (!packet) {continue;}// 成功取出 packet 后为其设置流的 IDpacket->m_nInfoField2 = rtmp->m_stream_id;// 7.发送数据,1 是开启内部缓冲,在收到结果前,packet 会保存在队列中if (!RTMP_SendPacket(rtmp, packet, 1)) {// RTMP 发送失败会自动断开与服务器的连接,因此就跳出循环LOGE("RTMP_SendPacket failed");releasePacket(&packet);break;}}// 考虑到可能从 if (!readyPushing) 跳出循环的情况,也要进行释放releasePacket(&packet);// 更新状态isStart = false;readyPushing = false;packets.setEnable(false);packets.clear();if (rtmp) {// 8.关闭连接RTMP_Close(rtmp);// 9.释放RTMP_Free(rtmp);}DELETE(url)return nullptr;
}

代码就是按照图片给出的顺序连接服务器,然后在推流状态下不断从队列中取出 RTMPPacket,通过 RTMP_SendPacket() 发送给服务器,从而实现推流。

3、音频编码与推流

3.1 采集音频

上层的 AudioChannel 控制 AudioRecord 录制音频:

class AudioChannel(private val mLivePusher: LivePusher,sampleRateInHz: Int,channels: Int
) {private val mExecutor: ExecutorServiceprivate var mAudioRecord: AudioRecord?// 比如单通道样本数是 1024,那么双声道就是 2048,再加上采样位数是 16 位,// 再乘 2 就是 4096,也就是 mInputSamples 的值了private val mInputSamples: Intprivate var mIsLiving = falseinit {mExecutor = Executors.newSingleThreadExecutor()// 通道配置val channelConfig = if (channels == 2)AudioFormat.CHANNEL_IN_STEREOelseAudioFormat.CHANNEL_IN_MONO// AudioRecord 所需最小的缓冲区大小val minBufferSize = AudioRecord.getMinBufferSize(sampleRateInHz,channelConfig,AudioFormat.ENCODING_PCM_16BIT // 采样位数设置为 16 位) * 2// 还需获取 faac 编码器缓冲区的大小,获取前需要先初始化 faac 编码器mLivePusher.nativeInitAudioEncoder(sampleRateInHz, channels)// 采样位数为 16 位,需要再乘以 2 个字节mInputSamples = mLivePusher.getInputSamples() * 2mAudioRecord = AudioRecord(MediaRecorder.AudioSource.MIC,sampleRateInHz,channelConfig,AudioFormat.ENCODING_PCM_16BIT,max(minBufferSize, mInputSamples) // 缓冲区取大的,如果取小了数据不完整)}fun startLive() {mIsLiving = truemExecutor.submit {// 开始录音mAudioRecord?.startRecording()// 录音缓冲区val bytes = ByteArray(mInputSamples)var len: Intwhile (mIsLiving) {len = mAudioRecord?.read(bytes, 0, bytes.size) ?: 0if (len > 0) {mLivePusher.nativePushAudio(bytes)}}mAudioRecord?.stop()}}fun stopLive() {mIsLiving = false}fun release() {mAudioRecord?.release()mAudioRecord = null}
}

3.2 Native 层

上层的 AudioChannel 在 init 时调用了 LivePusher 的 nativeInitAudioEncoder() 对音频编码器进行初始化,这项工作落到 Native 层的 AudioChannel 上:

void AudioChannel::initAudioEncoder(int sample_rate_in_hz, int channels) {this->channels = channels;// 1.打开 faac 编码器。前两个是入参,后两个是出参audioEncoder = faacEncOpen(sample_rate_in_hz, channels, &inputSamples, &maxOutputBytes);if (!audioEncoder) {LOGE("打开 faac 编码器失败");return;}// 2.配置编码器参数faacEncConfigurationPtr config = faacEncGetCurrentConfiguration(audioEncoder);// 使用 MPEG4 版本编码config->mpegVersion = MPEG4;// LC 标准,因为要求的质量不高,因此编解码较快config->aacObjectType = LOW;// 16 位config->inputFormat = FAAC_INPUT_16BIT;// 比特流输出格式为 Raw,设置为 1 的话就是 ADTSconfig->outputFormat = 0;// 开启 Temporal Noise Shaping,该音频编码技术用于减小编码后的音频中的噪声config->useTns = 1;// 禁用 Low-Frequency Effect 低频效果config->useLfe = 0;// 3.把参数设置给编码器if (!faacEncSetConfiguration(audioEncoder, config)) {LOGE("FAAC 音频编码器参数配置失败");return;}LOGD("FAAC 音频编码器初始化成功");buffer = new u_char(maxOutputBytes);
}

随后,上层的 AudioChannel 采集到一帧音频,就会调用 LivePusher 的 nativePushAudio() 让 Native 层编码:

extern "C"
JNIEXPORT void JNICALL
Java_com_rtmp_pusher_LivePusher_nativePushAudio(JNIEnv *env, jobject thiz, jbyteArray data_) {if (!audioChannel || !readyPushing) {return;}jbyte *data = env->GetByteArrayElements(data_, nullptr);audioChannel->encodeData(data);env->ReleaseByteArrayElements(data_, data, 0);
}

具体的编码工作还是由 Native 的 AudioChannel 完成:

void AudioChannel::encodeData(int8_t *data) {// 1.编码音频数据int byteLen = faacEncEncode(audioEncoder, reinterpret_cast<int32_t *>(data),inputSamples, buffer, maxOutputBytes);if (byteLen > 0) {// 2.实例化 RTMPPacket 并分配空间auto packet = new RTMPPacket;int body_size = 2 + byteLen;RTMPPacket_Alloc(packet, body_size);// 3.设置 RTMPPacket 内容// 声道数if (channels == 2) {packet->m_body[0] = 0xAF;} else {packet->m_body[0] = 0xAE;}// 编码出的音频数据都是 0x01packet->m_body[1] = 0x01;// 音频数据memcpy(&packet->m_body[2], buffer, byteLen);// 4.封包,与视频封包类似// 包类型为音频packet->m_packetType = RTMP_PACKET_TYPE_AUDIO;packet->m_nBodySize = body_size;packet->m_nChannel = 11;// 音频帧数据有时间戳packet->m_nTimeStamp = -1;// 一般不用绝对时间戳packet->m_hasAbsTimestamp = 0;// 帧数据类型一般是大包,如果是头信息,可以给一个中包或小包packet->m_headerType = RTMP_PACKET_SIZE_LARGE;// 5.回调加入队列audioCallback(packet);}
}

由于 RTMPPacket 本身是不区分音频与视频的,也就是说音视频数据编码完成后都可以存入 RTMPPacket 然后通过回调添加到 native-lib 的 RTMPPacket 队列中。最后的 audioCallback 在讲视频初始化时已经写过,回调函数与视频通道的回调函数相同,都是 native-lib 的 callback():

void callback(RTMPPacket *packet) {if (packet) {if (packet->m_nTimeStamp == -1) {// I 帧需要时间戳,但是 SPS 和 PPS 本身没有时间戳也就不需要packet->m_nTimeStamp = RTMP_GetTime() - start_time;}packets.put(packet);}
}

此外,回看视频推流一节中的 task_start(),在 while 循环开始发送 RTMPPacket 前,有通过 callback() 向队列中发送一个音频头:

	// 获取并将音频头存入队列,实测不用放这个头也是可以成功将音频推流的callback(audioChannel->getAudioSeqHeader());

当然实测时发现没有音频头拉流端也能正常播放音频。音频头具体设置如下:

RTMPPacket *AudioChannel::getAudioSeqHeader() {u_char *ppBuffer;// 这个 len 对于头而言固定为 2u_long len;faacEncGetDecoderSpecificInfo(audioEncoder, &ppBuffer, &len);auto packet = new RTMPPacket;int body_size = 2 + len;RTMPPacket_Alloc(packet, body_size);// 3.设置 RTMPPacket 内容// 声道数if (channels == 2) {packet->m_body[0] = 0xAF;} else {packet->m_body[0] = 0xAE;}// 序列头为 0x00packet->m_body[1] = 0x00;// 音频数据memcpy(&packet->m_body[2], ppBuffer, len);// 4.封包,与视频封包类似// 包类型为音频packet->m_packetType = RTMP_PACKET_TYPE_AUDIO;packet->m_nBodySize = body_size;packet->m_nChannel = 11;// 音频帧的头一般都是没有时间搓的packet->m_nTimeStamp = 0;// 一般不用绝对时间戳packet->m_hasAbsTimestamp = 0;// 帧数据类型一般是大包,如果是头信息,可以给一个中包或小包packet->m_headerType = RTMP_PACKET_SIZE_LARGE;return packet;
}

4、释放与测试结果

释放工作主要是 Native 层的各种成员变量。

首先 native-lib 中要删除两个 Channel 并等待线程 pid_start 执行完毕:

extern "C"
JNIEXPORT void JNICALL
Java_com_rtmp_pusher_LivePusher_nativeStop(JNIEnv *env, jobject thiz) {isStart = false;readyPushing = false;packets.setEnable(false);// 等待 pid_start 线程执行完再做后续的善后工作pthread_join(pid_start, nullptr);
}extern "C"
JNIEXPORT void JNICALL
Java_com_rtmp_pusher_LivePusher_nativeRelease(JNIEnv *env, jobject thiz) {DELETE(videoChannel)DELETE(audioChannel)
}

VideoChannel 的析构函数要销毁同步锁、关闭编码器:

VideoChannel::~VideoChannel() {pthread_mutex_destroy(&mutex);releaseVideoCodec();videoCallback = nullptr;
}void VideoChannel::releaseVideoCodec() {if (videoEncoder) {// 释放通过 x264_picture_alloc 分配内存的图片的相关资源x264_picture_clean(pic_in);// 关闭编码器x264_encoder_close(videoEncoder);// 释放编码器结构体,通过该函数将结构体内部的字段置空x264_encoder_parameters(videoEncoder, nullptr);videoEncoder = nullptr;}
}

AudioChannel 需要关闭音频编码器,回收 buffer:

AudioChannel::~AudioChannel() {DELETE(buffer)if (audioEncoder) {faacEncClose(audioEncoder);audioEncoder = nullptr;}audioCallback = nullptr;
}

最终的演示效果,手机端推流:

2024-4-22.直播效果4

拉流端通过 ffplay 拉流:

ffplay -i rtmp://xxx.xx.xx.xx/myapp

效果图:

2024-4-22.直播效果2

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:http://www.mzph.cn/news/831010.shtml

如若内容造成侵权/违法违规/事实不符,请联系多彩编程网进行投诉反馈email:809451989@qq.com,一经查实,立即删除!

相关文章

vue计算属性是什么 怎么选择?

写在computed对象中的属性&#xff0c;本质上是一个方法&#xff0c;不过使用时依旧当属性来使用 虽然模板语法使用非常便利&#xff0c;但是它是被设计成用于简单运算的。在模板中放入太多的逻辑会让模板过重且难以维护 无论计算属性还是methos&#xff0c;这两种实现方式的…

抽象类和接口的区别你知道吗

抽象类和接口是Java中两种重要的机制&#xff0c;它们都用于实现多态性和代码的灵活性&#xff0c;但在某些方面有着不同的特点。下面我将详细介绍抽象类和接口的区别&#xff0c;并通过示例进行说明。 抽象类&#xff08;Abstract Class&#xff09; 定义&#xff1a; 抽象类…

暴雨服务器引领信创算力新潮流

去年大模型的空前发展&#xff0c;人工智能也终于迎来了属于自己的“文艺复兴”&#xff0c;众多的模型相继发布&#xff0c;继而催生了整个行业对于智能算力需求的激增。 市场需求与技术驱动仿佛现实世界的左右脚&#xff0c;催动着世界文明的齿轮向前滚动。在全球经济角逐日…

力扣:61. 旋转链表(Java,双指针)

目录 题目描述&#xff1a;输入&#xff1a;输出&#xff1a;代码描述&#xff1a; 题目描述&#xff1a; 给你一个链表的头节点 head &#xff0c;旋转链表&#xff0c;将链表每个节点向右移动 k 个位置。 输入&#xff1a; head [1,2,3,4,5], k 2 输出&#xff1a; [4,5,1,…

Win系统常用操作

文章目录 1、常用快捷键 1、常用快捷键 锁屏&#xff1a;WinL截屏&#xff1a;WInShiftS新建文件夹&#xff1a;CtrlShiftN关闭程序&#xff1a;CtrlW放大/缩小页面&#xff1a;Ctrl加号/减号最小化当前窗口&#xff1a;Alt空格N返回桌面&#xff1a;WinD&#xff08;再次按Wi…

linux高性能服务器--定长内存池设计

内存池 内存池是指程序预先从操作系统申请一块足够大内存&#xff0c;此后当程序中需要再次申请内存的时候&#xff0c;直接从内存池中获取&#xff1b;当程序释放内存的时候&#xff0c;是返回给内存池保管。 开源内存池&#xff1a; tcmalloc 功能&#xff1a; 避免频繁分…

扒开kafka内部组件,咱瞅一瞅都有啥?

Apache Kafka 是一个分布式流处理平台&#xff0c;主要用于构建实时数据管道和流式应用程序。它由几个核心组件组成&#xff0c;这些组件共同工作以提供高吞吐量、持久性、容错性和可扩展性。关于 Kafka 与其它类似的中间件的对比&#xff0c;这里V 哥就不再阐述了&#xff0c;…

leetcode-滑动窗口的最大值-95

题目要求 思路 1.这个题是可以暴力求解的&#xff0c;但是时间复杂度比较高&#xff0c;因此&#xff0c;这里说一个时间复杂度为O(n)的方法 2.因为这个代码是优化后的结果&#xff0c;第一次写如果直接写成这样着实不容易&#xff0c;因此&#xff0c;我直接讲每一行的含义。…

Linux系统应用与设置(1):比较、生成补丁以及应用补丁(diff与patch指令)

1. 简述 在我们应用Linux系统或基于Linux系统做一些软件应用开发时&#xff0c;经常会遇到需要将文件的改动记录下来&#xff0c;并且讲这些改动和差异应用到其他的Linux系统时&#xff0c;我们会用到diff和patch指令。 在Linux中&#xff0c;diff和patch是两个非常有用的命令行…

【网络基础】深入理解TCP协议:协议段、可靠性、各种机制

文章目录 1. TCP协议段格式1.1. 如何解包 / 向上交付1.1.1. 交付1.1.2. 解包 1.2. 如何理解可靠性1.2.1. 确认应答机制&#xff08;ACK&#xff09;1.2.2. 序号 与 确认序号 2. TCP做到全双工的原因2.1. 16位窗口大小2.2. 6个标记位 3. 如何理解连接3.1 连接管理机制3.1.1. 三次…

44. UE5 RPG 初始化敌人的属性

在正常的游戏中&#xff0c;我们应该考虑如何去初始化角色属性&#xff0c;并且要给角色分好类型。比如&#xff0c;在我们游戏中&#xff0c;我们如何去初始化小兵的属性&#xff0c;并且还要实现小兵随着等级的增长而增加属性。而且就是小兵也有类型的区分&#xff0c;比如我…

【数组】Leetcode 26. 删除有序数组中的重复项【简单】

删除有序数组中的重复项 其他算法导航栏 给你一个 非严格递增排列 的数组 nums &#xff0c;请你 原地 删除重复出现的元素&#xff0c;使每个元素 只出现一次 &#xff0c;返回删除后数组的新长度。元素的 相对顺序 应该保持 一致 。然后返回 nums 中唯一元素的个数。 考虑 …

【C++STL】String类的常用函数用法总结

String类 一&#xff1a;String类的初始化方式二&#xff1a;String类的访问方式三&#xff1a;String的大小和容量四&#xff1a;string的插入&#xff1a;push_back() 和 insert()五&#xff1a;string拼接字符串&#xff1a;append() & 操作符六&#xff1a; string的删…

PhpAdmin-getshell

PhpAdmin-getshell 通过未授权成功写入&#xff0c;然后getshell 路径&#xff1a;C:\phpstudy_pro\Extensions\MySQL5.7.26\ 写入木马&#xff1a; into写入文件&#xff1a; 使用需看要secure_file_priv的值。 当value为“null”时&#xff0c;不允许读取任意文件 当value为…

Android 文件传输

经常写adb命令传文件&#xff0c;结果发现Android studio有自带的文件管理器&#xff0c;可以上传下载文件。

高扬程消防水泵在火灾中的关键作用/恒峰智慧科技

在火灾这一无情的灾难面前&#xff0c;每一秒都至关重要。而在这一分一秒的较量中&#xff0c;高扬程消防水泵无疑扮演着举足轻重的角色。它不仅是灭火战斗的得力助手&#xff0c;更是保障人民生命财产安全的守护神。 高扬程消防水泵&#xff0c;顾名思义&#xff0c;其扬程远超…

Gson打印按照想要的key顺序

默认大家都知道这个吧&#xff1f; val gson GsonBuilder().setPrettyPrinting().create() log(gson.toJson(bean))它是用于将对象bean&#xff0c;转成json以后&#xff0c;能够比较漂亮的打印出json的结构。我常用的是如下4个函数。 //就是jsonStr&#xff0c;使用该函数来…

qt对话框功能介绍

1、颜色对话框 //方式一QColor color QColorDialog::getColor(Qt::red, this, QString::fromLocal8Bit("颜色对话框"),QColorDialog::ShowAlphaChannel);qDebug() <<"color:" <<color;//方式二QColorDialog dialog(Qt::red, this); // 创建对…

通过自然语言处理执行特定任务的AI Agents;大模型控制NPC执行一系列的动作;个人化的电子邮件助手Panza

✨ 1: OpenAgents 通过自然语言处理执行特定任务的AI代理 OpenAgents是一个开放平台&#xff0c;旨在使语言代理&#xff08;即通过自然语言处理执行特定任务的AI代理&#xff09;的使用和托管变得更加便捷和实用。它特别适合于日常生活中对数据分析、工具插件获取和网络浏览…

vue2编写主体页面

目录 一. 导入两张图片 二. 新建主体vue 三. 修改路由 1. 新增主体界面Main.vue的路由 2. 完整router/index.js代码如下: 在Vue 2中编写一个主体页面通常意味着创建一个包含导航栏、侧边栏、内容区域等的布局。以下是使用Vue 2和Element UI框架来构建一个简单的主体页面的…