lustre/osd-zfs/osd_index.c

   1 /*
   2  * GPL HEADER START
   3  *
   4  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
   5  *
   6  * This program is free software; you can redistribute it and/or modify
   7  * it under the terms of the GNU General Public License version 2 only,
   8  * as published by the Free Software Foundation.
   9  *
  10  * This program is distributed in the hope that it will be useful, but
  11  * WITHOUT ANY WARRANTY; without even the implied warranty of
  12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  13  * General Public License version 2 for more details (a copy is included
  14  * in the LICENSE file that accompanied this code).
  15  *
  16  * You should have received a copy of the GNU General Public License
  17  * version 2 along with this program; If not, see
  18  * http://www.sun.com/software/products/lustre/docs/GPLv2.pdf
  19  *
  20  * Please contact Sun Microsystems, Inc., 4150 Network Circle, Santa Clara,
  21  * CA 95054 USA or visit www.sun.com if you need additional information or
  22  * have any questions.
  23  *
  24  * GPL HEADER END
  25  */
  26 /*
  27  * Copyright (c) 2009, 2010, Oracle and/or its affiliates. All rights reserved.
  28  * Use is subject to license terms.
  29  *
  30  * Copyright (c) 2012, 2015, Intel Corporation.
  31  */
  32 /*
  33  * This file is part of Lustre, http://www.lustre.org/
  34  * Lustre is a trademark of Sun Microsystems, Inc.
  35  *
  36  * lustre/osd-zfs/osd_index.c
  37  *
  38  * Author: Alex Zhuravlev <bzzz@whamcloud.com>
  39  * Author: Mike Pershin <tappro@whamcloud.com>
  40  */
  41
  42 #define DEBUG_SUBSYSTEM S_OSD
  43
  44 #include <lustre_ver.h>
  45 #include <libcfs/libcfs.h>
  46 #include <obd_support.h>
  47 #include <lustre_net.h>
  48 #include <obd.h>
  49 #include <obd_class.h>
  50 #include <lustre_disk.h>
  51 #include <lustre_fid.h>
  52
  53 #include "osd_internal.h"
  54
  55 #include <sys/dnode.h>
  56 #include <sys/dbuf.h>
  57 #include <sys/spa.h>
  58 #include <sys/stat.h>
  59 #include <sys/zap.h>
  60 #include <sys/spa_impl.h>
  61 #include <sys/zfs_znode.h>
  62 #include <sys/dmu_tx.h>
  63 #include <sys/dmu_objset.h>
  64 #include <sys/dsl_prop.h>
  65 #include <sys/sa_impl.h>
  66 #include <sys/txg.h>
  67
  68 static inline int osd_object_is_zap(dmu_buf_t *db)
  69 {
  70         dmu_buf_impl_t *dbi = (dmu_buf_impl_t *) db;
  71         dnode_t *dn;
  72         int rc;
  73
  74         DB_DNODE_ENTER(dbi);
  75         dn = DB_DNODE(dbi);
  76         rc = (dn->dn_type == DMU_OT_DIRECTORY_CONTENTS ||
  77                         dn->dn_type == DMU_OT_USERGROUP_USED);
  78         DB_DNODE_EXIT(dbi);
  79
  80         return rc;
  81 }
  82
  83 /* We don't actually have direct access to the zap_hashbits() function
  84  * so just pretend like we do for now.  If this ever breaks we can look at
  85  * it at that time. */
  86 #define zap_hashbits(zc) 48
  87 /*
  88  * ZFS hash format:
  89  * | cd (16 bits) | hash (48 bits) |
  90  * we need it in other form:
  91  * |0| hash (48 bit) | cd (15 bit) |
  92  * to be a full 64-bit ordered hash so that Lustre readdir can use it to merge
  93  * the readdir hashes from multiple directory stripes uniformly on the client.
  94  * Another point is sign bit, the hash range should be in [0, 2^63-1] because
  95  * loff_t (for llseek) needs to be a positive value.  This means the "cd" field
  96  * should only be the low 15 bits.
  97  */
  98 uint64_t osd_zap_cursor_serialize(zap_cursor_t *zc)
  99 {
 100         uint64_t zfs_hash = zap_cursor_serialize(zc) & (~0ULL >> 1);
 101
 102         return (zfs_hash >> zap_hashbits(zc)) |
 103                 (zfs_hash << (63 - zap_hashbits(zc)));
 104 }
 105
 106 void osd_zap_cursor_init_serialized(zap_cursor_t *zc, struct objset *os,
 107                                     uint64_t id, uint64_t dirhash)
 108 {
 109         uint64_t zfs_hash = ((dirhash << zap_hashbits(zc)) & (~0ULL >> 1)) |
 110                 (dirhash >> (63 - zap_hashbits(zc)));
 111
 112         zap_cursor_init_serialized(zc, os, id, zfs_hash);
 113 }
 114
 115 int osd_zap_cursor_init(zap_cursor_t **zc, struct objset *os,
 116                         uint64_t id, uint64_t dirhash)
 117 {
 118         zap_cursor_t *t;
 119
 120         OBD_ALLOC_PTR(t);
 121         if (unlikely(t == NULL))
 122                 return -ENOMEM;
 123
 124         osd_zap_cursor_init_serialized(t, os, id, dirhash);
 125         *zc = t;
 126
 127         return 0;
 128 }
 129
 130 void osd_zap_cursor_fini(zap_cursor_t *zc)
 131 {
 132         zap_cursor_fini(zc);
 133         OBD_FREE_PTR(zc);
 134 }
 135
 136 static inline void osd_obj_cursor_init_serialized(zap_cursor_t *zc,
 137                                                  struct osd_object *o,
 138                                                  uint64_t dirhash)
 139 {
 140         struct osd_device *d = osd_obj2dev(o);
 141         osd_zap_cursor_init_serialized(zc, d->od_os,
 142                                        o->oo_db->db_object, dirhash);
 143 }
 144
 145 static inline int osd_obj_cursor_init(zap_cursor_t **zc, struct osd_object *o,
 146                         uint64_t dirhash)
 147 {
 148         struct osd_device *d = osd_obj2dev(o);
 149         return osd_zap_cursor_init(zc, d->od_os, o->oo_db->db_object, dirhash);
 150 }
 151
 152 static struct dt_it *osd_index_it_init(const struct lu_env *env,
 153                                        struct dt_object *dt,
 154                                        __u32 unused)
 155 {
 156         struct osd_thread_info  *info = osd_oti_get(env);
 157         struct osd_zap_it       *it;
 158         struct osd_object       *obj = osd_dt_obj(dt);
 159         struct lu_object        *lo  = &dt->do_lu;
 160         int                      rc;
 161         ENTRY;
 162
 163         if (obj->oo_destroyed)
 164                 RETURN(ERR_PTR(-ENOENT));
 165
 166         LASSERT(lu_object_exists(lo));
 167         LASSERT(obj->oo_db);
 168         LASSERT(osd_object_is_zap(obj->oo_db));
 169         LASSERT(info);
 170
 171         OBD_SLAB_ALLOC_PTR_GFP(it, osd_zapit_cachep, GFP_NOFS);
 172         if (it == NULL)
 173                 RETURN(ERR_PTR(-ENOMEM));
 174
 175         rc = osd_obj_cursor_init(&it->ozi_zc, obj, 0);
 176         if (rc != 0) {
 177                 OBD_SLAB_FREE_PTR(it, osd_zapit_cachep);
 178                 RETURN(ERR_PTR(rc));
 179         }
 180
 181         it->ozi_obj   = obj;
 182         it->ozi_reset = 1;
 183         lu_object_get(lo);
 184
 185         RETURN((struct dt_it *)it);
 186 }
 187
 188 static void osd_index_it_fini(const struct lu_env *env, struct dt_it *di)
 189 {
 190         struct osd_zap_it       *it     = (struct osd_zap_it *)di;
 191         struct osd_object       *obj;
 192         ENTRY;
 193
 194         LASSERT(it);
 195         LASSERT(it->ozi_obj);
 196
 197         obj = it->ozi_obj;
 198
 199         osd_zap_cursor_fini(it->ozi_zc);
 200         lu_object_put(env, &obj->oo_dt.do_lu);
 201         OBD_SLAB_FREE_PTR(it, osd_zapit_cachep);
 202
 203         EXIT;
 204 }
 205
 206
 207 static void osd_index_it_put(const struct lu_env *env, struct dt_it *di)
 208 {
 209         /* PBS: do nothing : ref are incremented at retrive and decreamented
 210          *      next/finish. */
 211 }
 212
 213 static inline void osd_it_append_attrs(struct lu_dirent *ent, __u32 attr,
 214                                        int len, __u16 type)
 215 {
 216         const unsigned    align = sizeof(struct luda_type) - 1;
 217         struct luda_type *lt;
 218
 219         /* check if file type is required */
 220         if (attr & LUDA_TYPE) {
 221                 len = (len + align) & ~align;
 222
 223                 lt = (void *)ent->lde_name + len;
 224                 lt->lt_type = cpu_to_le16(DTTOIF(type));
 225                 ent->lde_attrs |= LUDA_TYPE;
 226         }
 227
 228         ent->lde_attrs = cpu_to_le32(ent->lde_attrs);
 229 }
 230
 231 /**
 232  * Get the object's FID from its LMA EA.
 233  *
 234  * \param[in] env       pointer to the thread context
 235  * \param[in] osd       pointer to the OSD device
 236  * \param[in] oid       the object's local identifier
 237  * \param[out] fid      the buffer to hold the object's FID
 238  *
 239  * \retval              0 for success
 240  * \retval              negative error number on failure
 241  */
 242 static int osd_get_fid_by_oid(const struct lu_env *env, struct osd_device *osd,
 243                               uint64_t oid, struct lu_fid *fid)
 244 {
 245         struct objset           *os       = osd->od_os;
 246         struct osd_thread_info  *oti      = osd_oti_get(env);
 247         struct lustre_mdt_attrs *lma      =
 248                         (struct lustre_mdt_attrs *)oti->oti_buf;
 249         struct lu_buf            buf;
 250         nvlist_t                *sa_xattr = NULL;
 251         sa_handle_t             *sa_hdl   = NULL;
 252         uchar_t                 *nv_value = NULL;
 253         uint64_t                 xattr    = ZFS_NO_OBJECT;
 254         int                      size     = 0;
 255         int                      rc;
 256         ENTRY;
 257
 258         rc = __osd_xattr_load(osd, oid, &sa_xattr);
 259         if (rc == -ENOENT)
 260                 goto regular;
 261
 262         if (rc != 0)
 263                 GOTO(out, rc);
 264
 265         rc = -nvlist_lookup_byte_array(sa_xattr, XATTR_NAME_LMA, &nv_value,
 266                                        &size);
 267         if (rc == -ENOENT)
 268                 goto regular;
 269
 270         if (rc != 0)
 271                 GOTO(out, rc);
 272
 273         if (unlikely(size > sizeof(oti->oti_buf)))
 274                 GOTO(out, rc = -ERANGE);
 275
 276         memcpy(lma, nv_value, size);
 277
 278         goto found;
 279
 280 regular:
 281         rc = -sa_handle_get(os, oid, NULL, SA_HDL_PRIVATE, &sa_hdl);
 282         if (rc != 0)
 283                 GOTO(out, rc);
 284
 285         rc = -sa_lookup(sa_hdl, SA_ZPL_XATTR(osd), &xattr, 8);
 286         sa_handle_destroy(sa_hdl);
 287         if (rc != 0)
 288                 GOTO(out, rc);
 289
 290         buf.lb_buf = lma;
 291         buf.lb_len = sizeof(oti->oti_buf);
 292         rc = __osd_xattr_get_large(env, osd, xattr, &buf,
 293                                    XATTR_NAME_LMA, &size);
 294         if (rc != 0)
 295                 GOTO(out, rc);
 296
 297 found:
 298         if (size < sizeof(*lma))
 299                 GOTO(out, rc = -EIO);
 300
 301         lustre_lma_swab(lma);
 302         if (unlikely((lma->lma_incompat & ~LMA_INCOMPAT_SUPP) ||
 303                      CFS_FAIL_CHECK(OBD_FAIL_OSD_LMA_INCOMPAT))) {
 304                 CWARN("%s: unsupported incompat LMA feature(s) %#x for "
 305                       "oid = "LPX64"\n", osd->od_svname,
 306                       lma->lma_incompat & ~LMA_INCOMPAT_SUPP, oid);
 307                 GOTO(out, rc = -EOPNOTSUPP);
 308         } else {
 309                 *fid = lma->lma_self_fid;
 310                 GOTO(out, rc = 0);
 311         }
 312
 313 out:
 314         if (sa_xattr != NULL)
 315                 nvlist_free(sa_xattr);
 316         return rc;
 317 }
 318
 319 /*
 320  * As we don't know FID, we can't use LU object, so this function
 321  * partially duplicate __osd_xattr_get() which is built around
 322  * LU-object and uses it to cache data like regular EA dnode, etc
 323  */
 324 static int osd_find_parent_by_dnode(const struct lu_env *env,
 325                                     struct dt_object *o,
 326                                     struct lu_fid *fid)
 327 {
 328         struct osd_device       *osd = osd_obj2dev(osd_dt_obj(o));
 329         sa_handle_t             *sa_hdl;
 330         uint64_t                 dnode = ZFS_NO_OBJECT;
 331         int                      rc;
 332         ENTRY;
 333
 334         /* first of all, get parent dnode from own attributes */
 335         LASSERT(osd_dt_obj(o)->oo_db);
 336         rc = -sa_handle_get(osd->od_os, osd_dt_obj(o)->oo_db->db_object,
 337                             NULL, SA_HDL_PRIVATE, &sa_hdl);
 338         if (rc != 0)
 339                 RETURN(rc);
 340
 341         rc = -sa_lookup(sa_hdl, SA_ZPL_PARENT(osd), &dnode, 8);
 342         sa_handle_destroy(sa_hdl);
 343         if (rc == 0)
 344                 rc = osd_get_fid_by_oid(env, osd, dnode, fid);
 345
 346         RETURN(rc);
 347 }
 348
 349 static int osd_find_parent_fid(const struct lu_env *env, struct dt_object *o,
 350                                struct lu_fid *fid)
 351 {
 352         struct link_ea_header  *leh;
 353         struct link_ea_entry   *lee;
 354         struct lu_buf           buf;
 355         int                     rc;
 356         ENTRY;
 357
 358         buf.lb_buf = osd_oti_get(env)->oti_buf;
 359         buf.lb_len = sizeof(osd_oti_get(env)->oti_buf);
 360
 361         rc = osd_xattr_get(env, o, &buf, XATTR_NAME_LINK);
 362         if (rc == -ERANGE) {
 363                 rc = osd_xattr_get(env, o, &LU_BUF_NULL, XATTR_NAME_LINK);
 364                 if (rc < 0)
 365                         RETURN(rc);
 366                 LASSERT(rc > 0);
 367                 OBD_ALLOC(buf.lb_buf, rc);
 368                 if (buf.lb_buf == NULL)
 369                         RETURN(-ENOMEM);
 370                 buf.lb_len = rc;
 371                 rc = osd_xattr_get(env, o, &buf, XATTR_NAME_LINK);
 372         }
 373         if (rc < 0)
 374                 GOTO(out, rc);
 375         if (rc < sizeof(*leh) + sizeof(*lee))
 376                 GOTO(out, rc = -EINVAL);
 377
 378         leh = buf.lb_buf;
 379         if (leh->leh_magic == __swab32(LINK_EA_MAGIC)) {
 380                 leh->leh_magic = LINK_EA_MAGIC;
 381                 leh->leh_reccount = __swab32(leh->leh_reccount);
 382                 leh->leh_len = __swab64(leh->leh_len);
 383         }
 384         if (leh->leh_magic != LINK_EA_MAGIC)
 385                 GOTO(out, rc = -EINVAL);
 386         if (leh->leh_reccount == 0)
 387                 GOTO(out, rc = -ENODATA);
 388
 389         lee = (struct link_ea_entry *)(leh + 1);
 390         fid_be_to_cpu(fid, (const struct lu_fid *)&lee->lee_parent_fid);
 391         rc = 0;
 392
 393 out:
 394         if (buf.lb_buf != osd_oti_get(env)->oti_buf)
 395                 OBD_FREE(buf.lb_buf, buf.lb_len);
 396
 397 #if 0
 398         /* this block can be enabled for additional verification
 399          * it's trying to match FID from LinkEA vs. FID from LMA */
 400         if (rc == 0) {
 401                 struct lu_fid fid2;
 402                 int rc2;
 403                 rc2 = osd_find_parent_by_dnode(env, o, &fid2);
 404                 if (rc2 == 0)
 405                         if (lu_fid_eq(fid, &fid2) == 0)
 406                                 CERROR("wrong parent: "DFID" != "DFID"\n",
 407                                        PFID(fid), PFID(&fid2));
 408         }
 409 #endif
 410
 411         /* no LinkEA is found, let's try to find the fid in parent's LMA */
 412         if (unlikely(rc != 0))
 413                 rc = osd_find_parent_by_dnode(env, o, fid);
 414
 415         RETURN(rc);
 416 }
 417
 418 static int osd_dir_lookup(const struct lu_env *env, struct dt_object *dt,
 419                           struct dt_rec *rec, const struct dt_key *key)
 420 {
 421         struct osd_thread_info *oti = osd_oti_get(env);
 422         struct osd_object  *obj = osd_dt_obj(dt);
 423         struct osd_device  *osd = osd_obj2dev(obj);
 424         char               *name = (char *)key;
 425         int                 rc;
 426         ENTRY;
 427
 428         LASSERT(osd_object_is_zap(obj->oo_db));
 429
 430         if (name[0] == '.') {
 431                 if (name[1] == 0) {
 432                         const struct lu_fid *f = lu_object_fid(&dt->do_lu);
 433                         memcpy(rec, f, sizeof(*f));
 434                         RETURN(1);
 435                 } else if (name[1] == '.' && name[2] == 0) {
 436                         rc = osd_find_parent_fid(env, dt, (struct lu_fid *)rec);
 437                         RETURN(rc == 0 ? 1 : rc);
 438                 }
 439         }
 440
 441         memset(&oti->oti_zde.lzd_fid, 0, sizeof(struct lu_fid));
 442         rc = -zap_lookup(osd->od_os, obj->oo_db->db_object,
 443                          (char *)key, 8, sizeof(oti->oti_zde) / 8,
 444                          (void *)&oti->oti_zde);
 445         if (rc != 0)
 446                 RETURN(rc);
 447
 448         if (likely(fid_is_sane(&oti->oti_zde.lzd_fid))) {
 449                 memcpy(rec, &oti->oti_zde.lzd_fid, sizeof(struct lu_fid));
 450                 RETURN(1);
 451         }
 452
 453         rc = osd_get_fid_by_oid(env, osd, oti->oti_zde.lzd_reg.zde_dnode,
 454                                 (struct lu_fid *)rec);
 455
 456         RETURN(rc == 0 ? 1 : (rc == -ENOENT ? -ENODATA : rc));
 457 }
 458
 459 static int osd_declare_dir_insert(const struct lu_env *env,
 460                                   struct dt_object *dt,
 461                                   const struct dt_rec *rec,
 462                                   const struct dt_key *key,
 463                                   struct thandle *th)
 464 {
 465         struct osd_object  *obj = osd_dt_obj(dt);
 466         struct osd_thandle *oh;
 467         uint64_t object;
 468         ENTRY;
 469
 470         LASSERT(th != NULL);
 471         oh = container_of0(th, struct osd_thandle, ot_super);
 472
 473         /* This is for inserting dot/dotdot for new created dir. */
 474         if (obj->oo_db == NULL)
 475                 object = DMU_NEW_OBJECT;
 476         else
 477                 object = obj->oo_db->db_object;
 478
 479         dmu_tx_hold_bonus(oh->ot_tx, object);
 480         dmu_tx_hold_zap(oh->ot_tx, object, TRUE, (char *)key);
 481
 482         RETURN(0);
 483 }
 484
 485 /**
 486  * Find the osd object for given fid.
 487  *
 488  * \param fid need to find the osd object having this fid
 489  *
 490  * \retval osd_object on success
 491  * \retval        -ve on error
 492  */
 493 struct osd_object *osd_object_find(const struct lu_env *env,
 494                                    struct dt_object *dt,
 495                                    const struct lu_fid *fid)
 496 {
 497         struct lu_device         *ludev = dt->do_lu.lo_dev;
 498         struct osd_object        *child = NULL;
 499         struct lu_object         *luch;
 500         struct lu_object         *lo;
 501
 502         /*
 503          * at this point topdev might not exist yet
 504          * (i.e. MGS is preparing profiles). so we can
 505          * not rely on topdev and instead lookup with
 506          * our device passed as topdev. this can't work
 507          * if the object isn't cached yet (as osd doesn't
 508          * allocate lu_header). IOW, the object must be
 509          * in the cache, otherwise lu_object_alloc() crashes
 510          * -bzzz
 511          */
 512         luch = lu_object_find_at(env, ludev, fid, NULL);
 513         if (IS_ERR(luch))
 514                 return (void *)luch;
 515
 516         if (lu_object_exists(luch)) {
 517                 lo = lu_object_locate(luch->lo_header, ludev->ld_type);
 518                 if (lo != NULL)
 519                         child = osd_obj(lo);
 520                 else
 521                         LU_OBJECT_DEBUG(D_ERROR, env, luch,
 522                                         "%s: object can't be located "DFID,
 523                                         osd_dev(ludev)->od_svname, PFID(fid));
 524
 525                 if (child == NULL) {
 526                         lu_object_put(env, luch);
 527                         CERROR("%s: Unable to get osd_object "DFID"\n",
 528                                osd_dev(ludev)->od_svname, PFID(fid));
 529                         child = ERR_PTR(-ENOENT);
 530                 }
 531         } else {
 532                 LU_OBJECT_DEBUG(D_ERROR, env, luch,
 533                                 "%s: lu_object does not exists "DFID,
 534                                 osd_dev(ludev)->od_svname, PFID(fid));
 535                 lu_object_put(env, luch);
 536                 child = ERR_PTR(-ENOENT);
 537         }
 538
 539         return child;
 540 }
 541
 542 /**
 543  * Put the osd object once done with it.
 544  *
 545  * \param obj osd object that needs to be put
 546  */
 547 static inline void osd_object_put(const struct lu_env *env,
 548                                   struct osd_object *obj)
 549 {
 550         lu_object_put(env, &obj->oo_dt.do_lu);
 551 }
 552
 553 static int osd_seq_exists(const struct lu_env *env, struct osd_device *osd,
 554                           u64 seq)
 555 {
 556         struct lu_seq_range     *range = &osd_oti_get(env)->oti_seq_range;
 557         struct seq_server_site  *ss = osd_seq_site(osd);
 558         int                     rc;
 559         ENTRY;
 560
 561         LASSERT(ss != NULL);
 562         LASSERT(ss->ss_server_fld != NULL);
 563
 564         rc = osd_fld_lookup(env, osd, seq, range);
 565         if (rc != 0) {
 566                 if (rc != -ENOENT)
 567                         CERROR("%s: Can not lookup fld for "LPX64"\n",
 568                                osd_name(osd), seq);
 569                 RETURN(0);
 570         }
 571
 572         RETURN(ss->ss_node_id == range->lsr_index);
 573 }
 574
 575 static int osd_remote_fid(const struct lu_env *env, struct osd_device *osd,
 576                           const struct lu_fid *fid)
 577 {
 578         struct seq_server_site  *ss = osd_seq_site(osd);
 579         ENTRY;
 580
 581         /* FID seqs not in FLDB, must be local seq */
 582         if (unlikely(!fid_seq_in_fldb(fid_seq(fid))))
 583                 RETURN(0);
 584
 585         /* If FLD is not being initialized yet, it only happens during the
 586          * initialization, likely during mgs initialization, and we assume
 587          * this is local FID. */
 588         if (ss == NULL || ss->ss_server_fld == NULL)
 589                 RETURN(0);
 590
 591         /* Only check the local FLDB here */
 592         if (osd_seq_exists(env, osd, fid_seq(fid)))
 593                 RETURN(0);
 594
 595         RETURN(1);
 596 }
 597
 598 /**
 599  *      Inserts (key, value) pair in \a directory object.
 600  *
 601  *      \param  dt      osd index object
 602  *      \param  key     key for index
 603  *      \param  rec     record reference
 604  *      \param  th      transaction handler
 605  *      \param  ignore_quota update should not affect quota
 606  *
 607  *      \retval  0  success
 608  *      \retval -ve failure
 609  */
 610 static int osd_dir_insert(const struct lu_env *env, struct dt_object *dt,
 611                           const struct dt_rec *rec, const struct dt_key *key,
 612                           struct thandle *th, int ignore_quota)
 613 {
 614         struct osd_thread_info *oti = osd_oti_get(env);
 615         struct osd_object   *parent = osd_dt_obj(dt);
 616         struct osd_device   *osd = osd_obj2dev(parent);
 617         struct dt_insert_rec *rec1 = (struct dt_insert_rec *)rec;
 618         const struct lu_fid *fid = rec1->rec_fid;
 619         struct osd_thandle  *oh;
 620         struct osd_object   *child = NULL;
 621         __u32                attr;
 622         char                *name = (char *)key;
 623         int                  rc;
 624         ENTRY;
 625
 626         LASSERT(parent->oo_db);
 627         LASSERT(osd_object_is_zap(parent->oo_db));
 628
 629         LASSERT(dt_object_exists(dt));
 630         LASSERT(osd_invariant(parent));
 631
 632         LASSERT(th != NULL);
 633         oh = container_of0(th, struct osd_thandle, ot_super);
 634
 635         rc = osd_remote_fid(env, osd, fid);
 636         if (rc < 0) {
 637                 CERROR("%s: Can not find object "DFID": rc = %d\n",
 638                        osd->od_svname, PFID(fid), rc);
 639                 RETURN(rc);
 640         }
 641
 642         if (unlikely(rc == 1)) {
 643                 /* Insert remote entry */
 644                 memset(&oti->oti_zde.lzd_reg, 0, sizeof(oti->oti_zde.lzd_reg));
 645                 oti->oti_zde.lzd_reg.zde_type = IFTODT(rec1->rec_type & S_IFMT);
 646         } else {
 647                 /*
 648                  * To simulate old Orion setups with ./..  stored in the
 649                  * directories
 650                  */
 651                 /* Insert local entry */
 652                 child = osd_object_find(env, dt, fid);
 653                 if (IS_ERR(child))
 654                         RETURN(PTR_ERR(child));
 655
 656                 LASSERT(child->oo_db);
 657                 if (name[0] == '.') {
 658                         if (name[1] == 0) {
 659                                 /* do not store ".", instead generate it
 660                                  * during iteration */
 661                                 GOTO(out, rc = 0);
 662                         } else if (name[1] == '.' && name[2] == 0) {
 663                                 if (OBD_FAIL_CHECK(OBD_FAIL_LFSCK_BAD_PARENT)) {
 664                                         struct lu_fid tfid = *fid;
 665
 666                                         osd_object_put(env, child);
 667                                         tfid.f_oid--;
 668                                         child = osd_object_find(env, dt, &tfid);
 669                                         if (IS_ERR(child))
 670                                                 RETURN(PTR_ERR(child));
 671
 672                                         LASSERT(child->oo_db);
 673                                 }
 674
 675                                 /* update parent dnode in the child.
 676                                  * later it will be used to generate ".." */
 677                                 rc = osd_object_sa_update(parent,
 678                                                  SA_ZPL_PARENT(osd),
 679                                                  &child->oo_db->db_object,
 680                                                  8, oh);
 681
 682                                 GOTO(out, rc);
 683                         }
 684                 }
 685                 CLASSERT(sizeof(oti->oti_zde.lzd_reg) == 8);
 686                 CLASSERT(sizeof(oti->oti_zde) % 8 == 0);
 687                 attr = child->oo_dt.do_lu.lo_header ->loh_attr;
 688                 oti->oti_zde.lzd_reg.zde_type = IFTODT(attr & S_IFMT);
 689                 oti->oti_zde.lzd_reg.zde_dnode = child->oo_db->db_object;
 690         }
 691
 692         oti->oti_zde.lzd_fid = *fid;
 693         /* Insert (key,oid) into ZAP */
 694         rc = -zap_add(osd->od_os, parent->oo_db->db_object,
 695                       (char *)key, 8, sizeof(oti->oti_zde) / 8,
 696                       (void *)&oti->oti_zde, oh->ot_tx);
 697         if (unlikely(rc == -EEXIST &&
 698                      name[0] == '.' && name[1] == '.' && name[2] == 0))
 699                 /* Update (key,oid) in ZAP */
 700                 rc = -zap_update(osd->od_os, parent->oo_db->db_object,
 701                                 (char *)key, 8, sizeof(oti->oti_zde) / 8,
 702                                 (void *)&oti->oti_zde, oh->ot_tx);
 703
 704 out:
 705         if (child != NULL)
 706                 osd_object_put(env, child);
 707
 708         RETURN(rc);
 709 }
 710
 711 static int osd_declare_dir_delete(const struct lu_env *env,
 712                                   struct dt_object *dt,
 713                                   const struct dt_key *key,
 714                                   struct thandle *th)
 715 {
 716         struct osd_object  *obj = osd_dt_obj(dt);
 717         struct osd_thandle *oh;
 718         uint64_t            dnode;
 719         ENTRY;
 720
 721         LASSERT(dt_object_exists(dt));
 722         LASSERT(osd_invariant(obj));
 723
 724         LASSERT(th != NULL);
 725         oh = container_of0(th, struct osd_thandle, ot_super);
 726
 727         if (dt_object_exists(dt)) {
 728                 LASSERT(obj->oo_db);
 729                 LASSERT(osd_object_is_zap(obj->oo_db));
 730                 dnode = obj->oo_db->db_object;
 731         } else {
 732                 dnode = DMU_NEW_OBJECT;
 733         }
 734         dmu_tx_hold_zap(oh->ot_tx, dnode, TRUE, (char *)key);
 735
 736         RETURN(0);
 737 }
 738
 739 static int osd_dir_delete(const struct lu_env *env, struct dt_object *dt,
 740                           const struct dt_key *key, struct thandle *th)
 741 {
 742         struct osd_object *obj = osd_dt_obj(dt);
 743         struct osd_device *osd = osd_obj2dev(obj);
 744         struct osd_thandle *oh;
 745         dmu_buf_t *zap_db = obj->oo_db;
 746         char      *name = (char *)key;
 747         int rc;
 748         ENTRY;
 749
 750         LASSERT(zap_db);
 751         LASSERT(osd_object_is_zap(zap_db));
 752
 753         LASSERT(th != NULL);
 754         oh = container_of0(th, struct osd_thandle, ot_super);
 755
 756         /*
 757          * In Orion . and .. were stored in the directory (not generated upon
 758          * request as now). we preserve them for backward compatibility
 759          */
 760         if (name[0] == '.') {
 761                 if (name[1] == 0) {
 762                         RETURN(0);
 763                 } else if (name[1] == '.' && name[2] == 0) {
 764                         RETURN(0);
 765                 }
 766         }
 767
 768         /* Remove key from the ZAP */
 769         rc = -zap_remove(osd->od_os, zap_db->db_object,
 770                          (char *) key, oh->ot_tx);
 771
 772         if (unlikely(rc && rc != -ENOENT))
 773                 CERROR("%s: zap_remove failed: rc = %d\n", osd->od_svname, rc);
 774
 775         RETURN(rc);
 776 }
 777
 778 static struct dt_it *osd_dir_it_init(const struct lu_env *env,
 779                                      struct dt_object *dt,
 780                                      __u32 unused)
 781 {
 782         struct osd_zap_it *it;
 783
 784         it = (struct osd_zap_it *)osd_index_it_init(env, dt, unused);
 785         if (!IS_ERR(it))
 786                 it->ozi_pos = 0;
 787
 788         RETURN((struct dt_it *)it);
 789 }
 790
 791 /**
 792  *  Move Iterator to record specified by \a key
 793  *
 794  *  \param  di      osd iterator
 795  *  \param  key     key for index
 796  *
 797  *  \retval +ve  di points to record with least key not larger than key
 798  *  \retval  0   di points to exact matched key
 799  *  \retval -ve  failure
 800  */
 801 static int osd_dir_it_get(const struct lu_env *env,
 802                           struct dt_it *di, const struct dt_key *key)
 803 {
 804         struct osd_zap_it *it = (struct osd_zap_it *)di;
 805         struct osd_object *obj = it->ozi_obj;
 806         char              *name = (char *)key;
 807         int                rc;
 808         ENTRY;
 809
 810         LASSERT(it);
 811         LASSERT(it->ozi_zc);
 812
 813         /* reset the cursor */
 814         zap_cursor_fini(it->ozi_zc);
 815         osd_obj_cursor_init_serialized(it->ozi_zc, obj, 0);
 816
 817         /* XXX: implementation of the API is broken at the moment */
 818         LASSERT(((const char *)key)[0] == 0);
 819
 820         if (name[0] == 0) {
 821                 it->ozi_pos = 0;
 822                 RETURN(1);
 823         }
 824
 825         if (name[0] == '.') {
 826                 if (name[1] == 0) {
 827                         it->ozi_pos = 1;
 828                         GOTO(out, rc = 1);
 829                 } else if (name[1] == '.' && name[2] == 0) {
 830                         it->ozi_pos = 2;
 831                         GOTO(out, rc = 1);
 832                 }
 833         }
 834
 835         /* neither . nor .. - some real record */
 836         it->ozi_pos = 3;
 837         rc = +1;
 838
 839 out:
 840         RETURN(rc);
 841 }
 842
 843 static void osd_dir_it_put(const struct lu_env *env, struct dt_it *di)
 844 {
 845         /* PBS: do nothing : ref are incremented at retrive and decreamented
 846          *      next/finish. */
 847 }
 848
 849 /*
 850  * in Orion . and .. were stored in the directory, while ZPL
 851  * and current osd-zfs generate them up on request. so, we
 852  * need to ignore previously stored . and ..
 853  */
 854 static int osd_index_retrieve_skip_dots(struct osd_zap_it *it,
 855                                         zap_attribute_t *za)
 856 {
 857         int rc, isdot;
 858
 859         do {
 860                 rc = -zap_cursor_retrieve(it->ozi_zc, za);
 861
 862                 isdot = 0;
 863                 if (unlikely(rc == 0 && za->za_name[0] == '.')) {
 864                         if (za->za_name[1] == 0) {
 865                                 isdot = 1;
 866                         } else if (za->za_name[1] == '.' &&
 867                                    za->za_name[2] == 0) {
 868                                 isdot = 1;
 869                         }
 870                         if (unlikely(isdot))
 871                                 zap_cursor_advance(it->ozi_zc);
 872                 }
 873         } while (unlikely(rc == 0 && isdot));
 874
 875         return rc;
 876 }
 877
 878 /**
 879  * to load a directory entry at a time and stored it in
 880  * iterator's in-memory data structure.
 881  *
 882  * \param di, struct osd_it_ea, iterator's in memory structure
 883  *
 884  * \retval +ve, iterator reached to end
 885  * \retval   0, iterator not reached to end
 886  * \retval -ve, on error
 887  */
 888 static int osd_dir_it_next(const struct lu_env *env, struct dt_it *di)
 889 {
 890         struct osd_zap_it *it = (struct osd_zap_it *)di;
 891         zap_attribute_t   *za = &osd_oti_get(env)->oti_za;
 892         int                rc;
 893
 894         ENTRY;
 895
 896         /* temp. storage should be enough for any key supported by ZFS */
 897         CLASSERT(sizeof(za->za_name) <= sizeof(it->ozi_name));
 898
 899         /*
 900          * the first ->next() moves the cursor to .
 901          * the second ->next() moves the cursor to ..
 902          * then we get to the real records and have to verify any exist
 903          */
 904         if (it->ozi_pos <= 2) {
 905                 it->ozi_pos++;
 906                 if (it->ozi_pos <=2)
 907                         RETURN(0);
 908
 909         } else {
 910                 zap_cursor_advance(it->ozi_zc);
 911         }
 912
 913         /*
 914          * According to current API we need to return error if its last entry.
 915          * zap_cursor_advance() does not return any value. So we need to call
 916          * retrieve to check if there is any record.  We should make
 917          * changes to Iterator API to not return status for this API
 918          */
 919         rc = osd_index_retrieve_skip_dots(it, za);
 920
 921         if (rc == -ENOENT) /* end of dir */
 922                 RETURN(+1);
 923
 924         RETURN(rc);
 925 }
 926
 927 static struct dt_key *osd_dir_it_key(const struct lu_env *env,
 928                                      const struct dt_it *di)
 929 {
 930         struct osd_zap_it *it = (struct osd_zap_it *)di;
 931         zap_attribute_t   *za = &osd_oti_get(env)->oti_za;
 932         int                rc = 0;
 933         ENTRY;
 934
 935         if (it->ozi_pos <= 1) {
 936                 it->ozi_pos = 1;
 937                 RETURN((struct dt_key *)".");
 938         } else if (it->ozi_pos == 2) {
 939                 RETURN((struct dt_key *)"..");
 940         }
 941
 942         if ((rc = -zap_cursor_retrieve(it->ozi_zc, za)))
 943                 RETURN(ERR_PTR(rc));
 944
 945         strcpy(it->ozi_name, za->za_name);
 946
 947         RETURN((struct dt_key *)it->ozi_name);
 948 }
 949
 950 static int osd_dir_it_key_size(const struct lu_env *env, const struct dt_it *di)
 951 {
 952         struct osd_zap_it *it = (struct osd_zap_it *)di;
 953         zap_attribute_t   *za = &osd_oti_get(env)->oti_za;
 954         int                rc;
 955         ENTRY;
 956
 957         if (it->ozi_pos <= 1) {
 958                 it->ozi_pos = 1;
 959                 RETURN(2);
 960         } else if (it->ozi_pos == 2) {
 961                 RETURN(3);
 962         }
 963
 964         if ((rc = -zap_cursor_retrieve(it->ozi_zc, za)) == 0)
 965                 rc = strlen(za->za_name);
 966
 967         RETURN(rc);
 968 }
 969
 970 static int osd_dir_it_rec(const struct lu_env *env, const struct dt_it *di,
 971                           struct dt_rec *dtrec, __u32 attr)
 972 {
 973         struct osd_zap_it   *it = (struct osd_zap_it *)di;
 974         struct lu_dirent    *lde = (struct lu_dirent *)dtrec;
 975         struct luz_direntry *zde = &osd_oti_get(env)->oti_zde;
 976         zap_attribute_t     *za = &osd_oti_get(env)->oti_za;
 977         int                  rc, namelen;
 978         ENTRY;
 979
 980         if (it->ozi_pos <= 1) {
 981                 lde->lde_hash = cpu_to_le64(1);
 982                 strcpy(lde->lde_name, ".");
 983                 lde->lde_namelen = cpu_to_le16(1);
 984                 lde->lde_fid = *lu_object_fid(&it->ozi_obj->oo_dt.do_lu);
 985                 lde->lde_attrs = LUDA_FID;
 986                 /* append lustre attributes */
 987                 osd_it_append_attrs(lde, attr, 1, IFTODT(S_IFDIR));
 988                 lde->lde_reclen = cpu_to_le16(lu_dirent_calc_size(1, attr));
 989                 it->ozi_pos = 1;
 990                 GOTO(out, rc = 0);
 991
 992         } else if (it->ozi_pos == 2) {
 993                 lde->lde_hash = cpu_to_le64(2);
 994                 strcpy(lde->lde_name, "..");
 995                 lde->lde_namelen = cpu_to_le16(2);
 996                 lde->lde_attrs = LUDA_FID;
 997                 /* append lustre attributes */
 998                 osd_it_append_attrs(lde, attr, 2, IFTODT(S_IFDIR));
 999                 lde->lde_reclen = cpu_to_le16(lu_dirent_calc_size(2, attr));
1000                 rc = osd_find_parent_fid(env, &it->ozi_obj->oo_dt, &lde->lde_fid);
1001
1002                 /* ENOENT happens at the root of filesystem so ignore it */
1003                 if (rc == -ENOENT)
1004                         rc = 0;
1005                 GOTO(out, rc);
1006         }
1007
1008         LASSERT(lde);
1009
1010         rc = -zap_cursor_retrieve(it->ozi_zc, za);
1011         if (unlikely(rc != 0))
1012                 GOTO(out, rc);
1013
1014         lde->lde_hash = cpu_to_le64(osd_zap_cursor_serialize(it->ozi_zc));
1015         namelen = strlen(za->za_name);
1016         if (namelen > NAME_MAX)
1017                 GOTO(out, rc = -EOVERFLOW);
1018         strcpy(lde->lde_name, za->za_name);
1019         lde->lde_namelen = cpu_to_le16(namelen);
1020
1021         if (za->za_integer_length != 8 || za->za_num_integers < 3) {
1022                 CERROR("%s: unsupported direntry format: %d %d\n",
1023                        osd_obj2dev(it->ozi_obj)->od_svname,
1024                        za->za_integer_length, (int)za->za_num_integers);
1025
1026                 GOTO(out, rc = -EIO);
1027         }
1028
1029         rc = -zap_lookup(it->ozi_zc->zc_objset, it->ozi_zc->zc_zapobj,
1030                          za->za_name, za->za_integer_length, 3, zde);
1031         if (rc)
1032                 GOTO(out, rc);
1033
1034         lde->lde_fid = zde->lzd_fid;
1035         lde->lde_attrs = LUDA_FID;
1036
1037         /* append lustre attributes */
1038         osd_it_append_attrs(lde, attr, namelen, zde->lzd_reg.zde_type);
1039
1040         lde->lde_reclen = cpu_to_le16(lu_dirent_calc_size(namelen, attr));
1041
1042 out:
1043         RETURN(rc);
1044 }
1045
1046 static int osd_dir_it_rec_size(const struct lu_env *env, const struct dt_it *di,
1047                                __u32 attr)
1048 {
1049         struct osd_zap_it   *it = (struct osd_zap_it *)di;
1050         zap_attribute_t     *za = &osd_oti_get(env)->oti_za;
1051         size_t               namelen = 0;
1052         int                  rc;
1053         ENTRY;
1054
1055         if (it->ozi_pos <= 1)
1056                 namelen = 1;
1057         else if (it->ozi_pos == 2)
1058                 namelen = 2;
1059
1060         if (namelen > 0) {
1061                 rc = lu_dirent_calc_size(namelen, attr);
1062                 RETURN(rc);
1063         }
1064
1065         rc = -zap_cursor_retrieve(it->ozi_zc, za);
1066         if (unlikely(rc != 0))
1067                 RETURN(rc);
1068
1069         if (za->za_integer_length != 8 || za->za_num_integers < 3) {
1070                 CERROR("%s: unsupported direntry format: %d %d\n",
1071                        osd_obj2dev(it->ozi_obj)->od_svname,
1072                        za->za_integer_length, (int)za->za_num_integers);
1073                 RETURN(-EIO);
1074         }
1075
1076         namelen = strlen(za->za_name);
1077         if (namelen > NAME_MAX)
1078                 RETURN(-EOVERFLOW);
1079
1080         rc = lu_dirent_calc_size(namelen, attr);
1081
1082         RETURN(rc);
1083 }
1084
1085 static __u64 osd_dir_it_store(const struct lu_env *env, const struct dt_it *di)
1086 {
1087         struct osd_zap_it *it = (struct osd_zap_it *)di;
1088         __u64              pos;
1089         ENTRY;
1090
1091         if (it->ozi_pos <= 2)
1092                 pos = it->ozi_pos;
1093         else
1094                 pos = osd_zap_cursor_serialize(it->ozi_zc);
1095
1096         RETURN(pos);
1097 }
1098
1099 /*
1100  * return status :
1101  *  rc == 0 -> end of directory.
1102  *  rc >  0 -> ok, proceed.
1103  *  rc <  0 -> error.  ( EOVERFLOW  can be masked.)
1104  */
1105 static int osd_dir_it_load(const struct lu_env *env,
1106                         const struct dt_it *di, __u64 hash)
1107 {
1108         struct osd_zap_it *it = (struct osd_zap_it *)di;
1109         struct osd_object *obj = it->ozi_obj;
1110         zap_attribute_t   *za = &osd_oti_get(env)->oti_za;
1111         int                rc;
1112         ENTRY;
1113
1114         /* reset the cursor */
1115         zap_cursor_fini(it->ozi_zc);
1116         osd_obj_cursor_init_serialized(it->ozi_zc, obj, hash);
1117
1118         if (hash <= 2) {
1119                 it->ozi_pos = hash;
1120                 rc = +1;
1121         } else {
1122                 it->ozi_pos = 3;
1123                 /* to return whether the end has been reached */
1124                 rc = osd_index_retrieve_skip_dots(it, za);
1125                 if (rc == 0)
1126                         rc = +1;
1127                 else if (rc == -ENOENT)
1128                         rc = 0;
1129         }
1130
1131         RETURN(rc);
1132 }
1133
1134 struct dt_index_operations osd_dir_ops = {
1135         .dio_lookup         = osd_dir_lookup,
1136         .dio_declare_insert = osd_declare_dir_insert,
1137         .dio_insert         = osd_dir_insert,
1138         .dio_declare_delete = osd_declare_dir_delete,
1139         .dio_delete         = osd_dir_delete,
1140         .dio_it     = {
1141                 .init     = osd_dir_it_init,
1142                 .fini     = osd_index_it_fini,
1143                 .get      = osd_dir_it_get,
1144                 .put      = osd_dir_it_put,
1145                 .next     = osd_dir_it_next,
1146                 .key      = osd_dir_it_key,
1147                 .key_size = osd_dir_it_key_size,
1148                 .rec      = osd_dir_it_rec,
1149                 .rec_size = osd_dir_it_rec_size,
1150                 .store    = osd_dir_it_store,
1151                 .load     = osd_dir_it_load
1152         }
1153 };
1154
1155 /*
1156  * Primitives for index files using binary keys.
1157  */
1158
1159 /* key integer_size is 8 */
1160 static int osd_prepare_key_uint64(struct osd_object *o, __u64 *dst,
1161                                   const struct dt_key *src)
1162 {
1163         int size;
1164
1165         LASSERT(dst);
1166         LASSERT(src);
1167
1168         /* align keysize to 64bit */
1169         size = (o->oo_keysize + sizeof(__u64) - 1) / sizeof(__u64);
1170         size *= sizeof(__u64);
1171
1172         LASSERT(size <= MAXNAMELEN);
1173
1174         if (unlikely(size > o->oo_keysize))
1175                 memset(dst + o->oo_keysize, 0, size - o->oo_keysize);
1176         memcpy(dst, (const char *)src, o->oo_keysize);
1177
1178         return (size/sizeof(__u64));
1179 }
1180
1181 static int osd_index_lookup(const struct lu_env *env, struct dt_object *dt,
1182                         struct dt_rec *rec, const struct dt_key *key)
1183 {
1184         struct osd_object *obj = osd_dt_obj(dt);
1185         struct osd_device *osd = osd_obj2dev(obj);
1186         __u64             *k = osd_oti_get(env)->oti_key64;
1187         int                rc;
1188         ENTRY;
1189
1190         rc = osd_prepare_key_uint64(obj, k, key);
1191
1192         rc = -zap_lookup_uint64(osd->od_os, obj->oo_db->db_object,
1193                                 k, rc, obj->oo_recusize, obj->oo_recsize,
1194                                 (void *)rec);
1195         RETURN(rc == 0 ? 1 : rc);
1196 }
1197
1198 static int osd_declare_index_insert(const struct lu_env *env,
1199                                     struct dt_object *dt,
1200                                     const struct dt_rec *rec,
1201                                     const struct dt_key *key,
1202                                     struct thandle *th)
1203 {
1204         struct osd_object  *obj = osd_dt_obj(dt);
1205         struct osd_thandle *oh;
1206         ENTRY;
1207
1208         LASSERT(th != NULL);
1209         oh = container_of0(th, struct osd_thandle, ot_super);
1210
1211         LASSERT(obj->oo_db);
1212
1213         dmu_tx_hold_bonus(oh->ot_tx, obj->oo_db->db_object);
1214
1215         /* It is not clear what API should be used for binary keys, so we pass
1216          * a null name which has the side effect of over-reserving space,
1217          * accounting for the worst case. See zap_count_write() */
1218         dmu_tx_hold_zap(oh->ot_tx, obj->oo_db->db_object, TRUE, NULL);
1219
1220         RETURN(0);
1221 }
1222
1223 static int osd_index_insert(const struct lu_env *env, struct dt_object *dt,
1224                             const struct dt_rec *rec, const struct dt_key *key,
1225                             struct thandle *th, int ignore_quota)
1226 {
1227         struct osd_object  *obj = osd_dt_obj(dt);
1228         struct osd_device  *osd = osd_obj2dev(obj);
1229         struct osd_thandle *oh;
1230         __u64              *k = osd_oti_get(env)->oti_key64;
1231         int                 rc;
1232         ENTRY;
1233
1234         LASSERT(obj->oo_db);
1235         LASSERT(dt_object_exists(dt));
1236         LASSERT(osd_invariant(obj));
1237         LASSERT(th != NULL);
1238
1239         oh = container_of0(th, struct osd_thandle, ot_super);
1240
1241         rc = osd_prepare_key_uint64(obj, k, key);
1242
1243         /* Insert (key,oid) into ZAP */
1244         rc = -zap_add_uint64(osd->od_os, obj->oo_db->db_object,
1245                              k, rc, obj->oo_recusize, obj->oo_recsize,
1246                              (void *)rec, oh->ot_tx);
1247         RETURN(rc);
1248 }
1249
1250 static int osd_declare_index_delete(const struct lu_env *env,
1251                                     struct dt_object *dt,
1252                                     const struct dt_key *key,
1253                                     struct thandle *th)
1254 {
1255         struct osd_object  *obj = osd_dt_obj(dt);
1256         struct osd_thandle *oh;
1257         ENTRY;
1258
1259         LASSERT(dt_object_exists(dt));
1260         LASSERT(osd_invariant(obj));
1261         LASSERT(th != NULL);
1262         LASSERT(obj->oo_db);
1263
1264         oh = container_of0(th, struct osd_thandle, ot_super);
1265         dmu_tx_hold_zap(oh->ot_tx, obj->oo_db->db_object, TRUE, NULL);
1266
1267         RETURN(0);
1268 }
1269
1270 static int osd_index_delete(const struct lu_env *env, struct dt_object *dt,
1271                             const struct dt_key *key, struct thandle *th)
1272 {
1273         struct osd_object  *obj = osd_dt_obj(dt);
1274         struct osd_device  *osd = osd_obj2dev(obj);
1275         struct osd_thandle *oh;
1276         __u64              *k = osd_oti_get(env)->oti_key64;
1277         int                 rc;
1278         ENTRY;
1279
1280         LASSERT(obj->oo_db);
1281         LASSERT(th != NULL);
1282         oh = container_of0(th, struct osd_thandle, ot_super);
1283
1284         rc = osd_prepare_key_uint64(obj, k, key);
1285
1286         /* Remove binary key from the ZAP */
1287         rc = -zap_remove_uint64(osd->od_os, obj->oo_db->db_object,
1288                                 k, rc, oh->ot_tx);
1289         RETURN(rc);
1290 }
1291
1292 static int osd_index_it_get(const struct lu_env *env, struct dt_it *di,
1293                             const struct dt_key *key)
1294 {
1295         struct osd_zap_it *it = (struct osd_zap_it *)di;
1296         struct osd_object *obj = it->ozi_obj;
1297         struct osd_device *osd = osd_obj2dev(obj);
1298         ENTRY;
1299
1300         LASSERT(it);
1301         LASSERT(it->ozi_zc);
1302
1303         /*
1304          * XXX: we need a binary version of zap_cursor_move_to_key()
1305          *      to implement this API */
1306         if (*((const __u64 *)key) != 0)
1307                 CERROR("NOT IMPLEMETED YET (move to "LPX64")\n",
1308                        *((__u64 *)key));
1309
1310         zap_cursor_fini(it->ozi_zc);
1311         zap_cursor_init(it->ozi_zc, osd->od_os, obj->oo_db->db_object);
1312         it->ozi_reset = 1;
1313
1314         RETURN(+1);
1315 }
1316
1317 static int osd_index_it_next(const struct lu_env *env, struct dt_it *di)
1318 {
1319         struct osd_zap_it *it = (struct osd_zap_it *)di;
1320         zap_attribute_t   *za = &osd_oti_get(env)->oti_za;
1321         int                rc;
1322         ENTRY;
1323
1324         if (it->ozi_reset == 0)
1325                 zap_cursor_advance(it->ozi_zc);
1326         it->ozi_reset = 0;
1327
1328         /*
1329          * According to current API we need to return error if it's last entry.
1330          * zap_cursor_advance() does not return any value. So we need to call
1331          * retrieve to check if there is any record.  We should make
1332          * changes to Iterator API to not return status for this API
1333          */
1334         rc = -zap_cursor_retrieve(it->ozi_zc, za);
1335         if (rc == -ENOENT)
1336                 RETURN(+1);
1337
1338         RETURN((rc));
1339 }
1340
1341 static struct dt_key *osd_index_it_key(const struct lu_env *env,
1342                                        const struct dt_it *di)
1343 {
1344         struct osd_zap_it *it = (struct osd_zap_it *)di;
1345         struct osd_object *obj = it->ozi_obj;
1346         zap_attribute_t   *za = &osd_oti_get(env)->oti_za;
1347         int                rc = 0;
1348         ENTRY;
1349
1350         it->ozi_reset = 0;
1351         rc = -zap_cursor_retrieve(it->ozi_zc, za);
1352         if (rc)
1353                 RETURN(ERR_PTR(rc));
1354
1355         /* the binary key is stored in the name */
1356         memcpy(&it->ozi_key, za->za_name, obj->oo_keysize);
1357
1358         RETURN((struct dt_key *)&it->ozi_key);
1359 }
1360
1361 static int osd_index_it_key_size(const struct lu_env *env,
1362                                 const struct dt_it *di)
1363 {
1364         struct osd_zap_it *it = (struct osd_zap_it *)di;
1365         struct osd_object *obj = it->ozi_obj;
1366         RETURN(obj->oo_keysize);
1367 }
1368
1369 static int osd_index_it_rec(const struct lu_env *env, const struct dt_it *di,
1370                             struct dt_rec *rec, __u32 attr)
1371 {
1372         zap_attribute_t   *za = &osd_oti_get(env)->oti_za;
1373         struct osd_zap_it *it = (struct osd_zap_it *)di;
1374         struct osd_object *obj = it->ozi_obj;
1375         struct osd_device *osd = osd_obj2dev(obj);
1376         __u64             *k = osd_oti_get(env)->oti_key64;
1377         int                rc;
1378         ENTRY;
1379
1380         it->ozi_reset = 0;
1381         rc = -zap_cursor_retrieve(it->ozi_zc, za);
1382         if (rc)
1383                 RETURN(rc);
1384
1385         rc = osd_prepare_key_uint64(obj, k, (const struct dt_key *)za->za_name);
1386
1387         rc = -zap_lookup_uint64(osd->od_os, obj->oo_db->db_object,
1388                                 k, rc, obj->oo_recusize, obj->oo_recsize,
1389                                 (void *)rec);
1390         RETURN(rc);
1391 }
1392
1393 static __u64 osd_index_it_store(const struct lu_env *env,
1394                                 const struct dt_it *di)
1395 {
1396         struct osd_zap_it *it = (struct osd_zap_it *)di;
1397
1398         it->ozi_reset = 0;
1399         RETURN((__u64)zap_cursor_serialize(it->ozi_zc));
1400 }
1401
1402 static int osd_index_it_load(const struct lu_env *env, const struct dt_it *di,
1403                              __u64 hash)
1404 {
1405         struct osd_zap_it *it = (struct osd_zap_it *)di;
1406         struct osd_object *obj = it->ozi_obj;
1407         struct osd_device *osd = osd_obj2dev(obj);
1408         zap_attribute_t   *za = &osd_oti_get(env)->oti_za;
1409         int                rc;
1410         ENTRY;
1411
1412         /* reset the cursor */
1413         zap_cursor_fini(it->ozi_zc);
1414         zap_cursor_init_serialized(it->ozi_zc, osd->od_os,
1415                                    obj->oo_db->db_object, hash);
1416         it->ozi_reset = 0;
1417
1418         rc = -zap_cursor_retrieve(it->ozi_zc, za);
1419         if (rc == 0)
1420                 RETURN(+1);
1421         else if (rc == -ENOENT)
1422                 RETURN(0);
1423
1424         RETURN(rc);
1425 }
1426
1427 static struct dt_index_operations osd_index_ops = {
1428         .dio_lookup             = osd_index_lookup,
1429         .dio_declare_insert     = osd_declare_index_insert,
1430         .dio_insert             = osd_index_insert,
1431         .dio_declare_delete     = osd_declare_index_delete,
1432         .dio_delete             = osd_index_delete,
1433         .dio_it = {
1434                 .init           = osd_index_it_init,
1435                 .fini           = osd_index_it_fini,
1436                 .get            = osd_index_it_get,
1437                 .put            = osd_index_it_put,
1438                 .next           = osd_index_it_next,
1439                 .key            = osd_index_it_key,
1440                 .key_size       = osd_index_it_key_size,
1441                 .rec            = osd_index_it_rec,
1442                 .store          = osd_index_it_store,
1443                 .load           = osd_index_it_load
1444         }
1445 };
1446
1447 struct osd_metadnode_it {
1448         struct osd_device       *mit_dev;
1449         __u64                    mit_pos;
1450         struct lu_fid            mit_fid;
1451         int                      mit_prefetched;
1452         __u64                    mit_prefetched_dnode;
1453 };
1454
1455 static struct dt_it *osd_zfs_otable_it_init(const struct lu_env *env,
1456                                             struct dt_object *dt, __u32 attr)
1457 {
1458         struct osd_device       *dev   = osd_dev(dt->do_lu.lo_dev);
1459         struct osd_metadnode_it *it;
1460         ENTRY;
1461
1462         OBD_ALLOC_PTR(it);
1463         if (unlikely(it == NULL))
1464                 RETURN(ERR_PTR(-ENOMEM));
1465
1466         it->mit_dev = dev;
1467
1468         /* XXX: dmu_object_next() does NOT find dnodes allocated
1469          *      in the current non-committed txg, so we force txg
1470          *      commit to find all existing dnodes ... */
1471         txg_wait_synced(dmu_objset_pool(dev->od_os), 0ULL);
1472
1473         RETURN((struct dt_it *)it);
1474 }
1475
1476 static void osd_zfs_otable_it_fini(const struct lu_env *env, struct dt_it *di)
1477 {
1478         struct osd_metadnode_it *it  = (struct osd_metadnode_it *)di;
1479
1480         OBD_FREE_PTR(it);
1481 }
1482
1483 static int osd_zfs_otable_it_get(const struct lu_env *env,
1484                                  struct dt_it *di, const struct dt_key *key)
1485 {
1486         return 0;
1487 }
1488
1489 static void osd_zfs_otable_it_put(const struct lu_env *env, struct dt_it *di)
1490 {
1491 }
1492
1493 #define OTABLE_PREFETCH         256
1494
1495 static void osd_zfs_otable_prefetch(const struct lu_env *env,
1496                                     struct osd_metadnode_it *it)
1497 {
1498         struct osd_device       *dev = it->mit_dev;
1499         int                      rc;
1500
1501         /* can go negative on the very first access to the iterator
1502          * or if some non-Lustre objects were found */
1503         if (unlikely(it->mit_prefetched < 0))
1504                 it->mit_prefetched = 0;
1505
1506         if (it->mit_prefetched >= (OTABLE_PREFETCH >> 1))
1507                 return;
1508
1509         if (it->mit_prefetched_dnode == 0)
1510                 it->mit_prefetched_dnode = it->mit_pos;
1511
1512         while (it->mit_prefetched < OTABLE_PREFETCH) {
1513                 rc = -dmu_object_next(dev->od_os, &it->mit_prefetched_dnode,
1514                                       B_FALSE, 0);
1515                 if (unlikely(rc != 0))
1516                         break;
1517
1518                 osd_dmu_prefetch(dev->od_os, it->mit_prefetched_dnode,
1519                                  0, 0, 0, ZIO_PRIORITY_ASYNC_READ);
1520
1521                 it->mit_prefetched++;
1522         }
1523 }
1524
1525 static int osd_zfs_otable_it_next(const struct lu_env *env, struct dt_it *di)
1526 {
1527         struct osd_metadnode_it *it  = (struct osd_metadnode_it *)di;
1528         struct lustre_mdt_attrs *lma;
1529         struct osd_device       *dev = it->mit_dev;
1530         nvlist_t                *nvbuf = NULL;
1531         uchar_t                 *v;
1532         __u64                    dnode;
1533         int                      rc, s;
1534
1535         memset(&it->mit_fid, 0, sizeof(it->mit_fid));
1536
1537         dnode = it->mit_pos;
1538         do {
1539                 rc = -dmu_object_next(dev->od_os, &it->mit_pos, B_FALSE, 0);
1540                 if (unlikely(rc != 0))
1541                         GOTO(out, rc = 1);
1542                 it->mit_prefetched--;
1543
1544                 /* LMA is required for this to be a Lustre object.
1545                  * If there is no xattr skip it. */
1546                 rc = __osd_xattr_load(dev, it->mit_pos, &nvbuf);
1547                 if (unlikely(rc != 0))
1548                         continue;
1549
1550                 LASSERT(nvbuf != NULL);
1551                 rc = -nvlist_lookup_byte_array(nvbuf, XATTR_NAME_LMA, &v, &s);
1552                 if (likely(rc == 0)) {
1553                         /* Lustre object */
1554                         lma = (struct lustre_mdt_attrs *)v;
1555                         lustre_lma_swab(lma);
1556                         it->mit_fid = lma->lma_self_fid;
1557                         nvlist_free(nvbuf);
1558                         break;
1559                 } else {
1560                         /* not a Lustre object, try next one */
1561                         nvlist_free(nvbuf);
1562                 }
1563
1564         } while (1);
1565
1566
1567         /* we aren't prefetching in the above loop because the number of
1568          * non-Lustre objects is very small and we will be repeating very
1569          * rare. in case we want to use this to iterate over non-Lustre
1570          * objects (i.e. when we convert regular ZFS in Lustre) it makes
1571          * sense to initiate prefetching in the loop */
1572
1573         /* 0 - there are more items, +1 - the end */
1574         if (likely(rc == 0))
1575                 osd_zfs_otable_prefetch(env, it);
1576
1577         CDEBUG(D_OTHER, "advance: %llu -> %llu "DFID": %d\n", dnode,
1578                it->mit_pos, PFID(&it->mit_fid), rc);
1579
1580 out:
1581         return rc;
1582 }
1583
1584 static struct dt_key *osd_zfs_otable_it_key(const struct lu_env *env,
1585                                             const struct dt_it *di)
1586 {
1587         return NULL;
1588 }
1589
1590 static int osd_zfs_otable_it_key_size(const struct lu_env *env,
1591                                       const struct dt_it *di)
1592 {
1593         return sizeof(__u64);
1594 }
1595
1596 static int osd_zfs_otable_it_rec(const struct lu_env *env,
1597                                  const struct dt_it *di,
1598                                  struct dt_rec *rec, __u32 attr)
1599 {
1600         struct osd_metadnode_it *it  = (struct osd_metadnode_it *)di;
1601         struct lu_fid *fid = (struct lu_fid *)rec;
1602         ENTRY;
1603
1604         *fid = it->mit_fid;
1605
1606         RETURN(0);
1607 }
1608
1609
1610 static __u64 osd_zfs_otable_it_store(const struct lu_env *env,
1611                                      const struct dt_it *di)
1612 {
1613         struct osd_metadnode_it *it  = (struct osd_metadnode_it *)di;
1614
1615         return it->mit_pos;
1616 }
1617
1618 static int osd_zfs_otable_it_load(const struct lu_env *env,
1619                                   const struct dt_it *di, __u64 hash)
1620 {
1621         struct osd_metadnode_it *it  = (struct osd_metadnode_it *)di;
1622
1623         it->mit_pos = hash;
1624         it->mit_prefetched = 0;
1625         it->mit_prefetched_dnode = 0;
1626
1627         return osd_zfs_otable_it_next(env, (struct dt_it *)di);
1628 }
1629
1630 static int osd_zfs_otable_it_key_rec(const struct lu_env *env,
1631                                      const struct dt_it *di, void *key_rec)
1632 {
1633         return 0;
1634 }
1635
1636 const struct dt_index_operations osd_zfs_otable_ops = {
1637         .dio_it = {
1638                 .init     = osd_zfs_otable_it_init,
1639                 .fini     = osd_zfs_otable_it_fini,
1640                 .get      = osd_zfs_otable_it_get,
1641                 .put      = osd_zfs_otable_it_put,
1642                 .next     = osd_zfs_otable_it_next,
1643                 .key      = osd_zfs_otable_it_key,
1644                 .key_size = osd_zfs_otable_it_key_size,
1645                 .rec      = osd_zfs_otable_it_rec,
1646                 .store    = osd_zfs_otable_it_store,
1647                 .load     = osd_zfs_otable_it_load,
1648                 .key_rec  = osd_zfs_otable_it_key_rec,
1649         }
1650 };
1651
1652 int osd_index_try(const struct lu_env *env, struct dt_object *dt,
1653                 const struct dt_index_features *feat)
1654 {
1655         struct osd_object *obj = osd_dt_obj(dt);
1656         int rc = 0;
1657         ENTRY;
1658
1659         down_read(&obj->oo_guard);
1660
1661         /*
1662          * XXX: implement support for fixed-size keys sorted with natural
1663          *      numerical way (not using internal hash value)
1664          */
1665         if (feat->dif_flags & DT_IND_RANGE)
1666                 GOTO(out, rc = -ERANGE);
1667
1668         if (unlikely(feat == &dt_otable_features)) {
1669                 dt->do_index_ops = &osd_zfs_otable_ops;
1670                 GOTO(out, rc = 0);
1671         }
1672
1673         LASSERT(!dt_object_exists(dt) || obj->oo_db != NULL);
1674         if (likely(feat == &dt_directory_features)) {
1675                 if (!dt_object_exists(dt) || osd_object_is_zap(obj->oo_db))
1676                         dt->do_index_ops = &osd_dir_ops;
1677                 else
1678                         GOTO(out, rc = -ENOTDIR);
1679         } else if (unlikely(feat == &dt_acct_features)) {
1680                 LASSERT(fid_is_acct(lu_object_fid(&dt->do_lu)));
1681                 dt->do_index_ops = &osd_acct_index_ops;
1682         } else if (dt->do_index_ops == NULL) {
1683                 /* For index file, we don't support variable key & record sizes
1684                  * and the key has to be unique */
1685                 if ((feat->dif_flags & ~DT_IND_UPDATE) != 0)
1686                         GOTO(out, rc = -EINVAL);
1687
1688                 if (feat->dif_keysize_max > ZAP_MAXNAMELEN)
1689                         GOTO(out, rc = -E2BIG);
1690                 if (feat->dif_keysize_max != feat->dif_keysize_min)
1691                         GOTO(out, rc = -EINVAL);
1692
1693                 /* As for the record size, it should be a multiple of 8 bytes
1694                  * and smaller than the maximum value length supported by ZAP.
1695                  */
1696                 if (feat->dif_recsize_max > ZAP_MAXVALUELEN)
1697                         GOTO(out, rc = -E2BIG);
1698                 if (feat->dif_recsize_max != feat->dif_recsize_min)
1699                         GOTO(out, rc = -EINVAL);
1700
1701                 obj->oo_keysize = feat->dif_keysize_max;
1702                 obj->oo_recsize = feat->dif_recsize_max;
1703                 obj->oo_recusize = 1;
1704
1705                 /* ZFS prefers to work with array of 64bits */
1706                 if ((obj->oo_recsize & 7) == 0) {
1707                         obj->oo_recsize >>= 3;
1708                         obj->oo_recusize = 8;
1709                 }
1710                 dt->do_index_ops = &osd_index_ops;
1711         }
1712
1713 out:
1714         up_read(&obj->oo_guard);
1715
1716         RETURN(rc);
1717 }