/* -*- mode: c; c-basic-offset: 8; indent-tabs-mode: nil; -*-
 * vim:expandtab:shiftwidth=8:tabstop=8:
 *
 *  lustre/mdd/mdd_lov.c
 *  Lustre Metadata Server (mds) handling of striped file data
 *
 *  Copyright (C) 2001-2006 Cluster File Systems, Inc.
 *   Author: Peter Braam <braam@clusterfs.com>
 *           wangdi <wangdi@clusterfs.com>
 *
 *   This file is part of the Lustre file system, http://www.lustre.org
 *   Lustre is a trademark of Cluster File Systems, Inc.
 *
 *   You may have signed or agreed to another license before downloading
 *   this software.  If so, you are bound by the terms and conditions
 *   of that agreement, and the following does not apply to you.  See the
 *   LICENSE file included with this distribution for more information.
 *
 *   If you did not agree to a different license, then this copy of Lustre
 *   is open source software; you can redistribute it and/or modify it
 *   under the terms of version 2 of the GNU General Public License as
 *   published by the Free Software Foundation.
 *
 *   In either case, Lustre 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
 *   license text for more details.
 */
#ifndef EXPORT_SYMTAB
# define EXPORT_SYMTAB
#endif
#define DEBUG_SUBSYSTEM S_MDS

#include <linux/module.h>
#include <obd.h>
#include <obd_class.h>
#include <lustre_ver.h>
#include <obd_support.h>
#include <obd_lov.h>
#include <lprocfs_status.h>
#include <lustre_mds.h>
#include <lustre_fid.h>
#include <lustre/lustre_idl.h>

#include "mdd_internal.h"

static int mdd_lov_update(struct obd_device *host,
                          struct obd_device *watched,
                          enum obd_notify_event ev, void *owner)
{
        struct mdd_device *mdd = owner;
        struct md_device *upcall_dev;
        int rc;
        ENTRY;

        LASSERT(owner != NULL);

        upcall_dev = mdd->mdd_md_dev.md_upcall.mu_upcall_dev;

        rc = upcall_dev->md_upcall.mu_upcall(NULL, upcall_dev, MD_LOV_SYNC);
        if (rc)
                RETURN(rc);
        
        rc = mdd_txn_init_credits(NULL, mdd);
        
        RETURN(rc);
}

/* The obd is created for handling data stack for mdd */
int mdd_init_obd(const struct lu_env *env, struct mdd_device *mdd,
                 struct lustre_cfg *cfg)
{
        struct lustre_cfg_bufs *bufs;
        struct lustre_cfg      *lcfg;
        struct obd_device      *obd;
        char                   *dev = lustre_cfg_string(cfg, 0);
        char                   *index_string = lustre_cfg_string(cfg, 2);
        char                   *name, *uuid, *p;
        int rc, name_size, uuid_size, index;
        ENTRY;

        LASSERT(index_string);

        index = simple_strtol(index_string, &p, 10);

        name_size = strlen(MDD_OBD_NAME) + 5;
        uuid_size = strlen(MDD_OBD_UUID) + 5;

        OBD_ALLOC(name, name_size);
        OBD_ALLOC(uuid, uuid_size);
        if (!name || !uuid) {
                if (name)
                        OBD_FREE(name, name_size);
                if (uuid)
                        OBD_FREE(uuid, uuid_size);
                RETURN(-ENOMEM);
        }

        OBD_ALLOC_PTR(bufs);
        if (!bufs) {
                GOTO(cleanup_mem, rc = -ENOMEM);
        }

        snprintf(name, strlen(MDD_OBD_NAME) + 5, "%s-%d",
                                              MDD_OBD_NAME, index);
        snprintf(uuid, strlen(MDD_OBD_UUID) + 5, "%s-%d",
                                              MDD_OBD_UUID, index);
        lustre_cfg_bufs_reset(bufs, name);
        lustre_cfg_bufs_set_string(bufs, 1, MDD_OBD_TYPE);
        lustre_cfg_bufs_set_string(bufs, 2, uuid);
        lustre_cfg_bufs_set_string(bufs, 3, (char*)dev/* MDD_OBD_PROFILE */);
        lustre_cfg_bufs_set_string(bufs, 4, (char*)dev);

