// SPDX-License-Identifier: GPL-2.0-or-later
/*
 * MGMTD Frontend Client Connection Adapter
 *
 * Copyright (C) 2021  Vmware, Inc.
 *		       Pushpasis Sarkar <spushpasis@vmware.com>
 * Copyright (c) 2023, LabN Consulting, L.L.C.
 */

#include <zebra.h>
#include "darr.h"
#include "frrstr.h"
#include "sockopt.h"
#include "network.h"
#include "libfrr.h"
#include "mgmt_fe_client.h"
#include "mgmt_msg.h"
#include "mgmt_msg_native.h"
#include "hash.h"
#include "jhash.h"
#include "mgmtd/mgmt.h"
#include "mgmtd/mgmt_ds.h"
#include "mgmtd/mgmt_memory.h"
#include "mgmtd/mgmt_fe_adapter.h"

#define _dbg(fmt, ...)	   DEBUGD(&mgmt_debug_fe, "FE-ADAPTER: %s: " fmt, __func__, ##__VA_ARGS__)
#define _log_err(fmt, ...) zlog_err("FE-ADAPTER: %s: ERROR: " fmt, __func__, ##__VA_ARGS__)

enum mgmt_session_event {
	MGMTD_FE_SESSION_CFG_TXN_CLNUP = 1,
	MGMTD_FE_SESSION_SHOW_TXN_CLNUP,
};

struct mgmt_fe_session_ctx {
	struct mgmt_fe_client_adapter *adapter;
	uint64_t session_id;
	uint64_t client_id;
	uint64_t txn_id;
	uint64_t cfg_txn_id;
	uint8_t notify_format;
	uint8_t ds_locked[MGMTD_DS_MAX_ID];
	const char **notify_xpaths;
	struct event *proc_cfg_txn_clnp;
	struct event *proc_show_txn_clnp;

	LIST_ENTRY(mgmt_fe_session_ctx) link;
};

struct mgmt_fe_client_adapter {
	struct msg_conn *conn;
	char name[MGMTD_CLIENT_NAME_MAX_LEN];

	LIST_ENTRY(mgmt_fe_client_adapter) link;

	/* List of sessions created and being maintained for this client. */
	LIST_HEAD(fe_session_list_head, mgmt_fe_session_ctx) sessions;

	/* NOTE: shared by all sessions, only works b/c one session configuring at a time */
	struct mgmt_commit_stats cmt_stats;
};

/*
 * A tree for storing unique notify-select strings.
 */
PREDECL_RBTREE_UNIQ(ns_string);
struct ns_string {
	struct ns_string_item link;
	struct list *sessions;
	char s[];
};
static uint32_t ns_string_compare(const struct ns_string *ns1, const struct ns_string *ns2);
DECLARE_RBTREE_UNIQ(ns_string, struct ns_string, link, ns_string_compare);

/* ---------- */
/* Prototypes */
/* ---------- */

static struct msg_conn *fe_adapter_create(int conn_fd, union sockunion *from);
static void fe_session_compute_commit_timers(struct mgmt_commit_stats *cmt_stats);

/* ---------------- */
/* Global variables */
/* ---------------- */

static struct event_loop *mgmt_loop;
static struct msg_server mgmt_fe_server = {.fd = -1};

LIST_HEAD(fe_adapter_list_head, mgmt_fe_client_adapter) fe_adapters;

static struct hash *mgmt_fe_sessions;
static uint64_t mgmt_fe_next_session_id;

static struct ns_string_head mgmt_fe_ns_strings;

/* ------------------------------ */
/* Notify select string functions */
/* ------------------------------ */

static uint32_t ns_string_compare(const struct ns_string *ns1, const struct ns_string *ns2)
{
	return strcmp(ns1->s, ns2->s);
}

static void mgmt_fe_free_ns_string(struct ns_string *ns)
{
	list_delete(&ns->sessions);
	XFREE(MTYPE_MGMTD_XPATH, ns);
}

static void mgmt_fe_free_ns_strings(struct ns_string_head *head)
{
	struct ns_string *ns;

	while ((ns = ns_string_pop(head)))
		mgmt_fe_free_ns_string(ns);
	ns_string_fini(head);
}

static uint64_t mgmt_fe_ns_string_remove_session(struct ns_string_head *head,
						 struct mgmt_fe_session_ctx *session)
{
	struct listnode *node;
	struct ns_string *ns;
	uint64_t clients = 0;

	frr_each_safe (ns_string, head, ns) {
		node = listnode_lookup(ns->sessions, session);
		if (!node)
			continue;
		list_delete_node(ns->sessions, node);
		if (list_isempty(ns->sessions)) {
			_dbg("do not notify session-id: %Lu on %s", session->session_id, ns->s);
			clients |= mgmt_be_interested_clients(ns->s, MGMT_BE_XPATH_SUBSCR_TYPE_OPER,
							      "add-notify-select");
			ns_string_del(head, ns);
			mgmt_fe_free_ns_string(ns);
		}
	}

	return clients;
}

static uint64_t mgmt_fe_add_ns_string(struct ns_string_head *head, const char *path, size_t plen,
				      struct mgmt_fe_session_ctx *session, uint64_t *all_matched)
{
	struct ns_string *e, *ns;
	uint64_t clients;

	ns = XCALLOC(MTYPE_MGMTD_XPATH, sizeof(*ns) + plen + 1);
	strlcpy(ns->s, path, plen + 1);

	_dbg("notify session-id: %Lu on %s", session->session_id, ns->s);
	clients = mgmt_be_interested_clients(ns->s, MGMT_BE_XPATH_SUBSCR_TYPE_OPER,
					     "add-notify-select");
	*all_matched |= clients;

	e = ns_string_add(head, ns);
	if (!e) {
		ns->sessions = list_new();
		listnode_add(ns->sessions, session);
	} else {
		clients = 0;
		XFREE(MTYPE_MGMTD_XPATH, ns);
		if (!listnode_lookup(e->sessions, session))
			listnode_add(e->sessions, session);
	}

	return clients;
}

char **mgmt_fe_get_all_selectors(void)
{
	char **selectors = NULL;
	struct ns_string *ns;

	frr_each (ns_string, &mgmt_fe_ns_strings, ns)
		*darr_append(selectors) = darr_strdup(ns->s);

	return selectors;
}

enum mgmt_result nb_error_to_mgmt_result(enum nb_error error)
{
	switch (error) {
	case NB_OK:
		return MGMTD_SUCCESS;
	case NB_ERR:
		return MGMTD_INTERNAL_ERROR;
	case NB_ERR_NO_CHANGES:
		return MGMTD_NO_CFG_CHANGES;
	case NB_ERR_NOT_FOUND:
		return MGMTD_INVALID_PARAM;
	case NB_ERR_EXISTS:
		return MGMTD_VALUE_EXISTS;
	case NB_ERR_LOCKED:
		return MGMTD_DS_LOCK_FAILED;
	case NB_ERR_VALIDATION:
		return MGMTD_VALIDATION_ERROR;
	case NB_ERR_RESOURCE:
		return MGMTD_INTERNAL_ERROR;
	case NB_ERR_INCONSISTENCY:
		return MGMTD_INTERNAL_ERROR;
	case NB_YIELD:
		return MGMTD_UNKNOWN_FAILURE;
	}
	return MGMTD_UNKNOWN_FAILURE;
}

int mgmt_result_to_error(enum mgmt_result result)
{
	switch (result) {
	case MGMTD_SUCCESS:
		return 0;
	case MGMTD_INVALID_PARAM:
		return -EBADMSG;
	case MGMTD_INTERNAL_ERROR:
	case MGMTD_DS_UNLOCK_FAILED:
	case MGMTD_UNKNOWN_FAILURE:
		return -EFAULT;
	case MGMTD_VALIDATION_ERROR:
		return -EINVAL;
	case MGMTD_NO_CFG_CHANGES:
		return -EALREADY;
	case MGMTD_VALUE_EXISTS:
		return -EEXIST;
	case MGMTD_DS_LOCK_FAILED:
		return -EBUSY;
	}
	return -EFAULT;
}

/* =========================== */
/* Frontend Session Management */
/* =========================== */

static int mgmt_fe_session_write_lock_ds(enum mgmt_ds_id ds_id, struct mgmt_ds_ctx *ds_ctx,
					 struct mgmt_fe_session_ctx *session)
{
	if (session->ds_locked[ds_id])
		zlog_warn("multiple lock taken by session-id: %" PRIu64
			  " on DS:%s",
			  session->session_id, mgmt_ds_id2name(ds_id));
	else {
		if (mgmt_ds_lock(ds_ctx, session->session_id)) {
			_log_err("Failed to lock the DS:%s for session-id: %" PRIu64 " from %s!",
				 mgmt_ds_id2name(ds_id), session->session_id,
				 session->adapter->name);
			return -1;
		}

		session->ds_locked[ds_id] = true;
		_dbg("Write-Locked the DS:%s for session-id: %" PRIu64 " from %s",
		     mgmt_ds_id2name(ds_id), session->session_id, session->adapter->name);
	}

	return 0;
}

static void mgmt_fe_session_unlock_ds(enum mgmt_ds_id ds_id, struct mgmt_ds_ctx *ds_ctx,
				      struct mgmt_fe_session_ctx *session)
{
	if (!session->ds_locked[ds_id])
		zlog_warn("unlock unlocked by session-id: %" PRIu64 " on DS:%s",
			  session->session_id, mgmt_ds_id2name(ds_id));

	session->ds_locked[ds_id] = false;
	mgmt_ds_unlock(ds_ctx, session->session_id);
	_dbg("Unlocked DS:%s write-locked earlier by session-id: %" PRIu64 " from %s",
	     mgmt_ds_id2name(ds_id), session->session_id, session->adapter->name);
}

static struct mgmt_fe_session_ctx *
fe_session_lookup_by_client_id(struct mgmt_fe_client_adapter *adapter, uint64_t client_id)
{
	struct mgmt_fe_session_ctx *session;

	LIST_FOREACH (session, &adapter->sessions, link) {
		if (session->client_id == client_id) {
			_dbg("Found session-id %" PRIu64 " using client-id %" PRIu64,
			     session->session_id, client_id);
			return session;
		}
	}
	_dbg("Session not found using client-id %" PRIu64, client_id);
	return NULL;
}

