[thirdparty/systemd.git] / src / test / test-nss-hosts.c

/* SPDX-License-Identifier: LGPL-2.1-or-later */

#include <net/if.h>
#include <stdlib.h>
#include <unistd.h>

#include "af-list.h"
#include "alloc-util.h"
#include "dlfcn-util.h"
#include "errno-list.h"
#include "format-util.h"
#include "hexdecoct.h"
#include "hostname-util.h"
#include "in-addr-util.h"
#include "local-addresses.h"
#include "log.h"
#include "main-func.h"
#include "nss-test-util.h"
#include "nss-util.h"
#include "parse-util.h"
#include "path-util.h"
#include "stdio-util.h"
#include "string-util.h"
#include "strv.h"
#include "tests.h"

static size_t arg_bufsize = 1024;

static const char* af_to_string(int family, char *buf, size_t buf_len) {
        const char *name;

        if (family == AF_UNSPEC)
                return "*";

        name = af_to_name(family);
        if (name)
                return name;

        snprintf(buf, buf_len, "%i", family);
        return buf;
}

static int print_gaih_addrtuples(const struct gaih_addrtuple *tuples) {
        int n = 0;

        for (const struct gaih_addrtuple *it = tuples; it; it = it->next) {
                _cleanup_free_ char *a = NULL;
                union in_addr_union u;
                int r;
                char family_name[DECIMAL_STR_MAX(int)];
                char ifname[IF_NAMESIZE + 1];

                memcpy(&u, it->addr, 16);
                r = in_addr_to_string(it->family, &u, &a);
                assert_se(IN_SET(r, 0, -EAFNOSUPPORT));
                if (r == -EAFNOSUPPORT)
                        assert_se(a = hexmem(it->addr, 16));

                if (it->scopeid == 0)
                        goto numerical_index;

                if (!format_ifname(it->scopeid, ifname)) {
                        log_warning_errno(errno, "if_indextoname(%d) failed: %m", it->scopeid);
                numerical_index:
                        xsprintf(ifname, "%i", it->scopeid);
                };

                log_info("        \"%s\" %s %s %%%s",
                         it->name,
                         af_to_string(it->family, family_name, sizeof family_name),
                         a,
                         ifname);
                n ++;
        }
        return n;
}

static void print_struct_hostent(struct hostent *host, const char *canon) {
        char **s;

        log_info("        \"%s\"", host->h_name);
        STRV_FOREACH(s, host->h_aliases)
                log_info("        alias \"%s\"", *s);
        STRV_FOREACH(s, host->h_addr_list) {
                union in_addr_union u;
                _cleanup_free_ char *a = NULL;
                char family_name[DECIMAL_STR_MAX(int)];
                int r;

                assert_se((unsigned) host->h_length == FAMILY_ADDRESS_SIZE(host->h_addrtype));
                memcpy(&u, *s, host->h_length);
                r = in_addr_to_string(host->h_addrtype, &u, &a);
                assert_se(r == 0);
                log_info("        %s %s",
                         af_to_string(host->h_addrtype, family_name, sizeof family_name),
                         a);
        }
        if (canon)
                log_info("        canonical: \"%s\"", canon);
}

static void test_gethostbyname4_r(void *handle, const char *module, const char *name) {
        const char *fname;
        _nss_gethostbyname4_r_t f;
        char buffer[arg_bufsize];
        struct gaih_addrtuple *pat = NULL;
        int errno1 = 999, errno2 = 999; /* nss-dns doesn't set those */
        int32_t ttl = INT32_MAX; /* nss-dns wants to return the lowest ttl,
                                    and will access this variable through *ttlp,
                                    so we need to set it to something.
                                    I'm not sure if this is a bug in nss-dns
                                    or not. */
        enum nss_status status;
        char pretty_status[DECIMAL_STR_MAX(enum nss_status)];
        int n;

        fname = strjoina("_nss_", module, "_gethostbyname4_r");
        f = dlsym(handle, fname);
        log_debug("dlsym(0x%p, %s) → 0x%p", handle, fname, f);
        if (!f) {
                log_info("%s not defined", fname);
                return;
        }

        status = f(name, &pat, buffer, sizeof buffer, &errno1, &errno2, &ttl);
        if (status == NSS_STATUS_SUCCESS) {
                log_info("%s(\"%s\") → status=%s%-20spat=buffer+0x%tx errno=%d/%s h_errno=%d/%s ttl=%"PRIi32,
                         fname, name,
                         nss_status_to_string(status, pretty_status, sizeof pretty_status), "\n",
                         pat ? (char*) pat - buffer : 0,
                         errno1, errno_to_name(errno1) ?: "---",
                         errno2, hstrerror(errno2),
                         ttl);
                n = print_gaih_addrtuples(pat);
        } else {
                log_info("%s(\"%s\") → status=%s%-20spat=0x%p errno=%d/%s h_errno=%d/%s",
                         fname, name,
                         nss_status_to_string(status, pretty_status, sizeof pretty_status), "\n",
                         pat,
                         errno1, errno_to_name(errno1) ?: "---",
                         errno2, hstrerror(errno2));
                n = 0;
        }

        if (STR_IN_SET(module, "resolve", "mymachines") && status == NSS_STATUS_UNAVAIL)
                return;

        if (STR_IN_SET(module, "myhostname", "resolve") && streq(name, "localhost")) {
                assert_se(status == NSS_STATUS_SUCCESS);
                assert_se(n == 2);
        }
}

