fabrics.c

// SPDX-License-Identifier: LGPL-2.1-or-later
/*
 * This file is part of libnvme.
 * Copyright (c) 2020 Western Digital Corporation or its affiliates.
 *
 * Authors: Keith Busch <keith.busch@wdc.com>
 * 	    Chaitanya Kulkarni <chaitanya.kulkarni@wdc.com>
 */

#include <dirent.h>
#include <errno.h>
#include <fcntl.h>
#include <fnmatch.h>
#include <inttypes.h>
#include <limits.h>
#include <stdbool.h>
#include <stddef.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>

#include <arpa/inet.h>
#include <net/if.h>
#include <netdb.h>
#include <sys/param.h>
#include <sys/stat.h>
#include <sys/types.h>

#include <ccan/array_size/array_size.h>
#include <ccan/endian/endian.h>
#include <ccan/list/list.h>
#include <ccan/str/str.h>

#include <libnvme.h>

#include "cleanup.h"
#include "cleanup-linux.h"
#include "private.h"
#include "private-fabrics.h"
#include "compiler-attributes.h"

const char *nvmf_dev = "/dev/nvme-fabrics";

static inline void free_uri(struct libnvmf_uri **uri)
{
	libnvmf_uri_free(*uri);
}
#define __cleanup_uri __cleanup(free_uri)

/**
 * strchomp() - Strip trailing spaces
 * @str: String to strip
 * @max: Maximum length of string
 */
static void strchomp(char *str, int max)
{
	int i;

	for (i = max - 1; i >= 0 && str[i] == ' '; i--) {
		str[i] = '\0';
	}
}

const char *arg_str(const char * const *strings,
		size_t array_size, size_t idx)
{
	if (idx < array_size && strings[idx])
		return strings[idx];
	return "unrecognized";
}

const char * const trtypes[] = {
	[NVMF_TRTYPE_RDMA]	= "rdma",
	[NVMF_TRTYPE_FC]	= "fc",
	[NVMF_TRTYPE_TCP]	= "tcp",
	[NVMF_TRTYPE_LOOP]	= "loop",
};

__public const char *libnvmf_trtype_str(__u8 trtype)
{
	return arg_str(trtypes, ARRAY_SIZE(trtypes), trtype);
}

static const char * const adrfams[] = {
	[NVMF_ADDR_FAMILY_PCI]	= "pci",
	[NVMF_ADDR_FAMILY_IP4]	= "ipv4",
	[NVMF_ADDR_FAMILY_IP6]	= "ipv6",
	[NVMF_ADDR_FAMILY_IB]	= "infiniband",
	[NVMF_ADDR_FAMILY_FC]	= "fibre-channel",
};

__public const char *libnvmf_adrfam_str(__u8 adrfam)
{
	return arg_str(adrfams, ARRAY_SIZE(adrfams), adrfam);
}

static const char * const subtypes[] = {
	[NVME_NQN_DISC]		= "discovery subsystem referral",
	[NVME_NQN_NVME]		= "nvme subsystem",
	[NVME_NQN_CURR]		= "current discovery subsystem",
};

__public const char *libnvmf_subtype_str(__u8 subtype)
{
	return arg_str(subtypes, ARRAY_SIZE(subtypes), subtype);
}

static const char * const treqs[] = {
	[NVMF_TREQ_NOT_SPECIFIED]	= "not specified",
	[NVMF_TREQ_REQUIRED]		= "required",
	[NVMF_TREQ_NOT_REQUIRED]	= "not required",
	[NVMF_TREQ_NOT_SPECIFIED |
	 NVMF_TREQ_DISABLE_SQFLOW]	= "not specified, "
				"sq flow control disable supported",
	[NVMF_TREQ_REQUIRED |
	 NVMF_TREQ_DISABLE_SQFLOW]	= "required, "
				"sq flow control disable supported",
	[NVMF_TREQ_NOT_REQUIRED |
	 NVMF_TREQ_DISABLE_SQFLOW]	= "not required, "
				"sq flow control disable supported",
};

__public const char *libnvmf_treq_str(__u8 treq)
{
	return arg_str(treqs, ARRAY_SIZE(treqs), treq);
}

static const char * const eflags_strings[] = {
	[NVMF_DISC_EFLAGS_NONE]		= "none",
	[NVMF_DISC_EFLAGS_EPCSD]	= "explicit discovery connections",
	[NVMF_DISC_EFLAGS_DUPRETINFO]	= "duplicate discovery information",
	[NVMF_DISC_EFLAGS_EPCSD |
	 NVMF_DISC_EFLAGS_DUPRETINFO]	= "explicit discovery connections, "
					  "duplicate discovery information",
	[NVMF_DISC_EFLAGS_NCC]		= "no cdc connectivity",
	[NVMF_DISC_EFLAGS_EPCSD |
	 NVMF_DISC_EFLAGS_NCC]		= "explicit discovery connections, "
					  "no cdc connectivity",
	[NVMF_DISC_EFLAGS_DUPRETINFO |
	 NVMF_DISC_EFLAGS_NCC]		= "duplicate discovery information, "
					  "no cdc connectivity",
	[NVMF_DISC_EFLAGS_EPCSD |
	 NVMF_DISC_EFLAGS_DUPRETINFO |
	 NVMF_DISC_EFLAGS_NCC]		= "explicit discovery connections, "
					  "duplicate discovery information, "
					  "no cdc connectivity",
};

__public const char *libnvmf_eflags_str(__u16 eflags)
{
	return arg_str(eflags_strings, ARRAY_SIZE(eflags_strings), eflags);
}

static const char * const sectypes[] = {
	[NVMF_TCP_SECTYPE_NONE]		= "none",
	[NVMF_TCP_SECTYPE_TLS]		= "tls",
	[NVMF_TCP_SECTYPE_TLS13]	= "tls13",
};

__public const char *libnvmf_sectype_str(__u8 sectype)
{
	return arg_str(sectypes, ARRAY_SIZE(sectypes), sectype);
}

static const char * const prtypes[] = {
	[NVMF_RDMA_PRTYPE_NOT_SPECIFIED]	= "not specified",
	[NVMF_RDMA_PRTYPE_IB]			= "infiniband",
	[NVMF_RDMA_PRTYPE_ROCE]			= "roce",
	[NVMF_RDMA_PRTYPE_ROCEV2]		= "roce-v2",
	[NVMF_RDMA_PRTYPE_IWARP]		= "iwarp",
};

__public const char *libnvmf_prtype_str(__u8 prtype)
{
	return arg_str(prtypes, ARRAY_SIZE(prtypes), prtype);
}

static const char * const qptypes[] = {
	[NVMF_RDMA_QPTYPE_CONNECTED]	= "connected",
	[NVMF_RDMA_QPTYPE_DATAGRAM]	= "datagram",
};

__public const char *libnvmf_qptype_str(__u8 qptype)
{
	return arg_str(qptypes, ARRAY_SIZE(qptypes), qptype);
}

static const char * const cms[] = {
	[NVMF_RDMA_CMS_RDMA_CM]	= "rdma-cm",
};

__public const char *libnvmf_cms_str(__u8 cm)
{
	return arg_str(cms, ARRAY_SIZE(cms), cm);
}

void libnvmf_default_config(struct libnvme_fabrics_config *cfg)
{
	cfg->tos = -1;
	cfg->ctrl_loss_tmo = NVMF_DEF_CTRL_LOSS_TMO;
}

__public int libnvmf_context_create(struct libnvme_global_ctx *ctx,
		bool (*decide_retry)(struct libnvmf_context *fctx, int err,
			void *user_data),
		void (*connected)(struct libnvmf_context *fctx,
			struct libnvme_ctrl *c, void *user_data),
		void (*already_connected)(struct libnvmf_context *fctx,
			struct libnvme_host *host, const char *subsysnqn,
			const char *transport, const char *traddr,
			const char *trsvcid, void *user_data),
		void *user_data, struct libnvmf_context **fctxp)
{
	struct libnvmf_context *fctx;

	fctx = calloc(1, sizeof(*fctx));
	if (!fctx)
		return -ENOMEM;

	fctx->ctx = ctx;

	libnvmf_default_config(&fctx->cfg);

	fctx->decide_retry = decide_retry;
	fctx->connected = connected;
	fctx->already_connected = already_connected;

	fctx->user_data = user_data;

	*fctxp = fctx;
	return 0;
}

__public void libnvmf_context_free(struct libnvmf_context *fctx)
{
	if (!fctx)
		return;

	free(fctx->tls_key);
	free(fctx);
}

__public int libnvmf_context_set_discovery_cbs(struct libnvmf_context *fctx,
		void (*discovery_log)(struct libnvmf_context *fctx,
			bool connect,
			struct nvmf_discovery_log *log,
			uint64_t numrec, void *user_data),
		int (*parser_init)(struct libnvmf_context *fctx,
			void *user_data),
		void (*parser_cleanup)(struct libnvmf_context *fctx,
			void *user_data),
		int (*parser_next_line)(struct libnvmf_context *fctx,
			void *user_data))
{
	fctx->discovery_log = discovery_log;
	fctx->parser_init = parser_init;
	fctx->parser_cleanup = parser_cleanup;
	fctx->parser_next_line = parser_next_line;

	return 0;
}

__public int libnvmf_context_set_discovery_defaults(struct libnvmf_context *fctx,
		int max_discovery_retries, int keep_alive_timeout)
{
	fctx->default_max_discovery_retries = max_discovery_retries;
	fctx->default_keep_alive_timeout = keep_alive_timeout;

	return 0;
}

__public int libnvmf_context_set_fabrics_config(struct libnvmf_context *fctx,
		struct libnvme_fabrics_config *cfg)
{
	fctx->cfg.queue_size = cfg->queue_size;
	fctx->cfg.nr_io_queues = cfg->nr_io_queues;
	fctx->cfg.reconnect_delay = cfg->reconnect_delay;
	fctx->cfg.ctrl_loss_tmo = cfg->ctrl_loss_tmo;
	fctx->cfg.fast_io_fail_tmo = cfg->fast_io_fail_tmo;
	fctx->cfg.keep_alive_tmo = cfg->keep_alive_tmo;
	fctx->cfg.nr_write_queues = cfg->nr_write_queues;
	fctx->cfg.nr_poll_queues = cfg->nr_poll_queues;
	fctx->cfg.tos = cfg->tos;
	fctx->cfg.keyring_id = cfg->keyring_id;
	fctx->cfg.tls_key_id = cfg->tls_key_id;
	fctx->cfg.tls_configured_key_id = cfg->tls_configured_key_id;
	fctx->cfg.duplicate_connect = cfg->duplicate_connect;
	fctx->cfg.disable_sqflow = cfg->disable_sqflow;
	fctx->cfg.hdr_digest = cfg->hdr_digest;
	fctx->cfg.data_digest = cfg->data_digest;
	fctx->cfg.tls = cfg->tls;
	fctx->cfg.concat = cfg->concat;

	return 0;
}

__public int libnvmf_context_set_connection(struct libnvmf_context *fctx,
		const char *subsysnqn, const char *transport,
		const char *traddr, const char *trsvcid,
		const char *host_traddr, const char *host_iface)
{
	fctx->subsysnqn = subsysnqn;
	fctx->transport = transport;
	fctx->traddr = traddr;
	fctx->trsvcid = trsvcid;
	fctx->host_traddr = host_traddr;
	fctx->host_iface = host_iface;

	return 0;
}

static const char *hostid_from_hostnqn(const char *hostnqn)
{
	const char *match;

	if (!hostnqn)
		return NULL;

	match = strstr(hostnqn, "uuid:");
	if (!match)
		return NULL;

	return match + strlen("uuid:");
}

__public int libnvmf_context_set_hostnqn(struct libnvmf_context *fctx,
		const char *hostnqn, const char *hostid)
{
	fctx->hostnqn = hostnqn;
	if (!hostid)
		hostid = hostid_from_hostnqn(hostnqn);
	fctx->hostid = hostid;

	return 0;
}

__public int libnvmf_context_set_crypto(struct libnvmf_context *fctx,
		const char *hostkey, const char *ctrlkey,
		const char *keyring, const char *tls_key,
		const char *tls_key_identity)
{
	int err;

	fctx->hostkey = hostkey;
	fctx->ctrlkey = ctrlkey;
	fctx->keyring = keyring;
	fctx->tls_key_identity = tls_key_identity;

	if (!tls_key)
		return 0;

	if (!strncmp(tls_key, "pin:", 4)) {
		__cleanup_free unsigned char *raw_secret = NULL;
		__cleanup_free char *encoded_key = NULL;
		int key_len = 32;

		err = libnvme_create_raw_secret(fctx->ctx, tls_key,
			key_len, &raw_secret);
		if (err)
			return err;

		err = libnvme_export_tls_key(fctx->ctx, raw_secret,
			key_len, &encoded_key);
		if (err)
			return err;

		fctx->tls_key = encoded_key;
		encoded_key = NULL;
		return 0;
	}

	fctx->tls_key = strdup(tls_key);
	return 0;
}

__public int libnvmf_context_set_persistent(struct libnvmf_context *fctx, bool persistent)
{
	fctx->persistent = persistent;

	return 0;
}

__public int libnvmf_context_set_device(struct libnvmf_context *fctx, const char *device)
{
	fctx->device = device;

	return 0;
}

__public struct libnvme_fabrics_config *libnvmf_context_get_fabrics_config(
		struct libnvmf_context *fctx)
{
	return &fctx->cfg;
}

__public struct libnvme_fabrics_config *libnvmf_ctrl_get_fabrics_config(
		libnvme_ctrl_t c)
{
	return &c->cfg;
}

/*
 * Derived from Linux's supported options (the opt_tokens table)
 * when the mechanism to report supported options was added (f18ee3d988157).
 * Not all of these options may actually be supported,
 * but we retain the old behavior of passing all that might be.
 */
