使用ffmpeg实现rtsp推流

以下是使用FFmpeg实现RTSP推流的C++代码示例： ```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等参数。

以下是使用FFmpeg实现RTSP推流的示例C代码： ``` #include <stdio.h> #include <stdlib.h> #include <string.h> #include <unistd.h> #include <libavformat/avformat.h> #include <libavutil/opt.h> int main(int argc, char **argv) { AVFormatContext *fmt_ctx = NULL; AVOutputFormat *out_fmt = NULL; AVStream *video_st = NULL; AVCodecContext *codec_ctx = NULL; AVCodec *codec = NULL; AVDictionary *options = NULL; int ret; const char *rtsp_url = "rtsp://your_rtsp_server_url"; const char *out_filename = "rtsp_stream_output"; // Register all available codecs and formats av_register_all(); avformat_network_init(); // Allocate output format context avformat_alloc_output_context2(&fmt_ctx, NULL, "rtsp", out_filename); if (!fmt_ctx) { fprintf(stderr, "Could not create output format context\n"); return -1; } // Find video encoder codec codec = avcodec_find_encoder_by_name("h264"); if (!codec) { fprintf(stderr, "Could not find video encoder codec\n"); return -1; } // Allocate codec context and set options codec_ctx = avcodec_alloc_context3(codec); if (!codec_ctx) { fprintf(stderr, "Could not allocate codec context\n"); return -1; } codec_ctx->bit_rate = 400000; codec_ctx->width = 640; codec_ctx->height = 480; codec_ctx->time_base = (AVRational){1, 25}; codec_ctx->gop_size = 10; codec_ctx->pix_fmt = AV_PIX_FMT_YUV420P; av_opt_set(codec_ctx->priv_data, "preset", "ultrafast", 0); av_opt_set(codec_ctx->priv_data, "tune", "zerolatency", 0); // Open codec if ((ret = avcodec_open2(codec_ctx, codec, &options)) < 0) { fprintf(stderr, "Could not open codec: %s\n", av_err2str(ret)); return -1; } // Add video stream to output format context video_st = avformat_new_stream(fmt_ctx, codec); if (!video_st) { fprintf(stderr, "Could not create new video stream\n"); return -1; } avcodec_parameters_from_context(video_st->codecpar, codec_ctx); // Open output file or URL if ((ret = avio_open(&fmt_ctx->pb, rtsp_url, AVIO_FLAG_WRITE)) < 0) { fprintf(stderr, "Could not open output file or URL: %s\n", av_err2str(ret)); return -1; } // Write header to output format context if ((ret = avformat_write_header(fmt_ctx, &options)) < 0) { fprintf(stderr, "Could not write header to output format context: %s\n", av_err2str(ret)); return -1; } // Main loop to write video frames for (int i = 0; i < 1000; i++) { AVFrame *frame = av_frame_alloc(); if (!frame) { fprintf(stderr, "Could not allocate video frame\n"); return -1; } // Fill video frame with data // ... // Set frame PTS and duration frame->