/*
 * GPL HEADER START
 *
 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License version 2 only,
 * as published by the Free Software Foundation.
 *
 * This program is distributed in the hope that it will be useful, but
 * WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * General Public License version 2 for more details (a copy is included
 * in the LICENSE file that accompanied this code).
 *
 * You should have received a copy of the GNU General Public License
 * version 2 along with this program; If not, see
 * http://www.sun.com/software/products/lustre/docs/GPLv2.pdf
 *
 * Please contact Sun Microsystems, Inc., 4150 Network Circle, Santa Clara,
 * CA 95054 USA or visit www.sun.com if you need additional information or
 * have any questions.
 *
 * GPL HEADER END
 */
/*
 * Copyright (c) 2009, 2010, Oracle and/or its affiliates. All rights reserved.
 * Use is subject to license terms.
 */
/*
 * Copyright (c) 2012, 2013, Intel Corporation.
 * Use is subject to license terms.
 */
/*
 * This file is part of Lustre, http://www.lustre.org/
 * Lustre is a trademark of Sun Microsystems, Inc.
 *
 * lustre/osd-zfs/osd_oi.c
 * OI functions to map fid to dnode
 *
 * Author: Alex Zhuravlev <bzzz@whamcloud.com>
 * Author: Mike Pershin <tappro@whamcloud.com>
 * Author: Di Wang <di.wang@intel.com>
 */

#define DEBUG_SUBSYSTEM S_OSD

#include <lustre_ver.h>
#include <libcfs/libcfs.h>
#include <obd_support.h>
#include <lustre_net.h>
#include <obd.h>
#include <obd_class.h>
#include <lustre_disk.h>
#include <lustre_fid.h>

#include "osd_internal.h"

#include <sys/dnode.h>
#include <sys/dbuf.h>
#include <sys/spa.h>
#include <sys/stat.h>
#include <sys/zap.h>
#include <sys/spa_impl.h>
#include <sys/zfs_znode.h>
#include <sys/dmu_tx.h>
#include <sys/dmu_objset.h>
#include <sys/dsl_prop.h>
#include <sys/sa_impl.h>
#include <sys/txg.h>

static char *oi_tag = "osd_mount, oi";

#define OSD_OI_FID_NR         (1UL << 7)
#define OSD_OI_FID_NR_MAX     (1UL << OSD_OI_FID_OID_BITS_MAX)
unsigned int osd_oi_count = OSD_OI_FID_NR;


/*
 * zfs osd maintains names for known fids in the name hierarchy
 * so that one can mount filesystem with regular ZFS stack and
 * access files
 */
struct named_oid {
	unsigned long	 oid;
	char		*name;
};

static const struct named_oid oids[] = {
	{ LAST_RECV_OID,		LAST_RCVD },
	{ OFD_LAST_GROUP_OID,		"LAST_GROUP" },
	{ LLOG_CATALOGS_OID,		"CATALOGS" },
	{ MGS_CONFIGS_OID,              NULL /*MOUNT_CONFIGS_DIR*/ },
	{ FID_SEQ_SRV_OID,              "seq_srv" },
	{ FID_SEQ_CTL_OID,              "seq_ctl" },
	{ FLD_INDEX_OID,                "fld" },
	{ MDD_LOV_OBJ_OID,		LOV_OBJID },
	{ OFD_HEALTH_CHECK_OID,		HEALTH_CHECK },
	{ ACCT_USER_OID,		"acct_usr_inode" },
	{ ACCT_GROUP_OID,		"acct_grp_inode" },
	{ 0,				NULL }
};

static char *oid2name(const unsigned long oid)
{
	int i = 0;

	while (oids[i].oid) {
		if (oids[i].oid == oid)
			return oids[i].name;
		i++;
	}
	return NULL;
}

/**
 * Lookup an existing OI by the given name.
 */
static int
osd_oi_lookup(const struct lu_env *env, struct osd_device *o,
	      uint64_t parent, const char *name, struct osd_oi *oi)
{
	struct zpl_direntry	*zde = &osd_oti_get(env)->oti_zde.lzd_reg;
	int			 rc;

	rc = -zap_lookup(o->od_objset.os, parent, name, 8, 1, (void *)zde);
	if (rc)
		return rc;

	strncpy(oi->oi_name, name, OSD_OI_NAME_SIZE - 1);
	oi->oi_zapid = zde->zde_dnode;

	return rc;
}