static const struct libnvme_fabric_options default_supported_options = {
	.ctrl_loss_tmo = true,
	.data_digest = true,
	.disable_sqflow = true,
	.discovery = true,
	.duplicate_connect = true,
	.fast_io_fail_tmo = true,
	.hdr_digest = true,
	.host_iface = true,
	.host_traddr = true,
	.hostid = true,
	.hostnqn = true,
	.keep_alive_tmo = true,
	.nqn = true,
	.nr_io_queues = true,
	.nr_poll_queues = true,
	.nr_write_queues = true,
	.queue_size = true,
	.reconnect_delay = true,
	.tos = true,
	.traddr = true,
	.transport = true,
	.trsvcid = true,
};

#define MERGE_CFG_OPTION(c, n, o, d)			\
	if ((c)->o == d) (c)->o = (n)->o
static void merge_config(libnvme_ctrl_t c,
		const struct libnvme_fabrics_config *cfg)
{
	struct libnvme_fabrics_config *ctrl_cfg =
		libnvmf_ctrl_get_fabrics_config(c);

	MERGE_CFG_OPTION(ctrl_cfg, cfg, nr_io_queues, 0);
	MERGE_CFG_OPTION(ctrl_cfg, cfg, nr_write_queues, 0);
	MERGE_CFG_OPTION(ctrl_cfg, cfg, nr_poll_queues, 0);
	MERGE_CFG_OPTION(ctrl_cfg, cfg, queue_size, 0);
	MERGE_CFG_OPTION(ctrl_cfg, cfg, keep_alive_tmo, 0);
	MERGE_CFG_OPTION(ctrl_cfg, cfg, reconnect_delay, 0);
	MERGE_CFG_OPTION(ctrl_cfg, cfg, ctrl_loss_tmo,
			  NVMF_DEF_CTRL_LOSS_TMO);
	MERGE_CFG_OPTION(ctrl_cfg, cfg, fast_io_fail_tmo, 0);
	MERGE_CFG_OPTION(ctrl_cfg, cfg, tos, -1);
	MERGE_CFG_OPTION(ctrl_cfg, cfg, keyring_id, 0);
	MERGE_CFG_OPTION(ctrl_cfg, cfg, tls_key_id, 0);
	MERGE_CFG_OPTION(ctrl_cfg, cfg, tls_configured_key_id, 0);
	MERGE_CFG_OPTION(ctrl_cfg, cfg, duplicate_connect, false);
	MERGE_CFG_OPTION(ctrl_cfg, cfg, disable_sqflow, false);
	MERGE_CFG_OPTION(ctrl_cfg, cfg, hdr_digest, false);
	MERGE_CFG_OPTION(ctrl_cfg, cfg, data_digest, false);
	MERGE_CFG_OPTION(ctrl_cfg, cfg, tls, false);
	MERGE_CFG_OPTION(ctrl_cfg, cfg, concat, false);
}

#define UPDATE_CFG_OPTION(c, n, o, d)			\
	if ((n)->o != d) (c)->o = (n)->o
static void update_config(libnvme_ctrl_t c,
		const struct libnvme_fabrics_config *cfg)
{
	struct libnvme_fabrics_config *ctrl_cfg =
		libnvmf_ctrl_get_fabrics_config(c);

	UPDATE_CFG_OPTION(ctrl_cfg, cfg, nr_io_queues, 0);
	UPDATE_CFG_OPTION(ctrl_cfg, cfg, nr_write_queues, 0);
	UPDATE_CFG_OPTION(ctrl_cfg, cfg, nr_poll_queues, 0);
	UPDATE_CFG_OPTION(ctrl_cfg, cfg, queue_size, 0);
	UPDATE_CFG_OPTION(ctrl_cfg, cfg, keep_alive_tmo, 0);
	UPDATE_CFG_OPTION(ctrl_cfg, cfg, reconnect_delay, 0);
	UPDATE_CFG_OPTION(ctrl_cfg, cfg, ctrl_loss_tmo,
			  NVMF_DEF_CTRL_LOSS_TMO);
	UPDATE_CFG_OPTION(ctrl_cfg, cfg, fast_io_fail_tmo, 0);
	UPDATE_CFG_OPTION(ctrl_cfg, cfg, tos, -1);
	UPDATE_CFG_OPTION(ctrl_cfg, cfg, keyring_id , 0);
	UPDATE_CFG_OPTION(ctrl_cfg, cfg, tls_key_id, 0);
	UPDATE_CFG_OPTION(ctrl_cfg, cfg, tls_configured_key_id, 0);
	UPDATE_CFG_OPTION(ctrl_cfg, cfg, duplicate_connect, false);
	UPDATE_CFG_OPTION(ctrl_cfg, cfg, disable_sqflow, false);
	UPDATE_CFG_OPTION(ctrl_cfg, cfg, hdr_digest, false);
	UPDATE_CFG_OPTION(ctrl_cfg, cfg, data_digest, false);
	UPDATE_CFG_OPTION(ctrl_cfg, cfg, tls, false);
	UPDATE_CFG_OPTION(ctrl_cfg, cfg, concat, false);
}

static int __add_bool_argument(char **argstr, char *tok, bool arg)
{
	char *nstr;

	if (!arg)
		return 0;
	if (asprintf(&nstr, "%s,%s", *argstr, tok) < 0) {
		return -ENOMEM;
	}
	free(*argstr);
	*argstr = nstr;

	return 0;
}

static int __add_hex_argument(char **argstr, char *tok, int arg, bool allow_zero)
{
	char *nstr;

	if (arg < 0 || (!arg && !allow_zero))
		return 0;
	if (asprintf(&nstr, "%s,%s=0x%08x", *argstr, tok, arg) < 0) {
		return -ENOMEM;
	}
	free(*argstr);
	*argstr = nstr;

	return 0;
}

static int __add_int_argument(char **argstr, char *tok, int arg, bool allow_zero)
{
	char *nstr;

	if (arg < 0 || (!arg && !allow_zero))
		return 0;
	if (asprintf(&nstr, "%s,%s=%d", *argstr, tok, arg) < 0) {
		return -ENOMEM;
	}
	free(*argstr);
	*argstr = nstr;

	return 0;
}

static int __add_int_or_minus_one_argument(char **argstr, char *tok, int arg)
{
	char *nstr;

	if (arg < -1)
		return 0;
	if (asprintf(&nstr, "%s,%s=%d", *argstr, tok, arg) < 0) {
		return -ENOMEM;
	}
	free(*argstr);
	*argstr = nstr;

	return 0;
}

static int __add_argument(char **argstr, const char *tok, const char *arg)
{
	char *nstr;

	if (!arg || arg[0] == '\0' || !strcmp(arg, "none"))
		return 0;
	if (asprintf(&nstr, "%s,%s=%s", *argstr, tok, arg) < 0) {
		return -ENOMEM;
	}
	free(*argstr);
	*argstr = nstr;

	return 0;
}

static int __nvmf_supported_options(struct libnvme_global_ctx *ctx);
#define nvmf_check_option(ctx, tok)					\
({									\
	!__nvmf_supported_options(ctx) && ctx->options->tok;		\
})

#define add_bool_argument(ctx, argstr, tok, arg)			\
({									\
	int ret;							\
	if (nvmf_check_option(ctx, tok)) {				\
		ret = __add_bool_argument(argstr,			\
					  stringify(tok),		\
					  arg);				\
	} else {							\
		libnvme_msg(ctx, LIBNVME_LOG_DEBUG,				\
			 "option \"%s\" ignored\n",			\
			 stringify(tok));				\
		ret = 0;						\
	}								\
	ret;								\
})

#define add_hex_argument(ctx, argstr, tok, arg, allow_zero)		\
({									\
	int ret;							\
	if (nvmf_check_option(ctx, tok)) {				\
		ret = __add_hex_argument(argstr,			\
					stringify(tok),			\
					arg,				\
					allow_zero);			\
	} else {							\
		libnvme_msg(ctx, LIBNVME_LOG_DEBUG,				\
			 "option \"%s\" ignored\n",			\
			 stringify(tok));				\
		ret = 0;						\
	}								\
	ret;								\
})

#define add_int_argument(ctx, argstr, tok, arg, allow_zero)		\
({									\
	int ret;							\
	if (nvmf_check_option(ctx, tok)) {				\
		ret = __add_int_argument(argstr,			\
					stringify(tok),			\
					arg,				\
					allow_zero);			\
	} else {							\
		libnvme_msg(ctx, LIBNVME_LOG_DEBUG,				\
			 "option \"%s\" ignored\n",			\
			 stringify(tok));				\
		ret = 0;						\
	}								\
	ret;								\
})

#define add_int_or_minus_one_argument(ctx, argstr, tok, arg)		\
({									\
	int ret;							\
	if (nvmf_check_option(ctx, tok)) {				\
		ret = __add_int_or_minus_one_argument(argstr,		\
						     stringify(tok),	\
						     arg);		\
	} else {							\
		libnvme_msg(ctx, LIBNVME_LOG_DEBUG,				\
			 "option \"%s\" ignored\n",			\
			 stringify(tok));				\
		ret = 0;						\
	}								\
	ret;								\
})

#define add_argument(ctx, argstr, tok, arg)				\
({									\
	int ret;							\
	if (nvmf_check_option(ctx, tok)) {				\
		ret = __add_argument(argstr,				\
				     stringify(tok),			\
				     arg);				\
	} else {							\
		libnvme_msg(ctx, LIBNVME_LOG_WARN,				\
			 "option \"%s\" ignored\n",			\
			 stringify(tok));				\
		ret = 0;						\
	}								\
	ret;								\
})

static int inet4_pton(const char *src, uint16_t port,
		      struct sockaddr_storage *addr)
{
	struct sockaddr_in *addr4 = (struct sockaddr_in *)addr;

	if (strlen(src) > INET_ADDRSTRLEN)
		return -EINVAL;

	if (inet_pton(AF_INET, src, &addr4->sin_addr.s_addr) <= 0)
		return -EINVAL;

	addr4->sin_family = AF_INET;
	addr4->sin_port = htons(port);

	return 0;
}

static int inet6_pton(struct libnvme_global_ctx *ctx, const char *src, uint16_t port,
		      struct sockaddr_storage *addr)
{
	struct sockaddr_in6 *addr6 = (struct sockaddr_in6 *)addr;
	const char *scope = NULL;
	char *p;

	if (strlen(src) > INET6_ADDRSTRLEN)
		return -EINVAL;

	__cleanup_free char *tmp = strdup(src);
	if (!tmp) {
		libnvme_msg(ctx, LIBNVME_LOG_ERR, "cannot copy: %s\n", src);
		return -ENOMEM;
	}

	p = strchr(tmp, '%');
	if (p) {
		*p = '\0';
		scope = src + (p - tmp) + 1;
	}

	if (inet_pton(AF_INET6, tmp, &addr6->sin6_addr) != 1)
		return -EINVAL;

	if (IN6_IS_ADDR_LINKLOCAL(&addr6->sin6_addr) && scope) {
		addr6->sin6_scope_id = if_nametoindex(scope);
		if (addr6->sin6_scope_id == 0) {
			libnvme_msg(ctx, LIBNVME_LOG_ERR,
				 "can't find iface index for: %s (%m)\n", scope);
			return -EINVAL;
		}
	}

	addr6->sin6_family = AF_INET6;
	addr6->sin6_port = htons(port);
	return 0;
}

/**
 * inet_pton_with_scope - convert an IPv4/IPv6 to socket address
 * @ctx: Global context
 * @af: address family, AF_INET, AF_INET6 or AF_UNSPEC for either
 * @src: the start of the address string
 * @trsvcid: transport service identifier
 * @addr: output socket address
 *
 * Return 0 on success, errno otherwise.
 */
static int inet_pton_with_scope(struct libnvme_global_ctx *ctx, int af,
				const char *src, const char * trsvcid,
				struct sockaddr_storage *addr)
{
	int      ret  = -EINVAL;
	uint16_t port = 0;

	if (trsvcid) {
		unsigned long long tmp = strtoull(trsvcid, NULL, 0);
		port = (uint16_t)tmp;
		if (tmp != port) {
			libnvme_msg(ctx, LIBNVME_LOG_ERR, "trsvcid out of range: %s\n",
				 trsvcid);
			return -ERANGE;
		}
	} else {
		port = 0;
	}

	switch (af) {
	case AF_INET:
		ret = inet4_pton(src, port, addr);
		break;
	case AF_INET6:
		ret = inet6_pton(ctx, src, port, addr);
		break;
	case AF_UNSPEC:
		ret = inet4_pton(src, port, addr);
		if (ret)
			ret = inet6_pton(ctx, src, port, addr);
		break;
	default:
		libnvme_msg(ctx, LIBNVME_LOG_ERR, "unexpected address family %d\n", af);
	}

	return ret;
}

bool traddr_is_hostname(struct libnvme_global_ctx *ctx,
		const char *transport, const char *traddr)
{
	struct sockaddr_storage addr;

	if (!traddr || !transport)
		return false;
	if (!strcmp(traddr, "none"))
		return false;
	if (strcmp(transport, "tcp") && strcmp(transport, "rdma"))
		return false;
	if (inet_pton_with_scope(ctx, AF_UNSPEC,
			traddr, NULL, &addr) == 0)  /* scope-aware */
		return false;

	return true;
}

