applied platformio structure

2026-03-13 17:03:22 +00:00
parent c5233cf15c
commit db7d90e736
3510 changed files with 691878 additions and 0 deletions
@@ -0,0 +1,108 @@
+#include <cassert>
+#include <vector>
+#include <mutex>
+
+#include "BellLogger.h"
+#include "MDNSService.h"
+
+#ifdef _WIN32
+#include <WinSock2.h>
+#include <iphlpapi.h>
+#pragma comment(lib, "IPHLPAPI.lib")
+#include "mdnssvc.h"
+#else
+#include <arpa/inet.h>
+#include "mdns.h"
+#endif
+
+using namespace bell;
+
+class implMDNSService : public MDNSService {
+ private:
+  struct mdns_service* service;
+  void unregisterService(void);
+
+ public:
+  static struct mdnsd* mdnsServer;
+  static std::atomic<size_t> instances;
+  implMDNSService(struct mdns_service* service) : service(service) { instances++; };
+};
+
+/**
+ * Win32 implementation of MDNSService
+ **/
+
+struct mdnsd* implMDNSService::mdnsServer = NULL;
+std::atomic<size_t> implMDNSService::instances = 0;
+
+static std::mutex registerMutex;
+
+void implMDNSService::unregisterService() {
+  mdns_service_remove(implMDNSService::mdnsServer, service);
+  if (!--instances && implMDNSService::mdnsServer) {
+    mdnsd_stop(implMDNSService::mdnsServer);
+    implMDNSService::mdnsServer = nullptr;
+  }   
+}
+
+std::unique_ptr<MDNSService> MDNSService::registerService(
+    const std::string& serviceName, const std::string& serviceType,
+    const std::string& serviceProto, const std::string& serviceHost,
+    int servicePort, const std::map<std::string, std::string> txtData) {
+  std::lock_guard lock(registerMutex);
+  if (!implMDNSService::mdnsServer) {
+    char hostname[128];
+    gethostname(hostname, sizeof(hostname));
+
+    struct sockaddr_in* host = NULL;
+    ULONG size = sizeof(IP_ADAPTER_ADDRESSES) * 64;
+    IP_ADAPTER_ADDRESSES* adapters = (IP_ADAPTER_ADDRESSES*)malloc(size);
+    int ret = GetAdaptersAddresses(AF_UNSPEC,
+                                   GAA_FLAG_INCLUDE_GATEWAYS |
+                                       GAA_FLAG_SKIP_MULTICAST |
+                                       GAA_FLAG_SKIP_ANYCAST,
+                                   0, adapters, &size);
+
+    for (PIP_ADAPTER_ADDRESSES adapter = adapters; adapter && !host;
+         adapter = adapter->Next) {
+      if (adapter->TunnelType == TUNNEL_TYPE_TEREDO)
+        continue;
+      if (adapter->OperStatus != IfOperStatusUp)
+        continue;
+
+      for (IP_ADAPTER_UNICAST_ADDRESS* unicast = adapter->FirstUnicastAddress;
+           unicast; unicast = unicast->Next) {
+        if (adapter->FirstGatewayAddress &&
+            unicast->Address.lpSockaddr->sa_family == AF_INET) {
+          host = (struct sockaddr_in*)unicast->Address.lpSockaddr;
+          BELL_LOG(info, "mdns", "mDNS on interface %s",
+                   inet_ntoa(host->sin_addr));
+          implMDNSService::mdnsServer = mdnsd_start(host->sin_addr, false);
+          break;
+        }
+      }
+    }
+
+    assert(implMDNSService::mdnsServer);
+    mdnsd_set_hostname(implMDNSService::mdnsServer, hostname, host->sin_addr);
+    free(adapters);
+  }
+
+  std::vector<const char*> txt;
+  std::vector<std::unique_ptr<std::string>> txtStr;
+
+  for (auto& [key, value] : txtData) {
+    auto str = make_unique<std::string>(key + "=" + value);
+    txtStr.push_back(std::move(str));
+    txt.push_back(txtStr.back()->c_str());
+  }
+  txt.push_back(NULL);
+
+  std::string type(serviceType + "." + serviceProto + ".local");
+
+  auto service =
+      mdnsd_register_svc(implMDNSService::mdnsServer, serviceName.c_str(),
+                         type.c_str(), servicePort, NULL, txt.data());
+
+  return service ? std::make_unique<implMDNSService>(service) : nullptr;
+}