static void test_gethostbyname3_r(void *handle, const char *module, const char *name, int af) {
        const char *fname;
        _nss_gethostbyname3_r_t f;
        char buffer[arg_bufsize];
        int errno1 = 999, errno2 = 999; /* nss-dns doesn't set those */
        int32_t ttl = INT32_MAX; /* nss-dns wants to return the lowest ttl,
                                    and will access this variable through *ttlp,
                                    so we need to set it to something.
                                    I'm not sure if this is a bug in nss-dns
                                    or not. */
        enum nss_status status;
        char pretty_status[DECIMAL_STR_MAX(enum nss_status)];
        struct hostent host;
        char *canon;
        char family_name[DECIMAL_STR_MAX(int)];

        fname = strjoina("_nss_", module, "_gethostbyname3_r");
        f = dlsym(handle, fname);
        log_debug("dlsym(0x%p, %s) → 0x%p", handle, fname, f);
        if (!f) {
                log_info("%s not defined", fname);
                return;
        }

        status = f(name, af, &host, buffer, sizeof buffer, &errno1, &errno2, &ttl, &canon);
        log_info("%s(\"%s\", %s) → status=%s%-20serrno=%d/%s h_errno=%d/%s ttl=%"PRIi32,
                 fname, name, af_to_string(af, family_name, sizeof family_name),
                 nss_status_to_string(status, pretty_status, sizeof pretty_status), "\n",
                 errno1, errno_to_name(errno1) ?: "---",
                 errno2, hstrerror(errno2),
                 ttl);
        if (status == NSS_STATUS_SUCCESS)
                print_struct_hostent(&host, canon);
}

static void test_gethostbyname2_r(void *handle, const char *module, const char *name, int af) {
        const char *fname;
        _nss_gethostbyname2_r_t f;
        char buffer[arg_bufsize];
        int errno1 = 999, errno2 = 999; /* nss-dns doesn't set those */
        enum nss_status status;
        char pretty_status[DECIMAL_STR_MAX(enum nss_status)];
        struct hostent host;
        char family_name[DECIMAL_STR_MAX(int)];

        fname = strjoina("_nss_", module, "_gethostbyname2_r");
        f = dlsym(handle, fname);
        log_debug("dlsym(0x%p, %s) → 0x%p", handle, fname, f);
        if (!f) {
                log_info("%s not defined", fname);
                return;
        }

        status = f(name, af, &host, buffer, sizeof buffer, &errno1, &errno2);
        log_info("%s(\"%s\", %s) → status=%s%-20serrno=%d/%s h_errno=%d/%s",
                 fname, name, af_to_string(af, family_name, sizeof family_name),
                 nss_status_to_string(status, pretty_status, sizeof pretty_status), "\n",
                 errno1, errno_to_name(errno1) ?: "---",
                 errno2, hstrerror(errno2));
        if (status == NSS_STATUS_SUCCESS)
                print_struct_hostent(&host, NULL);
}

static void test_gethostbyname_r(void *handle, const char *module, const char *name) {
        const char *fname;
        _nss_gethostbyname_r_t f;
        char buffer[arg_bufsize];
        int errno1 = 999, errno2 = 999; /* nss-dns doesn't set those */
        enum nss_status status;
        char pretty_status[DECIMAL_STR_MAX(enum nss_status)];
        struct hostent host;

        fname = strjoina("_nss_", module, "_gethostbyname_r");
        f = dlsym(handle, fname);
        log_debug("dlsym(0x%p, %s) → 0x%p", handle, fname, f);
        if (!f) {
                log_info("%s not defined", fname);
                return;
        }

        status = f(name, &host, buffer, sizeof buffer, &errno1, &errno2);
        log_info("%s(\"%s\") → status=%s%-20serrno=%d/%s h_errno=%d/%s",
                 fname, name,
                 nss_status_to_string(status, pretty_status, sizeof pretty_status), "\n",
                 errno1, errno_to_name(errno1) ?: "---",
                 errno2, hstrerror(errno2));
        if (status == NSS_STATUS_SUCCESS)
                print_struct_hostent(&host, NULL);
}

static void test_gethostbyaddr2_r(void *handle,
                                  const char *module,
                                  const void* addr, socklen_t len,
                                  int af) {

        const char *fname;
        _nss_gethostbyaddr2_r_t f;
        char buffer[arg_bufsize];
        int errno1 = 999, errno2 = 999; /* nss-dns doesn't set those */
        enum nss_status status;
        char pretty_status[DECIMAL_STR_MAX(enum nss_status)];
        struct hostent host;
        int32_t ttl = INT32_MAX;
        _cleanup_free_ char *addr_pretty = NULL;

        fname = strjoina("_nss_", module, "_gethostbyaddr2_r");
        f = dlsym(handle, fname);

        log_full_errno(f ? LOG_DEBUG : LOG_INFO,  errno,
                       "dlsym(0x%p, %s) → 0x%p: %m", handle, fname, f);
        if (!f) {
                log_info("%s not defined", fname);
                return;
        }

        assert_se(in_addr_to_string(af, addr, &addr_pretty) >= 0);

        status = f(addr, len, af, &host, buffer, sizeof buffer, &errno1, &errno2, &ttl);
        log_info("%s(\"%s\") → status=%s%-20serrno=%d/%s h_errno=%d/%s ttl=%"PRIi32,
                 fname, addr_pretty,
                 nss_status_to_string(status, pretty_status, sizeof pretty_status), "\n",
                 errno1, errno_to_name(errno1) ?: "---",
                 errno2, hstrerror(errno2),
                 ttl);
        if (status == NSS_STATUS_SUCCESS)
                print_struct_hostent(&host, NULL);
}