static unsigned int fe_session_hash_key(const void *data)
{
	const struct mgmt_fe_session_ctx *session = data;

	return jhash2((uint32_t *) &session->session_id,
		      sizeof(session->session_id) / sizeof(uint32_t), 0);
}

static bool fe_session_hash_cmp(const void *d1, const void *d2)
{
	const struct mgmt_fe_session_ctx *session1 = d1;
	const struct mgmt_fe_session_ctx *session2 = d2;

	return (session1->session_id == session2->session_id);
}

static inline struct mgmt_fe_session_ctx *fe_session_lookup(uint64_t session_id)
{
	struct mgmt_fe_session_ctx key = {0};
	struct mgmt_fe_session_ctx *session;

	if (session_id == MGMTD_SESSION_ID_NONE)
		return NULL;

	if (!mgmt_fe_sessions)
		return NULL;

	key.session_id = session_id;
	session = hash_lookup(mgmt_fe_sessions, &key);

	return session;
}

static struct mgmt_fe_session_ctx *fe_session_by_txn_id(uint64_t txn_id)
{
	uint64_t session_id = mgmt_txn_get_session_id(txn_id);

	if (session_id == MGMTD_SESSION_ID_NONE)
		return NULL;
	return fe_session_lookup(session_id);
}

static void fe_session_cleanup(struct mgmt_fe_session_ctx **sessionp)
{
	enum mgmt_ds_id ds_id;
	struct mgmt_ds_ctx *ds_ctx;
	struct mgmt_fe_session_ctx *session = *sessionp;

	/* XXXchopps what about RPC txns? */
	mgmt_destroy_txn(&session->cfg_txn_id);
	mgmt_destroy_txn(&session->txn_id);
	for (ds_id = 0; ds_id < MGMTD_DS_MAX_ID; ds_id++) {
		ds_ctx = mgmt_ds_get_ctx_by_id(mm, ds_id);
		if (ds_ctx && session->ds_locked[ds_id])
			mgmt_fe_session_unlock_ds(ds_id, ds_ctx, session);
	}

	LIST_REMOVE(session, link);

	mgmt_fe_ns_string_remove_session(&mgmt_fe_ns_strings, session);
	darr_free_free(session->notify_xpaths);
	hash_release(mgmt_fe_sessions, session);
	XFREE(MTYPE_MGMTD_FE_SESSION, session);
	*sessionp = NULL;
}

static struct mgmt_fe_session_ctx *fe_session_create(struct mgmt_fe_client_adapter *adapter,
						     uint8_t notify_format, uint64_t client_id)
{
	struct mgmt_fe_session_ctx *session;

	session = fe_session_lookup_by_client_id(adapter, client_id);
	if (session)
		fe_session_cleanup(&session);

	session = XCALLOC(MTYPE_MGMTD_FE_SESSION,
			sizeof(struct mgmt_fe_session_ctx));
	assert(session);
	session->client_id = client_id;
	session->adapter = adapter;
	session->notify_format = notify_format;
	session->txn_id = MGMTD_TXN_ID_NONE;
	session->cfg_txn_id = MGMTD_TXN_ID_NONE;
	LIST_INSERT_HEAD(&adapter->sessions, session, link);
	if (!mgmt_fe_next_session_id)
		mgmt_fe_next_session_id++;
	session->session_id = mgmt_fe_next_session_id++;
	hash_get(mgmt_fe_sessions, session, hash_alloc_intern);

	return session;
}


/* =============================== */
/* Frontend Message (API) Handling */
/* =============================== */

/*
 * Code structure: 3 functions per message type:
 *
 * 1) send reply back to FE client
 * 2) process result from Txn module to send back to FE client, call (1)
 * 3) handle request from FE client, either create TXN or call (1)
 *
 * For message types that don't require a transaction then (2) is elided.
 *
 * Txn's are used by any message that requires a fan-out to multiple backends:
 *
 * 1) FE handler creates Txn and calls Txn module to process request
 * 2) Txn module creates Txn request determines backends to contact and sends
 *    them each a message.
 * 3) Each backend client responds and the BE adapter calls TXN module
 * 4) Txn module aggregates results and calls FE reply function to process
 *    send results back to FE client.
 *
 *                         .-> (BE Daemon) -> [BE adapter/Txn Handler]
 *        (1)         (2) /                 o      (3)      \\   (4)
 *   [FE handler] => [TXN] ...              o                [TXN] => [FE reply]
 *     session            \                 o               //
 *                         `-> (BE Daemon) -> [BE adapter/Txn Handler]
 */

static int fe_adapter_send_msg(struct mgmt_fe_client_adapter *adapter, void *msg, size_t len,
			       bool short_circuit_ok)
{
	return msg_conn_send_msg(adapter->conn, MGMT_MSG_VERSION_NATIVE, msg, len, NULL,
				 short_circuit_ok);
}

static int fe_session_send_error(struct mgmt_fe_session_ctx *session, uint64_t req_id,
				 bool short_circuit_ok, int16_t error, const char *errfmt, ...)
	PRINTFRR(5, 6);

static int fe_session_send_error(struct mgmt_fe_session_ctx *session, uint64_t req_id,
				 bool short_circuit_ok, int16_t error, const char *errfmt, ...)
{
	va_list ap;
	int ret;

	va_start(ap, errfmt);
	ret = vmgmt_msg_native_send_error(session->adapter->conn,
					  session->session_id, req_id,
					  short_circuit_ok, error, errfmt, ap);
	va_end(ap);

	return ret;
}

/*
 * Transaction (txn) Error - send an error back to the FE client and cleanup any
 * in-progress txn.
 *
 * XXXchopps this is used in 2 places for GET-DATA and RPC errors from TXN
 * module just adapt those 2 normal reply fucntions to take an error result and
 * get rid of this function.
 */
int mgmt_fe_adapter_txn_error(uint64_t txn_id, uint64_t req_id, bool short_circuit_ok,
			      int16_t error, const char *errstr)
{
	struct mgmt_fe_session_ctx *session;
	int ret;

	session = fe_session_by_txn_id(txn_id);
	if (!session) {
		_log_err("failed sending error for txn-id %" PRIu64 " session not found", txn_id);
		return -ENOENT;
	}

	ret = fe_session_send_error(session, req_id, false, error, "%s", errstr);

	mgmt_destroy_txn(&session->txn_id);

	return ret;
}


/* -------------- */
/* COMMIT Message */
/* -------------- */

static void fe_session_send_commit_reply(struct mgmt_fe_session_ctx *session, uint64_t req_id,
					 uint8_t source, uint8_t target, uint8_t action,
					 bool unlock)
{
	struct mgmt_msg_commit_reply *msg;
	int ret;

	msg = mgmt_msg_native_alloc_msg(struct mgmt_msg_commit_reply, 0,
					MTYPE_MSG_NATIVE_COMMIT_REPLY);
	msg->refer_id = session->session_id;
	msg->req_id = req_id;
	msg->code = MGMT_MSG_CODE_COMMIT_REPLY;
	msg->source = source;
	msg->target = target;
	msg->action = action;
	msg->unlock = unlock;

	_dbg("Sending commit-reply session-id %Lu on %s req-id %Lu source-ds: %s target-ds: %s action: %u unlock: %d",
	     session->session_id, session->adapter->name, req_id, mgmt_ds_id2name(source),
	     mgmt_ds_id2name(target), action, unlock);

	ret = fe_adapter_send_msg(session->adapter, msg, mgmt_msg_native_get_msg_len(msg), false);
	mgmt_msg_native_free_msg(msg);
	if (ret) {
		_log_err("Failed to send COMMIT_REPLY to session-id %Lu", session->session_id);
		msg_conn_disconnect(session->adapter->conn, false);
	}
}

int mgmt_fe_send_commit_cfg_reply(uint64_t session_id, uint64_t txn_id, enum mgmt_ds_id src_ds_id,
				  enum mgmt_ds_id dst_ds_id, uint64_t req_id, bool validate_only,
				  bool unlock, enum mgmt_result result, const char *error_if_any)
{
	struct mgmt_fe_session_ctx *session;
	uint8_t action;
	int ret = 0;

	/*
	 * When a session is deleted (e.g., disconnects) while there's an active
	 * TXN we can get a NULL return for the session_id after the TXN
	 * completes. A commit message never implicitly locks datastores --
	 * those locks are managed by the client (or cleaned up on session
	 * disconnect).
	 *
	 * However, for the implicit commit (legacy CLI/non-transactional) case
	 * the intention is that each change is made to the candidate and
	 * running at the same (i.e., as the user enters commands they take
	 * affect). So a failure to apply the change to running means we should
	 * back that change out of the candidate DS, and the user will be
	 * presented with an error message.
	 *
	 * So, if we have a failiure to apply an implicit commit, we
	 * should restore the candidate DS, and we can do this by copying
	 * running back over candidate. This doesn't work in the transactional
	 * case b/c the candidate may be made up from multiple changes and a
	 * failure of the latest change shouldn't invalidate all the previous
	 * valid changes which is what would happen if we copied running back
	 * over candidate.
	 */
	session = fe_session_lookup(session_id);
	if (session && session->cfg_txn_id && session->cfg_txn_id != txn_id)
		session = NULL;

	if (validate_only)
		action = MGMT_MSG_COMMIT_VALIDATE;
	else
		action = MGMT_MSG_COMMIT_APPLY;

	/*
	 * Restore the source from the dest in case of error with an implicit commit.
	 * Currently we use the unlock feedback to identify an implicit commit.
	 */
	if (result != MGMTD_SUCCESS && result != MGMTD_NO_CFG_CHANGES && unlock)
		mgmt_ds_copy_dss(mgmt_ds_get_ctx_by_id(mm, src_ds_id),
				 mgmt_ds_get_ctx_by_id(mm, dst_ds_id), false);

	if (!session)
		return -ENOENT;

	if (mm->perf_stats_en)
		gettimeofday(&session->adapter->cmt_stats.last_end, NULL);
	fe_session_compute_commit_timers(&session->adapter->cmt_stats);

	if (result == MGMTD_SUCCESS || result == MGMTD_NO_CFG_CHANGES)
		fe_session_send_commit_reply(session, req_id, src_ds_id, dst_ds_id, action, unlock);
	else {
		ret = fe_session_send_error(
			session, req_id, false, mgmt_result_to_error(result),
			"commit failed session-id %Lu on %s req-id %Lu source-ds: %s target-ds: %s validate-only: %u: reason: '%s'",
			session->session_id, session->adapter->name, req_id,
			mgmt_ds_id2name(src_ds_id), mgmt_ds_id2name(dst_ds_id), validate_only,
			error_if_any ?: "");
	}

	assert(session->cfg_txn_id == txn_id);
	mgmt_destroy_txn(&session->cfg_txn_id);

	return ret;
}

