obs-studio/plugins/obs-outputs/rtmp-stream.c

/******************************************************************************
    Copyright (C) 2014 by Hugh Bailey <[email protected]>

    This program is free software: you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
    the Free Software Foundation, either version 2 of the License, or
    (at your option) any later version.

    This program is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    GNU General Public License for more details.

    You should have received a copy of the GNU General Public License
    along with this program.  If not, see <http://www.gnu.org/licenses/>.
******************************************************************************/

#include <obs.h>
#include <obs-avc.h>
#include <util/darray.h>
#include <util/dstr.h>
#include <util/threading.h>
#include "librtmp/rtmp.h"
#include "librtmp/log.h"
#include "flv-mux.h"

struct rtmp_stream {
	obs_output_t                  output;

	pthread_mutex_t               packets_mutex;
	DARRAY(struct encoder_packet) packets;

	bool                          connecting;
	bool                          active;
	pthread_t                     connect_thread;
	pthread_t                     send_thread;

	os_sem_t                      send_sem;
	os_event_t                    stop_event;

	struct dstr                   path, key;
	struct dstr                   username, password;

	RTMP                          rtmp;
};

static const char *rtmp_stream_getname(const char *locale)
{
	/* TODO: locale stuff */
	UNUSED_PARAMETER(locale);
	return "RTMP Stream";
}

static void log_rtmp(int level, const char *format, va_list args)
{
	blogva(LOG_DEBUG, format, args);

	UNUSED_PARAMETER(level);
}

static void rtmp_stream_destroy(void *data)
{
	struct rtmp_stream *stream = data;

	if (stream) {
		dstr_free(&stream->path);
		dstr_free(&stream->key);
		dstr_free(&stream->username);
		dstr_free(&stream->password);
		RTMP_Close(&stream->rtmp);
		os_event_destroy(stream->stop_event);
		os_sem_destroy(stream->send_sem);
		pthread_mutex_destroy(&stream->packets_mutex);
		bfree(stream);
	}
}

static void *rtmp_stream_create(obs_data_t settings, obs_output_t output)
{
	struct rtmp_stream *stream = bzalloc(sizeof(struct rtmp_stream));
	stream->output = output;
	pthread_mutex_init_value(&stream->packets_mutex);

	RTMP_Init(&stream->rtmp);
	RTMP_LogSetCallback(log_rtmp);

	if (pthread_mutex_init(&stream->packets_mutex, NULL) != 0)
		goto fail;
	if (os_event_init(&stream->stop_event, OS_EVENT_TYPE_MANUAL) != 0)
		goto fail;

	UNUSED_PARAMETER(settings);
	return stream;

fail:
	rtmp_stream_destroy(stream);
	return NULL;
}

static void rtmp_stream_stop(void *data)
{
	UNUSED_PARAMETER(data);
}

static void rtmp_stream_update(void *data, obs_data_t settings)
{
	UNUSED_PARAMETER(data);
	UNUSED_PARAMETER(settings);
}

static inline void set_rtmp_str(AVal *val, const char *str)
{
	bool valid  = (!str || !*str);
	val->av_val = valid ? (char*)str       : NULL;
	val->av_len = valid ? (int)strlen(str) : 0;
}

static inline void set_rtmp_dstr(AVal *val, struct dstr *str)
{
	bool valid  = !dstr_isempty(str);
	val->av_val = valid ? str->array    : NULL;
	val->av_len = valid ? (int)str->len : 0;
}

static inline bool get_next_packet(struct rtmp_stream *stream,
		struct encoder_packet *packet)
{
	bool new_packet = false;

	pthread_mutex_lock(&stream->packets_mutex);
	if (stream->packets.num) {
		*packet    = stream->packets.array[0];
		new_packet = true;
		da_erase(stream->packets, 0);
	}
	pthread_mutex_unlock(&stream->packets_mutex);

	return new_packet;
}

static void send_packet(struct rtmp_stream *stream,
		struct encoder_packet *packet, bool is_header)
{
	uint8_t *data;
	size_t  size;

	flv_packet_mux(packet, &data, &size, is_header);
	RTMP_Write(&stream->rtmp, (char*)data, (int)size);
	bfree(data);
}

static void *send_thread(void *data)
{
	struct rtmp_stream *stream = data;

	while (os_sem_wait(stream->send_sem) == 0) {
		struct encoder_packet packet;

		if (os_event_try(stream->stop_event) != EAGAIN)
			break;

		if (!get_next_packet(stream, &packet))
			continue;

		send_packet(stream, &packet, false);
		obs_free_encoder_packet(&packet);
	}

	return NULL;
}

#define MIN_SENDBUF_SIZE 65535

