Viewing: mgs_handler.c

// SPDX-License-Identifier: GPL-2.0

/*
 * Copyright (c) 2007, 2010, Oracle and/or its affiliates. All rights reserved.
 * Use is subject to license terms.
 *
 * Copyright (c) 2010, 2017, Intel Corporation.
 */

/*
 * This file is part of Lustre, http://www.lustre.org/
 *
 * Author: Nathan Rutman <nathan@clusterfs.com>
 * Author: Alex Zhuravlev <bzzz@whamcloud.com>
 * Author: Mikhail Pershin <tappro@whamcloud.com>
 */

#define DEBUG_SUBSYSTEM S_MGS
#define D_MGS D_CONFIG

#include <obd_class.h>
#include <lprocfs_status.h>
#include <uapi/linux/lustre/lustre_ioctl.h>
#include <lustre_ioctl_old.h>
#include <uapi/linux/lustre/lustre_param.h>
#include <uapi/linux/lustre/lustre_barrier_user.h>

#include "mgs_internal.h"

/*
 * Regular MGS handlers
 */
static int mgs_connect(struct tgt_session_info *tsi)
{
	struct ptlrpc_request	*req = tgt_ses_req(tsi);
	int			 rc;

	ENTRY;

	CFS_FAIL_TIMEOUT(OBD_FAIL_MGS_CONNECT_NET, cfs_fail_val);
	rc = tgt_connect(tsi);
	if (rc)
		RETURN(rc);

	if (lustre_msg_get_conn_cnt(req->rq_reqmsg) > 1)
		lustre_msg_add_op_flags(req->rq_repmsg, MSG_CONNECT_RECONNECT);

	RETURN(0);
}

static int mgs_disconnect(struct tgt_session_info *tsi)
{
	int rc;

	ENTRY;

	LASSERT(tsi->tsi_exp);

	rc = tgt_disconnect(tsi);
	if (rc)
		RETURN(err_serious(rc));
	RETURN(0);
}

static int mgs_exception(struct tgt_session_info *tsi)
{
	ENTRY;

	tgt_counter_incr(tsi->tsi_exp, LPROC_MGS_EXCEPTION);

	RETURN(0);
}

static inline bool str_starts_with(const char *str, const char *prefix)
{
	return strncmp(str, prefix, strlen(prefix)) == 0;
}

#if LUSTRE_VERSION_CODE < OBD_OCD_VERSION(2, 18, 53, 0)
static int mgs_set_info(struct tgt_session_info *tsi)
{
	struct mgs_thread_info *mgi;
	struct mgs_send_param *msp;
	size_t param_len;
	char *s;

	ENTRY;

	mgi = mgs_env_info(tsi->tsi_env);
	if (IS_ERR(mgi))
		RETURN(err_serious(PTR_ERR(mgi)));

	msp = req_capsule_client_get(tsi->tsi_pill, &RMF_MGS_SEND_PARAM);
	if (msp == NULL)
		RETURN(err_serious(-EFAULT));

	param_len = strnlen(msp->mgs_param, sizeof(msp->mgs_param));
	if (param_len == 0 || param_len == sizeof(msp->mgs_param))
		RETURN(-EINVAL);

	/* We only allow '*.lov.stripe{size,count,offset}=*' from an RPC. */
	s = strchr(msp->mgs_param, '.');
	if (s == NULL)
		RETURN(-EINVAL);

	if (!str_starts_with(s + 1, "lov.stripesize=") &&
	    !str_starts_with(s + 1, "lov.stripecount=") &&
	    !str_starts_with(s + 1, "lov.stripeoffset="))
		RETURN(-EINVAL);

	/* do nothing */
	CDEBUG(D_MGS, "%s: ignoring set info '%s'\n",
	       tgt_name(tsi->tsi_tgt), msp->mgs_param);
	RETURN(0);
}
#endif

enum ast_type {
	AST_CONFIG	= 1,
	AST_PARAMS	= 2,
	AST_IR		= 3,
	AST_BARRIER	= 4,
};

static int mgs_completion_ast_generic(struct ldlm_lock *lock, __u64 flags,
				     void *cbdata, enum ast_type type)
{
	ENTRY;

	if (!(flags & LDLM_FL_BLOCKED_MASK)) {
		struct fs_db *fsdb;

		/* l_ast_data is used as a marker to avoid cancel ldlm lock
		 * twice. See LU-2317.
		 */
		lock_res_and_lock(lock);
		fsdb = (struct fs_db *)lock->l_ast_data;
		lock->l_ast_data = NULL;
		unlock_res_and_lock(lock);

		if (fsdb != NULL) {
			struct lustre_handle lockh;

			switch (type) {
			case AST_CONFIG:
				/* clear the bit before lock put */
				clear_bit(FSDB_REVOKING_LOCK,
					  &fsdb->fsdb_flags);
				break;
			case AST_PARAMS:
				clear_bit(FSDB_REVOKING_PARAMS,
					  &fsdb->fsdb_flags);
				break;
			case AST_IR:
				mgs_ir_notify_complete(fsdb);
				break;
			case AST_BARRIER:
				break;
			default:
				LBUG();
			}

			ldlm_lock2handle(lock, &lockh);
			ldlm_lock_decref_and_cancel(&lockh, LCK_EX);
		}
	}

	RETURN(ldlm_completion_ast(lock, flags, cbdata));
}

static int mgs_completion_ast_config(struct ldlm_lock *lock, __u64 flags,
				     void *cbdata)
{
	return mgs_completion_ast_generic(lock, flags, cbdata, AST_CONFIG);
}

static int mgs_completion_ast_params(struct ldlm_lock *lock, __u64 flags,
				     void *cbdata)
{
	return mgs_completion_ast_generic(lock, flags, cbdata, AST_PARAMS);
}

static int mgs_completion_ast_ir(struct ldlm_lock *lock, __u64 flags,
				 void *cbdata)
{
	return mgs_completion_ast_generic(lock, flags, cbdata, AST_IR);
}

static int mgs_completion_ast_barrier(struct ldlm_lock *lock, __u64 flags,
				      void *cbdata)
{
	return mgs_completion_ast_generic(lock, flags, cbdata, AST_BARRIER);
}

void mgs_revoke_lock(struct mgs_device *mgs, struct fs_db *fsdb,
		     enum mgs_cfg_type type)
{
	ldlm_completion_callback cp = NULL;
	struct lustre_handle     lockh = {
		.cookie = 0,
	};
	struct ldlm_res_id       res_id;
	__u64 flags = LDLM_FL_ATOMIC_CB;
	int rc;

	ENTRY;

	LASSERT(fsdb->fsdb_name[0] != '\0');
	rc = mgc_fsname2resid(fsdb->fsdb_name, &res_id, type);
	LASSERT(rc == 0);
	switch (type) {
	case MGS_CFG_T_CONFIG:
	case MGS_CFG_T_NODEMAP:
		cp = mgs_completion_ast_config;
		if (test_and_set_bit(FSDB_REVOKING_LOCK, &fsdb->fsdb_flags))
			rc = -EALREADY;
		break;
	case MGS_CFG_T_PARAMS:
		cp = mgs_completion_ast_params;
		if (test_and_set_bit(FSDB_REVOKING_PARAMS, &fsdb->fsdb_flags))
			rc = -EALREADY;
		break;
	case MGS_CFG_T_RECOVER:
		cp = mgs_completion_ast_ir;
		break;
	case MGS_CFG_T_BARRIER:
		cp = mgs_completion_ast_barrier;
		break;
	default:
		break;
	}

	if (!rc) {
		LASSERT(cp != NULL);
		rc = ldlm_cli_enqueue_local(NULL, mgs->mgs_obd->obd_namespace,
					    &res_id, LDLM_PLAIN, NULL, LCK_EX,
					    &flags, ldlm_blocking_ast, cp,
					    NULL, fsdb, 0, LVB_T_NONE, NULL,
					    &lockh);
		if (rc != ELDLM_OK) {
			CERROR("%s: can't take cfg lock for %#llx/%#llx : rc = %d\n",
			       mgs->mgs_obd->obd_name,
			       le64_to_cpu(res_id.name[0]),
			       le64_to_cpu(res_id.name[1]), rc);

			if (type == MGS_CFG_T_CONFIG)
				clear_bit(FSDB_REVOKING_LOCK,
					  &fsdb->fsdb_flags);

			if (type == MGS_CFG_T_PARAMS)
				clear_bit(FSDB_REVOKING_PARAMS,
					  &fsdb->fsdb_flags);
		}
		/* lock has been cancelled in completion_ast. */
	}

	RETURN_EXIT;
}

/* Returns: 0 on Success
 * Returns: 1 means update
 * Returns: <0 means error
 */
static int mgs_check_target(const struct lu_env *env,
			    struct mgs_device *mgs,
			    struct mgs_target_info *mti)
{
	int rc;

	ENTRY;

