Whamcloud - gitweb
LU-10041 osd: osd-zfs to choose dnode size
[fs/lustre-release.git] / 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.gnu.org/licenses/gpl-2.0.html
19  *
20  * GPL HEADER END
21  */
22 /*
23  * Copyright (c) 2009, 2010, Oracle and/or its affiliates. All rights reserved.
24  * Use is subject to license terms.
25  *
26  * Copyright (c) 2012, 2016, Intel Corporation.
27  */
28 /*
29  * This file is part of Lustre, http://www.lustre.org/
30  * Lustre is a trademark of Sun Microsystems, Inc.
31  *
32  * lustre/osd-zfs/osd_handler.c
33  * Top-level entry points into osd module
34  *
35  * Author: Alex Zhuravlev <bzzz@whamcloud.com>
36  * Author: Mike Pershin <tappro@whamcloud.com>
37  * Author: Johann Lombardi <johann@whamcloud.com>
38  */
39
40 #define DEBUG_SUBSYSTEM S_OSD
41
42 #include <libcfs/libcfs.h>
43 #include <obd_support.h>
44 #include <lustre_net.h>
45 #include <obd.h>
46 #include <obd_class.h>
47 #include <lustre_disk.h>
48 #include <lustre_fid.h>
49 #include <uapi/linux/lustre/lustre_param.h>
50 #include <md_object.h>
51
52 #include "osd_internal.h"
53
54 #include <sys/dnode.h>
55 #include <sys/dbuf.h>
56 #include <sys/spa.h>
57 #include <sys/stat.h>
58 #include <sys/zap.h>
59 #include <sys/spa_impl.h>
60 #include <sys/zfs_znode.h>
61 #include <sys/dmu_tx.h>
62 #include <sys/dmu_objset.h>
63 #include <sys/dsl_prop.h>
64 #include <sys/sa_impl.h>
65 #include <sys/txg.h>
66
67 struct lu_context_key   osd_key;
68
69 /* Slab for OSD object allocation */
70 struct kmem_cache *osd_object_kmem;
71
72 /* Slab to allocate osd_zap_it */
73 struct kmem_cache *osd_zapit_cachep;
74
75 static struct lu_kmem_descr osd_caches[] = {
76         {
77                 .ckd_cache = &osd_object_kmem,
78                 .ckd_name  = "zfs_osd_obj",
79                 .ckd_size  = sizeof(struct osd_object)
80         },
81         {
82                 .ckd_cache = &osd_zapit_cachep,
83                 .ckd_name  = "osd_zapit_cache",
84                 .ckd_size  = sizeof(struct osd_zap_it)
85         },
86         {
87                 .ckd_cache = NULL
88         }
89 };
90
91 static void arc_prune_func(int64_t bytes, void *private)
92 {
93         struct osd_device *od = private;
94         struct lu_site    *site = &od->od_site;
95         struct lu_env      env;
96         int rc;
97
98         LASSERT(site->ls_obj_hash);
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         list_for_each_entry_safe(dcb, tmp, &oh->ot_dcb_list, dcb_linkage) {
152                 LASSERTF(dcb->dcb_magic == TRANS_COMMIT_CB_MAGIC,
153                          "commit callback entry: magic=%x name='%s'\n",
154                          dcb->dcb_magic, dcb->dcb_name);
155                 list_del_init(&dcb->dcb_linkage);
156                 dcb->dcb_func(NULL, th, dcb, error);
157         }
158
159         /* Unlike ldiskfs, zfs updates space accounting at commit time.
160          * As a consequence, op_end is called only now to inform the quota slave
161          * component that reserved quota space is now accounted in usage and
162          * should be released. Quota space won't be adjusted at this point since
163          * we can't provide a suitable environment. It will be performed
164          * asynchronously by a lquota thread. */
165         qsd_op_end(NULL, osd->od_quota_slave, &oh->ot_quota_trans);
166
167         lu_device_put(lud);
168         th->th_dev = NULL;
169         lu_context_exit(&th->th_ctx);
170         lu_context_fini(&th->th_ctx);
171         OBD_FREE_PTR(oh);
172
173         EXIT;
174 }
175
176 static int osd_trans_cb_add(struct thandle *th, struct dt_txn_commit_cb *dcb)
177 {
178         struct osd_thandle *oh = container_of0(th, struct osd_thandle,
179                                                ot_super);
180
181         LASSERT(dcb->dcb_magic == TRANS_COMMIT_CB_MAGIC);
182         LASSERT(&dcb->dcb_func != NULL);
183         if (dcb->dcb_flags & DCB_TRANS_STOP)
184                 list_add(&dcb->dcb_linkage, &oh->ot_stop_dcb_list);
185         else
186                 list_add(&dcb->dcb_linkage, &oh->ot_dcb_list);
187
188         return 0;
189 }
190
191 /*
192  * Concurrency: shouldn't matter.
193  */
194 static int osd_trans_start(const struct lu_env *env, struct dt_device *d,
195                            struct thandle *th)
196 {
197         struct osd_thandle      *oh;
198         int                     rc;
199         ENTRY;
200
201         oh = container_of0(th, struct osd_thandle, ot_super);
202         LASSERT(oh);
203         LASSERT(oh->ot_tx);
204
205         rc = dt_txn_hook_start(env, d, th);
206         if (rc != 0)
207                 RETURN(rc);
208
209         if (oh->ot_write_commit && OBD_FAIL_CHECK(OBD_FAIL_OST_MAPBLK_ENOSPC))
210                 /* Unlike ldiskfs, ZFS checks for available space and returns
211                  * -ENOSPC when assigning txg */
212                 RETURN(-ENOSPC);
213
214         rc = -dmu_tx_assign(oh->ot_tx, TXG_WAIT);
215         if (unlikely(rc != 0)) {
216                 struct osd_device *osd = osd_dt_dev(d);
217                 /* dmu will call commit callback with error code during abort */
218                 if (!lu_device_is_md(&d->dd_lu_dev) && rc == -ENOSPC)
219                         CERROR("%s: failed to start transaction due to ENOSPC"
220                                "\n", osd->od_svname);
221                 else
222                         CERROR("%s: can't assign tx: rc = %d\n",
223                                osd->od_svname, rc);
224         } else {
225                 /* add commit callback */
226                 dmu_tx_callback_register(oh->ot_tx, osd_trans_commit_cb, oh);
227                 oh->ot_assigned = 1;
228                 lu_context_init(&th->th_ctx, th->th_tags);
229                 lu_context_enter(&th->th_ctx);
230                 lu_device_get(&d->dd_lu_dev);
231         }
232
233         RETURN(rc);
234 }
235
236 static void osd_unlinked_list_emptify(const struct lu_env *env,
237                                       struct osd_device *osd,
238                                       struct list_head *list, bool free)
239 {
240         struct osd_object *obj;
241         uint64_t           oid;
242
243         while (!list_empty(list)) {
244                 obj = list_entry(list->next,
245                                  struct osd_object, oo_unlinked_linkage);
246                 LASSERT(obj->oo_dn != NULL);
247                 oid = obj->oo_dn->dn_object;
248
249                 list_del_init(&obj->oo_unlinked_linkage);
250                 if (free)
251                         (void)osd_unlinked_object_free(env, osd, oid);
252         }
253 }
254
255 static void osd_trans_stop_cb(struct osd_thandle *oth, int result)
256 {
257         struct dt_txn_commit_cb *dcb;
258         struct dt_txn_commit_cb *tmp;
259
260         /* call per-transaction stop callbacks if any */
261         list_for_each_entry_safe(dcb, tmp, &oth->ot_stop_dcb_list,
262                                  dcb_linkage) {
263                 LASSERTF(dcb->dcb_magic == TRANS_COMMIT_CB_MAGIC,
264                          "commit callback entry: magic=%x name='%s'\n",
265                          dcb->dcb_magic, dcb->dcb_name);
266                 list_del_init(&dcb->dcb_linkage);
267                 dcb->dcb_func(NULL, &oth->ot_super, dcb, result);
268         }
269 }
270
271 /*
272  * Concurrency: shouldn't matter.
273  */
274 static int osd_trans_stop(const struct lu_env *env, struct dt_device *dt,
275                           struct thandle *th)
276 {
277         struct osd_device       *osd = osd_dt_dev(th->th_dev);
278         bool                     sync = (th->th_sync != 0);
279         struct osd_thandle      *oh;
280         struct list_head         unlinked;
281         uint64_t                 txg;
282         int                      rc;
283         ENTRY;
284
285         oh = container_of0(th, struct osd_thandle, ot_super);
286         INIT_LIST_HEAD(&unlinked);
287         list_splice_init(&oh->ot_unlinked_list, &unlinked);
288         /* reset OI cache for safety */
289         osd_oti_get(env)->oti_ins_cache_used = 0;
290
291         if (oh->ot_assigned == 0) {
292                 LASSERT(oh->ot_tx);
293                 dmu_tx_abort(oh->ot_tx);
294                 osd_object_sa_dirty_rele(env, oh);
295                 osd_unlinked_list_emptify(env, osd, &unlinked, false);
296                 /* there won't be any commit, release reserved quota space now,
297                  * if any */
298                 qsd_op_end(env, osd->od_quota_slave, &oh->ot_quota_trans);
299                 OBD_FREE_PTR(oh);
300                 RETURN(0);
301         }
302
303         rc = dt_txn_hook_stop(env, th);
304         if (rc != 0)
305                 CDEBUG(D_OTHER, "%s: transaction hook failed: rc = %d\n",
306                        osd->od_svname, rc);
307
308         osd_trans_stop_cb(oh, rc);
309
310         LASSERT(oh->ot_tx);
311         txg = oh->ot_tx->tx_txg;
312
313         osd_object_sa_dirty_rele(env, oh);
314         /* XXX: Once dmu_tx_commit() called, oh/th could have been freed
315          * by osd_trans_commit_cb already. */
316         dmu_tx_commit(oh->ot_tx);
317
318         osd_unlinked_list_emptify(env, osd, &unlinked, true);
319
320         if (sync)
321                 txg_wait_synced(dmu_objset_pool(osd->od_os), txg);
322
323         RETURN(rc);
324 }
325
326 static struct thandle *osd_trans_create(const struct lu_env *env,
327                                         struct dt_device *dt)
328 {
329         struct osd_device       *osd = osd_dt_dev(dt);
330         struct osd_thandle      *oh;
331         struct thandle          *th;
332         dmu_tx_t                *tx;
333         ENTRY;
334
335         if (dt->dd_rdonly) {
336                 CERROR("%s: someone try to start transaction under "
337                        "readonly mode, should be disabled.\n",
338                        osd_name(osd_dt_dev(dt)));
339                 dump_stack();
340                 RETURN(ERR_PTR(-EROFS));
341         }
342
343         tx = dmu_tx_create(osd->od_os);
344         if (tx == NULL)
345                 RETURN(ERR_PTR(-ENOMEM));
346
347         /* alloc callback data */
348         OBD_ALLOC_PTR(oh);
349         if (oh == NULL) {
350                 dmu_tx_abort(tx);
351                 RETURN(ERR_PTR(-ENOMEM));
352         }
353
354         oh->ot_tx = tx;
355         INIT_LIST_HEAD(&oh->ot_dcb_list);
356         INIT_LIST_HEAD(&oh->ot_stop_dcb_list);
357         INIT_LIST_HEAD(&oh->ot_unlinked_list);
358         INIT_LIST_HEAD(&oh->ot_sa_list);
359         memset(&oh->ot_quota_trans, 0, sizeof(oh->ot_quota_trans));
360         th = &oh->ot_super;
361         th->th_dev = dt;
362         th->th_result = 0;
363         th->th_tags = LCT_TX_HANDLE;
364         RETURN(th);
365 }
366
367 /* Estimate the total number of objects from a number of blocks */
368 uint64_t osd_objs_count_estimate(uint64_t usedbytes, uint64_t usedobjs,
369                                  uint64_t nrblocks, uint64_t est_maxblockshift)
370 {
371         uint64_t est_totobjs, est_usedblocks, est_usedobjs;
372
373         /*
374          * If blocksize is below 64KB (e.g. MDT with recordsize=4096) then
375          * bump the free dnode estimate to assume blocks at least 64KB in
376          * case of a directory-heavy MDT (at 32KB/directory).
377          */
378         if (est_maxblockshift < 16) {
379                 nrblocks >>= (16 - est_maxblockshift);
380                 est_maxblockshift = 16;
381         }
382
383         /*
384          * Estimate the total number of dnodes from the total blocks count
385          * and the space used per dnode.  Since we don't know the overhead
386          * associated with each dnode (xattrs, SAs, VDEV overhead, etc.)
387          * just using DNODE_SHIFT isn't going to give a good estimate.
388          * Instead, compute the current average space usage per dnode, with
389          * an upper and lower cap to avoid unrealistic estimates..
390          *
391          * In case there aren't many dnodes or blocks used yet, add a small
392          * correction factor (OSD_DNODE_EST_{COUNT,BLKSHIFT}).  This factor
393          * gradually disappears as the number of real dnodes grows.  It also
394          * avoids the need to check for divide-by-zero computing dn_per_block.
395          */
396         CLASSERT(OSD_DNODE_MIN_BLKSHIFT > 0);
397         CLASSERT(OSD_DNODE_EST_BLKSHIFT > 0);
398
399         est_usedblocks = ((OSD_DNODE_EST_COUNT << OSD_DNODE_EST_BLKSHIFT) +
400                           usedbytes) >> est_maxblockshift;
401         est_usedobjs   = OSD_DNODE_EST_COUNT + usedobjs;
402
403         if (est_usedobjs <= est_usedblocks) {
404                 /*
405                  * Average space/dnode more than maximum block size, use max
406                  * block size to estimate free dnodes from adjusted free blocks
407                  * count.  OSTs typically use multiple blocks per dnode so this
408                  * case applies.
409                  */
410                 est_totobjs = nrblocks;
411
412         } else if (est_usedobjs >= (est_usedblocks << OSD_DNODE_MIN_BLKSHIFT)) {
413                 /*
414                  * Average space/dnode smaller than min dnode size (probably
415                  * due to metadnode compression), use min dnode size to
416                  * estimate object count.  MDTs may use only one block per node
417                  * so this case applies.
418                  */
419                 est_totobjs = nrblocks << OSD_DNODE_MIN_BLKSHIFT;
420
421         } else {
422                 /*
423                  * Between the extremes, use average space per existing dnode
424                  * to compute the number of dnodes that will fit into nrblocks:
425                  *
426                  *    est_totobjs = nrblocks * (est_usedobjs / est_usedblocks)
427                  *
428                  * this may overflow 64 bits or become 0 if not handled well.
429                  *
430                  * We know nrblocks is below 2^(64 - blkbits) bits, and
431                  * est_usedobjs is under 48 bits due to DN_MAX_OBJECT_SHIFT,
432                  * which means that multiplying them may get as large as
433                  * 2 ^ 96 for the minimum blocksize of 64KB allowed above.
434                  *
435                  * The ratio of dnodes per block (est_usedobjs / est_usedblocks)
436                  * is under 2^(blkbits - DNODE_SHIFT) = blocksize / 512 due to
437                  * the limit checks above, so we can safely compute this first.
438                  * We care more about accuracy on the MDT (many dnodes/block)
439                  * which is good because this is where truncation errors are
440                  * smallest.  Since both nrblocks and dn_per_block are a
441                  * function of blkbits, their product is at most:
442                  *
443                  *    2^(64 - blkbits) * 2^(blkbits - DNODE_SHIFT) = 2^(64 - 9)
444                  *
445                  * so we can safely use 7 bits to compute a fixed-point
446                  * fraction and est_totobjs can still fit in 64 bits.
447                  */
448                 unsigned dn_per_block = (est_usedobjs << 7) / est_usedblocks;
449
450                 est_totobjs = (nrblocks * dn_per_block) >> 7;
451         }
452         return est_totobjs;
453 }
454
455 static int osd_objset_statfs(struct osd_device *osd, struct obd_statfs *osfs)
456 {
457         struct objset *os = osd->od_os;
458         uint64_t usedbytes, availbytes, usedobjs, availobjs;
459         uint64_t est_availobjs;
460         uint64_t reserved;
461         uint64_t bshift;
462
463         dmu_objset_space(os, &usedbytes, &availbytes, &usedobjs, &availobjs);
464
465         memset(osfs, 0, sizeof(*osfs));
466
467         /* We're a zfs filesystem. */
468         osfs->os_type = UBERBLOCK_MAGIC;
469
470         /*
471          * ZFS allows multiple block sizes.  For statfs, Linux makes no
472          * proper distinction between bsize and frsize.  For calculations
473          * of free and used blocks incorrectly uses bsize instead of frsize,
474          * but bsize is also used as the optimal blocksize.  We return the
475          * largest possible block size as IO size for the optimum performance
476          * and scale the free and used blocks count appropriately.
477          */
478         osfs->os_bsize = osd->od_max_blksz;
479         bshift = fls64(osfs->os_bsize) - 1;
480
481         osfs->os_blocks = (usedbytes + availbytes) >> bshift;
482         osfs->os_bfree = availbytes >> bshift;
483         osfs->os_bavail = osfs->os_bfree; /* no extra root reservation */
484
485         /* Take replication (i.e. number of copies) into account */
486         if (os->os_copies != 0)
487                 osfs->os_bavail /= os->os_copies;
488
489         /*
490          * Reserve some space so we don't run into ENOSPC due to grants not
491          * accounting for metadata overhead in ZFS, and to avoid fragmentation.
492          * Rather than report this via os_bavail (which makes users unhappy if
493          * they can't fill the filesystem 100%), reduce os_blocks as well.
494          *
495          * Reserve 0.78% of total space, at least 16MB for small filesystems,
496          * for internal files to be created/unlinked when space is tight.
497          */
498         CLASSERT(OSD_STATFS_RESERVED_SIZE > 0);
499         reserved = OSD_STATFS_RESERVED_SIZE >> bshift;
500         if (likely(osfs->os_blocks >= reserved << OSD_STATFS_RESERVED_SHIFT))
501                 reserved = osfs->os_blocks >> OSD_STATFS_RESERVED_SHIFT;
502
503         osfs->os_blocks -= reserved;
504         osfs->os_bfree  -= min(reserved, osfs->os_bfree);
505         osfs->os_bavail -= min(reserved, osfs->os_bavail);
506
507         /*
508          * The availobjs value returned from dmu_objset_space() is largely
509          * useless, since it reports the number of objects that might
510          * theoretically still fit into the dataset, independent of minor
511          * issues like how much space is actually available in the pool.
512          * Compute a better estimate in udmu_objs_count_estimate().
513          */
514         est_availobjs = osd_objs_count_estimate(usedbytes, usedobjs,
515                                                 osfs->os_bfree, bshift);
516
517         osfs->os_ffree = min(availobjs, est_availobjs);
518         osfs->os_files = osfs->os_ffree + usedobjs;
519
520         /* ZFS XXX: fill in backing dataset FSID/UUID
521            memcpy(osfs->os_fsid, .... );*/
522
523         osfs->os_namelen = MAXNAMELEN;
524         osfs->os_maxbytes = OBD_OBJECT_EOF;
525
526         if (!spa_writeable(dmu_objset_spa(os)) ||
527             osd->od_dev_set_rdonly || osd->od_prop_rdonly)
528                 osfs->os_state |= OS_STATE_READONLY;
529
530         return 0;
531 }
532
533 /*
534  * Concurrency: shouldn't matter.
535  */
536 int osd_statfs(const struct lu_env *env, struct dt_device *d,
537                struct obd_statfs *osfs)
538 {
539         int                rc;
540         ENTRY;
541
542         rc = osd_objset_statfs(osd_dt_dev(d), osfs);
543         if (unlikely(rc != 0))
544                 RETURN(rc);
545
546         osfs->os_bavail -= min_t(u64,
547                                  OSD_GRANT_FOR_LOCAL_OIDS / osfs->os_bsize,
548                                  osfs->os_bavail);
549         RETURN(0);
550 }
551
552 static int osd_blk_insert_cost(struct osd_device *osd)
553 {
554         int max_blockshift, nr_blkptrshift, bshift;
555
556         /* max_blockshift is the log2 of the number of blocks needed to reach
557          * the maximum filesize (that's to say 2^64) */
558         bshift = osd_spa_maxblockshift(dmu_objset_spa(osd->od_os));
559         max_blockshift = DN_MAX_OFFSET_SHIFT - bshift;
560
561         /* nr_blkptrshift is the log2 of the number of block pointers that can
562          * be stored in an indirect block */
563         CLASSERT(DN_MAX_INDBLKSHIFT > SPA_BLKPTRSHIFT);
564         nr_blkptrshift = DN_MAX_INDBLKSHIFT - SPA_BLKPTRSHIFT;
565
566         /* max_blockshift / nr_blkptrshift is thus the maximum depth of the
567          * tree. We add +1 for rounding purpose.
568          * The tree depth times the indirect block size gives us the maximum
569          * cost of inserting a block in the tree */
570         return (max_blockshift / nr_blkptrshift + 1) * (1<<DN_MAX_INDBLKSHIFT);
571 }
572
573 /*
574  * Concurrency: doesn't access mutable data.
575  */
576 static void osd_conf_get(const struct lu_env *env,
577                          const struct dt_device *dev,
578                          struct dt_device_param *param)
579 {
580         struct osd_device *osd = osd_dt_dev(dev);
581
582         /*
583          * XXX should be taken from not-yet-existing fs abstraction layer.
584          */
585         param->ddp_max_name_len = MAXNAMELEN;
586         param->ddp_max_nlink    = 1 << 31; /* it's 8byte on a disk */
587         param->ddp_symlink_max  = PATH_MAX;
588         param->ddp_mount_type   = LDD_MT_ZFS;
589
590         param->ddp_mntopts      = MNTOPT_USERXATTR;
591         if (osd->od_posix_acl)
592                 param->ddp_mntopts |= MNTOPT_ACL;
593         param->ddp_max_ea_size  = DXATTR_MAX_ENTRY_SIZE;
594
595         /* for maxbytes, report same value as ZPL */
596         param->ddp_maxbytes     = MAX_LFS_FILESIZE;
597
598         /* inodes are dynamically allocated, so we report the per-inode space
599          * consumption to upper layers. This static value is not really accurate
600          * and we should use the same logic as in udmu_objset_statfs() to
601          * estimate the real size consumed by an object */
602         param->ddp_inodespace = OSD_DNODE_EST_COUNT;
603         /* Although ZFS isn't an extent-based filesystem, the metadata overhead
604          * (i.e. 7 levels of indirect blocks, see osd_blk_insert_cost()) should
605          * not be accounted for every single new block insertion.
606          * Instead, the maximum extent size is set to the number of blocks that
607          * can fit into a single contiguous indirect block. There would be some
608          * cases where this crosses indirect blocks, but it also won't have 7
609          * new levels of indirect blocks in that case either, so it will still
610          * have enough reserved space for the extra indirect block */
611         param->ddp_max_extent_blks =
612                 (1 << (DN_MAX_INDBLKSHIFT - SPA_BLKPTRSHIFT));
613         param->ddp_extent_tax = osd_blk_insert_cost(osd);
614 }
615
616 /*
617  * Concurrency: shouldn't matter.
618  */
619 static int osd_sync(const struct lu_env *env, struct dt_device *d)
620 {
621         if (!d->dd_rdonly) {
622                 struct osd_device  *osd = osd_dt_dev(d);
623
624                 CDEBUG(D_CACHE, "syncing OSD %s\n", LUSTRE_OSD_ZFS_NAME);
625                 txg_wait_synced(dmu_objset_pool(osd->od_os), 0ULL);
626                 CDEBUG(D_CACHE, "synced OSD %s\n", LUSTRE_OSD_ZFS_NAME);
627         }
628
629         return 0;
630 }
631
632 static int osd_commit_async(const struct lu_env *env, struct dt_device *dev)
633 {
634         struct osd_device *osd = osd_dt_dev(dev);
635         tx_state_t        *tx = &dmu_objset_pool(osd->od_os)->dp_tx;
636         uint64_t           txg;
637
638         mutex_enter(&tx->tx_sync_lock);
639         txg = tx->tx_open_txg + 1;
640         if (tx->tx_quiesce_txg_waiting < txg) {
641                 tx->tx_quiesce_txg_waiting = txg;
642                 cv_broadcast(&tx->tx_quiesce_more_cv);
643         }
644         mutex_exit(&tx->tx_sync_lock);
645
646         return 0;
647 }
648
649 /*
650  * Concurrency: shouldn't matter.
651  */
652 static int osd_ro(const struct lu_env *env, struct dt_device *d)
653 {
654         struct osd_device  *osd = osd_dt_dev(d);
655         ENTRY;
656
657         CERROR("%s: *** setting device %s read-only ***\n",
658                osd->od_svname, LUSTRE_OSD_ZFS_NAME);
659         osd->od_dev_set_rdonly = 1;
660         spa_freeze(dmu_objset_spa(osd->od_os));
661
662         RETURN(0);
663 }
664
665 static struct dt_device_operations osd_dt_ops = {
666         .dt_root_get            = osd_root_get,
667         .dt_statfs              = osd_statfs,
668         .dt_trans_create        = osd_trans_create,
669         .dt_trans_start         = osd_trans_start,
670         .dt_trans_stop          = osd_trans_stop,
671         .dt_trans_cb_add        = osd_trans_cb_add,
672         .dt_conf_get            = osd_conf_get,
673         .dt_sync                = osd_sync,
674         .dt_commit_async        = osd_commit_async,
675         .dt_ro                  = osd_ro,
676 };
677
678 /*
679  * DMU OSD device type methods
680  */
681 static int osd_type_init(struct lu_device_type *t)
682 {
683         LU_CONTEXT_KEY_INIT(&osd_key);
684         return lu_context_key_register(&osd_key);
685 }
686
687 static void osd_type_fini(struct lu_device_type *t)
688 {
689         lu_context_key_degister(&osd_key);
690 }
691
692 static void *osd_key_init(const struct lu_context *ctx,
693                           struct lu_context_key *key)
694 {
695         struct osd_thread_info *info;
696
697         OBD_ALLOC_PTR(info);
698         if (info != NULL)
699                 info->oti_env = container_of(ctx, struct lu_env, le_ctx);
700         else
701                 info = ERR_PTR(-ENOMEM);
702         return info;
703 }
704
705 static void osd_key_fini(const struct lu_context *ctx,
706                          struct lu_context_key *key, void *data)
707 {
708         struct osd_thread_info *info = data;
709         struct osd_idmap_cache *idc = info->oti_ins_cache;
710
711         if (idc != NULL) {
712                 LASSERT(info->oti_ins_cache_size > 0);
713                 OBD_FREE(idc, sizeof(*idc) * info->oti_ins_cache_size);
714                 info->oti_ins_cache = NULL;
715                 info->oti_ins_cache_size = 0;
716         }
717         lu_buf_free(&info->oti_xattr_lbuf);
718         OBD_FREE_PTR(info);
719 }
720
721 static void osd_key_exit(const struct lu_context *ctx,
722                          struct lu_context_key *key, void *data)
723 {
724 }
725
726 struct lu_context_key osd_key = {
727         .lct_tags = LCT_DT_THREAD | LCT_MD_THREAD | LCT_MG_THREAD | LCT_LOCAL,
728         .lct_init = osd_key_init,
729         .lct_fini = osd_key_fini,
730         .lct_exit = osd_key_exit
731 };
732
733 static void osd_fid_fini(const struct lu_env *env, struct osd_device *osd)
734 {
735         if (osd->od_cl_seq == NULL)
736                 return;
737
738         seq_client_fini(osd->od_cl_seq);
739         OBD_FREE_PTR(osd->od_cl_seq);
740         osd->od_cl_seq = NULL;
741 }
742
743 static int osd_shutdown(const struct lu_env *env, struct osd_device *o)
744 {
745         ENTRY;
746
747         /* shutdown quota slave instance associated with the device */
748         if (o->od_quota_slave != NULL) {
749                 /* complete all in-flight callbacks */
750                 osd_sync(env, &o->od_dt_dev);
751                 txg_wait_callbacks(spa_get_dsl(dmu_objset_spa(o->od_os)));
752                 qsd_fini(env, o->od_quota_slave);
753                 o->od_quota_slave = NULL;
754         }
755
756         osd_fid_fini(env, o);
757
758         RETURN(0);
759 }
760
761 static void osd_xattr_changed_cb(void *arg, uint64_t newval)
762 {
763         struct osd_device *osd = arg;
764
765         osd->od_xattr_in_sa = (newval == ZFS_XATTR_SA);
766 }
767
768 static void osd_recordsize_changed_cb(void *arg, uint64_t newval)
769 {
770         struct osd_device *osd = arg;
771
772         LASSERT(newval <= osd_spa_maxblocksize(dmu_objset_spa(osd->od_os)));
773         LASSERT(newval >= SPA_MINBLOCKSIZE);
774         LASSERT(ISP2(newval));
775
776         osd->od_max_blksz = newval;
777 }
778
779 static void osd_readonly_changed_cb(void *arg, uint64_t newval)
780 {
781         struct osd_device *osd = arg;
782
783         osd->od_prop_rdonly = !!newval;
784 }
785
786 static void osd_dnodesize_changed_cb(void *arg, uint64_t newval)
787 {
788         struct osd_device *osd = arg;
789
790         osd->od_dnsize = newval;
791 }
792
793 /*
794  * This function unregisters all registered callbacks.  It's harmless to
795  * unregister callbacks that were never registered so it is used to safely
796  * unwind a partially completed call to osd_objset_register_callbacks().
797  */
798 static void osd_objset_unregister_callbacks(struct osd_device *o)
799 {
800         struct dsl_dataset      *ds = dmu_objset_ds(o->od_os);
801
802         (void) dsl_prop_unregister(ds, zfs_prop_to_name(ZFS_PROP_XATTR),
803                                    osd_xattr_changed_cb, o);
804         (void) dsl_prop_unregister(ds, zfs_prop_to_name(ZFS_PROP_RECORDSIZE),
805                                    osd_recordsize_changed_cb, o);
806         (void) dsl_prop_unregister(ds, zfs_prop_to_name(ZFS_PROP_READONLY),
807                                    osd_readonly_changed_cb, o);
808         (void) dsl_prop_unregister(ds, zfs_prop_to_name(ZFS_PROP_DNODESIZE),
809                                    osd_readonly_changed_cb, o);
810
811         if (o->arc_prune_cb != NULL) {
812                 arc_remove_prune_callback(o->arc_prune_cb);
813                 o->arc_prune_cb = NULL;
814         }
815 }
816
817 /*
818  * Register the required callbacks to be notified when zfs properties
819  * are modified using the 'zfs(8)' command line utility.
820  */
821 static int osd_objset_register_callbacks(struct osd_device *o)
822 {
823         struct dsl_dataset      *ds = dmu_objset_ds(o->od_os);
824         dsl_pool_t              *dp = dmu_objset_pool(o->od_os);
825         int                     rc;
826
827         LASSERT(ds);
828         LASSERT(dp);
829
830         dsl_pool_config_enter(dp, FTAG);
831         rc = -dsl_prop_register(ds, zfs_prop_to_name(ZFS_PROP_XATTR),
832                                 osd_xattr_changed_cb, o);
833         if (rc)
834                 GOTO(err, rc);
835
836         rc = -dsl_prop_register(ds, zfs_prop_to_name(ZFS_PROP_RECORDSIZE),
837                                 osd_recordsize_changed_cb, o);
838         if (rc)
839                 GOTO(err, rc);
840
841         rc = -dsl_prop_register(ds, zfs_prop_to_name(ZFS_PROP_READONLY),
842                                 osd_readonly_changed_cb, o);
843         if (rc)
844                 GOTO(err, rc);
845
846         rc = -dsl_prop_register(ds, zfs_prop_to_name(ZFS_PROP_DNODESIZE),
847                                 osd_dnodesize_changed_cb, o);
848         if (rc)
849                 GOTO(err, rc);
850
851         o->arc_prune_cb = arc_add_prune_callback(arc_prune_func, o);
852 err:
853         dsl_pool_config_exit(dp, FTAG);
854         if (rc)
855                 osd_objset_unregister_callbacks(o);
856
857         RETURN(rc);
858 }
859
860 static int osd_objset_open(struct osd_device *o)
861 {
862         uint64_t        version = ZPL_VERSION;
863         uint64_t        sa_obj, unlink_obj;
864         int             rc;
865         ENTRY;
866
867         rc = -osd_dmu_objset_own(o->od_mntdev, DMU_OST_ZFS,
868                              o->od_dt_dev.dd_rdonly ? B_TRUE : B_FALSE,
869                              B_FALSE, o, &o->od_os);
870
871         if (rc) {
872                 CERROR("%s: can't open %s\n", o->od_svname, o->od_mntdev);
873                 o->od_os = NULL;
874
875                 GOTO(out, rc);
876         }
877
878         /* Check ZFS version */
879         rc = -zap_lookup(o->od_os, MASTER_NODE_OBJ,
880                          ZPL_VERSION_STR, 8, 1, &version);
881         if (rc) {
882                 CERROR("%s: Error looking up ZPL VERSION\n", o->od_mntdev);
883                 /*
884                  * We can't return ENOENT because that would mean the objset
885                  * didn't exist.
886                  */
887                 GOTO(out, rc = -EIO);
888         }
889
890         rc = -zap_lookup(o->od_os, MASTER_NODE_OBJ,
891                          ZFS_SA_ATTRS, 8, 1, &sa_obj);
892         if (rc)
893                 GOTO(out, rc);
894
895         rc = -sa_setup(o->od_os, sa_obj, zfs_attr_table,
896                        ZPL_END, &o->z_attr_table);
897         if (rc)
898                 GOTO(out, rc);
899
900         rc = -zap_lookup(o->od_os, MASTER_NODE_OBJ, ZFS_ROOT_OBJ,
901                          8, 1, &o->od_rootid);
902         if (rc) {
903                 CERROR("%s: lookup for root failed: rc = %d\n",
904                         o->od_svname, rc);
905                 GOTO(out, rc);
906         }
907
908         rc = -zap_lookup(o->od_os, MASTER_NODE_OBJ, ZFS_UNLINKED_SET,
909                          8, 1, &unlink_obj);
910         if (rc) {
911                 CERROR("%s: lookup for %s failed: rc = %d\n",
912                        o->od_svname, ZFS_UNLINKED_SET, rc);
913                 GOTO(out, rc);
914         }
915
916         /* Check that user/group usage tracking is supported */
917         if (!dmu_objset_userused_enabled(o->od_os) ||
918             DMU_USERUSED_DNODE(o->od_os)->dn_type != DMU_OT_USERGROUP_USED ||
919             DMU_GROUPUSED_DNODE(o->od_os)->dn_type != DMU_OT_USERGROUP_USED) {
920                 CERROR("%s: Space accounting not supported by this target, "
921                         "aborting\n", o->od_svname);
922                 GOTO(out, rc = -ENOTSUPP);
923         }
924
925         rc = __osd_obj2dnode(o->od_os, unlink_obj, &o->od_unlinked);
926         if (rc) {
927                 CERROR("%s: can't get dnode for unlinked: rc = %d\n",
928                        o->od_svname, rc);
929                 GOTO(out, rc);
930         }
931
932 out:
933         if (rc != 0 && o->od_os != NULL) {
934                 osd_dmu_objset_disown(o->od_os, B_FALSE, o);
935                 o->od_os = NULL;
936         }
937
938         RETURN(rc);
939 }
940
941 int osd_unlinked_object_free(const struct lu_env *env, struct osd_device *osd,
942                          uint64_t oid)
943 {
944         char *key = osd_oti_get(env)->oti_str;
945         int       rc;
946         dmu_tx_t *tx;
947
948         if (osd->od_dt_dev.dd_rdonly) {
949                 CERROR("%s: someone try to free objects under "
950                        "readonly mode, should be disabled.\n", osd_name(osd));
951                 dump_stack();
952
953                 return -EROFS;
954         }
955
956         rc = -dmu_free_long_range(osd->od_os, oid, 0, DMU_OBJECT_END);
957         if (rc != 0) {
958                 CWARN("%s: Cannot truncate %llu: rc = %d\n",
959                       osd->od_svname, oid, rc);
960                 return rc;
961         }
962
963         tx = dmu_tx_create(osd->od_os);
964         dmu_tx_hold_free(tx, oid, 0, DMU_OBJECT_END);
965         osd_tx_hold_zap(tx, osd->od_unlinked->dn_object, osd->od_unlinked,
966                         FALSE, NULL);
967         rc = -dmu_tx_assign(tx, TXG_WAIT);
968         if (rc != 0) {
969                 CWARN("%s: Cannot assign tx for %llu: rc = %d\n",
970                       osd->od_svname, oid, rc);
971                 goto failed;
972         }
973
974         snprintf(key, sizeof(osd_oti_get(env)->oti_str), "%llx", oid);
975         rc = osd_zap_remove(osd, osd->od_unlinked->dn_object,
976                             osd->od_unlinked, key, tx);
977         if (rc != 0) {
978                 CWARN("%s: Cannot remove %llu from unlinked set: rc = %d\n",
979                       osd->od_svname, oid, rc);
980                 goto failed;
981         }
982
983         rc = -dmu_object_free(osd->od_os, oid, tx);
984         if (rc != 0) {
985                 CWARN("%s: Cannot free %llu: rc = %d\n",
986                       osd->od_svname, oid, rc);
987                 goto failed;
988         }
989         dmu_tx_commit(tx);
990
991         return 0;
992
993 failed:
994         LASSERT(rc != 0);
995         dmu_tx_abort(tx);
996
997         return rc;
998 }
999
1000 static void
1001 osd_unlinked_drain(const struct lu_env *env, struct osd_device *osd)
1002 {
1003         zap_cursor_t     zc;
1004         zap_attribute_t *za = &osd_oti_get(env)->oti_za;
1005
1006         zap_cursor_init(&zc, osd->od_os, osd->od_unlinked->dn_object);
1007
1008         while (zap_cursor_retrieve(&zc, za) == 0) {
1009                 /* If cannot free the object, leave it in the unlinked set,
1010                  * until the OSD is mounted again when obd_unlinked_drain()
1011                  * will be called. */
1012                 if (osd_unlinked_object_free(env, osd, za->za_first_integer))
1013                         break;
1014                 zap_cursor_advance(&zc);
1015         }
1016
1017         zap_cursor_fini(&zc);
1018 }
1019
1020 static int osd_mount(const struct lu_env *env,
1021                      struct osd_device *o, struct lustre_cfg *cfg)
1022 {
1023         char                    *mntdev = lustre_cfg_string(cfg, 1);
1024         char                    *str    = lustre_cfg_string(cfg, 2);
1025         char                    *svname = lustre_cfg_string(cfg, 4);
1026         dnode_t *rootdn;
1027         const char              *opts;
1028         int                      rc;
1029         ENTRY;
1030
1031         if (o->od_os != NULL)
1032                 RETURN(0);
1033
1034         if (mntdev == NULL || svname == NULL)
1035                 RETURN(-EINVAL);
1036
1037         rc = strlcpy(o->od_mntdev, mntdev, sizeof(o->od_mntdev));
1038         if (rc >= sizeof(o->od_mntdev))
1039                 RETURN(-E2BIG);
1040
1041         rc = strlcpy(o->od_svname, svname, sizeof(o->od_svname));
1042         if (rc >= sizeof(o->od_svname))
1043                 RETURN(-E2BIG);
1044
1045         str = strstr(str, ":");
1046         if (str) {
1047                 unsigned long flags;
1048
1049                 rc = kstrtoul(str + 1, 10, &flags);
1050                 if (rc)
1051                         RETURN(-EINVAL);
1052
1053                 if (flags & LMD_FLG_DEV_RDONLY) {
1054                         o->od_dt_dev.dd_rdonly = 1;
1055                         LCONSOLE_WARN("%s: set dev_rdonly on this device\n",
1056                                       svname);
1057                 }
1058         }
1059
1060         if (server_name_is_ost(o->od_svname))
1061                 o->od_is_ost = 1;
1062
1063         rc = osd_objset_open(o);
1064         if (rc)
1065                 RETURN(rc);
1066
1067         o->od_xattr_in_sa = B_TRUE;
1068         o->od_max_blksz = osd_spa_maxblocksize(o->od_os->os_spa);
1069
1070         rc = __osd_obj2dnode(o->od_os, o->od_rootid, &rootdn);
1071         if (rc)
1072                 GOTO(err, rc);
1073         o->od_root = rootdn->dn_object;
1074         osd_dnode_rele(rootdn);
1075
1076         rc = __osd_obj2dnode(o->od_os, DMU_USERUSED_OBJECT,
1077                              &o->od_userused_dn);
1078         if (rc)
1079                 GOTO(err, rc);
1080
1081         rc = __osd_obj2dnode(o->od_os, DMU_GROUPUSED_OBJECT,
1082                              &o->od_groupused_dn);
1083         if (rc)
1084                 GOTO(err, rc);
1085
1086         /* 1. initialize oi before any file create or file open */
1087         rc = osd_oi_init(env, o);
1088         if (rc)
1089                 GOTO(err, rc);
1090
1091         rc = lu_site_init(&o->od_site, osd2lu_dev(o));
1092         if (rc)
1093                 GOTO(err, rc);
1094         o->od_site.ls_bottom_dev = osd2lu_dev(o);
1095
1096         rc = lu_site_init_finish(&o->od_site);
1097         if (rc)
1098                 GOTO(err, rc);
1099
1100         rc = osd_objset_register_callbacks(o);
1101         if (rc)
1102                 GOTO(err, rc);
1103
1104         rc = osd_procfs_init(o, o->od_svname);
1105         if (rc)
1106                 GOTO(err, rc);
1107
1108         /* initialize quota slave instance */
1109         o->od_quota_slave = qsd_init(env, o->od_svname, &o->od_dt_dev,
1110                                      o->od_proc_entry);
1111         if (IS_ERR(o->od_quota_slave)) {
1112                 rc = PTR_ERR(o->od_quota_slave);
1113                 o->od_quota_slave = NULL;
1114                 GOTO(err, rc);
1115         }
1116
1117 #ifdef HAVE_DMU_USEROBJ_ACCOUNTING
1118         if (!osd_dmu_userobj_accounting_available(o))
1119                 CWARN("%s: dnode accounting not enabled: "
1120                       "enable feature@userobj_accounting in pool\n",
1121                       o->od_mntdev);
1122 #endif
1123
1124         /* parse mount option "noacl", and enable ACL by default */
1125         opts = lustre_cfg_string(cfg, 3);
1126         if (opts == NULL || strstr(opts, "noacl") == NULL)
1127                 o->od_posix_acl = 1;
1128
1129         osd_unlinked_drain(env, o);
1130 err:
1131         if (rc && o->od_os) {
1132                 osd_dmu_objset_disown(o->od_os, B_FALSE, o);
1133                 o->od_os = NULL;
1134         }
1135
1136         RETURN(rc);
1137 }
1138
1139 static void osd_umount(const struct lu_env *env, struct osd_device *o)
1140 {
1141         ENTRY;
1142
1143         if (atomic_read(&o->od_zerocopy_alloc))
1144                 CERROR("%s: lost %d allocated page(s)\n", o->od_svname,
1145                        atomic_read(&o->od_zerocopy_alloc));
1146         if (atomic_read(&o->od_zerocopy_loan))
1147                 CERROR("%s: lost %d loaned abuf(s)\n", o->od_svname,
1148                        atomic_read(&o->od_zerocopy_loan));
1149         if (atomic_read(&o->od_zerocopy_pin))
1150                 CERROR("%s: lost %d pinned dbuf(s)\n", o->od_svname,
1151                        atomic_read(&o->od_zerocopy_pin));
1152
1153         if (o->od_unlinked) {
1154                 osd_dnode_rele(o->od_unlinked);
1155                 o->od_unlinked = NULL;
1156         }
1157         if (o->od_userused_dn) {
1158                 osd_dnode_rele(o->od_userused_dn);
1159                 o->od_userused_dn = NULL;
1160         }
1161         if (o->od_groupused_dn) {
1162                 osd_dnode_rele(o->od_groupused_dn);
1163                 o->od_groupused_dn = NULL;
1164         }
1165
1166         if (o->od_os != NULL) {
1167                 if (!o->od_dt_dev.dd_rdonly)
1168                         /* force a txg sync to get all commit callbacks */
1169                         txg_wait_synced(dmu_objset_pool(o->od_os), 0ULL);
1170
1171                 /* close the object set */
1172                 osd_dmu_objset_disown(o->od_os, B_FALSE, o);
1173                 o->od_os = NULL;
1174         }
1175
1176         EXIT;
1177 }
1178
1179 static int osd_device_init0(const struct lu_env *env,
1180                             struct osd_device *o,
1181                             struct lustre_cfg *cfg)
1182 {
1183         struct lu_device        *l = osd2lu_dev(o);
1184         int                      rc;
1185
1186         /* if the module was re-loaded, env can loose its keys */
1187         rc = lu_env_refill((struct lu_env *) env);
1188         if (rc)
1189                 GOTO(out, rc);
1190
1191         l->ld_ops = &osd_lu_ops;
1192         o->od_dt_dev.dd_ops = &osd_dt_ops;
1193
1194 out:
1195         RETURN(rc);
1196 }
1197
1198 static struct lu_device *osd_device_fini(const struct lu_env *env,
1199                                          struct lu_device *dev);
1200
1201 static struct lu_device *osd_device_alloc(const struct lu_env *env,
1202                                           struct lu_device_type *type,
1203                                           struct lustre_cfg *cfg)
1204 {
1205         struct osd_device       *dev;
1206         struct osd_seq_list     *osl;
1207         int                     rc;
1208
1209         OBD_ALLOC_PTR(dev);
1210         if (dev == NULL)
1211                 return ERR_PTR(-ENOMEM);
1212
1213         osl = &dev->od_seq_list;
1214         INIT_LIST_HEAD(&osl->osl_seq_list);
1215         rwlock_init(&osl->osl_seq_list_lock);
1216         sema_init(&osl->osl_seq_init_sem, 1);
1217
1218         rc = dt_device_init(&dev->od_dt_dev, type);
1219         if (rc == 0) {
1220                 rc = osd_device_init0(env, dev, cfg);
1221                 if (rc == 0) {
1222                         rc = osd_mount(env, dev, cfg);
1223                         if (rc)
1224                                 osd_device_fini(env, osd2lu_dev(dev));
1225                 }
1226                 if (rc)
1227                         dt_device_fini(&dev->od_dt_dev);
1228         }
1229
1230         if (unlikely(rc != 0))
1231                 OBD_FREE_PTR(dev);
1232
1233         return rc == 0 ? osd2lu_dev(dev) : ERR_PTR(rc);
1234 }
1235
1236 static struct lu_device *osd_device_free(const struct lu_env *env,
1237                                          struct lu_device *d)
1238 {
1239         struct osd_device *o = osd_dev(d);
1240         ENTRY;
1241
1242         /* XXX: make osd top device in order to release reference */
1243         d->ld_site->ls_top_dev = d;
1244         lu_site_purge(env, d->ld_site, -1);
1245         if (!cfs_hash_is_empty(d->ld_site->ls_obj_hash)) {
1246                 LIBCFS_DEBUG_MSG_DATA_DECL(msgdata, D_ERROR, NULL);
1247                 lu_site_print(env, d->ld_site, &msgdata, lu_cdebug_printer);
1248         }
1249         lu_site_fini(&o->od_site);
1250         dt_device_fini(&o->od_dt_dev);
1251         OBD_FREE_PTR(o);
1252
1253         RETURN (NULL);
1254 }
1255
1256 static struct lu_device *osd_device_fini(const struct lu_env *env,
1257                                          struct lu_device *d)
1258 {
1259         struct osd_device *o = osd_dev(d);
1260         int                rc;
1261         ENTRY;
1262
1263
1264         if (o->od_os) {
1265                 osd_objset_unregister_callbacks(o);
1266                 if (!o->od_dt_dev.dd_rdonly) {
1267                         osd_sync(env, lu2dt_dev(d));
1268                         txg_wait_callbacks(
1269                                         spa_get_dsl(dmu_objset_spa(o->od_os)));
1270                 }
1271         }
1272
1273         /* now with all the callbacks completed we can cleanup the remainings */
1274         osd_shutdown(env, o);
1275         osd_oi_fini(env, o);
1276
1277         rc = osd_procfs_fini(o);
1278         if (rc) {
1279                 CERROR("proc fini error %d\n", rc);
1280                 RETURN(ERR_PTR(rc));
1281         }
1282
1283         if (o->od_os)
1284                 osd_umount(env, o);
1285
1286         RETURN(NULL);
1287 }
1288
1289 static int osd_device_init(const struct lu_env *env, struct lu_device *d,
1290                            const char *name, struct lu_device *next)
1291 {
1292         return 0;
1293 }
1294
1295 /*
1296  * To be removed, setup is performed by osd_device_{init,alloc} and
1297  * cleanup is performed by osd_device_{fini,free).
1298  */
1299 static int osd_process_config(const struct lu_env *env,
1300                               struct lu_device *d, struct lustre_cfg *cfg)
1301 {
1302         struct osd_device       *o = osd_dev(d);
1303         int                     rc;
1304         ENTRY;
1305
1306         switch(cfg->lcfg_command) {
1307         case LCFG_SETUP:
1308                 rc = osd_mount(env, o, cfg);
1309                 break;
1310         case LCFG_CLEANUP:
1311                 rc = osd_shutdown(env, o);
1312                 break;
1313         case LCFG_PARAM: {
1314                 LASSERT(&o->od_dt_dev);
1315                 rc = class_process_proc_param(PARAM_OSD, lprocfs_osd_obd_vars,
1316                                               cfg, &o->od_dt_dev);
1317                 if (rc > 0 || rc == -ENOSYS) {
1318                         rc = class_process_proc_param(PARAM_OST,
1319                                                       lprocfs_osd_obd_vars,
1320                                                       cfg, &o->od_dt_dev);
1321                         if (rc > 0)
1322                                 rc = 0;
1323                 }
1324                 break;
1325         }
1326         default:
1327                 rc = -ENOTTY;
1328         }
1329
1330         RETURN(rc);
1331 }
1332
1333 static int osd_recovery_complete(const struct lu_env *env, struct lu_device *d)
1334 {
1335         struct osd_device       *osd = osd_dev(d);
1336         int                      rc = 0;
1337         ENTRY;
1338
1339         if (osd->od_quota_slave == NULL)
1340                 RETURN(0);
1341
1342         /* start qsd instance on recovery completion, this notifies the quota
1343          * slave code that we are about to process new requests now */
1344         rc = qsd_start(env, osd->od_quota_slave);
1345         RETURN(rc);
1346 }
1347
1348 /*
1349  * we use exports to track all osd users
1350  */
1351 static int osd_obd_connect(const struct lu_env *env, struct obd_export **exp,
1352                            struct obd_device *obd, struct obd_uuid *cluuid,
1353                            struct obd_connect_data *data, void *localdata)
1354 {
1355         struct osd_device    *osd = osd_dev(obd->obd_lu_dev);
1356         struct lustre_handle  conn;
1357         int                   rc;
1358         ENTRY;
1359
1360         CDEBUG(D_CONFIG, "connect #%d\n", osd->od_connects);
1361
1362         rc = class_connect(&conn, obd, cluuid);
1363         if (rc)
1364                 RETURN(rc);
1365
1366         *exp = class_conn2export(&conn);
1367
1368         spin_lock(&obd->obd_dev_lock);
1369         osd->od_connects++;
1370         spin_unlock(&obd->obd_dev_lock);
1371
1372         RETURN(0);
1373 }
1374
1375 /*
1376  * once last export (we don't count self-export) disappeared
1377  * osd can be released
1378  */
1379 static int osd_obd_disconnect(struct obd_export *exp)
1380 {
1381         struct obd_device *obd = exp->exp_obd;
1382         struct osd_device *osd = osd_dev(obd->obd_lu_dev);
1383         int                rc, release = 0;
1384         ENTRY;
1385
1386         /* Only disconnect the underlying layers on the final disconnect. */
1387         spin_lock(&obd->obd_dev_lock);
1388         osd->od_connects--;
1389         if (osd->od_connects == 0)
1390                 release = 1;
1391         spin_unlock(&obd->obd_dev_lock);
1392
1393         rc = class_disconnect(exp); /* bz 9811 */
1394
1395         if (rc == 0 && release)
1396                 class_manual_cleanup(obd);
1397         RETURN(rc);
1398 }
1399
1400 static int osd_fid_init(const struct lu_env *env, struct osd_device *osd)
1401 {
1402         struct seq_server_site  *ss = osd_seq_site(osd);
1403         int                     rc;
1404         ENTRY;
1405
1406         if (osd->od_is_ost || osd->od_cl_seq != NULL)
1407                 RETURN(0);
1408
1409         if (unlikely(ss == NULL))
1410                 RETURN(-ENODEV);
1411
1412         OBD_ALLOC_PTR(osd->od_cl_seq);
1413         if (osd->od_cl_seq == NULL)
1414                 RETURN(-ENOMEM);
1415
1416         rc = seq_client_init(osd->od_cl_seq, NULL, LUSTRE_SEQ_METADATA,
1417                              osd->od_svname, ss->ss_server_seq);
1418
1419         if (rc != 0) {
1420                 OBD_FREE_PTR(osd->od_cl_seq);
1421                 osd->od_cl_seq = NULL;
1422         }
1423
1424         RETURN(rc);
1425 }
1426
1427 static int osd_prepare(const struct lu_env *env, struct lu_device *pdev,
1428                        struct lu_device *dev)
1429 {
1430         struct osd_device       *osd = osd_dev(dev);
1431         int                      rc = 0;
1432         ENTRY;
1433
1434         if (osd->od_quota_slave != NULL) {
1435                 /* set up quota slave objects */
1436                 rc = qsd_prepare(env, osd->od_quota_slave);
1437                 if (rc != 0)
1438                         RETURN(rc);
1439         }
1440
1441         rc = osd_fid_init(env, osd);
1442
1443         RETURN(rc);
1444 }
1445
1446 struct lu_device_operations osd_lu_ops = {
1447         .ldo_object_alloc       = osd_object_alloc,
1448         .ldo_process_config     = osd_process_config,
1449         .ldo_recovery_complete  = osd_recovery_complete,
1450         .ldo_prepare            = osd_prepare,
1451 };
1452
1453 static void osd_type_start(struct lu_device_type *t)
1454 {
1455 }
1456
1457 static void osd_type_stop(struct lu_device_type *t)
1458 {
1459 }
1460
1461 int osd_fid_alloc(const struct lu_env *env, struct obd_export *exp,
1462                   struct lu_fid *fid, struct md_op_data *op_data)
1463 {
1464         struct osd_device *osd = osd_dev(exp->exp_obd->obd_lu_dev);
1465
1466         return seq_client_alloc_fid(env, osd->od_cl_seq, fid);
1467 }
1468
1469 static struct lu_device_type_operations osd_device_type_ops = {
1470         .ldto_init              = osd_type_init,
1471         .ldto_fini              = osd_type_fini,
1472
1473         .ldto_start             = osd_type_start,
1474         .ldto_stop              = osd_type_stop,
1475
1476         .ldto_device_alloc      = osd_device_alloc,
1477         .ldto_device_free       = osd_device_free,
1478
1479         .ldto_device_init       = osd_device_init,
1480         .ldto_device_fini       = osd_device_fini
1481 };
1482
1483 static struct lu_device_type osd_device_type = {
1484         .ldt_tags     = LU_DEVICE_DT,
1485         .ldt_name     = LUSTRE_OSD_ZFS_NAME,
1486         .ldt_ops      = &osd_device_type_ops,
1487         .ldt_ctx_tags = LCT_LOCAL
1488 };
1489
1490
1491 static struct obd_ops osd_obd_device_ops = {
1492         .o_owner       = THIS_MODULE,
1493         .o_connect      = osd_obd_connect,
1494         .o_disconnect   = osd_obd_disconnect,
1495         .o_fid_alloc    = osd_fid_alloc
1496 };
1497
1498 static int __init osd_init(void)
1499 {
1500         int rc;
1501
1502         rc = osd_options_init();
1503         if (rc)
1504                 return rc;
1505
1506         rc = lu_kmem_init(osd_caches);
1507         if (rc)
1508                 return rc;
1509
1510         rc = class_register_type(&osd_obd_device_ops, NULL, true, NULL,
1511                                  LUSTRE_OSD_ZFS_NAME, &osd_device_type);
1512         if (rc)
1513                 lu_kmem_fini(osd_caches);
1514         return rc;
1515 }
1516
1517 static void __exit osd_exit(void)
1518 {
1519         class_unregister_type(LUSTRE_OSD_ZFS_NAME);
1520         lu_kmem_fini(osd_caches);
1521 }
1522
1523 extern unsigned int osd_oi_count;
1524 module_param(osd_oi_count, int, 0444);
1525 MODULE_PARM_DESC(osd_oi_count, "Number of Object Index containers to be created, it's only valid for new filesystem.");
1526
1527 MODULE_AUTHOR("OpenSFS, Inc. <http://www.lustre.org/>");
1528 MODULE_DESCRIPTION("Lustre Object Storage Device ("LUSTRE_OSD_ZFS_NAME")");
1529 MODULE_VERSION(LUSTRE_VERSION_STRING);
1530 MODULE_LICENSE("GPL");
1531
1532 module_init(osd_init);
1533 module_exit(osd_exit);