static void test_gethostbyaddr_r(void *handle,
                                 const char *module,
                                 const void* addr, socklen_t len,
                                 int af) {

        const char *fname;
        _nss_gethostbyaddr_r_t f;
        char buffer[arg_bufsize];
        int errno1 = 999, errno2 = 999; /* nss-dns doesn't set those */
        enum nss_status status;
        char pretty_status[DECIMAL_STR_MAX(enum nss_status)];
        struct hostent host;
        _cleanup_free_ char *addr_pretty = NULL;

        fname = strjoina("_nss_", module, "_gethostbyaddr_r");
        f = dlsym(handle, fname);

        log_full_errno(f ? LOG_DEBUG : LOG_INFO,  errno,
                       "dlsym(0x%p, %s) → 0x%p: %m", handle, fname, f);
        if (!f) {
                log_info("%s not defined", fname);
                return;
        }

        assert_se(in_addr_to_string(af, addr, &addr_pretty) >= 0);

        status = f(addr, len, af, &host, buffer, sizeof buffer, &errno1, &errno2);
        log_info("%s(\"%s\") → status=%s%-20serrno=%d/%s h_errno=%d/%s",
                 fname, addr_pretty,
                 nss_status_to_string(status, pretty_status, sizeof pretty_status), "\n",
                 errno1, errno_to_name(errno1) ?: "---",
                 errno2, hstrerror(errno2));
        if (status == NSS_STATUS_SUCCESS)
                print_struct_hostent(&host, NULL);
}

static void test_byname(void *handle, const char *module, const char *name) {
        test_gethostbyname4_r(handle, module, name);
        puts("");

        test_gethostbyname3_r(handle, module, name, AF_INET);
        puts("");
        test_gethostbyname3_r(handle, module, name, AF_INET6);
        puts("");
        test_gethostbyname3_r(handle, module, name, AF_UNSPEC);
        puts("");
        test_gethostbyname3_r(handle, module, name, AF_LOCAL);
        puts("");

        test_gethostbyname2_r(handle, module, name, AF_INET);
        puts("");
        test_gethostbyname2_r(handle, module, name, AF_INET6);
        puts("");
        test_gethostbyname2_r(handle, module, name, AF_UNSPEC);
        puts("");
        test_gethostbyname2_r(handle, module, name, AF_LOCAL);
        puts("");

        test_gethostbyname_r(handle, module, name);
        puts("");
}

static void test_byaddr(void *handle,
                        const char *module,
                        const void* addr, socklen_t len,
                        int af) {
        test_gethostbyaddr2_r(handle, module, addr, len, af);
        puts("");

        test_gethostbyaddr_r(handle, module, addr, len, af);
        puts("");
}

static int make_addresses(struct local_address **addresses) {
        int n;
        _cleanup_free_ struct local_address *addrs = NULL;

        n = local_addresses(NULL, 0, AF_UNSPEC, &addrs);
        if (n < 0)
                log_info_errno(n, "Failed to query local addresses: %m");

        assert_se(GREEDY_REALLOC(addrs, n + 3));

        addrs[n++] = (struct local_address) { .family = AF_INET,
                                              .address.in = { htobe32(0x7F000001) } };
        addrs[n++] = (struct local_address) { .family = AF_INET,
                                              .address.in = { htobe32(0x7F000002) } };
        addrs[n++] = (struct local_address) { .family = AF_INET6,
                                              .address.in6 = in6addr_loopback };
        return 0;
}

static int test_one_module(const char *dir,
                           const char *module,
                           char **names,
                           struct local_address *addresses,
                           int n_addresses) {

        log_info("======== %s ========", module);

        _cleanup_(dlclosep) void *handle = nss_open_handle(dir, module, RTLD_LAZY|RTLD_NODELETE);
        if (!handle)
                return -EINVAL;

        char **name;
        STRV_FOREACH(name, names)
                test_byname(handle, module, *name);

        for (int i = 0; i < n_addresses; i++)
                test_byaddr(handle, module,
                            &addresses[i].address,
                            FAMILY_ADDRESS_SIZE(addresses[i].family),
                            addresses[i].family);

        log_info(" ");
        return 0;
}