        lcfg = lustre_cfg_new(LCFG_ATTACH, bufs);
        OBD_FREE_PTR(bufs);
        if (!lcfg)
                GOTO(cleanup_mem, rc = -ENOMEM);

        rc = class_attach(lcfg);
        if (rc)
                GOTO(lcfg_cleanup, rc);

        obd = class_name2obd(name);
        if (!obd) {
                CERROR("can not find obd %s \n", MDD_OBD_NAME);
                LBUG();
        }

        obd->u.mds.mds_id = index;
        obd->obd_recovering = 1;
        rc = class_setup(obd, lcfg);
        if (rc)
                GOTO(class_detach, rc);
        /*
         * Add here for obd notify mechiasm,
         * when adding a new ost, the mds will notify this mdd
         */
        obd->obd_upcall.onu_owner = mdd;
        obd->obd_upcall.onu_upcall = mdd_lov_update;
        mdd->mdd_obd_dev = obd;
class_detach:
        if (rc)
                class_detach(obd, lcfg);
lcfg_cleanup:
        lustre_cfg_free(lcfg);
cleanup_mem:
        OBD_FREE(name, name_size);
        OBD_FREE(uuid, uuid_size);
        RETURN(rc);
}

int mdd_fini_obd(const struct lu_env *env, struct mdd_device *mdd)
{
        struct lustre_cfg_bufs *bufs;
        struct lustre_cfg      *lcfg;
        struct obd_device      *obd;
        int rc;
        ENTRY;

        obd = mdd2obd_dev(mdd);
        LASSERT(obd);

        OBD_ALLOC_PTR(bufs);
        if (!bufs)
                RETURN(-ENOMEM);
        lustre_cfg_bufs_reset(bufs, MDD_OBD_NAME);
        lcfg = lustre_cfg_new(LCFG_ATTACH, bufs);
        OBD_FREE_PTR(bufs);
        if (!lcfg)
                RETURN(-ENOMEM);

        rc = class_cleanup(obd, lcfg);
        if (rc)
                GOTO(lcfg_cleanup, rc);

        rc = class_detach(obd, lcfg);
        if (rc)
                GOTO(lcfg_cleanup, rc);
        mdd->mdd_obd_dev = NULL;
lcfg_cleanup:
        lustre_cfg_free(lcfg);
        RETURN(rc);
}

int mdd_get_md(const struct lu_env *env, struct mdd_object *obj,
               void *md, int *md_size, const char *name)
{
        struct dt_object *next;
        int rc;
        ENTRY;

        next = mdd_object_child(obj);
        rc = next->do_ops->do_xattr_get(env, next,
                                        mdd_buf_get(env, md, *md_size), name,
                                        mdd_object_capa(env, obj));
        /*
         * XXX: handling of -ENODATA, the right way is to have ->do_md_get()
         * exported by dt layer.
         */
        if (rc == 0 || rc == -ENODATA) {
                *md_size = 0;
                rc = 0;
        } else if (rc < 0) {
                CERROR("Error %d reading eadata \n", rc);
        } else {
                /* FIXME convert lov EA but fixed after verification test */
                *md_size = rc;
        }

        RETURN (rc);
}

int mdd_get_md_locked(const struct lu_env *env, struct mdd_object *obj,
                      void *md, int *md_size, const char *name)
{
        int rc = 0;
        mdd_read_lock(env, obj);
        rc = mdd_get_md(env, obj, md, md_size, name);
        mdd_read_unlock(env, obj);
        return rc;
}

static int mdd_lov_set_stripe_md(const struct lu_env *env,
                                 struct mdd_object *obj, struct lu_buf *buf,
                                 struct thandle *handle)
{
        struct mdd_device       *mdd = mdo2mdd(&obj->mod_obj);
        struct obd_device       *obd = mdd2obd_dev(mdd);
        struct obd_export       *lov_exp = obd->u.mds.mds_osc_exp;
        struct lov_stripe_md    *lsm = NULL;
        int rc;
        ENTRY;