static int build_options(libnvme_host_t h, libnvme_ctrl_t c, char **argstr)
{
	struct libnvme_fabrics_config *cfg = libnvmf_ctrl_get_fabrics_config(c);
	const char *transport = libnvme_ctrl_get_transport(c);
	const char *hostnqn, *hostid, *hostkey, *ctrlkey = NULL;
	bool discover = false, discovery_nqn = false;
	struct libnvme_global_ctx *ctx = h->ctx;
	long keyring_id = 0;
	long key_id = 0;
	int ret;

	if (!transport) {
		libnvme_msg(ctx, LIBNVME_LOG_ERR, "need a transport (-t) argument\n");
		return -ENVME_CONNECT_TARG;
	}

	if (strncmp(transport, "loop", 4)) {
		if (!libnvme_ctrl_get_traddr(c)) {
			libnvme_msg(h->ctx, LIBNVME_LOG_ERR, "need a address (-a) argument\n");
			return -ENVME_CONNECT_AARG;
		}
	}

	/* always specify nqn as first arg - this will init the string */
	if (asprintf(argstr, "nqn=%s",
		     libnvme_ctrl_get_subsysnqn(c)) < 0) {
		return -ENOMEM;
	}

	if (!strcmp(libnvme_ctrl_get_subsysnqn(c), NVME_DISC_SUBSYS_NAME)) {
		libnvme_ctrl_set_discovery_ctrl(c, true);
		libnvme_ctrl_set_unique_discovery_ctrl(c, false);
		discovery_nqn = true;
	}

	if (libnvme_ctrl_get_discovery_ctrl(c))
		discover = true;

	hostnqn = libnvme_host_get_hostnqn(h);
	hostid = libnvme_host_get_hostid(h);
	hostkey = libnvme_host_get_dhchap_host_key(h);
	if (!hostkey)
		hostkey = libnvme_ctrl_get_dhchap_host_key(c);

	if (hostkey)
		ctrlkey = libnvme_ctrl_get_dhchap_ctrl_key(c);

	if (cfg->tls && cfg->concat) {
		libnvme_msg(h->ctx, LIBNVME_LOG_ERR, "cannot specify --tls and --concat together\n");
		return -ENVME_CONNECT_INVAL;
	}

	if (cfg->concat && !hostkey) {
		libnvme_msg(h->ctx, LIBNVME_LOG_ERR, "required argument [--dhchap-secret | -S] not specified with --concat\n");
		return -ENVME_CONNECT_INVAL;
	}

	if (cfg->tls) {
		ret = __libnvme_import_keys_from_config(h, c,
			&keyring_id, &key_id);
		if (ret)
			return ret;

		if (key_id == 0) {
			if (cfg->tls_configured_key_id)
				key_id = cfg->tls_configured_key_id;
			else
				key_id = cfg->tls_key_id;
		}
	}

	if (add_argument(ctx, argstr, transport, transport) ||
	    add_argument(ctx, argstr, traddr,
			 libnvme_ctrl_get_traddr(c)) ||
	    add_argument(ctx, argstr, host_traddr,
			 libnvme_ctrl_get_host_traddr(c)) ||
	    add_argument(ctx, argstr, host_iface,
			 libnvme_ctrl_get_host_iface(c)) ||
	    add_argument(ctx, argstr, trsvcid,
			 libnvme_ctrl_get_trsvcid(c)) ||
	    (hostnqn && add_argument(ctx, argstr, hostnqn, hostnqn)) ||
	    (hostid && add_argument(ctx, argstr, hostid, hostid)) ||
	    (discover && !discovery_nqn &&
	     add_bool_argument(ctx, argstr, discovery, true)) ||
	    (hostkey &&
	     add_argument(ctx, argstr, dhchap_secret, hostkey)) ||
	    (ctrlkey &&
	     add_argument(ctx, argstr, dhchap_ctrl_secret, ctrlkey)) ||
	    (!discover &&
	     add_int_argument(ctx, argstr, nr_io_queues,
			      cfg->nr_io_queues, false)) ||
	    (!discover &&
	     add_int_argument(ctx, argstr, nr_write_queues,
			      cfg->nr_write_queues, false)) ||
	    (!discover &&
	     add_int_argument(ctx, argstr, nr_poll_queues,
			      cfg->nr_poll_queues, false)) ||
	    (!discover &&
	     add_int_argument(ctx, argstr, queue_size,
			      cfg->queue_size, false)) ||
	    add_int_argument(ctx, argstr, keep_alive_tmo,
			     cfg->keep_alive_tmo, false) ||
	    add_int_argument(ctx, argstr, reconnect_delay,
			     cfg->reconnect_delay, false) ||
	    (strcmp(transport, "loop") &&
	     add_int_or_minus_one_argument(ctx, argstr, ctrl_loss_tmo,
			      cfg->ctrl_loss_tmo)) ||
	    (strcmp(transport, "loop") &&
	     add_int_argument(ctx, argstr, fast_io_fail_tmo,
			      cfg->fast_io_fail_tmo, false)) ||
	    (strcmp(transport, "loop") &&
	     add_int_argument(ctx, argstr, tos, cfg->tos, true)) ||
	    add_hex_argument(ctx, argstr, keyring, keyring_id, false) ||
	    (!strcmp(transport, "tcp") &&
	     add_hex_argument(ctx, argstr, tls_key, key_id, false)) ||
	    add_bool_argument(ctx, argstr, duplicate_connect,
			      cfg->duplicate_connect) ||
	    add_bool_argument(ctx, argstr, disable_sqflow,
			      cfg->disable_sqflow) ||
	    (!strcmp(transport, "tcp") &&
	     add_bool_argument(ctx, argstr, hdr_digest, cfg->hdr_digest)) ||
	    (!strcmp(transport, "tcp") &&
	     add_bool_argument(ctx, argstr, data_digest, cfg->data_digest)) ||
	    (!strcmp(transport, "tcp") &&
	     add_bool_argument(ctx, argstr, tls, cfg->tls)) ||
	    (!strcmp(transport, "tcp") &&
	     add_bool_argument(ctx, argstr, concat, cfg->concat))) {
		free(*argstr);
		return -1;
	}

	return 0;
}

#define parse_option(ctx, v, name)	   		\
	if (!strcmp(v, stringify(name))) {	 	\
		ctx->options->name = true;		\
		continue;		   		\
	}

static int __nvmf_supported_options(struct libnvme_global_ctx *ctx)
{
	char buf[0x1000], *options, *p, *v;
	__cleanup_fd int fd = -1;
	ssize_t len;

	if (ctx->options)
		return 0;

	ctx->options = calloc(1, sizeof(*ctx->options));
	if (!ctx->options)
		return -ENOMEM;

	fd = open(nvmf_dev, O_RDONLY);
	if (fd < 0) {
		libnvme_msg(ctx, LIBNVME_LOG_ERR, "Failed to open %s: %s\n",
			 nvmf_dev, libnvme_strerror(errno));
		return -ENVME_CONNECT_OPEN;
	}

	memset(buf, 0x0, sizeof(buf));
	len = read(fd, buf, sizeof(buf) - 1);
	if (len < 0) {
		if (errno == EINVAL) {
			/*
			 * Older Linux kernels don't allow reading from nvmf_dev
			 * to get supported options, so use a default set
			 */
			libnvme_msg(ctx, LIBNVME_LOG_DEBUG,
			         "Cannot read %s, using default options\n",
			         nvmf_dev);
			*ctx->options = default_supported_options;
			return 0;
		}

		libnvme_msg(ctx, LIBNVME_LOG_ERR, "Failed to read from %s: %s\n",
			 nvmf_dev, libnvme_strerror(errno));
		return -ENVME_CONNECT_READ;
	}

	buf[len] = '\0';
	options = buf;

	libnvme_msg(ctx, LIBNVME_LOG_DEBUG, "kernel supports: ");

	while ((p = strsep(&options, ",\n")) != NULL) {
		if (!*p)
			continue;
		v = strsep(&p, "= ");
		if (!v)
			continue;
		libnvme_msg(ctx, LIBNVME_LOG_DEBUG, "%s ", v);

		parse_option(ctx, v, cntlid);
		parse_option(ctx, v, concat);
		parse_option(ctx, v, ctrl_loss_tmo);
		parse_option(ctx, v, data_digest);
		parse_option(ctx, v, dhchap_ctrl_secret);
		parse_option(ctx, v, dhchap_secret);
		parse_option(ctx, v, disable_sqflow);
		parse_option(ctx, v, discovery);
		parse_option(ctx, v, duplicate_connect);
		parse_option(ctx, v, fast_io_fail_tmo);
		parse_option(ctx, v, hdr_digest);
		parse_option(ctx, v, host_iface);
		parse_option(ctx, v, host_traddr);
		parse_option(ctx, v, hostid);
		parse_option(ctx, v, hostnqn);
		parse_option(ctx, v, instance);
		parse_option(ctx, v, keep_alive_tmo);
		parse_option(ctx, v, keyring);
		parse_option(ctx, v, nqn);
		parse_option(ctx, v, nr_io_queues);
		parse_option(ctx, v, nr_poll_queues);
		parse_option(ctx, v, nr_write_queues);
		parse_option(ctx, v, queue_size);
		parse_option(ctx, v, reconnect_delay);
		parse_option(ctx, v, tls);
		parse_option(ctx, v, tls_key);
		parse_option(ctx, v, tos);
		parse_option(ctx, v, traddr);
		parse_option(ctx, v, transport);
		parse_option(ctx, v, trsvcid);
	}
	libnvme_msg(ctx, LIBNVME_LOG_DEBUG, "\n");
	return 0;
}

static int __nvmf_add_ctrl(struct libnvme_global_ctx *ctx, const char *argstr)
{
	__cleanup_fd int fd = -1;
	int ret, len = strlen(argstr);
	char buf[0x1000], *options, *p;

	fd = open(nvmf_dev, O_RDWR);
	if (fd < 0) {
		libnvme_msg(ctx, LIBNVME_LOG_ERR, "Failed to open %s: %s\n",
			 nvmf_dev, libnvme_strerror(errno));
		return -ENVME_CONNECT_OPEN;
	}

	libnvme_msg(ctx, LIBNVME_LOG_DEBUG, "connect ctrl, '%.*s'\n",
		 (int)strcspn(argstr,"\n"), argstr);
	ret = write(fd, argstr, len);
	if (ret != len) {
		libnvme_msg(ctx, LIBNVME_LOG_INFO, "Failed to write to %s: %s\n",
			 nvmf_dev, libnvme_strerror(errno));
		switch (errno) {
		case EALREADY:
			return -ENVME_CONNECT_ALREADY;
		case EINVAL:
			return -ENVME_CONNECT_INVAL;
		case EADDRINUSE:
			return -ENVME_CONNECT_ADDRINUSE;
		case ENODEV:
			return -ENVME_CONNECT_NODEV;
		case EOPNOTSUPP:
			return -ENVME_CONNECT_OPNOTSUPP;
		case ECONNREFUSED:
			return -ENVME_CONNECT_CONNREFUSED;
		case EADDRNOTAVAIL:
			return -ENVME_CONNECT_ADDRNOTAVAIL;
		case ENOKEY:
			return -ENVME_CONNECT_NOKEY;
		default:
			return -ENVME_CONNECT_WRITE;
		}
	}

	memset(buf, 0x0, sizeof(buf));
	len = read(fd, buf, sizeof(buf) - 1);
	if (len < 0) {
		libnvme_msg(ctx, LIBNVME_LOG_ERR, "Failed to read from %s: %s\n",
			 nvmf_dev, libnvme_strerror(errno));
		return -ENVME_CONNECT_READ;
	}
	libnvme_msg(ctx, LIBNVME_LOG_DEBUG, "connect ctrl, response '%.*s'\n",
		 (int)strcspn(buf, "\n"), buf);
	buf[len] = '\0';
	options = buf;
	while ((p = strsep(&options, ",\n")) != NULL) {
		if (!*p)
			continue;
		if (sscanf(p, "instance=%d", &ret) == 1)
			return ret;
	}

	libnvme_msg(ctx, LIBNVME_LOG_ERR, "Failed to parse ctrl info for \"%s\"\n", argstr);
	return -ENVME_CONNECT_PARSE;
}

static const char *lookup_context(struct libnvme_global_ctx *ctx, libnvme_ctrl_t c)
{

	libnvme_host_t h;
	libnvme_subsystem_t s;

	libnvme_for_each_host(ctx, h) {
		libnvme_for_each_subsystem(h, s) {
			struct libnvmf_context fctx = {
				.transport = libnvme_ctrl_get_transport(c),
				.traddr = libnvme_ctrl_get_traddr(c),
				.host_traddr = NULL,
				.host_iface = NULL,
				.trsvcid = libnvme_ctrl_get_trsvcid(c),
				.subsysnqn = NULL,
			};
			if (libnvme_ctrl_find(s, &fctx))
				return libnvme_subsystem_get_application(s);
		}
	}

	return NULL;
}

__public int libnvmf_create_ctrl(struct libnvme_global_ctx *ctx,
		struct libnvmf_context *fctx, libnvme_ctrl_t *cp)
{
	return _libnvme_create_ctrl(ctx, fctx, cp);
}