static int parse_argv(int argc, char **argv,
                      char ***the_modules,
                      char ***the_names,
                      struct local_address **the_addresses, int *n_addresses) {

        _cleanup_strv_free_ char **modules = NULL, **names = NULL;
        _cleanup_free_ struct local_address *addrs = NULL;
        const char *p;
        int r, n = 0;

        p = getenv("SYSTEMD_TEST_NSS_BUFSIZE");
        if (p) {
                r = safe_atozu(p, &arg_bufsize);
                if (r < 0)
                        return log_error_errno(r, "Failed to parse $SYSTEMD_TEST_NSS_BUFSIZE");
        }

        if (argc > 1)
                modules = strv_new(argv[1]);
        else
                modules = strv_new(
#if ENABLE_NSS_MYHOSTNAME
                                "myhostname",
#endif
#if ENABLE_NSS_RESOLVE
                                "resolve",
#endif
#if ENABLE_NSS_MYMACHINES
                                "mymachines",
#endif
                                "dns");
        assert_se(modules);

        if (argc > 2) {
                char **name;
                int family;
                union in_addr_union address;

                STRV_FOREACH(name, argv + 2) {
                        r = in_addr_from_string_auto(*name, &family, &address);
                        if (r < 0) {
                                /* assume this is a name */
                                r = strv_extend(&names, *name);
                                if (r < 0)
                                        return r;
                        } else {
                                assert_se(GREEDY_REALLOC0(addrs, n + 1));

                                addrs[n++] = (struct local_address) { .family = family,
                                                                      .address = address };
                        }
                }
        } else {
                _cleanup_free_ char *hostname;
                assert_se(hostname = gethostname_malloc());
                assert_se(names = strv_new("localhost", "_gateway", "_outbound", "foo_no_such_host", hostname));

                n = make_addresses(&addrs);
                assert_se(n >= 0);
        }

        *the_modules = TAKE_PTR(modules);
        *the_names = TAKE_PTR(names);
        *the_addresses = TAKE_PTR(addrs);
        *n_addresses = n;
        return 0;
}

static int run(int argc, char **argv) {
        _cleanup_free_ char *dir = NULL;
        _cleanup_strv_free_ char **modules = NULL, **names = NULL;
        _cleanup_free_ struct local_address *addresses = NULL;
        int n_addresses = 0;
        char **module;
        int r;

        test_setup_logging(LOG_INFO);

        r = parse_argv(argc, argv, &modules, &names, &addresses, &n_addresses);
        if (r < 0)
                return log_error_errno(r, "Failed to parse arguments: %m");

        assert_se(path_extract_directory(argv[0], &dir) >= 0);

        STRV_FOREACH(module, modules) {
                r = test_one_module(dir, *module, names, addresses, n_addresses);
                if (r < 0)
                        return r;
        }

        return 0;
}

DEFINE_MAIN_FUNCTION(run);
Commit	Line	Data
db9ecf05	1	/* SPDX-License-Identifier: LGPL-2.1-or-later */
2abb5b3b	2
2abb5b3b	3	#include <net/if.h>
eb43e8a7	4	#include <stdlib.h>
ca78ad1d	5	#include <unistd.h>
2abb5b3b	6
eb43e8a7 YW	7	#include "af-list.h"
eb43e8a7 YW	8	#include "alloc-util.h"
e2aa384b	9	#include "dlfcn-util.h"
eb43e8a7	10	#include "errno-list.h"
518a66ec	11	#include "format-util.h"
eb43e8a7 YW	12	#include "hexdecoct.h"
	13	#include "hostname-util.h"
	14	#include "in-addr-util.h"
	15	#include "local-addresses.h"
2abb5b3b	16	#include "log.h"
f68a2622	17	#include "main-func.h"
f0d12668	18	#include "nss-test-util.h"
2abb5b3b	19	#include "nss-util.h"
f0cb09bb	20	#include "parse-util.h"
2abb5b3b	21	#include "path-util.h"
2abb5b3b	22	#include "stdio-util.h"
eb43e8a7	23	#include "string-util.h"
2abb5b3b	24	#include "strv.h"
f68a2622	25	#include "tests.h"
2abb5b3b	26
f0cb09bb ZJS	27	static size_t arg_bufsize = 1024;
f0cb09bb ZJS	28
2abb5b3b ZJS	29	static const char* af_to_string(int family, char *buf, size_t buf_len) {
	30	const char *name;
	31
	32	if (family == AF_UNSPEC)
	33	return "*";
	34
	35	name = af_to_name(family);
	36	if (name)
	37	return name;
	38
	39	snprintf(buf, buf_len, "%i", family);
	40	return buf;
	41	}
	42
