/* -*- mode: c; c-basic-offset: 8; indent-tabs-mode: nil; -*-
 * vim:expandtab:shiftwidth=8:tabstop=8:
 *
 *  lustre/cmm/cmm_split.c
 *  Lustre splitting dir
 *
 *  Copyright (c) 2006 Cluster File Systems, Inc.
 *   Author: Alex thomas <alex@clusterfs.com>
 *           Wang Di     <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 <obd_class.h>
#include <lustre_fid.h>
#include <lustre_mds.h>
#include <lustre/lustre_idl.h>
#include "cmm_internal.h"
#include "mdc_internal.h"

#define CMM_NO_SPLIT_EXPECTED   0
#define CMM_EXPECT_SPLIT        1
#define CMM_NO_SPLITTABLE       2

enum {
        SPLIT_SIZE =  64*1024
};

static inline struct lu_fid* cmm2_fid(struct cmm_object *obj)
{
       return &(obj->cmo_obj.mo_lu.lo_header->loh_fid);
}

static int cmm_expect_splitting(const struct lu_env *env,
                                struct md_object *mo,
                                struct md_attr *ma)
{
        struct cmm_device *cmm = cmm_obj2dev(md2cmm_obj(mo));
        struct lu_fid *fid = NULL;
        int rc = CMM_EXPECT_SPLIT;
        ENTRY;

        if (cmm->cmm_tgt_count == 0)
                GOTO(cleanup, rc = CMM_NO_SPLIT_EXPECTED);

        if (ma->ma_attr.la_size < SPLIT_SIZE)
                GOTO(cleanup, rc = CMM_NO_SPLIT_EXPECTED);

        if (ma->ma_lmv_size)
                GOTO(cleanup, rc = CMM_NO_SPLIT_EXPECTED);
        OBD_ALLOC_PTR(fid);
        rc = cmm_child_ops(cmm)->mdo_root_get(env, cmm->cmm_child, fid);
        if (rc)
                GOTO(cleanup, rc);

        rc = CMM_EXPECT_SPLIT;

        if (lu_fid_eq(fid, cmm2_fid(md2cmm_obj(mo))))
                GOTO(cleanup, rc = CMM_NO_SPLIT_EXPECTED);

cleanup:
        if (fid)
                OBD_FREE_PTR(fid);
        RETURN(rc);
}

#define cmm_md_size(stripes)                            \
       (sizeof(struct lmv_stripe_md) + (stripes) * sizeof(struct lu_fid))

static int cmm_alloc_fid(const struct lu_env *env, struct cmm_device *cmm,
                         struct lu_fid *fid, int count)
{
        struct  mdc_device *mc, *tmp;
        int rc = 0, i = 0;

        LASSERT(count == cmm->cmm_tgt_count);
        /* FIXME: this spin_lock maybe not proper,
         * because fid_alloc may need RPC */
        spin_lock(&cmm->cmm_tgt_guard);
        list_for_each_entry_safe(mc, tmp, &cmm->cmm_targets,
                                 mc_linkage) {
                LASSERT(cmm->cmm_local_num != mc->mc_num);

                rc = obd_fid_alloc(mc->mc_desc.cl_exp, &fid[i], NULL);
                if (rc > 0) {
                        struct lu_site *ls;

                        ls = cmm->cmm_md_dev.md_lu_dev.ld_site;
                        rc = fld_client_create(ls->ls_client_fld,
                                               fid_seq(&fid[i]),
                                               mc->mc_num, env);
                }
                if (rc < 0) {
                        spin_unlock(&cmm->cmm_tgt_guard);
                        RETURN(rc);
                }
                i++;
        }
        spin_unlock(&cmm->cmm_tgt_guard);
        LASSERT(i == count);
        if (rc == 1)
                rc = 0;
        RETURN(rc);
}

struct cmm_object *cmm_object_find(const struct lu_env *env,
                                   struct cmm_device *d,
                                   const struct lu_fid *f,
                                   struct lustre_capa *capa)
{
        struct lu_object *o;
        struct cmm_object *m;
        ENTRY;