/**
 * Create a new OI with the given name.
 */
static int
osd_oi_create(const struct lu_env *env, struct osd_device *o,
	      uint64_t parent, const char *name, uint64_t *child)
{
	struct zpl_direntry	*zde = &osd_oti_get(env)->oti_zde.lzd_reg;
	struct lu_attr		*la = &osd_oti_get(env)->oti_la;
	dmu_buf_t		*db;
	dmu_tx_t		*tx;
	int			 rc;

	/* verify it doesn't already exist */
	rc = -zap_lookup(o->od_objset.os, parent, name, 8, 1, (void *)zde);
	if (rc == 0)
		return -EEXIST;

	/* create fid-to-dnode index */
	tx = dmu_tx_create(o->od_objset.os);
	if (tx == NULL)
		return -ENOMEM;

	dmu_tx_hold_zap(tx, DMU_NEW_OBJECT, 1, NULL);
	dmu_tx_hold_bonus(tx, parent);
	dmu_tx_hold_zap(tx, parent, TRUE, name);
	LASSERT(tx->tx_objset->os_sa);
	dmu_tx_hold_sa_create(tx, ZFS_SA_BASE_ATTR_SIZE);

	rc = -dmu_tx_assign(tx, TXG_WAIT);
	if (rc) {
		dmu_tx_abort(tx);
		return rc;
	}

	la->la_valid = LA_MODE | LA_UID | LA_GID;
	la->la_mode = S_IFDIR | S_IRUGO | S_IWUSR | S_IXUGO;
	la->la_uid = la->la_gid = 0;
	__osd_zap_create(env, &o->od_objset, &db, tx, la, parent, oi_tag, 0);

	zde->zde_dnode = db->db_object;
	zde->zde_pad = 0;
	zde->zde_type = IFTODT(S_IFDIR);

	rc = -zap_add(o->od_objset.os, parent, name, 8, 1, (void *)zde, tx);

	dmu_tx_commit(tx);

	*child = db->db_object;
	sa_buf_rele(db, oi_tag);

	return rc;
}

static int
osd_oi_find_or_create(const struct lu_env *env, struct osd_device *o,
		      uint64_t parent, const char *name, uint64_t *child)
{
	struct osd_oi	oi;
	int		rc;

	rc = osd_oi_lookup(env, o, parent, name, &oi);
	if (rc == 0)
		*child = oi.oi_zapid;
	else if (rc == -ENOENT)
		rc = osd_oi_create(env, o, parent, name, child);

	return rc;
}

/**
 * Lookup the target index/flags of the fid, so it will know where
 * the object is located (tgt index) and it is MDT or OST object.
 */
int osd_fld_lookup(const struct lu_env *env, struct osd_device *osd,
		   const struct lu_fid *fid, struct lu_seq_range *range)
{
	struct seq_server_site	*ss = osd_seq_site(osd);
	int			rc;

	if (fid_is_idif(fid)) {
		fld_range_set_ost(range);
		range->lsr_index = fid_idif_ost_idx(fid);
		return 0;
	}

	if (!fid_seq_in_fldb(fid_seq(fid))) {
		fld_range_set_mdt(range);
		if (ss != NULL)
			/* FIXME: If ss is NULL, it suppose not get lsr_index
			 * at all */
			range->lsr_index = ss->ss_node_id;
		return 0;
	}

	LASSERT(ss != NULL);
	fld_range_set_any(range);
	rc = fld_server_lookup(env, ss->ss_server_fld, fid_seq(fid), range);
	if (rc != 0)
		CERROR("%s: cannot find FLD range for "DFID": rc = %d\n",
		       osd_name(osd), PFID(fid), rc);
	return rc;
}