__public int libnvmf_add_ctrl(libnvme_host_t h, libnvme_ctrl_t c)
{
	libnvme_subsystem_t s;
	const char *root_app, *app;
	__cleanup_free char *argstr = NULL;
	int ret;

	/* Are duplicate connections allowed on existing controller */
	if (libnvme_ctrl_get_name(c) && !c->cfg.duplicate_connect)
		return -ENVME_CONNECT_ALREADY;

	/* apply configuration from config file (JSON) */
	s = libnvme_lookup_subsystem(h, NULL, libnvme_ctrl_get_subsysnqn(c));
	if (s) {
		libnvme_ctrl_t fc;
		struct libnvmf_context fctx = {
			.transport = libnvme_ctrl_get_transport(c),
			.traddr = libnvme_ctrl_get_traddr(c),
			.host_traddr = libnvme_ctrl_get_host_traddr(c),
			.host_iface = libnvme_ctrl_get_trsvcid(c),
			.trsvcid = libnvme_ctrl_get_trsvcid(c),
			.subsysnqn = NULL,
		};

		fc = libnvme_ctrl_find(s, &fctx);
		if (fc) {
			const char *key;

			merge_config(c, libnvmf_ctrl_get_fabrics_config(fc));
			/*
			 * An authentication key might already been set
			 * in @cfg, so ensure to update @c with the correct
			 * controller key.
			 */
			key = libnvme_ctrl_get_dhchap_host_key(fc);
			if (key)
				libnvme_ctrl_set_dhchap_host_key(c, key);
			key = libnvme_ctrl_get_dhchap_ctrl_key(fc);
			if (key)
				libnvme_ctrl_set_dhchap_ctrl_key(c, key);
			key = libnvme_ctrl_get_keyring(fc);
			if (key)
				libnvme_ctrl_set_keyring(c, key);
			key = libnvme_ctrl_get_tls_key_identity(fc);
			if (key)
				libnvme_ctrl_set_tls_key_identity(c, key);
			key = libnvme_ctrl_get_tls_key(fc);
			if (key)
				libnvme_ctrl_set_tls_key(c, key);
		}

	}

	root_app = libnvme_get_application(h->ctx);
	if (root_app) {
		app = libnvme_subsystem_get_application(s);
		if (!app && libnvme_ctrl_get_discovery_ctrl(c))
			app = lookup_context(h->ctx, c);

		/*
		 * configuration is managed by an application,
		 * refuse to act on subsystems which either have
		 * no application set or which habe a different
		 * application string.
		 */
		if (app && strcmp(app, root_app)) {
			libnvme_msg(h->ctx, LIBNVME_LOG_INFO, "skip %s, not managed by %s\n",
				 libnvme_subsystem_get_subsysnqn(s), root_app);
			return -ENVME_CONNECT_IGNORED;
		}
	}

	libnvme_ctrl_set_discovered(c, true);
	if (traddr_is_hostname(h->ctx, c->transport, c->traddr)) {
		char *traddr = c->traddr;

		if (hostname2traddr(h->ctx, traddr, &c->traddr)) {
			c->traddr = traddr;
			return -ENVME_CONNECT_TRADDR;
		}
		free(traddr);
	}

	ret = build_options(h, c, &argstr);
	if (ret)
		return ret;

	ret = __nvmf_add_ctrl(h->ctx, argstr);
	if (ret < 0)
		return ret;

	libnvme_msg(h->ctx, LIBNVME_LOG_INFO, "nvme%d: %s connected\n", ret,
		 libnvme_ctrl_get_subsysnqn(c));
	return libnvme_init_ctrl(h, c, ret);
}

__public int libnvmf_connect_ctrl(libnvme_ctrl_t c)
{
	__cleanup_free char *argstr = NULL;
	int ret;

	ret = build_options(c->s->h, c, &argstr);
	if (ret)
		return ret;

	ret = __nvmf_add_ctrl(c->s->h->ctx, argstr);
	if (ret < 0)
		return ret;

	return 0;
}

__public int libnvmf_disconnect_ctrl(libnvme_ctrl_t c)
{
	struct libnvme_global_ctx *ctx = c->s && c->s->h ? c->s->h->ctx : NULL;
	int ret;

	ret = libnvme_set_attr(libnvme_ctrl_get_sysfs_dir(c),
			    "delete_controller", "1");
	if (ret < 0) {
		libnvme_msg(ctx, LIBNVME_LOG_ERR,
			"%s: failed to disconnect, error %d\n", c->name, errno);
		return ret;
	}
	libnvme_msg(ctx, LIBNVME_LOG_INFO, "%s: %s disconnected\n",
		c->name, c->subsysnqn);
	nvme_deconfigure_ctrl(c);
	return 0;
}

static void nvmf_update_tls_concat(struct nvmf_disc_log_entry *e,
		libnvme_ctrl_t c, libnvme_host_t h)
{
	if (e->trtype != NVMF_TRTYPE_TCP ||
	    e->tsas.tcp.sectype == NVMF_TCP_SECTYPE_NONE)
		return;

	if (e->treq & NVMF_TREQ_REQUIRED) {
		libnvme_msg(h->ctx, LIBNVME_LOG_DEBUG,
			"setting --tls due to treq %s and sectype %s\n",
			libnvmf_treq_str(e->treq),
			libnvmf_sectype_str(e->tsas.tcp.sectype));

		c->cfg.tls = true;
		return;
	}

	if (e->treq & NVMF_TREQ_NOT_REQUIRED) {
		libnvme_msg(h->ctx, LIBNVME_LOG_DEBUG,
			"setting --concat due to treq %s and sectype %s\n",
			libnvmf_treq_str(e->treq),
			libnvmf_sectype_str(e->tsas.tcp.sectype));

		c->cfg.concat = true;
		return;
	}
}

static int nvmf_connect_disc_entry(libnvme_host_t h,
		struct nvmf_disc_log_entry *e,
		struct libnvmf_context *fctx,
		bool *discover, libnvme_ctrl_t *cp)
{
	libnvme_ctrl_t c;
	int ret;

	switch (e->trtype) {
	case NVMF_TRTYPE_RDMA:
	case NVMF_TRTYPE_TCP:
		switch (e->adrfam) {
		case NVMF_ADDR_FAMILY_IP4:
		case NVMF_ADDR_FAMILY_IP6:
			fctx->traddr = e->traddr;
			fctx->trsvcid = e->trsvcid;
			break;
		default:
			libnvme_msg(h->ctx, LIBNVME_LOG_ERR,
				 "skipping unsupported adrfam %d\n",
				 e->adrfam);
			return -EINVAL;
		}
		break;
        case NVMF_TRTYPE_FC:
		switch (e->adrfam) {
		case NVMF_ADDR_FAMILY_FC:
			fctx->traddr = e->traddr;
			break;
		default:
			libnvme_msg(h->ctx, LIBNVME_LOG_ERR,
				 "skipping unsupported adrfam %d\n",
				 e->adrfam);
			return -EINVAL;
		}
		break;
	case NVMF_TRTYPE_LOOP:
		fctx->traddr = strlen(e->traddr) ? e->traddr : NULL;
		break;
	default:
		libnvme_msg(h->ctx, LIBNVME_LOG_ERR, "skipping unsupported transport %d\n",
			 e->trtype);
		return -EINVAL;
	}

	fctx->transport = libnvmf_trtype_str(e->trtype);
	fctx->subsysnqn = e->subnqn;

	libnvme_msg(h->ctx, LIBNVME_LOG_DEBUG, "lookup ctrl "
		 "(transport: %s, traddr: %s, trsvcid %s)\n",
		 fctx->transport, fctx->traddr, fctx->trsvcid);

	ret = _libnvme_create_ctrl(h->ctx, fctx, &c);
	if (ret) {
		libnvme_msg(h->ctx, LIBNVME_LOG_DEBUG, "skipping discovery entry, "
			 "failed to allocate %s controller with traddr %s\n",
			 fctx->transport, fctx->traddr);
		return ret;
	}

	switch (e->subtype) {
	case NVME_NQN_CURR:
		libnvme_ctrl_set_discovered(c, true);
		libnvme_ctrl_set_unique_discovery_ctrl(c,
				strcmp(e->subnqn, NVME_DISC_SUBSYS_NAME));
		break;
	case NVME_NQN_DISC:
		if (discover)
			*discover = true;
		libnvme_ctrl_set_discovery_ctrl(c, true);
		libnvme_ctrl_set_unique_discovery_ctrl(c,
				strcmp(e->subnqn, NVME_DISC_SUBSYS_NAME));
		break;
	default:
		libnvme_msg(h->ctx, LIBNVME_LOG_ERR, "unsupported subtype %d\n",
			 e->subtype);
		fallthrough;
	case NVME_NQN_NVME:
		libnvme_ctrl_set_discovery_ctrl(c, false);
		libnvme_ctrl_set_unique_discovery_ctrl(c, false);
		break;
	}

	if (libnvme_ctrl_get_discovered(c)) {
		libnvme_free_ctrl(c);
		return -EAGAIN;
	}

	if (e->treq & NVMF_TREQ_DISABLE_SQFLOW &&
	    nvmf_check_option(h->ctx, disable_sqflow))
		c->cfg.disable_sqflow = true;

	/* update tls or concat */
	nvmf_update_tls_concat(e, c, h);

	ret = libnvmf_add_ctrl(h, c);
	if (!ret) {
		*cp = c;
		return 0;
	}

	if (ret == EINVAL && c->cfg.disable_sqflow) {
		/* disable_sqflow is unrecognized option on older kernels */
		libnvme_msg(h->ctx, LIBNVME_LOG_INFO, "failed to connect controller, "
			 "retry with disabling SQ flow control\n");
		c->cfg.disable_sqflow = false;
		ret = libnvmf_add_ctrl(h, c);
		if (!ret) {
			*cp = c;
			return 0;
		}
	}
	libnvme_free_ctrl(c);
	return -ENOENT;
}

/*
 * Most of nvmf_discovery_log is reserved, so only fetch the initial bytes.
 * 8 bytes for GENCTR, 8 for NUMREC, and 2 for RECFMT.
 * Since only multiples of 4 bytes are allowed, round 18 up to 20.
 */
#define DISCOVERY_HEADER_LEN 20

static int nvme_discovery_log(libnvme_ctrl_t ctrl,
			      const struct libnvmf_discovery_args *args,
			      struct nvmf_discovery_log **logp)
{
	struct libnvme_global_ctx *ctx = ctrl->ctx;
	struct nvmf_discovery_log *log;
	int retries = 0;
	int err;
	const char *name = libnvme_ctrl_get_name(ctrl);
	uint64_t genctr, numrec;
	struct libnvme_transport_handle *hdl;

	hdl = libnvme_ctrl_get_transport_handle(ctrl);
	struct libnvme_passthru_cmd cmd;

	log = __libnvme_alloc(sizeof(*log));
	if (!log) {
		libnvme_msg(ctx, LIBNVME_LOG_ERR,
			 "could not allocate memory for discovery log header\n");
		return -ENOMEM;
	}

	libnvme_msg(ctx, LIBNVME_LOG_DEBUG, "%s: get header (try %d/%d)\n",
		 name, retries, args->max_retries);
	nvme_init_get_log_discovery(&cmd, 0, log, DISCOVERY_HEADER_LEN);
	err = libnvme_get_log(hdl, &cmd, false, DISCOVERY_HEADER_LEN);
	if (err) {
		libnvme_msg(ctx, LIBNVME_LOG_INFO,
			 "%s: discover try %d/%d failed, errno %d status 0x%x\n",
			 name, retries, args->max_retries, errno, err);
		goto out_free_log;
	}

	do {
		size_t entries_size;

		numrec = le64_to_cpu(log->numrec);
		genctr = le64_to_cpu(log->genctr);

		if (numrec == 0)
			break;

		free(log);
		entries_size = sizeof(*log->entries) * numrec;
		log = __libnvme_alloc(sizeof(*log) + entries_size);
		if (!log) {
		libnvme_msg(ctx, LIBNVME_LOG_ERR,
				 "could not alloc memory for discovery log page\n");
			return -ENOMEM;
		}

		libnvme_msg(ctx, LIBNVME_LOG_DEBUG,
			 "%s: get %" PRIu64 " records (genctr %" PRIu64 ")\n",
			 name, numrec, genctr);

		nvme_init_get_log_discovery(&cmd, sizeof(*log), log->entries, entries_size);
		cmd.cdw10 |= NVME_FIELD_ENCODE(args->lsp,
					       NVME_LOG_CDW10_LSP_SHIFT,
					       NVME_LOG_CDW10_LSP_MASK);
		err = libnvme_get_log(hdl, &cmd, false, NVME_LOG_PAGE_PDU_SIZE);
		if (err) {
			libnvme_msg(ctx, LIBNVME_LOG_INFO,
				 "%s: discover try %d/%d failed, errno %d status 0x%x\n",
				 name, retries, args->max_retries, errno, err);
			goto out_free_log;
		}

		/*
		 * If the log page was read with multiple Get Log Page commands,
		 * genctr must be checked afterwards to ensure atomicity
		 */
		libnvme_msg(ctx, LIBNVME_LOG_DEBUG, "%s: get header again\n", name);

		nvme_init_get_log_discovery(&cmd, 0, log, DISCOVERY_HEADER_LEN);
		err = libnvme_get_log(hdl, &cmd, false, DISCOVERY_HEADER_LEN);
		if (err) {
			libnvme_msg(ctx, LIBNVME_LOG_INFO,
				 "%s: discover try %d/%d failed, errno %d status 0x%x\n",
				 name, retries, args->max_retries, errno, err);
			goto out_free_log;
		}
	} while (genctr != le64_to_cpu(log->genctr) &&
		 ++retries < args->max_retries);

	if (genctr != le64_to_cpu(log->genctr)) {
		libnvme_msg(ctx, LIBNVME_LOG_INFO, "%s: discover genctr mismatch\n", name);
		err = -EAGAIN;
	} else if (numrec != le64_to_cpu(log->numrec)) {
		libnvme_msg(ctx, LIBNVME_LOG_INFO,
			 "%s: numrec changed unexpectedly "
			 "from %" PRIu64 " to %" PRIu64 "\n",
			 name, numrec, le64_to_cpu(log->numrec));
		err = -EBADSLT;
	} else {
		*logp = log;
		return 0;
	}

out_free_log:
	free(log);
	return err;
}

static void sanitize_discovery_log_entry(struct libnvme_global_ctx *ctx,
		struct nvmf_disc_log_entry *e)
{
	strchomp(e->trsvcid, sizeof(e->trsvcid));
	strchomp(e->traddr, sizeof(e->traddr));

	/*
	 * Report traddr always in 'nn-0x:pn-0x' format, but some discovery logs
	 * provide 'nn-0x,pn-0x'.
	 */
	if (e->trtype == NVMF_TRTYPE_FC) {
		char *comma = strchr(e->traddr, ',');

		if (comma) {
			libnvme_msg(ctx, LIBNVME_LOG_WARN,
				"invalid traddr separator ',' instead ':', fixing it");
			*comma = ':';
		}
	}
}

__public int libnvmf_get_discovery_log(libnvme_ctrl_t ctrl,
				    const struct libnvmf_discovery_args *args,
				    struct nvmf_discovery_log **logp)
{
	static const struct libnvmf_discovery_args defaults = {
		.max_retries = 6,
		.lsp         = NVMF_LOG_DISC_LSP_NONE,
	};
	struct nvmf_discovery_log *log;
	int err;