static void send_meta_data(struct rtmp_stream *stream)
{
	uint8_t               *meta_data;
	size_t                meta_data_size;

	flv_meta_data(stream->output, &meta_data, &meta_data_size);
	RTMP_Write(&stream->rtmp, (char*)meta_data, (int)meta_data_size);
	bfree(meta_data);
}

static void send_audio_header(struct rtmp_stream *stream)
{
	obs_output_t          context  = stream->output;
	obs_encoder_t         aencoder = obs_output_get_audio_encoder(context);

	struct encoder_packet packet   = {
		.type = OBS_ENCODER_AUDIO,
		.timebase_den = 1
	};

	obs_encoder_get_extra_data(aencoder, &packet.data, &packet.size);
	send_packet(stream, &packet, true);
}

static void send_video_header(struct rtmp_stream *stream)
{
	obs_output_t          context  = stream->output;
	obs_encoder_t         vencoder = obs_output_get_video_encoder(context);
	uint8_t               *header;
	size_t                size;

	struct encoder_packet packet   = {
		.type = OBS_ENCODER_VIDEO,
		.timebase_den = 1
	};

	obs_encoder_get_extra_data(vencoder, &header, &size);
	packet.size = obs_create_avc_header(&packet.data, header, size);
	send_packet(stream, &packet, true);
	obs_free_encoder_packet(&packet);
}

static void send_headers(struct rtmp_stream *stream)
{
	send_meta_data(stream);
	send_audio_header(stream);
	send_video_header(stream);
}

static inline bool reset_semaphore(struct rtmp_stream *stream)
{
	os_sem_destroy(stream->send_sem);
	return os_sem_init(&stream->send_sem, 0) == 0;
}

#ifdef _WIN32
#define socklen_t int
#endif

static void init_send(struct rtmp_stream *stream)
{
	int cur_sendbuf_size = MIN_SENDBUF_SIZE;
	socklen_t size = sizeof(int);
	socklen_t ret;

	getsockopt(stream->rtmp.m_sb.sb_socket, SOL_SOCKET, SO_SNDBUF,
			(char*)&cur_sendbuf_size, &size);

	if (cur_sendbuf_size < MIN_SENDBUF_SIZE) {
		cur_sendbuf_size = 65535;
		setsockopt(stream->rtmp.m_sb.sb_socket, SOL_SOCKET, SO_SNDBUF,
				(const char*)&cur_sendbuf_size, size);
	}

	reset_semaphore(stream);

	ret = pthread_create(&stream->send_thread, NULL, send_thread, stream);
	if (ret != 0)
		bcrash("Failed to create send thread");

	send_headers(stream);
	obs_output_begin_data_capture(stream->output, 0);
}

static int try_connect(struct rtmp_stream *stream)
{
	if (!RTMP_SetupURL2(&stream->rtmp, stream->path.array,
				stream->key.array))
		return OBS_OUTPUT_BAD_PATH;

	set_rtmp_dstr(&stream->rtmp.Link.pubUser,   &stream->username);
	set_rtmp_dstr(&stream->rtmp.Link.pubPasswd, &stream->password);
	stream->rtmp.Link.swfUrl = stream->rtmp.Link.tcUrl;
	set_rtmp_str(&stream->rtmp.Link.flashVer,
			"FMLE/3.0 (compatible; FMSc/1.0)");

	stream->rtmp.m_outChunkSize       = 4096;
	stream->rtmp.m_bSendChunkSizeInfo = true;
	stream->rtmp.m_bUseNagle          = true;

	if (!RTMP_Connect(&stream->rtmp, NULL))
		return OBS_OUTPUT_CONNECT_FAILED;
	if (!RTMP_ConnectStream(&stream->rtmp, 0))
		return OBS_OUTPUT_INVALID_STREAM;

	init_send(stream);
	return OBS_OUTPUT_SUCCESS;
}

static void *connect_thread(void *data)
{
	UNUSED_PARAMETER(data);
	return NULL;
}

static bool rtmp_stream_start(void *data)
{
	struct rtmp_stream *stream = data;
	obs_data_t settings;

	if (!obs_output_can_begin_data_capture(stream->output, 0))
		return false;

	settings = obs_output_get_settings(stream->output);
	dstr_copy(&stream->path,     obs_data_getstring(settings, "path"));
	dstr_copy(&stream->key,      obs_data_getstring(settings, "key"));
	dstr_copy(&stream->username, obs_data_getstring(settings, "username"));
	dstr_copy(&stream->password, obs_data_getstring(settings, "password"));
	obs_data_release(settings);

	return pthread_create(&stream->connect_thread, NULL, connect_thread,
			stream) != 0;
}

static bool rtmp_stream_active(void *data)
{
	UNUSED_PARAMETER(data);
	return false;
}