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",
 213                                osd->od_dt_dev.dd_lu_dev.ld_obd->obd_name);
 214                 else
 215                         CERROR("%s: can't assign tx: rc = %d\n",
 216                                osd->od_dt_dev.dd_lu_dev.ld_obd->obd_name, rc);
 217         } else {
 218                 /* add commit callback */
 219                 dmu_tx_callback_register(oh->ot_tx, osd_trans_commit_cb, oh);
 220                 oh->ot_assigned = 1;
 221                 lu_context_init(&th->th_ctx, th->th_tags);
 222                 lu_context_enter(&th->th_ctx);
 223                 lu_device_get(&d->dd_lu_dev);
 224         }
 225
 226         RETURN(rc);
 227 }
 228
 229 /*
 230  * Concurrency: shouldn't matter.
 231  */
 232 static int osd_trans_stop(const struct lu_env *env, struct thandle *th)
 233 {
 234         struct osd_device       *osd = osd_dt_dev(th->th_dev);
 235         struct osd_thandle      *oh;
 236         uint64_t                 txg;
 237         int                      rc;
 238         ENTRY;
 239
 240         oh = container_of0(th, struct osd_thandle, ot_super);
 241
 242         if (oh->ot_assigned == 0) {
 243                 LASSERT(oh->ot_tx);
 244                 dmu_tx_abort(oh->ot_tx);
 245                 osd_object_sa_dirty_rele(oh);
 246                 /* there won't be any commit, release reserved quota space now,
 247                  * if any */
 248                 qsd_op_end(env, osd->od_quota_slave, &oh->ot_quota_trans);
 249                 OBD_FREE_PTR(oh);
 250                 RETURN(0);
 251         }
 252
 253         /* When doing our own inode accounting, the ZAPs storing per-uid/gid
 254          * usage are updated at operation execution time, so we should call
 255          * qsd_op_end() straight away. Otherwise (for blk accounting maintained
 256          * by ZFS and when #inode is estimated from #blks) accounting is updated
 257          * at commit time and the call to qsd_op_end() must be delayed */
 258         if (oh->ot_quota_trans.lqt_id_cnt > 0 &&
 259                         !oh->ot_quota_trans.lqt_ids[0].lqi_is_blk &&
 260                         !osd->od_quota_iused_est)
 261                 qsd_op_end(env, osd->od_quota_slave, &oh->ot_quota_trans);
 262
 263         rc = dt_txn_hook_stop(env, th);
 264         if (rc != 0)
 265                 CDEBUG(D_OTHER, "%s: transaction hook failed: rc = %d\n",
 266                        osd->od_svname, rc);
 267
 268         LASSERT(oh->ot_tx);
 269         txg = oh->ot_tx->tx_txg;
 270
 271         osd_object_sa_dirty_rele(oh);
 272         dmu_tx_commit(oh->ot_tx);
 273
 274         if (th->th_sync)
 275                 txg_wait_synced(dmu_objset_pool(osd->od_objset.os), txg);
 276
 277         RETURN(rc);
 278 }
 279
 280 static struct thandle *osd_trans_create(const struct lu_env *env,
 281                                         struct dt_device *dt)
 282 {
 283         struct osd_device       *osd = osd_dt_dev(dt);
 284         struct osd_thandle      *oh;
 285         struct thandle          *th;
 286         dmu_tx_t                *tx;
 287         ENTRY;
 288
 289         tx = dmu_tx_create(osd->od_objset.os);
 290         if (tx == NULL)
 291                 RETURN(ERR_PTR(-ENOMEM));
 292
 293         /* alloc callback data */
 294         OBD_ALLOC_PTR(oh);
 295         if (oh == NULL) {
 296                 dmu_tx_abort(tx);
 297                 RETURN(ERR_PTR(-ENOMEM));
 298         }
 299
 300         oh->ot_tx = tx;
 301         CFS_INIT_LIST_HEAD(&oh->ot_dcb_list);
 302         CFS_INIT_LIST_HEAD(&oh->ot_sa_list);
 303         cfs_sema_init(&oh->ot_sa_lock, 1);
 304         memset(&oh->ot_quota_trans, 0, sizeof(oh->ot_quota_trans));
 305         th = &oh->ot_super;
 306         th->th_dev = dt;
 307         th->th_result = 0;
 308         th->th_tags = LCT_TX_HANDLE;
 309         RETURN(th);
 310 }
 311
 312 /*
 313  * Concurrency: shouldn't matter.
 314  */
 315 int osd_statfs(const struct lu_env *env, struct dt_device *d,
 316                struct obd_statfs *osfs)
 317 {
 318         struct osd_device *osd = osd_dt_dev(d);
 319         int                rc;
 320         ENTRY;
 321
 322         rc = udmu_objset_statfs(&osd->od_objset, osfs);
 323         if (rc)
 324                 RETURN(rc);
 325         osfs->os_bavail -= min_t(obd_size,
 326                                  OSD_GRANT_FOR_LOCAL_OIDS / osfs->os_bsize,
 327                                  osfs->os_bavail);
 328         RETURN(0);
 329 }
 330
 331 /*
 332  * Concurrency: doesn't access mutable data.
 333  */
 334 static void osd_conf_get(const struct lu_env *env,
 335                          const struct dt_device *dev,
 336                          struct dt_device_param *param)
 337 {
 338         /*
 339          * XXX should be taken from not-yet-existing fs abstraction layer.
 340          */
 341         param->ddp_max_name_len  = MAXNAMELEN;
 342         param->ddp_max_nlink     = 1 << 31; /* it's 8byte on a disk */
 343         param->ddp_block_shift   = 12; /* XXX */
 344         param->ddp_mount_type    = LDD_MT_ZFS;
 345
 346         param->ddp_mntopts        = MNTOPT_USERXATTR | MNTOPT_ACL;
 347         param->ddp_max_ea_size    = DXATTR_MAX_ENTRY_SIZE;
 348
 349         /* for maxbytes, report same value as ZPL */
 350         param->ddp_maxbytes      = MAX_LFS_FILESIZE;
 351
 352         /* Default reserved fraction of the available space that should be kept
 353          * for error margin. Unfortunately, there are many factors that can
 354          * impact the overhead with zfs, so let's be very cautious for now and
 355          * reserve 20% of the available space which is not given out as grant.
 356          * This tunable can be changed on a live system via procfs if needed. */
 357         param->ddp_grant_reserved = 20;
 358
 359         /* inodes are dynamically allocated, so we report the per-inode space
 360          * consumption to upper layers. This static value is not really accurate
 361          * and we should use the same logic as in udmu_objset_statfs() to
 362          * estimate the real size consumed by an object */
 363         param->ddp_inodespace = OSD_DNODE_EST_COUNT;
 364         /* per-fragment overhead to be used by the client code */
 365         param->ddp_grant_frag = udmu_blk_insert_cost();
 366
 367         param->ddp_mnt = NULL;
 368 }
 369
 370 /*
 371  * Concurrency: shouldn't matter.
 372  */
 373 static int osd_sync(const struct lu_env *env, struct dt_device *d)
 374 {
 375         struct osd_device  *osd = osd_dt_dev(d);
 376         CDEBUG(D_HA, "syncing OSD %s\n", LUSTRE_OSD_ZFS_NAME);
 377         txg_wait_synced(dmu_objset_pool(osd->od_objset.os), 0ULL);
 378         return 0;
 379 }
 380
 381 static int osd_commit_async(const struct lu_env *env, struct dt_device *dev)
 382 {
 383         struct osd_device *osd = osd_dt_dev(dev);
 384         tx_state_t        *tx = &dmu_objset_pool(osd->od_objset.os)->dp_tx;
 385         uint64_t           txg;
 386
 387         txg = tx->tx_open_txg + 1;
 388         if (tx->tx_quiesce_txg_waiting < txg) {
 389                 tx->tx_quiesce_txg_waiting = txg;
 390                 cv_broadcast(&tx->tx_quiesce_more_cv);
 391         }
 392         mutex_exit(&tx->tx_sync_lock);
 393
 394         return 0;
 395 }
 396
 397 /*
 398  * Concurrency: shouldn't matter.
 399  */
 400 static int osd_ro(const struct lu_env *env, struct dt_device *d)
 401 {
 402         struct osd_device  *osd = osd_dt_dev(d);
 403         ENTRY;
 404
 405         CERROR("%s: *** setting device %s read-only ***\n",
 406                osd->od_svname, LUSTRE_OSD_ZFS_NAME);
 407         osd->od_rdonly = 1;
 408         spa_freeze(dmu_objset_spa(osd->od_objset.os));
 409
 410         RETURN(0);
 411 }
 412
 413 /*
 414  * Concurrency: serialization provided by callers.
 415  */
 416 static int osd_init_capa_ctxt(const struct lu_env *env, struct dt_device *d,
 417                               int mode, unsigned long timeout, __u32 alg,
 418                               struct lustre_capa_key *keys)
 419 {
 420         struct osd_device *dev = osd_dt_dev(d);
 421         ENTRY;
 422
 423         dev->od_fl_capa = mode;
 424         dev->od_capa_timeout = timeout;
 425         dev->od_capa_alg = alg;
 426         dev->od_capa_keys = keys;
 427
 428         RETURN(0);
 429 }
 430
 431 static struct dt_device_operations osd_dt_ops = {
 432         .dt_root_get            = osd_root_get,
 433         .dt_statfs              = osd_statfs,
 434         .dt_trans_create        = osd_trans_create,
 435         .dt_trans_start         = osd_trans_start,
 436         .dt_trans_stop          = osd_trans_stop,
 437         .dt_trans_cb_add        = osd_trans_cb_add,
 438         .dt_conf_get            = osd_conf_get,
 439         .dt_sync                = osd_sync,
 440         .dt_commit_async        = osd_commit_async,
 441         .dt_ro                  = osd_ro,
 442         .dt_init_capa_ctxt      = osd_init_capa_ctxt,
 443 };
 444
 445 /*
 446  * DMU OSD device type methods
 447  */
 448 static int osd_type_init(struct lu_device_type *t)
 449 {
 450         LU_CONTEXT_KEY_INIT(&osd_key);
 451         return lu_context_key_register(&osd_key);
 452 }
 453
 454 static void osd_type_fini(struct lu_device_type *t)
 455 {
 456         lu_context_key_degister(&osd_key);
 457 }
 458
 459 static void *osd_key_init(const struct lu_context *ctx,
 460                           struct lu_context_key *key)
 461 {
 462         struct osd_thread_info *info;
 463
 464         OBD_ALLOC_PTR(info);
 465         if (info != NULL)
 466                 info->oti_env = container_of(ctx, struct lu_env, le_ctx);
 467         else
 468                 info = ERR_PTR(-ENOMEM);
 469         return info;
 470 }
 471
 472 static void osd_key_fini(const struct lu_context *ctx,
 473                          struct lu_context_key *key, void *data)
 474 {
 475         struct osd_thread_info *info = data;
 476
 477         OBD_FREE_PTR(info);
 478 }
 479
 480 static void osd_key_exit(const struct lu_context *ctx,
 481                          struct lu_context_key *key, void *data)
 482 {
 483         struct osd_thread_info *info = data;
 484
 485         memset(info, 0, sizeof(*info));
 486 }
 487
 488 struct lu_context_key osd_key = {
 489         .lct_tags = LCT_DT_THREAD | LCT_MD_THREAD | LCT_MG_THREAD | LCT_LOCAL,
 490         .lct_init = osd_key_init,
 491         .lct_fini = osd_key_fini,
 492         .lct_exit = osd_key_exit
 493 };
 494
 495 static int osd_shutdown(const struct lu_env *env, struct osd_device *o)
 496 {
 497         ENTRY;
 498
 499         /* shutdown quota slave instance associated with the device */
 500         if (o->od_quota_slave != NULL) {
 501                 qsd_fini(env, o->od_quota_slave);
 502                 o->od_quota_slave = NULL;
 503         }
 504
 505         RETURN(0);
 506 }
 507
 508 static int osd_mount(const struct lu_env *env,
 509                      struct osd_device *o, struct lustre_cfg *cfg)
 510 {
 511         char      *dev  = lustre_cfg_string(cfg, 1);
 512         dmu_buf_t *rootdb;
 513         int        rc;
 514         ENTRY;
 515
 516         if (o->od_objset.os != NULL)
 517                 RETURN(0);
 518
 519         if (strlen(dev) >= sizeof(o->od_mntdev))
 520                 RETURN(-E2BIG);
 521
 522         strcpy(o->od_mntdev, dev);
 523         strncpy(o->od_svname, lustre_cfg_string(cfg, 4),
 524                 sizeof(o->od_svname) - 1);
 525
 526         rc = -udmu_objset_open(o->od_mntdev, &o->od_objset);
 527         if (rc) {
 528                 CERROR("can't open objset %s: %d\n", o->od_mntdev, rc);
 529                 RETURN(rc);
 530         }
 531
 532         rc = __osd_obj2dbuf(env, o->od_objset.os, o->od_objset.root,
 533                                 &rootdb, root_tag);
 534         if (rc) {
 535                 CERROR("udmu_obj2dbuf() failed with error %d\n", rc);
 536                 udmu_objset_close(&o->od_objset);
 537                 RETURN(rc);
 538         }
 539
 540         o->od_root = rootdb->db_object;
 541         sa_buf_rele(rootdb, root_tag);
 542
 543         /* 1. initialize oi before any file create or file open */
 544         rc = osd_oi_init(env, o);
 545         if (rc)
 546                 GOTO(err, rc);
 547
 548         rc = lu_site_init(&o->od_site, osd2lu_dev(o));
 549         if (rc)
 550                 GOTO(err, rc);
 551
 552         rc = lu_site_init_finish(&o->od_site);
 553         if (rc)
 554                 GOTO(err, rc);
 555
 556         /* Use our own ZAP for inode accounting by default, this can be changed
 557          * via procfs to estimate the inode usage from the block usage */
 558         o->od_quota_iused_est = 0;
 559
 560         rc = osd_procfs_init(o, o->od_svname);
 561         if (rc)
 562                 GOTO(err, rc);
 563
 564         o->arc_prune_cb = arc_add_prune_callback(arc_prune_func, o);
 565
 566 err:
 567         RETURN(rc);
 568 }
 569
 570 static void osd_umount(const struct lu_env *env, struct osd_device *o)
 571 {
 572         ENTRY;
 573
 574         if (cfs_atomic_read(&o->od_zerocopy_alloc))
 575                 CERROR("%s: lost %d allocated page(s)\n", o->od_svname,
 576                        cfs_atomic_read(&o->od_zerocopy_alloc));
 577         if (cfs_atomic_read(&o->od_zerocopy_loan))
 578                 CERROR("%s: lost %d loaned abuf(s)\n", o->od_svname,
 579                        cfs_atomic_read(&o->od_zerocopy_loan));
 580         if (cfs_atomic_read(&o->od_zerocopy_pin))
 581                 CERROR("%s: lost %d pinned dbuf(s)\n", o->od_svname,
 582                        cfs_atomic_read(&o->od_zerocopy_pin));
 583
 584         if (o->od_objset.os != NULL)
 585                 udmu_objset_close(&o->od_objset);
 586
 587         EXIT;
 588 }
 589
 590 static int osd_device_init0(const struct lu_env *env,
 591                             struct osd_device *o,
 592                             struct lustre_cfg *cfg)
 593 {
 594         struct lu_device        *l = osd2lu_dev(o);
 595         int                      rc;
 596
 597         /* if the module was re-loaded, env can loose its keys */
 598         rc = lu_env_refill((struct lu_env *) env);
 599         if (rc)
 600                 GOTO(out, rc);
 601
 602         l->ld_ops = &osd_lu_ops;
 603         o->od_dt_dev.dd_ops = &osd_dt_ops;
 604
 605         o->od_capa_hash = init_capa_hash();
 606         if (o->od_capa_hash == NULL)
 607                 GOTO(out, rc = -ENOMEM);
 608
 609 out:
 610         RETURN(rc);
 611 }
 612
 613 static struct lu_device *osd_device_fini(const struct lu_env *env,
 614                                          struct lu_device *dev);
 615
 616 static struct lu_device *osd_device_alloc(const struct lu_env *env,
 617                                           struct lu_device_type *type,
 618                                           struct lustre_cfg *cfg)
 619 {
 620         struct osd_device *dev;
 621         int                rc;
 622
 623         OBD_ALLOC_PTR(dev);
 624         if (dev == NULL)
 625                 return ERR_PTR(-ENOMEM);
 626
 627         rc = dt_device_init(&dev->od_dt_dev, type);
 628         if (rc == 0) {
 629                 rc = osd_device_init0(env, dev, cfg);
 630                 if (rc == 0) {
 631                         rc = osd_mount(env, dev, cfg);
 632                         if (rc)
 633                                 osd_device_fini(env, osd2lu_dev(dev));
 634                 }
 635                 if (rc)
 636                         dt_device_fini(&dev->od_dt_dev);
 637         }
 638
 639         if (unlikely(rc != 0))
 640                 OBD_FREE_PTR(dev);
 641
 642         return rc == 0 ? osd2lu_dev(dev) : ERR_PTR(rc);
 643 }
 644
 645 static struct lu_device *osd_device_free(const struct lu_env *env,
 646                                          struct lu_device *d)
 647 {
 648         struct osd_device *o = osd_dev(d);
 649         ENTRY;
 650
 651         cleanup_capa_hash(o->od_capa_hash);
 652         /* XXX: make osd top device in order to release reference */
 653         d->ld_site->ls_top_dev = d;
 654         lu_site_purge(env, d->ld_site, -1);
 655         if (!cfs_hash_is_empty(d->ld_site->ls_obj_hash)) {
 656                 LIBCFS_DEBUG_MSG_DATA_DECL(msgdata, D_ERROR, NULL);
 657                 lu_site_print(env, d->ld_site, &msgdata, lu_cdebug_printer);
 658         }
 659         lu_site_fini(&o->od_site);
 660         dt_device_fini(&o->od_dt_dev);
 661         OBD_FREE_PTR(o);
 662
 663         RETURN (NULL);
 664 }
 665
 666 static struct lu_device *osd_device_fini(const struct lu_env *env,
 667                                          struct lu_device *d)
 668 {
 669         struct osd_device *o = osd_dev(d);
 670         int                rc;
 671         ENTRY;
 672
 673
 674         osd_oi_fini(env, o);
 675
 676         if (o->od_objset.os) {
 677                 arc_remove_prune_callback(o->arc_prune_cb);
 678                 o->arc_prune_cb = NULL;
 679                 osd_sync(env, lu2dt_dev(d));
 680                 txg_wait_callbacks(spa_get_dsl(dmu_objset_spa(o->od_objset.os)));
 681         }
 682
 683         rc = osd_procfs_fini(o);
 684         if (rc) {
 685                 CERROR("proc fini error %d\n", rc);
 686                 RETURN(ERR_PTR(rc));
 687         }
 688
 689         if (o->od_objset.os)
 690                 osd_umount(env, o);
 691
 692         RETURN(NULL);
 693 }
 694
 695 static int osd_device_init(const struct lu_env *env, struct lu_device *d,
 696                            const char *name, struct lu_device *next)
 697 {
 698         return 0;
 699 }
 700
 701 /*
 702  * To be removed, setup is performed by osd_device_{init,alloc} and
 703  * cleanup is performed by osd_device_{fini,free).
 704  */
 705 static int osd_process_config(const struct lu_env *env,
 706                               struct lu_device *d, struct lustre_cfg *cfg)
 707 {
 708         struct osd_device       *o = osd_dev(d);
 709         int                      err;
 710         ENTRY;
 711
 712         switch(cfg->lcfg_command) {
 713         case LCFG_SETUP:
 714                 err = osd_mount(env, o, cfg);
 715                 break;
 716         case LCFG_CLEANUP:
 717                 err = osd_shutdown(env, o);
 718                 break;
 719         default:
 720                 err = -ENOTTY;
 721         }
 722
 723         RETURN(err);
 724 }
 725
 726 static int osd_recovery_complete(const struct lu_env *env, struct lu_device *d)
 727 {
 728         ENTRY;
 729         RETURN(0);
 730 }
 731
 732 /*
 733  * we use exports to track all osd users
 734  */
 735 static int osd_obd_connect(const struct lu_env *env, struct obd_export **exp,
 736                            struct obd_device *obd, struct obd_uuid *cluuid,
 737                            struct obd_connect_data *data, void *localdata)
 738 {
 739         struct osd_device    *osd = osd_dev(obd->obd_lu_dev);
 740         struct lustre_handle  conn;
 741         int                   rc;
 742         ENTRY;
 743
 744         CDEBUG(D_CONFIG, "connect #%d\n", osd->od_connects);
 745
 746         rc = class_connect(&conn, obd, cluuid);
 747         if (rc)
 748                 RETURN(rc);
 749
 750         *exp = class_conn2export(&conn);
 751
 752         cfs_spin_lock(&osd->od_objset.lock);
 753         osd->od_connects++;
 754         cfs_spin_unlock(&osd->od_objset.lock);
 755
 756         RETURN(0);
 757 }
 758
 759 /*
 760  * once last export (we don't count self-export) disappeared
 761  * osd can be released
 762  */
 763 static int osd_obd_disconnect(struct obd_export *exp)
 764 {
 765         struct obd_device *obd = exp->exp_obd;
 766         struct osd_device *osd = osd_dev(obd->obd_lu_dev);
 767         int                rc, release = 0;
 768         ENTRY;
 769
 770         /* Only disconnect the underlying layers on the final disconnect. */
 771         cfs_spin_lock(&osd->od_objset.lock);
 772         osd->od_connects--;
 773         if (osd->od_connects == 0)
 774                 release = 1;
 775         cfs_spin_unlock(&osd->od_objset.lock);
 776
 777         rc = class_disconnect(exp); /* bz 9811 */
 778
 779         if (rc == 0 && release)
 780                 class_manual_cleanup(obd);
 781         RETURN(rc);
 782 }
 783
 784 static int osd_prepare(const struct lu_env *env, struct lu_device *pdev,
 785                        struct lu_device *dev)
 786 {
 787         struct osd_device       *osd = osd_dev(dev);
 788         int                      rc = 0;
 789         ENTRY;
 790
 791         /* initialize quota slave instance */
 792         osd->od_quota_slave = qsd_init(env, osd->od_svname, &osd->od_dt_dev,
 793                                        osd->od_proc_entry);
 794         if (IS_ERR(osd->od_quota_slave)) {
 795                 rc = PTR_ERR(osd->od_quota_slave);
 796                 osd->od_quota_slave = NULL;
 797         }
 798
 799         RETURN(rc);
 800 }
 801
 802 struct lu_device_operations osd_lu_ops = {
 803         .ldo_object_alloc       = osd_object_alloc,
 804         .ldo_process_config     = osd_process_config,
 805         .ldo_recovery_complete  = osd_recovery_complete,
 806         .ldo_prepare            = osd_prepare,
 807 };
 808
 809 static void osd_type_start(struct lu_device_type *t)
 810 {
 811 }
 812
 813 static void osd_type_stop(struct lu_device_type *t)
 814 {
 815 }
 816
 817 static struct lu_device_type_operations osd_device_type_ops = {
 818         .ldto_init              = osd_type_init,
 819         .ldto_fini              = osd_type_fini,
 820
 821         .ldto_start             = osd_type_start,
 822         .ldto_stop              = osd_type_stop,
 823
 824         .ldto_device_alloc      = osd_device_alloc,
 825         .ldto_device_free       = osd_device_free,
 826
 827         .ldto_device_init       = osd_device_init,
 828         .ldto_device_fini       = osd_device_fini
 829 };
 830
 831 static struct lu_device_type osd_device_type = {
 832         .ldt_tags     = LU_DEVICE_DT,
 833         .ldt_name     = LUSTRE_OSD_ZFS_NAME,
 834         .ldt_ops      = &osd_device_type_ops,
 835         .ldt_ctx_tags = LCT_LOCAL
 836 };
 837
 838
 839 static struct obd_ops osd_obd_device_ops = {
 840         .o_owner       = THIS_MODULE,
 841         .o_connect      = osd_obd_connect,
 842         .o_disconnect   = osd_obd_disconnect
 843 };
 844
 845 int __init osd_init(void)
 846 {
 847         int rc;
 848
 849         rc = osd_options_init();
 850         if (rc)
 851                 return rc;
 852
 853         rc = lu_kmem_init(osd_caches);
 854         if (rc)
 855                 return rc;
 856
 857         rc = class_register_type(&osd_obd_device_ops, NULL,
 858                                  lprocfs_osd_module_vars,
 859                                  LUSTRE_OSD_ZFS_NAME, &osd_device_type);
 860         if (rc)
 861                 lu_kmem_fini(osd_caches);
 862         return rc;
 863 }
 864
 865 void __exit osd_exit(void)
 866 {
 867         class_unregister_type(LUSTRE_OSD_ZFS_NAME);
 868         lu_kmem_fini(osd_caches);
 869 }
 870
 871 extern unsigned int osd_oi_count;
 872 CFS_MODULE_PARM(osd_oi_count, "i", int, 0444,
 873                 "Number of Object Index containers to be created, "
 874                 "it's only valid for new filesystem.");
 875
 876 MODULE_AUTHOR("Sun Microsystems, Inc. <http://www.lustre.org/>");
 877 MODULE_DESCRIPTION("Lustre Object Storage Device ("LUSTRE_OSD_ZFS_NAME")");
 878 MODULE_LICENSE("GPL");
 879
 880 cfs_module(osd, LUSTRE_VERSION_STRING, osd_init, osd_exit);