forked from gronod/squeezelite-esp32
109 lines
3.3 KiB
C++
109 lines
3.3 KiB
C++
#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;
|
|
}
|