        o = lu_object_find(env, d->cmm_md_dev.md_lu_dev.ld_site, f,
                           capa);
        if (IS_ERR(o))
                m = (struct cmm_object *)o;
        else
                m = lu2cmm_obj(lu_object_locate(o->lo_header,
                               d->cmm_md_dev.md_lu_dev.ld_type));
        RETURN(m);
}

static inline void cmm_object_put(const struct lu_env *env,
                                  struct cmm_object *o)
{
        lu_object_put(env, &o->cmo_obj.mo_lu);
}

static int cmm_creat_remote_obj(const struct lu_env *env,
                                struct cmm_device *cmm,
                                struct lu_fid *fid, struct md_attr *ma,
                                const struct lmv_stripe_md *lmv,
                                int lmv_size)
{
        struct cmm_object *obj;
        struct md_create_spec *spec;
        int rc;
        ENTRY;

        /* XXX Since capablity will not work with split. so we
         * pass NULL capablity here */
        obj = cmm_object_find(env, cmm, fid, NULL);
        if (IS_ERR(obj))
                RETURN(PTR_ERR(obj));

        OBD_ALLOC_PTR(spec);

        spec->u.sp_ea.fid = fid;
        spec->u.sp_ea.eadata = lmv;
        spec->u.sp_ea.eadatalen = lmv_size;
        spec->sp_cr_flags |= MDS_CREATE_SLAVE_OBJ;
        rc = mo_object_create(env, md_object_next(&obj->cmo_obj),
                              spec, ma);
        OBD_FREE_PTR(spec);

        cmm_object_put(env, obj);
        RETURN(rc);
}

static int cmm_create_slave_objects(const struct lu_env *env,
                                    struct md_object *mo, struct md_attr *ma)
{
        struct cmm_device *cmm = cmm_obj2dev(md2cmm_obj(mo));
        struct lmv_stripe_md *lmv = NULL, *slave_lmv = NULL;
        int lmv_size, i, rc;
        struct lu_fid *lf = cmm2_fid(md2cmm_obj(mo));
        ENTRY;

        lmv_size = cmm_md_size(cmm->cmm_tgt_count + 1);

        /* This lmv will be free after finish splitting. */
        OBD_ALLOC(lmv, lmv_size);
        if (!lmv)
                RETURN(-ENOMEM);

        lmv->mea_master = cmm->cmm_local_num;
        lmv->mea_magic = MEA_MAGIC_HASH_SEGMENT;
        lmv->mea_count = cmm->cmm_tgt_count + 1;

        lmv->mea_ids[0] = *lf;

        rc = cmm_alloc_fid(env, cmm, &lmv->mea_ids[1],
                           cmm->cmm_tgt_count);
        if (rc)
                GOTO(cleanup, rc);

        OBD_ALLOC_PTR(slave_lmv);
        if (!slave_lmv)
                GOTO(cleanup, rc = -ENOMEM);

        slave_lmv->mea_master = cmm->cmm_local_num;
        slave_lmv->mea_magic = MEA_MAGIC_HASH_SEGMENT;
        slave_lmv->mea_count = 0;
        for (i = 1; i < cmm->cmm_tgt_count + 1; i ++) {
                rc = cmm_creat_remote_obj(env, cmm, &lmv->mea_ids[i], ma,
                                          slave_lmv, sizeof(slave_lmv));
                if (rc)
                        GOTO(cleanup, rc);
        }

        ma->ma_lmv_size = lmv_size;
        ma->ma_lmv = lmv;
cleanup:
        if (slave_lmv)
                OBD_FREE_PTR(slave_lmv);
        RETURN(rc);
}

static int cmm_send_split_pages(const struct lu_env *env,
                                struct md_object *mo, struct lu_rdpg *rdpg,
                                struct lu_fid *fid, int len)
{
        struct cmm_device *cmm = cmm_obj2dev(md2cmm_obj(mo));
        struct cmm_object *obj;
        int rc = 0;
        ENTRY;