	if (!args)
		args = &defaults;

	err = nvme_discovery_log(ctrl, args, &log);
	if (err)
		return err;

	for (int i = 0; i < le64_to_cpu(log->numrec); i++)
		sanitize_discovery_log_entry(ctrl->ctx, &log->entries[i]);

	*logp = log;
	return 0;
}


/**
 * nvmf_get_tel() - Calculate the amount of memory needed for a DIE.
 * @hostsymname:	Symbolic name (may be NULL)
 *
 * Each Discovery Information Entry (DIE) must contain at a minimum an
 * Extended Attribute for the HostID. The Entry may optionally contain an
 * Extended Attribute for the Symbolic Name.
 *
 * Return: Total Entry Length
 */
static __u32 nvmf_get_tel(const char *hostsymname)
{
	__u32 tel = sizeof(struct nvmf_ext_die);
	__u16 len;

	/* Host ID is mandatory */
	tel += libnvmf_exat_size(NVME_UUID_LEN);

	/* Symbolic name is optional */
	len = hostsymname ? strlen(hostsymname) : 0;
	if (len)
		tel += libnvmf_exat_size(len);

	return tel;
}

/**
 * nvmf_fill_die() - Fill a Discovery Information Entry.
 * @die:	Pointer to Discovery Information Entry to be filled
 * @h:		Pointer to the host data structure
 * @tel:	Length of the DIE
 * @trtype:	Transport type
 * @adrfam:	Address family
 * @reg_addr:	Address to register. Setting this to an empty string tells
 *		the DC to infer address from the source address of the socket.
 * @tsas:	Transport Specific Address Subtype for the address being
 *		registered.
 */
static void nvmf_fill_die(struct nvmf_ext_die *die, struct libnvme_host *h,
			  __u32 tel, __u8 trtype, __u8 adrfam,
			  const char *reg_addr, union nvmf_tsas *tsas)
{
	__u16 numexat = 0;
	size_t symname_len;
	struct nvmf_ext_attr *exat;

	die->tel = cpu_to_le32(tel);
	die->trtype = trtype;
	die->adrfam = adrfam;

	memcpy(die->nqn, h->hostnqn, MIN(sizeof(die->nqn), strlen(h->hostnqn)));
	memcpy(die->traddr, reg_addr, MIN(sizeof(die->traddr), strlen(reg_addr)));

	if (tsas)
		memcpy(&die->tsas, tsas, sizeof(die->tsas));

	/* Extended Attribute for the HostID (mandatory) */
	numexat++;
	exat = die->exat;
	exat->exattype = cpu_to_le16(NVMF_EXATTYPE_HOSTID);
	exat->exatlen  = cpu_to_le16(libnvmf_exat_len(NVME_UUID_LEN));
	libnvme_uuid_from_string(h->hostid, exat->exatval);

	/* Extended Attribute for the Symbolic Name (optional) */
	symname_len = h->hostsymname ? strlen(h->hostsymname) : 0;
	if (symname_len) {
		__u16 exatlen = libnvmf_exat_len(symname_len);

		numexat++;
		exat = libnvmf_exat_ptr_next(exat);
		exat->exattype = cpu_to_le16(NVMF_EXATTYPE_SYMNAME);
		exat->exatlen  = cpu_to_le16(exatlen);
		memcpy(exat->exatval, h->hostsymname, symname_len);
		/* Per Base specs, ASCII strings must be padded with spaces */
		memset(&exat->exatval[symname_len], ' ', exatlen - symname_len);
	}

	die->numexat = cpu_to_le16(numexat);
}

/**
 * nvmf_dim() - Explicit reg, dereg, reg-update issuing DIM
 * @c:		Host NVMe controller instance maintaining the admin queue used to
 *		submit the DIM command to the DC.
 * @tas:	Task field of the Command Dword 10 (cdw10). Indicates whether to
 *		perform a Registration, Deregistration, or Registration-update.
 * @trtype:	Transport type (&enum nvmf_trtype - must be NVMF_TRTYPE_TCP)
 * @adrfam:	Address family (&enum nvmf_addr_family)
 * @reg_addr:	Address to register. Setting this to an empty string tells
 *		the DC to infer address from the source address of the socket.
 * @tsas:	Transport Specific Address Subtype for the address being
 *		registered.
 * @result:	Location where to save the command-specific result returned by
 *		the discovery controller.
 *
 * Perform explicit registration, deregistration, or
 * registration-update (specified by @tas) by sending a Discovery
 * Information Management (DIM) command to the Discovery Controller
 * (DC).
 *
 * Return: 0 on success; on failure -1 is returned and errno is set
 */
static int nvmf_dim(libnvme_ctrl_t c, enum nvmf_dim_tas tas, __u8 trtype,
		    __u8 adrfam, const char *reg_addr, union nvmf_tsas *tsas,
		    __u32 *result)
{
	struct libnvme_global_ctx *ctx = c->s && c->s->h ? c->s->h->ctx : NULL;
	__cleanup_free struct nvmf_dim_data *dim = NULL;
	struct libnvme_transport_handle *hdl = libnvme_ctrl_get_transport_handle(c);
	struct libnvme_passthru_cmd cmd;
	struct nvmf_ext_die  *die;
	__u32 tdl;
	__u32 tel;
	int ret;

	if (!c->s) {
		libnvme_msg(ctx, LIBNVME_LOG_ERR,
			 "%s: failed to perform DIM. subsystem undefined.\n",
			 c->name);
		return -EINVAL;
	}

	if (!c->s->h) {
		libnvme_msg(ctx, LIBNVME_LOG_ERR,
			 "%s: failed to perform DIM. host undefined.\n",
			 c->name);
		return -EINVAL;
	}

	if (!c->s->h->hostid) {
		libnvme_msg(ctx, LIBNVME_LOG_ERR,
			 "%s: failed to perform DIM. hostid undefined.\n",
			 c->name);
		return -EINVAL;
	}

	if (!c->s->h->hostnqn) {
		libnvme_msg(ctx, LIBNVME_LOG_ERR,
			 "%s: failed to perform DIM. hostnqn undefined.\n",
			 c->name);
		return -EINVAL;
	}

	if (strcmp(c->transport, "tcp")) {
		libnvme_msg(ctx, LIBNVME_LOG_ERR,
			 "%s: DIM only supported for TCP connections.\n",
			 c->name);
		return -EINVAL;
	}

	/* Register one Discovery Information Entry (DIE) of size TEL */
	tel = nvmf_get_tel(c->s->h->hostsymname);
	tdl = sizeof(struct nvmf_dim_data) + tel;

	dim = (struct nvmf_dim_data *)calloc(1, tdl);
	if (!dim) {
		return -ENOMEM;
	}

	dim->tdl    = cpu_to_le32(tdl);
	dim->nument = cpu_to_le64(1);    /* only one DIE to register */
	dim->entfmt = cpu_to_le16(NVMF_DIM_ENTFMT_EXTENDED);
	dim->etype  = cpu_to_le16(NVMF_DIM_ETYPE_HOST);
	dim->ektype = cpu_to_le16(0x5F); /* must be 0x5F per specs */

	memcpy(dim->eid, c->s->h->hostnqn,
	       MIN(sizeof(dim->eid), strlen(c->s->h->hostnqn)));

	ret = get_entity_name(dim->ename, sizeof(dim->ename));
	if (ret < 0)
		libnvme_msg(ctx, LIBNVME_LOG_INFO, "%s: Failed to retrieve ENAME. %s.\n",
			 c->name, libnvme_strerror(-ret));
	else if (ret == 0)
		libnvme_msg(ctx, LIBNVME_LOG_INFO, "%s: Failed to retrieve ENAME.\n",
			 c->name);

	ret = get_entity_version(dim->ever, sizeof(dim->ever));
	if (ret <= 0)
		libnvme_msg(ctx, LIBNVME_LOG_INFO, "%s: Failed to retrieve EVER.\n", c->name);

	die = &dim->die->extended;
	nvmf_fill_die(die, c->s->h, tel, trtype, adrfam, reg_addr, tsas);

	nvme_init_dim_send(&cmd, tas, dim, tdl);
	return libnvme_submit_admin_passthru(hdl, &cmd);
}

/**
 * nvme_get_adrfam() - Get address family for the address we're registering
 * with the DC.
 *
 * We retrieve this info from the socket itself. If we can't get the source
 * address from the socket, then we'll infer the address family from the
 * address of the DC since the DC address has the same address family.
 *
 * @c: Host NVMe controller instance maintaining the admin queue used to
 *   submit the DIM command to the DC.
 *
 * Return: The address family of the source address associated with the
 *   socket connected to the DC.
 */
static __u8 nvme_get_adrfam(libnvme_ctrl_t c)
{
	struct sockaddr_storage addr;
	__u8 adrfam = NVMF_ADDR_FAMILY_IP4;
	struct libnvme_global_ctx *ctx = c->s && c->s->h ? c->s->h->ctx : NULL;

	if (!inet_pton_with_scope(ctx, AF_UNSPEC, c->traddr, c->trsvcid, &addr)) {
		if (addr.ss_family == AF_INET6)
			adrfam = NVMF_ADDR_FAMILY_IP6;
	}

	return adrfam;
}

/* These string definitions must match with the kernel */
static const char *cntrltype_str[] = {
	[NVME_CTRL_CNTRLTYPE_IO] = "io",
	[NVME_CTRL_CNTRLTYPE_DISCOVERY] = "discovery",
	[NVME_CTRL_CNTRLTYPE_ADMIN] = "admin",
};

static const char *dctype_str[] = {
	[NVME_CTRL_DCTYPE_NOT_REPORTED] = "none",
	[NVME_CTRL_DCTYPE_DDC] = "ddc",
	[NVME_CTRL_DCTYPE_CDC] = "cdc",
};

/**
 * nvme_fetch_cntrltype_dctype_from_id - Get cntrltype and dctype from identify command
 * @c:	Controller instance
 *
 * On legacy kernels the cntrltype and dctype are not exposed through the
 * sysfs. We must get them directly from the controller by performing an
 * identify command.
 */
static int nvme_fetch_cntrltype_dctype_from_id(libnvme_ctrl_t c)
{
	__cleanup_free struct nvme_id_ctrl *id = NULL;
	int ret;

	id = __libnvme_alloc(sizeof(*id));
	if (!id)
		return -ENOMEM;

	ret = libnvme_ctrl_identify(c, id);
	if (ret)
		return ret;

	if (!c->cntrltype) {
		if (id->cntrltype > NVME_CTRL_CNTRLTYPE_ADMIN || !cntrltype_str[id->cntrltype])
			c->cntrltype = strdup("reserved");
		else
			c->cntrltype = strdup(cntrltype_str[id->cntrltype]);
	}

	if (!c->dctype) {
		if (id->dctype > NVME_CTRL_DCTYPE_CDC || !dctype_str[id->dctype])
			c->dctype = strdup("reserved");
		else
			c->dctype = strdup(dctype_str[id->dctype]);
	}
	return 0;
}

__public bool libnvmf_is_registration_supported(libnvme_ctrl_t c)
{
	if (!c->cntrltype || !c->dctype)
		if (nvme_fetch_cntrltype_dctype_from_id(c))
			return false;

	return !strcmp(c->dctype, "ddc") || !strcmp(c->dctype, "cdc");
}

__public int libnvmf_register_ctrl(libnvme_ctrl_t c, enum nvmf_dim_tas tas, __u32 *result)
{
	if (!libnvmf_is_registration_supported(c))
		return -ENOTSUP;

	/* We're registering our source address with the DC. To do
	 * that, we can simply send an empty string. This tells the DC
	 * to retrieve the source address from the socket and use that
	 * as the registration address.
	 */
	return nvmf_dim(c, tas, NVMF_TRTYPE_TCP, nvme_get_adrfam(c), "", NULL, result);
}

#define IS_XDIGIT(c) ((c >= '0' && c <= '9') || \
		      (c >= 'A' && c <= 'F') || \
		      (c >= 'a' && c <= 'f'))
#define XDIGIT_VAL(c) ((c >= '0' && c <= '9') ? c - '0' : ( \
		       (c >= 'A' && c <= 'F') ? c - 'A' + 10 : c - 'a' + 10))

/* returns newly allocated string */
static char *unescape_uri(const char *str, int len)
{
	char *dst;
	int l;
	int i, j;

	l = len > 0 ? len : strlen(str);
	dst = malloc(l + 1);
	for (i = 0, j = 0; i < l; i++, j++) {
		if (str[i] == '%' && i + 2 < l &&
		    IS_XDIGIT(str[i + 1]) && IS_XDIGIT(str[i + 2])) {
			dst[j] = (XDIGIT_VAL(str[i + 1]) << 4) +
				  XDIGIT_VAL(str[i + 2]);
			i += 2;
		} else
			dst[j] = str[i];
	}
	dst[j] = '\0';
	return dst;
}