        LASSERT(S_ISDIR(mdd_object_type(obj)) || S_ISREG(mdd_object_type(obj)));
        rc = obd_iocontrol(OBD_IOC_LOV_SETSTRIPE, lov_exp, 0,
                           &lsm, buf->lb_buf);
        if (rc)
                RETURN(rc);
        obd_free_memmd(lov_exp, &lsm);

        rc = mdd_xattr_set_txn(env, obj, buf, MDS_LOV_MD_NAME, 0, handle);

        CDEBUG(D_INFO, "set lov ea of "DFID" rc %d \n", PFID(mdo2fid(obj)), rc);
        RETURN(rc);
}

static int mdd_lov_set_dir_md(const struct lu_env *env,
                              struct mdd_object *obj, struct lu_buf *buf,
                              struct thandle *handle)
{
        struct lov_user_md *lum = NULL;
        int rc = 0;
        ENTRY;

        /*TODO check permission*/
        LASSERT(S_ISDIR(mdd_object_type(obj)));
        lum = (struct lov_user_md*)buf->lb_buf;

        /* if { size, offset, count } = { 0, -1, 0 } (i.e. all default
         * values specified) then delete default striping from dir. */
        if ((lum->lmm_stripe_size == 0 && lum->lmm_stripe_count == 0 &&
             lum->lmm_stripe_offset == (typeof(lum->lmm_stripe_offset))(-1)) ||
             /* lmm_stripe_size == -1 is deprecated in 1.4.6 */
             lum->lmm_stripe_size == (typeof(lum->lmm_stripe_size))(-1)){
                rc = mdd_xattr_set_txn(env, obj, &LU_BUF_NULL,
                                       MDS_LOV_MD_NAME, 0, handle);
                if (rc == -ENODATA)
                        rc = 0;
                CDEBUG(D_INFO, "delete lov ea of "DFID" rc %d \n",
                                PFID(mdo2fid(obj)), rc);
        } else {
                rc = mdd_lov_set_stripe_md(env, obj, buf, handle);
        }
        RETURN(rc);
}

int mdd_lov_set_md(const struct lu_env *env, struct mdd_object *pobj,
                   struct mdd_object *child, struct lov_mds_md *lmmp,
                   int lmm_size, struct thandle *handle, int set_stripe)
{
        int rc = 0;
        umode_t mode;
        struct lu_buf *buf;
        ENTRY;

        buf = mdd_buf_get(env, lmmp, lmm_size);
        mode = mdd_object_type(child);
        if (S_ISREG(mode) && lmm_size > 0) {
                if (set_stripe) {
                        rc = mdd_lov_set_stripe_md(env, child, buf, handle);
                } else {
                        rc = mdd_xattr_set_txn(env, child, buf,
                                               MDS_LOV_MD_NAME, 0, handle);
                }
        } else  if (S_ISDIR(mode)) {
                if (lmmp == NULL && lmm_size == 0) {
                        struct lov_mds_md *lmm = &mdd_env_info(env)->mti_lmm;
                        int size = sizeof(*lmm);

                        /* Get parent dir stripe and set */
                        if (pobj != NULL)
                                rc = mdd_get_md(env, pobj, lmm, &size,
                                                MDS_LOV_MD_NAME);
                        if (rc > 0) {
                                buf = mdd_buf_get(env, lmm, size);
                                rc = mdd_xattr_set_txn(env, child, buf,
                                               MDS_LOV_MD_NAME, 0, handle);
                                if (rc)
                                        CERROR("error on copy stripe info: rc "
                                                "= %d\n", rc);
                        }
                } else {
                        LASSERT(lmmp != NULL && lmm_size > 0);
                        rc = mdd_lov_set_dir_md(env, child, buf, handle);
                }
        }
        CDEBUG(D_INFO, "Set lov md %p size %d for fid "DFID" rc %d\n",
                        lmmp, lmm_size, PFID(mdo2fid(child)), rc);
        RETURN(rc);
}

