当前位置：首页 > news >正文

FFmpeg实现rtp推流

news 来源：原创 2024/5/14 14:41:07

以下是一个简单的示例代码，演示了如何使用 UDP 或 TCP 进行音视频传输：
代码未经验证，供参考

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <libavformat/avformat.h>#define RTPPKT_LEN 1400
#define IPOUT_PAYNUM 1// UDP Socket
int udpSocket;// TCP Socket
int tcpSocket;// UDP 发送函数
int udp_write_buffer(void* opaque, uint8_t* buf, int buf_size) {struct sockaddr_in destAddr;memset(&destAddr, 0, sizeof(destAddr));destAddr.sin_family = AF_INET;destAddr.sin_port = htons(1234); // 设置目标端口号destAddr.sin_addr.s_addr = inet_addr("127.0.0.1"); // 设置目标IP地址ssize_t sentBytes = sendto(udpSocket, buf, buf_size, 0, (struct sockaddr*)&destAddr, sizeof(destAddr));if (sentBytes < 0) {perror("UDP 发送失败");return -1;}return 0;
}// TCP 发送函数
int tcp_write_buffer(void* opaque, uint8_t* buf, int buf_size) {ssize_t sentBytes = send(tcpSocket, buf, buf_size, 0);if (sentBytes < 0) {perror("TCP 发送失败");return -1;}return 0;
}int main() {av_register_all();avformat_network_init();// 创建输入格式上下文AVFormatContext* inputFormatCtx = avformat_alloc_context();// 打开输入文件或网络流const char* inputUrl = "input.mp4";int ret = avformat_open_input(&inputFormatCtx, inputUrl, nullptr, nullptr);if (ret != 0) {printf("无法打开输入文件：%s\n", inputUrl);return -1;}// 获取流信息ret = avformat_find_stream_info(inputFormatCtx, nullptr);if (ret < 0) {printf("无法获取流信息\n");return -1;}// 寻找视频流索引int videoStreamIndex = -1;for (unsigned int i = 0; i < inputFormatCtx->nb_streams; i++) {if (inputFormatCtx->streams[i]->codecpar->codec_type == AVMEDIA_TYPE_VIDEO) {videoStreamIndex = i;break;}}// 创建输出格式上下文AVFormatContext* outputFormatCtx = nullptr;ret = avformat_alloc_output_context2(&outputFormatCtx, nullptr, "rtp", "rtp://127.0.0.1:1234"); // 设置输出 URLif (ret < 0) {printf("无法创建输出格式上下文\n");return -1;}// 创建 AVIO 缓存空间unsigned char* outBuffer = (unsigned char*)av_malloc(RTPPKT_LEN * IPOUT_PAYNUM);// 创建 UDP SocketudpSocket = socket(AF_INET, SOCK_DGRAM, 0);if (udpSocket < 0) {perror("无法创建 UDP Socket");return -1;}// 创建 TCP SockettcpSocket = socket(AF_INET, SOCK_STREAM, 0);if (tcpSocket < 0) {perror("无法创建 TCP Socket");return -1;}// 创建 AVIO 上下文AVIOContext* avioOut;if (strcmp("UDP", "RTP/AVP") == 0) {avioOut = avio_alloc_context(outBuffer, RTPPKT_LEN * IPOUT_PAYNUM, 1, nullptr, nullptr, udp_write_buffer, nullptr);} else if (strcmp("TCP", "RTP/AVP") == 0) {avioOut = avio_alloc_context(outBuffer, RTPPKT_LEN * IPOUT_PAYNUM, 1, nullptr, nullptr, tcp_write_buffer, nullptr);}if (!avioOut) {printf("无法创建 AVIO 上下文\n");return -1;}avioOut->max_packet_size = RTPPKT_LEN;outputFormatCtx->pb = avioOut;outputFormatCtx->flags = AVFMT_FLAG_CUSTOM_IO;// 复制视频流参数到输出格式上下文AVStream* outputStream = avformat_new_stream(outputFormatCtx, nullptr);ret = avcodec_parameters_copy(outputStream->codecpar, inputFormatCtx->streams[videoStreamIndex]->codecpar);if (ret < 0) {printf("无法复制视频流参数\n");return -1;}// 写入输出头部ret = avformat_write_header(outputFormatCtx, nullptr);if (ret < 0) {printf("无法写入输出头部\n");return -1;}// 开始读取和发送数据包AVPacket packet;while (av_read_frame(inputFormatCtx, &packet) >= 0) {if (packet.stream_index == videoStreamIndex) {// 发送数据包到输出上下文ret = av_interleaved_write_frame(outputFormatCtx, &packet);if (ret < 0) {printf("无法发送数据包\n");return -1;}}av_packet_unref(&packet);}// 写入输出尾部ret = av_write_trailer(outputFormatCtx);if (ret < 0) {printf("无法写入输出尾部\n");return -1;}// 关闭 Socketclose(udpSocket);close(tcpSocket);// 释放资源avformat_close_input(&inputFormatCtx);avformat_free_context(inputFormatCtx);avformat_free_context(outputFormatCtx);avio_context_free(&avioOut);av_free(outBuffer);return 0;
}