__public int libnvmf_uri_parse(const char *str, struct libnvmf_uri **urip)
{
	__cleanup_uri struct libnvmf_uri *uri = NULL;
	__cleanup_free char *scheme = NULL;
	__cleanup_free char *authority = NULL;
	__cleanup_free char *path = NULL;
	__cleanup_free char *h = NULL;
	const char *host;
	int i;

	/* As defined in Boot Specification rev. 1.0:
	 *
	 * section 1.5.7: NVMe-oF URI Format
	 *  nvme+tcp://192.168.1.1:4420/
	 *  nvme+tcp://[FE80::1010]:4420/
	 *
	 * section 3.1.2.5.3: DHCP Root-Path - a hierarchical NVMe-oF URI Format
	 *  NVME<+PROTOCOL>://<SERVERNAME/IP>[:TRANSPORT PORT]/<SUBSYS NQN>/<NID>
	 * or
	 *  NVME<+PROTOCOL>://<DISCOVERY CONTROLLER ADDRESS>[:DISCOVERY-
	 *  -CONTROLLER PORT]/NQN.2014-08.ORG.NVMEXPRESS.DISCOVERY/<NID>
	 */

	uri = calloc(1, sizeof(struct libnvmf_uri));
	if (!uri)
		return -ENOMEM;

	if (sscanf(str, "%m[^:/]://%m[^/?#]%ms",
		   &scheme, &authority, &path) < 2)
		return -EINVAL;

	if (sscanf(scheme, "%m[^+]+%ms",
		   &uri->scheme, &uri->protocol) < 1)
		return -EINVAL;

	/* split userinfo */
	host = strrchr(authority, '@');
	if (host) {
		host++;
		uri->userinfo = unescape_uri(authority, host - authority);
	} else
		host = authority;

	/* try matching IPv6 address first */
	if (sscanf(host, "[%m[^]]]:%d",
		   &uri->host, &uri->port) < 1) {
		/* treat it as IPv4/hostname */
		if (sscanf(host, "%m[^:]:%d",
			   &h, &uri->port) < 1)
			return -EINVAL;
		uri->host = unescape_uri(h, 0);
	}

	/* split path into elements */
	if (path) {
		char *e, *elem;

		/* separate the fragment */
		e = strrchr(path, '#');
		if (e) {
			uri->fragment = unescape_uri(e + 1, 0);
			*e = '\0';
		}
		/* separate the query string */
		e = strrchr(path, '?');
		if (e) {
			uri->query = unescape_uri(e + 1, 0);
			*e = '\0';
		}

		/* count elements first */
		for (i = 0, e = path; *e; e++)
			if (*e == '/' && *(e + 1) != '/')
				i++;
		uri->path_segments = calloc(i + 2, sizeof(char *));

		i = 0;
		elem = strtok_r(path, "/", &e);
		if (elem)
			uri->path_segments[i++] = unescape_uri(elem, 0);
		while (elem && strlen(elem)) {
			elem = strtok_r(NULL, "/", &e);
			if (elem)
				uri->path_segments[i++] = unescape_uri(elem, 0);
		}
	}

	*urip = uri;
	uri = NULL;

	return 0;
}

__public void libnvmf_uri_free(struct libnvmf_uri *uri)
{
	char **s;

	if (!uri)
		return;
	free(uri->scheme);
	free(uri->protocol);
	free(uri->userinfo);
	free(uri->host);
	for (s = uri->path_segments; s && *s; s++)
		free(*s);
	free(uri->path_segments);
	free(uri->query);
	free(uri->fragment);
	free(uri);
}

static libnvme_ctrl_t lookup_ctrl(libnvme_host_t h, struct libnvmf_context *fctx)
{
	libnvme_subsystem_t s;
	libnvme_ctrl_t c;

	libnvme_for_each_subsystem(h, s) {
		c = libnvme_ctrl_find(s, fctx);
		if (c)
			return c;
	}

	return NULL;
}

static int lookup_host(struct libnvme_global_ctx *ctx,
		struct libnvmf_context *fctx, struct libnvme_host **host)
{
	__cleanup_free char *hnqn = NULL;
	__cleanup_free char *hid = NULL;
	struct libnvme_host *h;
	int err;

	err = libnvme_host_get_ids(ctx, fctx->hostnqn, fctx->hostid, &hnqn, &hid);
	if (err < 0)
		return err;

	h = libnvme_lookup_host(ctx, hnqn, hid);
	if (!h)
		return -ENOMEM;

	*host = h;

	return 0;
}

static int setup_connection(struct libnvmf_context *fctx, struct libnvme_host *h,
		bool discovery)
{
	if (fctx->hostkey)
		libnvme_host_set_dhchap_host_key(h, fctx->hostkey);

	if (!fctx->trsvcid)
		fctx->trsvcid = libnvmf_get_default_trsvcid(fctx->transport,
			discovery);

	return 0;
}


static int set_discovery_kato(struct libnvmf_context *fctx)
{
	int tmo = fctx->cfg.keep_alive_tmo;

	/* Set kato to NVMF_DEF_DISC_TMO for persistent controllers */
	if (fctx->persistent && !fctx->cfg.keep_alive_tmo)
		fctx->cfg.keep_alive_tmo = fctx->default_keep_alive_timeout;
	/* Set kato to zero for non-persistent controllers */
	else if (!fctx->persistent && (fctx->cfg.keep_alive_tmo > 0))
		fctx->cfg.keep_alive_tmo = 0;

	return tmo;
}

static void nvme_parse_tls_args(const char *keyring, const char *tls_key,
				const char *tls_key_identity,
				struct libnvme_fabrics_config *cfg, libnvme_ctrl_t c)
{
	if (keyring) {
		char *endptr;
		long id = strtol(keyring, &endptr, 0);

		if (endptr != keyring)
			cfg->keyring_id = id;
		else
			libnvme_ctrl_set_keyring(c, keyring);
	}

	if (tls_key_identity)
		libnvme_ctrl_set_tls_key_identity(c, tls_key_identity);

	if (tls_key) {
		char *endptr;
		long id = strtol(tls_key, &endptr, 0);

		if (endptr != tls_key)
			cfg->tls_key_id = id;
		else
			libnvme_ctrl_set_tls_key(c, tls_key);
	}
}


static int _nvmf_discovery(struct libnvme_global_ctx *ctx,
		struct libnvmf_context *fctx, bool connect,
		struct libnvme_ctrl *c)
{
	__cleanup_free struct nvmf_discovery_log *log = NULL;
	libnvme_subsystem_t s = libnvme_ctrl_get_subsystem(c);
	libnvme_host_t h = libnvme_subsystem_get_host(s);
	uint64_t numrec;
	int err;

	struct libnvmf_discovery_args args = {
		.max_retries = fctx->default_max_discovery_retries,
		.lsp = NVMF_LOG_DISC_LSP_NONE,
	};

	err = nvme_discovery_log(c, &args, &log);
	if (err) {
		libnvme_msg(ctx, LIBNVME_LOG_ERR, "failed to get discovery log: %s\n",
			libnvme_strerror(err));
		return err;
	}

	numrec = le64_to_cpu(log->numrec);
	if (fctx->discovery_log)
		fctx->discovery_log(fctx, connect, log, numrec,
			fctx->user_data);

	if (!connect)
		return 0;

	for (int i = 0; i < numrec; i++) {
		struct nvmf_disc_log_entry *e = &log->entries[i];
		libnvme_ctrl_t cl;
		bool discover = false;
		bool disconnect;
		libnvme_ctrl_t child = { 0 };
		int tmo = fctx->cfg.keep_alive_tmo;
		struct libnvmf_context nfctx = *fctx;

		sanitize_discovery_log_entry(c->ctx, e);

		nfctx.subsysnqn = e->subnqn;
		nfctx.transport = libnvmf_trtype_str(e->trtype);
		nfctx.traddr = e->traddr;
		nfctx.trsvcid = e->trsvcid;
		nfctx.cfg = fctx->cfg;

		/* Already connected ? */
		cl = lookup_ctrl(h, &nfctx);
		if (cl && libnvme_ctrl_get_name(cl))
			continue;

		/* Skip connect if the transport types don't match */
		if (strcmp(libnvme_ctrl_get_transport(c),
			   nfctx.transport))
			continue;

		if (e->subtype == NVME_NQN_DISC ||
		    e->subtype == NVME_NQN_CURR) {
			__u16 eflags = le16_to_cpu(e->eflags);
			/*
			 * Does this discovery controller return the
			 * same information?
			 */
			if (eflags & NVMF_DISC_EFLAGS_DUPRETINFO)
				continue;

			/*
			 * Are we supposed to keep the discovery
			 * controller around?
			 */
			disconnect = !nfctx.persistent;

			if (strcmp(e->subnqn, NVME_DISC_SUBSYS_NAME)) {
				/*
				 * Does this discovery controller doesn't
				 * support explicit persistent connection?
				 */
				if (!(eflags & NVMF_DISC_EFLAGS_EPCSD))
					disconnect = true;
				else
					disconnect = false;
			}

			set_discovery_kato(&nfctx);
		} else {
			/* NVME_NQN_NVME */
			disconnect = false;
		}

		err = nvmf_connect_disc_entry(h, e, &nfctx, &discover, &child);

		nfctx.cfg.keep_alive_tmo = tmo;

		if (!child) {
			if (discover)
				_nvmf_discovery(ctx, &nfctx, true, child);

			if (child && disconnect) {
				libnvmf_disconnect_ctrl(child);
				libnvme_free_ctrl(child);
			}
		} else if (err == -ENVME_CONNECT_ALREADY) {
			struct nvmf_disc_log_entry *e = &log->entries[i];

			nfctx.already_connected(&nfctx, h, e->subnqn,
				libnvmf_trtype_str(e->trtype), e->traddr,
				e->trsvcid, nfctx.user_data);
		}
	}

	return 0;
}

__public const char *libnvmf_get_default_trsvcid(const char *transport,
		bool discovery_ctrl)
{
	if (!transport)
		return NULL;
	if (!strcmp(transport, "tcp")) {
		if (discovery_ctrl)
			/* Default port for NVMe/TCP discovery controllers */
			return stringify(NVME_DISC_IP_PORT);
		/* Default port for NVMe/TCP io controllers */
		return stringify(NVME_RDMA_IP_PORT);
	} else if (!strcmp(transport, "rdma")) {
		/* Default port for NVMe/RDMA controllers */
		return stringify(NVME_RDMA_IP_PORT);
	}

	return NULL;
}

static bool is_persistent_discovery_ctrl(libnvme_host_t h, libnvme_ctrl_t c)
{
	if (libnvme_host_is_pdc_enabled(h, DEFAULT_PDC_ENABLED))
		return libnvme_ctrl_get_unique_discovery_ctrl(c);

	return false;
}

static int libnvme_add_ctrl(struct libnvmf_context *fctx,
		struct libnvme_host *h, struct libnvme_ctrl *c)
{
	int err;

retry:
	err = libnvmf_add_ctrl(h, c);
	if (!err)
		return 0;
	if (fctx->decide_retry(fctx, err, fctx->user_data))
		goto retry;

	return err;
}

static int __create_discovery_ctrl(struct libnvme_global_ctx *ctx,
		struct libnvmf_context *fctx, libnvme_host_t h,
		struct libnvme_ctrl **ctrl)
{
	libnvme_ctrl_t c;
	int tmo, ret;

	ret = _libnvme_create_ctrl(ctx, fctx, &c);
	if (ret)
		return ret;

	libnvme_ctrl_set_discovery_ctrl(c, true);
	libnvme_ctrl_set_unique_discovery_ctrl(c,
		     strcmp(fctx->subsysnqn, NVME_DISC_SUBSYS_NAME));
	tmo = set_discovery_kato(fctx);

	if (libnvme_ctrl_get_unique_discovery_ctrl(c) && fctx->hostkey) {
		libnvme_ctrl_set_dhchap_host_key(c, fctx->hostkey);
		if (fctx->ctrlkey)
			libnvme_ctrl_set_dhchap_ctrl_key(c, fctx->ctrlkey);
	}

	ret = libnvme_add_ctrl(fctx, h, c);
	fctx->cfg.keep_alive_tmo = tmo;
	if (ret) {
		libnvme_free_ctrl(c);
		return ret;
	}

	*ctrl = c;
	return 0;
}

static int nvmf_create_discovery_ctrl(struct libnvme_global_ctx *ctx,
		struct libnvmf_context *fctx, libnvme_host_t h,
		struct libnvme_ctrl **ctrl)
{
	__cleanup_free struct nvme_id_ctrl *id = NULL;
	struct libnvme_ctrl *c;
	int ret;

	ret = __create_discovery_ctrl(ctx, fctx, h, &c);
	if (ret)
		return ret;

	if (libnvme_ctrl_get_unique_discovery_ctrl(c)) {
		*ctrl = c;
		return 0;
	}

	id = __libnvme_alloc(sizeof(*id));
	if (!id) {
		libnvme_free_ctrl(c);
		return -ENOMEM;
	}

	ret = libnvme_open(ctx, c->name, &c->hdl);
	if (ret) {
		libnvme_msg(ctx, LIBNVME_LOG_ERR, "failed to open %s\n", c->name);
		return ret;
	}

	/* Find out the name of discovery controller */
	ret = libnvme_ctrl_identify(c, id);
	if (ret)  {
		libnvme_msg(ctx, LIBNVME_LOG_ERR,
			 "failed to identify controller, error %s\n",
			 libnvme_strerror(-ret));
		libnvmf_disconnect_ctrl(c);
		libnvme_free_ctrl(c);
		return ret;
	}

	if (!strcmp(id->subnqn, NVME_DISC_SUBSYS_NAME)) {
		*ctrl = c;
		return 0;
	}

	/*
	 * The subsysnqn is not the well-known name. Prefer the unique
	 * subsysnqn over the well-known one.
	 */
	libnvmf_disconnect_ctrl(c);
	libnvme_free_ctrl(c);

	fctx->subsysnqn = id->subnqn;
	ret = __create_discovery_ctrl(ctx, fctx, h, &c);
	if (ret)
		return ret;

	*ctrl = c;
	return 0;
}

