%% options copyright owner = Dirk Krause copyright year = 2015-xxxx license = bsd %% module #include "dk4sock.h" #include "dk4mem.h" #if DK4_HAVE_GETADDRINFO #include "dk4maowd.h" #else #include "dk4maiwdus.h" #endif #include "dk4rec24.h" $!trace-include #if DK4_CHAR_SIZE > 1 #if DK4_ON_WINDOWS /* +++++ +wchar_t +Windows */ /** Create listener socket set for a local port number. @param servicename Local service name. @param backlog Number of connection requests to queue. @param local Flag: Allow local connections only. @param numeric Flag: Service name is numeric. @param erp Error report, may be NULL. @return Socket set on success, NULL on error. */ static dk4_socket_set_t * dk4socket_wc_set_service_num_flag( const wchar_t *servicename, int backlog, int local, int numeric, dk4_er_t *erp ) { #if DK4_HAVE_GETADDRINFO /* +++++ +getaddrinfo */ ADDRINFOW hints; ADDRINFOW *addri = NULL; ADDRINFOW *paddr = NULL; dk4_socket_set_t *back = NULL; size_t numad = 0; int res = 0; #if DK4_HAVE_SETSOCKOPT && (defined(SO_REUSEPORT) || defined(SO_REUSEADDR)) int opt = 0; #endif if (NULL != servicename) { DK4_MEMRES(&hints, sizeof(hints)); hints.ai_family = AF_UNSPEC; hints.ai_socktype = SOCK_STREAM; hints.ai_flags = 0; hints.ai_protocol = 0; hints.ai_addrlen = 0; hints.ai_canonname = NULL; hints.ai_addr = NULL; hints.ai_next = NULL; if (0 == local) { hints.ai_flags |= AI_PASSIVE; } if (0 != numeric) { hints.ai_flags |= AI_NUMERICSERV; } res = GetAddrInfoW(NULL, servicename, &hints, &addri); if ((0 == res) && (NULL != addri)) { paddr = addri; while (NULL != paddr) { numad++; paddr = paddr->ai_next; } if (0 < numad) { paddr = addri; back = dk4socket_set_new(numad, erp); if (NULL != back) { while (NULL != paddr) { (back->pSockets)[back->szUsed] = dk4socket_open( paddr->ai_family, SOCK_STREAM, 0, erp ); if (INVALID_SOCKET != (back->pSockets)[back->szUsed]) { #if DK4_HAVE_SETSOCKOPT && defined(SO_REUSEPORT) $? ". SO_REUSEPORT" dk4socket_set_one_bytes(&opt, sizeof(opt)); res = setsockopt( (back->pSockets)[back->szUsed], SOL_SOCKET, SO_REUSEPORT, (void *)(&opt), sizeof(opt) ); if (0 != res) { $? "! failed to reuse port number" } #else #if DK4_HAVE_SETSOCKOPT && defined(SO_REUSEADDR) $? ". SO_REUSEADDR" dk4socket_set_one_bytes(&opt, sizeof(opt)); res = setsockopt( (back->pSockets)[back->szUsed], SOL_SOCKET, SO_REUSEADDR, (void *)(&opt), sizeof(opt) ); if (0 != res) { $? "! failed to reuse address" } #else $? ". neither SO_REUSEPORT nor SO_REUSEADDR" #endif #endif res = dk4socket_bind( (back->pSockets)[back->szUsed], paddr->ai_addr, paddr->ai_addrlen, erp ); if (DK4_SOCKET_RESULT_SUCCESS == res) { #if 0 /* Setting SO_REUSEADDR or SO_REUSEPORT must follow immediately on socket(). */ (void)dk4socket_option_set_reuse(back, 1, NULL); #endif res = dk4socket_listen( (back->pSockets)[back->szUsed], backlog, erp ); if (DK4_SOCKET_RESULT_SUCCESS == res) { back->szUsed += 1; } else { (void)dk4socket_close((back->pSockets)[back->szUsed], NULL); (back->pSockets)[back->szUsed] = INVALID_SOCKET; } } else { (void)dk4socket_close((back->pSockets)[back->szUsed], NULL); (back->pSockets)[back->szUsed] = INVALID_SOCKET; } } paddr = paddr->ai_next; } if (0 == back->szUsed) { dk4socket_set_delete(back, NULL); back = NULL; } } } FreeAddrInfo(addri); } else { dk4error_set_idetails(erp, DK4_E_SOCKET_GETADDRLOC, res); } } else { dk4error_set_simple_error_code(erp, DK4_E_INVALID_ARGUMENTS); } return back; /* ----- +getaddrinfo */ #else /* +++++ -getaddrinfo */ char asvcn[128]; const wchar_t *endptr = NULL; dk4_socket_set_t *back = NULL; struct servent *serv = NULL; int iport = 0; int res = 0; unsigned short portno = 0; if (NULL != servicename) { res = dk4ma_input_wc_dec_ushort( &portno, servicename, &endptr, 1, erp ); if (0 != res) { back = dk4socket_set_server_directly(portno, backlog, local, erp); } else { res = dk4recode_wchar_t_to_char(asvcn, sizeof(asvcn), servicename, erp); if (0 != res) { serv = getservbyname(asvcn, NULL); if (NULL != serv) { iport = serv->s_port; dk4socket_swap_bytes_if_not_bigendian((void *)(&iport),sizeof(iport)); portno = (unsigned short)iport; back = dk4socket_set_server_directly(portno, backlog, local, erp); } else { dk4socket_error_report(erp, DK4_E_SOCKET_GETSERVBYNAME); } } } } else { dk4error_set_simple_error_code(erp, DK4_E_INVALID_ARGUMENTS); } return back; /* ----- -getaddrinfo */ #endif } static dk4_socket_set_t * dk4socket_wc_set_server( unsigned short portno, int backlog, int local, dk4_er_t *erp ) { #if DK4_HAVE_GETADDRINFO /* +++++ +getaddrinfo */ wchar_t pnb[16*sizeof(dk4_um_t)]; dk4_socket_set_t *back = NULL; int res; if (0 < portno) { res = dk4ma_write_wc_decimal_unsigned( pnb, DK4_SIZEOF(pnb,wchar_t), (dk4_um_t)portno, 0, erp ); if (0 != res) { back = dk4socket_wc_set_service_num_flag( pnb, backlog, local, 1, erp ); } } else { dk4error_set_simple_error_code(erp, DK4_E_INVALID_ARGUMENTS); } return back; /* ----- +getaddrinfo */ #else /* +++++ -getaddrinfo */ return ( dk4socket_set_server_directly(portno, backlog, local, erp) ); /* ----- -getaddrinfo */ #endif } static dk4_socket_set_t * dk4socket_wc_set_service( const wchar_t *servicename, int backlog, int local, dk4_er_t *erp ) { return ( dk4socket_wc_set_service_num_flag(servicename, backlog, local, 0, erp) ); } /* ----- +wchar_t +Windows */ #else #error No networking functions using wchar_t host and service names available! #endif #endif dk4_socket_set_t * dk4socket_set_server( unsigned short portno, int backlog, int local, dk4_er_t *erp ) { #if DK4_CHAR_SIZE > 1 return ( dk4socket_wc_set_server(portno, backlog, local, erp) ); #else return ( dk4socket_c8_set_server(portno, backlog, local, erp) ); #endif } dk4_socket_set_t * dk4socket_set_service( const dkChar *servicename, int backlog, int local, dk4_er_t *erp ) { #if DK4_CHAR_SIZE > 1 return ( dk4socket_wc_set_service(servicename, backlog, local, erp) ); #else return ( dk4socket_c8_set_service(servicename, backlog, local, erp) ); #endif }