c++ 使用ffmpeg实现rtsp推流代码

以下是使用FFmpeg实现RTSP推流的C++代码示例：

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <signal.h>
#include <pthread.h>
#include <errno.h>

#include <libavcodec/avcodec.h>
#include <libavformat/avformat.h>
#include <libswscale/swscale.h>

const char* RTSP_URL = "rtsp://192.168.1.1:8554/test"; // 要推流的RTSP地址
const int FRAME_RATE = 25; // 视频帧率
const int VIDEO_WIDTH = 640; // 视频宽度
const int VIDEO_HEIGHT = 480; // 视频高度

int64_t start_time = 0;

int interrupt_cb(void* ctx) {
    int timeout = 10;
    if (av_gettime_relative() - start_time > timeout * 1000 * 1000) {
        return 1;
    }
    return 0;
}

void* push_thread(void* arg) {
    AVFormatContext* fmt_ctx = NULL;
    AVStream* video_stream = NULL;
    AVCodecContext* codec_ctx = NULL;
    AVCodec* codec = NULL;
    AVFrame* frame = NULL;
    AVPacket pkt;
    int ret = 0;

    avformat_network_init();

    // 打开输出RTSP流的上下文
    avformat_alloc_output_context2(&fmt_ctx, NULL, "rtsp", RTSP_URL);
    if (!fmt_ctx) {
        printf("avformat_alloc_output_context2 failed\n");
        goto end;
    }

    // 找到h.264编码器
    codec = avcodec_find_encoder_by_name("libx264");
    if (!codec) {
        printf("avcodec_find_encoder_by_name failed\n");
        goto end;
    }

    // 创建视频流
    video_stream = avformat_new_stream(fmt_ctx, codec);
    if (!video_stream) {
        printf("avformat_new_stream failed\n");
        goto end;
    }
    video_stream->codecpar->codec_id = codec->id;
    video_stream->codecpar->codec_type = AVMEDIA_TYPE_VIDEO;
    video_stream->codecpar->width = VIDEO_WIDTH;
    video_stream->codecpar->height = VIDEO_HEIGHT;
    video_stream->codecpar->format = AV_PIX_FMT_YUV420P;
    video_stream->codecpar->bit_rate = 500000;
    video_stream->codecpar->fps_num = FRAME_RATE;
    video_stream->codecpar->fps_den = 1;

    // 打开编码器
    codec_ctx = avcodec_alloc_context3(codec);
    if (!codec_ctx) {
        printf("avcodec_alloc_context3 failed\n");
        goto end;
    }
    avcodec_parameters_to_context(codec_ctx, video_stream->codecpar);
    if (avcodec_open2(codec_ctx, codec, NULL) < 0) {
        printf("avcodec_open2 failed\n");
        goto end;
    }

    // 创建帧
    frame = av_frame_alloc();
    if (!frame) {
        printf("av_frame_alloc failed\n");
        goto end;
    }
    frame->format = codec_ctx->pix_fmt;
    frame->width = VIDEO_WIDTH;
    frame->height = VIDEO_HEIGHT;
    if (av_frame_get_buffer(frame, 32) < 0) {
        printf("av_frame_get_buffer failed\n");
        goto end;
    }

    // 打开输出流
    if (avio_open(&fmt_ctx->pb, RTSP_URL, AVIO_FLAG_WRITE) < 0) {
        printf("avio_open failed\n");
        goto end;
    }

    // 写输出流头部
    avformat_write_header(fmt_ctx, NULL);

    // 推流
    while (1) {
        // 生成测试图像
        uint8_t* data[1];
        int linesize[1];
        int y_size = VIDEO_WIDTH * VIDEO_HEIGHT;
        data[0] = (uint8_t*)malloc(y_size * 3 / 2);
        memset(data[0], 0, y_size * 3 / 2);
        for (int i = 0; i < VIDEO_HEIGHT; i++) {
            memset(data[0] + i * VIDEO_WIDTH, i * 255 / (VIDEO_HEIGHT - 1), VIDEO_WIDTH);
        }
        for (int i = 0; i < VIDEO_HEIGHT / 2; i++) {
            memset(data[0] + y_size + i * VIDEO_WIDTH / 2, 128 + i * 127 / (VIDEO_HEIGHT / 2 - 1), VIDEO_WIDTH / 2);
        }

        // 将测试图像转换为AVFrame
        av_image_fill_arrays(frame->data, frame->linesize, data[0], codec_ctx->pix_fmt, VIDEO_WIDTH, VIDEO_HEIGHT, 32);
        frame->pts = av_rescale_q(av_gettime_relative() - start_time, (AVRational){1, AV_TIME_BASE}, video_stream->time_base);
        ret = avcodec_send_frame(codec_ctx, frame);
        if (ret < 0) {
            printf("avcodec_send_frame failed\n");
            goto end;
        }
        while (ret >= 0) {
            ret = avcodec_receive_packet(codec_ctx, &pkt);
            if (ret < 0) {
                break;
            }
            av_packet_rescale_ts(&pkt, codec_ctx->time_base, video_stream->time_base);
            pkt.stream_index = video_stream->index;
            av_interleaved_write_frame(fmt_ctx, &pkt);
            av_packet_unref(&pkt);
        }

        free(data[0]);

        if (av_gettime_relative() - start_time > 30 * 1000 * 1000) { // 推流30秒后退出
            break;
        }
    }

    // 写输出流尾部
    av_write_trailer(fmt_ctx);

end:
    if (frame) {
        av_frame_free(&frame);
    }
    if (codec_ctx) {
        avcodec_free_context(&codec_ctx);
    }
    if (fmt_ctx) {
        avio_close(fmt_ctx->pb);
        avformat_free_context(fmt_ctx);
    }

    return NULL;
}

int main(int argc, char* argv[]) {
    pthread_t pid;
    int ret = 0;

    // 初始化FFmpeg库
    av_register_all();
    avformat_network_init();
    avcodec_register_all();

    start_time = av_gettime_relative();

    // 创建推流线程
    ret = pthread_create(&pid, NULL, push_thread, NULL);
    if (ret != 0) {
        printf("pthread_create failed\n");
        return -1;
    }

    // 等待推流线程退出
    pthread_join(pid, NULL);

    return 0;
}

上述代码中使用libx264编码器，生成测试图像并将其推流到RTSP服务器。可以根据实际需要修改RTSP_URL、FRAME_RATE、VIDEO_WIDTH和VIDEO_HEIGHT等参数。