diff --git a/UI/window-basic-auto-config-test.cpp b/UI/window-basic-auto-config-test.cpp index 8c7a82a03..e56b19b0a 100644 --- a/UI/window-basic-auto-config-test.cpp +++ b/UI/window-basic-auto-config-test.cpp @@ -250,9 +250,10 @@ void AutoConfigTestPage::TestBandwidthThread() GetServers(servers); /* just use the first server if it only has one alternate server, - * or if using Restream due to their "auto" servers */ + * or if using Restream or Nimo TV due to their "auto" servers */ if (servers.size() < 3 || - wiz->serviceName.substr(0, 11) == "Restream.io") { + wiz->serviceName.substr(0, 11) == "Restream.io" || + wiz->serviceName == "Nimo TV") { servers.resize(1); } else if (wiz->service == AutoConfig::Service::Twitch && diff --git a/plugins/rtmp-services/CMakeLists.txt b/plugins/rtmp-services/CMakeLists.txt index 5232336b8..ed8f13634 100644 --- a/plugins/rtmp-services/CMakeLists.txt +++ b/plugins/rtmp-services/CMakeLists.txt @@ -9,6 +9,7 @@ include_directories(${OBS_JANSSON_INCLUDE_DIRS}) set(rtmp-services_SOURCES twitch.c younow.c + nimotv.c rtmp-common.c rtmp-custom.c rtmp-services-main.c) @@ -23,6 +24,7 @@ endif() set(rtmp-services_HEADERS twitch.h younow.h + nimotv.h rtmp-format-ver.h) set(RTMP_SERVICES_URL diff --git a/plugins/rtmp-services/nimotv.c b/plugins/rtmp-services/nimotv.c new file mode 100644 index 000000000..c03279822 --- /dev/null +++ b/plugins/rtmp-services/nimotv.c @@ -0,0 +1,145 @@ +#include +#include +#include +#include +#include +#include "util/base.h" +#include "nimotv.h" +#include + +struct nimotv_mem_struct { + char *memory; + size_t size; +}; +static char *current_ingest = NULL; +static time_t last_time = -1; + +static size_t nimotv_write_cb(void *contents, size_t size, size_t nmemb, + void *userp) +{ + size_t realsize = size * nmemb; + struct nimotv_mem_struct *mem = (struct nimotv_mem_struct *)userp; + + char *ptr = realloc(mem->memory, mem->size + realsize + 1); + if (ptr == NULL) { + blog(LOG_WARNING, "nimotv_write_cb: realloc returned NULL"); + return 0; + } + + mem->memory = ptr; + memcpy(&(mem->memory[mem->size]), contents, realsize); + mem->size += realsize; + mem->memory[mem->size] = 0; + + return realsize; +} + +static char *load_ingest(const char *json) +{ + json_t *root = json_loads(json, 0, NULL); + char *ingest = NULL; + if (!root) + return ingest; + + json_t *recommended_ingests = json_object_get(root, "ingests"); + if (recommended_ingests) { + json_t *recommended = json_array_get(recommended_ingests, 0); + if (recommended) { + json_t *item_url = json_object_get(recommended, "url"); + if (item_url) { + const char *url = json_string_value(item_url); + ingest = bstrdup(url); + } + } + } + + json_decref(root); + return ingest; +} + +const char *nimotv_get_ingest(const char *key) +{ + if (current_ingest != NULL) { + time_t now = time(NULL); + double diff = difftime(now, last_time); + if (diff < 2) { + blog(LOG_INFO, + "nimotv_get_ingest: returning ingest from cache: %s", + current_ingest); + return current_ingest; + } + } + + CURL *curl_handle; + CURLcode res; + struct nimotv_mem_struct chunk; + struct dstr uri; + long response_code; + + curl_handle = curl_easy_init(); + chunk.memory = malloc(1); /* will be grown as needed by realloc */ + chunk.size = 0; /* no data at this point */ + + char *encoded_key = curl_easy_escape(NULL, key, 0); + dstr_init(&uri); + dstr_copy(&uri, "https://globalcdnweb.nimo.tv/api/ingests/nimo?id="); + dstr_ncat(&uri, encoded_key, strlen(encoded_key)); + curl_free(encoded_key); + + curl_easy_setopt(curl_handle, CURLOPT_URL, uri.array); + curl_easy_setopt(curl_handle, CURLOPT_SSL_VERIFYPEER, true); + curl_easy_setopt(curl_handle, CURLOPT_SSL_VERIFYHOST, 2L); + curl_easy_setopt(curl_handle, CURLOPT_TIMEOUT, 3L); + curl_easy_setopt(curl_handle, CURLOPT_WRITEFUNCTION, nimotv_write_cb); + curl_easy_setopt(curl_handle, CURLOPT_WRITEDATA, (void *)&chunk); + curl_obs_set_revoke_setting(curl_handle); + +#if LIBCURL_VERSION_NUM >= 0x072400 + // A lot of servers don't yet support ALPN + curl_easy_setopt(curl_handle, CURLOPT_SSL_ENABLE_ALPN, 0); +#endif + + res = curl_easy_perform(curl_handle); + dstr_free(&uri); + + if (res != CURLE_OK) { + blog(LOG_WARNING, + "nimotv_get_ingest: curl_easy_perform() failed: %s", + curl_easy_strerror(res)); + curl_easy_cleanup(curl_handle); + free(chunk.memory); + return NULL; + } + + curl_easy_getinfo(curl_handle, CURLINFO_RESPONSE_CODE, &response_code); + if (response_code != 200) { + blog(LOG_WARNING, + "nimotv_get_ingest: curl_easy_perform() returned code: %ld", + response_code); + curl_easy_cleanup(curl_handle); + free(chunk.memory); + return NULL; + } + + curl_easy_cleanup(curl_handle); + + if (chunk.size == 0) { + blog(LOG_WARNING, + "nimotv_get_ingest: curl_easy_perform() returned empty response"); + free(chunk.memory); + return NULL; + } + + if (current_ingest != NULL) { + bfree(current_ingest); + } + + current_ingest = load_ingest(chunk.memory); + last_time = time(NULL); + + free(chunk.memory); + blog(LOG_INFO, "nimotv_get_ingest: returning ingest: %s", + current_ingest); + + return current_ingest; +} diff --git a/plugins/rtmp-services/nimotv.h b/plugins/rtmp-services/nimotv.h new file mode 100644 index 000000000..ac4d30ea6 --- /dev/null +++ b/plugins/rtmp-services/nimotv.h @@ -0,0 +1,3 @@ +#pragma once + +extern const char *nimotv_get_ingest(const char *key); diff --git a/plugins/rtmp-services/rtmp-common.c b/plugins/rtmp-services/rtmp-common.c index c572a88df..88ba6d125 100644 --- a/plugins/rtmp-services/rtmp-common.c +++ b/plugins/rtmp-services/rtmp-common.c @@ -6,6 +6,7 @@ #include "rtmp-format-ver.h" #include "twitch.h" #include "younow.h" +#include "nimotv.h" struct rtmp_common { char *service; @@ -345,6 +346,11 @@ static void fill_servers(obs_property_t *servers_prop, json_t *service, return; } + if (strcmp(name, "Nimo TV") == 0) { + obs_property_list_add_string( + servers_prop, obs_module_text("Server.Auto"), "auto"); + } + json_array_foreach (servers, index, server) { const char *server_name = get_string_val(server, "name"); const char *url = get_string_val(server, "url"); @@ -601,6 +607,12 @@ static const char *rtmp_common_url(void *data) } } + if (service->service && strcmp(service->service, "Nimo TV") == 0) { + if (service->server && strcmp(service->server, "auto") == 0) { + return nimotv_get_ingest(service->key); + } + } + return service->server; }