int fid_is_on_ost(const struct lu_env *env, struct osd_device *osd,
		  const struct lu_fid *fid)
{
	struct lu_seq_range *range = &osd_oti_get(env)->oti_seq_range;
	int rc;
	ENTRY;

	if (fid_is_idif(fid))
		RETURN(1);

	rc = osd_fld_lookup(env, osd, fid, range);
	if (rc != 0) {
		CERROR("%s: Can not lookup fld for "DFID"\n",
		       osd_name(osd), PFID(fid));
		RETURN(rc);
	}

	CDEBUG(D_INFO, "fid "DFID" range "DRANGE"\n", PFID(fid),
	       PRANGE(range));

	if (fld_range_is_ost(range))
		RETURN(1);

	RETURN(0);
}

static struct osd_seq *osd_seq_find_locked(struct osd_seq_list *seq_list,
					   obd_seq seq)
{
	struct osd_seq *osd_seq;

	cfs_list_for_each_entry(osd_seq, &seq_list->osl_seq_list, os_seq_list) {
		if (osd_seq->os_seq == seq)
			return osd_seq;
	}
	return NULL;
}

static struct osd_seq *osd_seq_find(struct osd_seq_list *seq_list,
				    obd_seq seq)
{
	struct osd_seq *osd_seq;

	read_lock(&seq_list->osl_seq_list_lock);
	osd_seq = osd_seq_find_locked(seq_list, seq);
	read_unlock(&seq_list->osl_seq_list_lock);

	return osd_seq;
}

static struct osd_seq *osd_find_or_add_seq(const struct lu_env *env,
					   struct osd_device *osd, obd_seq seq)
{
	struct osd_seq_list	*seq_list = &osd->od_seq_list;
	struct osd_seq		*osd_seq;
	char			*key = osd_oti_get(env)->oti_buf;
	char			*seq_name = osd_oti_get(env)->oti_str;
	struct osd_oi		oi;
	uint64_t		sdb, odb;
	int			i;
	int			rc = 0;
	ENTRY;

	osd_seq = osd_seq_find(seq_list, seq);
	if (osd_seq != NULL)
		RETURN(osd_seq);

	down(&seq_list->osl_seq_init_sem);
	/* Check again, in case some one else already add it
	 * to the list */
	osd_seq = osd_seq_find(seq_list, seq);
	if (osd_seq != NULL)
		GOTO(out, rc = 0);

	OBD_ALLOC_PTR(osd_seq);
	if (osd_seq == NULL)
		GOTO(out, rc = -ENOMEM);

	CFS_INIT_LIST_HEAD(&osd_seq->os_seq_list);
	osd_seq->os_seq = seq;

	/* Init subdir count to be 32, but each seq can have
	 * different subdir count */
	osd_seq->os_subdir_count = OSD_OST_MAP_SIZE;
	OBD_ALLOC(osd_seq->os_compat_dirs,
		  sizeof(uint64_t) * osd_seq->os_subdir_count);
	if (osd_seq->os_compat_dirs == NULL)
		GOTO(out, rc = -ENOMEM);

	rc = osd_oi_lookup(env, osd, osd->od_root, "O", &oi);
	if (rc != 0) {
		CERROR("%s: Can not find O: rc = %d\n", osd_name(osd), rc);
		GOTO(out, rc);
	}

	sprintf(seq_name, (fid_seq_is_rsvd(seq) ||
		fid_seq_is_mdt0(seq)) ?  LPU64 : LPX64i,
		fid_seq_is_idif(seq) ? 0 : seq);

	rc = osd_oi_find_or_create(env, osd, oi.oi_zapid, seq_name, &odb);
	if (rc != 0) {
		CERROR("%s: Can not create %s : rc = %d\n",
		       osd_name(osd), seq_name, rc);
		GOTO(out, rc);
	}

	for (i = 0; i < OSD_OST_MAP_SIZE; i++) {
		sprintf(key, "d%d", i);
		rc = osd_oi_find_or_create(env, osd, odb, key, &sdb);
		if (rc)
			GOTO(out, rc);
		osd_seq->os_compat_dirs[i] = sdb;
	}