static void fe_session_handle_commit(struct mgmt_fe_session_ctx *session, void *_msg,
				     size_t msg_len)
{
	struct mgmt_msg_commit *msg = _msg;
	struct mgmt_ds_ctx *src_ds_ctx, *dst_ds_ctx;
	uint64_t txn_id;

	_dbg("Got COMMIT for source-ds: %s target-ds: %s action: %s on session-id %Lu from '%s'",
	     mgmt_ds_id2name(msg->source), mgmt_ds_id2name(msg->target),
	     msg->action == MGMT_MSG_COMMIT_VALIDATE ? "validate"
	     : msg->action == MGMT_MSG_COMMIT_ABORT  ? "abort"
						     : "apply",
	     session->session_id, session->adapter->name);

	if (mm->perf_stats_en)
		gettimeofday(&session->adapter->cmt_stats.last_start, NULL);
	session->adapter->cmt_stats.commit_cnt++;

	/* Validate source and dest DS */
	if (msg->source != MGMTD_DS_CANDIDATE || msg->target != MGMTD_DS_RUNNING) {
		fe_session_send_error(session, msg->req_id, false, EINVAL,
				      "source/target for commit must be candidate/running");
		return;
	}

	src_ds_ctx = mgmt_ds_get_ctx_by_id(mm, msg->source);
	dst_ds_ctx = mgmt_ds_get_ctx_by_id(mm, msg->target);

	if (mgmt_ds_is_txn_locked(src_ds_ctx, &txn_id) ||
	    mgmt_ds_is_txn_locked(dst_ds_ctx, &txn_id)) {
		fe_session_send_error(session, msg->req_id, false, EBUSY,
				      "source/target datastore is locked by another transaction txn-id: %Lu",
				      txn_id);
		return;
	}

	/* User must always have lock on source */
	if (!session->ds_locked[msg->source]) {
		fe_session_send_error(session, msg->req_id, false, EBUSY,
				      "source not locked by session-id: %Lu on '%s'",
				      session->session_id, session->adapter->name);
		return;
	}

	/* For apply/abort user must also have lock on target */
	if (msg->action != MGMT_MSG_COMMIT_VALIDATE && !session->ds_locked[msg->target]) {
		fe_session_send_error(session, msg->req_id, false, EBUSY,
				      "target not locked for apply/abort by session-id: %Lu on '%s'",
				      session->session_id, session->adapter->name);
		return;
	}

	/*
	 * We get an exisitng cfg_txn_id if a check/validate was done first then
	 * an apply/abort later.
	 */
	if (session->cfg_txn_id == MGMTD_TXN_ID_NONE) {
		/* as we have the lock no-one else should have a config txn */
		assert(!mgmt_txn_config_in_progress());

		/*
		 * Start a CONFIG Transaction (if not started already)
		 */
		session->cfg_txn_id = mgmt_create_txn(session->session_id, MGMTD_TXN_TYPE_CONFIG);
		if (session->cfg_txn_id == MGMTD_SESSION_ID_NONE) {
			fe_session_send_error(session, msg->req_id, false, ENOMEM,
					      "failed to create config transaction for session-id: %Lu on '%s'",
					      session->session_id, session->adapter->name);
			return;
		}
		_dbg("created config txn-id: %Lu for session-id %Lu on '%s'", session->cfg_txn_id,
		     session->session_id, session->adapter->name);
	}

	/*
	 * Create COMMIT Config request under the transaction
	 */
	mgmt_txn_send_commit_config_req(session->cfg_txn_id, msg->req_id, msg->source, src_ds_ctx,
					msg->target, dst_ds_ctx,
					msg->action == MGMT_MSG_COMMIT_VALIDATE,
					msg->action == MGMT_MSG_COMMIT_ABORT, false /* implicit */,
					msg->unlock, NULL);
}

/* ------------ */
/* LOCK Message */
/* ------------ */

static int fe_session_send_lock_reply(struct mgmt_fe_session_ctx *session, uint64_t req_id,
				      uint8_t datastore, bool lock, bool short_circuit_ok)
{
	struct mgmt_msg_lock_reply *msg;
	int ret;

	msg = mgmt_msg_native_alloc_msg(struct mgmt_msg_lock_reply, 0, MTYPE_MSG_NATIVE_LOCK_REPLY);
	msg->code = MGMT_MSG_CODE_LOCK_REPLY;
	msg->refer_id = session->session_id;
	msg->req_id = req_id;
	msg->datastore = datastore;
	msg->lock = lock;

	_dbg("Sending lock-reply from adapter %s on session-id %Lu req-id %Lu datastore %u lock %u scok %u",
	     session->adapter->name, session->session_id, req_id, datastore, lock,
	     short_circuit_ok);

	ret = fe_adapter_send_msg(session->adapter, msg, mgmt_msg_native_get_msg_len(msg),
				  short_circuit_ok);
	mgmt_msg_native_free_msg(msg);

	return ret;
}

static void fe_session_handle_lock(struct mgmt_fe_session_ctx *session, void *_msg, size_t msg_len)
{
	const struct mgmt_msg_lock *msg = _msg;
	struct mgmt_ds_ctx *ds_ctx;
	bool short_circuit_ok = session->adapter->conn->is_short_circuit;
	uint8_t datastore = msg->datastore;
	uint64_t lock_session;
	uint64_t txn_id;
	bool lock = msg->lock;
	int ret;

	_dbg("Got %sLOCK for DS:%s for session-id %Lu from '%s'", msg->lock ? "" : "UN",
	     mgmt_ds_id2name(datastore), msg->refer_id, session->adapter->name);

	if (datastore != MGMTD_DS_CANDIDATE && datastore != MGMTD_DS_RUNNING) {
		fe_session_send_error(session, msg->req_id, short_circuit_ok, EINVAL,
				      "Lock/Unlock on DS other than candidate or running DS not supported");
		return;
	}

	ds_ctx = mgmt_ds_get_ctx_by_id(mm, datastore);
	assert(ds_ctx);

	if (lock && mgmt_ds_is_locked(ds_ctx, &lock_session, &txn_id) &&
	    lock_session != session->session_id) {
		fe_session_send_error(session, msg->req_id, short_circuit_ok, EBUSY,
				      "Lock already taken on datastore %s by session: %Lu txn-id: %Lu",
				      mgmt_ds_id2name(datastore), lock_session, txn_id);
		return;
	} else if (lock) {
		ret = mgmt_fe_session_write_lock_ds(datastore, ds_ctx, session);
		if (ret) {
			fe_session_send_error(session, msg->req_id, short_circuit_ok, EBUSY,
					      "Unexpected error %d trying to lock datastore by session-id: %Lu",
					      ret, session->session_id);
			return;
		}
	} else {
		/* unlock even if one or both of the lock indicators is wrong */
		mgmt_fe_session_unlock_ds(datastore, ds_ctx, session);
	}

	if (fe_session_send_lock_reply(session, msg->req_id, msg->datastore, msg->lock,
				       short_circuit_ok)) {
		assert(!short_circuit_ok);
		_log_err("Failed to send LOCK_REPLY to session-id %Lu", session->session_id);
		msg_conn_disconnect(session->adapter->conn, false);
	}
}

/* ---------------- */
/* GET-DATA Message */
/* ---------------- */

/*
 * Send result of get-tree request back to the FE client.
 */
static int fe_session_send_tree_data(struct mgmt_fe_session_ctx *session, uint64_t req_id,
				     bool short_circuit_ok, uint8_t result_type,
				     uint32_t wd_options, const struct lyd_node *tree,
				     int partial_error)

{
	struct mgmt_msg_tree_data *msg;
	uint8_t **darrp = NULL;
	int ret = 0;

	msg = mgmt_msg_native_alloc_msg(struct mgmt_msg_tree_data, 0, MTYPE_MSG_NATIVE_TREE_DATA);
	msg->refer_id = session->session_id;
	msg->req_id = req_id;
	msg->code = MGMT_MSG_CODE_TREE_DATA;
	msg->partial_error = partial_error;
	msg->result_type = result_type;

	darrp = mgmt_msg_native_get_darrp(msg);
	ret = yang_print_tree_append(darrp, tree, result_type,
				     (wd_options | LYD_PRINT_WITHSIBLINGS));
	if (ret != LY_SUCCESS) {
		_log_err("Error building get-tree result for client %s session-id %" PRIu64
			 " req-id %" PRIu64 " scok %d result type %u",
			 session->adapter->name, session->session_id, req_id, short_circuit_ok,
			 result_type);
		goto done;
	}

	_dbg("Sending get-tree result from adapter %s to session-id %" PRIu64 " req-id %" PRIu64
	     " scok %d result type %u len %u",
	     session->adapter->name, session->session_id, req_id, short_circuit_ok, result_type,
	     mgmt_msg_native_get_msg_len(msg));

	ret = fe_adapter_send_msg(session->adapter, msg, mgmt_msg_native_get_msg_len(msg),
				  short_circuit_ok);
done:
	mgmt_msg_native_free_msg(msg);

	return ret;
}