	rc = mgs_check_index(env, mgs, mti);
	if (rc == 0) {
		LCONSOLE_ERROR("%s claims to have registered, but this MGS does not know about it, preventing registration.\n",
			       mti->mti_svname);
		rc = -ENOENT;
	} else if (rc == MGS_ERR_EMPTY_CLIENT_LOG) {
		LCONSOLE_ERROR("Client log %s-client has disappeared! Regenerating all logs.\n",
			       mti->mti_fsname);
		mti->mti_flags |= LDD_F_WRITECONF;
		rc = 1;
	} else {
		/* Index is correctly marked as used */
		rc = 0;
	}

	RETURN(rc);
}

/* Ensure this is not a failover node that is connecting first*/
static int mgs_check_failover_reg(struct mgs_target_info *mti)
{
	struct lnet_nid nid;
	char *ptr;
	int i;

	ptr = mti->mti_params;
	while (class_find_param(ptr, PARAM_FAILNODE, &ptr) == 0) {
		while (class_parse_nid_quiet(ptr, &nid, &ptr) == 0) {
			for (i = 0; i < mti->mti_nid_count; i++) {
				struct lnet_nid nid2;
				int rc;

				if (target_supports_large_nid(mti)) {
					rc = libcfs_strnid(&nid2,
							   mti->mti_nidlist[i]);
					if (rc < 0) {
						LCONSOLE_WARN("NID %s is unsupported type or improper format\n",
							      libcfs_nidstr(&nid));
						return rc;
					}
				} else {
					lnet_nid4_to_nid(mti->mti_nids[i],
							 &nid2);
				}

				if (nid_same(&nid, &nid2)) {
					LCONSOLE_WARN("Denying initial registration attempt from nid %s, specified as failover\n",
						      libcfs_nidstr(&nid));
					return -EADDRNOTAVAIL;
				}
			}
		}
	}
	return 0;
}

