%% options copyright owner = Dirk Krause copyright year = 2015-xxxx license = bsd %% module #include "dk4sock.h" #include "dk4mem.h" #if DK4_HAVE_GETADDRINFO #include "dk4mao8d.h" #else #include "dk4mai8dus.h" #endif $!trace-include /** 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_c8_set_service_num_flag( const char *servicename, int backlog, int local, int numeric, dk4_er_t *erp ) { #if DK4_HAVE_GETADDRINFO /* +++++ +getaddrinfo */ struct addrinfo hints; struct addrinfo *addri = NULL; struct addrinfo *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 $? "+ dk4socket_c8_set_service_num_flag" 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 = getaddrinfo(NULL, servicename, &hints, &addri); if ((0 == res) && (NULL != addri)) { paddr = addri; while (NULL != paddr) { numad++; paddr = paddr->ai_next; } if (0 < numad) { $? ". numad = %u", (unsigned)numad paddr = addri; back = dk4socket_set_new(numad, erp); if (NULL != back) { while (NULL != paddr) { $? ". try next address" (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) { res = dk4socket_listen( (back->pSockets)[back->szUsed], backlog, erp ); if (DK4_SOCKET_RESULT_SUCCESS == res) { back->szUsed += 1; $? ". used = %u", (unsigned)(back->szUsed) } else { $? "! failed to listen" (void)dk4socket_close((back->pSockets)[back->szUsed], NULL); (back->pSockets)[back->szUsed] = INVALID_SOCKET; } } else { $? "! failed to bind address" (void)dk4socket_close((back->pSockets)[back->szUsed], NULL); (back->pSockets)[back->szUsed] = INVALID_SOCKET; } } else { $? "! failed to create 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); } $? "- dk4socket_c8_set_service_num_flag" return back; /* ----- +getaddrinfo */ #else /* +++++ -getaddrinfo */ const char *endptr = NULL; dk4_socket_set_t *back = NULL; struct servent *serv = NULL; int iport = 0; int res = 0; unsigned short portno = 0; $? "+ dk4socket_c8_set_service_num_flag" if (NULL != servicename) { res = dk4ma_input_c8_dec_ushort( &portno, servicename, &endptr, 1, NULL ); if (0 != res) { back = dk4socket_set_server_directly(portno, backlog, local, erp); } else { dk4socket_error_reset(); serv = getservbyname(servicename, 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); } $? "- dk4socket_c8_set_service_num_flag" return back; /* ----- -getaddrinfo */ #endif } dk4_socket_set_t * dk4socket_c8_set_server( unsigned short portno, int backlog, int local, dk4_er_t *erp ) { #if DK4_HAVE_GETADDRINFO /* +++++ +getaddrinfo */ char pnb[16*sizeof(dk4_um_t)]; dk4_socket_set_t *back = NULL; int res; if (0 < portno) { res = dk4ma_write_c8_decimal_unsigned( pnb, sizeof(pnb), (dk4_um_t)portno, 0, erp ); if (0 != res) { back = dk4socket_c8_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 } dk4_socket_set_t * dk4socket_c8_set_service( const char *servicename, int backlog, int local, dk4_er_t *erp ) { return ( dk4socket_c8_set_service_num_flag(servicename, backlog, local, 0, erp) ); }