2abb5b3b	43	static int print_gaih_addrtuples(const struct gaih_addrtuple *tuples) {
2abb5b3b ZJS	44	int n = 0;
2abb5b3b ZJS	45
890e5a4d	46	for (const struct gaih_addrtuple *it = tuples; it; it = it->next) {
2abb5b3b ZJS	47	_cleanup_free_ char *a = NULL;
	48	union in_addr_union u;
	49	int r;
	50	char family_name[DECIMAL_STR_MAX(int)];
518a66ec	51	char ifname[IF_NAMESIZE + 1];
2abb5b3b ZJS	52
	53	memcpy(&u, it->addr, 16);
	54	r = in_addr_to_string(it->family, &u, &a);
4c701096	55	assert_se(IN_SET(r, 0, -EAFNOSUPPORT));
2abb5b3b	56	if (r == -EAFNOSUPPORT)
dcd6361e	57	assert_se(a = hexmem(it->addr, 16));
2abb5b3b ZJS	58
	59	if (it->scopeid == 0)
	60	goto numerical_index;
	61
518a66ec	62	if (!format_ifname(it->scopeid, ifname)) {
25f027c5	63	log_warning_errno(errno, "if_indextoname(%d) failed: %m", it->scopeid);
2abb5b3b ZJS	64	numerical_index:
	65	xsprintf(ifname, "%i", it->scopeid);
	66	};
	67
	68	log_info(" \"%s\" %s %s %%%s",
	69	it->name,
	70	af_to_string(it->family, family_name, sizeof family_name),
	71	a,
	72	ifname);
	73	n ++;
	74	}
	75	return n;
	76	}
	77
	78	static void print_struct_hostent(struct hostent host, const char canon) {
	79	char **s;
	80
	81	log_info(" \"%s\"", host->h_name);
	82	STRV_FOREACH(s, host->h_aliases)
	83	log_info(" alias \"%s\"", *s);
	84	STRV_FOREACH(s, host->h_addr_list) {
	85	union in_addr_union u;
	86	_cleanup_free_ char *a = NULL;
	87	char family_name[DECIMAL_STR_MAX(int)];
	88	int r;
	89
	90	assert_se((unsigned) host->h_length == FAMILY_ADDRESS_SIZE(host->h_addrtype));
	91	memcpy(&u, *s, host->h_length);
	92	r = in_addr_to_string(host->h_addrtype, &u, &a);
	93	assert_se(r == 0);
	94	log_info(" %s %s",
	95	af_to_string(host->h_addrtype, family_name, sizeof family_name),
	96	a);
	97	}
	98	if (canon)
	99	log_info(" canonical: \"%s\"", canon);
	100	}
	101
	102	static void test_gethostbyname4_r(void handle, const char module, const char *name) {
	103	const char *fname;
	104	_nss_gethostbyname4_r_t f;
f0cb09bb	105	char buffer[arg_bufsize];
2abb5b3b ZJS	106	struct gaih_addrtuple *pat = NULL;
	107	int errno1 = 999, errno2 = 999; /* nss-dns doesn't set those */
	108	int32_t ttl = INT32_MAX; /* nss-dns wants to return the lowest ttl,
	109	and will access this variable through *ttlp,
	110	so we need to set it to something.
	111	I'm not sure if this is a bug in nss-dns
	112	or not. */
	113	enum nss_status status;
	114	char pretty_status[DECIMAL_STR_MAX(enum nss_status)];
	115	int n;
	116
	117	fname = strjoina("_nss_", module, "_gethostbyname4_r");
	118	f = dlsym(handle, fname);
	119	log_debug("dlsym(0x%p, %s) → 0x%p", handle, fname, f);
08554d47 ZJS	120	if (!f) {
	121	log_info("%s not defined", fname);
	122	return;
	123	}
2abb5b3b ZJS	124
	125	status = f(name, &pat, buffer, sizeof buffer, &errno1, &errno2, &ttl);
	126	if (status == NSS_STATUS_SUCCESS) {
	127	log_info("%s(\"%s\") → status=%s%-20spat=buffer+0x%tx errno=%d/%s h_errno=%d/%s ttl=%"PRIi32,
	128	fname, name,
	129	nss_status_to_string(status, pretty_status, sizeof pretty_status), "\n",
	130	pat ? (char*) pat - buffer : 0,
	131	errno1, errno_to_name(errno1) ?: "---",
	132	errno2, hstrerror(errno2),
	133	ttl);
	134	n = print_gaih_addrtuples(pat);
	135	} else {
	136	log_info("%s(\"%s\") → status=%s%-20spat=0x%p errno=%d/%s h_errno=%d/%s",
	137	fname, name,
	138	nss_status_to_string(status, pretty_status, sizeof pretty_status), "\n",
	139	pat,
	140	errno1, errno_to_name(errno1) ?: "---",
	141	errno2, hstrerror(errno2));
	142	n = 0;
	143	}
	144
	145	if (STR_IN_SET(module, "resolve", "mymachines") && status == NSS_STATUS_UNAVAIL)
	146	return;
	147
	148	if (STR_IN_SET(module, "myhostname", "resolve") && streq(name, "localhost")) {
	149	assert_se(status == NSS_STATUS_SUCCESS);
	150	assert_se(n == 2);
	151	}
	152	}
	153