void mgmt_fe_adapter_send_tree_data(uint64_t session_id, uint64_t txn_id, uint64_t req_id,
				    LYD_FORMAT result_type, uint32_t wd_options,
				    const struct lyd_node *tree, int partial_error,
				    bool short_circuit_ok)
{
	struct mgmt_fe_session_ctx *session;

	session = fe_session_lookup(session_id);
	if (!session)
		return;

	/* XXXchopps why do we care about this? Why not allow multple? */
	assert(session->txn_id == txn_id);

	if (fe_session_send_tree_data(session, req_id, short_circuit_ok, result_type, wd_options,
				      tree, partial_error))
		fe_session_send_error(session, req_id, false, -EIO,
				      "Failed sending GET-DATA reply");
	mgmt_destroy_txn(&session->txn_id);
}

static void fe_session_handle_get_data(struct mgmt_fe_session_ctx *session, void *_msg,
				       size_t msg_len)
{
	struct mgmt_msg_get_data *msg = _msg;
	const struct lysc_node **snodes = NULL;
	struct lyd_node *ylib = NULL;
	uint64_t req_id = msg->req_id;
	enum mgmt_ds_id ds_id;
	uint64_t clients = 0;
	uint32_t wd_options;
	bool in_oper = false;
	bool simple_xpath;
	LY_ERR err = 0;
	int ret;

	_dbg("Received get-data request from client %s for session-id %" PRIu64 " req-id %" PRIu64,
	     session->adapter->name, session->session_id, msg->req_id);

	if (!MGMT_MSG_VALIDATE_NUL_TERM(msg, msg_len)) {
		fe_session_send_error(session, req_id, false, -EINVAL,
				      "Invalid message rcvd from session-id: %" PRIu64,
				      session->session_id);
		goto done;
	}

	if (session->txn_id != MGMTD_TXN_ID_NONE) {
		fe_session_send_error(session, req_id, false, -EINPROGRESS,
				      "Transaction in progress txn-id: %" PRIu64
				      " for session-id: %" PRIu64,
				      session->txn_id, session->session_id);
		goto done;
	}

	switch (msg->defaults) {
	case GET_DATA_DEFAULTS_EXPLICIT:
		wd_options = LYD_PRINT_WD_EXPLICIT;
		break;
	case GET_DATA_DEFAULTS_TRIM:
		wd_options = LYD_PRINT_WD_TRIM;
		break;
	case GET_DATA_DEFAULTS_ALL:
		wd_options = LYD_PRINT_WD_ALL;
		break;
	case GET_DATA_DEFAULTS_ALL_ADD_TAG:
		wd_options = LYD_PRINT_WD_IMPL_TAG;
		break;
	default:
		fe_session_send_error(session, req_id, false, -EINVAL,
				      "Invalid defaults value %u for session-id: %" PRIu64,
				      msg->defaults, session->session_id);
		goto done;
	}

	/*
	 * Not shrinking can triple or more the size of the result, as a result
	 * we should not send indented results by default and have the FE client
	 * do this instead.
	 */
	wd_options |= LYD_PRINT_SHRINK;

	if (msg->datastore == MGMT_MSG_DATASTORE_OPERATIONAL)
		in_oper = true;

	/* Check for yang-library shortcut */
	if (in_oper && CHECK_FLAG(msg->flags, GET_DATA_FLAG_STATE) &&
	    (!strcmp("/*", msg->xpath) || nb_oper_is_yang_lib_query(msg->xpath))) {
		err = ly_ctx_get_yanglib_data(ly_native_ctx, &ylib, "%u",
					      ly_ctx_get_change_count(ly_native_ctx));
		if (err) {
			fe_session_send_error(session, req_id, false, err,
					      "Error getting yang-library data, session-id: %" PRIu64
					      " error: %s",
					      session->session_id, ly_last_errmsg());
			goto done;
		} else if (nb_oper_is_yang_lib_query(msg->xpath)) {
			struct lyd_node *result;

			yang_lyd_trim_xpath(&ylib, msg->xpath);
			result = ylib;
			if (CHECK_FLAG(msg->flags, GET_DATA_FLAG_EXACT))
				result = yang_dnode_get(result, msg->xpath);
			(void)fe_session_send_tree_data(session, req_id, false, msg->result_type,
							wd_options, result, 0);
			goto done;
		}
	}

	switch (msg->datastore) {
	case MGMT_MSG_DATASTORE_CANDIDATE:
		ds_id = MGMTD_DS_CANDIDATE;
		break;
	case MGMT_MSG_DATASTORE_RUNNING:
		ds_id = MGMTD_DS_RUNNING;
		break;
	case MGMT_MSG_DATASTORE_OPERATIONAL:
		ds_id = MGMTD_DS_OPERATIONAL;
		break;
	default:
		fe_session_send_error(session, req_id, false, -EINVAL,
				      "Unsupported datastore %" PRIu8
				      " requested from session-id: %" PRIu64,
				      msg->datastore, session->session_id);
		goto done;
	}

	err = yang_resolve_snode_xpath(ly_native_ctx, msg->xpath, &snodes,
				       &simple_xpath);
	if (err) {
		fe_session_send_error(session, req_id, false, -EINPROGRESS,
				      "XPath doesn't resolve for session-id: %" PRIu64,
				      session->session_id);
		goto done;
	}
	darr_free(snodes);

	if (in_oper)
		clients = mgmt_be_interested_clients(msg->xpath, MGMT_BE_XPATH_SUBSCR_TYPE_OPER,
						     "GET-DATA");

	if (!clients && !ylib && !CHECK_FLAG(msg->flags, GET_DATA_FLAG_CONFIG)) {
		_dbg("No backends provide xpath: %s for txn-id: %" PRIu64 " session-id: %" PRIu64,
		     msg->xpath, session->txn_id, session->session_id);

		fe_session_send_tree_data(session, req_id, false, msg->result_type, wd_options,
					  NULL, 0);
		goto done;
	}

	if (ylib)
		simple_xpath = false;

	/* Start a SHOW Transaction */
	session->txn_id = mgmt_create_txn(session->session_id,
					  MGMTD_TXN_TYPE_SHOW);
	if (session->txn_id == MGMTD_TXN_ID_NONE) {
		fe_session_send_error(session, req_id, false, -EINPROGRESS,
				      "failed to create a 'show' txn");
		goto done;
	}

	_dbg("Created new show txn-id: %" PRIu64 " for session-id: %" PRIu64, session->txn_id,
	     session->session_id);

	/* Create a GET-TREE request under the transaction */
	ret = mgmt_txn_send_get_tree(session->txn_id, req_id, clients, ds_id, msg->result_type,
				     msg->flags, wd_options, simple_xpath, &ylib, msg->xpath);
	if (ret) {
		/* destroy the just created txn */
		mgmt_destroy_txn(&session->txn_id);
		fe_session_send_error(session, req_id, false, -EINPROGRESS,
				      "failed to create a 'show' txn");
	}
done:
	if (ylib)
		lyd_free_all(ylib);
	darr_free(snodes);
}

/* ------------ */
/* EDIT Message */
/* ------------ */

static int fe_session_send_edit_reply(struct mgmt_fe_session_ctx *session, uint64_t req_id,
				      bool changed, bool created, const char *xpath,
				      const char *data)
{
	struct mgmt_msg_edit_reply *msg;
	int ret;

	msg = mgmt_msg_native_alloc_msg(struct mgmt_msg_edit_reply, 0, MTYPE_MSG_NATIVE_EDIT_REPLY);
	msg->refer_id = session->session_id;
	msg->req_id = req_id;
	msg->changed = changed;
	msg->created = created;
	msg->code = MGMT_MSG_CODE_EDIT_REPLY;

	mgmt_msg_native_xpath_encode(msg, xpath);

	if (data)
		mgmt_msg_native_append(msg, data, strlen(data) + 1);

	_dbg("Sending edit-reply from adapter %s to session-id %" PRIu64 " req-id %" PRIu64
	     " changed %u created %u len %u",
	     session->adapter->name, session->session_id, req_id, changed, created,
	     mgmt_msg_native_get_msg_len(msg));

	ret = fe_adapter_send_msg(session->adapter, msg, mgmt_msg_native_get_msg_len(msg), false);
	mgmt_msg_native_free_msg(msg);

	return ret;
}


int mgmt_fe_adapter_send_edit_reply(uint64_t session_id, uint64_t txn_id, uint64_t req_id,
				    struct mgmt_edit_req **edit, enum mgmt_result result,
				    const char *errstr)
{
	struct mgmt_fe_session_ctx *session;
	const enum mgmt_ds_id can_id = MGMTD_DS_CANDIDATE;
	const enum mgmt_ds_id run_id = MGMTD_DS_RUNNING;
	struct mgmt_ds_ctx *can_ds = mgmt_ds_get_ctx_by_id(mm, can_id);
	struct mgmt_ds_ctx *run_ds = mgmt_ds_get_ctx_by_id(mm, run_id);
	int ret;

	/*
	 * When a session is deleted (e.g., disconnects) while there's an
	 * active TXN we can get a NULL return here when the TXN completes. We
	 * still want to do any cleanup that the session cleanup could not
	 * accomplish b/c of the outstanding TXN.
	 */
	session = fe_session_lookup(session_id);
	if (session && session->cfg_txn_id && session->cfg_txn_id != txn_id)
		session = NULL;

	/*
	 * Deal with the backup candidate config. If the edit was successful we
	 * free the backup. If it failed we restore the datastore from the
	 * backup.
	 */
	if (result == MGMTD_SUCCESS || result == MGMTD_NO_CFG_CHANGES)
		nb_config_free((*edit)->nb_backup);
	else
		mgmt_ds_restore_nb_config(can_ds, (*edit)->nb_backup);

	if (!session)
		return -ENOENT;

	if ((*edit)->unlock_running)
		mgmt_fe_session_unlock_ds(run_id, run_ds, session);
	if ((*edit)->unlock_candidate)
		mgmt_fe_session_unlock_ds(can_id, can_ds, session);


	if (result != MGMTD_SUCCESS && result != MGMTD_NO_CFG_CHANGES)
		ret = fe_session_send_error(session, req_id, false, mgmt_result_to_error(result),
					    "%s", errstr);
	else
		ret = fe_session_send_edit_reply(session, req_id,
						 result == MGMTD_SUCCESS /*changed*/,
						 (*edit)->created, (*edit)->xpath_created, errstr);
	XFREE(MTYPE_MGMTD_TXN_REQ, *edit);

	assert(session->cfg_txn_id == txn_id);
	mgmt_destroy_txn(&session->cfg_txn_id);

	return ret;
}

