Automtically discover Gamestream servers on local network

CMakeLists.txt

@@ -21,6 +21,7 @@ find_package(Freescale)
 
 find_package(PkgConfig REQUIRED)
 pkg_check_modules(EVDEV REQUIRED libevdev)
+pkg_check_modules(AVAHI REQUIRED avahi-client)
 
 if(Broadcom_FOUND)
   aux_source_directory(./ilclient SRC_LIST)
@@ -41,5 +42,5 @@ endif()
 add_executable(moonlight ${SRC_LIST})
 set_property(TARGET moonlight PROPERTY C_STANDARD 11)
 
-include_directories(./moonlight-common-c ${OPUS_INCLUDE_DIRS} ${EVDEV_INCLUDE_DIRS})
-target_link_libraries (moonlight PUBLIC ${CMAKE_THREAD_LIBS_INIT} ${CURL_LIBRARIES} ${OPENSSL_LIBRARIES} ${EXPAT_LIBRARIES} ${EVDEV_LIBRARIES} ${ALSA_LIBRARY} ${OPUS_LIBRARY})
+include_directories(./moonlight-common-c ${OPUS_INCLUDE_DIRS} ${EVDEV_INCLUDE_DIRS} ${AVAHI_INCLUDE_DIRS})
+target_link_libraries (moonlight PUBLIC ${CMAKE_THREAD_LIBS_INIT} ${CURL_LIBRARIES} ${OPENSSL_LIBRARIES} ${EXPAT_LIBRARIES} ${EVDEV_LIBRARIES} ${ALSA_LIBRARY} ${OPUS_LIBRARY} ${AVAHI_LIBRARIES})

src/discover.c

@@ -0,0 +1,106 @@
+/*
+ * This file is part of Moonlight Embedded.
+ *
+ * Based on Avahi example client-browse-services
+ *
+ * Moonlight 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 3 of the License, or
+ * (at your option) any later version.
+ *
+ * Moonlight 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 Moonlight; if not, see <http://www.gnu.org/licenses/>.
+ */
+
+#include <avahi-client/client.h>
+#include <avahi-client/lookup.h>
+
+#include <avahi-common/simple-watch.h>
+#include <avahi-common/malloc.h>
+#include <avahi-common/error.h>
+
+#include <stdio.h>
+#include <stdlib.h>
+#include <stdbool.h>
+#include <time.h>
+
+static AvahiSimplePoll *simple_poll = NULL;
+
+static void discover_client_callback(AvahiClient *c, AvahiClientState state, void *userdata) {
+  if (state == AVAHI_CLIENT_FAILURE) {
+    fprintf(stderr, "Server connection failure: %s\n", avahi_strerror(avahi_client_errno(c)));
+    avahi_simple_poll_quit(simple_poll);
+  }
+}
+
+static void discover_resolve_callback(AvahiServiceResolver *r, AvahiIfIndex interface, AvahiProtocol protocol, AvahiResolverEvent event, const char *name, const char *type, const char *domain, const char *host_name, const AvahiAddress *address, uint16_t port, AvahiStringList *txt, AvahiLookupResultFlags flags, void *userdata) {
+  if (event == AVAHI_RESOLVER_FOUND) {
+    if (userdata != NULL) {
+      avahi_address_snprint(userdata, AVAHI_ADDRESS_STR_MAX, address);
+      avahi_simple_poll_quit(simple_poll);
+    } else {
+      char strAddress[AVAHI_ADDRESS_STR_MAX];
+      avahi_address_snprint(strAddress, sizeof(strAddress), address);
+      printf(" %s (%s)\n", host_name, strAddress);
+    }
+  }
+
+  avahi_service_resolver_free(r);
+}
+
+static void discover_browse_callback(AvahiServiceBrowser *b, AvahiIfIndex interface, AvahiProtocol protocol, AvahiBrowserEvent event, const char *name, const char *type, const char *domain, AvahiLookupResultFlags flags, void* userdata) {
+  AvahiClient *c = avahi_service_browser_get_client(b);
+
+  switch (event) {
+  case AVAHI_BROWSER_FAILURE:
+    fprintf(stderr, "(Discover) %s\n", avahi_strerror(avahi_client_errno(avahi_service_browser_get_client(b))));
+    avahi_simple_poll_quit(simple_poll);
+    break;
+  case AVAHI_BROWSER_NEW:
+    if (!(avahi_service_resolver_new(c, interface, protocol, name, type, domain, AVAHI_PROTO_UNSPEC, 0, discover_resolve_callback, userdata)))
+      fprintf(stderr, "Failed to resolve service '%s': %s\n", name, avahi_strerror(avahi_client_errno(c)));
+
+    break;
+  case AVAHI_BROWSER_REMOVE:
+    break;
+  }
+}
+
+void discover_server(char* dest) {
+  AvahiClient *client = NULL;
+  AvahiServiceBrowser *sb = NULL;
+
+  if (!(simple_poll = avahi_simple_poll_new())) {
+    fprintf(stderr, "Failed to create simple poll object.\n");
+    goto cleanup;
+  }
+
+  int error;
+  client = avahi_client_new(avahi_simple_poll_get(simple_poll), 0, discover_client_callback, NULL, &error);
+  if (!client) {
+    fprintf(stderr, "Failed to create client: %s\n", avahi_strerror(error));
+    goto cleanup;
+  }
+
+  if (!(sb = avahi_service_browser_new(client, AVAHI_IF_UNSPEC, AVAHI_PROTO_UNSPEC, "_nvstream._tcp", NULL, 0, discover_browse_callback, dest))) {
+    fprintf(stderr, "Failed to create service browser: %s\n", avahi_strerror(avahi_client_errno(client)));
+    goto cleanup;
+  }
+
+  avahi_simple_poll_loop(simple_poll);
+
+  cleanup:
+  if (sb)
+    avahi_service_browser_free(sb);
+
+  if (client)
+    avahi_client_free(client);
+
+  if (simple_poll)
+    avahi_simple_poll_free(simple_poll);
+}

src/discover.h

@@ -0,0 +1,20 @@
+/*
+ * This file is part of Moonlight Embedded.
+ *
+ * Moonlight 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 3 of the License, or
+ * (at your option) any later version.
+ *
+ * Moonlight 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 Moonlight; if not, see <http://www.gnu.org/licenses/>.
+ */
+
+#define MAX_ADDRESS_SIZE 40
+
+void discover_server(char* dest);

src/main.c

@@ -22,6 +22,7 @@
 #include "video.h"
 #include "audio.h"
 #include "input.h"
+#include "discover.h"
 
 #include "limelight-common/Limelight.h"
 
@@ -195,8 +196,17 @@ int main(int argc, char* argv[]) {
   }
 
   if (address == NULL) {
-    perror("No address given");
-    exit(-1);
+    address = malloc(MAX_ADDRESS_SIZE);
+    address[0] = 0;
+    if (address == NULL) {
+      perror("Not enough memory");
+      exit(-1);
+    }
+    discover_server(address);
+    if (address[0] == 0) {
+      perror("Can't find server");
+      exit(-1);
+    }
   }
 
   client_init(address);