From e0c6872cf40896c7be36b11dcc744620f10adf1d Mon Sep 17 00:00:00 2001 From: Norbert Preining Date: Mon, 2 Sep 2019 13:46:59 +0900 Subject: Initial commit --- support/dktools/dk4sock24.ctr | 232 ++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 232 insertions(+) create mode 100644 support/dktools/dk4sock24.ctr (limited to 'support/dktools/dk4sock24.ctr') diff --git a/support/dktools/dk4sock24.ctr b/support/dktools/dk4sock24.ctr new file mode 100644 index 0000000000..cce68e3b8d --- /dev/null +++ b/support/dktools/dk4sock24.ctr @@ -0,0 +1,232 @@ +%% 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) + ); +} + + + -- cgit v1.2.3