static void fe_session_handle_edit(struct mgmt_fe_session_ctx *session, void *_msg, size_t msg_len)
{
	struct mgmt_msg_edit *msg = _msg;
	const enum mgmt_ds_id can_id = MGMTD_DS_CANDIDATE;
	const enum mgmt_ds_id run_id = MGMTD_DS_RUNNING;
	struct mgmt_ds_ctx *can_ds = mgmt_ds_get_ctx_by_id(mm, can_id);
	struct mgmt_ds_ctx *run_ds = mgmt_ds_get_ctx_by_id(mm, run_id);
	struct mgmt_edit_req *edit;
	struct nb_config *nb_config;
	uint64_t txn_id = MGMTD_TXN_ID_NONE;
	bool implicit_can_lock = false;
	bool implicit_run_lock = false;
	const char *xpath, *data;
	char errstr[BUFSIZ];
	bool commit;
	int ret;

	/* grab the deprecated commit flag -- the lock flag isn't required at all */
	commit = CHECK_FLAG(msg->flags, EDIT_FLAG_IMPLICIT_COMMIT);

	/*
	 * Validate input args and obtain any needed locks
	 */

	commit = commit || msg->datastore == MGMT_MSG_DATASTORE_RUNNING;
	if (!commit && msg->datastore != MGMT_MSG_DATASTORE_CANDIDATE) {
		fe_session_send_error(session, msg->req_id, false, -EINVAL,
				      "Unsupported datastore");
		return;
	}

	/* Decode the target xpath and config changes */
	xpath = mgmt_msg_native_xpath_data_decode(msg, msg_len, data);
	if (!xpath) {
		fe_session_send_error(session, msg->req_id, false, -EINVAL, "Invalid message");
		return;
	}

	/* If committing, make sure no other txn has locks on the datastores */
	if (commit &&
	    (mgmt_ds_is_txn_locked(can_ds, &txn_id) || mgmt_ds_is_txn_locked(run_ds, &txn_id))) {
		fe_session_send_error(session, msg->req_id, false, -EBUSY,
				      "datastores are locked by another transaction txn-id: %Lu",
				      txn_id);
		return;
	}

	/* We always ensure the candidate DS is locked */
	if (!session->ds_locked[can_id]) {
		if (mgmt_fe_session_write_lock_ds(can_id, can_ds, session)) {
			fe_session_send_error(session, msg->req_id, false, -EBUSY,
					      "Candidate DS is locked by another session");
			return;
		}
		implicit_can_lock = true;
	}

	/* And if modifying running, ensure it is locked too */
	if (commit && !session->ds_locked[run_id]) {
		if (mgmt_fe_session_write_lock_ds(run_id, run_ds, session)) {
			if (implicit_can_lock)
				mgmt_fe_session_unlock_ds(can_id, can_ds, session);
			fe_session_send_error(session, msg->req_id, false, -EBUSY,
					      "Running DS is locked by another session");
			return;
		}
		implicit_run_lock = true;
	}

	/*
	 * Everythign valid and setup, proceed to make the edit.
	 */

	errstr[0] = '\0';
	nb_config = mgmt_ds_get_nb_config(can_ds);

	edit = XCALLOC(MTYPE_MGMTD_TXN_REQ, sizeof(struct mgmt_edit_req));
	edit->unlock_candidate = implicit_can_lock;
	edit->unlock_running = implicit_run_lock;
	edit->nb_backup = nb_config_dup(nb_config); /* keep a backup of candidate */

	/* Make edits to the candidate DS */
	ret = nb_candidate_edit_tree(nb_config, msg->operation, msg->request_type, xpath, data,
				     &edit->created, edit->xpath_created, errstr, sizeof(errstr));
	if (ret && ret == NB_ERR_NO_CHANGES)
		ret = NB_OK;
	else if (ret)
		_dbg("Edit failed for txn-id %Lu req-id %Lu: %s: restoring candidate", txn_id,
		     msg->req_id, errstr[0] ? errstr : "unknown reason");
	else if (commit) {
		/* Get a TXN for the commit */
		txn_id = mgmt_create_txn(session->session_id, MGMTD_TXN_TYPE_CONFIG);
		if (txn_id == MGMTD_SESSION_ID_NONE) {
			ret = NB_ERR_EXISTS; /* should not happen as we have the locks */
			goto reply;
		}
		session->cfg_txn_id = txn_id;

		_dbg("Created new config txn-id: %Lu for session-id: %Lu", txn_id,
		     session->session_id);

		/* And this is modifying the running */
		mgmt_txn_send_commit_config_req(txn_id, msg->req_id, can_id, can_ds, run_id,
						run_ds, false /*abort*/, false /*validate-only*/,
						true /*implicit*/, false /*unlock*/, edit);
		return;
	}
reply:
	mgmt_fe_adapter_send_edit_reply(session->session_id, txn_id, msg->req_id, &edit,
					nb_error_to_mgmt_result(ret), errstr);
}

/* --------------------- */
/* NOTIFY-SELECT Message */
/* --------------------- */

/*
 * Handle an Notify Select message - there's no reply for this message.
 */
static void fe_session_handle_notify_select(struct mgmt_fe_session_ctx *session, void *_msg,
					    size_t msg_len)
{
	struct mgmt_msg_notify_select *msg = _msg;
	uint64_t req_id = msg->req_id;
	struct nb_node **nb_nodes;
	const char **selectors = NULL;
	const char **new;
	const char **sp;
	char *selstr = NULL;
	uint64_t clients = 0;
	uint64_t all_matched = 0, rm_clients = 0;


	if (msg_len >= sizeof(*msg)) {
		selectors = mgmt_msg_native_strings_decode(msg, msg_len, msg->selectors);
		if (!selectors) {
			fe_session_send_error(session, req_id, false, -EINVAL, "Invalid message");
			return;
		}
	}

	/* Validate all selectors, they need to resolve to actual northbound_nodes */
	darr_foreach_p (selectors, sp) {
		nb_nodes = nb_nodes_find(*sp);
		if (!nb_nodes) {
			fe_session_send_error(session, req_id, false, -EINVAL,
					      "Selector doesn't resolve to a node: %s", *sp);
			darr_free_free(selectors);
			return;
		}
		darr_free(nb_nodes);
	}

	if (DEBUG_MODE_CHECK(&mgmt_debug_fe, DEBUG_MODE_ALL)) {
		selstr = frrstr_join(selectors, darr_len(selectors), ", ");
		if (!selstr)
			selstr = XSTRDUP(MTYPE_TMP, "");
	}

	if (msg->replace) {
		rm_clients = mgmt_fe_ns_string_remove_session(&mgmt_fe_ns_strings, session);
		// [ ] Keep a local tree to optimize sending selectors to BE?
		// [*] Or just KISS and fanout the original message to BEs?
		// mgmt_remove_add_notify_selectors(session->notify_xpaths, selectors);
		darr_free_free(session->notify_xpaths);
		session->notify_xpaths = selectors;
	} else if (selectors) {
		// [ ] Keep a local tree to optimize sending selectors to BE?
		// [*] Or just KISS and fanout the original message to BEs?
		// mgmt_remove_add_notify_selectors(session->notify_xpaths, selectors);
		new = darr_append_nz(session->notify_xpaths, darr_len(selectors));
		memcpy(new, selectors, darr_len(selectors) * sizeof(*selectors));
	} else {
		_log_err("Invalid msg from session-id: %Lu: no selectors present in non-replace msg",
			 session->session_id);
		darr_free_free(selectors);
		selectors = NULL;
		goto done;
	}


	if (session->notify_xpaths && DEBUG_MODE_CHECK(&mgmt_debug_fe, DEBUG_MODE_ALL)) {
		const char **sel = session->notify_xpaths;
		char *s = frrstr_join(sel, darr_len(sel), ", ");
		_dbg("New NOTIF %d selectors '%s' (replace: %d) for session-id: %Lu", darr_len(sel),
		     s, msg->replace, session->session_id);
		XFREE(MTYPE_TMP, s);
	}

	/*
	 * Add the new selectors to the global tree, track BE clients that
	 * haven't been given the selectors (that need to be), and also all the
	 * BE clients that provide state for the selectors (to query for initial
	 * dump)
	 */
	darr_foreach_p (selectors, sp)
		clients |= mgmt_fe_add_ns_string(&mgmt_fe_ns_strings, *sp, darr_strlen(*sp),
						 session, &all_matched);

	if (!(all_matched | rm_clients)) {
		_dbg("No backends publishing for selectors: '%s' session-id: %Lu", selstr,
		     session->session_id);
		goto done;
	}
	if (!(clients | rm_clients))
		_dbg("No backends to newly notify for selectors: '%s' session-id: %Lu", selstr,
		     session->session_id);
	else
		/* Send a message to set the selectors on the changed clients */
		mgmt_txn_send_notify_selectors(req_id, MGMTD_SESSION_ID_NONE,
					       (clients | rm_clients),
					       msg->replace ? NULL : selectors);

	if (!all_matched || !selectors)
		goto done;

	_dbg("Created new push for session-id: %Lu", session->session_id);

	/* Send a second message requesting a full state dump for the session */
	mgmt_txn_send_notify_selectors(req_id, session->session_id, all_matched, selectors);
done:
	if (session->notify_xpaths != selectors)
		darr_free(selectors);
	if (selstr)
		XFREE(MTYPE_TMP, selstr);
}

/* ----------- */
/* RPC Message */
/* ----------- */

