30#if defined(_WIN32) && !defined(_CRT_SECURE_NO_WARNINGS)
31#define _CRT_SECURE_NO_WARNINGS 1
41#define sleep(x) Sleep(x * 1000)
51#ifndef OCPN_ENABLE_MDNS_DEBUG
55static char addrbuffer[64];
56static char namebuffer[256];
57static char sendbuffer[1024];
59static struct sockaddr_in service_address_ipv4;
60static struct sockaddr_in6 service_address_ipv6;
62volatile sig_atomic_t running_server = 1;
66 mdns_string_t service;
67 mdns_string_t hostname;
68 mdns_string_t service_instance;
69 mdns_string_t hostname_qualified;
70 struct sockaddr_in address_ipv4;
71 struct sockaddr_in6 address_ipv6;
73 mdns_record_t record_ptr;
74 mdns_record_t record_srv;
75 mdns_record_t record_a;
76 mdns_record_t record_aaaa;
77 mdns_record_t txt_record[2];
81int ocpn_service_callback(
int sock,
const struct sockaddr* from,
size_t addrlen,
82 mdns_entry_type_t entry, uint16_t query_id,
83 uint16_t rtype, uint16_t rclass, uint32_t ttl,
84 const void* data,
size_t size,
size_t name_offset,
85 size_t name_length,
size_t record_offset,
86 size_t record_length,
void* user_data) {
88 if (entry != MDNS_ENTRYTYPE_QUESTION)
return 0;
90 const char dns_sd[] =
"_services._dns-sd._udp.local.";
93 mdns_string_t fromaddrstr =
94 ip_address_to_string(addrbuffer,
sizeof(addrbuffer), from, addrlen);
96 size_t offset = name_offset;
98 mdns_string_extract(data, size, &offset, namebuffer,
sizeof(namebuffer));
100 const char* record_name = 0;
101 if (rtype == MDNS_RECORDTYPE_PTR)
103 else if (rtype == MDNS_RECORDTYPE_SRV)
105 else if (rtype == MDNS_RECORDTYPE_A)
107 else if (rtype == MDNS_RECORDTYPE_AAAA)
108 record_name =
"AAAA";
109 else if (rtype == MDNS_RECORDTYPE_TXT)
111 else if (rtype == MDNS_RECORDTYPE_ANY)
115 printf(
"Query %s %.*s\n", record_name, MDNS_STRING_FORMAT(name));
117 if ((name.length == (
sizeof(dns_sd) - 1)) &&
118 (strncmp(name.str, dns_sd,
sizeof(dns_sd) - 1) == 0)) {
119 if ((rtype == MDNS_RECORDTYPE_PTR) || (rtype == MDNS_RECORDTYPE_ANY)) {
126 mdns_record_t answer;
128 answer.type = MDNS_RECORDTYPE_PTR;
129 answer.data.ptr.name = service->service;
132 uint16_t unicast = (rclass & MDNS_UNICAST_RESPONSE);
133 printf(
" --> answer %.*s (%s)\n",
134 MDNS_STRING_FORMAT(answer.data.ptr.name),
135 (unicast ?
"unicast" :
"multicast"));
138 mdns_query_answer_unicast(sock, from, addrlen, sendbuffer,
139 sizeof(sendbuffer), query_id,
140 (mdns_record_type_t)rtype, name.str,
141 name.length, answer, 0, 0, 0, 0);
143 mdns_query_answer_multicast(sock, sendbuffer,
sizeof(sendbuffer),
147 }
else if ((name.length == service->service.length) &&
148 (strncmp(name.str, service->service.str, name.length) == 0)) {
149 if ((rtype == MDNS_RECORDTYPE_PTR) || (rtype == MDNS_RECORDTYPE_ANY)) {
161 mdns_record_t answer = service->record_ptr;
163 mdns_record_t additional[5]{{}};
164 size_t additional_count = 0;
168 additional[additional_count++] = service->record_srv;
171 if (service->address_ipv4.sin_family == AF_INET)
172 additional[additional_count++] = service->record_a;
173 if (service->address_ipv6.sin6_family == AF_INET6)
174 additional[additional_count++] = service->record_aaaa;
183 uint16_t unicast = (rclass & MDNS_UNICAST_RESPONSE);
184 printf(
" --> answer %.*s (%s)\n",
185 MDNS_STRING_FORMAT(service->record_ptr.data.ptr.name),
186 (unicast ?
"unicast" :
"multicast"));
189 mdns_query_answer_unicast(
190 sock, from, addrlen, sendbuffer,
sizeof(sendbuffer), query_id,
191 (mdns_record_type)rtype, name.str, name.length, answer, 0, 0,
192 additional, additional_count);
194 mdns_query_answer_multicast(sock, sendbuffer,
sizeof(sendbuffer),
195 answer, 0, 0, additional, additional_count);
198 }
else if ((name.length == service->service_instance.length) &&
199 (strncmp(name.str, service->service_instance.str, name.length) ==
201 if ((rtype == MDNS_RECORDTYPE_SRV) || (rtype == MDNS_RECORDTYPE_ANY)) {
210 mdns_record_t answer = service->record_srv;
212 mdns_record_t additional[5]{{}};
213 size_t additional_count = 0;
216 if (service->address_ipv4.sin_family == AF_INET)
217 additional[additional_count++] = service->record_a;
218 if (service->address_ipv6.sin6_family == AF_INET6)
219 additional[additional_count++] = service->record_aaaa;
224 additional[additional_count++] = service->txt_record[0];
225 additional[additional_count++] = service->txt_record[1];
228 uint16_t unicast = (rclass & MDNS_UNICAST_RESPONSE);
229 printf(
" --> answer %.*s port %d (%s)\n",
230 MDNS_STRING_FORMAT(service->record_srv.data.srv.name),
231 service->port, (unicast ?
"unicast" :
"multicast"));
234 mdns_query_answer_unicast(
235 sock, from, addrlen, sendbuffer,
sizeof(sendbuffer), query_id,
236 (mdns_record_type)rtype, name.str, name.length, answer, 0, 0,
237 additional, additional_count);
239 mdns_query_answer_multicast(sock, sendbuffer,
sizeof(sendbuffer),
240 answer, 0, 0, additional, additional_count);
243 }
else if ((name.length == service->hostname_qualified.length) &&
244 (strncmp(name.str, service->hostname_qualified.str, name.length) ==
246 if (((rtype == MDNS_RECORDTYPE_A) || (rtype == MDNS_RECORDTYPE_ANY)) &&
247 (service->address_ipv4.sin_family == AF_INET)) {
254 mdns_record_t answer = service->record_a;
256 mdns_record_t additional[5]{{}};
257 size_t additional_count = 0;
260 if (service->address_ipv6.sin6_family == AF_INET6)
261 additional[additional_count++] = service->record_aaaa;
266 additional[additional_count++] = service->txt_record[0];
267 additional[additional_count++] = service->txt_record[1];
270 uint16_t unicast = (rclass & MDNS_UNICAST_RESPONSE);
271 mdns_string_t addrstr =
272 ip_address_to_string(addrbuffer,
sizeof(addrbuffer),
273 (
struct sockaddr*)&service->record_a.data.a.addr,
274 sizeof(service->record_a.data.a.addr));
275 printf(
" --> answer %.*s IPv4 %.*s (%s)\n",
276 MDNS_STRING_FORMAT(service->record_a.name),
277 MDNS_STRING_FORMAT(addrstr), (unicast ?
"unicast" :
"multicast"));
280 mdns_query_answer_unicast(
281 sock, from, addrlen, sendbuffer,
sizeof(sendbuffer), query_id,
282 (mdns_record_type)rtype, name.str, name.length, answer, 0, 0,
283 additional, additional_count);
285 mdns_query_answer_multicast(sock, sendbuffer,
sizeof(sendbuffer),
286 answer, 0, 0, additional, additional_count);
288 }
else if (((rtype == MDNS_RECORDTYPE_AAAA) ||
289 (rtype == MDNS_RECORDTYPE_ANY)) &&
290 (service->address_ipv6.sin6_family == AF_INET6)) {
297 mdns_record_t answer = service->record_aaaa;
299 mdns_record_t additional[5]{{}};
300 size_t additional_count = 0;
303 if (service->address_ipv4.sin_family == AF_INET)
304 additional[additional_count++] = service->record_a;
309 additional[additional_count++] = service->txt_record[0];
310 additional[additional_count++] = service->txt_record[1];
313 uint16_t unicast = (rclass & MDNS_UNICAST_RESPONSE);
314 mdns_string_t addrstr = ip_address_to_string(
315 addrbuffer,
sizeof(addrbuffer),
316 (
struct sockaddr*)&service->record_aaaa.data.aaaa.addr,
317 sizeof(service->record_aaaa.data.aaaa.addr));
318 printf(
" --> answer %.*s IPv6 %.*s (%s)\n",
319 MDNS_STRING_FORMAT(service->record_aaaa.name),
320 MDNS_STRING_FORMAT(addrstr), (unicast ?
"unicast" :
"multicast"));
323 mdns_query_answer_unicast(
324 sock, from, addrlen, sendbuffer,
sizeof(sendbuffer), query_id,
325 (mdns_record_type)rtype, name.str, name.length, answer, 0, 0,
326 additional, additional_count);
328 mdns_query_answer_multicast(sock, sendbuffer,
sizeof(sendbuffer),
329 answer, 0, 0, additional, additional_count);
337void service_mdns(
const char* hostname,
const char* service_name,
341 open_service_sockets(sockets,
sizeof(sockets) /
sizeof(sockets[0]));
342 if (num_sockets <= 0) {
343 printf(
"Failed to open any client sockets\n");
346 printf(
"Opened %d socket%s for mDNS service\n", num_sockets,
347 num_sockets ?
"s" :
"");
349 size_t service_name_length = strlen(service_name);
350 if (!service_name_length) {
351 printf(
"Invalid service name\n");
355 char* service_name_buffer = (
char*)malloc(service_name_length + 2);
356 memcpy(service_name_buffer, service_name, service_name_length);
357 if (service_name_buffer[service_name_length - 1] !=
'.')
358 service_name_buffer[service_name_length++] =
'.';
359 service_name_buffer[service_name_length] = 0;
360 service_name = service_name_buffer;
362 printf(
"Service mDNS: %s:%d\n", service_name, service_port);
363 printf(
"Hostname: %s\n", hostname);
365 size_t capacity = 2048;
366 void* buffer = malloc(capacity);
368 mdns_string_t service_string;
369 service_string.str = service_name;
370 service_string.length = strlen(service_name);
371 mdns_string_t hostname_string;
372 hostname_string.str = hostname;
373 hostname_string.length = strlen(hostname);
376 char service_instance_buffer[256] = {0};
377 snprintf(service_instance_buffer,
sizeof(service_instance_buffer) - 1,
378 "%.*s.%.*s", MDNS_STRING_FORMAT(hostname_string),
379 MDNS_STRING_FORMAT(service_string));
381 mdns_string_t service_instance_string;
382 service_instance_string.str = service_instance_buffer;
383 service_instance_string.length = strlen(service_instance_buffer);
386 char qualified_hostname_buffer[256] = {0};
387 snprintf(qualified_hostname_buffer,
sizeof(qualified_hostname_buffer) - 1,
388 "%.*s.local.", MDNS_STRING_FORMAT(hostname_string));
389 mdns_string_t hostname_qualified_string;
390 hostname_qualified_string.str = qualified_hostname_buffer;
391 hostname_qualified_string.length = strlen(qualified_hostname_buffer);
394 service.service = service_string;
395 service.hostname = hostname_string;
396 service.service_instance = service_instance_string;
397 service.hostname_qualified = hostname_qualified_string;
398 service.address_ipv4 = service_address_ipv4;
399 service.address_ipv6 = service_address_ipv6;
400 service.port = service_port;
406 service.record_ptr.name = service.service;
407 service.record_ptr.type = MDNS_RECORDTYPE_PTR;
408 service.record_ptr.rclass = 0;
409 service.record_ptr.ttl = 0;
410 service.record_ptr.data.ptr.name = service.service_instance;
414 service.record_srv.name = service.service_instance;
415 service.record_srv.type = MDNS_RECORDTYPE_SRV;
416 service.record_srv.data.srv.name = service.hostname_qualified;
417 service.record_srv.data.srv.port = service.port;
418 service.record_srv.data.srv.priority = 0;
419 service.record_srv.data.srv.weight = 0;
420 service.record_srv.rclass = 0;
421 service.record_srv.ttl = 0;
425 service.record_a.name = service.hostname_qualified;
426 service.record_a.type = MDNS_RECORDTYPE_A;
427 service.record_a.data.a.addr = service.address_ipv4;
428 service.record_a.rclass = 0;
429 service.record_a.ttl = 0;
431 service.record_aaaa.name = service.hostname_qualified;
432 service.record_aaaa.type = MDNS_RECORDTYPE_AAAA;
433 service.record_aaaa.data.aaaa.addr = service.address_ipv6;
434 service.record_aaaa.rclass = 0;
435 service.record_aaaa.ttl = 0;
439 printf(
"Sending announce\n");
440 mdns_record_t additional[5]{{}};
441 size_t additional_count = 0;
442 additional[additional_count++] = service.record_srv;
443 if (service.address_ipv4.sin_family == AF_INET)
444 additional[additional_count++] = service.record_a;
445 if (service.address_ipv6.sin6_family == AF_INET6)
446 additional[additional_count++] = service.record_aaaa;
450 for (
int isock = 0; isock < num_sockets; ++isock)
451 mdns_announce_multicast(sockets[isock], buffer, capacity,
452 service.record_ptr, 0, 0, additional,
457 while (running_server) {
461 for (
int isock = 0; isock < num_sockets; ++isock) {
462 if (sockets[isock] >= nfds) nfds = sockets[isock] + 1;
463 FD_SET(sockets[isock], &readfs);
466 struct timeval timeout;
468 timeout.tv_usec = 100000;
470 if (select(nfds, &readfs, 0, 0, &timeout) >= 0) {
471 for (
int isock = 0; isock < num_sockets; ++isock) {
472 if (FD_ISSET(sockets[isock], &readfs)) {
473 mdns_socket_listen(sockets[isock], buffer, capacity,
474 ocpn_service_callback, &service);
476 FD_SET(sockets[isock], &readfs);
485 printf(
"Sending goodbye\n");
486 mdns_record_t additional[5]{{}};
487 size_t additional_count = 0;
488 additional[additional_count++] = service.record_srv;
489 if (service.address_ipv4.sin_family == AF_INET)
490 additional[additional_count++] = service.record_a;
491 if (service.address_ipv6.sin6_family == AF_INET6)
492 additional[additional_count++] = service.record_aaaa;
496 for (
int isock = 0; isock < num_sockets; ++isock)
497 mdns_goodbye_multicast(sockets[isock], buffer, capacity,
498 service.record_ptr, 0, 0, additional,
503 free(service_name_buffer);
505 for (
int isock = 0; isock < num_sockets; ++isock)
506 mdns_socket_close(sockets[isock]);
507 printf(
"Closed socket%s\n", num_sockets ?
"s" :
"");
515int StartMDNSService(std::string hostname, std::string service_name,
518 service = service_name;
520 std::thread mdns_service_thread(service_mdns, host.c_str(), service.c_str(),
522 mdns_service_thread.detach();
527bool StopMDNSService() {
return true; }