/* Called whenever a target starts up.  Flags indicate first connect, etc. */
static int mgs_target_reg(struct tgt_session_info *tsi)
{
	struct obd_device *obd = tsi->tsi_exp->exp_obd;
	struct mgs_device *mgs = exp2mgs_dev(tsi->tsi_exp);
	struct mgs_target_info *mti, *reply_mti, *request_mti;
	struct mgs_target_nidlist *mtn = NULL;
	struct ptlrpc_bulk_desc *desc = NULL;
	struct fs_db *b_fsdb = NULL; /* barrier fsdb */
	struct fs_db *c_fsdb = NULL; /* config fsdb */
	char barrier_name[20];
	size_t mti_buflen, mti_alloc = 0;
	int opc;
	int rc = 0;
	bool nidlist;

	ENTRY;
	rc = lu_env_refill((struct lu_env *)tsi->tsi_env);
	if (rc)
		return err_serious(rc);

	tgt_counter_incr(tsi->tsi_exp, LPROC_MGS_TARGET_REG);

	nidlist = exp_connect_flags(tsi->tsi_exp) & OBD_CONNECT_MGS_NIDLIST;

	request_mti = req_capsule_client_get(tsi->tsi_pill,
					     &RMF_MGS_TARGET_INFO);
	if (!request_mti) {
		DEBUG_REQ(D_HA, tgt_ses_req(tsi), "no mgs_target_info");
		RETURN(err_serious(-EPROTO));
	}
	mti_buflen = req_capsule_get_size(tsi->tsi_pill, &RMF_MGS_TARGET_INFO,
					  RCL_CLIENT);
	if (nidlist) {
		req_capsule_extend(tsi->tsi_pill, &RQF_MGS_TARGET_REG_NIDLIST);
		if (!req_capsule_field_present(tsi->tsi_pill,
					       &RMF_MGS_TARGET_NIDLIST,
					       RCL_CLIENT)) {
			DEBUG_REQ(D_HA, tgt_ses_req(tsi),
				  "no mgs_target_nidlist");
			RETURN(err_serious(-EPROTO));
		}

		/* new protocol with nidlist */
		mtn = req_capsule_client_get(tsi->tsi_pill,
					     &RMF_MGS_TARGET_NIDLIST);

		/* Initial request may be with NIDs in mti, in that case
		 * NIDLIST buffer has just header without NIDs.
		 * Check it to avoid mismatch with client and set nidlist
		 * to false if so. Exception is a bulk request.
		 */
		if (req_capsule_get_size(tsi->tsi_pill,
					 &RMF_MGS_TARGET_NIDLIST,
					 RCL_CLIENT) <= sizeof(*mtn))
			nidlist = !!(mtn->mtn_flags & NIDLIST_IN_BULK);
	}

	/* Compatibility code for older targets, process mti as is */
	if (!nidlist || !target_supports_large_nid(request_mti)) {
		int limit;

		mti = request_mti;
		/* sanity check for mti_nid_count */
		if (mti_buflen > sizeof(*mti))
			limit = (mti_buflen - sizeof(*mti)) / MTN_NIDSTR_SIZE;
		else
			limit = MTI_NIDS_MAX;

		if (mti->mti_nid_count > limit) {
			CWARN("%s: bad NID count in mti: %d, req limit: %d\n",
			      mti->mti_svname, mti->mti_nid_count, limit);
			mti->mti_nid_count = limit;
		}
		goto process;
	}

	mti_alloc = sizeof(*mti) + NIDLIST_SIZE(mtn->mtn_nids);
	OBD_ALLOC_LARGE(mti, mti_alloc);
	if (!mti)
		RETURN(err_serious(-ENOMEM));

	if (mtn->mtn_flags & NIDLIST_IN_BULK) {
		int pages;
		size_t nidlist_size = NIDLIST_SIZE(mtn->mtn_nids);

		pages = DIV_ROUND_UP((sizeof(*mti) & ~PAGE_MASK) +
				     nidlist_size, PAGE_SIZE);
		desc = ptlrpc_prep_bulk_exp(tsi->tsi_pill->rc_req,
					    pages, PTLRPC_BULK_OPS_COUNT,
					    PTLRPC_BULK_GET_SINK,
					    MGS_BULK_PORTAL,
					    &ptlrpc_bulk_kiov_nopin_ops);
		if (!desc)
			GOTO(out_mti_free, rc = err_serious(-ENOMEM));

		desc->bd_frag_ops->add_iov_frag(desc, mti->mti_nidlist,
						nidlist_size);
		tsi->tsi_pill->rc_req->rq_bulk_write = 1;
		rc = sptlrpc_svc_prep_bulk(tsi->tsi_pill->rc_req, desc);
		if (rc != 0)
			GOTO(out_free, rc = err_serious(rc));

		rc = target_bulk_io(tsi->tsi_pill->rc_req->rq_export, desc);
		if (rc < 0)
			GOTO(out_free, rc = err_serious(rc));
	} else {
		memcpy(mti->mti_nidlist, mtn->mtn_inline_list,
		       NIDLIST_SIZE(mtn->mtn_nids));
	}
	*mti = *request_mti;
	mti->mti_nid_count = mtn->mtn_nids;
	mti->mti_flags |= LDD_F_LARGE_NID;

process:
	/* at this point all NIDs are in mti */
	down_read(&mgs->mgs_barrier_rwsem);

	if (OCD_HAS_FLAG(&tsi->tsi_exp->exp_connect_data, IMP_RECOV))
		opc = mti->mti_flags & LDD_F_OPC_MASK;
	else
		opc = LDD_F_OPC_REG;

	if (opc == LDD_F_OPC_READY) {
		CDEBUG(D_MGS, "fs: %s index: %d is ready to reconnect.\n",
			mti->mti_fsname, mti->mti_stripe_index);
		rc = mgs_ir_update(tsi->tsi_env, mgs, mti);
		if (rc) {
			LASSERT(!(mti->mti_flags & LDD_F_IR_CAPABLE));
			CERROR("%s: Update IR return failure: rc = %d\n",
			       mti->mti_fsname, rc);
		}

		GOTO(out_norevoke, rc);
	}

	/* Do not support unregistering right now. */
	if (opc != LDD_F_OPC_REG)
		GOTO(out_norevoke, rc = -EINVAL);

	snprintf(barrier_name, sizeof(barrier_name) - 1, "%s-%s",
		 mti->mti_fsname, BARRIER_FILENAME);
	rc = mgs_find_or_make_fsdb(tsi->tsi_env, mgs, barrier_name, &b_fsdb);
	if (rc) {
		CERROR("%s: Can't get db for %s: rc = %d\n",
		       mti->mti_fsname, barrier_name, rc);

		GOTO(out_norevoke, rc);
	}

	CDEBUG(D_MGS, "fs: %s index: %d is registered to MGS.\n",
	       mti->mti_fsname, mti->mti_stripe_index);

	if (mti->mti_flags & LDD_F_SV_TYPE_MDT) {
		if (b_fsdb->fsdb_barrier_status == BS_FREEZING_P1 ||
		    b_fsdb->fsdb_barrier_status == BS_FREEZING_P2 ||
		    b_fsdb->fsdb_barrier_status == BS_FROZEN) {
			LCONSOLE_WARN("%s: the system is in barrier, refuse the connection from MDT %s temporary\n",
				      obd->obd_name, mti->mti_svname);

			GOTO(out_norevoke, rc = -EBUSY);
		}

		if (!(exp_connect_flags(tsi->tsi_exp) & OBD_CONNECT_BARRIER) &&
		    !b_fsdb->fsdb_barrier_disabled) {
			LCONSOLE_WARN("%s: the MDT %s does not support write barrier, so disable barrier on the whole system.\n",
				      obd->obd_name, mti->mti_svname);

			b_fsdb->fsdb_barrier_disabled = 1;
		}
	}

	if (mti->mti_flags & LDD_F_NEED_INDEX)
		mti->mti_flags |= LDD_F_WRITECONF;

	/* Check if new target registration is allowed */
	if (!allow_register) {
		rc = mgs_check_index(tsi->tsi_env, mgs, mti);
		if (rc == 0) {
			/* Target index not found - this is a new target */
			rc = -EACCES;
			LCONSOLE_ERROR("%s: New target registration disabled. Use 'lctl set_param allow_register=1' to allow new target registrations.\n",
				       mti->mti_svname);
			GOTO(out_norevoke, rc);
		}

		if (rc < 0 && rc != MGS_ERR_EMPTY_CLIENT_LOG) {
			/* Error checking index */
			CERROR("%s: Error checking target index: rc = %d\n",
			       mti->mti_svname, rc);
			GOTO(out_norevoke, rc);
		}
		/* rc == 1 means target was previously registered, allow it */
		/* rc == MGS_ERR_EMPTY_CLIENT_LOG:
		 * empty client log, will be handled below
		 */
	}

	if (!(mti->mti_flags & (LDD_F_WRITECONF | LDD_F_UPDATE))) {
		/* We're just here as a startup ping. */
		CDEBUG(D_MGS, "Server %s is running on %s\n",
		       mti->mti_svname, obd_export_nid2str(tsi->tsi_exp));
		rc = mgs_check_target(tsi->tsi_env, mgs, mti);
		/* above will set appropriate mti flags */
		if (rc <= 0)
			/* Nothing wrong, or fatal error */
			GOTO(out_norevoke, rc);
	} else if (!(mti->mti_flags & LDD_F_NO_PRIMNODE)) {
		rc = mgs_check_failover_reg(mti);
		if (rc)
			GOTO(out_norevoke, rc);
	}

	CFS_FAIL_TIMEOUT(OBD_FAIL_MGS_PAUSE_TARGET_REG, 10);

	if (mti->mti_flags & LDD_F_WRITECONF) {
		if (mti->mti_flags & LDD_F_SV_TYPE_MDT &&
		    mti->mti_stripe_index == 0) {
			mgs_put_fsdb(mgs, b_fsdb);
			b_fsdb = NULL;
			rc = mgs_erase_logs(tsi->tsi_env, mgs,
					    mti->mti_fsname);
			LCONSOLE_WARN("%s: Logs for fs %s were removed by user request.  All servers must be restarted in order to regenerate the logs: rc = %d\n",
				      obd->obd_name, mti->mti_fsname, rc);
			if (rc && rc != -ENOENT)
				GOTO(out_norevoke, rc);

			rc = mgs_find_or_make_fsdb(tsi->tsi_env, mgs,
						   barrier_name, &b_fsdb);
			if (rc) {
				CERROR("Can't get db for %s: %d\n",
				       barrier_name, rc);

				GOTO(out_norevoke, rc);
			}

			if (!(exp_connect_flags(tsi->tsi_exp) &
			      OBD_CONNECT_BARRIER)) {
				LCONSOLE_WARN("%s: the MDT %s does not support write barrier, disable barrier on the whole system.\n",
					      obd->obd_name, mti->mti_svname);

				b_fsdb->fsdb_barrier_disabled = 1;
			}
		} else if (mti->mti_flags &
			   (LDD_F_SV_TYPE_OST | LDD_F_SV_TYPE_MDT)) {
			rc = mgs_erase_log(tsi->tsi_env, mgs, mti->mti_svname);
			LCONSOLE_WARN("%s: Regenerating %s log by user request: rc = %d\n",
				      obd->obd_name, mti->mti_svname, rc);
			if (rc)
				GOTO(out_norevoke, rc);
		}

		mti->mti_flags |= LDD_F_UPDATE;
	}

	rc = mgs_find_or_make_fsdb(tsi->tsi_env, mgs, mti->mti_fsname, &c_fsdb);
	if (rc) {
		CERROR("Can't get db for %s: %d\n", mti->mti_fsname, rc);

		GOTO(out_norevoke, rc);
	}

	/*
	 * Log writing contention is handled by the fsdb_mutex.
	 *
	 * It should be alright if someone was reading while we were
	 * updating the logs - if we revoke at the end they will just update
	 * from where they left off.
	 */
	if (mti->mti_flags & LDD_F_UPDATE) {
		CDEBUG(D_MGS, "updating %s, index=%d\n", mti->mti_svname,
		       mti->mti_stripe_index);

		/* create/update target log and update the client/mdt logs */
		rc = mgs_write_log_target(tsi->tsi_env, mgs, mti, c_fsdb);
		if (rc) {
			CERROR("Failed to write %s log (%d)\n",
			       mti->mti_svname, rc);
			GOTO(out, rc);
		}

		mti->mti_flags &= ~(LDD_F_VIRGIN | LDD_F_UPDATE |
				    LDD_F_NEED_INDEX | LDD_F_WRITECONF);
		mti->mti_flags |= LDD_F_REWRITE_LDD;
	}

out:
	mgs_revoke_lock(mgs, c_fsdb, MGS_CFG_T_CONFIG);

out_norevoke:
	if (!rc && mti->mti_flags & LDD_F_SV_TYPE_MDT && b_fsdb) {
		if (!c_fsdb) {
			rc = mgs_find_or_make_fsdb(tsi->tsi_env, mgs,
						   mti->mti_fsname, &c_fsdb);
			if (rc)
				CERROR("Fail to get db for %s: %d\n",
				       mti->mti_fsname, rc);
		}

		if (c_fsdb) {
			memcpy(b_fsdb->fsdb_mdt_index_map,
			       c_fsdb->fsdb_mdt_index_map, INDEX_MAP_SIZE);
			b_fsdb->fsdb_mdt_count = c_fsdb->fsdb_mdt_count;
		}
	}

	up_read(&mgs->mgs_barrier_rwsem);

	CDEBUG(D_MGS, "replying with %s, index=%d, rc=%d\n", mti->mti_svname,
	       mti->mti_stripe_index, rc);
	 /* An error flag is set in the mti reply rather than an error code */
	if (rc)
		mti->mti_flags |= LDD_F_ERROR;

	/* Compatibility code:
	 * if large mti was received, send back the same buffer size as that
	 * MGC expects, so avoid buffer size mismatch errors on MGC side
	 */
	if (mti_buflen > sizeof(*mti)) {
		int err;

		err = req_capsule_server_grow(tsi->tsi_pill,
					      &RMF_MGS_TARGET_INFO, mti_buflen);
		if (err < 0)
			GOTO(out_fsdb, rc = err_serious(err));
	}
	reply_mti = req_capsule_server_get(tsi->tsi_pill, &RMF_MGS_TARGET_INFO);
	*reply_mti = *mti;

	/* Flush logs to disk */
	dt_sync(tsi->tsi_env, mgs->mgs_bottom);

out_fsdb:
	if (b_fsdb)
		mgs_put_fsdb(mgs, b_fsdb);
	if (c_fsdb)
		mgs_put_fsdb(mgs, c_fsdb);
out_free:
	ptlrpc_free_bulk(desc);
out_mti_free:
	if (mti_alloc)
		OBD_FREE_LARGE(mti, mti_alloc);

	RETURN(rc);
}

/* Called whenever a target cleans up. */
static int mgs_target_del(struct tgt_session_info *tsi)
{
	ENTRY;

	tgt_counter_incr(tsi->tsi_exp, LPROC_MGS_TARGET_DEL);

	RETURN(0);
}

static int mgs_config_read(struct tgt_session_info *tsi)
{
	struct ptlrpc_request	*req = tgt_ses_req(tsi);
	struct mgs_config_body	*body;
	int			 rc;

	ENTRY;

	body = req_capsule_client_get(tsi->tsi_pill, &RMF_MGS_CONFIG_BODY);
	if (body == NULL) {
		DEBUG_REQ(D_HA, req, "no mgs_config_body");
		RETURN(err_serious(-EFAULT));
	}

	switch (body->mcb_type) {
	case MGS_CFG_T_RECOVER:
		rc = mgs_get_ir_logs(req);
		break;
	case MGS_CFG_T_NODEMAP:
		rc = nodemap_get_config_req(req->rq_export->exp_obd, req);
		break;
	case MGS_CFG_T_CONFIG:
		rc = -EOPNOTSUPP;
		break;
	default:
		rc = -EINVAL;
		break;
	}

	RETURN(rc);
}