	write_lock(&seq_list->osl_seq_list_lock);
	cfs_list_add(&osd_seq->os_seq_list, &seq_list->osl_seq_list);
	write_unlock(&seq_list->osl_seq_list_lock);
out:
	up(&seq_list->osl_seq_init_sem);
	if (rc != 0) {
		if (osd_seq != NULL && osd_seq->os_compat_dirs != NULL)
			OBD_FREE(osd_seq->os_compat_dirs,
				 sizeof(uint64_t) * osd_seq->os_subdir_count);
		if (osd_seq != NULL)
			OBD_FREE_PTR(osd_seq);
		osd_seq = ERR_PTR(rc);
	}
	RETURN(osd_seq);
}

/*
 * objects w/o a natural reference (unlike a file on a MDS)
 * are put under a special hierarchy /O/<seq>/d0..dXX
 * this function returns a directory specific fid belongs to
 */
static uint64_t
osd_get_idx_for_ost_obj(const struct lu_env *env, struct osd_device *osd,
			const struct lu_fid *fid, char *buf)
{
	struct osd_seq	*osd_seq;
	unsigned long	b;
	int		rc;

	osd_seq = osd_find_or_add_seq(env, osd, fid_seq(fid));
	if (IS_ERR(osd_seq)) {
		CERROR("%s: Can not find seq group "DFID"\n", osd_name(osd),
		       PFID(fid));
		return PTR_ERR(osd_seq);
	}

	rc = fid_to_ostid(fid, &osd_oti_get(env)->oti_ostid);
	LASSERT(rc == 0); /* we should not get here with IGIF */
	b = ostid_id(&osd_oti_get(env)->oti_ostid) % OSD_OST_MAP_SIZE;
	LASSERT(osd_seq->os_compat_dirs[b]);

	sprintf(buf, LPU64, ostid_id(&osd_oti_get(env)->oti_ostid));

	return osd_seq->os_compat_dirs[b];
}

/* XXX: f_ver is not counted, but may differ too */
static void osd_fid2str(char *buf, const struct lu_fid *fid)
{
	sprintf(buf, DFID_NOBRACE, PFID(fid));
}

/*
 * Determine the zap object id which is being used as the OI for the
 * given fid.  The lowest N bits in the sequence ID are used as the
 * index key.  On failure 0 is returned which zfs treats internally
 * as an invalid object id.
 */
static uint64_t
osd_get_idx_for_fid(struct osd_device *osd, const struct lu_fid *fid,
		    char *buf)
{
	struct osd_oi *oi;

	LASSERT(osd->od_oi_table != NULL);
	oi = osd->od_oi_table[fid_seq(fid) & (osd->od_oi_count - 1)];
	osd_fid2str(buf, fid);

	return oi->oi_zapid;
}

uint64_t osd_get_name_n_idx(const struct lu_env *env, struct osd_device *osd,
			    const struct lu_fid *fid, char *buf)
{
	uint64_t zapid;

	LASSERT(fid);
	LASSERT(buf);

	if (fid_is_on_ost(env, osd, fid) == 1 || fid_seq(fid) == FID_SEQ_ECHO) {
		zapid = osd_get_idx_for_ost_obj(env, osd, fid, buf);
	} else if (unlikely(fid_seq(fid) == FID_SEQ_LOCAL_FILE)) {
		/* special objects with fixed known fids get their name */
		char *name = oid2name(fid_oid(fid));

		if (name) {
			zapid = osd->od_root;
			strcpy(buf, name);
			if (fid_is_acct(fid))
				zapid = MASTER_NODE_OBJ;
		} else {
			zapid = osd_get_idx_for_fid(osd, fid, buf);
		}
	} else {
		zapid = osd_get_idx_for_fid(osd, fid, buf);
	}

	return zapid;
}

static inline int fid_is_fs_root(const struct lu_fid *fid)
{
	/* Map root inode to special local object FID */
	return fid_seq(fid) == FID_SEQ_LOCAL_FILE &&
		fid_oid(fid) == OSD_FS_ROOT_OID;
}

