mirror of
https://github.com/obsproject/obs-studio.git
synced 2026-09-17 17:16:40 +08:00
pkv
1451554fb8
Fixes #10504. There was a bug in FFmpeg implementation which was hidden by a bug in libsrt; it was fixed in a recent commit [1]. When we ported FFmpeg libsrt.c to obs, we brought the said bug along. When starting an SRT stream in listener mode, if no connection is made by a client, there were two issues: - 1) obs was stuck into a connecting loop, - 2) the socket was not closed when exiting OBS. This fixes the issue so that SRT is displaying that a stream started when in listener mode even if NO client is connected. This is the correct behaviour for a listener. The stream now closes properly. [1] https://git.videolan.org/?p=ffmpeg.git;a=commit;h=87677c2195e86b126c3438439a05d0a46ae5bb50 Signed-off-by: pkv <pkv@obsproject.com>
121 lines
2.7 KiB
C
121 lines
2.7 KiB
C
#pragma once
|
|
|
|
#include <libavutil/opt.h>
|
|
#include <libavutil/pixdesc.h>
|
|
#include <libavcodec/avcodec.h>
|
|
#include <libavformat/avformat.h>
|
|
#include <libswscale/swscale.h>
|
|
#ifdef NEW_MPEGTS_OUTPUT
|
|
#include "obs-ffmpeg-url.h"
|
|
#endif
|
|
|
|
struct ffmpeg_cfg {
|
|
const char *url;
|
|
const char *format_name;
|
|
const char *format_mime_type;
|
|
const char *muxer_settings;
|
|
const char *protocol_settings; // not used yet for SRT nor RIST
|
|
int gop_size;
|
|
int video_bitrate;
|
|
int audio_bitrate;
|
|
const char *video_encoder;
|
|
int video_encoder_id;
|
|
const char *audio_encoder;
|
|
int audio_encoder_id;
|
|
int audio_bitrates[MAX_AUDIO_MIXES]; // multi-track
|
|
const char *video_settings;
|
|
const char *audio_settings;
|
|
int audio_mix_count;
|
|
int audio_tracks;
|
|
const char *audio_stream_names[MAX_AUDIO_MIXES];
|
|
enum AVPixelFormat format;
|
|
enum AVColorRange color_range;
|
|
enum AVColorPrimaries color_primaries;
|
|
enum AVColorTransferCharacteristic color_trc;
|
|
enum AVColorSpace colorspace;
|
|
int max_luminance;
|
|
int scale_width;
|
|
int scale_height;
|
|
int width;
|
|
int height;
|
|
int frame_size; // audio frame size
|
|
const char *username;
|
|
const char *password;
|
|
const char *stream_id;
|
|
const char *encrypt_passphrase;
|
|
bool is_srt;
|
|
bool is_rist;
|
|
};
|
|
|
|
struct ffmpeg_audio_info {
|
|
AVStream *stream;
|
|
AVCodecContext *ctx;
|
|
};
|
|
|
|
struct ffmpeg_data {
|
|
AVStream *video;
|
|
AVCodecContext *video_ctx;
|
|
struct ffmpeg_audio_info *audio_infos;
|
|
const AVCodec *acodec;
|
|
const AVCodec *vcodec;
|
|
AVFormatContext *output;
|
|
struct SwsContext *swscale;
|
|
|
|
int64_t total_frames;
|
|
AVFrame *vframe;
|
|
int frame_size;
|
|
|
|
uint64_t start_timestamp;
|
|
|
|
int64_t total_samples[MAX_AUDIO_MIXES];
|
|
uint32_t audio_samplerate;
|
|
enum audio_format audio_format;
|
|
size_t audio_planes;
|
|
size_t audio_size;
|
|
int num_audio_streams;
|
|
|
|
/* audio_tracks is a bitmask storing the indices of the mixes */
|
|
int audio_tracks;
|
|
struct deque excess_frames[MAX_AUDIO_MIXES][MAX_AV_PLANES];
|
|
uint8_t *samples[MAX_AUDIO_MIXES][MAX_AV_PLANES];
|
|
AVFrame *aframe[MAX_AUDIO_MIXES];
|
|
|
|
struct ffmpeg_cfg config;
|
|
|
|
bool initialized;
|
|
|
|
char *last_error;
|
|
};
|
|
|
|
struct ffmpeg_output {
|
|
obs_output_t *output;
|
|
volatile bool active;
|
|
struct ffmpeg_data ff_data;
|
|
|
|
bool connecting;
|
|
pthread_t start_thread;
|
|
|
|
uint64_t total_bytes;
|
|
|
|
uint64_t audio_start_ts;
|
|
uint64_t video_start_ts;
|
|
uint64_t stop_ts;
|
|
volatile bool stopping;
|
|
|
|
bool write_thread_active;
|
|
pthread_mutex_t write_mutex;
|
|
pthread_t write_thread;
|
|
os_sem_t *write_sem;
|
|
os_event_t *stop_event;
|
|
|
|
DARRAY(AVPacket *) packets;
|
|
#ifdef NEW_MPEGTS_OUTPUT
|
|
/* used for SRT & RIST */
|
|
URLContext *h;
|
|
AVIOContext *s;
|
|
bool got_headers;
|
|
#endif
|
|
};
|
|
bool ffmpeg_data_init(struct ffmpeg_data *data, struct ffmpeg_cfg *config);
|
|
void ffmpeg_data_free(struct ffmpeg_data *data);
|