static int mgs_llog_open(struct tgt_session_info *tsi)
{
	struct mgs_thread_info	*mgi;
	struct ptlrpc_request	*req = tgt_ses_req(tsi);
	char			*logname;
	int			 rc;

	ENTRY;

	rc = tgt_llog_open(tsi);
	if (rc)
		RETURN(rc);

	/*
	 * For old clients there is no direct way of knowing which file system
	 * a client is operating at the MGS side. But we need to pick up those
	 * clients so that the MGS can mark the corresponding file system as
	 * non-IR capable because old clients are not ready to be notified.
	 *
	 * Therefore we attempt to detect the file systems name by hacking the
	 * llog operation which is currently used by the clients to fetch
	 * configuration logs. At present this is fine because this is the
	 * ONLY llog operation between mgc and the MGS.
	 *
	 * If extra llog operation are going to be added, this function needs
	 * further work.
	 *
	 * When releases prior than 2.0 are not supported, the following code
	 * can be removed.
	 */
	mgi = mgs_env_info(tsi->tsi_env);
	if (IS_ERR(mgi))
		RETURN(PTR_ERR(mgi));

	logname = req_capsule_client_get(tsi->tsi_pill, &RMF_NAME);
	if (logname) {
		char *ptr = strrchr(logname, '-');
		int   len = (ptr != NULL) ? (int)(ptr - logname) : 0;

		if (ptr == NULL || len >= sizeof(mgi->mgi_fsname)) {
			if (strcmp(logname, PARAMS_FILENAME) != 0)
				LCONSOLE_WARN("%s: non-config logname received: %s\n",
					      tgt_name(tsi->tsi_tgt),
					      logname);
			/* not error, this can be llog test name */
		} else {
			strncpy(mgi->mgi_fsname, logname, len);
			mgi->mgi_fsname[len] = 0;

			rc = mgs_fsc_attach(tsi->tsi_env, tsi->tsi_exp,
					    mgi->mgi_fsname);
			if (rc && rc != -EEXIST) {
				LCONSOLE_WARN("%s: Unable to add client %s to file system %s: %d\n",
					      tgt_name(tsi->tsi_tgt),
					      libcfs_nidstr(&req->rq_peer.nid),
					      mgi->mgi_fsname, rc);
			} else {
				rc = 0;
			}
		}
	} else {
		CERROR("%s: no logname in request\n", tgt_name(tsi->tsi_tgt));
		RETURN(-EINVAL);
	}
	RETURN(rc);
}

static inline int mgs_init_export(struct obd_export *exp)
{
	struct mgs_export_data *data = &exp->u.eu_mgs_data;

	/* init mgs_export_data for fsc */
	spin_lock_init(&data->med_lock);
	INIT_LIST_HEAD(&data->med_clients);

	spin_lock(&exp->exp_lock);
	exp->exp_connecting = 1;
	spin_unlock(&exp->exp_lock);

	/* self-export doesn't need client data and ldlm initialization */
	if (unlikely(obd_uuid_equals(&exp->exp_obd->obd_uuid,
				     &exp->exp_client_uuid)))
		return 0;

	spin_lock_init(&exp->exp_target_data.ted_nodemap_lock);
	INIT_LIST_HEAD(&exp->exp_target_data.ted_nodemap_member);

	return ldlm_init_export(exp);
}

static inline int mgs_destroy_export(struct obd_export *exp)
{
	ENTRY;

	target_destroy_export(exp);
	mgs_client_free(exp);

	if (unlikely(obd_uuid_equals(&exp->exp_obd->obd_uuid,
				     &exp->exp_client_uuid)))
		RETURN(0);

	ldlm_destroy_export(exp);

	RETURN(0);
}

static int mgs_extract_fs_pool(char *arg, char *fsname, char *poolname)
{
	size_t len;
	char *ptr;

	ENTRY;
	/* Validate name */
	for (ptr = arg; *ptr != '\0'; ptr++) {
		if (!isalnum(*ptr) && *ptr != '_' && *ptr != '-' && *ptr != '.')
			return -EINVAL;
	}

	/* Test for fsname.poolname format. strlen test if poolname is empty */
	ptr = strchr(arg, '.');
	if (!ptr || !strlen(ptr))
		return -EINVAL;
	ptr++;

	/* Check pool name validity. */
	if (ptr[0] == '\0' || lov_pool_is_reserved(ptr))
		return -EINVAL;
	/* Also make sure poolname is not to long. */
	if (strlen(ptr) > LOV_MAXPOOLNAME)
		return -ENAMETOOLONG;
	strscpy(poolname, ptr, LOV_MAXPOOLNAME + 1);

	/* Test if fsname is empty */
	len = strlen(arg) - strlen(ptr) - 1;
	if (!len)
		return -EINVAL;

	/* or too long */
	if (len > LUSTRE_MAXFSNAME)
		return -ENAMETOOLONG;

	strncpy(fsname, arg, len);

	RETURN(0);
}

/**
 * __llog_fileset_cleanup_apply() - Forge the lustre_cfg to disable
 * nodemap.NM_NAME.fileset on all server targets by invoking "mgs_set_param()"
 *
 * @env: thread context
 * @mgs: mgs device
 * @nodemap_name: name of the nodemap to cleanup
 *
 * Return:
 * * %0 on success
 * * %-negative error code on failure
 */
static int __llog_fileset_cleanup_apply(const struct lu_env *env,
					struct mgs_device *mgs,
					const char *nodemap_name)
{
	struct lustre_cfg_bufs *bufs = NULL;
	struct lustre_cfg *lcfg = NULL;
	char *lcfg_param = NULL;
	const char *lcfg_format;
	size_t lcfg_param_size;
	int rc = 0;

	if (!nodemap_name || nodemap_name[0] == '\0')
		RETURN(-EINVAL);

	lcfg_format = "nodemap.%s.fileset=";
	lcfg_param_size = snprintf(NULL, 0, lcfg_format, nodemap_name) + 1;

	OBD_ALLOC(lcfg_param, lcfg_param_size);
	if (!lcfg_param)
		RETURN(-ENOMEM);

	snprintf(lcfg_param, lcfg_param_size, lcfg_format, nodemap_name);

	OBD_ALLOC_PTR(bufs);
	if (!bufs)
		GOTO(out_cleanup, rc = -ENOMEM);

	/* lcfg for all targets */
	lustre_cfg_bufs_reset(bufs, LUSTRE_CFG_ALL_TARGETS);
	lustre_cfg_bufs_set_string(bufs, 1, lcfg_param);

	OBD_ALLOC(lcfg, lustre_cfg_len(bufs->lcfg_bufcount, bufs->lcfg_buflen));
	if (!lcfg)
		GOTO(out_cleanup, rc = -ENOMEM);

	lustre_cfg_init(lcfg, LCFG_SET_PARAM, bufs);

	rc = mgs_set_param(env, mgs, lcfg);
	if (rc == -ENOENT)
		rc = 0;

out_cleanup:
	if (lcfg)
		OBD_FREE(lcfg, lustre_cfg_len(lcfg->lcfg_bufcount,
					      lcfg->lcfg_buflens));
	/* lustre_cfg maintains its own buffers */
	OBD_FREE_PTR(bufs);
	OBD_FREE(lcfg_param, lcfg_param_size);

	return rc;
}

/**
 * mgs_llog_fileset_cleanup() - Cleanup the fileset entry from the params
 * llog on all server targets
 *
 * @env: thread context
 * @mgs: mgs device
 * @data: ioctl data containing the nodemap name
 *
 * This function is necessary to provide backward compatibility with old
 * clients versions that still use the params llog to set the fileset. If
 * the new ioctl based nodemap fileset functions are used, the llog entry
 * would not be removed, and could re-appear for a new similar named nodemap.
 *
 * Return:
 * * %0 on success
 * * %-negative error code on failure
 */
static int mgs_llog_fileset_cleanup(const struct lu_env *env,
				    struct mgs_device *mgs,
				    struct obd_ioctl_data *data)
{
	struct lustre_cfg *lcfg_in;
	char *nodemap_name;
	int rc;

	if (data->ioc_plen1 > PAGE_SIZE)
		GOTO(out, rc = -E2BIG);

	OBD_ALLOC(lcfg_in, data->ioc_plen1);
	if (!lcfg_in)
		GOTO(out, rc = -ENOMEM);

	if (copy_from_user(lcfg_in, data->ioc_pbuf1, data->ioc_plen1))
		GOTO(out_cleanup, rc = -EFAULT);
	if (lustre_cfg_sanity_check(lcfg_in, data->ioc_plen1))
		GOTO(out_cleanup, rc = -EINVAL);

	if (lcfg_in->lcfg_bufcount < 2)
		GOTO(out_cleanup, rc = -EINVAL);

	nodemap_name = lustre_cfg_string(lcfg_in, 1);
	rc = __llog_fileset_cleanup_apply(env, mgs, nodemap_name);
	if (rc)
		CWARN("%s: failed to cleanup llog fileset for nodemap %s: %d\n",
		      mgs->mgs_obd->obd_name, nodemap_name, rc);

out_cleanup:
	OBD_FREE(lcfg_in, data->ioc_plen1);
out:
	return rc;
}

/**
 * mgs_has_local_targets() - check if MGS is co-located with an MDT or OST
 *
 * Check if there are any MDT or OST targets on the MGS node. This is done by
 * checking whether the device types are registered.
 *
 * Return: true if MDT or OST type exists, false otherwise
 */