int osd_fid_lookup(const struct lu_env *env, struct osd_device *dev,
		   const struct lu_fid *fid, uint64_t *oid)
{
	struct osd_thread_info	*info = osd_oti_get(env);
	char			*buf = info->oti_buf;
	uint64_t		zapid;
	int			rc = 0;
	ENTRY;

	if (OBD_FAIL_CHECK(OBD_FAIL_OST_ENOENT))
		RETURN(-ENOENT);

	if (unlikely(fid_is_acct(fid))) {
		if (fid_oid(fid) == ACCT_USER_OID)
			*oid = dev->od_iusr_oid;
		else
			*oid = dev->od_igrp_oid;
	} else if (unlikely(fid_is_fs_root(fid))) {
		*oid = dev->od_root;
	} else {
		zapid = osd_get_name_n_idx(env, dev, fid, buf);

		rc = -zap_lookup(dev->od_objset.os, zapid, buf,
				8, 1, &info->oti_zde);
		if (rc)
			RETURN(rc);
		*oid = info->oti_zde.lzd_reg.zde_dnode;
	}

	RETURN(rc);
}

/**
 * Close an entry in a specific slot.
 */
static void
osd_oi_remove_table(const struct lu_env *env, struct osd_device *o, int key)
{
	struct osd_oi *oi;

	LASSERT(key < o->od_oi_count);

	oi = o->od_oi_table[key];
	if (oi) {
		OBD_FREE_PTR(oi);
		o->od_oi_table[key] = NULL;
	}
}

/**
 * Allocate and open a new entry in the specified unused slot.
 */
static int
osd_oi_add_table(const struct lu_env *env, struct osd_device *o,
		 char *name, int key)
{
	struct osd_oi *oi;
	int rc;

	LASSERT(key < o->od_oi_count);
	LASSERT(o->od_oi_table[key] == NULL);

	OBD_ALLOC_PTR(oi);
	if (oi == NULL)
		return -ENOMEM;

	rc = osd_oi_lookup(env, o, o->od_root, name, oi);
	if (rc) {
		OBD_FREE_PTR(oi);
		return rc;
	}

	o->od_oi_table[key] = oi;

	return 0;
}

/**
 * Depopulate the OI table.
 */
static void
osd_oi_close_table(const struct lu_env *env, struct osd_device *o)
{
	int i;

	for (i = 0; i < o->od_oi_count; i++)
		osd_oi_remove_table(env, o, i);
}

/**
 * Populate the OI table based.
 */
static int
osd_oi_open_table(const struct lu_env *env, struct osd_device *o, int count)
{
	char name[16];
	int  i, rc = 0;
	ENTRY;

	for (i = 0; i < count; i++) {
		sprintf(name, "%s.%d", DMU_OSD_OI_NAME_BASE, i);
		rc = osd_oi_add_table(env, o, name, i);
		if (rc) {
			osd_oi_close_table(env, o);
			break;
		}
	}

	RETURN(rc);
}

/**
 * Determine if the type and number of OIs used by this file system.
 */
static int
osd_oi_probe(const struct lu_env *env, struct osd_device *o, int *count)
{
	uint64_t	root_oid = o->od_root;
	struct osd_oi	oi;
	char		name[16];
	int		rc;
	ENTRY;

	/*
	 * Check for multiple OIs and determine the count.  There is no
	 * gap handling, if an OI is missing the wrong size can be returned.
	 * The only safeguard is that we know the number of OIs must be a
	 * power of two and this is checked for basic sanity.
	 */
	for (*count = 0; *count < OSD_OI_FID_NR_MAX; (*count)++) {
		sprintf(name, "%s.%d", DMU_OSD_OI_NAME_BASE, *count);
		rc = osd_oi_lookup(env, o, root_oid, name, &oi);
		if (rc == 0)
			continue;

		if (rc == -ENOENT) {
			if (*count == 0)
				break;

			if ((*count & (*count - 1)) != 0)
				RETURN(-EDOM);

			RETURN(0);
		}

		RETURN(rc);
	}

	/*
	 * No OIs exist, this must be a new filesystem.
	 */
	*count = 0;

	RETURN(0);
}

static void osd_ost_seq_init(const struct lu_env *env, struct osd_device *osd)
{
	struct osd_seq_list *osl = &osd->od_seq_list;

	CFS_INIT_LIST_HEAD(&osl->osl_seq_list);
	rwlock_init(&osl->osl_seq_list_lock);
	sema_init(&osl->osl_seq_init_sem, 1);
}

