From 28d841e7d06a3a8e03d3914f3836b1d5e0879eb2 Mon Sep 17 00:00:00 2001 From: Carlos Gomes Martinho Date: Wed, 26 Feb 2020 23:16:09 +0100 Subject: [PATCH] chore: make test_package smaller --- recipes/mdns/all/test_package/test_package.c | 328 +------------------ 1 file changed, 4 insertions(+), 324 deletions(-) diff --git a/recipes/mdns/all/test_package/test_package.c b/recipes/mdns/all/test_package/test_package.c index 9909ca691ddf9..b27d9acd4f04b 100644 --- a/recipes/mdns/all/test_package/test_package.c +++ b/recipes/mdns/all/test_package/test_package.c @@ -3,7 +3,7 @@ # define _CRT_SECURE_NO_WARNINGS 1 #endif -#include +#include "mdns.h" #include #include @@ -15,165 +15,6 @@ # include #endif -static char addrbuffer[64]; -static char namebuffer[256]; -static char sendbuffer[256]; -static mdns_record_txt_t txtbuffer[128]; - -typedef struct { - const char* service; - const char* hostname; - uint32_t address_ipv4; - uint8_t* address_ipv6; - int port; -} service_record_t; - -static mdns_string_t -ipv4_address_to_string(char* buffer, size_t capacity, const struct sockaddr_in* addr, size_t addrlen) { - char host[NI_MAXHOST] = {0}; - char service[NI_MAXSERV] = {0}; - int ret = getnameinfo((const struct sockaddr*)addr, addrlen, - host, NI_MAXHOST, service, NI_MAXSERV, - NI_NUMERICSERV | NI_NUMERICHOST); - int len = 0; - if (ret == 0) { - if (addr->sin_port != 0) - len = snprintf(buffer, capacity, "%s:%s", host, service); - else - len = snprintf(buffer, capacity, "%s", host); - } - if (len >= (int)capacity) - len = (int)capacity - 1; - mdns_string_t str = {buffer, len}; - return str; -} - -static mdns_string_t -ipv6_address_to_string(char* buffer, size_t capacity, const struct sockaddr_in6* addr, size_t addrlen) { - char host[NI_MAXHOST] = {0}; - char service[NI_MAXSERV] = {0}; - int ret = getnameinfo((const struct sockaddr*)addr, addrlen, - host, NI_MAXHOST, service, NI_MAXSERV, - NI_NUMERICSERV | NI_NUMERICHOST); - int len = 0; - if (ret == 0) { - if (addr->sin6_port != 0) - len = snprintf(buffer, capacity, "[%s]:%s", host, service); - else - len = snprintf(buffer, capacity, "%s", host); - } - if (len >= (int)capacity) - len = (int)capacity - 1; - mdns_string_t str = {buffer, len}; - return str; -} - -static mdns_string_t -ip_address_to_string(char* buffer, size_t capacity, const struct sockaddr* addr, size_t addrlen) { - if (addr->sa_family == AF_INET6) - return ipv6_address_to_string(buffer, capacity, (const struct sockaddr_in6*)addr, addrlen); - return ipv4_address_to_string(buffer, capacity, (const struct sockaddr_in*)addr, addrlen); -} - -static int -query_callback(int sock, const struct sockaddr* from, size_t addrlen, - mdns_entry_type_t entry, uint16_t transaction_id, - uint16_t rtype, uint16_t rclass, uint32_t ttl, - const void* data, size_t size, size_t offset, size_t length, - void* user_data) { - mdns_string_t fromaddrstr = ip_address_to_string(addrbuffer, sizeof(addrbuffer), from, addrlen); - const char* entrytype = (entry == MDNS_ENTRYTYPE_ANSWER) ? "answer" : - ((entry == MDNS_ENTRYTYPE_AUTHORITY) ? "authority" : "additional"); - if (rtype == MDNS_RECORDTYPE_PTR) { - mdns_string_t namestr = mdns_record_parse_ptr(data, size, offset, length, - namebuffer, sizeof(namebuffer)); - printf("%.*s : %s PTR %.*s rclass 0x%x ttl %u length %d\n", - MDNS_STRING_FORMAT(fromaddrstr), entrytype, - MDNS_STRING_FORMAT(namestr), rclass, ttl, (int)length); - } - else if (rtype == MDNS_RECORDTYPE_SRV) { - mdns_record_srv_t srv = mdns_record_parse_srv(data, size, offset, length, - namebuffer, sizeof(namebuffer)); - printf("%.*s : %s SRV %.*s priority %d weight %d port %d\n", - MDNS_STRING_FORMAT(fromaddrstr), entrytype, - MDNS_STRING_FORMAT(srv.name), srv.priority, srv.weight, srv.port); - } - else if (rtype == MDNS_RECORDTYPE_A) { - struct sockaddr_in addr; - mdns_record_parse_a(data, size, offset, length, &addr); - mdns_string_t addrstr = ipv4_address_to_string(namebuffer, sizeof(namebuffer), &addr, sizeof(addr)); - printf("%.*s : %s A %.*s\n", - MDNS_STRING_FORMAT(fromaddrstr), entrytype, - MDNS_STRING_FORMAT(addrstr)); - } - else if (rtype == MDNS_RECORDTYPE_AAAA) { - struct sockaddr_in6 addr; - mdns_record_parse_aaaa(data, size, offset, length, &addr); - mdns_string_t addrstr = ipv6_address_to_string(namebuffer, sizeof(namebuffer), &addr, sizeof(addr)); - printf("%.*s : %s AAAA %.*s\n", - MDNS_STRING_FORMAT(fromaddrstr), entrytype, - MDNS_STRING_FORMAT(addrstr)); - } - else if (rtype == MDNS_RECORDTYPE_TXT) { - size_t parsed = mdns_record_parse_txt(data, size, offset, length, - txtbuffer, sizeof(txtbuffer) / sizeof(mdns_record_txt_t)); - for (size_t itxt = 0; itxt < parsed; ++itxt) { - if (txtbuffer[itxt].value.length) { - printf("%.*s : %s TXT %.*s = %.*s\n", - MDNS_STRING_FORMAT(fromaddrstr), entrytype, - MDNS_STRING_FORMAT(txtbuffer[itxt].key), - MDNS_STRING_FORMAT(txtbuffer[itxt].value)); - } - else { - printf("%.*s : %s TXT %.*s\n", - MDNS_STRING_FORMAT(fromaddrstr), entrytype, - MDNS_STRING_FORMAT(txtbuffer[itxt].key)); - } - } - } - else { - printf("%.*s : %s type %u rclass 0x%x ttl %u length %d\n", - MDNS_STRING_FORMAT(fromaddrstr), entrytype, - rtype, rclass, ttl, (int)length); - } - return 0; -} - -static int -service_callback(int sock, const struct sockaddr* from, size_t addrlen, - mdns_entry_type_t entry, uint16_t transaction_id, - uint16_t rtype, uint16_t rclass, uint32_t ttl, - const void* data, size_t size, size_t offset, size_t length, - void* user_data) { - if (entry != MDNS_ENTRYTYPE_QUESTION) - return 0; - mdns_string_t fromaddrstr = ip_address_to_string(addrbuffer, sizeof(addrbuffer), from, addrlen); - if (rtype == MDNS_RECORDTYPE_PTR) { - mdns_string_t service = mdns_string_extract(data, size, &offset, - namebuffer, sizeof(namebuffer)); - printf("%.*s : question PTR %.*s\n", - MDNS_STRING_FORMAT(fromaddrstr), MDNS_STRING_FORMAT(service)); - - const char dns_sd[] = "_services._dns-sd._udp.local."; - const service_record_t* service_record = (const service_record_t*)user_data; - size_t service_length = strlen(service_record->service); - if ((service.length == (sizeof(dns_sd) - 1)) && (strcmp(service.str, dns_sd) == 0)) { - printf(" --> answer %s\n", service_record->service); - mdns_discovery_answer(sock, from, addrlen, sendbuffer, - sizeof(sendbuffer), service_record->service, service_length); - } - else if ((service.length == service_length) && (strcmp(service.str, service_record->service) == 0)) { - printf(" --> answer %s.%s port %d\n", service_record->hostname, service_record->service, service_record->port); - mdns_query_answer(sock, from, addrlen, sendbuffer, sizeof(sendbuffer), - transaction_id, service_record->service, service_length, - service_record->hostname, strlen(service_record->hostname), - service_record->address_ipv4, service_record->address_ipv6, - (uint16_t)service_record->port, 0, 0); - } - } - return 0; -} - int main(int argc, const char* const* argv) { #ifdef _WIN32 @@ -185,179 +26,18 @@ main(int argc, const char* const* argv) { } #endif - size_t capacity = 2048; - void* buffer = malloc(capacity); - void* user_data = 0; - size_t records; - - int mode = 0; - const char* service = "_test_mdns._tcp.local."; - const char* hostname = "dummy-host"; - int service_port = 42424; - for (int iarg = 0; iarg < argc; ++iarg) { - if (strcmp(argv[iarg], "--discovery") == 0) { - mode = 0; - } - else if (strcmp(argv[iarg], "--query") == 0) { - mode = 1; - ++iarg; - if (iarg < argc) - service = argv[iarg]; - } - else if (strcmp(argv[iarg], "--service") == 0) { - mode = 2; - ++iarg; - if (iarg < argc) - service = argv[iarg]; - } - else if (strcmp(argv[iarg], "--hostname") == 0) { - ++iarg; - if (iarg < argc) - hostname = argv[iarg]; - } - else if (strcmp(argv[iarg], "--port") == 0) { - ++iarg; - if (iarg < argc) - service_port = atoi(argv[iarg]); - } - } - - int port = (mode == 2) ? MDNS_PORT : 0; + int port = 0; int sock = mdns_socket_open_ipv4(port); if (sock < 0) { printf("Failed to open socket: %s\n", strerror(errno)); return -1; } - printf("Opened IPv4 socket for mDNS/DNS-SD\n"); - - if (mode == 0) { - printf("Sending DNS-SD discovery\n"); - if (mdns_discovery_send(sock)) { - printf("Failed to send DNS-DS discovery: %s\n", strerror(errno)); - goto quit; - } - - printf("Reading DNS-SD replies\n"); - for (int i = 0; i < 10; ++i) { - do { - records = mdns_discovery_recv(sock, buffer, capacity, query_callback, - user_data); - } while (records); - if (records) - i = 0; - sleep(1); - } - } else if (mode == 1) { - printf("Sending mDNS query: %s\n", service); - if (mdns_query_send(sock, MDNS_RECORDTYPE_PTR, - service, strlen(service), - buffer, capacity)) { - printf("Failed to send mDNS query: %s\n", strerror(errno)); - goto quit; - } - - printf("Reading mDNS replies\n"); - for (int i = 0; i < 5; ++i) { - do { - records = mdns_query_recv(sock, buffer, capacity, query_callback, user_data, 1); - } while (records); - if (records) - i = 0; - sleep(1); - } - } else if (mode == 2) { - printf("Service mDNS: %s\n", service); -#ifdef _WIN32 - unsigned long param = 0; - ioctlsocket(sock, FIONBIO, ¶m); -#else - const int flags = fcntl(sock, F_GETFL, 0); - fcntl(sock, F_SETFL, flags & ~O_NONBLOCK); -#endif - - uint32_t address_ipv4 = 0; - uint8_t* address_ipv6 = 0; - uint8_t address_ipv6_buffer[16]; - -#ifdef _WIN32 - IP_ADAPTER_ADDRESSES* adapter_address = 0; - unsigned int address_size = 8000; - unsigned int ret; - unsigned int num_retries = 4; - do { - adapter_address = malloc(address_size); - ret = GetAdaptersAddresses(AF_UNSPEC, GAA_FLAG_SKIP_MULTICAST | GAA_FLAG_SKIP_ANYCAST, 0, - adapter_address, &address_size); - if (ret == ERROR_BUFFER_OVERFLOW) { - free(adapter_address); - adapter_address = 0; - } - else { - break; - } - } - while (num_retries-- > 0); - - if (!adapter_address || (ret != NO_ERROR)) { - free(adapter_address); - printf("Failed to get network adapter addresses\n"); - goto quit; - } - - for (PIP_ADAPTER_ADDRESSES adapter = adapter_address; adapter; 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 (unicast->Address.lpSockaddr->sa_family == AF_INET) { - struct sockaddr_in* saddr = (struct sockaddr_in*)unicast->Address.lpSockaddr; - if ((saddr->sin_addr.S_un.S_un_b.s_b1 != 127) || (saddr->sin_addr.S_un.S_un_b.s_b2 != 0) || - (saddr->sin_addr.S_un.S_un_b.s_b3 != 0) || (saddr->sin_addr.S_un.S_un_b.s_b4 != 1)) { - mdns_string_t addr = ipv4_address_to_string(buffer, capacity, saddr, sizeof(struct sockaddr_in)); - printf("Local IPv4 address: %.*s\n", MDNS_STRING_FORMAT(addr)); - address_ipv4 = saddr->sin_addr.S_un.S_addr; - } - } - else if (unicast->Address.lpSockaddr->sa_family == AF_INET6) { - if ((unicast->DadState == NldsPreferred) && !address_ipv6) { - struct sockaddr_in6* saddr = (struct sockaddr_in6*)unicast->Address.lpSockaddr; - memcpy(address_ipv6_buffer, &saddr->sin6_addr, 16); - address_ipv6 = address_ipv6_buffer; - mdns_string_t addr = ipv6_address_to_string(buffer, capacity, saddr, sizeof(struct sockaddr_in6)); - printf("Local IPv6 address: %.*s\n", MDNS_STRING_FORMAT(addr)); - } - } - } - } -#endif - - service_record_t service_record = { - service, - hostname, - address_ipv4, - address_ipv6, - service_port - }; - - int error_code = 0; - do { - mdns_socket_listen(sock, buffer, capacity, service_callback, &service_record); - int error_code_size = sizeof(error_code); - getsockopt(sock, SOL_SOCKET, SO_ERROR, (char*)&error_code, (socklen_t*)&error_code_size); - } while (!error_code); - } - -quit: - free(buffer); + printf("socket cleanup\n"); mdns_socket_close(sock); - printf("Closed socket\n"); - #ifdef _WIN32 WSACleanup(); #endif return 0; -} \ No newline at end of file +}