static bool mgs_has_local_targets(void)
{
	struct obd_type *mdt_type;
	struct obd_type *ost_type;
	bool has_targets = false;

	mdt_type = class_search_type(LUSTRE_MDT_NAME);
	if (mdt_type) {
		kobject_put(&mdt_type->typ_kobj);
		has_targets = true;
	}

	if (!has_targets) {
		ost_type = class_search_type(LUSTRE_OST_NAME);
		if (ost_type) {
			kobject_put(&ost_type->typ_kobj);
			has_targets = true;
		}
	}

	return has_targets;
}

static int mgs_iocontrol_nodemap(const struct lu_env *env,
				 struct mgs_device *mgs,
				 struct obd_ioctl_data *data)
{
	bool clean_llog_fileset = false;
	bool dynamic = false;
	bool ro_cmd = false;
	struct fs_db *fsdb;
	int rc;

	ENTRY;

	if (data->ioc_type != LUSTRE_CFG_TYPE) {
		CERROR("%s: unknown cfg record type: %d\n",
		       mgs->mgs_obd->obd_name, data->ioc_type);
		GOTO(out, rc = -EINVAL);
	}

	rc = server_iocontrol_nodemap(mgs->mgs_obd, data, &dynamic,
				      &clean_llog_fileset, &ro_cmd);
	if (rc)
		GOTO(out, rc);

	/* For dyn. nodemap and ro commands, skip nodemap config distribution */
	if (dynamic || ro_cmd)
		GOTO(out, rc);

	/* A llog fileset entry might still exist and needs to be removed */
	if (clean_llog_fileset) {
		int rc2;

		/* Attempt to clean up the llog fileset and provide a warning.
		 * It is not serious enough to fail the original request which
		 * was already applied on the MGS above.
		 */
		rc2 = mgs_llog_fileset_cleanup(env, mgs, data);
		if (rc2)
			CWARN("%s: failed to cleanup llog fileset: %d\n",
			      mgs->mgs_obd->obd_name, rc2);
	}

	/* if MGS is co-located with an MDT or OST, clear dynamic nodemaps */
	if (nodemap_has_dynamic_nodemaps() && mgs_has_local_targets())
		nodemap_clear_dynamic_nodemaps();

	/* revoke nodemap lock */
	rc = mgs_find_or_make_fsdb(env, mgs, LUSTRE_NODEMAP_NAME, &fsdb);
	if (rc < 0) {
		CWARN("%s: cannot make nodemap fsdb: rc = %d\n",
		      mgs->mgs_obd->obd_name, rc);
	} else {
		/* require targets to fetch the nodemap config from MGS */
		mgs_revoke_lock(mgs, fsdb, MGS_CFG_T_NODEMAP);
		mgs_put_fsdb(mgs, fsdb);
	}

out:
	RETURN(rc);
}

static int mgs_iocontrol_pool(const struct lu_env *env,
			      struct mgs_device *mgs,
			      struct obd_ioctl_data *data)
{
	struct mgs_thread_info *mgi = mgs_env_info(env);
	int rc;
	struct lustre_cfg *lcfg = NULL;
	char *poolname = NULL;

	ENTRY;

	OBD_ALLOC(poolname, LOV_MAXPOOLNAME + 1);
	if (poolname == NULL)
		RETURN(-ENOMEM);

	if (data->ioc_type != LUSTRE_CFG_TYPE) {
		CERROR("%s: unknown cfg record type: %d\n",
		       mgs->mgs_obd->obd_name, data->ioc_type);
		GOTO(out_pool, rc = -EINVAL);
	}

	if (data->ioc_plen1 > PAGE_SIZE)
		GOTO(out_pool, rc = -E2BIG);

	OBD_ALLOC(lcfg, data->ioc_plen1);
	if (lcfg == NULL)
		GOTO(out_pool, rc = -ENOMEM);

	if (copy_from_user(lcfg, data->ioc_pbuf1, data->ioc_plen1))
		GOTO(out_lcfg, rc = -EFAULT);

	rc = lustre_cfg_sanity_check(lcfg, data->ioc_plen1);
	if (rc)
		GOTO(out_lcfg, rc);

	if (lcfg->lcfg_bufcount < 2)
		GOTO(out_lcfg, rc = -EINVAL);

	/* first arg is always <fsname>.<poolname> */
	rc = mgs_extract_fs_pool(lustre_cfg_string(lcfg, 1), mgi->mgi_fsname,
				 poolname);
	if (rc)
		GOTO(out_lcfg, rc);

	switch (lcfg->lcfg_command) {
	case LCFG_POOL_NEW:
		if (lcfg->lcfg_bufcount != 2)
			GOTO(out_lcfg, rc = -EINVAL);
		rc = mgs_pool_cmd(env, mgs, LCFG_POOL_NEW, mgi->mgi_fsname,
				  poolname, NULL);
		break;
	case LCFG_POOL_ADD:
		if (lcfg->lcfg_bufcount != 3)
			GOTO(out_lcfg, rc = -EINVAL);
		rc = mgs_pool_cmd(env, mgs, LCFG_POOL_ADD, mgi->mgi_fsname,
				  poolname, lustre_cfg_string(lcfg, 2));
		break;
	case LCFG_POOL_REM:
		if (lcfg->lcfg_bufcount != 3)
			GOTO(out_lcfg, rc = -EINVAL);
		rc = mgs_pool_cmd(env, mgs, LCFG_POOL_REM, mgi->mgi_fsname,
				  poolname, lustre_cfg_string(lcfg, 2));
		break;
	case LCFG_POOL_DEL:
		if (lcfg->lcfg_bufcount != 2)
			GOTO(out_lcfg, rc = -EINVAL);
		rc = mgs_pool_cmd(env, mgs, LCFG_POOL_DEL, mgi->mgi_fsname,
				  poolname, NULL);
		break;
	default:
		 rc = -EINVAL;
	}

	if (rc) {
		CERROR("OBD_IOC_POOL err %d, cmd %X for pool %s.%s\n",
		       rc, lcfg->lcfg_command, mgi->mgi_fsname, poolname);
		GOTO(out_lcfg, rc);
	}

out_lcfg:
	OBD_FREE(lcfg, data->ioc_plen1);
out_pool:
	OBD_FREE(poolname, LOV_MAXPOOLNAME + 1);
	RETURN(rc);
}