static int fe_session_send_rpc_reply(struct mgmt_fe_session_ctx *session, uint64_t req_id,
				     uint8_t result_type, bool restconf,
				     const struct lyd_node *result)
{
	struct mgmt_msg_rpc_reply *msg;
	uint8_t **darrp = NULL;
	int ret;

	msg = mgmt_msg_native_alloc_msg(struct mgmt_msg_rpc_reply, 0, MTYPE_MSG_NATIVE_RPC_REPLY);
	msg->refer_id = session->session_id;
	msg->req_id = req_id;
	msg->code = MGMT_MSG_CODE_RPC_REPLY;
	msg->result_type = result_type;
	msg->restconf = restconf;

	if (result) {
		darrp = mgmt_msg_native_get_darrp(msg);
		ret = yang_print_tree_append(darrp, result, result_type,
					     (LYD_PRINT_SHRINK | LYD_PRINT_WD_EXPLICIT |
					      LYD_PRINT_WITHSIBLINGS));
		if (ret != LY_SUCCESS) {
			_log_err("Error building rpc-reply result for client %s session-id %" PRIu64
				 " req-id %" PRIu64 " result type %u",
				 session->adapter->name, session->session_id, req_id, result_type);
			goto done;
		}
	}

	_dbg("Sending rpc-reply from adapter %s to session-id %" PRIu64 " req-id %" PRIu64
	     " len %u",
	     session->adapter->name, session->session_id, req_id, mgmt_msg_native_get_msg_len(msg));

	ret = fe_adapter_send_msg(session->adapter, msg, mgmt_msg_native_get_msg_len(msg), false);
done:
	mgmt_msg_native_free_msg(msg);

	return ret;
}

void mgmt_fe_adapter_send_rpc_reply(uint64_t session_id, uint64_t txn_id, uint64_t req_id,
				    LYD_FORMAT result_type, bool restconf,
				    const struct lyd_node *result)
{
	struct mgmt_fe_session_ctx *session;

	session = fe_session_lookup(session_id);
	if (!session)
		return;

	/* XXXchopps why do we care about this? Why not allow multple? */
	assert(session->txn_id == txn_id);

	if (fe_session_send_rpc_reply(session, req_id, result_type, restconf, result))
		fe_session_send_error(session, req_id, false, -EIO, "Failed sending RPC reply");

	mgmt_destroy_txn(&session->txn_id);
}

static void fe_session_handle_rpc(struct mgmt_fe_session_ctx *session, void *_msg, size_t msg_len)
{
	struct mgmt_msg_rpc *msg = _msg;
	const struct lysc_node *snode;
	const char *xpath, *data;
	uint64_t req_id = msg->req_id;
	uint64_t clients;

	_dbg("Received RPC request from client %s for session-id %" PRIu64 " req-id %" PRIu64,
	     session->adapter->name, session->session_id, msg->req_id);

	xpath = mgmt_msg_native_xpath_data_decode(msg, msg_len, data);
	if (!xpath) {
		fe_session_send_error(session, req_id, false, -EINVAL, "Invalid message");
		return;
	}

	if (session->txn_id != MGMTD_TXN_ID_NONE) {
		fe_session_send_error(session, req_id, false, -EINPROGRESS,
				      "Transaction in progress txn-id: %" PRIu64
				      " for session-id: %" PRIu64,
				      session->txn_id, session->session_id);
		return;
	}

	snode = lys_find_path(ly_native_ctx, NULL, xpath, 0);
	if (!snode) {
		fe_session_send_error(session, req_id, false, -ENOENT, "No such path: %s", xpath);
		return;
	}

	if (snode->nodetype != LYS_RPC && snode->nodetype != LYS_ACTION) {
		fe_session_send_error(session, req_id, false, -EINVAL,
				      "Not an RPC or action path: %s", xpath);
		return;
	}

	clients = mgmt_be_interested_clients(xpath, MGMT_BE_XPATH_SUBSCR_TYPE_RPC, "RPC");
	if (!clients) {
		_dbg("No backends implement xpath: %s for txn-id: %" PRIu64 " session-id: %" PRIu64,
		     xpath, session->txn_id, session->session_id);

		fe_session_send_error(session, req_id, false, -ENOENT,
				      "No backends implement xpath: %s", xpath);
		return;
	}

	/* Start a RPC Transaction */
	session->txn_id = mgmt_create_txn(session->session_id,
					  MGMTD_TXN_TYPE_RPC);
	if (session->txn_id == MGMTD_SESSION_ID_NONE) {
		fe_session_send_error(session, req_id, false, -EINPROGRESS,
				      "Failed to create an RPC transaction");
		return;
	}

	_dbg("Created new rpc txn-id: %" PRIu64 " for session-id: %" PRIu64, session->txn_id,
	     session->session_id);

	/* Create an RPC request under the transaction */
	mgmt_txn_send_rpc(session->txn_id, req_id, clients, msg->request_type, msg->restconf,
			  xpath, data, mgmt_msg_native_data_len_decode(msg, msg_len));
}

/* -------------------- */
/* New Session Requests */
/* -------------------- */

static int fe_adapter_conn_send_error(struct msg_conn *conn, uint64_t session_id, uint64_t req_id,
				      bool short_circuit_ok, int16_t error, const char *errfmt,
				      ...) PRINTFRR(6, 7);
static int fe_adapter_conn_send_error(struct msg_conn *conn, uint64_t session_id, uint64_t req_id,
				      bool short_circuit_ok, int16_t error, const char *errfmt, ...)
{
	va_list ap;
	int ret;

	va_start(ap, errfmt);

	ret = vmgmt_msg_native_send_error(conn, session_id, req_id, short_circuit_ok, error,
					  errfmt, ap);
	va_end(ap);

	return ret;
}

static int fe_adapter_send_session_reply(struct mgmt_fe_client_adapter *adapter,
					 uint64_t session_id, uint64_t req_id, bool created)
{
	struct mgmt_msg_session_reply *msg;
	bool scok = adapter->conn->is_short_circuit;
	int ret;

	msg = mgmt_msg_native_alloc_msg(struct mgmt_msg_session_reply, 0,
					MTYPE_MSG_NATIVE_SESSION_REPLY);
	msg->refer_id = session_id;
	msg->req_id = req_id;
	msg->code = MGMT_MSG_CODE_SESSION_REPLY;
	msg->created = created;

	_dbg("Sending session-reply from adapter %s to session-id %Lu req-id %Lu created %u scok %u",
	     adapter->name, session_id, req_id, created, scok);

	ret = fe_adapter_send_msg(adapter, msg, mgmt_msg_native_get_msg_len(msg), scok);
	mgmt_msg_native_free_msg(msg);

	return ret;
}

/*
 * Handle a session-req message from a FE client.
 */
static void fe_adapter_handle_session_req(struct mgmt_fe_client_adapter *adapter, void *_msg,
					  size_t msg_len)
{
	const struct mgmt_msg_session_req *msg = _msg;
	struct mgmt_fe_session_ctx *session;
	bool scok = adapter->conn->is_short_circuit;
	uint64_t client_id;

	_dbg("Got session-req is create %u req-id %Lu for refer-id %Lu notify-fmt %u from '%s'",
	     msg->refer_id == 0, msg->req_id, msg->refer_id, msg->notify_format, adapter->name);

	/*
	 * It's important that any error has its refer_id set to 0 for create
	 * case and set to the passed in msg->refer_id otherwise (the destroy
	 * case). For non-error return pass the session_id for create or destroy.
	 */

	if (msg->refer_id) {
		uint64_t session_id = msg->refer_id;

		session = fe_session_lookup(session_id);
		if (!session) {
			fe_adapter_conn_send_error(adapter->conn, session_id, msg->req_id, scok,
						   EINVAL,
						   "No session to delete for session-id: %" PRIu64,
						   session_id);
			return;
		}
		fe_adapter_send_session_reply(adapter, session_id, msg->req_id, false);
		fe_session_cleanup(&session);
		return;
	}

	client_id = msg->req_id;

	/* Default notification format */
	if (msg->notify_format && msg->notify_format > MGMT_MSG_FORMAT_LAST) {
		fe_adapter_conn_send_error(adapter->conn, 0, msg->req_id, scok, EINVAL,
					   "Unrecognized notify format: %u", msg->notify_format);
		return;
	}

	/* See if we have a client name to register */
	if (msg_len > sizeof(*msg)) {
		if (!MGMT_MSG_VALIDATE_NUL_TERM(msg, msg_len)) {
			fe_adapter_conn_send_error(adapter->conn, 0, msg->req_id, scok, EINVAL,
						   "Corrupt session-req msg from client-id: %Lu",
						   client_id);
			return;
		}
		_dbg("Set client-name to '%s'", msg->client_name);
		strlcpy(adapter->name, msg->client_name, sizeof(adapter->name));
	}

	session = fe_session_create(adapter, msg->notify_format ?: DEFAULT_NOTIFY_FORMAT,
				    client_id);
	fe_adapter_send_session_reply(adapter, session->session_id, msg->req_id, true);
}

/*
 * Handle a message from the FE client.
 */
static void fe_adapter_process_msg(uint8_t version, uint8_t *data, size_t msg_len,
				   struct msg_conn *conn)
{
	struct mgmt_fe_client_adapter *adapter = conn->user;
	struct mgmt_fe_session_ctx *session;
	struct mgmt_msg_header *msg = (typeof(msg))data;
	size_t min_size;

	if (version != MGMT_MSG_VERSION_NATIVE) {
		_log_err("Protobuf not supported for frontend messages (adapter: %s)",
			 adapter->name);
		return;
	}

	if (msg_len < sizeof(*msg)) {
		_log_err("native message to adapter %s too short %zu", adapter->name, msg_len);
		return;
	}

	min_size = mgmt_msg_get_min_size(msg->code);
	if (msg_len < min_size) {
		if (!min_size)
			_log_err("adapter %s: recv msg refer-id %" PRIu64 " unknown message type %u",
				 adapter->name, msg->refer_id, msg->code);
		else
			_log_err("adapter %s: recv msg refer-id %" PRIu64
				 " short (%zu<%zu) msg for type %u",
				 adapter->name, msg->refer_id, msg_len, min_size, msg->code);
		return;
	}