2abb5b3b ZJS	154	static void test_gethostbyname3_r(void handle, const char module, const char *name, int af) {
	155	const char *fname;
	156	_nss_gethostbyname3_r_t f;
f0cb09bb	157	char buffer[arg_bufsize];
2abb5b3b ZJS	158	int errno1 = 999, errno2 = 999; /* nss-dns doesn't set those */
	159	int32_t ttl = INT32_MAX; /* nss-dns wants to return the lowest ttl,
	160	and will access this variable through *ttlp,
	161	so we need to set it to something.
	162	I'm not sure if this is a bug in nss-dns
	163	or not. */
	164	enum nss_status status;
	165	char pretty_status[DECIMAL_STR_MAX(enum nss_status)];
	166	struct hostent host;
	167	char *canon;
	168	char family_name[DECIMAL_STR_MAX(int)];
	169
	170	fname = strjoina("_nss_", module, "_gethostbyname3_r");
	171	f = dlsym(handle, fname);
	172	log_debug("dlsym(0x%p, %s) → 0x%p", handle, fname, f);
08554d47 ZJS	173	if (!f) {
	174	log_info("%s not defined", fname);
	175	return;
	176	}
2abb5b3b ZJS	177
	178	status = f(name, af, &host, buffer, sizeof buffer, &errno1, &errno2, &ttl, &canon);
	179	log_info("%s(\"%s\", %s) → status=%s%-20serrno=%d/%s h_errno=%d/%s ttl=%"PRIi32,
	180	fname, name, af_to_string(af, family_name, sizeof family_name),
	181	nss_status_to_string(status, pretty_status, sizeof pretty_status), "\n",
	182	errno1, errno_to_name(errno1) ?: "---",
	183	errno2, hstrerror(errno2),
	184	ttl);
	185	if (status == NSS_STATUS_SUCCESS)
	186	print_struct_hostent(&host, canon);
	187	}
	188
	189	static void test_gethostbyname2_r(void handle, const char module, const char *name, int af) {
	190	const char *fname;
	191	_nss_gethostbyname2_r_t f;
f0cb09bb	192	char buffer[arg_bufsize];
2abb5b3b ZJS	193	int errno1 = 999, errno2 = 999; /* nss-dns doesn't set those */
	194	enum nss_status status;
	195	char pretty_status[DECIMAL_STR_MAX(enum nss_status)];
	196	struct hostent host;
	197	char family_name[DECIMAL_STR_MAX(int)];
	198
	199	fname = strjoina("_nss_", module, "_gethostbyname2_r");
	200	f = dlsym(handle, fname);
	201	log_debug("dlsym(0x%p, %s) → 0x%p", handle, fname, f);
08554d47 ZJS	202	if (!f) {
	203	log_info("%s not defined", fname);
	204	return;
	205	}
2abb5b3b ZJS	206
	207	status = f(name, af, &host, buffer, sizeof buffer, &errno1, &errno2);
	208	log_info("%s(\"%s\", %s) → status=%s%-20serrno=%d/%s h_errno=%d/%s",
	209	fname, name, af_to_string(af, family_name, sizeof family_name),
	210	nss_status_to_string(status, pretty_status, sizeof pretty_status), "\n",
	211	errno1, errno_to_name(errno1) ?: "---",
	212	errno2, hstrerror(errno2));
	213	if (status == NSS_STATUS_SUCCESS)
	214	print_struct_hostent(&host, NULL);
	215	}
	216
	217	static void test_gethostbyname_r(void handle, const char module, const char *name) {
	218	const char *fname;
	219	_nss_gethostbyname_r_t f;
f0cb09bb	220	char buffer[arg_bufsize];
2abb5b3b ZJS	221	int errno1 = 999, errno2 = 999; /* nss-dns doesn't set those */
	222	enum nss_status status;
	223	char pretty_status[DECIMAL_STR_MAX(enum nss_status)];
	224	struct hostent host;
	225
	226	fname = strjoina("_nss_", module, "_gethostbyname_r");
	227	f = dlsym(handle, fname);
	228	log_debug("dlsym(0x%p, %s) → 0x%p", handle, fname, f);
08554d47 ZJS	229	if (!f) {
	230	log_info("%s not defined", fname);
	231	return;
	232	}
2abb5b3b ZJS	233
	234	status = f(name, &host, buffer, sizeof buffer, &errno1, &errno2);
	235	log_info("%s(\"%s\") → status=%s%-20serrno=%d/%s h_errno=%d/%s",
	236	fname, name,
	237	nss_status_to_string(status, pretty_status, sizeof pretty_status), "\n",
	238	errno1, errno_to_name(errno1) ?: "---",
	239	errno2, hstrerror(errno2));
	240	if (status == NSS_STATUS_SUCCESS)
	241	print_struct_hostent(&host, NULL);
	242	}
	243
	244	static void test_gethostbyaddr2_r(void *handle,
	245	const char *module,
	246	const void* addr, socklen_t len,
	247	int af) {
	248
	249	const char *fname;
	250	_nss_gethostbyaddr2_r_t f;
f0cb09bb	251	char buffer[arg_bufsize];
2abb5b3b ZJS	252	int errno1 = 999, errno2 = 999; /* nss-dns doesn't set those */
	253	enum nss_status status;
	254	char pretty_status[DECIMAL_STR_MAX(enum nss_status)];
	255	struct hostent host;
	256	int32_t ttl = INT32_MAX;
	257	_cleanup_free_ char *addr_pretty = NULL;
	258
	259	fname = strjoina("_nss_", module, "_gethostbyaddr2_r");
	260	f = dlsym(handle, fname);
	261
	262	log_full_errno(f ? LOG_DEBUG : LOG_INFO, errno,
	263	"dlsym(0x%p, %s) → 0x%p: %m", handle, fname, f);
08554d47 ZJS	264	if (!f) {
08554d47 ZJS	265	log_info("%s not defined", fname);
2abb5b3b	266	return;
08554d47	267	}
2abb5b3b ZJS	268
	269	assert_se(in_addr_to_string(af, addr, &addr_pretty) >= 0);
	270
	271	status = f(addr, len, af, &host, buffer, sizeof buffer, &errno1, &errno2, &ttl);
	272	log_info("%s(\"%s\") → status=%s%-20serrno=%d/%s h_errno=%d/%s ttl=%"PRIi32,
	273	fname, addr_pretty,
	274	nss_status_to_string(status, pretty_status, sizeof pretty_status), "\n",
	275	errno1, errno_to_name(errno1) ?: "---",
	276	errno2, hstrerror(errno2),
	277	ttl);
	278	if (status == NSS_STATUS_SUCCESS)
	279	print_struct_hostent(&host, NULL);
	280	}
	281
	282	static void test_gethostbyaddr_r(void *handle,
	283	const char *module,
	284	const void* addr, socklen_t len,
	285	int af) {
	286
	287	const char *fname;
	288	_nss_gethostbyaddr_r_t f;
f0cb09bb	289	char buffer[arg_bufsize];
2abb5b3b ZJS	290	int errno1 = 999, errno2 = 999; /* nss-dns doesn't set those */
	291	enum nss_status status;
	292	char pretty_status[DECIMAL_STR_MAX(enum nss_status)];
	293	struct hostent host;
	294	_cleanup_free_ char *addr_pretty = NULL;
	295
	296	fname = strjoina("_nss_", module, "_gethostbyaddr_r");
	297	f = dlsym(handle, fname);
	298
	299	log_full_errno(f ? LOG_DEBUG : LOG_INFO, errno,
	300	"dlsym(0x%p, %s) → 0x%p: %m", handle, fname, f);
08554d47 ZJS	301	if (!f) {
08554d47 ZJS	302	log_info("%s not defined", fname);
2abb5b3b	303	return;
08554d47	304	}
2abb5b3b ZJS	305
	306	assert_se(in_addr_to_string(af, addr, &addr_pretty) >= 0);
	307
	308	status = f(addr, len, af, &host, buffer, sizeof buffer, &errno1, &errno2);
	309	log_info("%s(\"%s\") → status=%s%-20serrno=%d/%s h_errno=%d/%s",
	310	fname, addr_pretty,
	311	nss_status_to_string(status, pretty_status, sizeof pretty_status), "\n",
	312	errno1, errno_to_name(errno1) ?: "---",
	313	errno2, hstrerror(errno2));
	314	if (status == NSS_STATUS_SUCCESS)
	315	print_struct_hostent(&host, NULL);
	316	}
	317
	318	static void test_byname(void handle, const char module, const char *name) {
	319	test_gethostbyname4_r(handle, module, name);
	320	puts("");
	321
	322	test_gethostbyname3_r(handle, module, name, AF_INET);
	323	puts("");
	324	test_gethostbyname3_r(handle, module, name, AF_INET6);
	325	puts("");
	326	test_gethostbyname3_r(handle, module, name, AF_UNSPEC);
	327	puts("");
	328	test_gethostbyname3_r(handle, module, name, AF_LOCAL);
	329	puts("");
	330
	331	test_gethostbyname2_r(handle, module, name, AF_INET);
	332	puts("");
	333	test_gethostbyname2_r(handle, module, name, AF_INET6);
	334	puts("");
	335	test_gethostbyname2_r(handle, module, name, AF_UNSPEC);
	336	puts("");
	337	test_gethostbyname2_r(handle, module, name, AF_LOCAL);
	338	puts("");
	339
	340	test_gethostbyname_r(handle, module, name);
	341	puts("");
	342	}
	343
	344	static void test_byaddr(void *handle,
	345	const char *module,
	346	const void* addr, socklen_t len,
	347	int af) {
	348	test_gethostbyaddr2_r(handle, module, addr, len, af);
	349	puts("");
	350
	351	test_gethostbyaddr_r(handle, module, addr, len, af);
	352	puts("");
	353	}
	354
