21xrx.com
2025-04-07 17:02:04 Monday
文章检索 我的文章 写文章
使用ffmpeg进行UDP推流的代码示例
2023-08-14 01:44:09 深夜i     40     0
ffmpeg UDP推流 代码示例

在视频流媒体领域，使用UDP（User Datagram Protocol）进行推流是一种常见的方法。其中，FFmpeg是一个功能强大的开源多媒体框架，它可以在各种平台上进行音视频的采集、编码、解码、传输等操作。

下面是一个使用FFmpeg进行UDP推流的代码示例：

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <libavcodec/avcodec.h>
#include <libavformat/avformat.h>
#include <libswscale/swscale.h>
#define OUTPUT_URL "udp://localhost:1234" // UDP推流目标地址
int main()
{
  // 注册所有的编解码器和格式
  av_register_all();
  AVFormatContext *formatContext = NULL;
  AVOutputFormat *outputFormat = NULL;
  AVStream *videoStream = NULL;
  AVCodecContext *codecContext = NULL;
  AVCodec *codec = NULL;
  AVFrame *frame = NULL;
  AVPacket packet;
  // 创建输出格式上下文
  avformat_alloc_output_context2(&formatContext, NULL, "mpegts", OUTPUT_URL);
  if (!formatContext) {
    printf("Could not create output context\n");
    return -1;
  }
  // 添加视频流
  outputFormat = formatContext->oformat;
  videoStream = avformat_new_stream(formatContext, NULL);
  if (!videoStream) {
    printf("Failed allocating output stream\n");
    return -1;
  }
  codecContext = videoStream->codec;
  codecContext->codec_id = outputFormat->video_codec;
  codecContext->codec_type = AVMEDIA_TYPE_VIDEO;
  codecContext->pix_fmt = AV_PIX_FMT_YUV420P;
  codecContext->width = 640;
  codecContext->height = 480;
  codecContext->time_base.num = 1;
  codecContext->time_base.den = 25;
  codecContext->bit_rate = 400000;
  codec = avcodec_find_encoder(codecContext->codec_id);
  if (!codec) {
    printf("Codec not found\n");
    return -1;
  }
  // 打开编码器
  if (avcodec_open2(codecContext, codec, NULL) < 0) {
    printf("Could not open codec\n");
    return -1;
  }
  // 分配视频帧内存和设置属性
  frame = av_frame_alloc();
  frame->format = codecContext->pix_fmt;
  frame->width = codecContext->width;
  frame->height = codecContext->height;
  if (av_frame_get_buffer(frame, 0) < 0) {
    printf("Could not allocate the video frame data\n");
    return -1;
  }
  // 打开输出URL
  if (avio_open(&formatContext->pb, OUTPUT_URL, AVIO_FLAG_WRITE) < 0) {
    printf("Could not open output URL\n");
    return -1;
  }
  // 写入输出文件头
  if (avformat_write_header(formatContext, NULL) < 0) {
    printf("Error occurred when opening output URL\n");
    return -1;
  }
  // 准备数据并发送
  AVFrame *tmpFrame = av_frame_alloc();
  uint8_t *buffer = (uint8_t *) malloc(av_image_get_buffer_size(codecContext->pix_fmt, codecContext->width, codecContext->height, 1));
  av_image_fill_arrays(tmpFrame->data, tmpFrame->linesize, buffer, codecContext->pix_fmt, codecContext->width, codecContext->height, 1);
  int frameIndex = 0;
  while (frameIndex < 100) {
    // 生成测试图像
    for (int y = 0; y < codecContext->height; y++) {
      for (int x = 0; x < codecContext->width; x++) {
        tmpFrame->data[0][y * tmpFrame->linesize[0] + x] = x + y + frameIndex * 3;
      }
    }
    // 编码图像
    AVPacket packet;
    av_init_packet(&packet);
    packet.data = NULL;
    packet.size = 0;
    int ret = avcodec_send_frame(codecContext, tmpFrame);
    if (ret < 0) {
      printf("Error sending frame for encoding\n");
      return -1;
    }
    while (ret >= 0) {
      ret = avcodec_receive_packet(codecContext, &packet);
      if (ret == AVERROR(EAGAIN) || ret == AVERROR_EOF) {
        break;
      } else if (ret < 0) {
        printf("Error during encoding\n");
        return -1;
      }
      // 发送数据包
      av_write_frame(formatContext, &packet);
    }
    frameIndex++;
  }
  // 写入输出文件尾
  av_write_trailer(formatContext);
  // 清理资源
  avcodec_free_context(&codecContext);
  av_frame_free(&frame);
  av_frame_free(&tmpFrame);
  free(buffer);
  avio_close(formatContext->pb);
  avformat_free_context(formatContext);
  return 0;
}

以上示例代码演示了在C语言下使用FFmpeg进行UDP推流的过程。首先，我们注册所有的编解码器和格式。然后，创建输出格式上下文和视频流，并设置视频流的编解码器和相关属性。接着，打开编码器并分配视频帧内存。然后，打开输出URL，并写入输出文件头。接下来，准备图像数据并发送。最后，写入输出文件尾，并清理相关资源。

通过以上代码示例，我们可以了解到如何使用FFmpeg进行UDP推流操作，为视频流媒体提供了一个基础的参考。当然，根据实际需求，具体的参数和配置可能会有所不同。因此，在实际应用中，需要根据具体情况进行相应的调整和优化。

  
  

评论区

请求出错了