/*
 * XXX: this is for create lsm object id, which should identify the lsm object
 * unique in the whole mds, as I see. But it seems, we still not need it
 * now. Right? So just borrow the ll_fid_build_ino().
 */
static obd_id mdd_lov_create_id(const struct lu_fid *fid)
{
        return ((fid_seq(fid) - 1) * LUSTRE_SEQ_MAX_WIDTH + fid_oid(fid));
}

static int mdd_lov_objid_alloc(const struct lu_env *env,
                               struct mdd_device *mdd)
{
        struct mdd_thread_info *info = mdd_env_info(env);
        struct mds_obd *mds = &mdd->mdd_obd_dev->u.mds;

        OBD_ALLOC(info->mti_oti.oti_objid,
                  mds->mds_lov_desc.ld_tgt_count * sizeof(obd_id));
        return (info->mti_oti.oti_objid == NULL ? -ENOMEM : 0);
}

static void mdd_lov_objid_update(const struct lu_env *env,
                          struct mdd_device *mdd)
{
        struct mdd_thread_info *info = mdd_env_info(env);
        mds_lov_update_objids(mdd->mdd_obd_dev, info->mti_oti.oti_objid);
}

static void mdd_lov_objid_from_lmm(const struct lu_env *env,
                                   struct mdd_device *mdd,
                                   struct lov_mds_md *lmm)
{
        struct mds_obd *mds = &mdd->mdd_obd_dev->u.mds;
        struct mdd_thread_info *info = mdd_env_info(env);
        mds_objids_from_lmm(info->mti_oti.oti_objid, lmm, &mds->mds_lov_desc);
}

static void mdd_lov_objid_free(const struct lu_env *env,
                               struct mdd_device *mdd)
{
        struct mdd_thread_info *info = mdd_env_info(env);
        struct mds_obd *mds = &mdd->mdd_obd_dev->u.mds;

        OBD_FREE(info->mti_oti.oti_objid,
                 mds->mds_lov_desc.ld_tgt_count * sizeof(obd_id));
        info->mti_oti.oti_objid = NULL;
}

void mdd_lov_create_finish(const struct lu_env *env,
                           struct mdd_device *mdd, int rc)
{
        struct mdd_thread_info *info = mdd_env_info(env);

        if (info->mti_oti.oti_objid != NULL) {
                if (rc == 0)
                        mdd_lov_objid_update(env, mdd);
                mdd_lov_objid_free(env, mdd);
        }
}