int _discovery_config_json(struct libnvme_global_ctx *ctx,
		struct libnvmf_context *fctx, libnvme_host_t h, libnvme_ctrl_t c,
		bool connect, bool force)
{
	struct libnvmf_context nfctx = *fctx;
	libnvme_ctrl_t cn;
	int ret = 0;

	nfctx.transport = libnvme_ctrl_get_transport(c);
	nfctx.traddr = libnvme_ctrl_get_traddr(c);
	nfctx.host_traddr = libnvme_ctrl_get_host_traddr(c);
	nfctx.host_iface = libnvme_ctrl_get_host_iface(c);

	if (!nfctx.transport && !nfctx.traddr)
		return 0;

	/* ignore none fabric transports */
	if (strcmp(nfctx.transport, "tcp") &&
	    strcmp(nfctx.transport, "rdma") &&
	    strcmp(nfctx.transport, "fc"))
		return 0;

	/* ignore if no host_traddr for fc */
	if (!strcmp(nfctx.transport, "fc")) {
		if (!nfctx.host_traddr) {
			libnvme_msg(ctx, LIBNVME_LOG_ERR,
				 "host_traddr required for fc\n");
			return 0;
		}
	}

	/* ignore if host_iface set for any transport other than tcp */
	if (!strcmp(nfctx.transport, "rdma") ||
	    !strcmp(nfctx.transport, "fc")) {
		if (nfctx.host_iface) {
			libnvme_msg(ctx, LIBNVME_LOG_ERR,
				 "host_iface not permitted for rdma or fc\n");
			return 0;
		}
	}

	nfctx.trsvcid = libnvme_ctrl_get_trsvcid(c);
	if (!nfctx.trsvcid || !strcmp(nfctx.trsvcid, ""))
		nfctx.trsvcid =
			libnvmf_get_default_trsvcid(nfctx.transport, true);

	if (force)
		nfctx.subsysnqn = libnvme_ctrl_get_subsysnqn(c);
	else
		nfctx.subsysnqn = NVME_DISC_SUBSYS_NAME;

	if (libnvme_ctrl_get_persistent(c))
		nfctx.persistent = true;

	if (!force) {
		cn = lookup_ctrl(h, &nfctx);
		if (cn) {
			nfctx.persistent = true;
			_nvmf_discovery(ctx, &nfctx, connect, cn);
			return 0;
		}
	}

	ret = nvmf_create_discovery_ctrl(ctx, &nfctx, h, &cn);
	if (ret)
		return 0;

	_nvmf_discovery(ctx, &nfctx, connect, cn);
	if (!(fctx->persistent || is_persistent_discovery_ctrl(h, cn)))
		ret = libnvmf_disconnect_ctrl(cn);
	libnvme_free_ctrl(cn);

	return ret;
}

__public int libnvmf_discovery_config_json(struct libnvme_global_ctx *ctx,
		struct libnvmf_context *fctx, bool connect, bool force)
{
	const char *hnqn, *hid;
	struct libnvme_subsystem *s;
	struct libnvme_host *h;
	struct libnvme_ctrl *c;
	int ret = 0, err;

	err = lookup_host(ctx, fctx, &h);
	if (err)
		return err;

	err = setup_connection(fctx, h, false);
	if (err)
		return err;

	libnvme_for_each_host(ctx, h) {
		libnvme_for_each_subsystem(h, s) {
			hnqn = libnvme_host_get_hostnqn(h);
			if (fctx->hostnqn && hnqn &&
					strcmp(fctx->hostnqn, hnqn))
				continue;
			hid = libnvme_host_get_hostid(h);
			if (fctx->hostid && hid &&
					strcmp(fctx->hostid, hid))
				continue;

			libnvme_subsystem_for_each_ctrl(s, c) {
				err = _discovery_config_json(ctx, fctx, h, c,
					connect, force);
				if (err) {
					libnvme_msg(ctx, LIBNVME_LOG_ERR,
						"failed to connect to hostnqn=%s,nqn=%s,%s\n",
						libnvme_host_get_hostnqn(h),
						libnvme_subsystem_get_name(s),
						libnvme_ctrl_get_traddr(c));

					if (!ret)
						ret = err;
				}
			}
		}
	}

	return ret;
}

__public int libnvmf_connect_config_json(struct libnvme_global_ctx *ctx,
		struct libnvmf_context *fctx)
{
	const char *hnqn, *hid;
	const char *transport;
	libnvme_host_t h;
	libnvme_subsystem_t s;
	libnvme_ctrl_t c, _c;
	int ret = 0, err;

	err = lookup_host(ctx, fctx, &h);
	if (err)
		return err;

	err = setup_connection(fctx, h, false);
	if (err)
		return err;

	libnvme_for_each_host(ctx, h) {
		libnvme_for_each_subsystem(h, s) {
			hnqn = libnvme_host_get_hostnqn(h);
			if (fctx->hostnqn && hnqn &&
					strcmp(fctx->hostnqn, hnqn))
				continue;
			hid = libnvme_host_get_hostid(h);
			if (fctx->hostid && hid &&
					strcmp(fctx->hostid, hid))
				continue;

			libnvme_subsystem_for_each_ctrl_safe(s, c, _c) {
				transport = libnvme_ctrl_get_transport(c);

				/* ignore none fabric transports */
				if (strcmp(transport, "tcp") &&
				    strcmp(transport, "rdma") &&
				    strcmp(transport, "fc"))
					continue;

				err = libnvmf_connect_ctrl(c);
				if (err) {
					if (err == -ENVME_CONNECT_ALREADY)
						continue;

					libnvme_msg(ctx, LIBNVME_LOG_ERR,
						 "failed to connect to hostnqn=%s,nqn=%s,%s\n",
						 libnvme_host_get_hostnqn(h),
						 libnvme_subsystem_get_name(s),
						 libnvme_ctrl_get_traddr(c));

					if (!ret)
						ret = err;
				}
			}
		}
	}

	return ret;
}

__public int libnvmf_discovery_config_file(struct libnvme_global_ctx *ctx,
		struct libnvmf_context *fctx, bool connect, bool force)
{
	struct libnvme_host *h;
	struct libnvme_ctrl *c;
	int err;

	err = lookup_host(ctx, fctx, &h);
	if (err)
		return err;

	err = setup_connection(fctx, h, false);
	if (err)
		return err;

	err = fctx->parser_init(fctx, fctx->user_data);
	if (err)
		return err;

	do {
		err = fctx->parser_next_line(fctx, fctx->user_data);
		if (err)
			break;

		struct libnvmf_context nfctx = *fctx;

		if (!force) {
			c = lookup_ctrl(h, &nfctx);
			if (c) {
				_nvmf_discovery(ctx, &nfctx, connect, c);
				continue;
			}
		}

		err = nvmf_create_discovery_ctrl(ctx, &nfctx, h, &c);
		if (err)
			continue;

		_nvmf_discovery(ctx, &nfctx, connect, c);
		if (!(nfctx.persistent ||
		      is_persistent_discovery_ctrl(h, c)))
			err = libnvmf_disconnect_ctrl(c);
		libnvme_free_ctrl(c);
	} while (!err);

	fctx->parser_cleanup(fctx, fctx->user_data);

	if (err != -EOF)
		return err;

	return 0;
}

__public int libnvmf_config_modify(struct libnvme_global_ctx *ctx,
		struct libnvmf_context *fctx)
{
	__cleanup_free char *hnqn = NULL;
	__cleanup_free char *hid = NULL;
	struct libnvme_host *h;
	struct libnvme_subsystem *s;
	struct libnvme_ctrl *c;

	if (!fctx->hostnqn)
		fctx->hostnqn = hnqn = libnvme_read_hostnqn();
	if (!fctx->hostid && hnqn)
		fctx->hostid = hid = libnvme_read_hostid();

	h = libnvme_lookup_host(ctx, fctx->hostnqn, fctx->hostid);
	if (!h) {
		libnvme_msg(ctx, LIBNVME_LOG_ERR, "Failed to lookup host '%s'\n",
			fctx->hostnqn);
		return -ENODEV;
	}

	if (fctx->hostkey)
		libnvme_host_set_dhchap_host_key(h, fctx->hostkey);

	s = libnvme_lookup_subsystem(h, NULL, fctx->subsysnqn);
	if (!s) {
		libnvme_msg(ctx, LIBNVME_LOG_ERR, "Failed to lookup subsystem '%s'\n",
			fctx->subsysnqn);
		return -ENODEV;
	}

	c = libnvme_lookup_ctrl(s, fctx, NULL);
	if (!c) {
		libnvme_msg(ctx, LIBNVME_LOG_ERR, "Failed to lookup controller\n");
		return -ENODEV;
	}
	if (fctx->ctrlkey)
		libnvme_ctrl_set_dhchap_ctrl_key(c, fctx->ctrlkey);

	nvme_parse_tls_args(fctx->keyring, fctx->tls_key,
			    fctx->tls_key_identity, &fctx->cfg, c);

	update_config(c, &fctx->cfg);

	return 0;
}

#define NBFT_SYSFS_FILENAME	"NBFT*"

static int nbft_filter(const struct dirent *dent)
{
	return !fnmatch(NBFT_SYSFS_FILENAME, dent->d_name, FNM_PATHNAME);
}

__public int libnvmf_nbft_read_files(struct libnvme_global_ctx *ctx, char *path,
		struct nbft_file_entry **head)
{
	struct nbft_file_entry *entry = NULL;
	struct libnbft_info *nbft;
	struct dirent **dent;
	char filename[PATH_MAX];
	int i, count, ret;

	count = scandir(path, &dent, nbft_filter, NULL);
	if (count < 0)
		return -errno;

	for (i = 0; i < count; i++) {
		snprintf(filename, sizeof(filename), "%s/%s", path,
			dent[i]->d_name);

		ret = libnvme_read_nbft(ctx, &nbft, filename);
		if (!ret) {
			struct nbft_file_entry *new;

			new = calloc(1, sizeof(*new));
			if (!new)
				return -ENOMEM;
			new->nbft = nbft;
			if (entry) {
				entry->next = new;
				entry = entry->next;
			} else {
				entry = new;
				*head = entry;
			}
		}
		free(dent[i]);
	}
	free(dent);
	return 0;
}

__public void libnvmf_nbft_free(struct libnvme_global_ctx *ctx, struct nbft_file_entry *head)
{
	if (!head)
		return;

	while (head) {
		struct nbft_file_entry *next = head->next;

		libnvme_free_nbft(ctx, head->nbft);
		free(head);

		head = next;
	}
}

static bool validate_uri(struct libnvme_global_ctx *ctx,
			 struct libnbft_discovery *dd,
			 struct libnvmf_uri *uri)
{
	if (!uri) {
		libnvme_msg(ctx, LIBNVME_LOG_ERR,
			 "Discovery Descriptor %d: failed to parse URI %s\n",
			 dd->index, dd->uri);
		return false;
	}
	if (strcmp(uri->scheme, "nvme") != 0) {
		libnvme_msg(ctx, LIBNVME_LOG_ERR,
			 "Discovery Descriptor %d: unsupported scheme '%s'\n",
			 dd->index, uri->scheme);
		return false;
	}
	if (!uri->protocol || strcmp(uri->protocol, "tcp") != 0) {
		libnvme_msg(ctx, LIBNVME_LOG_ERR,
			 "Discovery Descriptor %d: unsupported transport '%s'\n",
			 dd->index, uri->protocol);
		return false;
	}

	return true;
}

static int nbft_connect(struct libnvme_global_ctx *ctx,
		struct libnvmf_context *fctx, struct libnvme_host *h,
		struct nvmf_disc_log_entry *e,
		struct libnbft_subsystem_ns *ss)
{
	libnvme_ctrl_t c;
	int saved_log_level;
	bool saved_log_tstamp;
	bool saved_log_pid;
	int ret;

	saved_log_level = libnvme_get_logging_level(ctx, &saved_log_tstamp,
		&saved_log_pid);

	c = lookup_ctrl(h, fctx);
	if (c && libnvme_ctrl_get_name(c))
		return 0;

	ret = _libnvme_create_ctrl(ctx, fctx, &c);
	if (ret)
		return ret;

	/* Pause logging for unavailable SSNSs */
	if (ss && ss->unavailable && saved_log_level < 1)
		libnvme_set_logging_level(ctx, -1, false, false);

	/* Update tls or concat */
	nvmf_update_tls_concat(e, c, h);

	ret = libnvmf_add_ctrl(h, c);

	/* Resume logging */
	if (ss && ss->unavailable && saved_log_level < 1)
		libnvme_set_logging_level(ctx,
				  saved_log_level,
				  saved_log_pid,
				  saved_log_tstamp);

	if (ret) {
		libnvme_free_ctrl(c);
		/*
		 * In case this SSNS was marked as 'unavailable' and
		 * our connection attempt has failed, ignore it.
		 */
		if (ss && ss->unavailable) {
			libnvme_msg(ctx, LIBNVME_LOG_INFO,
				"SSNS %d reported as unavailable, skipping\n",
				ss->index);
			return 0;
		}
		return ret;
	}

	if (fctx->connected)
		fctx->connected(fctx, c, fctx->user_data);

	return 0;
}

static int nbft_discovery(struct libnvme_global_ctx *ctx,
		struct libnvmf_context *fctx, struct libnbft_discovery *dd,
		struct libnvme_host *h, struct libnvme_ctrl *c)
{
	struct nvmf_discovery_log *log = NULL;
	int ret;
	int i;

	struct libnvmf_discovery_args args = {
		.max_retries = 10 /* MAX_DISC_RETRIES */,
		.lsp = NVMF_LOG_DISC_LSP_NONE,
	};

	ret = nvme_discovery_log(c, &args, &log);
	if (ret) {
		libnvme_msg(ctx, LIBNVME_LOG_ERR,
			"Discovery Descriptor %d: failed to get discovery log: %s\n",
			dd->index, libnvme_strerror(ret));
		return ret;
	}