/* from mdt_iocontrol */
static int mgs_iocontrol(unsigned int cmd, struct obd_export *exp, int len,
			 void *karg, void __user *uarg)
{
	struct obd_device *obd = exp->exp_obd;
	struct mgs_device *mgs = exp2mgs_dev(exp);
	struct obd_ioctl_data *data;
	struct lu_env env;
	int rc = -EINVAL;

	ENTRY;
	CDEBUG(D_IOCTL, "%s: cmd=%x len=%u karg=%pK uarg=%pK\n",
	       obd->obd_name, cmd, len, karg, uarg);
	if (unlikely(karg == NULL))
		RETURN(OBD_IOC_ERROR(obd->obd_name, cmd, "karg=NULL", rc));
	data = karg;

	rc = lu_env_init(&env, LCT_MG_THREAD);
	if (rc)
		RETURN(rc);
	rc = lu_env_add(&env);
	if (unlikely(rc))
		GOTO(out_fini, rc);

	rc = -EINVAL;
	switch (cmd) {
	case OBD_IOC_PARAM: {
		struct lustre_cfg *lcfg;

		if (data->ioc_type != LUSTRE_CFG_TYPE) {
			CERROR("%s: unknown cfg record type '%x': rc = %d\n",
			       obd->obd_name, data->ioc_type, rc);
			GOTO(out, rc);
		}

		OBD_ALLOC(lcfg, data->ioc_plen1);
		if (lcfg == NULL)
			GOTO(out, rc = -ENOMEM);
		if (copy_from_user(lcfg, data->ioc_pbuf1, data->ioc_plen1))
			GOTO(out_free, rc = -EFAULT);

		rc = lustre_cfg_sanity_check(lcfg, data->ioc_plen1);
		if (rc)
			GOTO(out_free, rc);

		if (lcfg->lcfg_bufcount < 1)
			GOTO(out_free, rc = -EINVAL);

		rc = mgs_set_param(&env, mgs, lcfg);
		if (rc)
			CERROR("%s: setparam err: rc = %d\n",
			       obd->obd_name, rc);
out_free:
		OBD_FREE(lcfg, data->ioc_plen1);
		break;
	}
	case OBD_IOC_REPLACE_NIDS:
		if (!data->ioc_inllen1 || !data->ioc_inlbuf1) {
			CERROR("%s: no device or fsname specified: rc = %d\n",
			       obd->obd_name, rc);
			break;
		}

		if (data->ioc_inllen1 > MTI_NAME_MAXLEN) {
			rc = -EOVERFLOW;
			CERROR("%s: device or fsname is too long: rc = %d\n",
			       obd->obd_name, rc);
			break;
		}

		if (data->ioc_inlbuf1[data->ioc_inllen1 - 1] != 0) {
			CERROR("%s: unterminated device or fsname: rc = %d\n",
			       obd->obd_name, rc);
			break;
		}

		if (!data->ioc_inllen2 || !data->ioc_inlbuf2) {
			CERROR("%s: no NIDs specified: rc = %d\n",
			       obd->obd_name, rc);
			break;
		}

		if (data->ioc_inlbuf2[data->ioc_inllen2 - 1] != 0) {
			CERROR("%s: NID list is not NUL terminated: rc = %d\n",
			       obd->obd_name, rc);
			break;
		}

		/* replace nids in llog */
		rc = mgs_replace_nids(&env, mgs, data->ioc_inlbuf1,
				      data->ioc_inlbuf2);
		if (rc)
			CERROR("%s: error replacing NIDs for '%s': rc = %d\n",
			       obd->obd_name, data->ioc_inlbuf1, rc);

		break;
	case OBD_IOC_CLEAR_CONFIGS:
		if (!data->ioc_inllen1 || !data->ioc_inlbuf1) {
			CERROR("%s: no device or fsname specified: rc = %d\n",
			       obd->obd_name, rc);
			break;
		}

		if (data->ioc_inllen1 > MTI_NAME_MAXLEN) {
			rc = -EOVERFLOW;
			CERROR("%s: device or fsname is too long: rc = %d\n",
			       obd->obd_name, rc);
			break;
		}

		if (data->ioc_inlbuf1[data->ioc_inllen1 - 1] != 0) {
			CERROR("%s: unterminated device or fsname: rc = %d\n",
			       obd->obd_name, rc);
			break;
		}

		/* remove records marked SKIP from config logs */
		rc = mgs_clear_configs(&env, mgs, data->ioc_inlbuf1);
		if (rc)
			CERROR("%s: error clearing config log: rc = %d\n",
			       obd->obd_name, rc);

		break;
	case OBD_IOC_POOL:
		rc = mgs_iocontrol_pool(&env, mgs, data);
		break;
#ifdef OBD_IOC_BARRIER
	case_OBD_IOC_DEPRECATED_FT(OBD_IOC_BARRIER, obd->obd_name, 2, 17);
#endif
	case OBD_IOC_BARRIER_V2:
		rc = mgs_iocontrol_barrier(&env, mgs, data);
		break;
	case OBD_IOC_NODEMAP:
		rc = mgs_iocontrol_nodemap(&env, mgs, data);
		break;
	case OBD_IOC_LCFG_FORK:
		rc = mgs_lcfg_fork(&env, mgs, data->ioc_inlbuf1,
				   data->ioc_inlbuf2);
		break;
	case OBD_IOC_LCFG_ERASE:
		rc = mgs_lcfg_erase(&env, mgs, data->ioc_inlbuf1);
		break;
	case OBD_IOC_CATLOGLIST:
		rc = mgs_list_logs(&env, mgs, data);
		break;
	case OBD_IOC_LLOG_CANCEL:
	case OBD_IOC_LLOG_REMOVE:
	case OBD_IOC_LLOG_CHECK:
	case OBD_IOC_LLOG_INFO:
	case OBD_IOC_LLOG_PRINT: {
		struct llog_ctxt *ctxt;

		ctxt = llog_get_context(mgs->mgs_obd, LLOG_CONFIG_ORIG_CTXT);
		rc = llog_ioctl(&env, ctxt, cmd, data);
		llog_ctxt_put(ctxt);
		break;
	}
	default:
		rc = OBD_IOC_ERROR(obd->obd_name, cmd, "unrecognized", -ENOTTY);
		break;
	}
out:
	lu_env_remove(&env);
out_fini:
	lu_env_fini(&env);
	RETURN(rc);
}

static int mgs_connect_to_osd(struct mgs_device *m, const char *nextdev)
{
	struct obd_connect_data *data = NULL;
	struct obd_device       *obd;
	int                      rc;

	ENTRY;

	OBD_ALLOC_PTR(data);
	if (data == NULL)
		RETURN(-ENOMEM);

	obd = class_name2obd(nextdev);
	if (obd == NULL) {
		CERROR("can't locate next device: %s\n", nextdev);
		GOTO(out, rc = -ENOTCONN);
	}

	data->ocd_version = LUSTRE_VERSION_CODE;

	rc = obd_connect(NULL, &m->mgs_bottom_exp, obd,
			 &obd->obd_uuid, data, NULL);
	if (rc) {
		CERROR("cannot connect to next dev %s (%d)\n", nextdev, rc);
		GOTO(out, rc);
	}

	m->mgs_bottom = lu2dt_dev(m->mgs_bottom_exp->exp_obd->obd_lu_dev);
	m->mgs_dt_dev.dd_lu_dev.ld_site = m->mgs_bottom->dd_lu_dev.ld_site;
	LASSERT(m->mgs_dt_dev.dd_lu_dev.ld_site);
out:
	OBD_FREE_PTR(data);
	RETURN(rc);
}

static struct tgt_handler mgs_mgs_handlers[] = {
TGT_RPC_HANDLER(MGS_FIRST_OPC,
		0,			MGS_CONNECT,	 mgs_connect,
		&RQF_CONNECT, LUSTRE_OBD_VERSION),
TGT_RPC_HANDLER(MGS_FIRST_OPC,
		0,			MGS_DISCONNECT,	 mgs_disconnect,
		&RQF_MDS_DISCONNECT, LUSTRE_OBD_VERSION),
TGT_MGS_HDL_VAR(0,			MGS_EXCEPTION,	 mgs_exception),
#if LUSTRE_VERSION_CODE < OBD_OCD_VERSION(2, 18, 53, 0)
TGT_MGS_HDL(HAS_REPLY | IS_MUTABLE,	MGS_SET_INFO,	 mgs_set_info),
#endif
TGT_MGS_HDL(HAS_REPLY | IS_MUTABLE,	MGS_TARGET_REG,	 mgs_target_reg),
TGT_MGS_HDL_VAR(0,			MGS_TARGET_DEL,	 mgs_target_del),
TGT_MGS_HDL(HAS_REPLY,			MGS_CONFIG_READ, mgs_config_read),
};

static struct tgt_handler mgs_obd_handlers[] = {
TGT_OBD_HDL(0,	OBD_PING,	tgt_obd_ping),
};

static struct tgt_handler mgs_dlm_handlers[] = {
[LDLM_ENQUEUE - LDLM_FIRST_OPC] = {
	.th_name = "LDLM_ENQUEUE",
	/* don't use th_fail_id for MGS to don't interfere with MDS tests.
	 * There are no tests for MGS with OBD_FAIL_LDLM_ENQUEUE_NET so it
	 * is safe. If such tests will be needed we have to distinguish
	 * MDS and MGS fail ids, e.g use OBD_FAIL_MGS_ENQUEUE_NET for MGS
	 * instead of common OBD_FAIL_LDLM_ENQUEUE_NET
	 */
	.th_fail_id = 0,
	.th_opc = LDLM_ENQUEUE,
	.th_flags = HAS_KEY,
	.th_act = tgt_enqueue,
	.th_fmt = &RQF_LDLM_ENQUEUE,
	.th_version = LUSTRE_DLM_VERSION,
	},
};

static struct tgt_handler mgs_llog_handlers[] = {
TGT_LLOG_HDL(0,	LLOG_ORIGIN_HANDLE_CREATE,	mgs_llog_open),
TGT_LLOG_HDL(0,	LLOG_ORIGIN_HANDLE_NEXT_BLOCK,	tgt_llog_next_block),
TGT_LLOG_HDL(0,	LLOG_ORIGIN_HANDLE_READ_HEADER,	tgt_llog_read_header),
TGT_LLOG_HDL(0,	LLOG_ORIGIN_HANDLE_PREV_BLOCK,	tgt_llog_prev_block),
};

static struct tgt_opc_slice mgs_common_slice[] = {
	{
		.tos_opc_start = MGS_FIRST_OPC,
		.tos_opc_end   = MGS_LAST_OPC,
		.tos_hs        = mgs_mgs_handlers
	},
	{
		.tos_opc_start = OBD_FIRST_OPC,
		.tos_opc_end   = OBD_LAST_OPC,
		.tos_hs        = mgs_obd_handlers
	},
	{
		.tos_opc_start = LDLM_FIRST_OPC,
		.tos_opc_end   = LDLM_LAST_OPC,
		.tos_hs        = mgs_dlm_handlers
	},
	{
		.tos_opc_start = LLOG_FIRST_OPC,
		.tos_opc_end   = LLOG_LAST_OPC,
		.tos_hs        = mgs_llog_handlers
	},
	{
		.tos_opc_start = SEC_FIRST_OPC,
		.tos_opc_end   = SEC_LAST_OPC,
		.tos_hs        = tgt_sec_ctx_handlers
	},
	{
		.tos_hs        = NULL
	}
};

static int mgs_init0(const struct lu_env *env, struct mgs_device *mgs,
		     struct lu_device_type *ldt, struct lustre_cfg *lcfg)
{
	struct ptlrpc_service_conf	 conf;
	struct obd_device		*obd;
	struct lustre_mount_info	*lmi;
	struct llog_ctxt		*ctxt;
	struct lustre_sb_info *lsi;
	int				 rc;