        obj = cmm_object_find(env, cmm, fid, NULL);
        if (IS_ERR(obj))
                RETURN(PTR_ERR(obj));

        rc = mdc_send_page(cmm, env, md_object_next(&obj->cmo_obj),
                           rdpg->rp_pages[0], len);
        cmm_object_put(env, obj);
        RETURN(rc);
}

static int cmm_remove_entries(const struct lu_env *env,
                              struct md_object *mo, struct lu_rdpg *rdpg,
                              __u32 hash_end, __u32 *len)
{
        struct lu_dirpage *dp;
        struct lu_dirent  *ent;
        int rc = 0;
        ENTRY;

        kmap(rdpg->rp_pages[0]);
        dp = page_address(rdpg->rp_pages[0]);
        for (ent = lu_dirent_start(dp); ent != NULL;
                          ent = lu_dirent_next(ent)) {
                if (ent->lde_hash < hash_end) {
                        if (strncmp(ent->lde_name, ".", ent->lde_namelen) &&
                            strncmp(ent->lde_name, "..", ent->lde_namelen)) {
                                char *name;
                                /* FIXME: Here we allocate name for each name,
                                 * maybe stupid, but can not find better way.
                                 * will find better way */
                                OBD_ALLOC(name, ent->lde_namelen + 1);
                                memcpy(name, ent->lde_name, ent->lde_namelen);
                                rc = mdo_name_remove(env, md_object_next(mo),
                                                     name);
                                OBD_FREE(name, ent->lde_namelen + 1);
                        }
                        if (rc) {
                                /* FIXME: Do not know why it return -ENOENT
                                 * in some case
                                 * */
                                if (rc != -ENOENT)
                                        GOTO(unmap, rc);
                        }
                } else {
                        if (ent != lu_dirent_start(dp))
                                *len = (int)((__u32)ent - (__u32)dp);
                        else
                                *len = 0;
                        GOTO(unmap, rc);
                }
        }
        *len = CFS_PAGE_SIZE;
unmap:
        kunmap(rdpg->rp_pages[0]);
        RETURN(rc);
}

static int cmm_split_entries(const struct lu_env *env,
                             struct md_object *mo, struct lu_rdpg *rdpg,
                             struct lu_fid *lf, __u32 end)
{
        int rc, done = 0;
        ENTRY;

        LASSERTF(rdpg->rp_npages == 1, "Now Only support split 1 page each time"
                        "npages %d \n", rdpg->rp_npages);
        /* Read splitted page and send them to the slave master */
        do {
                struct lu_dirpage *ldp;
                __u32  len = 0;

                /* init page with '0' */
                memset(kmap(rdpg->rp_pages[0]), 0, CFS_PAGE_SIZE);
                kunmap(rdpg->rp_pages[0]);

                rc = mo_readpage(env, md_object_next(mo), rdpg);
                /* -E2BIG means it already reach the end of the dir */
                if (rc) {
                        if (rc != -ERANGE) {
                                if (rc == -E2BIG)
                                        rc = 0;
                                RETURN(rc);
                        }
                }

                /* Remove the old entries */
                rc = cmm_remove_entries(env, mo, rdpg, end, &len);
                if (rc)
                        RETURN(rc);

                /* Send page to slave object */
                if (len > 0) {
                        rc = cmm_send_split_pages(env, mo, rdpg, lf, len);
                        if (rc)
                                RETURN(rc);
                }

                kmap(rdpg->rp_pages[0]);
                ldp = page_address(rdpg->rp_pages[0]);
                if (ldp->ldp_hash_end >= end) {
                        done = 1;
                }
                rdpg->rp_hash = ldp->ldp_hash_end;
                kunmap(rdpg->rp_pages[0]);
        } while (!done);

