obs-studio/plugins/rtmp-services/service-specific/dacast.c

#include <file-updater/file-updater.h>
#include <util/threading.h>
#include <util/platform.h>
#include <util/dstr.h>
#include <jansson.h>

#include "dacast.h"

#ifndef SEC_TO_NSEC
#define SEC_TO_NSEC 1000000000ULL
#endif

static update_info_t *dacast_update_info = NULL;
static pthread_mutex_t mutex = PTHREAD_MUTEX_INITIALIZER;
static bool ingests_loaded = false;

struct dacast_ingest_info {
	char *key;
	uint64_t last_time;
	struct dacast_ingest ingest;
};

struct dacast_ingest dacast_invalid_ingest = {"rtmp://dacast", "", "",
					      "fake_key"};

static DARRAY(struct dacast_ingest_info) cur_ingests;

static void free_ingest(struct dacast_ingest ingest)
{
	bfree((void *)ingest.url);
	bfree((void *)ingest.username);
	bfree((void *)ingest.password);
	bfree((void *)ingest.streamkey);
}

static void free_ingests(void)
{
	for (size_t i = 0; i < cur_ingests.num; i++) {
		struct dacast_ingest_info *info = &cur_ingests.array[i];
		bfree(info->key);
		free_ingest(info->ingest);
	}
	da_free(cur_ingests);
}

static struct dacast_ingest_info *find_ingest(const char *key)
{
	struct dacast_ingest_info *ret = NULL;
	for (size_t i = 0; i < cur_ingests.num; i++) {
		struct dacast_ingest_info *info = &cur_ingests.array[i];
		if (strcmp(info->key, key) == 0) {
			ret = info;
			break;
		}
	}
	return (struct dacast_ingest_info *)ret;
}

static bool load_ingests(const char *json, const char *key)
{
	json_t *root;
	json_t *stream;
	bool success = false;
	struct dacast_ingest_info *info = find_ingest(key);
	if (!info) {
		info = da_push_back_new(cur_ingests);
		info->key = bstrdup(key);
	} else {
		free_ingest(info->ingest);
	}

	root = json_loads(json, 0, NULL);
	if (!root)
		goto finish;

	stream = json_object_get(root, "stream");
	if (!stream)
		goto finish;

	json_t *item_server = json_object_get(stream, "server");
	json_t *item_username = json_object_get(stream, "username");
	json_t *item_password = json_object_get(stream, "password");
	json_t *item_streamkey = json_object_get(stream, "streamkey");

	if (!item_server || !item_username || !item_password || !item_streamkey)
		goto finish;

	const char *server = json_string_value(item_server);
	const char *username = json_string_value(item_username);
	const char *password = json_string_value(item_password);
	const char *streamkey = json_string_value(item_streamkey);

	info->ingest.url = bstrdup(server);
	info->ingest.username = bstrdup(username);
	info->ingest.password = bstrdup(password);
	info->ingest.streamkey = bstrdup(streamkey);

	info->last_time = os_gettime_ns() / SEC_TO_NSEC;

	success = true;

finish:
	if (root)
		json_decref(root);
	return success;
}

static bool dacast_ingest_update(void *param, struct file_download_data *data)
{
	bool success;

	pthread_mutex_lock(&mutex);
	success = load_ingests((const char *)data->buffer.array,
			       (const char *)param);
	pthread_mutex_unlock(&mutex);

	if (success) {
		os_atomic_set_bool(&ingests_loaded, true);
	}

	return true;
}

struct dacast_ingest *dacast_ingest(const char *key)
{
	pthread_mutex_lock(&mutex);
	struct dacast_ingest_info *info = find_ingest(key);
	pthread_mutex_unlock(&mutex);
	return info == NULL ? &dacast_invalid_ingest : &info->ingest;
}

void init_dacast_data(void)
{
	da_init(cur_ingests);
	pthread_mutex_init(&mutex, NULL);
}

extern const char *get_module_name(void);

#define TIMEOUT_SEC 3

void dacast_ingests_load_data(const char *server, const char *key)
{
	struct dstr uri = {0};

	os_atomic_set_bool(&ingests_loaded, false);

	dstr_copy(&uri, server);
	dstr_cat(&uri, key);

	if (dacast_update_info) {
		update_info_destroy(dacast_update_info);
		dacast_update_info = NULL;
	}

	dacast_update_info = update_info_create_single(
		"[dacast ingest load data] ", get_module_name(), uri.array,
		dacast_ingest_update, (void *)key);

	if (!os_atomic_load_bool(&ingests_loaded)) {
		for (int i = 0; i < TIMEOUT_SEC * 100; i++) {
			if (os_atomic_load_bool(&ingests_loaded)) {
				break;
			}
			os_sleep_ms(10);
		}
	}

	dstr_free(&uri);
}

void unload_dacast_data(void)
{
	update_info_destroy(dacast_update_info);
	free_ingests();
	pthread_mutex_destroy(&mutex);
}