9f7672b3 ZJS	355	static int make_addresses(struct local_address **addresses) {
9f7672b3 ZJS	356	int n;
9f7672b3 ZJS	357	_cleanup_free_ struct local_address *addrs = NULL;
	358
	359	n = local_addresses(NULL, 0, AF_UNSPEC, &addrs);
	360	if (n < 0)
	361	log_info_errno(n, "Failed to query local addresses: %m");
	362
319a4f4b	363	assert_se(GREEDY_REALLOC(addrs, n + 3));
9f7672b3 ZJS	364
	365	addrs[n++] = (struct local_address) { .family = AF_INET,
	366	.address.in = { htobe32(0x7F000001) } };
	367	addrs[n++] = (struct local_address) { .family = AF_INET,
	368	.address.in = { htobe32(0x7F000002) } };
	369	addrs[n++] = (struct local_address) { .family = AF_INET6,
	370	.address.in6 = in6addr_loopback };
	371	return 0;
	372	}
	373
890e5a4d	374	static int test_one_module(const char *dir,
9f7672b3 ZJS	375	const char *module,
	376	char **names,
	377	struct local_address *addresses,
	378	int n_addresses) {
9f7672b3	379
9f7672b3 ZJS	380	log_info("======== %s ========", module);
9f7672b3 ZJS	381
e2aa384b	382	_cleanup_(dlclosep) void *handle = nss_open_handle(dir, module, RTLD_LAZY\|RTLD_NODELETE);
9f7672b3 ZJS	383	if (!handle)
	384	return -EINVAL;
	385
e2aa384b	386	char **name;
9f7672b3 ZJS	387	STRV_FOREACH(name, names)
	388	test_byname(handle, module, *name);
	389
890e5a4d	390	for (int i = 0; i < n_addresses; i++)
9f7672b3 ZJS	391	test_byaddr(handle, module,
	392	&addresses[i].address,
	393	FAMILY_ADDRESS_SIZE(addresses[i].family),
	394	addresses[i].family);
	395
	396	log_info(" ");
9f7672b3 ZJS	397	return 0;
	398	}
	399
	400	static int parse_argv(int argc, char **argv,
	401	char ***the_modules,
	402	char ***the_names,
	403	struct local_address *the_addresses, int n_addresses) {
	404
9f7672b3 ZJS	405	_cleanup_strv_free_ char modules = NULL, names = NULL;
9f7672b3 ZJS	406	_cleanup_free_ struct local_address *addrs = NULL;
f0cb09bb	407	const char *p;
f0d12668	408	int r, n = 0;
9f7672b3	409
f0cb09bb ZJS	410	p = getenv("SYSTEMD_TEST_NSS_BUFSIZE");
	411	if (p) {
	412	r = safe_atozu(p, &arg_bufsize);
	413	if (r < 0)
	414	return log_error_errno(r, "Failed to parse $SYSTEMD_TEST_NSS_BUFSIZE");
	415	}
	416
9f7672b3	417	if (argc > 1)
bea1a013	418	modules = strv_new(argv[1]);
9f7672b3 ZJS	419	else
9f7672b3 ZJS	420	modules = strv_new(
08540a95	421	#if ENABLE_NSS_MYHOSTNAME
9f7672b3	422	"myhostname",
2abb5b3b	423	#endif
08540a95	424	#if ENABLE_NSS_RESOLVE
9f7672b3	425	"resolve",
2abb5b3b	426	#endif
08540a95	427	#if ENABLE_NSS_MYMACHINES
9f7672b3	428	"mymachines",
2abb5b3b	429	#endif
bea1a013	430	"dns");
e2aa384b	431	assert_se(modules);
9f7672b3 ZJS	432
	433	if (argc > 2) {
	434	char **name;
	435	int family;
	436	union in_addr_union address;
	437
	438	STRV_FOREACH(name, argv + 2) {
	439	r = in_addr_from_string_auto(*name, &family, &address);
	440	if (r < 0) {
	441	/* assume this is a name */
	442	r = strv_extend(&names, *name);
	443	if (r < 0)
	444	return r;
	445	} else {
319a4f4b	446	assert_se(GREEDY_REALLOC0(addrs, n + 1));
9f7672b3 ZJS	447
	448	addrs[n++] = (struct local_address) { .family = family,
	449	.address = address };
	450	}
	451	}
	452	} else {
	453	_cleanup_free_ char *hostname;
e2aa384b	454	assert_se(hostname = gethostname_malloc());
a1fdbcbe	455	assert_se(names = strv_new("localhost", "_gateway", "_outbound", "foo_no_such_host", hostname));
9f7672b3 ZJS	456
9f7672b3 ZJS	457	n = make_addresses(&addrs);
e2aa384b	458	assert_se(n >= 0);
9f7672b3 ZJS	459	}
9f7672b3 ZJS	460
f0d12668 ZJS	461	*the_modules = TAKE_PTR(modules);
	462	*the_names = TAKE_PTR(names);
	463	*the_addresses = TAKE_PTR(addrs);
9f7672b3	464	*n_addresses = n;
9f7672b3 ZJS	465	return 0;
	466	}
	467
