mirror of
https://github.com/obsproject/obs-studio.git
synced 2024-07-20 01:59:17 +00:00
247 lines
7.6 KiB
C
247 lines
7.6 KiB
C
#include <obs-module.h>
|
|
#include <util/platform.h>
|
|
#include <util/deque.h>
|
|
#include <util/dstr.h>
|
|
#include <util/threading.h>
|
|
#include <inttypes.h>
|
|
#include "librtmp/rtmp.h"
|
|
#include "librtmp/log.h"
|
|
#include "flv-mux.h"
|
|
#include "net-if.h"
|
|
|
|
#ifdef _WIN32
|
|
#include <Iphlpapi.h>
|
|
#else
|
|
#include <sys/ioctl.h>
|
|
#endif
|
|
|
|
#define do_log(level, format, ...) \
|
|
blog(level, "[rtmp stream: '%s'] " format, \
|
|
obs_output_get_name(stream->output), ##__VA_ARGS__)
|
|
|
|
#define warn(format, ...) do_log(LOG_WARNING, format, ##__VA_ARGS__)
|
|
#define info(format, ...) do_log(LOG_INFO, format, ##__VA_ARGS__)
|
|
#define debug(format, ...) do_log(LOG_DEBUG, format, ##__VA_ARGS__)
|
|
|
|
#define OPT_DYN_BITRATE "dyn_bitrate"
|
|
#define OPT_DROP_THRESHOLD "drop_threshold_ms"
|
|
#define OPT_PFRAME_DROP_THRESHOLD "pframe_drop_threshold_ms"
|
|
#define OPT_MAX_SHUTDOWN_TIME_SEC "max_shutdown_time_sec"
|
|
#define OPT_BIND_IP "bind_ip"
|
|
#define OPT_IP_FAMILY "ip_family"
|
|
#define OPT_NEWSOCKETLOOP_ENABLED "new_socket_loop_enabled"
|
|
#define OPT_LOWLATENCY_ENABLED "low_latency_mode_enabled"
|
|
#define OPT_METADATA_MULTITRACK "metadata_multitrack"
|
|
|
|
//#define TEST_FRAMEDROPS
|
|
//#define TEST_FRAMEDROPS_WITH_BITRATE_SHORTCUTS
|
|
|
|
#ifdef TEST_FRAMEDROPS
|
|
|
|
#define DROPTEST_MAX_KBPS 3000
|
|
#define DROPTEST_MAX_BYTES (DROPTEST_MAX_KBPS * 1000 / 8)
|
|
|
|
struct droptest_info {
|
|
uint64_t ts;
|
|
size_t size;
|
|
};
|
|
#endif
|
|
|
|
struct dbr_frame {
|
|
uint64_t send_beg;
|
|
uint64_t send_end;
|
|
size_t size;
|
|
};
|
|
|
|
struct rtmp_stream {
|
|
obs_output_t *output;
|
|
|
|
pthread_mutex_t packets_mutex;
|
|
struct deque packets;
|
|
bool sent_headers;
|
|
|
|
bool got_first_video;
|
|
int64_t start_dts_offset;
|
|
|
|
volatile bool connecting;
|
|
pthread_t connect_thread;
|
|
|
|
volatile bool active;
|
|
volatile bool disconnected;
|
|
volatile bool encode_error;
|
|
pthread_t send_thread;
|
|
|
|
int max_shutdown_time_sec;
|
|
|
|
os_sem_t *send_sem;
|
|
os_event_t *stop_event;
|
|
uint64_t stop_ts;
|
|
uint64_t shutdown_timeout_ts;
|
|
|
|
struct dstr path, key;
|
|
struct dstr username, password;
|
|
struct dstr encoder_name;
|
|
struct dstr bind_ip;
|
|
socklen_t addrlen_hint; /* hint IPv4 vs IPv6 */
|
|
|
|
/* frame drop variables */
|
|
int64_t drop_threshold_usec;
|
|
int64_t pframe_drop_threshold_usec;
|
|
int min_priority;
|
|
float congestion;
|
|
|
|
int64_t last_dts_usec;
|
|
|
|
uint64_t total_bytes_sent;
|
|
int dropped_frames;
|
|
|
|
#ifdef TEST_FRAMEDROPS
|
|
struct deque droptest_info;
|
|
uint64_t droptest_last_key_check;
|
|
size_t droptest_max;
|
|
size_t droptest_size;
|
|
#endif
|
|
|
|
pthread_mutex_t dbr_mutex;
|
|
struct deque dbr_frames;
|
|
size_t dbr_data_size;
|
|
uint64_t dbr_inc_timeout;
|
|
long audio_bitrate;
|
|
long dbr_est_bitrate;
|
|
long dbr_orig_bitrate;
|
|
long dbr_prev_bitrate;
|
|
long dbr_cur_bitrate;
|
|
long dbr_inc_bitrate;
|
|
bool dbr_enabled;
|
|
|
|
enum audio_id_t audio_codec[MAX_OUTPUT_AUDIO_ENCODERS];
|
|
enum video_id_t video_codec[MAX_OUTPUT_VIDEO_ENCODERS];
|
|
|
|
RTMP rtmp;
|
|
|
|
bool new_socket_loop;
|
|
bool low_latency_mode;
|
|
bool disable_send_window_optimization;
|
|
bool socket_thread_active;
|
|
pthread_t socket_thread;
|
|
uint8_t *write_buf;
|
|
size_t write_buf_len;
|
|
size_t write_buf_size;
|
|
pthread_mutex_t write_buf_mutex;
|
|
os_event_t *buffer_space_available_event;
|
|
os_event_t *buffer_has_data_event;
|
|
os_event_t *socket_available_event;
|
|
os_event_t *send_thread_signaled_exit;
|
|
};
|
|
|
|
#ifdef _WIN32
|
|
void *socket_thread_windows(void *data);
|
|
#endif
|
|
|
|
/* Adapted from FFmpeg's libavutil/pixfmt.h
|
|
*
|
|
* Renamed to make it apparent that these are not imported as this module does
|
|
* not use or link against FFmpeg.
|
|
*/
|
|
|
|
/**
|
|
* Chromaticity coordinates of the source primaries.
|
|
* These values match the ones defined by ISO/IEC 23091-2_2019 subclause 8.1 and ITU-T H.273.
|
|
*/
|
|
enum OBSColorPrimaries {
|
|
OBSCOL_PRI_RESERVED0 = 0,
|
|
OBSCOL_PRI_BT709 =
|
|
1, ///< also ITU-R BT1361 / IEC 61966-2-4 / SMPTE RP 177 Annex B
|
|
OBSCOL_PRI_UNSPECIFIED = 2,
|
|
OBSCOL_PRI_RESERVED = 3,
|
|
OBSCOL_PRI_BT470M =
|
|
4, ///< also FCC Title 47 Code of Federal Regulations 73.682 (a)(20)
|
|
OBSCOL_PRI_BT470BG =
|
|
5, ///< also ITU-R BT601-6 625 / ITU-R BT1358 625 / ITU-R BT1700 625 PAL & SECAM
|
|
OBSCOL_PRI_SMPTE170M =
|
|
6, ///< also ITU-R BT601-6 525 / ITU-R BT1358 525 / ITU-R BT1700 NTSC
|
|
OBSCOL_PRI_SMPTE240M =
|
|
7, ///< identical to above, also called "SMPTE C" even though it uses D65
|
|
OBSCOL_PRI_FILM = 8, ///< colour filters using Illuminant C
|
|
OBSCOL_PRI_BT2020 = 9, ///< ITU-R BT2020
|
|
OBSCOL_PRI_SMPTE428 = 10, ///< SMPTE ST 428-1 (CIE 1931 XYZ)
|
|
OBSCOL_PRI_SMPTEST428_1 = OBSCOL_PRI_SMPTE428,
|
|
OBSCOL_PRI_SMPTE431 = 11, ///< SMPTE ST 431-2 (2011) / DCI P3
|
|
OBSCOL_PRI_SMPTE432 =
|
|
12, ///< SMPTE ST 432-1 (2010) / P3 D65 / Display P3
|
|
OBSCOL_PRI_EBU3213 =
|
|
22, ///< EBU Tech. 3213-E (nothing there) / one of JEDEC P22 group phosphors
|
|
OBSCOL_PRI_JEDEC_P22 = OBSCOL_PRI_EBU3213,
|
|
OBSCOL_PRI_NB ///< Not part of ABI
|
|
};
|
|
|
|
/**
|
|
* Color Transfer Characteristic.
|
|
* These values match the ones defined by ISO/IEC 23091-2_2019 subclause 8.2.
|
|
*/
|
|
enum OBSColorTransferCharacteristic {
|
|
OBSCOL_TRC_RESERVED0 = 0,
|
|
OBSCOL_TRC_BT709 = 1, ///< also ITU-R BT1361
|
|
OBSCOL_TRC_UNSPECIFIED = 2,
|
|
OBSCOL_TRC_RESERVED = 3,
|
|
OBSCOL_TRC_GAMMA22 =
|
|
4, ///< also ITU-R BT470M / ITU-R BT1700 625 PAL & SECAM
|
|
OBSCOL_TRC_GAMMA28 = 5, ///< also ITU-R BT470BG
|
|
OBSCOL_TRC_SMPTE170M =
|
|
6, ///< also ITU-R BT601-6 525 or 625 / ITU-R BT1358 525 or 625 / ITU-R BT1700 NTSC
|
|
OBSCOL_TRC_SMPTE240M = 7,
|
|
OBSCOL_TRC_LINEAR = 8, ///< "Linear transfer characteristics"
|
|
OBSCOL_TRC_LOG =
|
|
9, ///< "Logarithmic transfer characteristic (100:1 range)"
|
|
OBSCOL_TRC_LOG_SQRT =
|
|
10, ///< "Logarithmic transfer characteristic (100 * Sqrt(10) : 1 range)"
|
|
OBSCOL_TRC_IEC61966_2_4 = 11, ///< IEC 61966-2-4
|
|
OBSCOL_TRC_BT1361_ECG = 12, ///< ITU-R BT1361 Extended Colour Gamut
|
|
OBSCOL_TRC_IEC61966_2_1 = 13, ///< IEC 61966-2-1 (sRGB or sYCC)
|
|
OBSCOL_TRC_BT2020_10 = 14, ///< ITU-R BT2020 for 10-bit system
|
|
OBSCOL_TRC_BT2020_12 = 15, ///< ITU-R BT2020 for 12-bit system
|
|
OBSCOL_TRC_SMPTE2084 =
|
|
16, ///< SMPTE ST 2084 for 10-, 12-, 14- and 16-bit systems
|
|
OBSCOL_TRC_SMPTEST2084 = OBSCOL_TRC_SMPTE2084,
|
|
OBSCOL_TRC_SMPTE428 = 17, ///< SMPTE ST 428-1
|
|
OBSCOL_TRC_SMPTEST428_1 = OBSCOL_TRC_SMPTE428,
|
|
OBSCOL_TRC_ARIB_STD_B67 =
|
|
18, ///< ARIB STD-B67, known as "Hybrid log-gamma"
|
|
OBSCOL_TRC_NB ///< Not part of ABI
|
|
};
|
|
|
|
/**
|
|
* YUV colorspace type.
|
|
* These values match the ones defined by ISO/IEC 23091-2_2019 subclause 8.3.
|
|
*/
|
|
enum OBSColorSpace {
|
|
OBSCOL_SPC_RGB =
|
|
0, ///< order of coefficients is actually GBR, also IEC 61966-2-1 (sRGB), YZX and ST 428-1
|
|
OBSCOL_SPC_BT709 =
|
|
1, ///< also ITU-R BT1361 / IEC 61966-2-4 xvYCC709 / derived in SMPTE RP 177 Annex B
|
|
OBSCOL_SPC_UNSPECIFIED = 2,
|
|
OBSCOL_SPC_RESERVED =
|
|
3, ///< reserved for future use by ITU-T and ISO/IEC just like 15-255 are
|
|
OBSCOL_SPC_FCC =
|
|
4, ///< FCC Title 47 Code of Federal Regulations 73.682 (a)(20)
|
|
OBSCOL_SPC_BT470BG =
|
|
5, ///< also ITU-R BT601-6 625 / ITU-R BT1358 625 / ITU-R BT1700 625 PAL & SECAM / IEC 61966-2-4 xvYCC601
|
|
OBSCOL_SPC_SMPTE170M =
|
|
6, ///< also ITU-R BT601-6 525 / ITU-R BT1358 525 / ITU-R BT1700 NTSC / functionally identical to above
|
|
OBSCOL_SPC_SMPTE240M =
|
|
7, ///< derived from 170M primaries and D65 white point, 170M is derived from BT470 System M's primaries
|
|
OBSCOL_SPC_YCGCO =
|
|
8, ///< used by Dirac / VC-2 and H.264 FRext, see ITU-T SG16
|
|
OBSCOL_SPC_YCOCG = OBSCOL_SPC_YCGCO,
|
|
OBSCOL_SPC_BT2020_NCL =
|
|
9, ///< ITU-R BT2020 non-constant luminance system
|
|
OBSCOL_SPC_BT2020_CL = 10, ///< ITU-R BT2020 constant luminance system
|
|
OBSCOL_SPC_SMPTE2085 = 11, ///< SMPTE 2085, Y'D'zD'x
|
|
OBSCOL_SPC_CHROMA_DERIVED_NCL =
|
|
12, ///< Chromaticity-derived non-constant luminance system
|
|
OBSCOL_SPC_CHROMA_DERIVED_CL =
|
|
13, ///< Chromaticity-derived constant luminance system
|
|
OBSCOL_SPC_ICTCP = 14, ///< ITU-R BT.2100-0, ICtCp
|
|
OBSCOL_SPC_NB ///< Not part of ABI
|
|
};
|