	/*
	 * Handle new session requests up front.
	 */

	if (msg->code == MGMT_MSG_CODE_SESSION_REQ) {
		_dbg("adapter %s: session-id %Lu received SESSION_REQ message", adapter->name,
		     msg->refer_id);
		fe_adapter_handle_session_req(adapter, msg, msg_len);
		return;
	}

	/*
	 * Get session and handle all other message types.
	 */

	session = fe_session_lookup(msg->refer_id);
	if (!session) {
		_log_err("adapter %s: recv msg unknown session-id %" PRIu64, adapter->name,
			 msg->refer_id);
		return;
	}
	assert(session->adapter == adapter);

	switch (msg->code) {
	case MGMT_MSG_CODE_COMMIT:
		_dbg("adapter %s: session-id %Lu received COMMIT message", adapter->name,
		     msg->refer_id);
		fe_session_handle_commit(session, msg, msg_len);
		break;
	case MGMT_MSG_CODE_EDIT:
		_dbg("adapter %s: session-id %" PRIu64 " received EDIT message", adapter->name,
		     msg->refer_id);
		fe_session_handle_edit(session, msg, msg_len);
		break;
	case MGMT_MSG_CODE_LOCK:
		_dbg("adapter %s: session-id %Lu received LOCK message", adapter->name,
		     msg->refer_id);
		fe_session_handle_lock(session, msg, msg_len);
		break;
	case MGMT_MSG_CODE_NOTIFY_SELECT:
		_dbg("adapter %s: session-id %" PRIu64 " received NOTIFY_SELECT message",
		     adapter->name, msg->refer_id);
		fe_session_handle_notify_select(session, msg, msg_len);
		break;
	case MGMT_MSG_CODE_GET_DATA:
		_dbg("adapter %s: session-id %" PRIu64 " received GET_DATA message", adapter->name,
		     msg->refer_id);
		fe_session_handle_get_data(session, msg, msg_len);
		break;
	case MGMT_MSG_CODE_RPC:
		_dbg("adapter %s: session-id %" PRIu64 " received RPC message", adapter->name,
		     msg->refer_id);
		fe_session_handle_rpc(session, msg, msg_len);
		break;
	default:
		_log_err("unknown native message session-id %" PRIu64 " req-id %" PRIu64
			 " code %u to FE adapter %s",
			 msg->refer_id, msg->req_id, msg->code, adapter->name);
		break;
	}
}

/* ============================= */
/* Async Notification Processing */
/* ============================= */

static struct mgmt_msg_notify_data *assure_notify_msg_cache(const struct mgmt_msg_notify_data *msg,
							    size_t msglen, struct lyd_node **tree,
							    uint8_t format,
							    struct mgmt_msg_notify_data **cache)

{
	uint32_t parse_options = LYD_PARSE_STRICT | LYD_PARSE_ONLY;
	struct mgmt_msg_notify_data *new_msg;
	const struct lyd_node *root;
	uint8_t **darrp = NULL;
	const char *data, *xpath;
	LY_ERR err;

	if (cache[format])
		return cache[format];

	_dbg("creating notify msg cache for format %u", format);

	xpath = mgmt_msg_native_xpath_data_decode(msg, msglen, data);

	/* Get a libyang data tree if we haven't yet */
	if (!*tree) {
#ifdef LYD_PARSE_LYB_SKIP_CTX_CHECK
		if (msg->result_type == LYD_LYB)
			parse_options |= LYD_PARSE_LYB_SKIP_CTX_CHECK;
#endif
		err = lyd_parse_data_mem(ly_native_ctx, data, msg->result_type, parse_options, 0,
					 tree);
		assert(err == LY_SUCCESS);
	}

	root = *tree;

	/* Copy original message (fixed-part), update format */
	new_msg = mgmt_msg_native_alloc_msg(struct mgmt_msg_notify_data, 0, MTYPE_MSG_NATIVE_NOTIFY);
	*new_msg = *msg;
	new_msg->result_type = format;

	/* Append the xpath string */
	mgmt_msg_native_xpath_encode(new_msg, xpath);

	/*
	 * Append new `format`ed data
	 */

	/* For JSON result top node starts at the xpath target */
	if (format == LYD_JSON) {
		root = yang_dnode_get(*tree, xpath);
		assert(root);
	}

	darrp = mgmt_msg_native_get_darrp(new_msg);
	err = yang_print_tree_append(darrp, root, format, LYD_PRINT_WITHSIBLINGS);
	assert(err == LY_SUCCESS);

	cache[format] = new_msg;
	return new_msg;
}

static void cleanup_notify_msg_cache(struct mgmt_msg_notify_data *msg, struct lyd_node **tree,
				     struct mgmt_msg_notify_data **cache)

{
	if (*tree) {
		lyd_free_all(*tree);
		*tree = NULL;
	}

	for (uint i = 0; i <= MGMT_MSG_FORMAT_LAST; i++) {
		if (cache[i] && cache[i] != msg) {
			_dbg("freeing notify msg cache for format %u", i);
			mgmt_msg_native_free_msg(cache[i]);
		}
	}
}

void mgmt_fe_adapter_send_notify(struct mgmt_msg_notify_data *msg, size_t msglen)
{
	struct mgmt_msg_notify_data *cache[MGMT_MSG_FORMAT_LAST + 1] = {};
	struct mgmt_msg_notify_data *send_msg;
	struct mgmt_fe_client_adapter *adapter;
	struct mgmt_fe_session_ctx **sessions = NULL;
	struct mgmt_fe_session_ctx *session;
	struct nb_node *nb_node = NULL;
	struct lyd_node *tree = NULL;
	struct listnode *node;
	struct ns_string *ns;
	const char *notif;
	uint i, sel_len, notif_len, nb_xpath_len;

	cache[msg->result_type] = msg;

	notif = mgmt_msg_native_xpath_decode(msg, msglen);
	if (!notif) {
		_log_err("Corrupt notify msg");
		return;
	}

	/* We don't support root level notifications, no backend should send this */
	assert(strcmp(notif, "/"));

	/*
	 * We need the nb_node to obtain a path which does not include any
	 * specific list entry selectors
	 */
	nb_node = nb_node_find(notif);
	if (!nb_node) {
		_log_err("No schema found for notification: %s", notif);
		return;
	}

	/*
	 * Handle notify "get" data case. When a FE session subscribes to DS
	 * notifications it first gets a dump of all the subscribed state.
	 */
	if (msg->refer_id != MGMTD_SESSION_ID_NONE) {
		session = fe_session_lookup(msg->refer_id);
		if (!session || !session->notify_xpaths) {
			_dbg("No session listening for notify 'get' data: %Lu", msg->refer_id);
			return;
		}

		send_msg = assure_notify_msg_cache(msg, msglen, &tree, session->notify_format,
						   cache);
		(void)fe_adapter_send_msg(session->adapter, send_msg, msglen, false);
		goto done;
	}

	/*
	 * Normal notification case.
	 */

	notif_len = strlen(notif);
	nb_xpath_len = strlen(nb_node->xpath);
	frr_each (ns_string, &mgmt_fe_ns_strings, ns) {
		sel_len = strlen(ns->s);
		/*
		 * Notify if:
		 * 1) the selector covers (is prefix of) the specific notified path.
		 * 2) the selector covers (is prefix of) the schema path of the
		 * notified path. this means the selector is generic (contains no keys)
		 *
		 * Also check if the selector is contained by the notification path
		 * (i.e., it's a prefix of).
		 */
		if (/* selector contains (specific or schema) notification path */
		    strncmp(ns->s, notif, sel_len) && strncmp(ns->s, nb_node->xpath, sel_len) &&
		    /* notify (specific or schema) contains selector */
		    strncmp(notif, ns->s, notif_len) && strncmp(nb_node->xpath, ns->s, nb_xpath_len))
			continue;

		for (ALL_LIST_ELEMENTS_RO(ns->sessions, node, session))
			darr_push_uniq(sessions, session);
	}
	/* Send to all interested sessions */
	darr_foreach_i (sessions, i) {
		send_msg = assure_notify_msg_cache(msg, msglen, &tree, sessions[i]->notify_format,
						   cache);
		send_msg->refer_id = sessions[i]->session_id;
		(void)fe_adapter_send_msg(sessions[i]->adapter, send_msg, msglen, false);
	}
	darr_free(sessions);

	/*
	 * Send all YANG defined notifications to all sesisons with *no*
	 * selectors as well (i.e., original NETCONF/RESTCONF notification
	 * scheme).
	 */
	if (CHECK_FLAG(nb_node->snode->nodetype, LYS_NOTIF)) {
		LIST_FOREACH (adapter, &fe_adapters, link) {
			LIST_FOREACH (session, &adapter->sessions, link) {
				if (session->notify_xpaths)
					continue;
				send_msg = assure_notify_msg_cache(msg, msglen, &tree,
								   session->notify_format, cache);
				send_msg->refer_id = session->session_id;
				(void)fe_adapter_send_msg(adapter, send_msg, msglen, false);
			}
		}
	}

	msg->refer_id = 0;

done:
	cleanup_notify_msg_cache(msg, &tree, cache);
}


/* =========================== */
/* Frontend VTY and Statistics */
/* =========================== */

static void fe_session_compute_commit_timers(struct mgmt_commit_stats *cmt_stats)
{
	cmt_stats->last_exec_tm = timeval_elapsed(cmt_stats->last_end, cmt_stats->last_start);
	if (cmt_stats->last_exec_tm > cmt_stats->max_tm) {
		cmt_stats->max_tm = cmt_stats->last_exec_tm;
		cmt_stats->max_batch_cnt = cmt_stats->last_batch_cnt;
	}

	if (cmt_stats->last_exec_tm < cmt_stats->min_tm) {
		cmt_stats->min_tm = cmt_stats->last_exec_tm;
		cmt_stats->min_batch_cnt = cmt_stats->last_batch_cnt;
	}
}

struct mgmt_commit_stats *mgmt_fe_get_session_commit_stats(uint64_t session_id)
{
	struct mgmt_fe_session_ctx *session;