static void osd_ost_seq_fini(const struct lu_env *env, struct osd_device *osd)
{
	struct osd_seq_list	*osl = &osd->od_seq_list;
	struct osd_seq		*osd_seq, *tmp;

	write_lock(&osl->osl_seq_list_lock);
	cfs_list_for_each_entry_safe(osd_seq, tmp, &osl->osl_seq_list,
				     os_seq_list) {
		cfs_list_del(&osd_seq->os_seq_list);
		OBD_FREE(osd_seq->os_compat_dirs,
			 sizeof(uint64_t) * osd_seq->os_subdir_count);
		OBD_FREE(osd_seq, sizeof(*osd_seq));
	}
	write_unlock(&osl->osl_seq_list_lock);

	return;
}

/**
 * Create /O subdirectory to map legacy OST objects for compatibility.
 */
static int
osd_oi_init_compat(const struct lu_env *env, struct osd_device *o)
{
	uint64_t	 odb, sdb;
	int		 rc;
	ENTRY;

	rc = osd_oi_find_or_create(env, o, o->od_root, "O", &sdb);
	if (rc)
		RETURN(rc);

	osd_ost_seq_init(env, o);
	/* Create on-disk indexes to maintain per-UID/GID inode usage.
	 * Those new indexes are created in the top-level ZAP outside the
	 * namespace in order not to confuse ZPL which might interpret those
	 * indexes as directories and assume the values are object IDs */
	rc = osd_oi_find_or_create(env, o, MASTER_NODE_OBJ,
			oid2name(ACCT_USER_OID), &odb);
	if (rc)
		RETURN(rc);
	o->od_iusr_oid = odb;

	rc = osd_oi_find_or_create(env, o, MASTER_NODE_OBJ,
			oid2name(ACCT_GROUP_OID), &odb);
	if (rc)
		RETURN(rc);
	o->od_igrp_oid = odb;

	RETURN(rc);
}

static char *root2convert = "ROOT";
/*
 * due to DNE requirements we have to change sequence of /ROOT object
 * so that it doesn't belong to the local sequence FID_SEQ_LOCAL_FILE
 * but a normal sequence living on MDS#0
 * this is the sole purpose of this function.
 *
 * This is only needed for pre-production 2.4 ZFS filesystems, and
 * can be removed in the future.
 */
int osd_convert_root_to_new_seq(const struct lu_env *env,
					struct osd_device *o)
{
	struct luz_direntry *lze = &osd_oti_get(env)->oti_zde;
	char		    *buf = osd_oti_get(env)->oti_str;
	struct lu_fid	     newfid;
	uint64_t	     zapid;
	dmu_tx_t	    *tx = NULL;
	int		     rc;
	ENTRY;

	/* ignore OSTs */
	if (strstr(o->od_svname, "MDT") == NULL)
		RETURN(0);

	/* lookup /ROOT */
	rc = -zap_lookup(o->od_objset.os, o->od_root, root2convert, 8,
			 sizeof(*lze) / 8, (void *)lze);
	/* doesn't exist or let actual user to handle the error */
	if (rc)
		RETURN(0);

	CDEBUG(D_OTHER, "%s: /ROOT -> "DFID" -> "LPU64"\n", o->od_svname,
	       PFID(&lze->lzd_fid), (long long int) lze->lzd_reg.zde_dnode);

	/* already right one? */
	if (fid_seq(&lze->lzd_fid) == FID_SEQ_ROOT)
		return 0;

	tx = dmu_tx_create(o->od_objset.os);
	if (tx == NULL)
		return -ENOMEM;

	dmu_tx_hold_bonus(tx, o->od_root);

	/* declare delete/insert of the name */
	dmu_tx_hold_zap(tx, o->od_root, TRUE, root2convert);
	dmu_tx_hold_zap(tx, o->od_root, FALSE, root2convert);

	/* declare that we'll remove object from fid-dnode mapping */
	zapid = osd_get_name_n_idx(env, o, &lze->lzd_fid, buf);
	dmu_tx_hold_bonus(tx, zapid);
	dmu_tx_hold_zap(tx, zapid, FALSE, buf);

