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