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