int mdd_lov_create(const struct lu_env *env, struct mdd_device *mdd,
                   struct mdd_object *parent, struct mdd_object *child,
                   struct lov_mds_md **lmm, int *lmm_size,
                   const struct md_create_spec *spec, struct lu_attr *la)
{
        struct obd_device     *obd = mdd2obd_dev(mdd);
        struct obd_export     *lov_exp = obd->u.mds.mds_osc_exp;
        struct obdo           *oa;
        struct lov_stripe_md  *lsm = NULL;
        const void            *eadata = spec->u.sp_ea.eadata;
        __u32                  create_flags = spec->sp_cr_flags;
        int                    rc = 0;
        struct obd_trans_info *oti = &mdd_env_info(env)->mti_oti;
        struct dt_object      *next;
        ENTRY;

        if (create_flags & MDS_OPEN_DELAY_CREATE ||
            !(create_flags & FMODE_WRITE))
                RETURN(0);

        oti_init(oti, NULL);
        rc = mdd_lov_objid_alloc(env, mdd);
        if (rc != 0)
                RETURN(rc);

        /* replay case, should get lov from eadata */
        if (spec->u.sp_ea.no_lov_create != 0) {
                mdd_lov_objid_from_lmm(env, mdd, (struct lov_mds_md *)eadata);
                RETURN(0);
        }

        if (OBD_FAIL_CHECK_ONCE(OBD_FAIL_MDS_ALLOC_OBDO))
                GOTO(out_ids, rc = -ENOMEM);

        LASSERT(lov_exp != NULL);
        oa = obdo_alloc();
        if (oa == NULL)
                GOTO(out_ids, rc = -ENOMEM);

        oa->o_uid = 0; /* must have 0 uid / gid on OST */
        oa->o_gid = 0;
        oa->o_gr = FILTER_GROUP_MDS0 + mdd2lu_dev(mdd)->ld_site->ls_node_id;
        oa->o_mode = S_IFREG | 0600;
        oa->o_id = mdd_lov_create_id(lu_object_fid(mdd2lu_obj(child)));
        oa->o_valid = OBD_MD_FLID | OBD_MD_FLTYPE | OBD_MD_FLFLAGS |
                OBD_MD_FLMODE | OBD_MD_FLUID | OBD_MD_FLGID | OBD_MD_FLGROUP;
        oa->o_size = 0;

        next = mdd_object_child(child);
        if (!(create_flags & MDS_OPEN_HAS_OBJS)) {
                if (create_flags & MDS_OPEN_HAS_EA) {
                        LASSERT(eadata != NULL);
                        rc = obd_iocontrol(OBD_IOC_LOV_SETSTRIPE, lov_exp,
                                           0, &lsm, (void*)eadata);
                        if (rc)
                                GOTO(out_oa, rc);
                        lsm->lsm_object_id = oa->o_id;
                        lsm->lsm_object_gr = oa->o_gr;
                } else if (parent != NULL) {
                        /* get lov ea from parent and set to lov */
                        struct lov_mds_md *__lmm;
                        int __lmm_size, returned_lmm_size;
                        
                        __lmm_size = mdd_lov_mdsize(env, mdd);
                        returned_lmm_size = __lmm_size;

                        OBD_ALLOC(__lmm, __lmm_size);
                        if (__lmm == NULL)
                                GOTO(out_oa, rc = -ENOMEM);

                        rc = mdd_get_md_locked(env, parent, __lmm,
                                               &returned_lmm_size,
                                               MDS_LOV_MD_NAME);
                        if (rc > 0)
                                rc = obd_iocontrol(OBD_IOC_LOV_SETSTRIPE,
                                                   lov_exp, 0, &lsm, __lmm);
                        OBD_FREE(__lmm, __lmm_size);
                        if (rc)
                                GOTO(out_oa, rc);
                }

                rc = obd_create(lov_exp, oa, &lsm, oti);
                if (rc) {
                        if (rc > 0) {
                                CERROR("Create error for "DFID": %d\n",
                                       PFID(mdo2fid(child)), rc);
                                rc = -EIO;
                        }
                        GOTO(out_oa, rc);
                }
                LASSERT(lsm->lsm_object_gr >= FILTER_GROUP_MDS0);
        } else {
                LASSERT(eadata != NULL);
                rc = obd_iocontrol(OBD_IOC_LOV_SETEA, lov_exp, 0, &lsm,
                                   (void*)eadata);
                if (rc)
                        GOTO(out_oa, rc);
                lsm->lsm_object_id = oa->o_id;
                lsm->lsm_object_gr = oa->o_gr;
        }
        
        /*
         * Sometimes, we may truncate some object(without lsm) then open (with
         * write flags)it, so creating lsm above.  The Nonzero(truncated) size
         * should tell ost. since size attr is in charged by OST.
         */
        if (la->la_size && la->la_valid & LA_SIZE) {
                struct obd_info *oinfo = &mdd_env_info(env)->mti_oi;

                memset(oinfo, 0, sizeof(*oinfo));

                oa->o_size = la->la_size;
                
                /* When setting attr to ost, FLBKSZ is not needed. */
                oa->o_valid &= ~OBD_MD_FLBLKSZ;
                obdo_from_la(oa, la, OBD_MD_FLTYPE | OBD_MD_FLATIME |
                             OBD_MD_FLMTIME | OBD_MD_FLCTIME | OBD_MD_FLSIZE);

                /*
                 * XXX: Pack lustre id to OST, in OST, it will be packed by
                 * filter_fid, but can not see what is the usages. So just pack
                 * o_seq o_ver here, maybe fix it after this cycle.
                 */
                oa->o_fid = fid_seq(lu_object_fid(mdd2lu_obj(child)));
                oa->o_generation = fid_oid(lu_object_fid(mdd2lu_obj(child)));
                oa->o_valid |= OBD_MD_FLFID | OBD_MD_FLGENER;
                oinfo->oi_oa = oa;
                oinfo->oi_md = lsm;
                oinfo->oi_capa = next->do_ops->do_capa_get(env, next, NULL,
                                                          CAPA_OPC_MDS_DEFAULT);
                if (IS_ERR(oinfo->oi_capa))
                        oinfo->oi_capa = NULL;

                rc = obd_setattr(lov_exp, oinfo, oti);
                capa_put(oinfo->oi_capa);
                if (rc) {
                        CERROR("Error setting attrs for "DFID": rc %d\n",
                               PFID(mdo2fid(child)), rc);
                        if (rc > 0) {
                                CERROR("obd_setattr for "DFID" rc %d\n",
                                        PFID(mdo2fid(child)), rc);
                                rc = -EIO;
                        }
                        GOTO(out_oa, rc);
                }
        }
        
        /* blksize should be changed after create data object */
        la->la_valid |= LA_BLKSIZE;
        la->la_blksize = oa->o_blksize;

        rc = obd_packmd(lov_exp, lmm, lsm);
        if (rc < 0) {
                CERROR("cannot pack lsm, err = %d\n", rc);
                GOTO(out_oa, rc);
        }
        *lmm_size = rc;
        rc = 0;
        EXIT;
out_oa:
        oti_free_cookies(oti);
        obdo_free(oa);
out_ids:
        if (lsm)
                obd_free_memmd(lov_exp, &lsm);
        if (rc != 0)
                mdd_lov_objid_free(env, mdd);
        return rc;
}

