obs-outputs: Add Bind-To-IP wrapper functions

Allows enumerating IP addresses so you can select the adapter/IP from
which to stream from.

plugins/obs-outputs/CMakeLists.txt

@@ -62,6 +62,7 @@ endif()
 set(obs-outputs_HEADERS
 	obs-output-ver.h
 	rtmp-helpers.h
+	net-if.h
 	flv-mux.h
 	flv-output.h
 	librtmp)
@@ -69,7 +70,8 @@ set(obs-outputs_SOURCES
 	obs-outputs.c
 	rtmp-stream.c
 	flv-output.c
-	flv-mux.c)
+	flv-mux.c
+	net-if.c)
 	
 add_library(obs-outputs MODULE
 	${obs-outputs_SOURCES}

plugins/obs-outputs/net-if.c

@@ -0,0 +1,270 @@
+/******************************************************************************
+    Copyright (C) 2016
+
+    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 "net-if.h"
+#include <util/platform.h>
+#include <util/dstr.h>
+
+#define do_log(level, format, ...) \
+	blog(level, "[net if] " format, ##__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__)
+
+static inline void netif_saddr_data_push_back(struct netif_saddr_data *sd,
+		const char *ip, const char *adapter)
+{
+	struct netif_saddr_item item;
+	struct dstr full_name = {0};
+	char *ip_dup = bstrdup(ip);
+
+	if (adapter && *adapter)
+		dstr_printf(&full_name, "[%s] %s", adapter, ip);
+	else
+		dstr_copy(&full_name, ip);
+
+	item.name = full_name.array;
+	item.addr = ip_dup;
+
+	da_push_back(sd->addrs, &item);
+}
+
+static void netif_convert_to_string(char *dest,
+		struct sockaddr_storage *byte_address)
+{
+	int family = byte_address->ss_family;
+	char temp_char[INET6_ADDRSTRLEN] = {0};
+
+#ifndef _WIN32
+	if (family == AF_INET)
+		inet_ntop(family, &(((struct sockaddr_in*)byte_address)->sin_addr),
+				temp_char, INET6_ADDRSTRLEN);
+	else if (family == AF_INET6)
+		inet_ntop(family, &(((struct sockaddr_in*)byte_address)->sin_addr),
+				temp_char, INET6_ADDRSTRLEN);
+#else
+	if (family == AF_INET)
+		InetNtopA(family, &(((SOCKADDR_IN *)byte_address)->sin_addr),
+				temp_char, INET6_ADDRSTRLEN);
+	else if (family == AF_INET6)
+		InetNtopA(family, &(((SOCKADDR_IN6 *)byte_address)->sin6_addr),
+				temp_char, INET6_ADDRSTRLEN);
+#endif
+	strncpy(dest, temp_char, INET6_ADDRSTRLEN);
+}
+
+static void netif_push(struct sockaddr *copy_source,
+				struct netif_saddr_data *saddr_d,
+				const char *adapter)
+{
+	char temp_char[INET6_ADDRSTRLEN] = {0};
+	struct sockaddr_storage sa = {0};
+
+	if (copy_source->sa_family == AF_INET)
+		memcpy(&sa, copy_source, sizeof(struct sockaddr_in));
+	else if (copy_source->sa_family == AF_INET6)
+		memcpy(&sa, copy_source, sizeof(struct sockaddr_in6));
+
+	netif_convert_to_string(temp_char, &sa);
+	netif_saddr_data_push_back(saddr_d, temp_char, adapter);
+}
+
+void netif_log_saddrs(struct netif_saddr_data *sd)
+{
+	for(size_t i = 0; i < sd->addrs.num; i++)
+		info("\t\t%s", sd->addrs.array[i].name);
+}
+
+bool netif_str_to_addr(struct sockaddr_storage *out, int *addr_len,
+		const char *addr)
+{
+	bool ipv6;
+
+	memset(out, 0, sizeof(*out));
+	*addr_len = 0;
+
+	if (!addr)
+		return false;
+
+	ipv6 = (strchr(addr, ':') != NULL);
+	out->ss_family = ipv6 ? AF_INET6 : AF_INET;
+	*addr_len = sizeof(*out);
+
+#ifdef _WIN32
+	int ret = WSAStringToAddressA((LPSTR)addr, out->ss_family, NULL,
+			(LPSOCKADDR)out, addr_len);
+	if (ret == SOCKET_ERROR)
+		warn("Could not parse address, error code: %d", GetLastError());
+	return ret != SOCKET_ERROR;
+#else
+	struct sockaddr_in *sin = (struct sockaddr_in *)out;
+	if (inet_pton(out->ss_family, addr, &sin->sin_addr)) {
+		*addr_len = ipv6 ?
+			sizeof(struct sockaddr_in6) :
+			sizeof(struct sockaddr_in);
+		return true;
+	}
+
+	return false;
+#endif
+}
+
+#ifndef _WIN32
+static inline bool is_loopback(struct ifaddrs *ifa)
+{
+	const char *n = ifa->ifa_name;
+	return n && (strcmp(n, "lo") == 0 || strcmp(n, "lo0") == 0);
+}
+
+static inline void netif_get_addrs_nix(struct netif_saddr_data *ifaddrs)
+{
+	struct ifaddrs *ifaddr, *ifa;
+	unsigned int family, s;
+	char host[NI_MAXHOST];
+
+	if (getifaddrs(&ifaddr) == -1) {
+		warn("getifaddrs() failed");
+		return;
+	}
+
+	for (ifa = ifaddr; ifa != NULL; ifa = ifa->ifa_next) {
+		if (ifa->ifa_addr == NULL || is_loopback(ifa))
+			continue;
+
+		family = ifa->ifa_addr->sa_family;
+
+		if ((family == AF_INET) || (family == AF_INET6)) {
+			s = getnameinfo(ifa->ifa_addr,
+				(family == AF_INET) ?
+				sizeof(struct sockaddr_in) :
+				sizeof(struct sockaddr_in6),
+				host, NI_MAXHOST, NULL, 0, NI_NUMERICHOST);
+			if (s != 0) {
+				warn("getnameinfo() failed: %s",
+						gai_strerror(s));
+				continue;
+			}
+
+			netif_push(ifa->ifa_addr, ifaddrs, ifa->ifa_name);
+		}
+	}
+
+	freeifaddrs(ifaddr);
+}
+
+#else
+
+static inline PIP_ADAPTER_ADDRESSES get_adapters(void)
+{
+	PIP_ADAPTER_ADDRESSES adapter = NULL;
+	unsigned long ret = 0;
+	unsigned long out_buf_len = 4096;
+	unsigned long flags =
+		GAA_FLAG_SKIP_ANYCAST |
+		GAA_FLAG_SKIP_MULTICAST |
+		GAA_FLAG_SKIP_DNS_SERVER;
+	const int max_tries = 3;
+	int i = 0;
+
+	do {
+		adapter = (IP_ADAPTER_ADDRESSES*)bmalloc(out_buf_len);
+		if (!adapter)
+			return NULL;
+
+		ret = GetAdaptersAddresses(AF_UNSPEC, flags, NULL, adapter,
+				&out_buf_len);
+		if (ret == ERROR_BUFFER_OVERFLOW) {
+			bfree(adapter);
+			adapter = NULL;
+		} else {
+			break;
+		}
+		i++;
+	} while ((ret == ERROR_BUFFER_OVERFLOW) && (i < max_tries));
+
+	if (ret != NO_ERROR && ret != ERROR_NO_DATA) {
+		LPSTR msg_buf = NULL;
+
+		bfree(adapter);
+		adapter = NULL;
+
+		FormatMessageA(
+				FORMAT_MESSAGE_ALLOCATE_BUFFER |
+				FORMAT_MESSAGE_FROM_SYSTEM |
+				FORMAT_MESSAGE_IGNORE_INSERTS,
+				NULL, ret,
+				MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT),
+				msg_buf, 0, NULL);
+		if (msg_buf) {
+			warn("Call to GetAdaptersAddresses failed: %s (%d)",
+					msg_buf, ret);
+			LocalFree(msg_buf);
+		}
+	}
+
+	return adapter;
+}
+
+static inline void netif_get_addrs_win32(struct netif_saddr_data *ifaddrs)
+{
+	PIP_ADAPTER_ADDRESSES adapter = get_adapters();
+	PIP_ADAPTER_UNICAST_ADDRESS unicast = NULL;
+	PIP_ADAPTER_ADDRESSES cur_adap = NULL;
+	SOCKET_ADDRESS socket_addr;
+	int family;
+
+	if (!adapter)
+		return;
+
+	for (cur_adap = adapter; !!cur_adap; cur_adap = cur_adap->Next) {
+		char *adap_name = NULL;
+
+		if (cur_adap->OperStatus != IfOperStatusUp ||
+		    cur_adap->IfType == IF_TYPE_SOFTWARE_LOOPBACK)
+			continue;
+
+		os_wcs_to_utf8_ptr(cur_adap->FriendlyName, 0, &adap_name);
+
+		unicast = cur_adap->FirstUnicastAddress;
+
+		for (; !!unicast; unicast = unicast->Next) {
+			socket_addr = unicast->Address;
+
+			family = socket_addr.lpSockaddr->sa_family;
+			if (family == AF_INET || family == AF_INET6)
+				netif_push(socket_addr.lpSockaddr, ifaddrs,
+						adap_name);
+		}
+
+		bfree(adap_name);
+	}
+
+	bfree(adapter);
+}
+#endif
+
+void netif_get_addrs(struct netif_saddr_data *ifaddrs)
+{
+	da_init(ifaddrs->addrs);
+
+#ifdef _WIN32
+	netif_get_addrs_win32(ifaddrs);
+#else
+	netif_get_addrs_nix(ifaddrs);
+#endif
+}

plugins/obs-outputs/net-if.h

@@ -0,0 +1,76 @@
+/******************************************************************************
+    Copyright (C) 2016
+
+    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/>.
+******************************************************************************/
+
+#pragma once
+
+#include <util/darray.h>
+
+#ifdef _WIN32
+#  include <ws2tcpip.h>
+#  include <winsock2.h>
+#  include <ws2ipdef.h>
+#  include <iphlpapi.h>
+#else
+
+#  ifdef __linux__
+#    include <linux/if_link.h>
+#  elif __FreeBSD__
+#    ifndef _GNU_SOURCE
+#      define _GNU_SOURCE
+#      define __NET_IF_GNU_SOURCE__
+#    endif //_GNU_SOURCE
+#  endif //__FreeBSD__
+
+#  include <ifaddrs.h>
+#  include <netdb.h>
+#  include <stdio.h>
+#  include <stdlib.h>
+#  include <unistd.h>
+#  include <arpa/inet.h>
+#  include <sys/socket.h>
+
+#  ifdef __FreeBSD__
+#    ifdef ___NET_IF_GNU_SOURCE__
+#      undef ___NET_IF_GNU_SOURCE__
+#      undef _GNU_SOURCE
+#    endif
+#  endif
+
+#endif
+
+struct netif_saddr_item {
+	char *name;
+	char *addr;
+};
+
+struct netif_saddr_data {
+	DARRAY(struct netif_saddr_item) addrs;
+};
+
+static inline void netif_saddr_data_free(struct netif_saddr_data *data)
+{
+	for (size_t i = 0; i < data->addrs.num; i++) {
+		bfree(data->addrs.array[i].name);
+		bfree(data->addrs.array[i].addr);
+	}
+	da_free(data->addrs);
+}
+
+extern bool netif_str_to_addr(struct sockaddr_storage *out, int *addr_len,
+		const char *addr);
+extern void netif_get_addrs(struct netif_saddr_data *ifaddrs);
+extern void netif_log_saddrs(struct netif_saddr_data *sd);