f68a2622	468	static int run(int argc, char **argv) {
9f7672b3 ZJS	469	_cleanup_free_ char *dir = NULL;
9f7672b3 ZJS	470	_cleanup_strv_free_ char modules = NULL, names = NULL;
2abb5b3b	471	_cleanup_free_ struct local_address *addresses = NULL;
6a909d41	472	int n_addresses = 0;
9f7672b3 ZJS	473	char **module;
9f7672b3 ZJS	474	int r;
2abb5b3b	475
f68a2622	476	test_setup_logging(LOG_INFO);
2abb5b3b	477
9f7672b3	478	r = parse_argv(argc, argv, &modules, &names, &addresses, &n_addresses);
e2aa384b ZJS	479	if (r < 0)
e2aa384b ZJS	480	return log_error_errno(r, "Failed to parse arguments: %m");
2abb5b3b	481
e2aa384b	482	assert_se(path_extract_directory(argv[0], &dir) >= 0);
2abb5b3b	483
9f7672b3 ZJS	484	STRV_FOREACH(module, modules) {
	485	r = test_one_module(dir, *module, names, addresses, n_addresses);
	486	if (r < 0)
f68a2622	487	return r;
2abb5b3b ZJS	488	}
2abb5b3b ZJS	489
f68a2622	490	return 0;
2abb5b3b	491	}
f68a2622 ZJS	492
f68a2622 ZJS	493	DEFINE_MAIN_FUNCTION(run);