	ENTRY;

	lmi = server_get_mount(lustre_cfg_string(lcfg, 0));
	if (lmi == NULL)
		RETURN(-ENODEV);

	mgs->mgs_dt_dev.dd_lu_dev.ld_ops = &mgs_lu_ops;

	rc = mgs_connect_to_osd(mgs, lustre_cfg_string(lcfg, 3));
	if (rc)
		GOTO(err_lmi, rc);

	obd = class_name2obd(lustre_cfg_string(lcfg, 0));
	LASSERT(obd);
	mgs->mgs_obd = obd;
	mgs->mgs_obd->obd_lu_dev = &mgs->mgs_dt_dev.dd_lu_dev;

	obd_obt_init(obd);

	/* namespace for mgs llog */
	obd->obd_namespace = ldlm_namespace_new(obd, "MGS",
						LDLM_NAMESPACE_SERVER,
						LDLM_NAMESPACE_MODEST,
						LDLM_NS_TYPE_MGT);
	if (IS_ERR(obd->obd_namespace)) {
		rc = PTR_ERR(obd->obd_namespace);
		CERROR("%s: unable to create server namespace: rc = %d\n",
		       obd->obd_name, rc);
		obd->obd_namespace = NULL;
		GOTO(err_ops, rc);
	}

	/* No recovery for MGCs */
	clear_bit(OBDF_REPLAYABLE, obd->obd_flags);
	lsi = s2lsi(lmi->lmi_sb);
	if (test_bit(LMD_FLG_NO_RCLNT, lsi->lsi_lmd->lmd_flags))
		obd->obd_no_conn = 1;

	rc = tgt_init(env, &mgs->mgs_lut, obd, mgs->mgs_bottom,
		      mgs_common_slice, OBD_FAIL_MGS_ALL_REQUEST_NET,
		      OBD_FAIL_MGS_ALL_REPLY_NET);
	if (rc)
		GOTO(err_ns, rc);

	rc = mgs_fs_setup(env, mgs);
	if (rc) {
		CERROR("%s: MGS filesystem method init failed: rc = %d\n",
		       obd->obd_name, rc);
		GOTO(err_tgt, rc);
	}

	rc = llog_setup(env, obd, &obd->obd_olg, LLOG_CONFIG_ORIG_CTXT,
			obd, &llog_osd_ops);
	if (rc)
		GOTO(err_fs, rc);

	/* XXX: we need this trick till N:1 stack is supported
	 * set "current" directory for named llogs
	 */
	ctxt = llog_get_context(mgs->mgs_obd, LLOG_CONFIG_ORIG_CTXT);
	LASSERT(ctxt);
	ctxt->loc_dir = mgs->mgs_configs_dir;
	llog_ctxt_put(ctxt);

	/* Internal mgs setup */
	mgs_init_fsdb_list(mgs);
	mutex_init(&mgs->mgs_mutex);
	mgs->mgs_start_time = ktime_get_real_seconds();
	spin_lock_init(&mgs->mgs_lock);
	mutex_init(&mgs->mgs_health_mutex);
	init_rwsem(&mgs->mgs_barrier_rwsem);

	rc = mgs_lcfg_rename(env, mgs);
	if (rc)
		GOTO(err_llog, rc);

	rc = lproc_mgs_setup(mgs, lustre_cfg_string(lcfg, 3));
	if (rc != 0) {
		CERROR("%s: cannot initialize proc entry: rc = %d\n",
		       obd->obd_name, rc);
		GOTO(err_llog, rc);
	}

	/* Setup params fsdb and log, so that other servers can make a local
	 * copy successfully when they are mounted. See LU-4783
	 */
	rc = mgs_params_fsdb_setup(env, mgs);
	if (rc)
		/* params fsdb and log can be setup later */
		CERROR("%s: %s fsdb and log setup failed: rc = %d\n",
		       obd->obd_name, PARAMS_FILENAME, rc);

	/* Setup _mgs fsdb, useful for srpc */
	mgs__mgs_fsdb_setup(env, mgs);

	ptlrpc_init_client(LDLM_CB_REQUEST_PORTAL, LDLM_CB_REPLY_PORTAL,
			   "mgs_ldlm_client", &obd->obd_ldlm_client);

	conf = (typeof(conf)) {
		.psc_name		= LUSTRE_MGS_NAME,
		.psc_watchdog_factor	= MGS_SERVICE_WATCHDOG_FACTOR,
		.psc_buf		= {
			.bc_nbufs		= MGS_NBUFS,
			.bc_buf_size		= MGS_BUFSIZE,
			.bc_req_max_size	= MGS_MAXREQSIZE,
			.bc_rep_max_size	= MGS_MAXREPSIZE,
			.bc_req_portal		= MGS_REQUEST_PORTAL,
			.bc_rep_portal		= MGC_REPLY_PORTAL,
		},
		.psc_thr		= {
			.tc_thr_name		= "ll_mgs",
			.tc_nthrs_init		= MGS_NTHRS_INIT,
			.tc_nthrs_max		= MGS_NTHRS_MAX,
			.tc_ctx_tags		= LCT_MG_THREAD,
		},
		.psc_ops		= {
			.so_req_handler		= tgt_request_handle,
			.so_req_printer		= target_print_req,
		},
	};

	/* Start the service threads */
	mgs->mgs_service = ptlrpc_register_service(&conf, &obd->obd_kset,
						   obd->obd_debugfs_entry);
	if (IS_ERR(mgs->mgs_service)) {
		rc = PTR_ERR(mgs->mgs_service);
		CERROR("failed to start mgs service: %d\n", rc);
		mgs->mgs_service = NULL;
		GOTO(err_lproc, rc);
	}

	ping_evictor_start();

	CDEBUG(D_INFO, "MGS %s started\n", obd->obd_name);

	/* device stack is not yet fully setup to keep no objects behind */
	lu_site_purge(env, mgs2lu_dev(mgs)->ld_site, ~0);
	RETURN(0);
err_lproc:
	mgs_params_fsdb_cleanup(env, mgs);
	lproc_mgs_cleanup(mgs);
err_llog:
	ctxt = llog_get_context(mgs->mgs_obd, LLOG_CONFIG_ORIG_CTXT);
	if (ctxt) {
		ctxt->loc_dir = NULL;
		llog_cleanup(env, ctxt);
	}
err_tgt:
	tgt_fini(env, &mgs->mgs_lut);
err_fs:
	/* No extra cleanup needed for llog_init_commit_thread() */
	mgs_fs_cleanup(env, mgs);
err_ns:
	ldlm_namespace_free(obd->obd_namespace, NULL, 0);
	obd->obd_namespace = NULL;
err_ops:
	lu_site_purge(env, mgs2lu_dev(mgs)->ld_site, ~0);
	lu_site_print(env, mgs2lu_dev(mgs)->ld_site,
		      &mgs2lu_dev(mgs)->ld_site->ls_obj_hash.nelems,
		      D_OTHER, lu_cdebug_printer);
	obd_disconnect(mgs->mgs_bottom_exp);
err_lmi:
	if (lmi)
		server_put_mount(lustre_cfg_string(lcfg, 0), true);
	RETURN(rc);
}

static struct lu_device *mgs_device_free(const struct lu_env *env,
					 struct lu_device *lu)
{
	struct mgs_device *mgs = lu2mgs_dev(lu);

	ENTRY;

	dt_device_fini(&mgs->mgs_dt_dev);
	OBD_FREE_PTR(mgs);
	RETURN(NULL);
}

static int mgs_process_config(const struct lu_env *env,
			      struct lu_device *dev,
			      struct lustre_cfg *lcfg)
{
	LBUG();
	return 0;
}

static int mgs_object_init(const struct lu_env *env, struct lu_object *o,
			   const struct lu_object_conf *unused)
{
	struct mgs_device *d = lu2mgs_dev(o->lo_dev);
	struct lu_device  *under;
	struct lu_object  *below;
	int                rc = 0;

	ENTRY;

	/* do no set .do_ops as mgs calls to bottom osd directly */
	CDEBUG(D_INFO, "object init, fid = "DFID"\n",
			PFID(lu_object_fid(o)));

	under = &d->mgs_bottom->dd_lu_dev;
	below = under->ld_ops->ldo_object_alloc(env, o->lo_header, under);
	if (below != NULL)
		lu_object_add(o, below);
	else
		rc = -ENOMEM;

	return rc;
}

static void mgs_object_free(const struct lu_env *env, struct lu_object *o)
{
	struct mgs_object *obj = lu2mgs_obj(o);
	struct lu_object_header *h = o->lo_header;

	dt_object_fini(&obj->mgo_obj);
	lu_object_header_fini(h);
	OBD_FREE_RCU(obj, sizeof(*obj), mgo_header.loh_rcu);
}

static int mgs_object_print(const struct lu_env *env, void *cookie,
			    lu_printer_t p, const struct lu_object *l)
{
	const struct mgs_object *o = lu2mgs_obj((struct lu_object *) l);

	return (*p)(env, cookie, LUSTRE_MGS_NAME"-object@%p", o);
}