	for (i = 0; i < le64_to_cpu(log->numrec); i++) {
		struct nvmf_disc_log_entry *e = &log->entries[i];
		struct libnvmf_context nfctx = *fctx;
		libnvme_ctrl_t cl;
		int tmo = fctx->cfg.keep_alive_tmo;

		sanitize_discovery_log_entry(c->ctx, e);

		nfctx.subsysnqn = e->subnqn;
		nfctx.transport = libnvmf_trtype_str(e->trtype);
		nfctx.traddr = e->traddr;
		nfctx.trsvcid = e->trsvcid;

		if (e->subtype == NVME_NQN_CURR)
			continue;

		/* Already connected ? */
		cl = lookup_ctrl(h, &nfctx);
		if (cl && libnvme_ctrl_get_name(cl))
			continue;

		/* Skip connect if the transport types don't match */
		if (strcmp(libnvme_ctrl_get_transport(c),
			   nfctx.transport))
			continue;

		if (e->subtype == NVME_NQN_DISC) {
			libnvme_ctrl_t child;

			ret = nvmf_connect_disc_entry(h, e, &nfctx,
				NULL, &child);
			if (ret)
				continue;
			nbft_discovery(ctx, &nfctx, dd, h, child);
			libnvmf_disconnect_ctrl(child);
			libnvme_free_ctrl(child);
		} else {
			ret = nbft_connect(ctx, &nfctx, h, e, NULL);

			/*
			 * With TCP/DHCP, it can happen that the OS
			 * obtains a different local IP address than the
			 * firmware had. Retry without host_traddr.
			 */
			if (ret == -ENVME_CONNECT_ADDRNOTAVAIL &&
			    !strcmp(nfctx.transport, "tcp") &&
			    strlen(dd->hfi->tcp_info.dhcp_server_ipaddr) > 0) {
				const char *htradr = nfctx.host_traddr;

				nfctx.host_traddr = NULL;
				ret = nbft_connect(ctx, &nfctx, h, e, NULL);

				if (ret == 0)
					libnvme_msg(ctx, LIBNVME_LOG_INFO,
						"Discovery Descriptor %d: connect with host_traddr=\"%s\" failed, success after omitting host_traddr\n",
						dd->index,
						htradr);
			}

			if (ret)
				libnvme_msg(ctx, LIBNVME_LOG_ERR,
					"Discovery Descriptor %d: no controller found\n",
					dd->index);
			if (ret == -ENOMEM)
				break;
		}

		fctx->cfg.keep_alive_tmo = tmo;
	}

	free(log);
	return 0;
}

__public int libnvmf_discovery_nbft(struct libnvme_global_ctx *ctx,
		struct libnvmf_context *fctx, bool connect, char *nbft_path)
{
	const char *hostnqn = NULL, *hostid = NULL, *host_traddr = NULL;
	char uuid[NVME_UUID_LEN_STRING];
	struct nbft_file_entry *entry = NULL;
	struct libnbft_subsystem_ns **ss;
	struct libnbft_hfi *hfi;
	struct libnbft_discovery **dd;
	struct libnvme_host *h;
	int ret, rr, i;

	ret = lookup_host(ctx, fctx, &h);
	if (ret)
		return ret;

	ret = setup_connection(fctx, h, false);
	if (ret)
		return ret;

	if (!connect)
		/* TODO: print discovery-type info from NBFT tables */
		return 0;

	ret = libnvmf_nbft_read_files(ctx, nbft_path, &entry);
	if (ret) {
		if (ret != -ENOENT)
			libnvme_msg(ctx, LIBNVME_LOG_ERR,
				"Failed to access ACPI tables directory\n");
		else
			ret = 0;  /* nothing to connect */
		goto out_free;
	}

	for (; entry; entry = entry->next) {
		if (fctx->hostnqn)
			hostnqn = fctx->hostnqn;
		else {
			hostnqn = entry->nbft->host.nqn;
			if (!hostnqn)
				hostnqn = fctx->hostnqn;
		}

		if (fctx->hostid)
			hostid = fctx->hostid;
		else if (*entry->nbft->host.id) {
			ret = libnvme_uuid_to_string(entry->nbft->host.id, uuid);
			if (!ret)
				hostid = uuid;
			else
				hostid = fctx->hostid;
		}

		h = libnvme_lookup_host(ctx, hostnqn, hostid);
		if (!h) {
			ret = -ENOENT;
			goto out_free;
		}

		/* Subsystem Namespace Descriptor List */
		for (ss = entry->nbft->subsystem_ns_list; ss && *ss; ss++)
			for (i = 0; i < (*ss)->num_hfis; i++) {
				struct libnvmf_context nfctx = *fctx;

				hfi = (*ss)->hfis[i];

				/* Skip discovery NQN records */
				if (strcmp((*ss)->subsys_nqn,
						NVME_DISC_SUBSYS_NAME) == 0) {
					libnvme_msg(ctx, LIBNVME_LOG_INFO,
						"SSNS %d points to well-known discovery NQN, skipping\n",
						(*ss)->index);
					continue;
				}

				nfctx.host_traddr = NULL;
				if (!fctx->host_traddr &&
				    !strncmp((*ss)->transport, "tcp", 3))
					nfctx.host_traddr =
						hfi->tcp_info.ipaddr;

				nfctx.subsysnqn = (*ss)->subsys_nqn;
				nfctx.transport = (*ss)->transport;
				nfctx.traddr = (*ss)->traddr;
				nfctx.trsvcid = (*ss)->trsvcid;
				nfctx.host_iface = NULL;

				rr = nbft_connect(ctx, &nfctx, h, NULL, *ss);

				/*
				 * With TCP/DHCP, it can happen that the OS
				 * obtains a different local IP address than the
				 * firmware had. Retry without host_traddr.
				 */
				if (rr == -ENVME_CONNECT_ADDRNOTAVAIL &&
				    !strcmp(nfctx.transport, "tcp") &&
				    strlen(hfi->tcp_info.dhcp_server_ipaddr) > 0) {
					nfctx.host_traddr = NULL;

					rr = nbft_connect(ctx, &nfctx, h, NULL,
						*ss);

					if (rr == 0)
						libnvme_msg(ctx, LIBNVME_LOG_INFO,
							"SSNS %d: connect with host_traddr=\"%s\" failed, success after omitting host_traddr\n",
							(*ss)->index,
							host_traddr);
				}

				if (rr) {
					libnvme_msg(ctx, LIBNVME_LOG_ERR,
						"SSNS %d: no controller found\n",
						(*ss)->index);
					/* report an error */
					ret = rr;
				}

				if (rr == -ENOMEM)
					goto out_free;
			}

		/* Discovery Descriptor List */
		for (dd = entry->nbft->discovery_list; dd && *dd; dd++) {
			__cleanup_uri struct libnvmf_uri *uri = NULL;
			__cleanup_free char *trsvcid = NULL;
			struct libnvmf_context nfctx = *fctx;
			bool persistent = false;
			bool linked = false;
			libnvme_ctrl_t c;

			/* only perform discovery when no SSNS record references it */
			for (ss = entry->nbft->subsystem_ns_list;
					ss && *ss; ss++)
				if ((*ss)->discovery &&
				    (*ss)->discovery->index == (*dd)->index &&
				    /* unavailable boot attempts are not discovered
				     * and may get transferred along with a well-known
				     * discovery NQN into an SSNS record.
				     */
				    strcmp((*ss)->subsys_nqn,
						NVME_DISC_SUBSYS_NAME) != 0) {
					linked = true;
					break;
				}
			if (linked)
				continue;

			hfi = (*dd)->hfi;
			ret = libnvmf_uri_parse((*dd)->uri, &uri);
			if (ret)
				continue;
			if (!validate_uri(ctx, *dd, uri))
				continue;

			host_traddr = NULL;
			if (!fctx->host_traddr &&
			    !strncmp(uri->protocol, "tcp", 3))
				host_traddr = hfi->tcp_info.ipaddr;
			if (uri->port > 0) {
				if (asprintf(&trsvcid, "%d", uri->port) < 0) {
					ret = -ENOMEM;
					goto out_free;
				}
			} else
				trsvcid =
					strdup(libnvmf_get_default_trsvcid(
						uri->protocol, true));

			nfctx.subsysnqn = NVME_DISC_SUBSYS_NAME;
			nfctx.transport =  uri->protocol;
			nfctx.traddr = uri->host;
			nfctx.trsvcid = trsvcid;
			nfctx.host_traddr = host_traddr;
			nfctx.host_iface = NULL;

			/* Lookup existing discovery controller */
			c = lookup_ctrl(h, &nfctx);
			if (c && libnvme_ctrl_get_name(c))
				persistent = true;

			if (!c) {
				ret = nvmf_create_discovery_ctrl(ctx, &nfctx,
					h, &c);
				if (ret == -ENVME_CONNECT_ADDRNOTAVAIL &&
				    !strcmp(nfctx.transport, "tcp") &&
				    strlen(hfi->tcp_info.dhcp_server_ipaddr) > 0) {
					nfctx.traddr = NULL;
					ret = nvmf_create_discovery_ctrl(ctx,
						&nfctx, h, &c);
				}
			} else
				ret = 0;

			if (ret) {
				libnvme_msg(ctx, LIBNVME_LOG_ERR,
					"Discovery Descriptor %d: failed to add discovery controller: %s\n",
					(*dd)->index, libnvme_strerror(-ret));
				goto out_free;
			}

			rr = nbft_discovery(ctx, &nfctx, *dd, h, c);
			if (!persistent)
				libnvmf_disconnect_ctrl(c);
			libnvme_free_ctrl(c);
			if (rr == -ENOMEM) {
				ret = rr;
				goto out_free;
			}
		}
	}
out_free:
	libnvmf_nbft_free(ctx, entry);
	return ret;
}

__public int libnvmf_discovery(struct libnvme_global_ctx *ctx, struct libnvmf_context *fctx,
		bool connect, bool force)
{
	struct libnvme_ctrl *c = NULL;
	struct libnvme_host *h;
	int ret;

	ret = lookup_host(ctx, fctx, &h);
	if (ret)
		return ret;

	ret = setup_connection(fctx, h, true);
	if (ret)
		return ret;

	if (fctx->device && !force) {
		ret = libnvme_scan_ctrl(ctx, fctx->device, &c);
		if (!ret) {
			/* Check if device matches command-line options */
			if (!_libnvme_ctrl_match_config(c, fctx)) {
				libnvme_msg(ctx, LIBNVME_LOG_ERR,
				    "ctrl device %s found, ignoring non matching command-line options\n",
				    fctx->device);
			}

			if (!libnvme_ctrl_get_discovery_ctrl(c)) {
				libnvme_msg(
					ctx, LIBNVME_LOG_ERR,
					"ctrl device %s found, ignoring non discovery controller\n",
					fctx->device);

				libnvme_free_ctrl(c);
				c = NULL;
				fctx->persistent = false;
			} else {
				/*
				 * If the controller device is found it must
				 * be persistent, and shouldn't be disconnected
				 * on exit.
				 */
				fctx->persistent = true;
				/*
				 * When --host-traddr/--host-iface are not specified on the
				 * command line, use the discovery controller's (c) host-
				 * traddr/host-iface for the connections to controllers
				 * returned in the Discovery Log Pages. This is essential
				 * when invoking "connect-all" with --device to reuse an
				 * existing persistent discovery controller (as is done
				 * for the udev rules). This ensures that host-traddr/
				 * host-iface are consistent with the discovery controller (c).
				 */
				if (!fctx->host_traddr)
					fctx->host_traddr = (char *)
						libnvme_ctrl_get_host_traddr(c);
				if (!fctx->host_iface)
					fctx->host_iface = (char *)
						libnvme_ctrl_get_host_iface(c);
			}
		} else {
			/*
			 * No controller found, fall back to create one.
			 * But that controller cannot be persistent.
			 */
			libnvme_msg(ctx, LIBNVME_LOG_ERR,
				"ctrl device %s not found%s\n", fctx->device,
				fctx->persistent ? ", ignoring --persistent" : "");
			fctx->persistent = false;
		}
	}

	if (!c && !force) {
		c = lookup_ctrl(h, fctx);
		if (c)
			fctx->persistent = true;
	}
	if (!c) {
		/* No device or non-matching device, create a new controller */
		ret = nvmf_create_discovery_ctrl(ctx, fctx, h, &c);
		if (ret) {
			if (ret != -ENVME_CONNECT_IGNORED)
				libnvme_msg(ctx, LIBNVME_LOG_ERR,
					 "failed to add controller, error %s\n",
					 libnvme_strerror(-ret));
			return ret;
		}
	}

	ret = _nvmf_discovery(ctx, fctx, connect, c);
	if (!(fctx->persistent || is_persistent_discovery_ctrl(h, c)))
		libnvmf_disconnect_ctrl(c);
	libnvme_free_ctrl(c);

	return ret;
}

__public int libnvmf_connect(struct libnvme_global_ctx *ctx, struct libnvmf_context *fctx)
{
	struct libnvme_host *h;
	struct libnvme_ctrl *c;
	int err;

	err = lookup_host(ctx, fctx, &h);
	if (err)
		return err;

	err = setup_connection(fctx, h, false);
	if (err)
		return err;

	c = lookup_ctrl(h, fctx);
	if (c && libnvme_ctrl_get_name(c) && !fctx->cfg.duplicate_connect) {
		fctx->already_connected(fctx, h, libnvme_ctrl_get_subsysnqn(c),
			libnvme_ctrl_get_transport(c), libnvme_ctrl_get_traddr(c),
			libnvme_ctrl_get_trsvcid(c), fctx->user_data);
		return -EALREADY;
	}

	err = _libnvme_create_ctrl(ctx, fctx, &c);
	if (err)
		return err;

	if (fctx->hostkey) {
		libnvme_ctrl_set_dhchap_host_key(c, fctx->hostkey);
		if (fctx->ctrlkey)
			libnvme_ctrl_set_dhchap_ctrl_key(c, fctx->ctrlkey);
	}

	nvme_parse_tls_args(fctx->keyring, fctx->tls_key,
		fctx->tls_key_identity, &fctx->cfg, c);

	/*
	 * We are connecting to a discovery controller, so let's treat
	 * this as a persistent connection and specify a KATO.
	 */
	if (!strcmp(fctx->subsysnqn, NVME_DISC_SUBSYS_NAME)) {
		fctx->persistent = true;

		set_discovery_kato(fctx);
	}

	err = libnvme_add_ctrl(fctx, h, c);
	if (err) {
		libnvme_msg(ctx, LIBNVME_LOG_ERR, "could not add new controller: %s\n",
			libnvme_strerror(-err));
		libnvme_free_ctrl(c);
		return err;
	}

	fctx->connected(fctx, c, fctx->user_data);

	return 0;
}