int mdd_unlink_log(const struct lu_env *env, struct mdd_device *mdd,
                   struct mdd_object *mdd_cobj, struct md_attr *ma)
{
        struct obd_device *obd = mdd2obd_dev(mdd);

        LASSERT(ma->ma_valid & MA_LOV);

        if ((ma->ma_cookie_size > 0) &&
            (mds_log_op_unlink(obd, ma->ma_lmm, ma->ma_lmm_size,
                              ma->ma_cookie, ma->ma_cookie_size) > 0)) {
                ma->ma_valid |= MA_COOKIE;
        }
        return 0;
}

int mdd_lov_setattr_async(const struct lu_env *env, struct mdd_object *obj,
                          struct lov_mds_md *lmm, int lmm_size)
{
        struct mdd_device       *mdd = mdo2mdd(&obj->mod_obj);
        struct obd_device       *obd = mdd2obd_dev(mdd);
        struct lu_attr          *tmp_la = &mdd_env_info(env)->mti_la;
        struct dt_object        *next = mdd_object_child(obj);
        __u32  seq  = lu_object_fid(mdd2lu_obj(obj))->f_seq;
        __u32  oid  = lu_object_fid(mdd2lu_obj(obj))->f_oid;
        struct obd_capa *oc;
        int rc = 0;
        ENTRY;

        rc = next->do_ops->do_attr_get(env, next, tmp_la,
                                       mdd_object_capa(env, obj));
        if (rc)
                RETURN(rc);

        oc = next->do_ops->do_capa_get(env, next, NULL, CAPA_OPC_MDS_DEFAULT);
        if (IS_ERR(oc))
                oc = NULL;
        rc = mds_osc_setattr_async(obd, tmp_la->la_uid, tmp_la->la_gid, lmm,
                                   lmm_size, NULL, seq, oid, oc);
        capa_put(oc);

        RETURN(rc);
}