hammer2 - Slightly reduce LZ4 output buffer limit
[dragonfly.git] / sys / vfs / hammer2 / hammer2_strategy.c
1 /*
2  * Copyright (c) 2011-2015 The DragonFly Project.  All rights reserved.
3  *
4  * This code is derived from software contributed to The DragonFly Project
5  * by Matthew Dillon <dillon@dragonflybsd.org>
6  * by Venkatesh Srinivas <vsrinivas@dragonflybsd.org>
7  * by Daniel Flores (GSOC 2013 - mentored by Matthew Dillon, compression) 
8  *
9  * Redistribution and use in source and binary forms, with or without
10  * modification, are permitted provided that the following conditions
11  * are met:
12  *
13  * 1. Redistributions of source code must retain the above copyright
14  *    notice, this list of conditions and the following disclaimer.
15  * 2. Redistributions in binary form must reproduce the above copyright
16  *    notice, this list of conditions and the following disclaimer in
17  *    the documentation and/or other materials provided with the
18  *    distribution.
19  * 3. Neither the name of The DragonFly Project nor the names of its
20  *    contributors may be used to endorse or promote products derived
21  *    from this software without specific, prior written permission.
22  *
23  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
24  * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
25  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
26  * FOR A PARTICULAR PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE
27  * COPYRIGHT HOLDERS OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
28  * INCIDENTAL, SPECIAL, EXEMPLARY OR CONSEQUENTIAL DAMAGES (INCLUDING,
29  * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
30  * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED
31  * AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
32  * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT
33  * OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
34  * SUCH DAMAGE.
35  */
36 /*
37  * This module handles low level logical file I/O (strategy) which backs
38  * the logical buffer cache.
39  *
40  * [De]compression, zero-block, check codes, and buffer cache operations
41  * for file data is handled here.
42  *
43  * Live dedup makes its home here as well.
44  */
45
46 #include <sys/param.h>
47 #include <sys/systm.h>
48 #include <sys/kernel.h>
49 #include <sys/fcntl.h>
50 #include <sys/buf.h>
51 #include <sys/proc.h>
52 #include <sys/namei.h>
53 #include <sys/mount.h>
54 #include <sys/vnode.h>
55 #include <sys/mountctl.h>
56 #include <sys/dirent.h>
57 #include <sys/uio.h>
58 #include <sys/objcache.h>
59 #include <sys/event.h>
60 #include <sys/file.h>
61 #include <vfs/fifofs/fifo.h>
62
63 #include "hammer2.h"
64 #include "hammer2_lz4.h"
65
66 #include "zlib/hammer2_zlib.h"
67
68 struct objcache *cache_buffer_read;
69 struct objcache *cache_buffer_write;
70
71 /*
72  * Strategy code (async logical file buffer I/O from system)
73  *
74  * Except for the transaction init (which should normally not block),
75  * we essentially run the strategy operation asynchronously via a XOP.
76  *
77  * XXX This isn't supposed to be able to deadlock against vfs_sync vfsync()
78  *     calls but it has in the past when multiple flushes are queued.
79  *
80  * XXX We currently terminate the transaction once we get a quorum, otherwise
81  *     the frontend can stall, but this can leave the remaining nodes with
82  *     a potential flush conflict.  We need to delay flushes on those nodes
83  *     until running transactions complete separately from the normal
84  *     transaction sequencing.  FIXME TODO.
85  */
86 static void hammer2_strategy_xop_read(hammer2_thread_t *thr,
87                                 hammer2_xop_t *arg);
88 static void hammer2_strategy_xop_write(hammer2_thread_t *thr,
89                                 hammer2_xop_t *arg);
90 static int hammer2_strategy_read(struct vop_strategy_args *ap);
91 static int hammer2_strategy_write(struct vop_strategy_args *ap);
92 static void hammer2_strategy_read_completion(hammer2_chain_t *chain,
93                                 char *data, struct bio *bio);
94
95 static hammer2_off_t hammer2_dedup_lookup(hammer2_dev_t *hmp,
96                         char **datap, int pblksize);
97
98 int
99 hammer2_vop_strategy(struct vop_strategy_args *ap)
100 {
101         struct bio *biop;
102         struct buf *bp;
103         int error;
104
105         biop = ap->a_bio;
106         bp = biop->bio_buf;
107
108         switch(bp->b_cmd) {
109         case BUF_CMD_READ:
110                 error = hammer2_strategy_read(ap);
111                 ++hammer2_iod_file_read;
112                 break;
113         case BUF_CMD_WRITE:
114                 error = hammer2_strategy_write(ap);
115                 ++hammer2_iod_file_write;
116                 break;
117         default:
118                 bp->b_error = error = EINVAL;
119                 bp->b_flags |= B_ERROR;
120                 biodone(biop);
121                 break;
122         }
123         return (error);
124 }
125
126 /*
127  * Return the largest contiguous physical disk range for the logical
128  * request, in bytes.
129  *
130  * (struct vnode *vp, off_t loffset, off_t *doffsetp, int *runp, int *runb)
131  *
132  * Basically disabled, the logical buffer write thread has to deal with
133  * buffers one-at-a-time.  Note that this should not prevent cluster_read()
134  * from reading-ahead, it simply prevents it from trying form a single
135  * cluster buffer for the logical request.  H2 already uses 64KB buffers!
136  */
137 int
138 hammer2_vop_bmap(struct vop_bmap_args *ap)
139 {
140         *ap->a_doffsetp = NOOFFSET;
141         if (ap->a_runp)
142                 *ap->a_runp = 0;
143         if (ap->a_runb)
144                 *ap->a_runb = 0;
145         return (EOPNOTSUPP);
146 }
147
148 /****************************************************************************
149  *                              READ SUPPORT                                *
150  ****************************************************************************/
151 /* 
152  * Callback used in read path in case that a block is compressed with LZ4.
153  */
154 static
155 void
156 hammer2_decompress_LZ4_callback(const char *data, u_int bytes, struct bio *bio)
157 {
158         struct buf *bp;
159         char *compressed_buffer;
160         int compressed_size;
161         int result;
162
163         bp = bio->bio_buf;
164
165 #if 0
166         if bio->bio_caller_info2.index &&
167               bio->bio_caller_info1.uvalue32 !=
168               crc32(bp->b_data, bp->b_bufsize) --- return error
169 #endif
170
171         KKASSERT(bp->b_bufsize <= HAMMER2_PBUFSIZE);
172         compressed_size = *(const int *)data;
173         KKASSERT((uint32_t)compressed_size <= bytes - sizeof(int));
174
175         compressed_buffer = objcache_get(cache_buffer_read, M_INTWAIT);
176         result = LZ4_decompress_safe(__DECONST(char *, &data[sizeof(int)]),
177                                      compressed_buffer,
178                                      compressed_size,
179                                      bp->b_bufsize);
180         if (result < 0) {
181                 kprintf("READ PATH: Error during decompression."
182                         "bio %016jx/%d\n",
183                         (intmax_t)bio->bio_offset, bytes);
184                 /* make sure it isn't random garbage */
185                 bzero(compressed_buffer, bp->b_bufsize);
186         }
187         KKASSERT(result <= bp->b_bufsize);
188         bcopy(compressed_buffer, bp->b_data, bp->b_bufsize);
189         if (result < bp->b_bufsize)
190                 bzero(bp->b_data + result, bp->b_bufsize - result);
191         objcache_put(cache_buffer_read, compressed_buffer);
192         bp->b_resid = 0;
193         bp->b_flags |= B_AGE;
194 }
195
196 /*
197  * Callback used in read path in case that a block is compressed with ZLIB.
198  * It is almost identical to LZ4 callback, so in theory they can be unified,
199  * but we didn't want to make changes in bio structure for that.
200  */
201 static
202 void
203 hammer2_decompress_ZLIB_callback(const char *data, u_int bytes, struct bio *bio)
204 {
205         struct buf *bp;
206         char *compressed_buffer;
207         z_stream strm_decompress;
208         int result;
209         int ret;
210
211         bp = bio->bio_buf;
212
213         KKASSERT(bp->b_bufsize <= HAMMER2_PBUFSIZE);
214         strm_decompress.avail_in = 0;
215         strm_decompress.next_in = Z_NULL;
216
217         ret = inflateInit(&strm_decompress);
218
219         if (ret != Z_OK)
220                 kprintf("HAMMER2 ZLIB: Fatal error in inflateInit.\n");
221
222         compressed_buffer = objcache_get(cache_buffer_read, M_INTWAIT);
223         strm_decompress.next_in = __DECONST(char *, data);
224
225         /* XXX supply proper size, subset of device bp */
226         strm_decompress.avail_in = bytes;
227         strm_decompress.next_out = compressed_buffer;
228         strm_decompress.avail_out = bp->b_bufsize;
229
230         ret = inflate(&strm_decompress, Z_FINISH);
231         if (ret != Z_STREAM_END) {
232                 kprintf("HAMMER2 ZLIB: Fatar error during decompression.\n");
233                 bzero(compressed_buffer, bp->b_bufsize);
234         }
235         bcopy(compressed_buffer, bp->b_data, bp->b_bufsize);
236         result = bp->b_bufsize - strm_decompress.avail_out;
237         if (result < bp->b_bufsize)
238                 bzero(bp->b_data + result, strm_decompress.avail_out);
239         objcache_put(cache_buffer_read, compressed_buffer);
240         ret = inflateEnd(&strm_decompress);
241
242         bp->b_resid = 0;
243         bp->b_flags |= B_AGE;
244 }
245
246 /*
247  * Logical buffer I/O, async read.
248  */
249 static
250 int
251 hammer2_strategy_read(struct vop_strategy_args *ap)
252 {
253         hammer2_xop_strategy_t *xop;
254         struct buf *bp;
255         struct bio *bio;
256         struct bio *nbio;
257         hammer2_inode_t *ip;
258         hammer2_key_t lbase;
259
260         bio = ap->a_bio;
261         bp = bio->bio_buf;
262         ip = VTOI(ap->a_vp);
263         nbio = push_bio(bio);
264
265         lbase = bio->bio_offset;
266         KKASSERT(((int)lbase & HAMMER2_PBUFMASK) == 0);
267
268         xop = hammer2_xop_alloc(ip, HAMMER2_XOP_STRATEGY);
269         xop->finished = 0;
270         xop->bio = bio;
271         xop->lbase = lbase;
272         hammer2_mtx_init(&xop->lock, "h2bior");
273         hammer2_xop_start(&xop->head, hammer2_strategy_xop_read);
274         /* asynchronous completion */
275
276         return(0);
277 }
278
279 /*
280  * Per-node XOP (threaded), do a synchronous lookup of the chain and
281  * its data.  The frontend is asynchronous, so we are also responsible
282  * for racing to terminate the frontend.
283  */
284 static
285 void
286 hammer2_strategy_xop_read(hammer2_thread_t *thr, hammer2_xop_t *arg)
287 {
288         hammer2_xop_strategy_t *xop = &arg->xop_strategy;
289         hammer2_chain_t *parent;
290         hammer2_chain_t *chain;
291         hammer2_key_t key_dummy;
292         hammer2_key_t lbase;
293         struct bio *bio;
294         struct buf *bp;
295         int error;
296
297         /*
298          * Note that we can race completion of the bio supplied by
299          * the front-end so we cannot access it until we determine
300          * that we are the ones finishing it up.
301          */
302         lbase = xop->lbase;
303
304         /*
305          * This is difficult to optimize.  The logical buffer might be
306          * partially dirty (contain dummy zero-fill pages), which would
307          * mess up our crc calculation if we were to try a direct read.
308          * So for now we always double-buffer through the underlying
309          * storage.
310          *
311          * If not for the above problem we could conditionalize on
312          * (1) 64KB buffer, (2) one chain (not multi-master) and
313          * (3) !hammer2_double_buffer, and issue a direct read into the
314          * logical buffer.
315          */
316         parent = hammer2_inode_chain(xop->head.ip1, thr->clindex,
317                                      HAMMER2_RESOLVE_ALWAYS |
318                                      HAMMER2_RESOLVE_SHARED);
319         if (parent) {
320                 chain = hammer2_chain_lookup(&parent, &key_dummy,
321                                              lbase, lbase,
322                                              &error,
323                                              HAMMER2_LOOKUP_ALWAYS |
324                                              HAMMER2_LOOKUP_SHARED);
325                 if (chain)
326                         error = chain->error;
327         } else {
328                 error = HAMMER2_ERROR_EIO;
329                 chain = NULL;
330         }
331         error = hammer2_xop_feed(&xop->head, chain, thr->clindex, error);
332         if (chain) {
333                 hammer2_chain_unlock(chain);
334                 hammer2_chain_drop(chain);
335         }
336         if (parent) {
337                 hammer2_chain_unlock(parent);
338                 hammer2_chain_drop(parent);
339         }
340         chain = NULL;   /* safety */
341         parent = NULL;  /* safety */
342
343         /*
344          * Race to finish the frontend.  First-to-complete.  bio is only
345          * valid if we are determined to be the ones able to complete
346          * the operation.
347          */
348         if (xop->finished)
349                 return;
350         hammer2_mtx_ex(&xop->lock);
351         if (xop->finished) {
352                 hammer2_mtx_unlock(&xop->lock);
353                 return;
354         }
355         bio = xop->bio;
356         bp = bio->bio_buf;
357
358         /*
359          * Async operation has not completed and we now own the lock.
360          * Determine if we can complete the operation by issuing the
361          * frontend collection non-blocking.
362          *
363          * H2 double-buffers the data, setting B_NOTMETA on the logical
364          * buffer hints to the OS that the logical buffer should not be
365          * swapcached (since the device buffer can be).
366          *
367          * Also note that even for compressed data we would rather the
368          * kernel cache/swapcache device buffers more and (decompressed)
369          * logical buffers less, since that will significantly improve
370          * the amount of end-user data that can be cached.
371          */
372         error = hammer2_xop_collect(&xop->head, HAMMER2_XOP_COLLECT_NOWAIT);
373
374         switch(error) {
375         case 0:
376                 xop->finished = 1;
377                 hammer2_mtx_unlock(&xop->lock);
378                 bp->b_flags |= B_NOTMETA;
379                 chain = xop->head.cluster.focus;
380                 hammer2_strategy_read_completion(chain, (char *)chain->data,
381                                                  xop->bio);
382                 biodone(bio);
383                 hammer2_xop_retire(&xop->head, HAMMER2_XOPMASK_VOP);
384                 break;
385         case HAMMER2_ERROR_ENOENT:
386                 xop->finished = 1;
387                 hammer2_mtx_unlock(&xop->lock);
388                 bp->b_flags |= B_NOTMETA;
389                 bp->b_resid = 0;
390                 bp->b_error = 0;
391                 bzero(bp->b_data, bp->b_bcount);
392                 biodone(bio);
393                 hammer2_xop_retire(&xop->head, HAMMER2_XOPMASK_VOP);
394                 break;
395         case HAMMER2_ERROR_EINPROGRESS:
396                 hammer2_mtx_unlock(&xop->lock);
397                 break;
398         default:
399                 kprintf("strategy_xop_read: error %08x loff=%016jx\n",
400                         error, bp->b_loffset);
401                 xop->finished = 1;
402                 hammer2_mtx_unlock(&xop->lock);
403                 bp->b_flags |= B_ERROR;
404                 bp->b_error = EIO;
405                 biodone(bio);
406                 hammer2_xop_retire(&xop->head, HAMMER2_XOPMASK_VOP);
407                 break;
408         }
409 }
410
411 static
412 void
413 hammer2_strategy_read_completion(hammer2_chain_t *chain, char *data,
414                                  struct bio *bio)
415 {
416         struct buf *bp = bio->bio_buf;
417
418         if (chain->bref.type == HAMMER2_BREF_TYPE_INODE) {
419                 /*
420                  * Copy from in-memory inode structure.
421                  */
422                 bcopy(((hammer2_inode_data_t *)data)->u.data,
423                       bp->b_data, HAMMER2_EMBEDDED_BYTES);
424                 bzero(bp->b_data + HAMMER2_EMBEDDED_BYTES,
425                       bp->b_bcount - HAMMER2_EMBEDDED_BYTES);
426                 bp->b_resid = 0;
427                 bp->b_error = 0;
428         } else if (chain->bref.type == HAMMER2_BREF_TYPE_DATA) {
429                 /*
430                  * Data is on-media, record for live dedup.  Release the
431                  * chain (try to free it) when done.  The data is still
432                  * cached by both the buffer cache in front and the
433                  * block device behind us.  This leaves more room in the
434                  * LRU chain cache for meta-data chains which we really
435                  * want to retain.
436                  *
437                  * NOTE: Deduplication cannot be safely recorded for
438                  *       records without a check code.
439                  */
440                 hammer2_dedup_record(chain, NULL, data);
441                 atomic_set_int(&chain->flags, HAMMER2_CHAIN_RELEASE);
442
443                 /*
444                  * Decompression and copy.
445                  */
446                 switch (HAMMER2_DEC_COMP(chain->bref.methods)) {
447                 case HAMMER2_COMP_LZ4:
448                         hammer2_decompress_LZ4_callback(data, chain->bytes,
449                                                         bio);
450                         /* b_resid set by call */
451                         break;
452                 case HAMMER2_COMP_ZLIB:
453                         hammer2_decompress_ZLIB_callback(data, chain->bytes,
454                                                          bio);
455                         /* b_resid set by call */
456                         break;
457                 case HAMMER2_COMP_NONE:
458                         KKASSERT(chain->bytes <= bp->b_bcount);
459                         bcopy(data, bp->b_data, chain->bytes);
460                         if (chain->bytes < bp->b_bcount) {
461                                 bzero(bp->b_data + chain->bytes,
462                                       bp->b_bcount - chain->bytes);
463                         }
464                         bp->b_resid = 0;
465                         bp->b_error = 0;
466                         break;
467                 default:
468                         panic("hammer2_strategy_read: "
469                               "unknown compression type");
470                 }
471         } else {
472                 panic("hammer2_strategy_read: unknown bref type");
473         }
474 }
475
476 /****************************************************************************
477  *                              WRITE SUPPORT                               *
478  ****************************************************************************/
479
480 /* 
481  * Functions for compression in threads,
482  * from hammer2_vnops.c
483  */
484 static void hammer2_write_file_core(char *data, hammer2_inode_t *ip,
485                                 hammer2_chain_t **parentp,
486                                 hammer2_key_t lbase, int ioflag, int pblksize,
487                                 hammer2_tid_t mtid, int *errorp);
488 static void hammer2_compress_and_write(char *data, hammer2_inode_t *ip,
489                                 hammer2_chain_t **parentp,
490                                 hammer2_key_t lbase, int ioflag, int pblksize,
491                                 hammer2_tid_t mtid, int *errorp,
492                                 int comp_algo, int check_algo);
493 static void hammer2_zero_check_and_write(char *data, hammer2_inode_t *ip,
494                                 hammer2_chain_t **parentp,
495                                 hammer2_key_t lbase, int ioflag, int pblksize,
496                                 hammer2_tid_t mtid, int *errorp,
497                                 int check_algo);
498 static int test_block_zeros(const char *buf, size_t bytes);
499 static void zero_write(char *data, hammer2_inode_t *ip,
500                                 hammer2_chain_t **parentp,
501                                 hammer2_key_t lbase,
502                                 hammer2_tid_t mtid, int *errorp);
503 static void hammer2_write_bp(hammer2_chain_t *chain, char *data,
504                                 int ioflag, int pblksize,
505                                 hammer2_tid_t mtid, int *errorp,
506                                 int check_algo);
507
508 static
509 int
510 hammer2_strategy_write(struct vop_strategy_args *ap)
511 {       
512         hammer2_xop_strategy_t *xop;
513         hammer2_pfs_t *pmp;
514         struct bio *bio;
515         struct buf *bp;
516         hammer2_inode_t *ip;
517         
518         bio = ap->a_bio;
519         bp = bio->bio_buf;
520         ip = VTOI(ap->a_vp);
521         pmp = ip->pmp;
522         
523         hammer2_lwinprog_ref(pmp);
524         hammer2_trans_assert_strategy(pmp);
525         hammer2_trans_init(pmp, HAMMER2_TRANS_BUFCACHE);
526
527         xop = hammer2_xop_alloc(ip, HAMMER2_XOP_MODIFYING |
528                                     HAMMER2_XOP_STRATEGY);
529         xop->finished = 0;
530         xop->bio = bio;
531         xop->lbase = bio->bio_offset;
532         hammer2_mtx_init(&xop->lock, "h2biow");
533         hammer2_xop_start(&xop->head, hammer2_strategy_xop_write);
534         /* asynchronous completion */
535
536         hammer2_lwinprog_wait(pmp, hammer2_flush_pipe);
537
538         return(0);
539 }
540
541 /*
542  * Per-node XOP (threaded).  Write the logical buffer to the media.
543  *
544  * This is a bit problematic because there may be multiple target and
545  * any of them may be able to release the bp.  In addition, if our
546  * particulr target is offline we don't want to block the bp (and thus
547  * the frontend).  To accomplish this we copy the data to the per-thr
548  * scratch buffer.
549  */
550 static
551 void
552 hammer2_strategy_xop_write(hammer2_thread_t *thr, hammer2_xop_t *arg)
553 {
554         hammer2_xop_strategy_t *xop = &arg->xop_strategy;
555         hammer2_chain_t *parent;
556         hammer2_key_t lbase;
557         hammer2_inode_t *ip;
558         struct bio *bio;
559         struct buf *bp;
560         int error;
561         int lblksize;
562         int pblksize;
563         hammer2_off_t bio_offset;
564         char *bio_data;
565
566         /*
567          * We can only access the bp/bio if the frontend has not yet
568          * completed.
569          */
570         if (xop->finished)
571                 return;
572         hammer2_mtx_sh(&xop->lock);
573         if (xop->finished) {
574                 hammer2_mtx_unlock(&xop->lock);
575                 return;
576         }
577
578         lbase = xop->lbase;
579         bio = xop->bio;                 /* ephermal */
580         bp = bio->bio_buf;              /* ephermal */
581         ip = xop->head.ip1;             /* retained by ref */
582         bio_offset = bio->bio_offset;
583         bio_data = thr->scratch;
584
585         /* hammer2_trans_init(parent->hmp->spmp, HAMMER2_TRANS_BUFCACHE); */
586
587         lblksize = hammer2_calc_logical(ip, bio->bio_offset, &lbase, NULL);
588         pblksize = hammer2_calc_physical(ip, lbase);
589         bcopy(bp->b_data, bio_data, lblksize);
590
591         hammer2_mtx_unlock(&xop->lock);
592         bp = NULL;      /* safety, illegal to access after unlock */
593         bio = NULL;     /* safety, illegal to access after unlock */
594
595         /*
596          * Actual operation
597          */
598         parent = hammer2_inode_chain(ip, thr->clindex, HAMMER2_RESOLVE_ALWAYS);
599         hammer2_write_file_core(bio_data, ip, &parent,
600                                 lbase, IO_ASYNC, pblksize,
601                                 xop->head.mtid, &error);
602         if (parent) {
603                 hammer2_chain_unlock(parent);
604                 hammer2_chain_drop(parent);
605                 parent = NULL;  /* safety */
606         }
607         hammer2_xop_feed(&xop->head, NULL, thr->clindex, error);
608
609         /*
610          * Try to complete the operation on behalf of the front-end.
611          */
612         if (xop->finished)
613                 return;
614         hammer2_mtx_ex(&xop->lock);
615         if (xop->finished) {
616                 hammer2_mtx_unlock(&xop->lock);
617                 return;
618         }
619
620         /*
621          * Async operation has not completed and we now own the lock.
622          * Determine if we can complete the operation by issuing the
623          * frontend collection non-blocking.
624          *
625          * H2 double-buffers the data, setting B_NOTMETA on the logical
626          * buffer hints to the OS that the logical buffer should not be
627          * swapcached (since the device buffer can be).
628          */
629         error = hammer2_xop_collect(&xop->head, HAMMER2_XOP_COLLECT_NOWAIT);
630
631         if (error == HAMMER2_ERROR_EINPROGRESS) {
632                 hammer2_mtx_unlock(&xop->lock);
633                 return;
634         }
635
636         /*
637          * Async operation has completed.
638          */
639         xop->finished = 1;
640         hammer2_mtx_unlock(&xop->lock);
641
642         bio = xop->bio;         /* now owned by us */
643         bp = bio->bio_buf;      /* now owned by us */
644
645         if (error == HAMMER2_ERROR_ENOENT || error == 0) {
646                 bp->b_flags |= B_NOTMETA;
647                 bp->b_resid = 0;
648                 bp->b_error = 0;
649                 biodone(bio);
650         } else {
651                 kprintf("strategy_xop_write: error %d loff=%016jx\n",
652                         error, bp->b_loffset);
653                 bp->b_flags |= B_ERROR;
654                 bp->b_error = EIO;
655                 biodone(bio);
656         }
657         hammer2_xop_retire(&xop->head, HAMMER2_XOPMASK_VOP);
658         hammer2_trans_assert_strategy(ip->pmp);
659         hammer2_lwinprog_drop(ip->pmp);
660         hammer2_trans_done(ip->pmp);
661 }
662
663 /*
664  * Wait for pending I/O to complete
665  */
666 void
667 hammer2_bioq_sync(hammer2_pfs_t *pmp)
668 {
669         hammer2_lwinprog_wait(pmp, 0);
670 }
671
672 /* 
673  * Assign physical storage at (cparent, lbase), returning a suitable chain
674  * and setting *errorp appropriately.
675  *
676  * If no error occurs, the returned chain will be in a modified state.
677  *
678  * If an error occurs, the returned chain may or may not be NULL.  If
679  * not-null any chain->error (if not 0) will also be rolled up into *errorp.
680  * So the caller only needs to test *errorp.
681  *
682  * cparent can wind up being anything.
683  *
684  * If datap is not NULL, *datap points to the real data we intend to write.
685  * If we can dedup the storage location we set *datap to NULL to indicate
686  * to the caller that a dedup occurred.
687  *
688  * NOTE: Special case for data embedded in inode.
689  */
690 static
691 hammer2_chain_t *
692 hammer2_assign_physical(hammer2_inode_t *ip, hammer2_chain_t **parentp,
693                         hammer2_key_t lbase, int pblksize,
694                         hammer2_tid_t mtid, char **datap, int *errorp)
695 {
696         hammer2_chain_t *chain;
697         hammer2_key_t key_dummy;
698         hammer2_off_t dedup_off;
699         int pradix = hammer2_getradix(pblksize);
700
701         /*
702          * Locate the chain associated with lbase, return a locked chain.
703          * However, do not instantiate any data reference (which utilizes a
704          * device buffer) because we will be using direct IO via the
705          * logical buffer cache buffer.
706          */
707         KKASSERT(pblksize >= HAMMER2_ALLOC_MIN);
708
709         chain = hammer2_chain_lookup(parentp, &key_dummy,
710                                      lbase, lbase,
711                                      errorp,
712                                      HAMMER2_LOOKUP_NODATA);
713
714         /*
715          * The lookup code should not return a DELETED chain to us, unless
716          * its a short-file embedded in the inode.  Then it is possible for
717          * the lookup to return a deleted inode.
718          */
719         if (chain && (chain->flags & HAMMER2_CHAIN_DELETED) &&
720             chain->bref.type != HAMMER2_BREF_TYPE_INODE) {
721                 kprintf("assign physical deleted chain @ "
722                         "%016jx (%016jx.%02x) ip %016jx\n",
723                         lbase, chain->bref.data_off, chain->bref.type,
724                         ip->meta.inum);
725                 Debugger("bleh");
726         }
727
728         if (chain == NULL) {
729                 /*
730                  * We found a hole, create a new chain entry.
731                  *
732                  * NOTE: DATA chains are created without device backing
733                  *       store (nor do we want any).
734                  */
735                 dedup_off = hammer2_dedup_lookup((*parentp)->hmp, datap,
736                                                  pblksize);
737                 *errorp |= hammer2_chain_create(parentp, &chain,
738                                                 ip->pmp,
739                                        HAMMER2_ENC_CHECK(ip->meta.check_algo) |
740                                        HAMMER2_ENC_COMP(HAMMER2_COMP_NONE),
741                                                 lbase, HAMMER2_PBUFRADIX,
742                                                 HAMMER2_BREF_TYPE_DATA,
743                                                 pblksize, mtid,
744                                                 dedup_off, 0);
745                 if (chain == NULL)
746                         goto failed;
747                 /*ip->delta_dcount += pblksize;*/
748         } else if (chain->error == 0) {
749                 switch (chain->bref.type) {
750                 case HAMMER2_BREF_TYPE_INODE:
751                         /*
752                          * The data is embedded in the inode, which requires
753                          * a bit more finess.
754                          */
755                         *errorp |= hammer2_chain_modify_ip(ip, chain, mtid, 0);
756                         break;
757                 case HAMMER2_BREF_TYPE_DATA:
758                         dedup_off = hammer2_dedup_lookup(chain->hmp, datap,
759                                                          pblksize);
760                         if (chain->bytes != pblksize) {
761                                 *errorp |= hammer2_chain_resize(chain,
762                                                      mtid, dedup_off,
763                                                      pradix,
764                                                      HAMMER2_MODIFY_OPTDATA);
765                                 if (*errorp)
766                                         break;
767                         }
768
769                         /*
770                          * DATA buffers must be marked modified whether the
771                          * data is in a logical buffer or not.  We also have
772                          * to make this call to fixup the chain data pointers
773                          * after resizing in case this is an encrypted or
774                          * compressed buffer.
775                          */
776                         *errorp |= hammer2_chain_modify(chain, mtid, dedup_off,
777                                                         HAMMER2_MODIFY_OPTDATA);
778                         break;
779                 default:
780                         panic("hammer2_assign_physical: bad type");
781                         /* NOT REACHED */
782                         break;
783                 }
784         } else {
785                 *errorp = chain->error;
786         }
787 failed:
788         return (chain);
789 }
790
791 /* 
792  * hammer2_write_file_core() - hammer2_write_thread() helper
793  *
794  * The core write function which determines which path to take
795  * depending on compression settings.  We also have to locate the
796  * related chains so we can calculate and set the check data for
797  * the blockref.
798  */
799 static
800 void
801 hammer2_write_file_core(char *data, hammer2_inode_t *ip,
802                         hammer2_chain_t **parentp,
803                         hammer2_key_t lbase, int ioflag, int pblksize,
804                         hammer2_tid_t mtid, int *errorp)
805 {
806         hammer2_chain_t *chain;
807         char *bdata;
808
809         *errorp = 0;
810
811         switch(HAMMER2_DEC_ALGO(ip->meta.comp_algo)) {
812         case HAMMER2_COMP_NONE:
813                 /*
814                  * We have to assign physical storage to the buffer
815                  * we intend to dirty or write now to avoid deadlocks
816                  * in the strategy code later.
817                  *
818                  * This can return NOOFFSET for inode-embedded data.
819                  * The strategy code will take care of it in that case.
820                  */
821                 bdata = data;
822                 chain = hammer2_assign_physical(ip, parentp, lbase, pblksize,
823                                                 mtid, &bdata, errorp);
824                 if (*errorp) {
825                         /* skip modifications */
826                 } else if (chain->bref.type == HAMMER2_BREF_TYPE_INODE) {
827                         hammer2_inode_data_t *wipdata;
828
829                         wipdata = &chain->data->ipdata;
830                         KKASSERT(wipdata->meta.op_flags &
831                                  HAMMER2_OPFLAG_DIRECTDATA);
832                         bcopy(data, wipdata->u.data, HAMMER2_EMBEDDED_BYTES);
833                         ++hammer2_iod_file_wembed;
834                 } else if (bdata == NULL) {
835                         /*
836                          * Copy of data already present on-media.
837                          */
838                         chain->bref.methods =
839                                 HAMMER2_ENC_COMP(HAMMER2_COMP_NONE) +
840                                 HAMMER2_ENC_CHECK(ip->meta.check_algo);
841                         hammer2_chain_setcheck(chain, data);
842                 } else {
843                         hammer2_write_bp(chain, data, ioflag, pblksize,
844                                          mtid, errorp, ip->meta.check_algo);
845                 }
846                 if (chain) {
847                         hammer2_chain_unlock(chain);
848                         hammer2_chain_drop(chain);
849                 }
850                 break;
851         case HAMMER2_COMP_AUTOZERO:
852                 /*
853                  * Check for zero-fill only
854                  */
855                 hammer2_zero_check_and_write(data, ip, parentp,
856                                              lbase, ioflag, pblksize,
857                                              mtid, errorp,
858                                              ip->meta.check_algo);
859                 break;
860         case HAMMER2_COMP_LZ4:
861         case HAMMER2_COMP_ZLIB:
862         default:
863                 /*
864                  * Check for zero-fill and attempt compression.
865                  */
866                 hammer2_compress_and_write(data, ip, parentp,
867                                            lbase, ioflag, pblksize,
868                                            mtid, errorp,
869                                            ip->meta.comp_algo,
870                                            ip->meta.check_algo);
871                 break;
872         }
873 }
874
875 /*
876  * Helper
877  *
878  * Generic function that will perform the compression in compression
879  * write path. The compression algorithm is determined by the settings
880  * obtained from inode.
881  */
882 static
883 void
884 hammer2_compress_and_write(char *data, hammer2_inode_t *ip,
885         hammer2_chain_t **parentp,
886         hammer2_key_t lbase, int ioflag, int pblksize,
887         hammer2_tid_t mtid, int *errorp, int comp_algo, int check_algo)
888 {
889         hammer2_chain_t *chain;
890         int comp_size;
891         int comp_block_size;
892         char *comp_buffer;
893         char *bdata;
894
895         /*
896          * An all-zeros write creates a hole unless the check code
897          * is disabled.  When the check code is disabled all writes
898          * are done in-place, including any all-zeros writes.
899          *
900          * NOTE: A snapshot will still force a copy-on-write
901          *       (see the HAMMER2_CHECK_NONE in hammer2_chain.c).
902          */
903         if (check_algo != HAMMER2_CHECK_NONE &&
904             test_block_zeros(data, pblksize)) {
905                 zero_write(data, ip, parentp, lbase, mtid, errorp);
906                 return;
907         }
908
909         /*
910          * Compression requested.  Try to compress the block.  We store
911          * the data normally if we cannot sufficiently compress it.
912          *
913          * We have a heuristic to detect files which are mostly
914          * uncompressable and avoid the compression attempt in that
915          * case.  If the compression heuristic is turned off, we always
916          * try to compress.
917          */
918         comp_size = 0;
919         comp_buffer = NULL;
920
921         KKASSERT(pblksize / 2 <= 32768);
922                 
923         if (ip->comp_heuristic < 8 || (ip->comp_heuristic & 7) == 0 ||
924             hammer2_always_compress) {
925                 z_stream strm_compress;
926                 int comp_level;
927                 int ret;
928
929                 switch(HAMMER2_DEC_ALGO(comp_algo)) {
930                 case HAMMER2_COMP_LZ4:
931                         /*
932                          * We need to prefix with the size, LZ4
933                          * doesn't do it for us.  Add the related
934                          * overhead.
935                          *
936                          * NOTE: The LZ4 code seems to assume at least an
937                          *       8-byte buffer size granularity and may
938                          *       overrun the buffer if given a 4-byte
939                          *       granularity.
940                          */
941                         comp_buffer = objcache_get(cache_buffer_write,
942                                                    M_INTWAIT);
943                         comp_size = LZ4_compress_limitedOutput(
944                                         data,
945                                         &comp_buffer[sizeof(int)],
946                                         pblksize,
947                                         pblksize / 2 - sizeof(int64_t));
948                         *(int *)comp_buffer = comp_size;
949                         if (comp_size)
950                                 comp_size += sizeof(int);
951                         break;
952                 case HAMMER2_COMP_ZLIB:
953                         comp_level = HAMMER2_DEC_LEVEL(comp_algo);
954                         if (comp_level == 0)
955                                 comp_level = 6; /* default zlib compression */
956                         else if (comp_level < 6)
957                                 comp_level = 6;
958                         else if (comp_level > 9)
959                                 comp_level = 9;
960                         ret = deflateInit(&strm_compress, comp_level);
961                         if (ret != Z_OK) {
962                                 kprintf("HAMMER2 ZLIB: fatal error "
963                                         "on deflateInit.\n");
964                         }
965
966                         comp_buffer = objcache_get(cache_buffer_write,
967                                                    M_INTWAIT);
968                         strm_compress.next_in = data;
969                         strm_compress.avail_in = pblksize;
970                         strm_compress.next_out = comp_buffer;
971                         strm_compress.avail_out = pblksize / 2;
972                         ret = deflate(&strm_compress, Z_FINISH);
973                         if (ret == Z_STREAM_END) {
974                                 comp_size = pblksize / 2 -
975                                             strm_compress.avail_out;
976                         } else {
977                                 comp_size = 0;
978                         }
979                         ret = deflateEnd(&strm_compress);
980                         break;
981                 default:
982                         kprintf("Error: Unknown compression method.\n");
983                         kprintf("Comp_method = %d.\n", comp_algo);
984                         break;
985                 }
986         }
987
988         if (comp_size == 0) {
989                 /*
990                  * compression failed or turned off
991                  */
992                 comp_block_size = pblksize;     /* safety */
993                 if (++ip->comp_heuristic > 128)
994                         ip->comp_heuristic = 8;
995         } else {
996                 /*
997                  * compression succeeded
998                  */
999                 ip->comp_heuristic = 0;
1000                 if (comp_size <= 1024) {
1001                         comp_block_size = 1024;
1002                 } else if (comp_size <= 2048) {
1003                         comp_block_size = 2048;
1004                 } else if (comp_size <= 4096) {
1005                         comp_block_size = 4096;
1006                 } else if (comp_size <= 8192) {
1007                         comp_block_size = 8192;
1008                 } else if (comp_size <= 16384) {
1009                         comp_block_size = 16384;
1010                 } else if (comp_size <= 32768) {
1011                         comp_block_size = 32768;
1012                 } else {
1013                         panic("hammer2: WRITE PATH: "
1014                               "Weird comp_size value.");
1015                         /* NOT REACHED */
1016                         comp_block_size = pblksize;
1017                 }
1018
1019                 /*
1020                  * Must zero the remainder or dedup (which operates on a
1021                  * physical block basis) will not find matches.
1022                  */
1023                 if (comp_size < comp_block_size) {
1024                         bzero(comp_buffer + comp_size,
1025                               comp_block_size - comp_size);
1026                 }
1027         }
1028
1029         /*
1030          * Assign physical storage, data will be set to NULL if a live-dedup
1031          * was successful.
1032          */
1033         bdata = comp_size ? comp_buffer : data;
1034         chain = hammer2_assign_physical(ip, parentp, lbase, comp_block_size,
1035                                         mtid, &bdata, errorp);
1036
1037         if (*errorp) {
1038                 goto done;
1039         }
1040
1041         if (chain->bref.type == HAMMER2_BREF_TYPE_INODE) {
1042                 hammer2_inode_data_t *wipdata;
1043
1044                 *errorp = hammer2_chain_modify_ip(ip, chain, mtid, 0);
1045                 if (*errorp == 0) {
1046                         wipdata = &chain->data->ipdata;
1047                         KKASSERT(wipdata->meta.op_flags &
1048                                  HAMMER2_OPFLAG_DIRECTDATA);
1049                         bcopy(data, wipdata->u.data, HAMMER2_EMBEDDED_BYTES);
1050                         ++hammer2_iod_file_wembed;
1051                 }
1052         } else if (bdata == NULL) {
1053                 /*
1054                  * Live deduplication, a copy of the data is already present
1055                  * on the media.
1056                  */
1057                 if (comp_size) {
1058                         chain->bref.methods =
1059                                 HAMMER2_ENC_COMP(comp_algo) +
1060                                 HAMMER2_ENC_CHECK(check_algo);
1061                 } else {
1062                         chain->bref.methods =
1063                                 HAMMER2_ENC_COMP(
1064                                         HAMMER2_COMP_NONE) +
1065                                 HAMMER2_ENC_CHECK(check_algo);
1066                 }
1067                 bdata = comp_size ? comp_buffer : data;
1068                 hammer2_chain_setcheck(chain, bdata);
1069                 atomic_clear_int(&chain->flags, HAMMER2_CHAIN_INITIAL);
1070         } else {
1071                 hammer2_io_t *dio;
1072
1073                 KKASSERT(chain->flags & HAMMER2_CHAIN_MODIFIED);
1074
1075                 switch(chain->bref.type) {
1076                 case HAMMER2_BREF_TYPE_INODE:
1077                         panic("hammer2_write_bp: unexpected inode\n");
1078                         break;
1079                 case HAMMER2_BREF_TYPE_DATA:
1080                         /*
1081                          * Optimize out the read-before-write
1082                          * if possible.
1083                          */
1084                         *errorp = hammer2_io_newnz(chain->hmp,
1085                                                    chain->bref.type,
1086                                                    chain->bref.data_off,
1087                                                    chain->bytes,
1088                                                    &dio);
1089                         if (*errorp) {
1090                                 hammer2_io_brelse(&dio);
1091                                 kprintf("hammer2: WRITE PATH: "
1092                                         "dbp bread error\n");
1093                                 break;
1094                         }
1095                         bdata = hammer2_io_data(dio, chain->bref.data_off);
1096
1097                         /*
1098                          * When loading the block make sure we don't
1099                          * leave garbage after the compressed data.
1100                          */
1101                         if (comp_size) {
1102                                 chain->bref.methods =
1103                                         HAMMER2_ENC_COMP(comp_algo) +
1104                                         HAMMER2_ENC_CHECK(check_algo);
1105                                 bcopy(comp_buffer, bdata, comp_size);
1106                         } else {
1107                                 chain->bref.methods =
1108                                         HAMMER2_ENC_COMP(
1109                                                 HAMMER2_COMP_NONE) +
1110                                         HAMMER2_ENC_CHECK(check_algo);
1111                                 bcopy(data, bdata, pblksize);
1112                         }
1113
1114                         /*
1115                          * The flush code doesn't calculate check codes for
1116                          * file data (doing so can result in excessive I/O),
1117                          * so we do it here.
1118                          */
1119                         hammer2_chain_setcheck(chain, bdata);
1120
1121                         /*
1122                          * Device buffer is now valid, chain is no longer in
1123                          * the initial state.
1124                          *
1125                          * (No blockref table worries with file data)
1126                          */
1127                         atomic_clear_int(&chain->flags, HAMMER2_CHAIN_INITIAL);
1128                         hammer2_dedup_record(chain, dio, bdata);
1129
1130                         /* Now write the related bdp. */
1131                         if (ioflag & IO_SYNC) {
1132                                 /*
1133                                  * Synchronous I/O requested.
1134                                  */
1135                                 hammer2_io_bwrite(&dio);
1136                         /*
1137                         } else if ((ioflag & IO_DIRECT) &&
1138                                    loff + n == pblksize) {
1139                                 hammer2_io_bdwrite(&dio);
1140                         */
1141                         } else if (ioflag & IO_ASYNC) {
1142                                 hammer2_io_bawrite(&dio);
1143                         } else {
1144                                 hammer2_io_bdwrite(&dio);
1145                         }
1146                         break;
1147                 default:
1148                         panic("hammer2_write_bp: bad chain type %d\n",
1149                                 chain->bref.type);
1150                         /* NOT REACHED */
1151                         break;
1152                 }
1153         }
1154 done:
1155         if (chain) {
1156                 hammer2_chain_unlock(chain);
1157                 hammer2_chain_drop(chain);
1158         }
1159         if (comp_buffer)
1160                 objcache_put(cache_buffer_write, comp_buffer);
1161 }
1162
1163 /*
1164  * Helper
1165  *
1166  * Function that performs zero-checking and writing without compression,
1167  * it corresponds to default zero-checking path.
1168  */
1169 static
1170 void
1171 hammer2_zero_check_and_write(char *data, hammer2_inode_t *ip,
1172         hammer2_chain_t **parentp,
1173         hammer2_key_t lbase, int ioflag, int pblksize,
1174         hammer2_tid_t mtid, int *errorp,
1175         int check_algo)
1176 {
1177         hammer2_chain_t *chain;
1178         char *bdata;
1179
1180         if (check_algo != HAMMER2_CHECK_NONE &&
1181             test_block_zeros(data, pblksize)) {
1182                 /*
1183                  * An all-zeros write creates a hole unless the check code
1184                  * is disabled.  When the check code is disabled all writes
1185                  * are done in-place, including any all-zeros writes.
1186                  *
1187                  * NOTE: A snapshot will still force a copy-on-write
1188                  *       (see the HAMMER2_CHECK_NONE in hammer2_chain.c).
1189                  */
1190                 zero_write(data, ip, parentp, lbase, mtid, errorp);
1191         } else {
1192                 /*
1193                  * Normal write
1194                  */
1195                 bdata = data;
1196                 chain = hammer2_assign_physical(ip, parentp, lbase, pblksize,
1197                                                 mtid, &bdata, errorp);
1198                 if (*errorp) {
1199                         /* do nothing */
1200                 } else if (bdata) {
1201                         hammer2_write_bp(chain, data, ioflag, pblksize,
1202                                          mtid, errorp, check_algo);
1203                 } else {
1204                         /* dedup occurred */
1205                         chain->bref.methods =
1206                                 HAMMER2_ENC_COMP(HAMMER2_COMP_NONE) +
1207                                 HAMMER2_ENC_CHECK(check_algo);
1208                         hammer2_chain_setcheck(chain, data);
1209                 }
1210                 if (chain) {
1211                         hammer2_chain_unlock(chain);
1212                         hammer2_chain_drop(chain);
1213                 }
1214         }
1215 }
1216
1217 /*
1218  * Helper
1219  *
1220  * A function to test whether a block of data contains only zeros,
1221  * returns TRUE (non-zero) if the block is all zeros.
1222  */
1223 static
1224 int
1225 test_block_zeros(const char *buf, size_t bytes)
1226 {
1227         size_t i;
1228
1229         for (i = 0; i < bytes; i += sizeof(long)) {
1230                 if (*(const long *)(buf + i) != 0)
1231                         return (0);
1232         }
1233         return (1);
1234 }
1235
1236 /*
1237  * Helper
1238  *
1239  * Function to "write" a block that contains only zeros.
1240  */
1241 static
1242 void
1243 zero_write(char *data, hammer2_inode_t *ip,
1244            hammer2_chain_t **parentp,
1245            hammer2_key_t lbase, hammer2_tid_t mtid, int *errorp)
1246 {
1247         hammer2_chain_t *chain;
1248         hammer2_key_t key_dummy;
1249
1250         chain = hammer2_chain_lookup(parentp, &key_dummy,
1251                                      lbase, lbase,
1252                                      errorp,
1253                                      HAMMER2_LOOKUP_NODATA);
1254         if (chain) {
1255                 if (chain->bref.type == HAMMER2_BREF_TYPE_INODE) {
1256                         hammer2_inode_data_t *wipdata;
1257
1258                         if (*errorp == 0) {
1259                                 *errorp = hammer2_chain_modify_ip(ip, chain,
1260                                                                   mtid, 0);
1261                         }
1262                         if (*errorp == 0) {
1263                                 wipdata = &chain->data->ipdata;
1264                                 KKASSERT(wipdata->meta.op_flags &
1265                                          HAMMER2_OPFLAG_DIRECTDATA);
1266                                 bzero(wipdata->u.data, HAMMER2_EMBEDDED_BYTES);
1267                                 ++hammer2_iod_file_wembed;
1268                         }
1269                 } else {
1270                         /* chain->error ok for deletion */
1271                         hammer2_chain_delete(*parentp, chain,
1272                                              mtid, HAMMER2_DELETE_PERMANENT);
1273                         ++hammer2_iod_file_wzero;
1274                 }
1275                 hammer2_chain_unlock(chain);
1276                 hammer2_chain_drop(chain);
1277         } else {
1278                 ++hammer2_iod_file_wzero;
1279         }
1280 }
1281
1282 /*
1283  * Helper
1284  *
1285  * Function to write the data as it is, without performing any sort of
1286  * compression. This function is used in path without compression and
1287  * default zero-checking path.
1288  */
1289 static
1290 void
1291 hammer2_write_bp(hammer2_chain_t *chain, char *data, int ioflag,
1292                  int pblksize,
1293                  hammer2_tid_t mtid, int *errorp, int check_algo)
1294 {
1295         hammer2_inode_data_t *wipdata;
1296         hammer2_io_t *dio;
1297         char *bdata;
1298         int error;
1299
1300         error = 0;      /* XXX TODO below */
1301
1302         KKASSERT(chain->flags & HAMMER2_CHAIN_MODIFIED);
1303
1304         switch(chain->bref.type) {
1305         case HAMMER2_BREF_TYPE_INODE:
1306                 wipdata = &chain->data->ipdata;
1307                 KKASSERT(wipdata->meta.op_flags & HAMMER2_OPFLAG_DIRECTDATA);
1308                 bcopy(data, wipdata->u.data, HAMMER2_EMBEDDED_BYTES);
1309                 error = 0;
1310                 ++hammer2_iod_file_wembed;
1311                 break;
1312         case HAMMER2_BREF_TYPE_DATA:
1313                 error = hammer2_io_newnz(chain->hmp,
1314                                          chain->bref.type,
1315                                          chain->bref.data_off,
1316                                          chain->bytes, &dio);
1317                 if (error) {
1318                         hammer2_io_bqrelse(&dio);
1319                         kprintf("hammer2: WRITE PATH: "
1320                                 "dbp bread error\n");
1321                         break;
1322                 }
1323                 bdata = hammer2_io_data(dio, chain->bref.data_off);
1324
1325                 chain->bref.methods = HAMMER2_ENC_COMP(HAMMER2_COMP_NONE) +
1326                                       HAMMER2_ENC_CHECK(check_algo);
1327                 bcopy(data, bdata, chain->bytes);
1328
1329                 /*
1330                  * The flush code doesn't calculate check codes for
1331                  * file data (doing so can result in excessive I/O),
1332                  * so we do it here.
1333                  */
1334                 hammer2_chain_setcheck(chain, bdata);
1335
1336                 /*
1337                  * Device buffer is now valid, chain is no longer in
1338                  * the initial state.
1339                  *
1340                  * (No blockref table worries with file data)
1341                  */
1342                 atomic_clear_int(&chain->flags, HAMMER2_CHAIN_INITIAL);
1343                 hammer2_dedup_record(chain, dio, bdata);
1344
1345                 if (ioflag & IO_SYNC) {
1346                         /*
1347                          * Synchronous I/O requested.
1348                          */
1349                         hammer2_io_bwrite(&dio);
1350                 /*
1351                 } else if ((ioflag & IO_DIRECT) &&
1352                            loff + n == pblksize) {
1353                         hammer2_io_bdwrite(&dio);
1354                 */
1355                 } else if (ioflag & IO_ASYNC) {
1356                         hammer2_io_bawrite(&dio);
1357                 } else {
1358                         hammer2_io_bdwrite(&dio);
1359                 }
1360                 break;
1361         default:
1362                 panic("hammer2_write_bp: bad chain type %d\n",
1363                       chain->bref.type);
1364                 /* NOT REACHED */
1365                 error = 0;
1366                 break;
1367         }
1368         *errorp = error;
1369 }
1370
1371 /*
1372  * LIVE DEDUP HEURISTICS
1373  *
1374  * Record media and crc information for possible dedup operation.  Note
1375  * that the dedup mask bits must also be set in the related DIO for a dedup
1376  * to be fully validated (which is handled in the freemap allocation code).
1377  *
1378  * WARNING! This code is SMP safe but the heuristic allows SMP collisions.
1379  *          All fields must be loaded into locals and validated.
1380  *
1381  * WARNING! Should only be used for file data and directory entries,
1382  *          hammer2_chain_modify() only checks for the dedup case on data
1383  *          chains.  Also, dedup data can only be recorded for committed
1384  *          chains (so NOT strategy writes which can undergo further
1385  *          modification after the fact!).
1386  */
1387 void
1388 hammer2_dedup_record(hammer2_chain_t *chain, hammer2_io_t *dio, char *data)
1389 {
1390         hammer2_dev_t *hmp;
1391         hammer2_dedup_t *dedup;
1392         uint64_t crc;
1393         uint64_t mask;
1394         int best = 0;
1395         int i;
1396         int dticks;
1397
1398         /*
1399          * We can only record a dedup if we have media data to test against.
1400          * If dedup is not enabled, return early, which allows a chain to
1401          * remain marked MODIFIED (which might have benefits in special
1402          * situations, though typically it does not).
1403          */
1404         if (hammer2_dedup_enable == 0)
1405                 return;
1406         if (dio == NULL) {
1407                 dio = chain->dio;
1408                 if (dio == NULL)
1409                         return;
1410         }
1411
1412         hmp = chain->hmp;
1413
1414         switch(HAMMER2_DEC_CHECK(chain->bref.methods)) {
1415         case HAMMER2_CHECK_ISCSI32:
1416                 /*
1417                  * XXX use the built-in crc (the dedup lookup sequencing
1418                  * needs to be fixed so the check code is already present
1419                  * when dedup_lookup is called)
1420                  */
1421 #if 0
1422                 crc = (uint64_t)(uint32_t)chain->bref.check.iscsi32.value;
1423 #endif
1424                 crc = XXH64(data, chain->bytes, XXH_HAMMER2_SEED);
1425                 break;
1426         case HAMMER2_CHECK_XXHASH64:
1427                 crc = chain->bref.check.xxhash64.value;
1428                 break;
1429         case HAMMER2_CHECK_SHA192:
1430                 /*
1431                  * XXX use the built-in crc (the dedup lookup sequencing
1432                  * needs to be fixed so the check code is already present
1433                  * when dedup_lookup is called)
1434                  */
1435 #if 0
1436                 crc = ((uint64_t *)chain->bref.check.sha192.data)[0] ^
1437                       ((uint64_t *)chain->bref.check.sha192.data)[1] ^
1438                       ((uint64_t *)chain->bref.check.sha192.data)[2];
1439 #endif
1440                 crc = XXH64(data, chain->bytes, XXH_HAMMER2_SEED);
1441                 break;
1442         default:
1443                 /*
1444                  * Cannot dedup without a check code
1445                  *
1446                  * NOTE: In particular, CHECK_NONE allows a sector to be
1447                  *       overwritten without copy-on-write, recording
1448                  *       a dedup block for a CHECK_NONE object would be
1449                  *       a disaster!
1450                  */
1451                 return;
1452         }
1453
1454         atomic_set_int(&chain->flags, HAMMER2_CHAIN_DEDUPABLE);
1455
1456         dedup = &hmp->heur_dedup[crc & (HAMMER2_DEDUP_HEUR_MASK & ~3)];
1457         for (i = 0; i < 4; ++i) {
1458                 if (dedup[i].data_crc == crc) {
1459                         best = i;
1460                         break;
1461                 }
1462                 dticks = (int)(dedup[i].ticks - dedup[best].ticks);
1463                 if (dticks < 0 || dticks > hz * 60 * 30)
1464                         best = i;
1465         }
1466         dedup += best;
1467         if (hammer2_debug & 0x40000) {
1468                 kprintf("REC %04x %016jx %016jx\n",
1469                         (int)(dedup - hmp->heur_dedup),
1470                         crc,
1471                         chain->bref.data_off);
1472         }
1473         dedup->ticks = ticks;
1474         dedup->data_off = chain->bref.data_off;
1475         dedup->data_crc = crc;
1476
1477         /*
1478          * Set the valid bits for the dedup only after we know the data
1479          * buffer has been updated.  The alloc bits were set (and the valid
1480          * bits cleared) when the media was allocated.
1481          *
1482          * This is done in two stages becuase the bulkfree code can race
1483          * the gap between allocation and data population.  Both masks must
1484          * be set before a bcmp/dedup operation is able to use the block.
1485          */
1486         mask = hammer2_dedup_mask(dio, chain->bref.data_off, chain->bytes);
1487         atomic_set_64(&dio->dedup_valid, mask);
1488
1489 #if 0
1490         /*
1491          * XXX removed. MODIFIED is an integral part of the flush code,
1492          * lets not just clear it
1493          */
1494         /*
1495          * Once we record the dedup the chain must be marked clean to
1496          * prevent reuse of the underlying block.   Remember that this
1497          * write occurs when the buffer cache is flushed (i.e. on sync(),
1498          * fsync(), filesystem periodic sync, or when the kernel needs to
1499          * flush a buffer), and not whenever the user write()s.
1500          */
1501         if (chain->flags & HAMMER2_CHAIN_MODIFIED) {
1502                 atomic_clear_int(&chain->flags, HAMMER2_CHAIN_MODIFIED);
1503                 atomic_add_long(&hammer2_count_modified_chains, -1);
1504                 if (chain->pmp)
1505                         hammer2_pfs_memory_wakeup(chain->pmp);
1506         }
1507 #endif
1508 }
1509
1510 static
1511 hammer2_off_t
1512 hammer2_dedup_lookup(hammer2_dev_t *hmp, char **datap, int pblksize)
1513 {
1514         hammer2_dedup_t *dedup;
1515         hammer2_io_t *dio;
1516         hammer2_off_t off;
1517         uint64_t crc;
1518         uint64_t mask;
1519         char *data;
1520         char *dtmp;
1521         int i;
1522
1523         if (hammer2_dedup_enable == 0)
1524                 return 0;
1525         data = *datap;
1526         if (data == NULL)
1527                 return 0;
1528
1529         /*
1530          * XXX use the built-in crc (the dedup lookup sequencing
1531          * needs to be fixed so the check code is already present
1532          * when dedup_lookup is called)
1533          */
1534         crc = XXH64(data, pblksize, XXH_HAMMER2_SEED);
1535         dedup = &hmp->heur_dedup[crc & (HAMMER2_DEDUP_HEUR_MASK & ~3)];
1536
1537         if (hammer2_debug & 0x40000) {
1538                 kprintf("LOC %04x/4 %016jx\n",
1539                         (int)(dedup - hmp->heur_dedup),
1540                         crc);
1541         }
1542
1543         for (i = 0; i < 4; ++i) {
1544                 off = dedup[i].data_off;
1545                 cpu_ccfence();
1546                 if (dedup[i].data_crc != crc)
1547                         continue;
1548                 if ((1 << (int)(off & HAMMER2_OFF_MASK_RADIX)) != pblksize)
1549                         continue;
1550                 dio = hammer2_io_getquick(hmp, off, pblksize);
1551                 if (dio) {
1552                         dtmp = hammer2_io_data(dio, off),
1553                         mask = hammer2_dedup_mask(dio, off, pblksize);
1554                         if ((dio->dedup_alloc & mask) == mask &&
1555                             (dio->dedup_valid & mask) == mask &&
1556                             bcmp(data, dtmp, pblksize) == 0) {
1557                                 if (hammer2_debug & 0x40000) {
1558                                         kprintf("DEDUP SUCCESS %016jx\n",
1559                                                 (intmax_t)off);
1560                                 }
1561                                 hammer2_io_putblk(&dio);
1562                                 *datap = NULL;
1563                                 dedup[i].ticks = ticks;   /* update use */
1564                                 atomic_add_long(&hammer2_iod_file_wdedup,
1565                                                 pblksize);
1566
1567                                 return off;             /* RETURN */
1568                         }
1569                         hammer2_io_putblk(&dio);
1570                 }
1571         }
1572         return 0;
1573 }
1574
1575 /*
1576  * Poof.  Races are ok, if someone gets in and reuses a dedup offset
1577  * before or while we are clearing it they will also recover the freemap
1578  * entry (set it to fully allocated), so a bulkfree race can only set it
1579  * to a possibly-free state.
1580  *
1581  * XXX ok, well, not really sure races are ok but going to run with it
1582  *     for the moment.
1583  */
1584 void
1585 hammer2_dedup_clear(hammer2_dev_t *hmp)
1586 {
1587         int i;
1588
1589         for (i = 0; i < HAMMER2_DEDUP_HEUR_SIZE; ++i) {
1590                 hmp->heur_dedup[i].data_off = 0;
1591                 hmp->heur_dedup[i].ticks = ticks - 1;
1592         }
1593 }