        RETURN(rc);
}
#define SPLIT_PAGE_COUNT 1
static int cmm_scan_and_split(const struct lu_env *env,
                              struct md_object *mo, struct md_attr *ma)
{
        struct cmm_device *cmm = cmm_obj2dev(md2cmm_obj(mo));
        __u32 hash_segement;
        struct lu_rdpg   *rdpg = NULL;
        int rc = 0, i;

        OBD_ALLOC_PTR(rdpg);
        if (!rdpg)
                RETURN(-ENOMEM);

        rdpg->rp_npages = SPLIT_PAGE_COUNT;
        rdpg->rp_count  = CFS_PAGE_SIZE * rdpg->rp_npages;

        OBD_ALLOC(rdpg->rp_pages, rdpg->rp_npages * sizeof rdpg->rp_pages[0]);
        if (rdpg->rp_pages == NULL)
                GOTO(free_rdpg, rc = -ENOMEM);

        for (i = 0; i < rdpg->rp_npages; i++) {
                rdpg->rp_pages[i] = alloc_pages(GFP_KERNEL, 0);
                if (rdpg->rp_pages[i] == NULL)
                        GOTO(cleanup, rc = -ENOMEM);
        }

        hash_segement = MAX_HASH_SIZE / (cmm->cmm_tgt_count + 1);
        for (i = 1; i < cmm->cmm_tgt_count + 1; i++) {
                struct lu_fid *lf = &ma->ma_lmv->mea_ids[i];
                __u32 hash_end;

                rdpg->rp_hash = i * hash_segement;
                hash_end = rdpg->rp_hash + hash_segement;
                rc = cmm_split_entries(env, mo, rdpg, lf, hash_end);
                if (rc)
                        GOTO(cleanup, rc);
        }
cleanup:
        for (i = 0; i < rdpg->rp_npages; i++)
                if (rdpg->rp_pages[i] != NULL)
                        __free_pages(rdpg->rp_pages[i], 0);
        if (rdpg->rp_pages)
                OBD_FREE(rdpg->rp_pages, rdpg->rp_npages *
                                         sizeof rdpg->rp_pages[0]);
free_rdpg:
        if (rdpg)
                OBD_FREE_PTR(rdpg);

        RETURN(rc);
}

int cml_try_to_split(const struct lu_env *env, struct md_object *mo)
{
        struct cmm_device *cmm = cmm_obj2dev(md2cmm_obj(mo));
        struct md_attr *ma;
        int rc = 0;
        ENTRY;

        LASSERT(S_ISDIR(lu_object_attr(&mo->mo_lu)));

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

        ma->ma_need = MA_INODE|MA_LMV;
        rc = mo_attr_get(env, mo, ma);
        if (rc)
                GOTO(cleanup, ma);

        /* step1: checking whether the dir need to be splitted */
        rc = cmm_expect_splitting(env, mo, ma);
        if (rc != CMM_EXPECT_SPLIT)
                GOTO(cleanup, rc = 0);

        /* Disable trans for splitting, since there will be
         * so many trans in this one ops, confilct with current
         * recovery design */
        rc = cmm_upcall(env, &cmm->cmm_md_dev, MD_NO_TRANS);
        if (rc)
                GOTO(cleanup, rc = 0);

        /* step2: create slave objects */
        rc = cmm_create_slave_objects(env, mo, ma);
        if (rc)
                GOTO(cleanup, ma);

        /* step3: scan and split the object */
        rc = cmm_scan_and_split(env, mo, ma);
        if (rc)
                GOTO(cleanup, ma);

        /* step4: set mea to the master object */
        rc = mo_xattr_set(env, md_object_next(mo), ma->ma_lmv,
                          ma->ma_lmv_size, MDS_LMV_MD_NAME, 0);

        if (rc == -ERESTART)
                CWARN("Dir"DFID" has been split \n",
                                PFID(lu_object_fid(&mo->mo_lu)));
cleanup:
        if (ma->ma_lmv_size && ma->ma_lmv)
                OBD_FREE(ma->ma_lmv, ma->ma_lmv_size);

        OBD_FREE_PTR(ma);
        RETURN(rc);
}