	/* declare that we'll add object to fid-dnode mapping */
	newfid.f_seq = FID_SEQ_ROOT;
	newfid.f_oid = 1;
	newfid.f_ver = 0;
	zapid = osd_get_name_n_idx(env, o, &newfid, buf);
	dmu_tx_hold_bonus(tx, zapid);
	dmu_tx_hold_zap(tx, zapid, TRUE, buf);

	rc = -dmu_tx_assign(tx, TXG_WAIT);
	if (rc)
		GOTO(err, rc);

	rc = -zap_remove(o->od_objset.os, o->od_root, root2convert, tx);
	if (rc)
		GOTO(err, rc);

	/* remove from OI */
	zapid = osd_get_name_n_idx(env, o, &lze->lzd_fid, buf);
	rc = -zap_remove(o->od_objset.os, zapid, buf, tx);
	if (rc)
		GOTO(err, rc);

	lze->lzd_fid = newfid;
	rc = -zap_add(o->od_objset.os, o->od_root, root2convert,
		      8, sizeof(*lze) / 8, (void *)lze, tx);
	if (rc)
		GOTO(err, rc);

	/* add to OI with the new fid */
	zapid = osd_get_name_n_idx(env, o, &newfid, buf);
	rc = -zap_add(o->od_objset.os, zapid, buf, 8, 1, &lze->lzd_reg, tx);
	if (rc)
		GOTO(err, rc);


	/* LMA will be updated in mdd_compat_fixes */
	dmu_tx_commit(tx);

	RETURN(rc);

err:
	if (tx)
		dmu_tx_abort(tx);
	CERROR("%s: can't convert to new fid: rc = %d\n", o->od_svname, rc);
	RETURN(rc);
}

/**
 * Initialize the OIs by either opening or creating them as needed.
 */
int osd_oi_init(const struct lu_env *env, struct osd_device *o)
{
	char	*key = osd_oti_get(env)->oti_buf;
	int	 i, rc, count = 0;
	ENTRY;

	rc = osd_oi_probe(env, o, &count);
	if (rc)
		RETURN(rc);

	if (count == 0) {
		uint64_t odb, sdb;

		count = osd_oi_count;
		odb = o->od_root;

		for (i = 0; i < count; i++) {
			sprintf(key, "%s.%d", DMU_OSD_OI_NAME_BASE, i);
			rc = osd_oi_find_or_create(env, o, odb, key, &sdb);
			if (rc)
				RETURN(rc);
		}
	}

	rc = osd_oi_init_compat(env, o);
	if (rc)
		RETURN(rc);

	LASSERT((count & (count - 1)) == 0);
	o->od_oi_count = count;
	OBD_ALLOC(o->od_oi_table, sizeof(struct osd_oi *) * count);
	if (o->od_oi_table == NULL)
		RETURN(-ENOMEM);

	rc = osd_oi_open_table(env, o, count);
	if (rc) {
		OBD_FREE(o->od_oi_table, sizeof(struct osd_oi *) * count);
		o->od_oi_table = NULL;
	}

	RETURN(rc);
}

void osd_oi_fini(const struct lu_env *env, struct osd_device *o)
{
	ENTRY;

	osd_ost_seq_fini(env, o);

	if (o->od_oi_table != NULL) {
		(void) osd_oi_close_table(env, o);
		OBD_FREE(o->od_oi_table,
			 sizeof(struct osd_oi *) * o->od_oi_count);
		o->od_oi_table = NULL;
		o->od_oi_count = 0;
	}

	EXIT;
}

int osd_options_init(void)
{
	/* osd_oi_count - Default number of OIs, 128 works well for ZFS */
	if (osd_oi_count == 0 || osd_oi_count > OSD_OI_FID_NR_MAX)
		osd_oi_count = OSD_OI_FID_NR;

	if ((osd_oi_count & (osd_oi_count - 1)) != 0) {
		LCONSOLE_WARN("Round up osd_oi_count %d to power2 %d\n",
			osd_oi_count, size_roundup_power2(osd_oi_count));
		osd_oi_count = size_roundup_power2(osd_oi_count);
	}

	return 0;
}