lustre/osd-zfs/osd_handler.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 /*
  31  * Copyright (c) 2011, 2012 Whamcloud, Inc.
  32  * Use is subject to license terms.
  33  *
  34  */
  35 /*
  36  * This file is part of Lustre, http://www.lustre.org/
  37  * Lustre is a trademark of Sun Microsystems, Inc.
  38  *
  39  * lustre/osd-zfs/osd_handler.c
  40  * Top-level entry points into osd module
  41  *
  42  * Author: Alex Zhuravlev <bzzz@whamcloud.com>
  43  * Author: Mike Pershin <tappro@whamcloud.com>
  44  * Author: Johann Lombardi <johann@whamcloud.com>
  45  */
  46
  47 #ifndef EXPORT_SYMTAB
  48 # define EXPORT_SYMTAB
  49 #endif
  50 #define DEBUG_SUBSYSTEM S_OSD
  51
  52 #include <lustre_ver.h>
  53 #include <libcfs/libcfs.h>
  54 #include <obd_support.h>
  55 #include <lustre_net.h>
  56 #include <obd.h>
  57 #include <obd_class.h>
  58 #include <lustre_disk.h>
  59 #include <lustre_fid.h>
  60
  61 #include "osd_internal.h"
  62
  63 #include <sys/dnode.h>
  64 #include <sys/dbuf.h>
  65 #include <sys/spa.h>
  66 #include <sys/stat.h>
  67 #include <sys/zap.h>
  68 #include <sys/spa_impl.h>
  69 #include <sys/zfs_znode.h>
  70 #include <sys/dmu_tx.h>
  71 #include <sys/dmu_objset.h>
  72 #include <sys/dsl_prop.h>
  73 #include <sys/sa_impl.h>
  74 #include <sys/txg.h>
  75
  76 struct lu_context_key   osd_key;
  77
  78 static char *root_tag = "osd_mount, rootdb";
  79
  80 /* Slab for OSD object allocation */
  81 cfs_mem_cache_t *osd_object_kmem;
  82
  83 static struct lu_kmem_descr osd_caches[] = {
  84         {
  85                 .ckd_cache = &osd_object_kmem,
  86                 .ckd_name  = "zfs_osd_obj",
  87                 .ckd_size  = sizeof(struct osd_object)
  88         },
  89         {
  90                 .ckd_cache = NULL
  91         }
  92 };
  93
  94 static void arc_prune_func(int64_t bytes, void *private)
  95 {
  96         struct osd_device *od = private;
  97         struct lu_site    *site = &od->od_site;
  98         struct lu_env      env;
  99         int rc;
 100
 101         rc = lu_env_init(&env, LCT_SHRINKER);
 102         if (rc) {
 103                 CERROR("%s: can't initialize shrinker env: rc = %d\n",
 104                        od->od_svname, rc);
 105                 return;
 106         }
 107
 108         lu_site_purge(&env, site, (bytes >> 10));
 109
 110         lu_env_fini(&env);
 111 }
 112
 113 /*
 114  * Concurrency: doesn't access mutable data
 115  */
 116 static int osd_root_get(const struct lu_env *env,
 117                         struct dt_device *dev, struct lu_fid *f)
 118 {
 119         lu_local_obj_fid(f, OSD_FS_ROOT_OID);
 120         return 0;
 121 }
 122
 123 /*
 124  * OSD object methods.
 125  */
 126
 127 /*
 128  * Concurrency: shouldn't matter.
 129  */
 130 static void osd_trans_commit_cb(void *cb_data, int error)
 131 {
 132         struct osd_thandle      *oh = cb_data;
 133         struct thandle          *th = &oh->ot_super;
 134         struct osd_device       *osd = osd_dt_dev(th->th_dev);
 135         struct lu_device        *lud = &th->th_dev->dd_lu_dev;
 136         struct dt_txn_commit_cb *dcb, *tmp;
 137
 138         ENTRY;
 139
 140         if (error) {
 141                 if (error == ECANCELED)
 142                         CWARN("%s: transaction @0x%p was aborted\n",
 143                               osd_dt_dev(th->th_dev)->od_svname, th);
 144                 else
 145                         CERROR("%s: transaction @0x%p commit error: rc = %d\n",
 146                                 osd_dt_dev(th->th_dev)->od_svname, th, error);
 147         }
 148
 149         dt_txn_hook_commit(th);
 150
 151         /* call per-transaction callbacks if any */
 152         cfs_list_for_each_entry_safe(dcb, tmp, &oh->ot_dcb_list, dcb_linkage)
 153                 dcb->dcb_func(NULL, th, dcb, error);
 154
 155         /* Unlike ldiskfs, zfs updates space accounting at commit time.
 156          * As a consequence, op_end is called only now to inform the quota slave
 157          * component that reserved quota space is now accounted in usage and
 158          * should be released. Quota space won't be adjusted at this point since
 159          * we can't provide a suitable environment. It will be performed
 160          * asynchronously by a lquota thread. */
 161         qsd_op_end(NULL, osd->od_quota_slave, &oh->ot_quota_trans);
 162
 163         lu_device_put(lud);
 164         th->th_dev = NULL;
 165         lu_context_exit(&th->th_ctx);
 166         lu_context_fini(&th->th_ctx);
 167         OBD_FREE_PTR(oh);
 168
 169         EXIT;
 170 }
 171
 172 static int osd_trans_cb_add(struct thandle *th, struct dt_txn_commit_cb *dcb)
 173 {
 174         struct osd_thandle *oh;
 175
 176         oh = container_of0(th, struct osd_thandle, ot_super);
 177         cfs_list_add(&dcb->dcb_linkage, &oh->ot_dcb_list);
 178
 179         return 0;
 180 }
 181
 182 /*
 183  * Concurrency: shouldn't matter.
 184  */
 185 static int osd_trans_start(const struct lu_env *env, struct dt_device *d,
 186                            struct thandle *th)
 187 {
 188         struct osd_thandle      *oh;
 189         int                     rc;
 190         ENTRY;
 191
 192         oh = container_of0(th, struct osd_thandle, ot_super);
 193         LASSERT(oh);
 194         LASSERT(oh->ot_tx);
 195
 196         rc = dt_txn_hook_start(env, d, th);
 197         if (rc != 0)
 198                 RETURN(rc);
 199
 200         if (oh->ot_write_commit && OBD_FAIL_CHECK(OBD_FAIL_OST_MAPBLK_ENOSPC))
 201                 /* Unlike ldiskfs, ZFS checks for available space and returns
 202                  * -ENOSPC when assigning txg */
 203                 RETURN(-ENOSPC);
 204
 205         rc = -dmu_tx_assign(oh->ot_tx, TXG_WAIT);
 206         if (unlikely(rc != 0)) {
 207                 struct osd_device *osd = osd_dt_dev(d);
 208                 /* dmu will call commit callback with error code during abort */
 209                 if (!lu_device_is_md(&d->dd_lu_dev) && rc == -ENOSPC)
 210                         CERROR("%s: failed to start transaction due to ENOSPC. "
 211                                "Metadata overhead is underestimated or "
 212                                "grant_ratio is too low.\n", osd->od_svname);
 213                 else
 214                         CERROR("%s: can't assign tx: rc = %d\n",
 215                                osd->od_svname, rc);
 216         } else {
 217                 /* add commit callback */
 218                 dmu_tx_callback_register(oh->ot_tx, osd_trans_commit_cb, oh);
 219                 oh->ot_assigned = 1;
 220                 lu_context_init(&th->th_ctx, th->th_tags);
 221                 lu_context_enter(&th->th_ctx);
 222                 lu_device_get(&d->dd_lu_dev);
 223         }
 224
 225         RETURN(rc);
 226 }
 227
 228 /*
 229  * Concurrency: shouldn't matter.
 230  */
 231 static int osd_trans_stop(const struct lu_env *env, struct thandle *th)
 232 {
 233         struct osd_device       *osd = osd_dt_dev(th->th_dev);
 234         struct osd_thandle      *oh;
 235         uint64_t                 txg;
 236         int                      rc;
 237         ENTRY;
 238
 239         oh = container_of0(th, struct osd_thandle, ot_super);
 240
 241         if (oh->ot_assigned == 0) {
 242                 LASSERT(oh->ot_tx);
 243                 dmu_tx_abort(oh->ot_tx);
 244                 osd_object_sa_dirty_rele(oh);
 245                 /* there won't be any commit, release reserved quota space now,
 246                  * if any */
 247                 qsd_op_end(env, osd->od_quota_slave, &oh->ot_quota_trans);
 248                 OBD_FREE_PTR(oh);
 249                 RETURN(0);
 250         }
 251
 252         /* When doing our own inode accounting, the ZAPs storing per-uid/gid
 253          * usage are updated at operation execution time, so we should call
 254          * qsd_op_end() straight away. Otherwise (for blk accounting maintained
 255          * by ZFS and when #inode is estimated from #blks) accounting is updated
 256          * at commit time and the call to qsd_op_end() must be delayed */
 257         if (oh->ot_quota_trans.lqt_id_cnt > 0 &&
 258                         !oh->ot_quota_trans.lqt_ids[0].lqi_is_blk &&
 259                         !osd->od_quota_iused_est)
 260                 qsd_op_end(env, osd->od_quota_slave, &oh->ot_quota_trans);
 261
 262         rc = dt_txn_hook_stop(env, th);
 263         if (rc != 0)
 264                 CDEBUG(D_OTHER, "%s: transaction hook failed: rc = %d\n",
 265                        osd->od_svname, rc);
 266
 267         LASSERT(oh->ot_tx);
 268         txg = oh->ot_tx->tx_txg;
 269
 270         osd_object_sa_dirty_rele(oh);
 271         dmu_tx_commit(oh->ot_tx);
 272
 273         if (th->th_sync)
 274                 txg_wait_synced(dmu_objset_pool(osd->od_objset.os), txg);
 275
 276         RETURN(rc);
 277 }
 278
 279 static struct thandle *osd_trans_create(const struct lu_env *env,
 280                                         struct dt_device *dt)
 281 {
 282         struct osd_device       *osd = osd_dt_dev(dt);
 283         struct osd_thandle      *oh;
 284         struct thandle          *th;
 285         dmu_tx_t                *tx;
 286         ENTRY;
 287
 288         tx = dmu_tx_create(osd->od_objset.os);
 289         if (tx == NULL)
 290                 RETURN(ERR_PTR(-ENOMEM));
 291
 292         /* alloc callback data */
 293         OBD_ALLOC_PTR(oh);
 294         if (oh == NULL) {
 295                 dmu_tx_abort(tx);
 296                 RETURN(ERR_PTR(-ENOMEM));
 297         }
 298
 299         oh->ot_tx = tx;
 300         CFS_INIT_LIST_HEAD(&oh->ot_dcb_list);
 301         CFS_INIT_LIST_HEAD(&oh->ot_sa_list);
 302         cfs_sema_init(&oh->ot_sa_lock, 1);
 303         memset(&oh->ot_quota_trans, 0, sizeof(oh->ot_quota_trans));
 304         th = &oh->ot_super;
 305         th->th_dev = dt;
 306         th->th_result = 0;
 307         th->th_tags = LCT_TX_HANDLE;
 308         RETURN(th);
 309 }
 310
 311 /*
 312  * Concurrency: shouldn't matter.
 313  */
 314 int osd_statfs(const struct lu_env *env, struct dt_device *d,
 315                struct obd_statfs *osfs)
 316 {
 317         struct osd_device *osd = osd_dt_dev(d);
 318         int                rc;
 319         ENTRY;
 320
 321         rc = udmu_objset_statfs(&osd->od_objset, osfs);
 322         if (rc)
 323                 RETURN(rc);
 324         osfs->os_bavail -= min_t(obd_size,
 325                                  OSD_GRANT_FOR_LOCAL_OIDS / osfs->os_bsize,
 326                                  osfs->os_bavail);
 327         RETURN(0);
 328 }
 329
 330 /*
 331  * Concurrency: doesn't access mutable data.
 332  */
 333 static void osd_conf_get(const struct lu_env *env,
 334                          const struct dt_device *dev,
 335                          struct dt_device_param *param)
 336 {
 337         /*
 338          * XXX should be taken from not-yet-existing fs abstraction layer.
 339          */
 340         param->ddp_max_name_len  = MAXNAMELEN;
 341         param->ddp_max_nlink     = 1 << 31; /* it's 8byte on a disk */
 342         param->ddp_block_shift   = 12; /* XXX */
 343         param->ddp_mount_type    = LDD_MT_ZFS;
 344
 345         param->ddp_mntopts        = MNTOPT_USERXATTR | MNTOPT_ACL;
 346         param->ddp_max_ea_size    = DXATTR_MAX_ENTRY_SIZE;
 347
 348         /* for maxbytes, report same value as ZPL */
 349         param->ddp_maxbytes      = MAX_LFS_FILESIZE;
 350
 351         /* Default reserved fraction of the available space that should be kept
 352          * for error margin. Unfortunately, there are many factors that can
 353          * impact the overhead with zfs, so let's be very cautious for now and
 354          * reserve 20% of the available space which is not given out as grant.
 355          * This tunable can be changed on a live system via procfs if needed. */
 356         param->ddp_grant_reserved = 20;
 357
 358         /* inodes are dynamically allocated, so we report the per-inode space
 359          * consumption to upper layers. This static value is not really accurate
 360          * and we should use the same logic as in udmu_objset_statfs() to
 361          * estimate the real size consumed by an object */
 362         param->ddp_inodespace = OSD_DNODE_EST_COUNT;
 363         /* per-fragment overhead to be used by the client code */
 364         param->ddp_grant_frag = udmu_blk_insert_cost();
 365
 366         param->ddp_mnt = NULL;
 367 }
 368
 369 /*
 370  * Concurrency: shouldn't matter.
 371  */
 372 static int osd_sync(const struct lu_env *env, struct dt_device *d)
 373 {
 374         struct osd_device  *osd = osd_dt_dev(d);
 375         CDEBUG(D_HA, "syncing OSD %s\n", LUSTRE_OSD_ZFS_NAME);
 376         txg_wait_synced(dmu_objset_pool(osd->od_objset.os), 0ULL);
 377         return 0;
 378 }
 379
 380 static int osd_commit_async(const struct lu_env *env, struct dt_device *dev)
 381 {
 382         struct osd_device *osd = osd_dt_dev(dev);
 383         tx_state_t        *tx = &dmu_objset_pool(osd->od_objset.os)->dp_tx;
 384         uint64_t           txg;
 385
 386         txg = tx->tx_open_txg + 1;
 387         if (tx->tx_quiesce_txg_waiting < txg) {
 388                 tx->tx_quiesce_txg_waiting = txg;
 389                 cv_broadcast(&tx->tx_quiesce_more_cv);
 390         }
 391         mutex_exit(&tx->tx_sync_lock);
 392
 393         return 0;
 394 }
 395
 396 /*
 397  * Concurrency: shouldn't matter.
 398  */
 399 static int osd_ro(const struct lu_env *env, struct dt_device *d)
 400 {
 401         struct osd_device  *osd = osd_dt_dev(d);
 402         ENTRY;
 403
 404         CERROR("%s: *** setting device %s read-only ***\n",
 405                osd->od_svname, LUSTRE_OSD_ZFS_NAME);
 406         osd->od_rdonly = 1;
 407         spa_freeze(dmu_objset_spa(osd->od_objset.os));
 408
 409         RETURN(0);
 410 }
 411
 412 /*
 413  * Concurrency: serialization provided by callers.
 414  */
 415 static int osd_init_capa_ctxt(const struct lu_env *env, struct dt_device *d,
 416                               int mode, unsigned long timeout, __u32 alg,
 417                               struct lustre_capa_key *keys)
 418 {
 419         struct osd_device *dev = osd_dt_dev(d);
 420         ENTRY;
 421
 422         dev->od_fl_capa = mode;
 423         dev->od_capa_timeout = timeout;
 424         dev->od_capa_alg = alg;
 425         dev->od_capa_keys = keys;
 426
 427         RETURN(0);
 428 }
 429
 430 static struct dt_device_operations osd_dt_ops = {
 431         .dt_root_get            = osd_root_get,
 432         .dt_statfs              = osd_statfs,
 433         .dt_trans_create        = osd_trans_create,
 434         .dt_trans_start         = osd_trans_start,
 435         .dt_trans_stop          = osd_trans_stop,
 436         .dt_trans_cb_add        = osd_trans_cb_add,
 437         .dt_conf_get            = osd_conf_get,
 438         .dt_sync                = osd_sync,
 439         .dt_commit_async        = osd_commit_async,
 440         .dt_ro                  = osd_ro,
 441         .dt_init_capa_ctxt      = osd_init_capa_ctxt,
 442 };
 443
 444 /*
 445  * DMU OSD device type methods
 446  */
 447 static int osd_type_init(struct lu_device_type *t)
 448 {
 449         LU_CONTEXT_KEY_INIT(&osd_key);
 450         return lu_context_key_register(&osd_key);
 451 }
 452
 453 static void osd_type_fini(struct lu_device_type *t)
 454 {
 455         lu_context_key_degister(&osd_key);
 456 }
 457
 458 static void *osd_key_init(const struct lu_context *ctx,
 459                           struct lu_context_key *key)
 460 {
 461         struct osd_thread_info *info;
 462
 463         OBD_ALLOC_PTR(info);
 464         if (info != NULL)
 465                 info->oti_env = container_of(ctx, struct lu_env, le_ctx);
 466         else
 467                 info = ERR_PTR(-ENOMEM);
 468         return info;
 469 }
 470
 471 static void osd_key_fini(const struct lu_context *ctx,
 472                          struct lu_context_key *key, void *data)
 473 {
 474         struct osd_thread_info *info = data;
 475
 476         OBD_FREE_PTR(info);
 477 }
 478
 479 static void osd_key_exit(const struct lu_context *ctx,
 480                          struct lu_context_key *key, void *data)
 481 {
 482         struct osd_thread_info *info = data;
 483
 484         memset(info, 0, sizeof(*info));
 485 }
 486
 487 struct lu_context_key osd_key = {
 488         .lct_tags = LCT_DT_THREAD | LCT_MD_THREAD | LCT_MG_THREAD | LCT_LOCAL,
 489         .lct_init = osd_key_init,
 490         .lct_fini = osd_key_fini,
 491         .lct_exit = osd_key_exit
 492 };
 493
 494 static int osd_shutdown(const struct lu_env *env, struct osd_device *o)
 495 {
 496         ENTRY;
 497
 498         /* shutdown quota slave instance associated with the device */
 499         if (o->od_quota_slave != NULL) {
 500                 qsd_fini(env, o->od_quota_slave);
 501                 o->od_quota_slave = NULL;
 502         }
 503
 504         RETURN(0);
 505 }
 506
 507 static void osd_xattr_changed_cb(void *arg, uint64_t newval)
 508 {
 509         struct osd_device *osd = arg;
 510
 511         osd->od_xattr_in_sa = (newval == ZFS_XATTR_SA);
 512 }
 513
 514 static int osd_mount(const struct lu_env *env,
 515                      struct osd_device *o, struct lustre_cfg *cfg)
 516 {
 517         struct dsl_dataset *ds;
 518         char      *dev  = lustre_cfg_string(cfg, 1);
 519         dmu_buf_t *rootdb;
 520         int        rc;
 521         ENTRY;
 522
 523         if (o->od_objset.os != NULL)
 524                 RETURN(0);
 525
 526         if (strlen(dev) >= sizeof(o->od_mntdev))
 527                 RETURN(-E2BIG);
 528
 529         strcpy(o->od_mntdev, dev);
 530         strncpy(o->od_svname, lustre_cfg_string(cfg, 4),
 531                 sizeof(o->od_svname) - 1);
 532
 533         rc = -udmu_objset_open(o->od_mntdev, &o->od_objset);
 534         if (rc) {
 535                 CERROR("can't open objset %s: %d\n", o->od_mntdev, rc);
 536                 RETURN(rc);
 537         }
 538
 539         ds = dmu_objset_ds(o->od_objset.os);
 540         LASSERT(ds);
 541         rc = dsl_prop_register(ds, "xattr", osd_xattr_changed_cb, o);
 542         if (rc)
 543                 CERROR("%s: cat not register xattr callback, ignore: %d\n",
 544                        o->od_svname, rc);
 545
 546         rc = __osd_obj2dbuf(env, o->od_objset.os, o->od_objset.root,
 547                                 &rootdb, root_tag);
 548         if (rc) {
 549                 CERROR("udmu_obj2dbuf() failed with error %d\n", rc);
 550                 udmu_objset_close(&o->od_objset);
 551                 RETURN(rc);
 552         }
 553
 554         o->od_root = rootdb->db_object;
 555         sa_buf_rele(rootdb, root_tag);
 556
 557         /* 1. initialize oi before any file create or file open */
 558         rc = osd_oi_init(env, o);
 559         if (rc)
 560                 GOTO(err, rc);
 561
 562         rc = lu_site_init(&o->od_site, osd2lu_dev(o));
 563         if (rc)
 564                 GOTO(err, rc);
 565         o->od_site.ls_bottom_dev = osd2lu_dev(o);
 566
 567         rc = lu_site_init_finish(&o->od_site);
 568         if (rc)
 569                 GOTO(err, rc);
 570
 571         /* Use our own ZAP for inode accounting by default, this can be changed
 572          * via procfs to estimate the inode usage from the block usage */
 573         o->od_quota_iused_est = 0;
 574
 575         rc = osd_procfs_init(o, o->od_svname);
 576         if (rc)
 577                 GOTO(err, rc);
 578
 579         o->arc_prune_cb = arc_add_prune_callback(arc_prune_func, o);
 580
 581         /* initialize quota slave instance */
 582         o->od_quota_slave = qsd_init(env, o->od_svname, &o->od_dt_dev,
 583                                      o->od_proc_entry);
 584         if (IS_ERR(o->od_quota_slave)) {
 585                 rc = PTR_ERR(o->od_quota_slave);
 586                 o->od_quota_slave = NULL;
 587                 GOTO(err, rc);
 588         }
 589 err:
 590         RETURN(rc);
 591 }
 592
 593 static void osd_umount(const struct lu_env *env, struct osd_device *o)
 594 {
 595         ENTRY;
 596
 597         if (cfs_atomic_read(&o->od_zerocopy_alloc))
 598                 CERROR("%s: lost %d allocated page(s)\n", o->od_svname,
 599                        cfs_atomic_read(&o->od_zerocopy_alloc));
 600         if (cfs_atomic_read(&o->od_zerocopy_loan))
 601                 CERROR("%s: lost %d loaned abuf(s)\n", o->od_svname,
 602                        cfs_atomic_read(&o->od_zerocopy_loan));
 603         if (cfs_atomic_read(&o->od_zerocopy_pin))
 604                 CERROR("%s: lost %d pinned dbuf(s)\n", o->od_svname,
 605                        cfs_atomic_read(&o->od_zerocopy_pin));
 606
 607         if (o->od_objset.os != NULL)
 608                 udmu_objset_close(&o->od_objset);
 609
 610         EXIT;
 611 }
 612
 613 static int osd_device_init0(const struct lu_env *env,
 614                             struct osd_device *o,
 615                             struct lustre_cfg *cfg)
 616 {
 617         struct lu_device        *l = osd2lu_dev(o);
 618         int                      rc;
 619
 620         /* if the module was re-loaded, env can loose its keys */
 621         rc = lu_env_refill((struct lu_env *) env);
 622         if (rc)
 623                 GOTO(out, rc);
 624
 625         l->ld_ops = &osd_lu_ops;
 626         o->od_dt_dev.dd_ops = &osd_dt_ops;
 627
 628         o->od_capa_hash = init_capa_hash();
 629         if (o->od_capa_hash == NULL)
 630                 GOTO(out, rc = -ENOMEM);
 631
 632 out:
 633         RETURN(rc);
 634 }
 635
 636 static struct lu_device *osd_device_fini(const struct lu_env *env,
 637                                          struct lu_device *dev);
 638
 639 static struct lu_device *osd_device_alloc(const struct lu_env *env,
 640                                           struct lu_device_type *type,
 641                                           struct lustre_cfg *cfg)
 642 {
 643         struct osd_device *dev;
 644         int                rc;
 645
 646         OBD_ALLOC_PTR(dev);
 647         if (dev == NULL)
 648                 return ERR_PTR(-ENOMEM);
 649
 650         rc = dt_device_init(&dev->od_dt_dev, type);
 651         if (rc == 0) {
 652                 rc = osd_device_init0(env, dev, cfg);
 653                 if (rc == 0) {
 654                         rc = osd_mount(env, dev, cfg);
 655                         if (rc)
 656                                 osd_device_fini(env, osd2lu_dev(dev));
 657                 }
 658                 if (rc)
 659                         dt_device_fini(&dev->od_dt_dev);
 660         }
 661
 662         if (unlikely(rc != 0))
 663                 OBD_FREE_PTR(dev);
 664
 665         return rc == 0 ? osd2lu_dev(dev) : ERR_PTR(rc);
 666 }
 667
 668 static struct lu_device *osd_device_free(const struct lu_env *env,
 669                                          struct lu_device *d)
 670 {
 671         struct osd_device *o = osd_dev(d);
 672         ENTRY;
 673
 674         cleanup_capa_hash(o->od_capa_hash);
 675         /* XXX: make osd top device in order to release reference */
 676         d->ld_site->ls_top_dev = d;
 677         lu_site_purge(env, d->ld_site, -1);
 678         if (!cfs_hash_is_empty(d->ld_site->ls_obj_hash)) {
 679                 LIBCFS_DEBUG_MSG_DATA_DECL(msgdata, D_ERROR, NULL);
 680                 lu_site_print(env, d->ld_site, &msgdata, lu_cdebug_printer);
 681         }
 682         lu_site_fini(&o->od_site);
 683         dt_device_fini(&o->od_dt_dev);
 684         OBD_FREE_PTR(o);
 685
 686         RETURN (NULL);
 687 }
 688
 689 static struct lu_device *osd_device_fini(const struct lu_env *env,
 690                                          struct lu_device *d)
 691 {
 692         struct osd_device *o = osd_dev(d);
 693         struct dsl_dataset *ds;
 694         int                rc;
 695         ENTRY;
 696
 697
 698         osd_shutdown(env, o);
 699         osd_oi_fini(env, o);
 700
 701         if (o->od_objset.os) {
 702                 ds = dmu_objset_ds(o->od_objset.os);
 703                 rc = dsl_prop_unregister(ds, "xattr", osd_xattr_changed_cb, o);
 704                 if (rc)
 705                         CERROR("%s: dsl_prop_unregister xattr error %d\n",
 706                                 o->od_svname, rc);
 707                 arc_remove_prune_callback(o->arc_prune_cb);
 708                 o->arc_prune_cb = NULL;
 709                 osd_sync(env, lu2dt_dev(d));
 710                 txg_wait_callbacks(spa_get_dsl(dmu_objset_spa(o->od_objset.os)));
 711         }
 712
 713         rc = osd_procfs_fini(o);
 714         if (rc) {
 715                 CERROR("proc fini error %d\n", rc);
 716                 RETURN(ERR_PTR(rc));
 717         }
 718
 719         if (o->od_objset.os)
 720                 osd_umount(env, o);
 721
 722         RETURN(NULL);
 723 }
 724
 725 static int osd_device_init(const struct lu_env *env, struct lu_device *d,
 726                            const char *name, struct lu_device *next)
 727 {
 728         return 0;
 729 }
 730
 731 /*
 732  * To be removed, setup is performed by osd_device_{init,alloc} and
 733  * cleanup is performed by osd_device_{fini,free).
 734  */
 735 static int osd_process_config(const struct lu_env *env,
 736                               struct lu_device *d, struct lustre_cfg *cfg)
 737 {
 738         struct osd_device       *o = osd_dev(d);
 739         int                      err;
 740         ENTRY;
 741
 742         switch(cfg->lcfg_command) {
 743         case LCFG_SETUP:
 744                 err = osd_mount(env, o, cfg);
 745                 break;
 746         case LCFG_CLEANUP:
 747                 err = osd_shutdown(env, o);
 748                 break;
 749         default:
 750                 err = -ENOTTY;
 751         }
 752
 753         RETURN(err);
 754 }
 755
 756 static int osd_recovery_complete(const struct lu_env *env, struct lu_device *d)
 757 {
 758         struct osd_device       *osd = osd_dev(d);
 759         int                      rc = 0;
 760         ENTRY;
 761
 762         if (osd->od_quota_slave == NULL)
 763                 RETURN(0);
 764
 765         /* start qsd instance on recovery completion, this notifies the quota
 766          * slave code that we are about to process new requests now */
 767         rc = qsd_start(env, osd->od_quota_slave);
 768         RETURN(rc);
 769 }
 770
 771 /*
 772  * we use exports to track all osd users
 773  */
 774 static int osd_obd_connect(const struct lu_env *env, struct obd_export **exp,
 775                            struct obd_device *obd, struct obd_uuid *cluuid,
 776                            struct obd_connect_data *data, void *localdata)
 777 {
 778         struct osd_device    *osd = osd_dev(obd->obd_lu_dev);
 779         struct lustre_handle  conn;
 780         int                   rc;
 781         ENTRY;
 782
 783         CDEBUG(D_CONFIG, "connect #%d\n", osd->od_connects);
 784
 785         rc = class_connect(&conn, obd, cluuid);
 786         if (rc)
 787                 RETURN(rc);
 788
 789         *exp = class_conn2export(&conn);
 790
 791         cfs_spin_lock(&osd->od_objset.lock);
 792         osd->od_connects++;
 793         cfs_spin_unlock(&osd->od_objset.lock);
 794
 795         RETURN(0);
 796 }
 797
 798 /*
 799  * once last export (we don't count self-export) disappeared
 800  * osd can be released
 801  */
 802 static int osd_obd_disconnect(struct obd_export *exp)
 803 {
 804         struct obd_device *obd = exp->exp_obd;
 805         struct osd_device *osd = osd_dev(obd->obd_lu_dev);
 806         int                rc, release = 0;
 807         ENTRY;
 808
 809         /* Only disconnect the underlying layers on the final disconnect. */
 810         cfs_spin_lock(&osd->od_objset.lock);
 811         osd->od_connects--;
 812         if (osd->od_connects == 0)
 813                 release = 1;
 814         cfs_spin_unlock(&osd->od_objset.lock);
 815
 816         rc = class_disconnect(exp); /* bz 9811 */
 817
 818         if (rc == 0 && release)
 819                 class_manual_cleanup(obd);
 820         RETURN(rc);
 821 }
 822
 823 static int osd_prepare(const struct lu_env *env, struct lu_device *pdev,
 824                        struct lu_device *dev)
 825 {
 826         struct osd_device       *osd = osd_dev(dev);
 827         int                      rc = 0;
 828         ENTRY;
 829
 830         if (dev->ld_site && lu_device_is_md(dev->ld_site->ls_top_dev)) {
 831                 /* MDT/MDD still use old infrastructure to create
 832                  * special files */
 833                 rc = llo_local_objects_setup(env, lu2md_dev(pdev),
 834                                              lu2dt_dev(dev));
 835                 if (rc)
 836                         RETURN(rc);
 837         }
 838
 839         if (osd->od_quota_slave != NULL)
 840                 /* set up quota slave objects */
 841                 rc = qsd_prepare(env, osd->od_quota_slave);
 842
 843         RETURN(rc);
 844 }
 845
 846 struct lu_device_operations osd_lu_ops = {
 847         .ldo_object_alloc       = osd_object_alloc,
 848         .ldo_process_config     = osd_process_config,
 849         .ldo_recovery_complete  = osd_recovery_complete,
 850         .ldo_prepare            = osd_prepare,
 851 };
 852
 853 static void osd_type_start(struct lu_device_type *t)
 854 {
 855 }
 856
 857 static void osd_type_stop(struct lu_device_type *t)
 858 {
 859 }
 860
 861 static struct lu_device_type_operations osd_device_type_ops = {
 862         .ldto_init              = osd_type_init,
 863         .ldto_fini              = osd_type_fini,
 864
 865         .ldto_start             = osd_type_start,
 866         .ldto_stop              = osd_type_stop,
 867
 868         .ldto_device_alloc      = osd_device_alloc,
 869         .ldto_device_free       = osd_device_free,
 870
 871         .ldto_device_init       = osd_device_init,
 872         .ldto_device_fini       = osd_device_fini
 873 };
 874
 875 static struct lu_device_type osd_device_type = {
 876         .ldt_tags     = LU_DEVICE_DT,
 877         .ldt_name     = LUSTRE_OSD_ZFS_NAME,
 878         .ldt_ops      = &osd_device_type_ops,
 879         .ldt_ctx_tags = LCT_LOCAL
 880 };
 881
 882
 883 static struct obd_ops osd_obd_device_ops = {
 884         .o_owner       = THIS_MODULE,
 885         .o_connect      = osd_obd_connect,
 886         .o_disconnect   = osd_obd_disconnect
 887 };
 888
 889 int __init osd_init(void)
 890 {
 891         int rc;
 892
 893         rc = osd_options_init();
 894         if (rc)
 895                 return rc;
 896
 897         rc = lu_kmem_init(osd_caches);
 898         if (rc)
 899                 return rc;
 900
 901         rc = class_register_type(&osd_obd_device_ops, NULL,
 902                                  lprocfs_osd_module_vars,
 903                                  LUSTRE_OSD_ZFS_NAME, &osd_device_type);
 904         if (rc)
 905                 lu_kmem_fini(osd_caches);
 906         return rc;
 907 }
 908
 909 void __exit osd_exit(void)
 910 {
 911         class_unregister_type(LUSTRE_OSD_ZFS_NAME);
 912         lu_kmem_fini(osd_caches);
 913 }
 914
 915 extern unsigned int osd_oi_count;
 916 CFS_MODULE_PARM(osd_oi_count, "i", int, 0444,
 917                 "Number of Object Index containers to be created, "
 918                 "it's only valid for new filesystem.");
 919
 920 MODULE_AUTHOR("Sun Microsystems, Inc. <http://www.lustre.org/>");
 921 MODULE_DESCRIPTION("Lustre Object Storage Device ("LUSTRE_OSD_ZFS_NAME")");
 922 MODULE_LICENSE("GPL");
 923
 924 cfs_module(osd, LUSTRE_VERSION_STRING, osd_init, osd_exit);