	session = fe_session_lookup(session_id);
	if (!session || !session->adapter)
		return NULL;

	return &session->adapter->cmt_stats;
}

static void _cmt_stats_write(struct vty *vty, struct mgmt_fe_client_adapter *adapter)
{
	char buf[MGMT_LONG_TIME_MAX_LEN];

	if (!mm->perf_stats_en)
		return;

	vty_out(vty, "    Num-Commits: \t\t\t%lu\n",
		adapter->cmt_stats.commit_cnt);
	if (adapter->cmt_stats.commit_cnt > 0) {
		if (mm->perf_stats_en)
			vty_out(vty, "    Max-Commit-Duration: \t\t%lu uSecs\n",
				adapter->cmt_stats.max_tm);
		vty_out(vty, "    Max-Commit-Batch-Size: \t\t%lu\n",
			adapter->cmt_stats.max_batch_cnt);
		if (mm->perf_stats_en)
			vty_out(vty, "    Min-Commit-Duration: \t\t%lu uSecs\n",
				adapter->cmt_stats.min_tm);
		vty_out(vty, "    Min-Commit-Batch-Size: \t\t%lu\n",
			adapter->cmt_stats.min_batch_cnt);
		if (mm->perf_stats_en)
			vty_out(vty,
				"    Last-Commit-Duration: \t\t%lu uSecs\n",
				adapter->cmt_stats.last_exec_tm);
		vty_out(vty, "    Last-Commit-Batch-Size: \t\t%lu\n",
			adapter->cmt_stats.last_batch_cnt);
		vty_out(vty, "    Last-Commit-CfgData-Reqs: \t\t%lu\n",
			adapter->cmt_stats.last_num_cfgdata_reqs);
		vty_out(vty, "    Last-Commit-CfgApply-Reqs: \t\t%lu\n",
			adapter->cmt_stats.last_num_apply_reqs);
		if (mm->perf_stats_en) {
			vty_out(vty, "    Last-Commit-Details:\n");
			vty_out(vty, "      Commit Start: \t\t\t%s\n",
				mgmt_realtime_to_string(
					&adapter->cmt_stats.last_start, buf,
					sizeof(buf)));
			vty_out(vty, "        Prep-Config Start: \t\t%s\n",
				mgmt_realtime_to_string(
					&adapter->cmt_stats.prep_cfg_start, buf,
					sizeof(buf)));
			vty_out(vty, "        Txn-Create Start: \t\t%s\n",
				mgmt_realtime_to_string(
					&adapter->cmt_stats.txn_create_start,
					buf, sizeof(buf)));
			vty_out(vty, "        Apply-Config Start: \t\t%s\n",
				mgmt_realtime_to_string(
					&adapter->cmt_stats.apply_cfg_start,
					buf, sizeof(buf)));
			vty_out(vty, "        Apply-Config End: \t\t%s\n",
				mgmt_realtime_to_string(
					&adapter->cmt_stats.apply_cfg_end, buf,
					sizeof(buf)));
			vty_out(vty, "        Txn-Delete Start: \t\t%s\n",
				mgmt_realtime_to_string(
					&adapter->cmt_stats.txn_del_start, buf,
					sizeof(buf)));
			vty_out(vty, "      Commit End: \t\t\t%s\n",
				mgmt_realtime_to_string(
					&adapter->cmt_stats.last_end, buf,
					sizeof(buf)));
		}
	}
}

void mgmt_fe_adapter_status_write(struct vty *vty, bool detail)
{
	struct mgmt_fe_client_adapter *adapter;
	struct mgmt_fe_session_ctx *session;
	enum mgmt_ds_id ds_id;
	bool locked = false;
	uint acount = 0;
	uint scount;

	vty_out(vty, "MGMTD Frontend Adpaters\n");

	LIST_FOREACH (adapter, &fe_adapters, link) {
		vty_out(vty, "  Client: \t\t\t\t%s\n", adapter->name);
		vty_out(vty, "    Conn-FD: \t\t\t\t%d\n", adapter->conn->fd);
		if (detail) {
			_cmt_stats_write(vty, adapter);
		}
		scount = 0;
		vty_out(vty, "    Sessions\n");
		LIST_FOREACH (session, &adapter->sessions, link) {
			vty_out(vty, "      Session: \t\t\t\t%p\n", session);
			vty_out(vty, "        Client-Id: \t\t\t%" PRIu64 "\n",
				session->client_id);
			vty_out(vty, "        Session-Id: \t\t\t%" PRIu64 "\n",
				session->session_id);
			vty_out(vty, "        DS-Locks:\n");
			FOREACH_MGMTD_DS_ID (ds_id) {
				if (session->ds_locked[ds_id]) {
					locked = true;
					vty_out(vty, "          %s\n",
						mgmt_ds_id2name(ds_id));
				}
			}
			if (!locked)
				vty_out(vty, "          None\n");
			scount++;
		}
		vty_out(vty, "    Total-Sessions: \t\t\t%u\n", scount);
		vty_out(vty, "    Msg-Recvd: \t\t\t\t%" PRIu64 "\n",
			adapter->conn->mstate.nrxm);
		vty_out(vty, "    Bytes-Recvd: \t\t\t%" PRIu64 "\n",
			adapter->conn->mstate.nrxb);
		vty_out(vty, "    Msg-Sent: \t\t\t\t%" PRIu64 "\n",
			adapter->conn->mstate.ntxm);
		vty_out(vty, "    Bytes-Sent: \t\t\t%" PRIu64 "\n",
			adapter->conn->mstate.ntxb);
	}
	vty_out(vty, "  Total: %u\n", acount);
}

void mgmt_fe_adapter_perf_measurement(struct vty *vty, bool config)
{
	mm->perf_stats_en = config;
}

void mgmt_fe_adapter_reset_perf_stats(struct vty *vty)
{
	struct mgmt_fe_client_adapter *adapter;

	LIST_FOREACH (adapter, &fe_adapters, link)
		memset(&adapter->cmt_stats, 0, sizeof(adapter->cmt_stats));
}

/* =================== */
/* Frontend Management */
/* =================== */

void mgmt_fe_adapter_toggle_client_debug(bool set)
{
	struct mgmt_fe_client_adapter *adapter;

	LIST_FOREACH (adapter, &fe_adapters, link)
		adapter->conn->debug = set;
}

static struct mgmt_fe_client_adapter *fe_adapter_lookup_by_fd(int conn_fd)
{
	struct mgmt_fe_client_adapter *adapter;

	LIST_FOREACH (adapter, &fe_adapters, link)
		if (adapter->conn->fd == conn_fd)
			return adapter;
	return NULL;
}

static void fe_adapter_delete(struct mgmt_fe_client_adapter *adapter)
{
	struct mgmt_fe_session_ctx *session, *next;

	_dbg("Deleting client adapter '%s'", adapter->name);

	/* TODO: notify about client disconnect for appropriate cleanup */
	LIST_FOREACH_SAFE (session, &adapter->sessions, link, next)
		fe_session_cleanup(&session);
	LIST_REMOVE(adapter, link);
	msg_server_conn_delete(adapter->conn);
	XFREE(MTYPE_MGMTD_FE_ADPATER, adapter);
}

static int fe_adapter_notify_disconnect(struct msg_conn *conn)
{
	struct mgmt_fe_client_adapter *adapter = conn->user;

	_dbg("notify disconnect for client adapter '%s'", adapter->name);

	fe_adapter_delete(adapter);

	return 0;
}

/*
 * The server accepted a new connection
 */
static struct msg_conn *fe_adapter_create(int conn_fd, union sockunion *from)
{
	struct mgmt_fe_client_adapter *adapter = NULL;

	adapter = fe_adapter_lookup_by_fd(conn_fd);
	if (!adapter) {
		adapter = XCALLOC(MTYPE_MGMTD_FE_ADPATER, sizeof(struct mgmt_fe_client_adapter));
		snprintf(adapter->name, sizeof(adapter->name), "Unknown-FD-%d", conn_fd);

		LIST_INSERT_HEAD(&fe_adapters, adapter, link);

		adapter->conn =
			msg_server_conn_create(mgmt_loop, conn_fd, fe_adapter_notify_disconnect,
					       fe_adapter_process_msg, MGMTD_FE_MAX_NUM_MSG_PROC,
					       MGMTD_FE_MAX_NUM_MSG_WRITE, MGMTD_FE_MAX_MSG_LEN,
					       adapter, "FE-ADAPTER-CONN");

		adapter->conn->debug = DEBUG_MODE_CHECK(&mgmt_debug_fe, DEBUG_MODE_ALL);

		adapter->cmt_stats.min_tm = ULONG_MAX;
		_dbg("Added new MGMTD Frontend adapter '%s'", adapter->name);
	}
	return adapter->conn;
}


/*
 * Initialize the FE adapter module
 */
void mgmt_fe_adapter_init(struct event_loop *tm)
{
	char server_path[MAXPATHLEN];

	assert(!mgmt_loop);
	mgmt_loop = tm;

	assert(!mgmt_fe_sessions);
	mgmt_fe_sessions = hash_create(fe_session_hash_key, fe_session_hash_cmp,
				       "MGMT Frontend Sessions");

	ns_string_init(&mgmt_fe_ns_strings);

	snprintf(server_path, sizeof(server_path), MGMTD_FE_SOCK_NAME);

	if (msg_server_init(&mgmt_fe_server, server_path, tm, fe_adapter_create, "frontend",
			    &mgmt_debug_fe)) {
		zlog_err("cannot initialize frontend server");
		exit(1);
	}
}

/*
 * Destroy the FE adapter module
 */
void mgmt_fe_adapter_destroy(void)
{
	struct mgmt_fe_client_adapter *adapter, *next;

	msg_server_cleanup(&mgmt_fe_server);


	/* Deleting the adapters will delete all the sessions */
	LIST_FOREACH_SAFE (adapter, &fe_adapters, link, next)
		fe_adapter_delete(adapter);

	mgmt_fe_free_ns_strings(&mgmt_fe_ns_strings);

	hash_free(mgmt_fe_sessions);
}