static const struct lu_object_operations mgs_lu_obj_ops = {
	.loo_object_init	= mgs_object_init,
	.loo_object_free	= mgs_object_free,
	.loo_object_print	= mgs_object_print,
};

static struct lu_object *mgs_object_alloc(const struct lu_env *env,
					  const struct lu_object_header *hdr,
					  struct lu_device *d)
{
	struct lu_object_header *h;
	struct mgs_object       *o;
	struct lu_object        *l;

	LASSERT(hdr == NULL);

	OBD_ALLOC_PTR(o);
	if (o != NULL) {
		l = &o->mgo_obj.do_lu;
		h = &o->mgo_header;

		lu_object_header_init(h);
		dt_object_init(&o->mgo_obj, h, d);
		lu_object_add_top(h, l);

		l->lo_ops = &mgs_lu_obj_ops;

		return l;
	} else {
		return NULL;
	}
}

const struct lu_device_operations mgs_lu_ops = {
	.ldo_object_alloc	= mgs_object_alloc,
	.ldo_process_config	= mgs_process_config,
};

static struct lu_device *mgs_device_alloc(const struct lu_env *env,
					  struct lu_device_type *type,
					  struct lustre_cfg *lcfg)
{
	struct mgs_device *mgs;
	struct lu_device  *ludev;

	OBD_ALLOC_PTR(mgs);
	if (mgs == NULL) {
		ludev = ERR_PTR(-ENOMEM);
	} else {
		int rc;

		ludev = mgs2lu_dev(mgs);
		dt_device_init(&mgs->mgs_dt_dev, type);
		rc = mgs_init0(env, mgs, type, lcfg);
		if (rc != 0) {
			mgs_device_free(env, ludev);
			ludev = ERR_PTR(rc);
		}
	}
	return ludev;
}

static struct lu_device *mgs_device_fini(const struct lu_env *env,
					 struct lu_device *d)
{
	struct mgs_device	*mgs = lu2mgs_dev(d);
	struct obd_device	*obd = mgs->mgs_obd;
	struct llog_ctxt	*ctxt;

	ENTRY;

	LASSERT(mgs->mgs_bottom);

	class_disconnect_exports(obd);

	ping_evictor_stop();

	mutex_lock(&mgs->mgs_health_mutex);
	ptlrpc_unregister_service(mgs->mgs_service);
	mutex_unlock(&mgs->mgs_health_mutex);

	mgs_params_fsdb_cleanup(env, mgs);
	mgs_cleanup_fsdb_list(mgs);

	ldlm_namespace_free_prior(obd->obd_namespace, NULL, 1);
	obd_exports_barrier(obd);
	obd_zombie_barrier();

	tgt_fini(env, &mgs->mgs_lut);
	lproc_mgs_cleanup(mgs);

	ctxt = llog_get_context(mgs->mgs_obd, LLOG_CONFIG_ORIG_CTXT);
	if (ctxt) {
		ctxt->loc_dir = NULL;
		llog_cleanup(env, ctxt);
	}

	mgs_fs_cleanup(env, mgs);

	ldlm_namespace_free_post(obd->obd_namespace);
	obd->obd_namespace = NULL;

	lu_site_purge(env, d->ld_site, ~0);
	lu_site_print(env, d->ld_site, &d->ld_site->ls_obj_hash.nelems,
		      D_OTHER, lu_cdebug_printer);
	LASSERT(mgs->mgs_bottom_exp);
	obd_disconnect(mgs->mgs_bottom_exp);

	server_put_mount(obd->obd_name, true);

	RETURN(NULL);
}

/* context key constructor/destructor: mgs_key_init, mgs_key_fini */
LU_KEY_INIT_FINI(mgs, struct mgs_thread_info);

LU_TYPE_INIT_FINI(mgs, &mgs_thread_key);

LU_CONTEXT_KEY_DEFINE(mgs, LCT_MG_THREAD);

static const struct lu_device_type_operations mgs_device_type_ops = {
	.ldto_init		= mgs_type_init,
	.ldto_fini		= mgs_type_fini,

	.ldto_start		= mgs_type_start,
	.ldto_stop		= mgs_type_stop,

	.ldto_device_alloc	= mgs_device_alloc,
	.ldto_device_free	= mgs_device_free,

	.ldto_device_fini	= mgs_device_fini
};

static struct lu_device_type mgs_device_type = {
	.ldt_tags	= LU_DEVICE_DT,
	.ldt_name	= LUSTRE_MGS_NAME,
	.ldt_ops	= &mgs_device_type_ops,
	.ldt_ctx_tags	= LCT_MG_THREAD
};

static int mgs_obd_reconnect(const struct lu_env *env, struct obd_export *exp,
			     struct obd_device *obd, struct obd_uuid *cluuid,
			     struct obd_connect_data *data, void *localdata)
{
	struct ptlrpc_request *req = localdata;
	struct ptlrpc_svc_ctx *svc_ctx = NULL;
	struct lnet_nid *client_nid = NULL;
	int rc;

	ENTRY;

	if (exp == NULL || obd == NULL || cluuid == NULL)
		RETURN(-EINVAL);

	tgt_counter_incr(exp, LPROC_MGS_CONNECT);

	if (data != NULL) {
		data->ocd_connect_flags &= MGS_CONNECT_SUPPORTED;

		if (data->ocd_connect_flags & OBD_CONNECT_FLAGS2)
			data->ocd_connect_flags2 &= MGS_CONNECT_SUPPORTED2;

		exp->exp_connect_data = *data;
		data->ocd_version = LUSTRE_VERSION_CODE;
	}

	if (req) {
		svc_ctx = req->rq_svc_ctx;
		client_nid = &req->rq_peer.nid;
	}

	if (svc_ctx || client_nid) {
		rc = nodemap_add_member(svc_ctx, client_nid, exp);
		if (rc != 0 && rc != -EEXIST)
			RETURN(rc);
	} else {
		CDEBUG(D_HA,
		       "%s: cannot find nodemap for client %s: svc_ctx and nid are null\n",
		       obd->obd_name, cluuid->uuid);
	}

	RETURN(mgs_export_stats_init(obd, exp, client_nid));
}

static int mgs_obd_connect(const struct lu_env *env, struct obd_export **exp,
			   struct obd_device *obd, struct obd_uuid *cluuid,
			   struct obd_connect_data *data, void *localdata)
{
	struct obd_export	*lexp;
	struct lustre_handle	 conn = {
		.cookie = 0,
	};
	int			 rc;

	ENTRY;

	if (exp == NULL || obd == NULL || cluuid == NULL)
		RETURN(-EINVAL);

	rc = class_connect(&conn, obd, cluuid);
	if (rc)
		RETURN(rc);

	lexp = class_conn2export(&conn);
	if (lexp == NULL)
		RETURN(-EFAULT);

	rc = mgs_obd_reconnect(env, lexp, obd, cluuid, data, localdata);
	if (rc)
		GOTO(out_disconnect, rc);

	*exp = lexp;

	RETURN(rc);

out_disconnect:
	class_disconnect(lexp);
	nodemap_del_member(lexp);

	return rc;
}

static int mgs_obd_disconnect(struct obd_export *exp)
{
	int rc;

	nodemap_del_member(exp);

	ENTRY;

	LASSERT(exp);

	mgs_fsc_cleanup(exp);

	class_export_get(exp);
	tgt_counter_incr(exp, LPROC_MGS_DISCONNECT);

	rc = server_disconnect_export(exp);
	class_export_put(exp);
	RETURN(rc);
}

static int mgs_health_check(const struct lu_env *env, struct obd_device *obd)
{
	struct mgs_device *mgs = lu2mgs_dev(obd->obd_lu_dev);
	int rc = 0;

	mutex_lock(&mgs->mgs_health_mutex);
	rc |= ptlrpc_service_health_check(mgs->mgs_service);
	mutex_unlock(&mgs->mgs_health_mutex);

	return rc != 0 ? 1 : 0;
}

/* use obd ops to offer management infrastructure */
static const struct obd_ops mgs_obd_device_ops = {
	.o_owner		= THIS_MODULE,
	.o_connect		= mgs_obd_connect,
	.o_reconnect		= mgs_obd_reconnect,
	.o_disconnect		= mgs_obd_disconnect,
	.o_init_export		= mgs_init_export,
	.o_destroy_export	= mgs_destroy_export,
	.o_iocontrol		= mgs_iocontrol,
	.o_health_check		= mgs_health_check,
};

static int __init mgs_init(void)
{
	int rc;

	rc = libcfs_setup();
	if (rc)
		return rc;

	return class_register_type(&mgs_obd_device_ops, NULL, false,
				   LUSTRE_MGS_NAME, &mgs_device_type);
}

static void __exit mgs_exit(void)
{
	class_unregister_type(LUSTRE_MGS_NAME);
}

MODULE_AUTHOR("OpenSFS, Inc. <http://www.lustre.org/>");
MODULE_DESCRIPTION("Lustre Management Server (MGS)");
MODULE_VERSION(LUSTRE_VERSION_STRING);
MODULE_LICENSE("GPL");

late_initcall_sync(mgs_init);
module_exit(mgs_exit);