14c8ff709SChao Yu // SPDX-License-Identifier: GPL-2.0 24c8ff709SChao Yu /* 34c8ff709SChao Yu * f2fs compress support 44c8ff709SChao Yu * 54c8ff709SChao Yu * Copyright (c) 2019 Chao Yu <chao@kernel.org> 64c8ff709SChao Yu */ 74c8ff709SChao Yu 84c8ff709SChao Yu #include <linux/fs.h> 94c8ff709SChao Yu #include <linux/f2fs_fs.h> 104c8ff709SChao Yu #include <linux/writeback.h> 114c8ff709SChao Yu #include <linux/backing-dev.h> 124c8ff709SChao Yu #include <linux/lzo.h> 134c8ff709SChao Yu #include <linux/lz4.h> 1450cfa66fSChao Yu #include <linux/zstd.h> 156ce19affSChao Yu #include <linux/pagevec.h> 164c8ff709SChao Yu 174c8ff709SChao Yu #include "f2fs.h" 184c8ff709SChao Yu #include "node.h" 196ce19affSChao Yu #include "segment.h" 204c8ff709SChao Yu #include <trace/events/f2fs.h> 214c8ff709SChao Yu 22c68d6c88SChao Yu static struct kmem_cache *cic_entry_slab; 23c68d6c88SChao Yu static struct kmem_cache *dic_entry_slab; 24c68d6c88SChao Yu 2531083031SChao Yu static void *page_array_alloc(struct inode *inode, int nr) 2631083031SChao Yu { 2731083031SChao Yu struct f2fs_sb_info *sbi = F2FS_I_SB(inode); 2831083031SChao Yu unsigned int size = sizeof(struct page *) * nr; 2931083031SChao Yu 3031083031SChao Yu if (likely(size <= sbi->page_array_slab_size)) 31*32410577SChao Yu return f2fs_kmem_cache_alloc(sbi->page_array_slab, 32*32410577SChao Yu GFP_F2FS_ZERO, false, F2FS_I_SB(inode)); 3331083031SChao Yu return f2fs_kzalloc(sbi, size, GFP_NOFS); 3431083031SChao Yu } 3531083031SChao Yu 3631083031SChao Yu static void page_array_free(struct inode *inode, void *pages, int nr) 3731083031SChao Yu { 3831083031SChao Yu struct f2fs_sb_info *sbi = F2FS_I_SB(inode); 3931083031SChao Yu unsigned int size = sizeof(struct page *) * nr; 4031083031SChao Yu 4131083031SChao Yu if (!pages) 4231083031SChao Yu return; 4331083031SChao Yu 4431083031SChao Yu if (likely(size <= sbi->page_array_slab_size)) 4531083031SChao Yu kmem_cache_free(sbi->page_array_slab, pages); 4631083031SChao Yu else 4731083031SChao Yu kfree(pages); 4831083031SChao Yu } 4931083031SChao Yu 504c8ff709SChao Yu struct f2fs_compress_ops { 514c8ff709SChao Yu int (*init_compress_ctx)(struct compress_ctx *cc); 524c8ff709SChao Yu void (*destroy_compress_ctx)(struct compress_ctx *cc); 534c8ff709SChao Yu int (*compress_pages)(struct compress_ctx *cc); 5423b1faaaSChao Yu int (*init_decompress_ctx)(struct decompress_io_ctx *dic); 5523b1faaaSChao Yu void (*destroy_decompress_ctx)(struct decompress_io_ctx *dic); 564c8ff709SChao Yu int (*decompress_pages)(struct decompress_io_ctx *dic); 574c8ff709SChao Yu }; 584c8ff709SChao Yu 594c8ff709SChao Yu static unsigned int offset_in_cluster(struct compress_ctx *cc, pgoff_t index) 604c8ff709SChao Yu { 614c8ff709SChao Yu return index & (cc->cluster_size - 1); 624c8ff709SChao Yu } 634c8ff709SChao Yu 644c8ff709SChao Yu static pgoff_t cluster_idx(struct compress_ctx *cc, pgoff_t index) 654c8ff709SChao Yu { 664c8ff709SChao Yu return index >> cc->log_cluster_size; 674c8ff709SChao Yu } 684c8ff709SChao Yu 694c8ff709SChao Yu static pgoff_t start_idx_of_cluster(struct compress_ctx *cc) 704c8ff709SChao Yu { 714c8ff709SChao Yu return cc->cluster_idx << cc->log_cluster_size; 724c8ff709SChao Yu } 734c8ff709SChao Yu 744c8ff709SChao Yu bool f2fs_is_compressed_page(struct page *page) 754c8ff709SChao Yu { 764c8ff709SChao Yu if (!PagePrivate(page)) 774c8ff709SChao Yu return false; 784c8ff709SChao Yu if (!page_private(page)) 794c8ff709SChao Yu return false; 80b763f3beSChao Yu if (page_private_nonpointer(page)) 814c8ff709SChao Yu return false; 8229b993c7SYu Changchun 834c8ff709SChao Yu f2fs_bug_on(F2FS_M_SB(page->mapping), 844c8ff709SChao Yu *((u32 *)page_private(page)) != F2FS_COMPRESSED_PAGE_MAGIC); 854c8ff709SChao Yu return true; 864c8ff709SChao Yu } 874c8ff709SChao Yu 884c8ff709SChao Yu static void f2fs_set_compressed_page(struct page *page, 89887347a0SChao Yu struct inode *inode, pgoff_t index, void *data) 904c8ff709SChao Yu { 91b763f3beSChao Yu attach_page_private(page, (void *)data); 924c8ff709SChao Yu 934c8ff709SChao Yu /* i_crypto_info and iv index */ 944c8ff709SChao Yu page->index = index; 954c8ff709SChao Yu page->mapping = inode->i_mapping; 964c8ff709SChao Yu } 974c8ff709SChao Yu 984c8ff709SChao Yu static void f2fs_drop_rpages(struct compress_ctx *cc, int len, bool unlock) 994c8ff709SChao Yu { 1004c8ff709SChao Yu int i; 1014c8ff709SChao Yu 1024c8ff709SChao Yu for (i = 0; i < len; i++) { 1034c8ff709SChao Yu if (!cc->rpages[i]) 1044c8ff709SChao Yu continue; 1054c8ff709SChao Yu if (unlock) 1064c8ff709SChao Yu unlock_page(cc->rpages[i]); 1074c8ff709SChao Yu else 1084c8ff709SChao Yu put_page(cc->rpages[i]); 1094c8ff709SChao Yu } 1104c8ff709SChao Yu } 1114c8ff709SChao Yu 1124c8ff709SChao Yu static void f2fs_put_rpages(struct compress_ctx *cc) 1134c8ff709SChao Yu { 1144c8ff709SChao Yu f2fs_drop_rpages(cc, cc->cluster_size, false); 1154c8ff709SChao Yu } 1164c8ff709SChao Yu 1174c8ff709SChao Yu static void f2fs_unlock_rpages(struct compress_ctx *cc, int len) 1184c8ff709SChao Yu { 1194c8ff709SChao Yu f2fs_drop_rpages(cc, len, true); 1204c8ff709SChao Yu } 1214c8ff709SChao Yu 1224c8ff709SChao Yu static void f2fs_put_rpages_wbc(struct compress_ctx *cc, 1234c8ff709SChao Yu struct writeback_control *wbc, bool redirty, int unlock) 1244c8ff709SChao Yu { 1254c8ff709SChao Yu unsigned int i; 1264c8ff709SChao Yu 1274c8ff709SChao Yu for (i = 0; i < cc->cluster_size; i++) { 1284c8ff709SChao Yu if (!cc->rpages[i]) 1294c8ff709SChao Yu continue; 1304c8ff709SChao Yu if (redirty) 1314c8ff709SChao Yu redirty_page_for_writepage(wbc, cc->rpages[i]); 1324c8ff709SChao Yu f2fs_put_page(cc->rpages[i], unlock); 1334c8ff709SChao Yu } 1344c8ff709SChao Yu } 1354c8ff709SChao Yu 1364c8ff709SChao Yu struct page *f2fs_compress_control_page(struct page *page) 1374c8ff709SChao Yu { 1384c8ff709SChao Yu return ((struct compress_io_ctx *)page_private(page))->rpages[0]; 1394c8ff709SChao Yu } 1404c8ff709SChao Yu 1414c8ff709SChao Yu int f2fs_init_compress_ctx(struct compress_ctx *cc) 1424c8ff709SChao Yu { 143adfc6943SJaegeuk Kim if (cc->rpages) 1444c8ff709SChao Yu return 0; 1454c8ff709SChao Yu 14631083031SChao Yu cc->rpages = page_array_alloc(cc->inode, cc->cluster_size); 1474c8ff709SChao Yu return cc->rpages ? 0 : -ENOMEM; 1484c8ff709SChao Yu } 1494c8ff709SChao Yu 1508bfbfb0dSChao Yu void f2fs_destroy_compress_ctx(struct compress_ctx *cc, bool reuse) 1514c8ff709SChao Yu { 15231083031SChao Yu page_array_free(cc->inode, cc->rpages, cc->cluster_size); 1534c8ff709SChao Yu cc->rpages = NULL; 1544c8ff709SChao Yu cc->nr_rpages = 0; 1554c8ff709SChao Yu cc->nr_cpages = 0; 1568bfbfb0dSChao Yu if (!reuse) 1574c8ff709SChao Yu cc->cluster_idx = NULL_CLUSTER; 1584c8ff709SChao Yu } 1594c8ff709SChao Yu 1604c8ff709SChao Yu void f2fs_compress_ctx_add_page(struct compress_ctx *cc, struct page *page) 1614c8ff709SChao Yu { 1624c8ff709SChao Yu unsigned int cluster_ofs; 1634c8ff709SChao Yu 1644c8ff709SChao Yu if (!f2fs_cluster_can_merge_page(cc, page->index)) 1654c8ff709SChao Yu f2fs_bug_on(F2FS_I_SB(cc->inode), 1); 1664c8ff709SChao Yu 1674c8ff709SChao Yu cluster_ofs = offset_in_cluster(cc, page->index); 1684c8ff709SChao Yu cc->rpages[cluster_ofs] = page; 1694c8ff709SChao Yu cc->nr_rpages++; 1704c8ff709SChao Yu cc->cluster_idx = cluster_idx(cc, page->index); 1714c8ff709SChao Yu } 1724c8ff709SChao Yu 1734c8ff709SChao Yu #ifdef CONFIG_F2FS_FS_LZO 1744c8ff709SChao Yu static int lzo_init_compress_ctx(struct compress_ctx *cc) 1754c8ff709SChao Yu { 1764c8ff709SChao Yu cc->private = f2fs_kvmalloc(F2FS_I_SB(cc->inode), 1774c8ff709SChao Yu LZO1X_MEM_COMPRESS, GFP_NOFS); 1784c8ff709SChao Yu if (!cc->private) 1794c8ff709SChao Yu return -ENOMEM; 1804c8ff709SChao Yu 1814c8ff709SChao Yu cc->clen = lzo1x_worst_compress(PAGE_SIZE << cc->log_cluster_size); 1824c8ff709SChao Yu return 0; 1834c8ff709SChao Yu } 1844c8ff709SChao Yu 1854c8ff709SChao Yu static void lzo_destroy_compress_ctx(struct compress_ctx *cc) 1864c8ff709SChao Yu { 1874c8ff709SChao Yu kvfree(cc->private); 1884c8ff709SChao Yu cc->private = NULL; 1894c8ff709SChao Yu } 1904c8ff709SChao Yu 1914c8ff709SChao Yu static int lzo_compress_pages(struct compress_ctx *cc) 1924c8ff709SChao Yu { 1934c8ff709SChao Yu int ret; 1944c8ff709SChao Yu 1954c8ff709SChao Yu ret = lzo1x_1_compress(cc->rbuf, cc->rlen, cc->cbuf->cdata, 1964c8ff709SChao Yu &cc->clen, cc->private); 1974c8ff709SChao Yu if (ret != LZO_E_OK) { 1984c8ff709SChao Yu printk_ratelimited("%sF2FS-fs (%s): lzo compress failed, ret:%d\n", 1994c8ff709SChao Yu KERN_ERR, F2FS_I_SB(cc->inode)->sb->s_id, ret); 2004c8ff709SChao Yu return -EIO; 2014c8ff709SChao Yu } 2024c8ff709SChao Yu return 0; 2034c8ff709SChao Yu } 2044c8ff709SChao Yu 2054c8ff709SChao Yu static int lzo_decompress_pages(struct decompress_io_ctx *dic) 2064c8ff709SChao Yu { 2074c8ff709SChao Yu int ret; 2084c8ff709SChao Yu 2094c8ff709SChao Yu ret = lzo1x_decompress_safe(dic->cbuf->cdata, dic->clen, 2104c8ff709SChao Yu dic->rbuf, &dic->rlen); 2114c8ff709SChao Yu if (ret != LZO_E_OK) { 2124c8ff709SChao Yu printk_ratelimited("%sF2FS-fs (%s): lzo decompress failed, ret:%d\n", 2134c8ff709SChao Yu KERN_ERR, F2FS_I_SB(dic->inode)->sb->s_id, ret); 2144c8ff709SChao Yu return -EIO; 2154c8ff709SChao Yu } 2164c8ff709SChao Yu 2174c8ff709SChao Yu if (dic->rlen != PAGE_SIZE << dic->log_cluster_size) { 2184c8ff709SChao Yu printk_ratelimited("%sF2FS-fs (%s): lzo invalid rlen:%zu, " 2194c8ff709SChao Yu "expected:%lu\n", KERN_ERR, 2204c8ff709SChao Yu F2FS_I_SB(dic->inode)->sb->s_id, 2214c8ff709SChao Yu dic->rlen, 2224c8ff709SChao Yu PAGE_SIZE << dic->log_cluster_size); 2234c8ff709SChao Yu return -EIO; 2244c8ff709SChao Yu } 2254c8ff709SChao Yu return 0; 2264c8ff709SChao Yu } 2274c8ff709SChao Yu 2284c8ff709SChao Yu static const struct f2fs_compress_ops f2fs_lzo_ops = { 2294c8ff709SChao Yu .init_compress_ctx = lzo_init_compress_ctx, 2304c8ff709SChao Yu .destroy_compress_ctx = lzo_destroy_compress_ctx, 2314c8ff709SChao Yu .compress_pages = lzo_compress_pages, 2324c8ff709SChao Yu .decompress_pages = lzo_decompress_pages, 2334c8ff709SChao Yu }; 2344c8ff709SChao Yu #endif 2354c8ff709SChao Yu 2364c8ff709SChao Yu #ifdef CONFIG_F2FS_FS_LZ4 2374c8ff709SChao Yu static int lz4_init_compress_ctx(struct compress_ctx *cc) 2384c8ff709SChao Yu { 2393fde13f8SChao Yu unsigned int size = LZ4_MEM_COMPRESS; 2403fde13f8SChao Yu 2413fde13f8SChao Yu #ifdef CONFIG_F2FS_FS_LZ4HC 2423fde13f8SChao Yu if (F2FS_I(cc->inode)->i_compress_flag >> COMPRESS_LEVEL_OFFSET) 2433fde13f8SChao Yu size = LZ4HC_MEM_COMPRESS; 2443fde13f8SChao Yu #endif 2453fde13f8SChao Yu 2463fde13f8SChao Yu cc->private = f2fs_kvmalloc(F2FS_I_SB(cc->inode), size, GFP_NOFS); 2474c8ff709SChao Yu if (!cc->private) 2484c8ff709SChao Yu return -ENOMEM; 2494c8ff709SChao Yu 250f6644143SChao Yu /* 251f6644143SChao Yu * we do not change cc->clen to LZ4_compressBound(inputsize) to 252f6644143SChao Yu * adapt worst compress case, because lz4 compressor can handle 253f6644143SChao Yu * output budget properly. 254f6644143SChao Yu */ 255f6644143SChao Yu cc->clen = cc->rlen - PAGE_SIZE - COMPRESS_HEADER_SIZE; 2564c8ff709SChao Yu return 0; 2574c8ff709SChao Yu } 2584c8ff709SChao Yu 2594c8ff709SChao Yu static void lz4_destroy_compress_ctx(struct compress_ctx *cc) 2604c8ff709SChao Yu { 2614c8ff709SChao Yu kvfree(cc->private); 2624c8ff709SChao Yu cc->private = NULL; 2634c8ff709SChao Yu } 2644c8ff709SChao Yu 2653fde13f8SChao Yu #ifdef CONFIG_F2FS_FS_LZ4HC 2663fde13f8SChao Yu static int lz4hc_compress_pages(struct compress_ctx *cc) 2673fde13f8SChao Yu { 2683fde13f8SChao Yu unsigned char level = F2FS_I(cc->inode)->i_compress_flag >> 2693fde13f8SChao Yu COMPRESS_LEVEL_OFFSET; 2703fde13f8SChao Yu int len; 2713fde13f8SChao Yu 2723fde13f8SChao Yu if (level) 2733fde13f8SChao Yu len = LZ4_compress_HC(cc->rbuf, cc->cbuf->cdata, cc->rlen, 2743fde13f8SChao Yu cc->clen, level, cc->private); 2753fde13f8SChao Yu else 2763fde13f8SChao Yu len = LZ4_compress_default(cc->rbuf, cc->cbuf->cdata, cc->rlen, 2773fde13f8SChao Yu cc->clen, cc->private); 2783fde13f8SChao Yu if (!len) 2793fde13f8SChao Yu return -EAGAIN; 2803fde13f8SChao Yu 2813fde13f8SChao Yu cc->clen = len; 2823fde13f8SChao Yu return 0; 2833fde13f8SChao Yu } 2843fde13f8SChao Yu #endif 2853fde13f8SChao Yu 2864c8ff709SChao Yu static int lz4_compress_pages(struct compress_ctx *cc) 2874c8ff709SChao Yu { 2884c8ff709SChao Yu int len; 2894c8ff709SChao Yu 2903fde13f8SChao Yu #ifdef CONFIG_F2FS_FS_LZ4HC 2913fde13f8SChao Yu return lz4hc_compress_pages(cc); 2923fde13f8SChao Yu #endif 2934c8ff709SChao Yu len = LZ4_compress_default(cc->rbuf, cc->cbuf->cdata, cc->rlen, 2944c8ff709SChao Yu cc->clen, cc->private); 295f6644143SChao Yu if (!len) 296f6644143SChao Yu return -EAGAIN; 297f6644143SChao Yu 2984c8ff709SChao Yu cc->clen = len; 2994c8ff709SChao Yu return 0; 3004c8ff709SChao Yu } 3014c8ff709SChao Yu 3024c8ff709SChao Yu static int lz4_decompress_pages(struct decompress_io_ctx *dic) 3034c8ff709SChao Yu { 3044c8ff709SChao Yu int ret; 3054c8ff709SChao Yu 3064c8ff709SChao Yu ret = LZ4_decompress_safe(dic->cbuf->cdata, dic->rbuf, 3074c8ff709SChao Yu dic->clen, dic->rlen); 3084c8ff709SChao Yu if (ret < 0) { 3094c8ff709SChao Yu printk_ratelimited("%sF2FS-fs (%s): lz4 decompress failed, ret:%d\n", 3104c8ff709SChao Yu KERN_ERR, F2FS_I_SB(dic->inode)->sb->s_id, ret); 3114c8ff709SChao Yu return -EIO; 3124c8ff709SChao Yu } 3134c8ff709SChao Yu 3144c8ff709SChao Yu if (ret != PAGE_SIZE << dic->log_cluster_size) { 3154c8ff709SChao Yu printk_ratelimited("%sF2FS-fs (%s): lz4 invalid rlen:%zu, " 3164c8ff709SChao Yu "expected:%lu\n", KERN_ERR, 3174c8ff709SChao Yu F2FS_I_SB(dic->inode)->sb->s_id, 3184c8ff709SChao Yu dic->rlen, 3194c8ff709SChao Yu PAGE_SIZE << dic->log_cluster_size); 3204c8ff709SChao Yu return -EIO; 3214c8ff709SChao Yu } 3224c8ff709SChao Yu return 0; 3234c8ff709SChao Yu } 3244c8ff709SChao Yu 3254c8ff709SChao Yu static const struct f2fs_compress_ops f2fs_lz4_ops = { 3264c8ff709SChao Yu .init_compress_ctx = lz4_init_compress_ctx, 3274c8ff709SChao Yu .destroy_compress_ctx = lz4_destroy_compress_ctx, 3284c8ff709SChao Yu .compress_pages = lz4_compress_pages, 3294c8ff709SChao Yu .decompress_pages = lz4_decompress_pages, 3304c8ff709SChao Yu }; 3314c8ff709SChao Yu #endif 3324c8ff709SChao Yu 33350cfa66fSChao Yu #ifdef CONFIG_F2FS_FS_ZSTD 33450cfa66fSChao Yu #define F2FS_ZSTD_DEFAULT_CLEVEL 1 33550cfa66fSChao Yu 33650cfa66fSChao Yu static int zstd_init_compress_ctx(struct compress_ctx *cc) 33750cfa66fSChao Yu { 33850cfa66fSChao Yu ZSTD_parameters params; 33950cfa66fSChao Yu ZSTD_CStream *stream; 34050cfa66fSChao Yu void *workspace; 34150cfa66fSChao Yu unsigned int workspace_size; 3423fde13f8SChao Yu unsigned char level = F2FS_I(cc->inode)->i_compress_flag >> 3433fde13f8SChao Yu COMPRESS_LEVEL_OFFSET; 34450cfa66fSChao Yu 3453fde13f8SChao Yu if (!level) 3463fde13f8SChao Yu level = F2FS_ZSTD_DEFAULT_CLEVEL; 3473fde13f8SChao Yu 3483fde13f8SChao Yu params = ZSTD_getParams(level, cc->rlen, 0); 34950cfa66fSChao Yu workspace_size = ZSTD_CStreamWorkspaceBound(params.cParams); 35050cfa66fSChao Yu 35150cfa66fSChao Yu workspace = f2fs_kvmalloc(F2FS_I_SB(cc->inode), 35250cfa66fSChao Yu workspace_size, GFP_NOFS); 35350cfa66fSChao Yu if (!workspace) 35450cfa66fSChao Yu return -ENOMEM; 35550cfa66fSChao Yu 35650cfa66fSChao Yu stream = ZSTD_initCStream(params, 0, workspace, workspace_size); 35750cfa66fSChao Yu if (!stream) { 35850cfa66fSChao Yu printk_ratelimited("%sF2FS-fs (%s): %s ZSTD_initCStream failed\n", 35950cfa66fSChao Yu KERN_ERR, F2FS_I_SB(cc->inode)->sb->s_id, 36050cfa66fSChao Yu __func__); 36150cfa66fSChao Yu kvfree(workspace); 36250cfa66fSChao Yu return -EIO; 36350cfa66fSChao Yu } 36450cfa66fSChao Yu 36550cfa66fSChao Yu cc->private = workspace; 36650cfa66fSChao Yu cc->private2 = stream; 36750cfa66fSChao Yu 36850cfa66fSChao Yu cc->clen = cc->rlen - PAGE_SIZE - COMPRESS_HEADER_SIZE; 36950cfa66fSChao Yu return 0; 37050cfa66fSChao Yu } 37150cfa66fSChao Yu 37250cfa66fSChao Yu static void zstd_destroy_compress_ctx(struct compress_ctx *cc) 37350cfa66fSChao Yu { 37450cfa66fSChao Yu kvfree(cc->private); 37550cfa66fSChao Yu cc->private = NULL; 37650cfa66fSChao Yu cc->private2 = NULL; 37750cfa66fSChao Yu } 37850cfa66fSChao Yu 37950cfa66fSChao Yu static int zstd_compress_pages(struct compress_ctx *cc) 38050cfa66fSChao Yu { 38150cfa66fSChao Yu ZSTD_CStream *stream = cc->private2; 38250cfa66fSChao Yu ZSTD_inBuffer inbuf; 38350cfa66fSChao Yu ZSTD_outBuffer outbuf; 38450cfa66fSChao Yu int src_size = cc->rlen; 38550cfa66fSChao Yu int dst_size = src_size - PAGE_SIZE - COMPRESS_HEADER_SIZE; 38650cfa66fSChao Yu int ret; 38750cfa66fSChao Yu 38850cfa66fSChao Yu inbuf.pos = 0; 38950cfa66fSChao Yu inbuf.src = cc->rbuf; 39050cfa66fSChao Yu inbuf.size = src_size; 39150cfa66fSChao Yu 39250cfa66fSChao Yu outbuf.pos = 0; 39350cfa66fSChao Yu outbuf.dst = cc->cbuf->cdata; 39450cfa66fSChao Yu outbuf.size = dst_size; 39550cfa66fSChao Yu 39650cfa66fSChao Yu ret = ZSTD_compressStream(stream, &outbuf, &inbuf); 39750cfa66fSChao Yu if (ZSTD_isError(ret)) { 39850cfa66fSChao Yu printk_ratelimited("%sF2FS-fs (%s): %s ZSTD_compressStream failed, ret: %d\n", 39950cfa66fSChao Yu KERN_ERR, F2FS_I_SB(cc->inode)->sb->s_id, 40050cfa66fSChao Yu __func__, ZSTD_getErrorCode(ret)); 40150cfa66fSChao Yu return -EIO; 40250cfa66fSChao Yu } 40350cfa66fSChao Yu 40450cfa66fSChao Yu ret = ZSTD_endStream(stream, &outbuf); 40550cfa66fSChao Yu if (ZSTD_isError(ret)) { 40650cfa66fSChao Yu printk_ratelimited("%sF2FS-fs (%s): %s ZSTD_endStream returned %d\n", 40750cfa66fSChao Yu KERN_ERR, F2FS_I_SB(cc->inode)->sb->s_id, 40850cfa66fSChao Yu __func__, ZSTD_getErrorCode(ret)); 40950cfa66fSChao Yu return -EIO; 41050cfa66fSChao Yu } 41150cfa66fSChao Yu 4121454c978SChao Yu /* 4131454c978SChao Yu * there is compressed data remained in intermediate buffer due to 4141454c978SChao Yu * no more space in cbuf.cdata 4151454c978SChao Yu */ 4161454c978SChao Yu if (ret) 4171454c978SChao Yu return -EAGAIN; 4181454c978SChao Yu 41950cfa66fSChao Yu cc->clen = outbuf.pos; 42050cfa66fSChao Yu return 0; 42150cfa66fSChao Yu } 42250cfa66fSChao Yu 42350cfa66fSChao Yu static int zstd_init_decompress_ctx(struct decompress_io_ctx *dic) 42450cfa66fSChao Yu { 42550cfa66fSChao Yu ZSTD_DStream *stream; 42650cfa66fSChao Yu void *workspace; 42750cfa66fSChao Yu unsigned int workspace_size; 4280e2b7385SChao Yu unsigned int max_window_size = 4290e2b7385SChao Yu MAX_COMPRESS_WINDOW_SIZE(dic->log_cluster_size); 43050cfa66fSChao Yu 4310e2b7385SChao Yu workspace_size = ZSTD_DStreamWorkspaceBound(max_window_size); 43250cfa66fSChao Yu 43350cfa66fSChao Yu workspace = f2fs_kvmalloc(F2FS_I_SB(dic->inode), 43450cfa66fSChao Yu workspace_size, GFP_NOFS); 43550cfa66fSChao Yu if (!workspace) 43650cfa66fSChao Yu return -ENOMEM; 43750cfa66fSChao Yu 4380e2b7385SChao Yu stream = ZSTD_initDStream(max_window_size, workspace, workspace_size); 43950cfa66fSChao Yu if (!stream) { 44050cfa66fSChao Yu printk_ratelimited("%sF2FS-fs (%s): %s ZSTD_initDStream failed\n", 44150cfa66fSChao Yu KERN_ERR, F2FS_I_SB(dic->inode)->sb->s_id, 44250cfa66fSChao Yu __func__); 44350cfa66fSChao Yu kvfree(workspace); 44450cfa66fSChao Yu return -EIO; 44550cfa66fSChao Yu } 44650cfa66fSChao Yu 44750cfa66fSChao Yu dic->private = workspace; 44850cfa66fSChao Yu dic->private2 = stream; 44950cfa66fSChao Yu 45050cfa66fSChao Yu return 0; 45150cfa66fSChao Yu } 45250cfa66fSChao Yu 45350cfa66fSChao Yu static void zstd_destroy_decompress_ctx(struct decompress_io_ctx *dic) 45450cfa66fSChao Yu { 45550cfa66fSChao Yu kvfree(dic->private); 45650cfa66fSChao Yu dic->private = NULL; 45750cfa66fSChao Yu dic->private2 = NULL; 45850cfa66fSChao Yu } 45950cfa66fSChao Yu 46050cfa66fSChao Yu static int zstd_decompress_pages(struct decompress_io_ctx *dic) 46150cfa66fSChao Yu { 46250cfa66fSChao Yu ZSTD_DStream *stream = dic->private2; 46350cfa66fSChao Yu ZSTD_inBuffer inbuf; 46450cfa66fSChao Yu ZSTD_outBuffer outbuf; 46550cfa66fSChao Yu int ret; 46650cfa66fSChao Yu 46750cfa66fSChao Yu inbuf.pos = 0; 46850cfa66fSChao Yu inbuf.src = dic->cbuf->cdata; 46950cfa66fSChao Yu inbuf.size = dic->clen; 47050cfa66fSChao Yu 47150cfa66fSChao Yu outbuf.pos = 0; 47250cfa66fSChao Yu outbuf.dst = dic->rbuf; 47350cfa66fSChao Yu outbuf.size = dic->rlen; 47450cfa66fSChao Yu 47550cfa66fSChao Yu ret = ZSTD_decompressStream(stream, &outbuf, &inbuf); 47650cfa66fSChao Yu if (ZSTD_isError(ret)) { 47750cfa66fSChao Yu printk_ratelimited("%sF2FS-fs (%s): %s ZSTD_compressStream failed, ret: %d\n", 47850cfa66fSChao Yu KERN_ERR, F2FS_I_SB(dic->inode)->sb->s_id, 47950cfa66fSChao Yu __func__, ZSTD_getErrorCode(ret)); 48050cfa66fSChao Yu return -EIO; 48150cfa66fSChao Yu } 48250cfa66fSChao Yu 48350cfa66fSChao Yu if (dic->rlen != outbuf.pos) { 48450cfa66fSChao Yu printk_ratelimited("%sF2FS-fs (%s): %s ZSTD invalid rlen:%zu, " 48550cfa66fSChao Yu "expected:%lu\n", KERN_ERR, 48650cfa66fSChao Yu F2FS_I_SB(dic->inode)->sb->s_id, 48750cfa66fSChao Yu __func__, dic->rlen, 48850cfa66fSChao Yu PAGE_SIZE << dic->log_cluster_size); 48950cfa66fSChao Yu return -EIO; 49050cfa66fSChao Yu } 49150cfa66fSChao Yu 49250cfa66fSChao Yu return 0; 49350cfa66fSChao Yu } 49450cfa66fSChao Yu 49550cfa66fSChao Yu static const struct f2fs_compress_ops f2fs_zstd_ops = { 49650cfa66fSChao Yu .init_compress_ctx = zstd_init_compress_ctx, 49750cfa66fSChao Yu .destroy_compress_ctx = zstd_destroy_compress_ctx, 49850cfa66fSChao Yu .compress_pages = zstd_compress_pages, 49950cfa66fSChao Yu .init_decompress_ctx = zstd_init_decompress_ctx, 50050cfa66fSChao Yu .destroy_decompress_ctx = zstd_destroy_decompress_ctx, 50150cfa66fSChao Yu .decompress_pages = zstd_decompress_pages, 50250cfa66fSChao Yu }; 50350cfa66fSChao Yu #endif 50450cfa66fSChao Yu 5056d92b201SChao Yu #ifdef CONFIG_F2FS_FS_LZO 5066d92b201SChao Yu #ifdef CONFIG_F2FS_FS_LZORLE 5076d92b201SChao Yu static int lzorle_compress_pages(struct compress_ctx *cc) 5086d92b201SChao Yu { 5096d92b201SChao Yu int ret; 5106d92b201SChao Yu 5116d92b201SChao Yu ret = lzorle1x_1_compress(cc->rbuf, cc->rlen, cc->cbuf->cdata, 5126d92b201SChao Yu &cc->clen, cc->private); 5136d92b201SChao Yu if (ret != LZO_E_OK) { 5146d92b201SChao Yu printk_ratelimited("%sF2FS-fs (%s): lzo-rle compress failed, ret:%d\n", 5156d92b201SChao Yu KERN_ERR, F2FS_I_SB(cc->inode)->sb->s_id, ret); 5166d92b201SChao Yu return -EIO; 5176d92b201SChao Yu } 5186d92b201SChao Yu return 0; 5196d92b201SChao Yu } 5206d92b201SChao Yu 5216d92b201SChao Yu static const struct f2fs_compress_ops f2fs_lzorle_ops = { 5226d92b201SChao Yu .init_compress_ctx = lzo_init_compress_ctx, 5236d92b201SChao Yu .destroy_compress_ctx = lzo_destroy_compress_ctx, 5246d92b201SChao Yu .compress_pages = lzorle_compress_pages, 5256d92b201SChao Yu .decompress_pages = lzo_decompress_pages, 5266d92b201SChao Yu }; 5276d92b201SChao Yu #endif 5286d92b201SChao Yu #endif 5296d92b201SChao Yu 5304c8ff709SChao Yu static const struct f2fs_compress_ops *f2fs_cops[COMPRESS_MAX] = { 5314c8ff709SChao Yu #ifdef CONFIG_F2FS_FS_LZO 5324c8ff709SChao Yu &f2fs_lzo_ops, 5334c8ff709SChao Yu #else 5344c8ff709SChao Yu NULL, 5354c8ff709SChao Yu #endif 5364c8ff709SChao Yu #ifdef CONFIG_F2FS_FS_LZ4 5374c8ff709SChao Yu &f2fs_lz4_ops, 5384c8ff709SChao Yu #else 5394c8ff709SChao Yu NULL, 5404c8ff709SChao Yu #endif 54150cfa66fSChao Yu #ifdef CONFIG_F2FS_FS_ZSTD 54250cfa66fSChao Yu &f2fs_zstd_ops, 54350cfa66fSChao Yu #else 54450cfa66fSChao Yu NULL, 54550cfa66fSChao Yu #endif 5466d92b201SChao Yu #if defined(CONFIG_F2FS_FS_LZO) && defined(CONFIG_F2FS_FS_LZORLE) 5476d92b201SChao Yu &f2fs_lzorle_ops, 5486d92b201SChao Yu #else 5496d92b201SChao Yu NULL, 5506d92b201SChao Yu #endif 5514c8ff709SChao Yu }; 5524c8ff709SChao Yu 5534c8ff709SChao Yu bool f2fs_is_compress_backend_ready(struct inode *inode) 5544c8ff709SChao Yu { 5554c8ff709SChao Yu if (!f2fs_compressed_file(inode)) 5564c8ff709SChao Yu return true; 5574c8ff709SChao Yu return f2fs_cops[F2FS_I(inode)->i_compress_algorithm]; 5584c8ff709SChao Yu } 5594c8ff709SChao Yu 56099bbe307SJaegeuk Kim static mempool_t *compress_page_pool; 5615e6bbde9SChao Yu static int num_compress_pages = 512; 5625e6bbde9SChao Yu module_param(num_compress_pages, uint, 0444); 5635e6bbde9SChao Yu MODULE_PARM_DESC(num_compress_pages, 5645e6bbde9SChao Yu "Number of intermediate compress pages to preallocate"); 5655e6bbde9SChao Yu 5665e6bbde9SChao Yu int f2fs_init_compress_mempool(void) 5675e6bbde9SChao Yu { 5685e6bbde9SChao Yu compress_page_pool = mempool_create_page_pool(num_compress_pages, 0); 5695e6bbde9SChao Yu if (!compress_page_pool) 5705e6bbde9SChao Yu return -ENOMEM; 5715e6bbde9SChao Yu 5725e6bbde9SChao Yu return 0; 5735e6bbde9SChao Yu } 5745e6bbde9SChao Yu 5755e6bbde9SChao Yu void f2fs_destroy_compress_mempool(void) 5765e6bbde9SChao Yu { 5775e6bbde9SChao Yu mempool_destroy(compress_page_pool); 5785e6bbde9SChao Yu } 5795e6bbde9SChao Yu 5805e6bbde9SChao Yu static struct page *f2fs_compress_alloc_page(void) 5814c8ff709SChao Yu { 5824c8ff709SChao Yu struct page *page; 5834c8ff709SChao Yu 5845e6bbde9SChao Yu page = mempool_alloc(compress_page_pool, GFP_NOFS); 5854c8ff709SChao Yu lock_page(page); 5865e6bbde9SChao Yu 5874c8ff709SChao Yu return page; 5884c8ff709SChao Yu } 5894c8ff709SChao Yu 5905e6bbde9SChao Yu static void f2fs_compress_free_page(struct page *page) 5915e6bbde9SChao Yu { 5925e6bbde9SChao Yu if (!page) 5935e6bbde9SChao Yu return; 594b763f3beSChao Yu detach_page_private(page); 5955e6bbde9SChao Yu page->mapping = NULL; 5965e6bbde9SChao Yu unlock_page(page); 5975e6bbde9SChao Yu mempool_free(page, compress_page_pool); 5985e6bbde9SChao Yu } 5995e6bbde9SChao Yu 6006fcaebacSDaeho Jeong #define MAX_VMAP_RETRIES 3 6016fcaebacSDaeho Jeong 6026fcaebacSDaeho Jeong static void *f2fs_vmap(struct page **pages, unsigned int count) 6036fcaebacSDaeho Jeong { 6046fcaebacSDaeho Jeong int i; 6056fcaebacSDaeho Jeong void *buf = NULL; 6066fcaebacSDaeho Jeong 6076fcaebacSDaeho Jeong for (i = 0; i < MAX_VMAP_RETRIES; i++) { 6086fcaebacSDaeho Jeong buf = vm_map_ram(pages, count, -1); 6096fcaebacSDaeho Jeong if (buf) 6106fcaebacSDaeho Jeong break; 6116fcaebacSDaeho Jeong vm_unmap_aliases(); 6126fcaebacSDaeho Jeong } 6136fcaebacSDaeho Jeong return buf; 6146fcaebacSDaeho Jeong } 6156fcaebacSDaeho Jeong 6164c8ff709SChao Yu static int f2fs_compress_pages(struct compress_ctx *cc) 6174c8ff709SChao Yu { 6184c8ff709SChao Yu struct f2fs_inode_info *fi = F2FS_I(cc->inode); 6194c8ff709SChao Yu const struct f2fs_compress_ops *cops = 6204c8ff709SChao Yu f2fs_cops[fi->i_compress_algorithm]; 62131083031SChao Yu unsigned int max_len, new_nr_cpages; 62231083031SChao Yu struct page **new_cpages; 623b28f047bSChao Yu u32 chksum = 0; 6244c8ff709SChao Yu int i, ret; 6254c8ff709SChao Yu 6264c8ff709SChao Yu trace_f2fs_compress_pages_start(cc->inode, cc->cluster_idx, 6274c8ff709SChao Yu cc->cluster_size, fi->i_compress_algorithm); 6284c8ff709SChao Yu 62923b1faaaSChao Yu if (cops->init_compress_ctx) { 6304c8ff709SChao Yu ret = cops->init_compress_ctx(cc); 6314c8ff709SChao Yu if (ret) 6324c8ff709SChao Yu goto out; 63323b1faaaSChao Yu } 6344c8ff709SChao Yu 6354c8ff709SChao Yu max_len = COMPRESS_HEADER_SIZE + cc->clen; 6364c8ff709SChao Yu cc->nr_cpages = DIV_ROUND_UP(max_len, PAGE_SIZE); 6374c8ff709SChao Yu 63831083031SChao Yu cc->cpages = page_array_alloc(cc->inode, cc->nr_cpages); 6394c8ff709SChao Yu if (!cc->cpages) { 6404c8ff709SChao Yu ret = -ENOMEM; 6414c8ff709SChao Yu goto destroy_compress_ctx; 6424c8ff709SChao Yu } 6434c8ff709SChao Yu 6444c8ff709SChao Yu for (i = 0; i < cc->nr_cpages; i++) { 6455e6bbde9SChao Yu cc->cpages[i] = f2fs_compress_alloc_page(); 6464c8ff709SChao Yu if (!cc->cpages[i]) { 6474c8ff709SChao Yu ret = -ENOMEM; 6484c8ff709SChao Yu goto out_free_cpages; 6494c8ff709SChao Yu } 6504c8ff709SChao Yu } 6514c8ff709SChao Yu 6526fcaebacSDaeho Jeong cc->rbuf = f2fs_vmap(cc->rpages, cc->cluster_size); 6534c8ff709SChao Yu if (!cc->rbuf) { 6544c8ff709SChao Yu ret = -ENOMEM; 6554c8ff709SChao Yu goto out_free_cpages; 6564c8ff709SChao Yu } 6574c8ff709SChao Yu 6586fcaebacSDaeho Jeong cc->cbuf = f2fs_vmap(cc->cpages, cc->nr_cpages); 6594c8ff709SChao Yu if (!cc->cbuf) { 6604c8ff709SChao Yu ret = -ENOMEM; 6614c8ff709SChao Yu goto out_vunmap_rbuf; 6624c8ff709SChao Yu } 6634c8ff709SChao Yu 6644c8ff709SChao Yu ret = cops->compress_pages(cc); 6654c8ff709SChao Yu if (ret) 6664c8ff709SChao Yu goto out_vunmap_cbuf; 6674c8ff709SChao Yu 6684c8ff709SChao Yu max_len = PAGE_SIZE * (cc->cluster_size - 1) - COMPRESS_HEADER_SIZE; 6694c8ff709SChao Yu 6704c8ff709SChao Yu if (cc->clen > max_len) { 6714c8ff709SChao Yu ret = -EAGAIN; 6724c8ff709SChao Yu goto out_vunmap_cbuf; 6734c8ff709SChao Yu } 6744c8ff709SChao Yu 6754c8ff709SChao Yu cc->cbuf->clen = cpu_to_le32(cc->clen); 6764c8ff709SChao Yu 677b28f047bSChao Yu if (fi->i_compress_flag & 1 << COMPRESS_CHKSUM) 678b28f047bSChao Yu chksum = f2fs_crc32(F2FS_I_SB(cc->inode), 679b28f047bSChao Yu cc->cbuf->cdata, cc->clen); 680b28f047bSChao Yu cc->cbuf->chksum = cpu_to_le32(chksum); 681b28f047bSChao Yu 6824c8ff709SChao Yu for (i = 0; i < COMPRESS_DATA_RESERVED_SIZE; i++) 6834c8ff709SChao Yu cc->cbuf->reserved[i] = cpu_to_le32(0); 6844c8ff709SChao Yu 68531083031SChao Yu new_nr_cpages = DIV_ROUND_UP(cc->clen + COMPRESS_HEADER_SIZE, PAGE_SIZE); 68631083031SChao Yu 68731083031SChao Yu /* Now we're going to cut unnecessary tail pages */ 68831083031SChao Yu new_cpages = page_array_alloc(cc->inode, new_nr_cpages); 68931083031SChao Yu if (!new_cpages) { 69031083031SChao Yu ret = -ENOMEM; 69131083031SChao Yu goto out_vunmap_cbuf; 69231083031SChao Yu } 6937fa6d598SEric Biggers 6947fa6d598SEric Biggers /* zero out any unused part of the last page */ 6957fa6d598SEric Biggers memset(&cc->cbuf->cdata[cc->clen], 0, 69631083031SChao Yu (new_nr_cpages * PAGE_SIZE) - 69731083031SChao Yu (cc->clen + COMPRESS_HEADER_SIZE)); 6987fa6d598SEric Biggers 6996fcaebacSDaeho Jeong vm_unmap_ram(cc->cbuf, cc->nr_cpages); 7006fcaebacSDaeho Jeong vm_unmap_ram(cc->rbuf, cc->cluster_size); 7014c8ff709SChao Yu 70231083031SChao Yu for (i = 0; i < cc->nr_cpages; i++) { 70331083031SChao Yu if (i < new_nr_cpages) { 70431083031SChao Yu new_cpages[i] = cc->cpages[i]; 70531083031SChao Yu continue; 70631083031SChao Yu } 7075e6bbde9SChao Yu f2fs_compress_free_page(cc->cpages[i]); 7084c8ff709SChao Yu cc->cpages[i] = NULL; 7094c8ff709SChao Yu } 7104c8ff709SChao Yu 71123b1faaaSChao Yu if (cops->destroy_compress_ctx) 71209ff4801SChao Yu cops->destroy_compress_ctx(cc); 71309ff4801SChao Yu 71431083031SChao Yu page_array_free(cc->inode, cc->cpages, cc->nr_cpages); 71531083031SChao Yu cc->cpages = new_cpages; 71631083031SChao Yu cc->nr_cpages = new_nr_cpages; 7174c8ff709SChao Yu 7184c8ff709SChao Yu trace_f2fs_compress_pages_end(cc->inode, cc->cluster_idx, 7194c8ff709SChao Yu cc->clen, ret); 7204c8ff709SChao Yu return 0; 7214c8ff709SChao Yu 7224c8ff709SChao Yu out_vunmap_cbuf: 7236fcaebacSDaeho Jeong vm_unmap_ram(cc->cbuf, cc->nr_cpages); 7244c8ff709SChao Yu out_vunmap_rbuf: 7256fcaebacSDaeho Jeong vm_unmap_ram(cc->rbuf, cc->cluster_size); 7264c8ff709SChao Yu out_free_cpages: 7274c8ff709SChao Yu for (i = 0; i < cc->nr_cpages; i++) { 7284c8ff709SChao Yu if (cc->cpages[i]) 7295e6bbde9SChao Yu f2fs_compress_free_page(cc->cpages[i]); 7304c8ff709SChao Yu } 73131083031SChao Yu page_array_free(cc->inode, cc->cpages, cc->nr_cpages); 7324c8ff709SChao Yu cc->cpages = NULL; 7334c8ff709SChao Yu destroy_compress_ctx: 73423b1faaaSChao Yu if (cops->destroy_compress_ctx) 7354c8ff709SChao Yu cops->destroy_compress_ctx(cc); 7364c8ff709SChao Yu out: 7374c8ff709SChao Yu trace_f2fs_compress_pages_end(cc->inode, cc->cluster_idx, 7384c8ff709SChao Yu cc->clen, ret); 7394c8ff709SChao Yu return ret; 7404c8ff709SChao Yu } 7414c8ff709SChao Yu 7426ce19affSChao Yu void f2fs_decompress_cluster(struct decompress_io_ctx *dic) 7434c8ff709SChao Yu { 7444c8ff709SChao Yu struct f2fs_sb_info *sbi = F2FS_I_SB(dic->inode); 7454c8ff709SChao Yu struct f2fs_inode_info *fi = F2FS_I(dic->inode); 7464c8ff709SChao Yu const struct f2fs_compress_ops *cops = 7474c8ff709SChao Yu f2fs_cops[fi->i_compress_algorithm]; 7484c8ff709SChao Yu int ret; 749b2f57a8eSChao Yu int i; 7504c8ff709SChao Yu 7514c8ff709SChao Yu trace_f2fs_decompress_pages_start(dic->inode, dic->cluster_idx, 7524c8ff709SChao Yu dic->cluster_size, fi->i_compress_algorithm); 7534c8ff709SChao Yu 7544c8ff709SChao Yu if (dic->failed) { 7554c8ff709SChao Yu ret = -EIO; 7567f59b277SEric Biggers goto out_end_io; 7574c8ff709SChao Yu } 7584c8ff709SChao Yu 75931083031SChao Yu dic->tpages = page_array_alloc(dic->inode, dic->cluster_size); 760b2f57a8eSChao Yu if (!dic->tpages) { 761b2f57a8eSChao Yu ret = -ENOMEM; 7627f59b277SEric Biggers goto out_end_io; 763b2f57a8eSChao Yu } 764b2f57a8eSChao Yu 765b2f57a8eSChao Yu for (i = 0; i < dic->cluster_size; i++) { 766b2f57a8eSChao Yu if (dic->rpages[i]) { 767b2f57a8eSChao Yu dic->tpages[i] = dic->rpages[i]; 768b2f57a8eSChao Yu continue; 769b2f57a8eSChao Yu } 770b2f57a8eSChao Yu 771b2f57a8eSChao Yu dic->tpages[i] = f2fs_compress_alloc_page(); 772b2f57a8eSChao Yu if (!dic->tpages[i]) { 773b2f57a8eSChao Yu ret = -ENOMEM; 7747f59b277SEric Biggers goto out_end_io; 775b2f57a8eSChao Yu } 776b2f57a8eSChao Yu } 777b2f57a8eSChao Yu 77823b1faaaSChao Yu if (cops->init_decompress_ctx) { 77923b1faaaSChao Yu ret = cops->init_decompress_ctx(dic); 78023b1faaaSChao Yu if (ret) 7817f59b277SEric Biggers goto out_end_io; 78223b1faaaSChao Yu } 78323b1faaaSChao Yu 7846fcaebacSDaeho Jeong dic->rbuf = f2fs_vmap(dic->tpages, dic->cluster_size); 7854c8ff709SChao Yu if (!dic->rbuf) { 7864c8ff709SChao Yu ret = -ENOMEM; 7877f59b277SEric Biggers goto out_destroy_decompress_ctx; 7884c8ff709SChao Yu } 7894c8ff709SChao Yu 7906fcaebacSDaeho Jeong dic->cbuf = f2fs_vmap(dic->cpages, dic->nr_cpages); 7914c8ff709SChao Yu if (!dic->cbuf) { 7924c8ff709SChao Yu ret = -ENOMEM; 7934c8ff709SChao Yu goto out_vunmap_rbuf; 7944c8ff709SChao Yu } 7954c8ff709SChao Yu 7964c8ff709SChao Yu dic->clen = le32_to_cpu(dic->cbuf->clen); 7974c8ff709SChao Yu dic->rlen = PAGE_SIZE << dic->log_cluster_size; 7984c8ff709SChao Yu 7994c8ff709SChao Yu if (dic->clen > PAGE_SIZE * dic->nr_cpages - COMPRESS_HEADER_SIZE) { 8004c8ff709SChao Yu ret = -EFSCORRUPTED; 8014c8ff709SChao Yu goto out_vunmap_cbuf; 8024c8ff709SChao Yu } 8034c8ff709SChao Yu 8044c8ff709SChao Yu ret = cops->decompress_pages(dic); 8054c8ff709SChao Yu 80675e91c88SChao Yu if (!ret && (fi->i_compress_flag & 1 << COMPRESS_CHKSUM)) { 807b28f047bSChao Yu u32 provided = le32_to_cpu(dic->cbuf->chksum); 808b28f047bSChao Yu u32 calculated = f2fs_crc32(sbi, dic->cbuf->cdata, dic->clen); 809b28f047bSChao Yu 810b28f047bSChao Yu if (provided != calculated) { 811b28f047bSChao Yu if (!is_inode_flag_set(dic->inode, FI_COMPRESS_CORRUPT)) { 812b28f047bSChao Yu set_inode_flag(dic->inode, FI_COMPRESS_CORRUPT); 813b28f047bSChao Yu printk_ratelimited( 814b28f047bSChao Yu "%sF2FS-fs (%s): checksum invalid, nid = %lu, %x vs %x", 815b28f047bSChao Yu KERN_INFO, sbi->sb->s_id, dic->inode->i_ino, 816b28f047bSChao Yu provided, calculated); 817b28f047bSChao Yu } 818b28f047bSChao Yu set_sbi_flag(sbi, SBI_NEED_FSCK); 819b28f047bSChao Yu } 820b28f047bSChao Yu } 821b28f047bSChao Yu 8224c8ff709SChao Yu out_vunmap_cbuf: 8236fcaebacSDaeho Jeong vm_unmap_ram(dic->cbuf, dic->nr_cpages); 8244c8ff709SChao Yu out_vunmap_rbuf: 8256fcaebacSDaeho Jeong vm_unmap_ram(dic->rbuf, dic->cluster_size); 8267f59b277SEric Biggers out_destroy_decompress_ctx: 82723b1faaaSChao Yu if (cops->destroy_decompress_ctx) 82823b1faaaSChao Yu cops->destroy_decompress_ctx(dic); 8297f59b277SEric Biggers out_end_io: 8304c8ff709SChao Yu trace_f2fs_decompress_pages_end(dic->inode, dic->cluster_idx, 8314c8ff709SChao Yu dic->clen, ret); 8327f59b277SEric Biggers f2fs_decompress_end_io(dic, ret); 8337f59b277SEric Biggers } 8347f59b277SEric Biggers 8357f59b277SEric Biggers /* 8367f59b277SEric Biggers * This is called when a page of a compressed cluster has been read from disk 8377f59b277SEric Biggers * (or failed to be read from disk). It checks whether this page was the last 8387f59b277SEric Biggers * page being waited on in the cluster, and if so, it decompresses the cluster 8397f59b277SEric Biggers * (or in the case of a failure, cleans up without actually decompressing). 8407f59b277SEric Biggers */ 8416ce19affSChao Yu void f2fs_end_read_compressed_page(struct page *page, bool failed, 8426ce19affSChao Yu block_t blkaddr) 8437f59b277SEric Biggers { 8447f59b277SEric Biggers struct decompress_io_ctx *dic = 8457f59b277SEric Biggers (struct decompress_io_ctx *)page_private(page); 8467f59b277SEric Biggers struct f2fs_sb_info *sbi = F2FS_I_SB(dic->inode); 8477f59b277SEric Biggers 8487f59b277SEric Biggers dec_page_count(sbi, F2FS_RD_DATA); 8497f59b277SEric Biggers 8507f59b277SEric Biggers if (failed) 8517f59b277SEric Biggers WRITE_ONCE(dic->failed, true); 8526ce19affSChao Yu else if (blkaddr) 8536ce19affSChao Yu f2fs_cache_compressed_page(sbi, page, 8546ce19affSChao Yu dic->inode->i_ino, blkaddr); 8557f59b277SEric Biggers 8567f59b277SEric Biggers if (atomic_dec_and_test(&dic->remaining_pages)) 8577f59b277SEric Biggers f2fs_decompress_cluster(dic); 8584c8ff709SChao Yu } 8594c8ff709SChao Yu 8604c8ff709SChao Yu static bool is_page_in_cluster(struct compress_ctx *cc, pgoff_t index) 8614c8ff709SChao Yu { 8624c8ff709SChao Yu if (cc->cluster_idx == NULL_CLUSTER) 8634c8ff709SChao Yu return true; 8644c8ff709SChao Yu return cc->cluster_idx == cluster_idx(cc, index); 8654c8ff709SChao Yu } 8664c8ff709SChao Yu 8674c8ff709SChao Yu bool f2fs_cluster_is_empty(struct compress_ctx *cc) 8684c8ff709SChao Yu { 8694c8ff709SChao Yu return cc->nr_rpages == 0; 8704c8ff709SChao Yu } 8714c8ff709SChao Yu 8724c8ff709SChao Yu static bool f2fs_cluster_is_full(struct compress_ctx *cc) 8734c8ff709SChao Yu { 8744c8ff709SChao Yu return cc->cluster_size == cc->nr_rpages; 8754c8ff709SChao Yu } 8764c8ff709SChao Yu 8774c8ff709SChao Yu bool f2fs_cluster_can_merge_page(struct compress_ctx *cc, pgoff_t index) 8784c8ff709SChao Yu { 8794c8ff709SChao Yu if (f2fs_cluster_is_empty(cc)) 8804c8ff709SChao Yu return true; 8814c8ff709SChao Yu return is_page_in_cluster(cc, index); 8824c8ff709SChao Yu } 8834c8ff709SChao Yu 8845db479f0SChao Yu static bool cluster_has_invalid_data(struct compress_ctx *cc) 8854c8ff709SChao Yu { 8864c8ff709SChao Yu loff_t i_size = i_size_read(cc->inode); 8874c8ff709SChao Yu unsigned nr_pages = DIV_ROUND_UP(i_size, PAGE_SIZE); 8884c8ff709SChao Yu int i; 8894c8ff709SChao Yu 8904c8ff709SChao Yu for (i = 0; i < cc->cluster_size; i++) { 8914c8ff709SChao Yu struct page *page = cc->rpages[i]; 8924c8ff709SChao Yu 8938af85f71SChao Yu f2fs_bug_on(F2FS_I_SB(cc->inode), !page); 8944c8ff709SChao Yu 8954c8ff709SChao Yu /* beyond EOF */ 8964c8ff709SChao Yu if (page->index >= nr_pages) 8974c8ff709SChao Yu return true; 8984c8ff709SChao Yu } 8995db479f0SChao Yu return false; 9005db479f0SChao Yu } 9014c8ff709SChao Yu 90291f0fb69SChao Yu static int __f2fs_cluster_blocks(struct inode *inode, 90391f0fb69SChao Yu unsigned int cluster_idx, bool compr) 9044c8ff709SChao Yu { 9054c8ff709SChao Yu struct dnode_of_data dn; 90691f0fb69SChao Yu unsigned int cluster_size = F2FS_I(inode)->i_cluster_size; 90791f0fb69SChao Yu unsigned int start_idx = cluster_idx << 90891f0fb69SChao Yu F2FS_I(inode)->i_log_cluster_size; 9094c8ff709SChao Yu int ret; 9104c8ff709SChao Yu 91191f0fb69SChao Yu set_new_dnode(&dn, inode, NULL, NULL, 0); 91291f0fb69SChao Yu ret = f2fs_get_dnode_of_data(&dn, start_idx, LOOKUP_NODE); 9134c8ff709SChao Yu if (ret) { 9144c8ff709SChao Yu if (ret == -ENOENT) 9154c8ff709SChao Yu ret = 0; 9164c8ff709SChao Yu goto fail; 9174c8ff709SChao Yu } 9184c8ff709SChao Yu 9194c8ff709SChao Yu if (dn.data_blkaddr == COMPRESS_ADDR) { 9204c8ff709SChao Yu int i; 9214c8ff709SChao Yu 9224c8ff709SChao Yu ret = 1; 92391f0fb69SChao Yu for (i = 1; i < cluster_size; i++) { 9244c8ff709SChao Yu block_t blkaddr; 9254c8ff709SChao Yu 926a2ced1ceSChao Yu blkaddr = data_blkaddr(dn.inode, 9274c8ff709SChao Yu dn.node_page, dn.ofs_in_node + i); 9281a67cbe1SChao Yu if (compr) { 9291a67cbe1SChao Yu if (__is_valid_data_blkaddr(blkaddr)) 9301a67cbe1SChao Yu ret++; 9311a67cbe1SChao Yu } else { 9324c8ff709SChao Yu if (blkaddr != NULL_ADDR) 9334c8ff709SChao Yu ret++; 9344c8ff709SChao Yu } 9354c8ff709SChao Yu } 9368f1d4983SChao Yu 9378f1d4983SChao Yu f2fs_bug_on(F2FS_I_SB(inode), 938c6140415SJaegeuk Kim !compr && ret != cluster_size && 939c6140415SJaegeuk Kim !is_inode_flag_set(inode, FI_COMPRESS_RELEASED)); 9401a67cbe1SChao Yu } 9414c8ff709SChao Yu fail: 9424c8ff709SChao Yu f2fs_put_dnode(&dn); 9434c8ff709SChao Yu return ret; 9444c8ff709SChao Yu } 9454c8ff709SChao Yu 9461a67cbe1SChao Yu /* return # of compressed blocks in compressed cluster */ 9471a67cbe1SChao Yu static int f2fs_compressed_blocks(struct compress_ctx *cc) 9481a67cbe1SChao Yu { 94991f0fb69SChao Yu return __f2fs_cluster_blocks(cc->inode, cc->cluster_idx, true); 9501a67cbe1SChao Yu } 9511a67cbe1SChao Yu 9521a67cbe1SChao Yu /* return # of valid blocks in compressed cluster */ 9534c8ff709SChao Yu int f2fs_is_compressed_cluster(struct inode *inode, pgoff_t index) 9544c8ff709SChao Yu { 95591f0fb69SChao Yu return __f2fs_cluster_blocks(inode, 95691f0fb69SChao Yu index >> F2FS_I(inode)->i_log_cluster_size, 95791f0fb69SChao Yu false); 9584c8ff709SChao Yu } 9594c8ff709SChao Yu 9604c8ff709SChao Yu static bool cluster_may_compress(struct compress_ctx *cc) 9614c8ff709SChao Yu { 962602a16d5SDaeho Jeong if (!f2fs_need_compress_data(cc->inode)) 9634c8ff709SChao Yu return false; 9644c8ff709SChao Yu if (f2fs_is_atomic_file(cc->inode)) 9654c8ff709SChao Yu return false; 9664c8ff709SChao Yu if (!f2fs_cluster_is_full(cc)) 9674c8ff709SChao Yu return false; 968dc35d73aSChao Yu if (unlikely(f2fs_cp_error(F2FS_I_SB(cc->inode)))) 969dc35d73aSChao Yu return false; 9705db479f0SChao Yu return !cluster_has_invalid_data(cc); 9714c8ff709SChao Yu } 9724c8ff709SChao Yu 9734c8ff709SChao Yu static void set_cluster_writeback(struct compress_ctx *cc) 9744c8ff709SChao Yu { 9754c8ff709SChao Yu int i; 9764c8ff709SChao Yu 9774c8ff709SChao Yu for (i = 0; i < cc->cluster_size; i++) { 9784c8ff709SChao Yu if (cc->rpages[i]) 9794c8ff709SChao Yu set_page_writeback(cc->rpages[i]); 9804c8ff709SChao Yu } 9814c8ff709SChao Yu } 9824c8ff709SChao Yu 9834c8ff709SChao Yu static void set_cluster_dirty(struct compress_ctx *cc) 9844c8ff709SChao Yu { 9854c8ff709SChao Yu int i; 9864c8ff709SChao Yu 9874c8ff709SChao Yu for (i = 0; i < cc->cluster_size; i++) 9884c8ff709SChao Yu if (cc->rpages[i]) 9894c8ff709SChao Yu set_page_dirty(cc->rpages[i]); 9904c8ff709SChao Yu } 9914c8ff709SChao Yu 9924c8ff709SChao Yu static int prepare_compress_overwrite(struct compress_ctx *cc, 9934c8ff709SChao Yu struct page **pagep, pgoff_t index, void **fsdata) 9944c8ff709SChao Yu { 9954c8ff709SChao Yu struct f2fs_sb_info *sbi = F2FS_I_SB(cc->inode); 9964c8ff709SChao Yu struct address_space *mapping = cc->inode->i_mapping; 9974c8ff709SChao Yu struct page *page; 9984c8ff709SChao Yu sector_t last_block_in_bio; 9994c8ff709SChao Yu unsigned fgp_flag = FGP_LOCK | FGP_WRITE | FGP_CREAT; 10004c8ff709SChao Yu pgoff_t start_idx = start_idx_of_cluster(cc); 10014c8ff709SChao Yu int i, ret; 10024c8ff709SChao Yu 10034c8ff709SChao Yu retry: 100491f0fb69SChao Yu ret = f2fs_is_compressed_cluster(cc->inode, start_idx); 10054c8ff709SChao Yu if (ret <= 0) 10064c8ff709SChao Yu return ret; 10074c8ff709SChao Yu 10084c8ff709SChao Yu ret = f2fs_init_compress_ctx(cc); 10094c8ff709SChao Yu if (ret) 10104c8ff709SChao Yu return ret; 10114c8ff709SChao Yu 10124c8ff709SChao Yu /* keep page reference to avoid page reclaim */ 10134c8ff709SChao Yu for (i = 0; i < cc->cluster_size; i++) { 10144c8ff709SChao Yu page = f2fs_pagecache_get_page(mapping, start_idx + i, 10154c8ff709SChao Yu fgp_flag, GFP_NOFS); 10164c8ff709SChao Yu if (!page) { 10174c8ff709SChao Yu ret = -ENOMEM; 10184c8ff709SChao Yu goto unlock_pages; 10194c8ff709SChao Yu } 10204c8ff709SChao Yu 10214c8ff709SChao Yu if (PageUptodate(page)) 1022a949dc5fSChao Yu f2fs_put_page(page, 1); 10234c8ff709SChao Yu else 10244c8ff709SChao Yu f2fs_compress_ctx_add_page(cc, page); 10254c8ff709SChao Yu } 10264c8ff709SChao Yu 10274c8ff709SChao Yu if (!f2fs_cluster_is_empty(cc)) { 10284c8ff709SChao Yu struct bio *bio = NULL; 10294c8ff709SChao Yu 10304c8ff709SChao Yu ret = f2fs_read_multi_pages(cc, &bio, cc->cluster_size, 10310683728aSChao Yu &last_block_in_bio, false, true); 1032a949dc5fSChao Yu f2fs_put_rpages(cc); 10338bfbfb0dSChao Yu f2fs_destroy_compress_ctx(cc, true); 10344c8ff709SChao Yu if (ret) 1035a949dc5fSChao Yu goto out; 10364c8ff709SChao Yu if (bio) 10374c8ff709SChao Yu f2fs_submit_bio(sbi, bio, DATA); 10384c8ff709SChao Yu 10394c8ff709SChao Yu ret = f2fs_init_compress_ctx(cc); 10404c8ff709SChao Yu if (ret) 1041a949dc5fSChao Yu goto out; 10424c8ff709SChao Yu } 10434c8ff709SChao Yu 10444c8ff709SChao Yu for (i = 0; i < cc->cluster_size; i++) { 10454c8ff709SChao Yu f2fs_bug_on(sbi, cc->rpages[i]); 10464c8ff709SChao Yu 10474c8ff709SChao Yu page = find_lock_page(mapping, start_idx + i); 1048a949dc5fSChao Yu if (!page) { 1049a949dc5fSChao Yu /* page can be truncated */ 1050a949dc5fSChao Yu goto release_and_retry; 1051a949dc5fSChao Yu } 10524c8ff709SChao Yu 10534c8ff709SChao Yu f2fs_wait_on_page_writeback(page, DATA, true, true); 10544c8ff709SChao Yu f2fs_compress_ctx_add_page(cc, page); 10554c8ff709SChao Yu 10564c8ff709SChao Yu if (!PageUptodate(page)) { 1057a949dc5fSChao Yu release_and_retry: 1058a949dc5fSChao Yu f2fs_put_rpages(cc); 10594c8ff709SChao Yu f2fs_unlock_rpages(cc, i + 1); 10608bfbfb0dSChao Yu f2fs_destroy_compress_ctx(cc, true); 10614c8ff709SChao Yu goto retry; 10624c8ff709SChao Yu } 10634c8ff709SChao Yu } 10644c8ff709SChao Yu 10654c8ff709SChao Yu if (likely(!ret)) { 10664c8ff709SChao Yu *fsdata = cc->rpages; 10674c8ff709SChao Yu *pagep = cc->rpages[offset_in_cluster(cc, index)]; 10684c8ff709SChao Yu return cc->cluster_size; 10694c8ff709SChao Yu } 10704c8ff709SChao Yu 10714c8ff709SChao Yu unlock_pages: 1072a949dc5fSChao Yu f2fs_put_rpages(cc); 10734c8ff709SChao Yu f2fs_unlock_rpages(cc, i); 10748bfbfb0dSChao Yu f2fs_destroy_compress_ctx(cc, true); 1075a949dc5fSChao Yu out: 10764c8ff709SChao Yu return ret; 10774c8ff709SChao Yu } 10784c8ff709SChao Yu 10794c8ff709SChao Yu int f2fs_prepare_compress_overwrite(struct inode *inode, 10804c8ff709SChao Yu struct page **pagep, pgoff_t index, void **fsdata) 10814c8ff709SChao Yu { 10824c8ff709SChao Yu struct compress_ctx cc = { 10834c8ff709SChao Yu .inode = inode, 10844c8ff709SChao Yu .log_cluster_size = F2FS_I(inode)->i_log_cluster_size, 10854c8ff709SChao Yu .cluster_size = F2FS_I(inode)->i_cluster_size, 10864c8ff709SChao Yu .cluster_idx = index >> F2FS_I(inode)->i_log_cluster_size, 10874c8ff709SChao Yu .rpages = NULL, 10884c8ff709SChao Yu .nr_rpages = 0, 10894c8ff709SChao Yu }; 10904c8ff709SChao Yu 10914c8ff709SChao Yu return prepare_compress_overwrite(&cc, pagep, index, fsdata); 10924c8ff709SChao Yu } 10934c8ff709SChao Yu 10944c8ff709SChao Yu bool f2fs_compress_write_end(struct inode *inode, void *fsdata, 10954c8ff709SChao Yu pgoff_t index, unsigned copied) 10964c8ff709SChao Yu 10974c8ff709SChao Yu { 10984c8ff709SChao Yu struct compress_ctx cc = { 109931083031SChao Yu .inode = inode, 11004c8ff709SChao Yu .log_cluster_size = F2FS_I(inode)->i_log_cluster_size, 11014c8ff709SChao Yu .cluster_size = F2FS_I(inode)->i_cluster_size, 11024c8ff709SChao Yu .rpages = fsdata, 11034c8ff709SChao Yu }; 11044c8ff709SChao Yu bool first_index = (index == cc.rpages[0]->index); 11054c8ff709SChao Yu 11064c8ff709SChao Yu if (copied) 11074c8ff709SChao Yu set_cluster_dirty(&cc); 11084c8ff709SChao Yu 11094c8ff709SChao Yu f2fs_put_rpages_wbc(&cc, NULL, false, 1); 11108bfbfb0dSChao Yu f2fs_destroy_compress_ctx(&cc, false); 11114c8ff709SChao Yu 11124c8ff709SChao Yu return first_index; 11134c8ff709SChao Yu } 11144c8ff709SChao Yu 11153265d3dbSChao Yu int f2fs_truncate_partial_cluster(struct inode *inode, u64 from, bool lock) 11163265d3dbSChao Yu { 11173265d3dbSChao Yu void *fsdata = NULL; 11183265d3dbSChao Yu struct page *pagep; 11193265d3dbSChao Yu int log_cluster_size = F2FS_I(inode)->i_log_cluster_size; 11203265d3dbSChao Yu pgoff_t start_idx = from >> (PAGE_SHIFT + log_cluster_size) << 11213265d3dbSChao Yu log_cluster_size; 11223265d3dbSChao Yu int err; 11233265d3dbSChao Yu 11243265d3dbSChao Yu err = f2fs_is_compressed_cluster(inode, start_idx); 11253265d3dbSChao Yu if (err < 0) 11263265d3dbSChao Yu return err; 11273265d3dbSChao Yu 11283265d3dbSChao Yu /* truncate normal cluster */ 11293265d3dbSChao Yu if (!err) 11303265d3dbSChao Yu return f2fs_do_truncate_blocks(inode, from, lock); 11313265d3dbSChao Yu 11323265d3dbSChao Yu /* truncate compressed cluster */ 11333265d3dbSChao Yu err = f2fs_prepare_compress_overwrite(inode, &pagep, 11343265d3dbSChao Yu start_idx, &fsdata); 11353265d3dbSChao Yu 11363265d3dbSChao Yu /* should not be a normal cluster */ 11373265d3dbSChao Yu f2fs_bug_on(F2FS_I_SB(inode), err == 0); 11383265d3dbSChao Yu 11393265d3dbSChao Yu if (err <= 0) 11403265d3dbSChao Yu return err; 11413265d3dbSChao Yu 11423265d3dbSChao Yu if (err > 0) { 11433265d3dbSChao Yu struct page **rpages = fsdata; 11443265d3dbSChao Yu int cluster_size = F2FS_I(inode)->i_cluster_size; 11453265d3dbSChao Yu int i; 11463265d3dbSChao Yu 11473265d3dbSChao Yu for (i = cluster_size - 1; i >= 0; i--) { 11483265d3dbSChao Yu loff_t start = rpages[i]->index << PAGE_SHIFT; 11493265d3dbSChao Yu 11503265d3dbSChao Yu if (from <= start) { 11513265d3dbSChao Yu zero_user_segment(rpages[i], 0, PAGE_SIZE); 11523265d3dbSChao Yu } else { 11533265d3dbSChao Yu zero_user_segment(rpages[i], from - start, 11543265d3dbSChao Yu PAGE_SIZE); 11553265d3dbSChao Yu break; 11563265d3dbSChao Yu } 11573265d3dbSChao Yu } 11583265d3dbSChao Yu 11593265d3dbSChao Yu f2fs_compress_write_end(inode, fsdata, start_idx, true); 11603265d3dbSChao Yu } 11613265d3dbSChao Yu return 0; 11623265d3dbSChao Yu } 11633265d3dbSChao Yu 11644c8ff709SChao Yu static int f2fs_write_compressed_pages(struct compress_ctx *cc, 11654c8ff709SChao Yu int *submitted, 11664c8ff709SChao Yu struct writeback_control *wbc, 11674c8ff709SChao Yu enum iostat_type io_type) 11684c8ff709SChao Yu { 11694c8ff709SChao Yu struct inode *inode = cc->inode; 11704c8ff709SChao Yu struct f2fs_sb_info *sbi = F2FS_I_SB(inode); 11714c8ff709SChao Yu struct f2fs_inode_info *fi = F2FS_I(inode); 11724c8ff709SChao Yu struct f2fs_io_info fio = { 11734c8ff709SChao Yu .sbi = sbi, 11744c8ff709SChao Yu .ino = cc->inode->i_ino, 11754c8ff709SChao Yu .type = DATA, 11764c8ff709SChao Yu .op = REQ_OP_WRITE, 11774c8ff709SChao Yu .op_flags = wbc_to_write_flags(wbc), 11784c8ff709SChao Yu .old_blkaddr = NEW_ADDR, 11794c8ff709SChao Yu .page = NULL, 11804c8ff709SChao Yu .encrypted_page = NULL, 11814c8ff709SChao Yu .compressed_page = NULL, 11824c8ff709SChao Yu .submitted = false, 11834c8ff709SChao Yu .io_type = io_type, 11844c8ff709SChao Yu .io_wbc = wbc, 118527aacd28SSatya Tangirala .encrypted = fscrypt_inode_uses_fs_layer_crypto(cc->inode), 11864c8ff709SChao Yu }; 11874c8ff709SChao Yu struct dnode_of_data dn; 11884c8ff709SChao Yu struct node_info ni; 11894c8ff709SChao Yu struct compress_io_ctx *cic; 11904c8ff709SChao Yu pgoff_t start_idx = start_idx_of_cluster(cc); 11914c8ff709SChao Yu unsigned int last_index = cc->cluster_size - 1; 11924c8ff709SChao Yu loff_t psize; 11934c8ff709SChao Yu int i, err; 11944c8ff709SChao Yu 1195ee68d271SChao Yu /* we should bypass data pages to proceed the kworkder jobs */ 1196ee68d271SChao Yu if (unlikely(f2fs_cp_error(sbi))) { 1197ee68d271SChao Yu mapping_set_error(cc->rpages[0]->mapping, -EIO); 1198ee68d271SChao Yu goto out_free; 1199ee68d271SChao Yu } 1200ee68d271SChao Yu 120179963d96SChao Yu if (IS_NOQUOTA(inode)) { 120279963d96SChao Yu /* 120379963d96SChao Yu * We need to wait for node_write to avoid block allocation during 120479963d96SChao Yu * checkpoint. This can only happen to quota writes which can cause 120579963d96SChao Yu * the below discard race condition. 120679963d96SChao Yu */ 120779963d96SChao Yu down_read(&sbi->node_write); 120879963d96SChao Yu } else if (!f2fs_trylock_op(sbi)) { 120931083031SChao Yu goto out_free; 121079963d96SChao Yu } 12114c8ff709SChao Yu 1212df77fbd8SChao Yu set_new_dnode(&dn, cc->inode, NULL, NULL, 0); 12134c8ff709SChao Yu 12144c8ff709SChao Yu err = f2fs_get_dnode_of_data(&dn, start_idx, LOOKUP_NODE); 12154c8ff709SChao Yu if (err) 12164c8ff709SChao Yu goto out_unlock_op; 12174c8ff709SChao Yu 12184c8ff709SChao Yu for (i = 0; i < cc->cluster_size; i++) { 1219a2ced1ceSChao Yu if (data_blkaddr(dn.inode, dn.node_page, 12204c8ff709SChao Yu dn.ofs_in_node + i) == NULL_ADDR) 12214c8ff709SChao Yu goto out_put_dnode; 12224c8ff709SChao Yu } 12234c8ff709SChao Yu 12244c8ff709SChao Yu psize = (loff_t)(cc->rpages[last_index]->index + 1) << PAGE_SHIFT; 12254c8ff709SChao Yu 12264c8ff709SChao Yu err = f2fs_get_node_info(fio.sbi, dn.nid, &ni); 12274c8ff709SChao Yu if (err) 12284c8ff709SChao Yu goto out_put_dnode; 12294c8ff709SChao Yu 12304c8ff709SChao Yu fio.version = ni.version; 12314c8ff709SChao Yu 1232*32410577SChao Yu cic = f2fs_kmem_cache_alloc(cic_entry_slab, GFP_F2FS_ZERO, false, sbi); 12334c8ff709SChao Yu if (!cic) 12344c8ff709SChao Yu goto out_put_dnode; 12354c8ff709SChao Yu 12364c8ff709SChao Yu cic->magic = F2FS_COMPRESSED_PAGE_MAGIC; 12374c8ff709SChao Yu cic->inode = inode; 1238e6c3948dSChao Yu atomic_set(&cic->pending_pages, cc->nr_cpages); 123931083031SChao Yu cic->rpages = page_array_alloc(cc->inode, cc->cluster_size); 12404c8ff709SChao Yu if (!cic->rpages) 12414c8ff709SChao Yu goto out_put_cic; 12424c8ff709SChao Yu 12434c8ff709SChao Yu cic->nr_rpages = cc->cluster_size; 12444c8ff709SChao Yu 12454c8ff709SChao Yu for (i = 0; i < cc->nr_cpages; i++) { 12464c8ff709SChao Yu f2fs_set_compressed_page(cc->cpages[i], inode, 1247887347a0SChao Yu cc->rpages[i + 1]->index, cic); 12484c8ff709SChao Yu fio.compressed_page = cc->cpages[i]; 1249f567adb0SChao Yu 1250f567adb0SChao Yu fio.old_blkaddr = data_blkaddr(dn.inode, dn.node_page, 1251f567adb0SChao Yu dn.ofs_in_node + i + 1); 1252f567adb0SChao Yu 1253f567adb0SChao Yu /* wait for GCed page writeback via META_MAPPING */ 1254f567adb0SChao Yu f2fs_wait_on_block_writeback(inode, fio.old_blkaddr); 1255f567adb0SChao Yu 12564c8ff709SChao Yu if (fio.encrypted) { 12574c8ff709SChao Yu fio.page = cc->rpages[i + 1]; 12584c8ff709SChao Yu err = f2fs_encrypt_one_page(&fio); 12594c8ff709SChao Yu if (err) 12604c8ff709SChao Yu goto out_destroy_crypt; 12614c8ff709SChao Yu cc->cpages[i] = fio.encrypted_page; 12624c8ff709SChao Yu } 12634c8ff709SChao Yu } 12644c8ff709SChao Yu 12654c8ff709SChao Yu set_cluster_writeback(cc); 12664c8ff709SChao Yu 12674c8ff709SChao Yu for (i = 0; i < cc->cluster_size; i++) 12684c8ff709SChao Yu cic->rpages[i] = cc->rpages[i]; 12694c8ff709SChao Yu 12704c8ff709SChao Yu for (i = 0; i < cc->cluster_size; i++, dn.ofs_in_node++) { 12714c8ff709SChao Yu block_t blkaddr; 12724c8ff709SChao Yu 1273a2ced1ceSChao Yu blkaddr = f2fs_data_blkaddr(&dn); 127495978caaSChao Yu fio.page = cc->rpages[i]; 12754c8ff709SChao Yu fio.old_blkaddr = blkaddr; 12764c8ff709SChao Yu 12774c8ff709SChao Yu /* cluster header */ 12784c8ff709SChao Yu if (i == 0) { 12794c8ff709SChao Yu if (blkaddr == COMPRESS_ADDR) 12804c8ff709SChao Yu fio.compr_blocks++; 12814c8ff709SChao Yu if (__is_valid_data_blkaddr(blkaddr)) 12824c8ff709SChao Yu f2fs_invalidate_blocks(sbi, blkaddr); 12834c8ff709SChao Yu f2fs_update_data_blkaddr(&dn, COMPRESS_ADDR); 12844c8ff709SChao Yu goto unlock_continue; 12854c8ff709SChao Yu } 12864c8ff709SChao Yu 12874c8ff709SChao Yu if (fio.compr_blocks && __is_valid_data_blkaddr(blkaddr)) 12884c8ff709SChao Yu fio.compr_blocks++; 12894c8ff709SChao Yu 12904c8ff709SChao Yu if (i > cc->nr_cpages) { 12914c8ff709SChao Yu if (__is_valid_data_blkaddr(blkaddr)) { 12924c8ff709SChao Yu f2fs_invalidate_blocks(sbi, blkaddr); 12934c8ff709SChao Yu f2fs_update_data_blkaddr(&dn, NEW_ADDR); 12944c8ff709SChao Yu } 12954c8ff709SChao Yu goto unlock_continue; 12964c8ff709SChao Yu } 12974c8ff709SChao Yu 12984c8ff709SChao Yu f2fs_bug_on(fio.sbi, blkaddr == NULL_ADDR); 12994c8ff709SChao Yu 13004c8ff709SChao Yu if (fio.encrypted) 13014c8ff709SChao Yu fio.encrypted_page = cc->cpages[i - 1]; 13024c8ff709SChao Yu else 13034c8ff709SChao Yu fio.compressed_page = cc->cpages[i - 1]; 13044c8ff709SChao Yu 13054c8ff709SChao Yu cc->cpages[i - 1] = NULL; 13064c8ff709SChao Yu f2fs_outplace_write_data(&dn, &fio); 13074c8ff709SChao Yu (*submitted)++; 13084c8ff709SChao Yu unlock_continue: 13094c8ff709SChao Yu inode_dec_dirty_pages(cc->inode); 13104c8ff709SChao Yu unlock_page(fio.page); 13114c8ff709SChao Yu } 13124c8ff709SChao Yu 13134c8ff709SChao Yu if (fio.compr_blocks) 13144c8ff709SChao Yu f2fs_i_compr_blocks_update(inode, fio.compr_blocks - 1, false); 13154c8ff709SChao Yu f2fs_i_compr_blocks_update(inode, cc->nr_cpages, true); 13165ac443e2SDaeho Jeong add_compr_block_stat(inode, cc->nr_cpages); 13174c8ff709SChao Yu 13184c8ff709SChao Yu set_inode_flag(cc->inode, FI_APPEND_WRITE); 13194c8ff709SChao Yu if (cc->cluster_idx == 0) 13204c8ff709SChao Yu set_inode_flag(inode, FI_FIRST_BLOCK_WRITTEN); 13214c8ff709SChao Yu 13224c8ff709SChao Yu f2fs_put_dnode(&dn); 132379963d96SChao Yu if (IS_NOQUOTA(inode)) 132479963d96SChao Yu up_read(&sbi->node_write); 132579963d96SChao Yu else 13264c8ff709SChao Yu f2fs_unlock_op(sbi); 13274c8ff709SChao Yu 1328c10c9820SChao Yu spin_lock(&fi->i_size_lock); 13294c8ff709SChao Yu if (fi->last_disk_size < psize) 13304c8ff709SChao Yu fi->last_disk_size = psize; 1331c10c9820SChao Yu spin_unlock(&fi->i_size_lock); 13324c8ff709SChao Yu 13334c8ff709SChao Yu f2fs_put_rpages(cc); 133431083031SChao Yu page_array_free(cc->inode, cc->cpages, cc->nr_cpages); 133531083031SChao Yu cc->cpages = NULL; 13368bfbfb0dSChao Yu f2fs_destroy_compress_ctx(cc, false); 13374c8ff709SChao Yu return 0; 13384c8ff709SChao Yu 13394c8ff709SChao Yu out_destroy_crypt: 134031083031SChao Yu page_array_free(cc->inode, cic->rpages, cc->cluster_size); 13414c8ff709SChao Yu 13424c8ff709SChao Yu for (--i; i >= 0; i--) 13434c8ff709SChao Yu fscrypt_finalize_bounce_page(&cc->cpages[i]); 13444c8ff709SChao Yu for (i = 0; i < cc->nr_cpages; i++) { 13454c8ff709SChao Yu if (!cc->cpages[i]) 13464c8ff709SChao Yu continue; 1347a12cc5b4SChao Yu f2fs_compress_free_page(cc->cpages[i]); 1348a12cc5b4SChao Yu cc->cpages[i] = NULL; 13494c8ff709SChao Yu } 13504c8ff709SChao Yu out_put_cic: 1351c68d6c88SChao Yu kmem_cache_free(cic_entry_slab, cic); 13524c8ff709SChao Yu out_put_dnode: 13534c8ff709SChao Yu f2fs_put_dnode(&dn); 13544c8ff709SChao Yu out_unlock_op: 135579963d96SChao Yu if (IS_NOQUOTA(inode)) 135679963d96SChao Yu up_read(&sbi->node_write); 135779963d96SChao Yu else 13584c8ff709SChao Yu f2fs_unlock_op(sbi); 135931083031SChao Yu out_free: 136031083031SChao Yu page_array_free(cc->inode, cc->cpages, cc->nr_cpages); 136131083031SChao Yu cc->cpages = NULL; 13624c8ff709SChao Yu return -EAGAIN; 13634c8ff709SChao Yu } 13644c8ff709SChao Yu 13654c8ff709SChao Yu void f2fs_compress_write_end_io(struct bio *bio, struct page *page) 13664c8ff709SChao Yu { 13674c8ff709SChao Yu struct f2fs_sb_info *sbi = bio->bi_private; 13684c8ff709SChao Yu struct compress_io_ctx *cic = 13694c8ff709SChao Yu (struct compress_io_ctx *)page_private(page); 13704c8ff709SChao Yu int i; 13714c8ff709SChao Yu 13724c8ff709SChao Yu if (unlikely(bio->bi_status)) 13734c8ff709SChao Yu mapping_set_error(cic->inode->i_mapping, -EIO); 13744c8ff709SChao Yu 13755e6bbde9SChao Yu f2fs_compress_free_page(page); 13764c8ff709SChao Yu 13774c8ff709SChao Yu dec_page_count(sbi, F2FS_WB_DATA); 13784c8ff709SChao Yu 1379e6c3948dSChao Yu if (atomic_dec_return(&cic->pending_pages)) 13804c8ff709SChao Yu return; 13814c8ff709SChao Yu 13824c8ff709SChao Yu for (i = 0; i < cic->nr_rpages; i++) { 13834c8ff709SChao Yu WARN_ON(!cic->rpages[i]); 1384b763f3beSChao Yu clear_page_private_gcing(cic->rpages[i]); 13854c8ff709SChao Yu end_page_writeback(cic->rpages[i]); 13864c8ff709SChao Yu } 13874c8ff709SChao Yu 138831083031SChao Yu page_array_free(cic->inode, cic->rpages, cic->nr_rpages); 1389c68d6c88SChao Yu kmem_cache_free(cic_entry_slab, cic); 13904c8ff709SChao Yu } 13914c8ff709SChao Yu 13924c8ff709SChao Yu static int f2fs_write_raw_pages(struct compress_ctx *cc, 13934c8ff709SChao Yu int *submitted, 13944c8ff709SChao Yu struct writeback_control *wbc, 13954c8ff709SChao Yu enum iostat_type io_type) 13964c8ff709SChao Yu { 13974c8ff709SChao Yu struct address_space *mapping = cc->inode->i_mapping; 13984c8ff709SChao Yu int _submitted, compr_blocks, ret; 13994c8ff709SChao Yu int i = -1, err = 0; 14004c8ff709SChao Yu 14014c8ff709SChao Yu compr_blocks = f2fs_compressed_blocks(cc); 14024c8ff709SChao Yu if (compr_blocks < 0) { 14034c8ff709SChao Yu err = compr_blocks; 14044c8ff709SChao Yu goto out_err; 14054c8ff709SChao Yu } 14064c8ff709SChao Yu 14074c8ff709SChao Yu for (i = 0; i < cc->cluster_size; i++) { 14084c8ff709SChao Yu if (!cc->rpages[i]) 14094c8ff709SChao Yu continue; 14104c8ff709SChao Yu retry_write: 14114c8ff709SChao Yu if (cc->rpages[i]->mapping != mapping) { 14124c8ff709SChao Yu unlock_page(cc->rpages[i]); 14134c8ff709SChao Yu continue; 14144c8ff709SChao Yu } 14154c8ff709SChao Yu 14164c8ff709SChao Yu BUG_ON(!PageLocked(cc->rpages[i])); 14174c8ff709SChao Yu 14184c8ff709SChao Yu ret = f2fs_write_single_data_page(cc->rpages[i], &_submitted, 14194c8ff709SChao Yu NULL, NULL, wbc, io_type, 14203afae09fSChao Yu compr_blocks, false); 14214c8ff709SChao Yu if (ret) { 14224c8ff709SChao Yu if (ret == AOP_WRITEPAGE_ACTIVATE) { 14234c8ff709SChao Yu unlock_page(cc->rpages[i]); 14244c8ff709SChao Yu ret = 0; 14254c8ff709SChao Yu } else if (ret == -EAGAIN) { 1426466357dcSChao Yu /* 1427466357dcSChao Yu * for quota file, just redirty left pages to 1428466357dcSChao Yu * avoid deadlock caused by cluster update race 1429466357dcSChao Yu * from foreground operation. 1430466357dcSChao Yu */ 1431466357dcSChao Yu if (IS_NOQUOTA(cc->inode)) { 1432466357dcSChao Yu err = 0; 1433466357dcSChao Yu goto out_err; 1434466357dcSChao Yu } 14354c8ff709SChao Yu ret = 0; 14364c8ff709SChao Yu cond_resched(); 14375df7731fSChao Yu congestion_wait(BLK_RW_ASYNC, 14385df7731fSChao Yu DEFAULT_IO_TIMEOUT); 14394c8ff709SChao Yu lock_page(cc->rpages[i]); 1440eb1353cfSChao Yu 1441eb1353cfSChao Yu if (!PageDirty(cc->rpages[i])) { 1442eb1353cfSChao Yu unlock_page(cc->rpages[i]); 1443eb1353cfSChao Yu continue; 1444eb1353cfSChao Yu } 1445eb1353cfSChao Yu 14464c8ff709SChao Yu clear_page_dirty_for_io(cc->rpages[i]); 14474c8ff709SChao Yu goto retry_write; 14484c8ff709SChao Yu } 14494c8ff709SChao Yu err = ret; 1450466357dcSChao Yu goto out_err; 14514c8ff709SChao Yu } 14524c8ff709SChao Yu 14534c8ff709SChao Yu *submitted += _submitted; 14544c8ff709SChao Yu } 14553afae09fSChao Yu 14563afae09fSChao Yu f2fs_balance_fs(F2FS_M_SB(mapping), true); 14573afae09fSChao Yu 14584c8ff709SChao Yu return 0; 14594c8ff709SChao Yu out_err: 14604c8ff709SChao Yu for (++i; i < cc->cluster_size; i++) { 14614c8ff709SChao Yu if (!cc->rpages[i]) 14624c8ff709SChao Yu continue; 14634c8ff709SChao Yu redirty_page_for_writepage(wbc, cc->rpages[i]); 14644c8ff709SChao Yu unlock_page(cc->rpages[i]); 14654c8ff709SChao Yu } 14664c8ff709SChao Yu return err; 14674c8ff709SChao Yu } 14684c8ff709SChao Yu 14694c8ff709SChao Yu int f2fs_write_multi_pages(struct compress_ctx *cc, 14704c8ff709SChao Yu int *submitted, 14714c8ff709SChao Yu struct writeback_control *wbc, 14724c8ff709SChao Yu enum iostat_type io_type) 14734c8ff709SChao Yu { 14744c8ff709SChao Yu int err; 14754c8ff709SChao Yu 14764c8ff709SChao Yu *submitted = 0; 14774c8ff709SChao Yu if (cluster_may_compress(cc)) { 14784c8ff709SChao Yu err = f2fs_compress_pages(cc); 14794c8ff709SChao Yu if (err == -EAGAIN) { 14804c8ff709SChao Yu goto write; 14814c8ff709SChao Yu } else if (err) { 14824c8ff709SChao Yu f2fs_put_rpages_wbc(cc, wbc, true, 1); 14834c8ff709SChao Yu goto destroy_out; 14844c8ff709SChao Yu } 14854c8ff709SChao Yu 14864c8ff709SChao Yu err = f2fs_write_compressed_pages(cc, submitted, 14874c8ff709SChao Yu wbc, io_type); 14884c8ff709SChao Yu if (!err) 14894c8ff709SChao Yu return 0; 14904c8ff709SChao Yu f2fs_bug_on(F2FS_I_SB(cc->inode), err != -EAGAIN); 14914c8ff709SChao Yu } 14924c8ff709SChao Yu write: 14934c8ff709SChao Yu f2fs_bug_on(F2FS_I_SB(cc->inode), *submitted); 14944c8ff709SChao Yu 14954c8ff709SChao Yu err = f2fs_write_raw_pages(cc, submitted, wbc, io_type); 14964c8ff709SChao Yu f2fs_put_rpages_wbc(cc, wbc, false, 0); 14974c8ff709SChao Yu destroy_out: 14988bfbfb0dSChao Yu f2fs_destroy_compress_ctx(cc, false); 14994c8ff709SChao Yu return err; 15004c8ff709SChao Yu } 15014c8ff709SChao Yu 15027f59b277SEric Biggers static void f2fs_free_dic(struct decompress_io_ctx *dic); 15037f59b277SEric Biggers 15044c8ff709SChao Yu struct decompress_io_ctx *f2fs_alloc_dic(struct compress_ctx *cc) 15054c8ff709SChao Yu { 15064c8ff709SChao Yu struct decompress_io_ctx *dic; 15074c8ff709SChao Yu pgoff_t start_idx = start_idx_of_cluster(cc); 15084c8ff709SChao Yu int i; 15094c8ff709SChao Yu 1510*32410577SChao Yu dic = f2fs_kmem_cache_alloc(dic_entry_slab, GFP_F2FS_ZERO, 1511*32410577SChao Yu false, F2FS_I_SB(cc->inode)); 15124c8ff709SChao Yu if (!dic) 15134c8ff709SChao Yu return ERR_PTR(-ENOMEM); 15144c8ff709SChao Yu 151531083031SChao Yu dic->rpages = page_array_alloc(cc->inode, cc->cluster_size); 15164c8ff709SChao Yu if (!dic->rpages) { 1517c68d6c88SChao Yu kmem_cache_free(dic_entry_slab, dic); 15184c8ff709SChao Yu return ERR_PTR(-ENOMEM); 15194c8ff709SChao Yu } 15204c8ff709SChao Yu 15214c8ff709SChao Yu dic->magic = F2FS_COMPRESSED_PAGE_MAGIC; 15224c8ff709SChao Yu dic->inode = cc->inode; 15237f59b277SEric Biggers atomic_set(&dic->remaining_pages, cc->nr_cpages); 15244c8ff709SChao Yu dic->cluster_idx = cc->cluster_idx; 15254c8ff709SChao Yu dic->cluster_size = cc->cluster_size; 15264c8ff709SChao Yu dic->log_cluster_size = cc->log_cluster_size; 15274c8ff709SChao Yu dic->nr_cpages = cc->nr_cpages; 15287f59b277SEric Biggers refcount_set(&dic->refcnt, 1); 15294c8ff709SChao Yu dic->failed = false; 15307f59b277SEric Biggers dic->need_verity = f2fs_need_verity(cc->inode, start_idx); 15314c8ff709SChao Yu 15324c8ff709SChao Yu for (i = 0; i < dic->cluster_size; i++) 15334c8ff709SChao Yu dic->rpages[i] = cc->rpages[i]; 15344c8ff709SChao Yu dic->nr_rpages = cc->cluster_size; 15354c8ff709SChao Yu 153631083031SChao Yu dic->cpages = page_array_alloc(dic->inode, dic->nr_cpages); 15374c8ff709SChao Yu if (!dic->cpages) 15384c8ff709SChao Yu goto out_free; 15394c8ff709SChao Yu 15404c8ff709SChao Yu for (i = 0; i < dic->nr_cpages; i++) { 15414c8ff709SChao Yu struct page *page; 15424c8ff709SChao Yu 15435e6bbde9SChao Yu page = f2fs_compress_alloc_page(); 15444c8ff709SChao Yu if (!page) 15454c8ff709SChao Yu goto out_free; 15464c8ff709SChao Yu 15474c8ff709SChao Yu f2fs_set_compressed_page(page, cc->inode, 1548887347a0SChao Yu start_idx + i + 1, dic); 15494c8ff709SChao Yu dic->cpages[i] = page; 15504c8ff709SChao Yu } 15514c8ff709SChao Yu 15524c8ff709SChao Yu return dic; 15534c8ff709SChao Yu 15544c8ff709SChao Yu out_free: 15554c8ff709SChao Yu f2fs_free_dic(dic); 15564c8ff709SChao Yu return ERR_PTR(-ENOMEM); 15574c8ff709SChao Yu } 15584c8ff709SChao Yu 15597f59b277SEric Biggers static void f2fs_free_dic(struct decompress_io_ctx *dic) 15604c8ff709SChao Yu { 15614c8ff709SChao Yu int i; 15624c8ff709SChao Yu 15634c8ff709SChao Yu if (dic->tpages) { 15644c8ff709SChao Yu for (i = 0; i < dic->cluster_size; i++) { 15654c8ff709SChao Yu if (dic->rpages[i]) 15664c8ff709SChao Yu continue; 15678908e753SChao Yu if (!dic->tpages[i]) 15688908e753SChao Yu continue; 15695e6bbde9SChao Yu f2fs_compress_free_page(dic->tpages[i]); 15704c8ff709SChao Yu } 157131083031SChao Yu page_array_free(dic->inode, dic->tpages, dic->cluster_size); 15724c8ff709SChao Yu } 15734c8ff709SChao Yu 15744c8ff709SChao Yu if (dic->cpages) { 15754c8ff709SChao Yu for (i = 0; i < dic->nr_cpages; i++) { 15764c8ff709SChao Yu if (!dic->cpages[i]) 15774c8ff709SChao Yu continue; 15785e6bbde9SChao Yu f2fs_compress_free_page(dic->cpages[i]); 15794c8ff709SChao Yu } 158031083031SChao Yu page_array_free(dic->inode, dic->cpages, dic->nr_cpages); 15814c8ff709SChao Yu } 15824c8ff709SChao Yu 158331083031SChao Yu page_array_free(dic->inode, dic->rpages, dic->nr_rpages); 1584c68d6c88SChao Yu kmem_cache_free(dic_entry_slab, dic); 15854c8ff709SChao Yu } 15864c8ff709SChao Yu 15877f59b277SEric Biggers static void f2fs_put_dic(struct decompress_io_ctx *dic) 15887f59b277SEric Biggers { 15897f59b277SEric Biggers if (refcount_dec_and_test(&dic->refcnt)) 15907f59b277SEric Biggers f2fs_free_dic(dic); 15917f59b277SEric Biggers } 15927f59b277SEric Biggers 15937f59b277SEric Biggers /* 15947f59b277SEric Biggers * Update and unlock the cluster's pagecache pages, and release the reference to 15957f59b277SEric Biggers * the decompress_io_ctx that was being held for I/O completion. 15967f59b277SEric Biggers */ 15977f59b277SEric Biggers static void __f2fs_decompress_end_io(struct decompress_io_ctx *dic, bool failed) 15984c8ff709SChao Yu { 15994c8ff709SChao Yu int i; 16004c8ff709SChao Yu 16017f59b277SEric Biggers for (i = 0; i < dic->cluster_size; i++) { 16027f59b277SEric Biggers struct page *rpage = dic->rpages[i]; 16034c8ff709SChao Yu 16044c8ff709SChao Yu if (!rpage) 16054c8ff709SChao Yu continue; 16064c8ff709SChao Yu 16077f59b277SEric Biggers /* PG_error was set if verity failed. */ 16087f59b277SEric Biggers if (failed || PageError(rpage)) { 16094c8ff709SChao Yu ClearPageUptodate(rpage); 16107f59b277SEric Biggers /* will re-read again later */ 16114c8ff709SChao Yu ClearPageError(rpage); 16127f59b277SEric Biggers } else { 16137f59b277SEric Biggers SetPageUptodate(rpage); 16147f59b277SEric Biggers } 16154c8ff709SChao Yu unlock_page(rpage); 16164c8ff709SChao Yu } 16177f59b277SEric Biggers 16187f59b277SEric Biggers f2fs_put_dic(dic); 16197f59b277SEric Biggers } 16207f59b277SEric Biggers 16217f59b277SEric Biggers static void f2fs_verify_cluster(struct work_struct *work) 16227f59b277SEric Biggers { 16237f59b277SEric Biggers struct decompress_io_ctx *dic = 16247f59b277SEric Biggers container_of(work, struct decompress_io_ctx, verity_work); 16257f59b277SEric Biggers int i; 16267f59b277SEric Biggers 16277f59b277SEric Biggers /* Verify the cluster's decompressed pages with fs-verity. */ 16287f59b277SEric Biggers for (i = 0; i < dic->cluster_size; i++) { 16297f59b277SEric Biggers struct page *rpage = dic->rpages[i]; 16307f59b277SEric Biggers 16317f59b277SEric Biggers if (rpage && !fsverity_verify_page(rpage)) 16327f59b277SEric Biggers SetPageError(rpage); 16337f59b277SEric Biggers } 16347f59b277SEric Biggers 16357f59b277SEric Biggers __f2fs_decompress_end_io(dic, false); 16367f59b277SEric Biggers } 16377f59b277SEric Biggers 16387f59b277SEric Biggers /* 16397f59b277SEric Biggers * This is called when a compressed cluster has been decompressed 16407f59b277SEric Biggers * (or failed to be read and/or decompressed). 16417f59b277SEric Biggers */ 16427f59b277SEric Biggers void f2fs_decompress_end_io(struct decompress_io_ctx *dic, bool failed) 16437f59b277SEric Biggers { 16447f59b277SEric Biggers if (!failed && dic->need_verity) { 16457f59b277SEric Biggers /* 16467f59b277SEric Biggers * Note that to avoid deadlocks, the verity work can't be done 16477f59b277SEric Biggers * on the decompression workqueue. This is because verifying 16487f59b277SEric Biggers * the data pages can involve reading metadata pages from the 16497f59b277SEric Biggers * file, and these metadata pages may be compressed. 16507f59b277SEric Biggers */ 16517f59b277SEric Biggers INIT_WORK(&dic->verity_work, f2fs_verify_cluster); 16527f59b277SEric Biggers fsverity_enqueue_verify_work(&dic->verity_work); 16537f59b277SEric Biggers } else { 16547f59b277SEric Biggers __f2fs_decompress_end_io(dic, failed); 16557f59b277SEric Biggers } 16567f59b277SEric Biggers } 16577f59b277SEric Biggers 16587f59b277SEric Biggers /* 16597f59b277SEric Biggers * Put a reference to a compressed page's decompress_io_ctx. 16607f59b277SEric Biggers * 16617f59b277SEric Biggers * This is called when the page is no longer needed and can be freed. 16627f59b277SEric Biggers */ 16637f59b277SEric Biggers void f2fs_put_page_dic(struct page *page) 16647f59b277SEric Biggers { 16657f59b277SEric Biggers struct decompress_io_ctx *dic = 16667f59b277SEric Biggers (struct decompress_io_ctx *)page_private(page); 16677f59b277SEric Biggers 16687f59b277SEric Biggers f2fs_put_dic(dic); 16694c8ff709SChao Yu } 167031083031SChao Yu 167194afd6d6SChao Yu /* 167294afd6d6SChao Yu * check whether cluster blocks are contiguous, and add extent cache entry 167394afd6d6SChao Yu * only if cluster blocks are logically and physically contiguous. 167494afd6d6SChao Yu */ 167594afd6d6SChao Yu unsigned int f2fs_cluster_blocks_are_contiguous(struct dnode_of_data *dn) 167694afd6d6SChao Yu { 167794afd6d6SChao Yu bool compressed = f2fs_data_blkaddr(dn) == COMPRESS_ADDR; 167894afd6d6SChao Yu int i = compressed ? 1 : 0; 167994afd6d6SChao Yu block_t first_blkaddr = data_blkaddr(dn->inode, dn->node_page, 168094afd6d6SChao Yu dn->ofs_in_node + i); 168194afd6d6SChao Yu 168294afd6d6SChao Yu for (i += 1; i < F2FS_I(dn->inode)->i_cluster_size; i++) { 168394afd6d6SChao Yu block_t blkaddr = data_blkaddr(dn->inode, dn->node_page, 168494afd6d6SChao Yu dn->ofs_in_node + i); 168594afd6d6SChao Yu 168694afd6d6SChao Yu if (!__is_valid_data_blkaddr(blkaddr)) 168794afd6d6SChao Yu break; 168894afd6d6SChao Yu if (first_blkaddr + i - (compressed ? 1 : 0) != blkaddr) 168994afd6d6SChao Yu return 0; 169094afd6d6SChao Yu } 169194afd6d6SChao Yu 169294afd6d6SChao Yu return compressed ? i - 1 : i; 169394afd6d6SChao Yu } 169494afd6d6SChao Yu 16956ce19affSChao Yu const struct address_space_operations f2fs_compress_aops = { 16966ce19affSChao Yu .releasepage = f2fs_release_page, 16976ce19affSChao Yu .invalidatepage = f2fs_invalidate_page, 16986ce19affSChao Yu }; 16996ce19affSChao Yu 17006ce19affSChao Yu struct address_space *COMPRESS_MAPPING(struct f2fs_sb_info *sbi) 17016ce19affSChao Yu { 17026ce19affSChao Yu return sbi->compress_inode->i_mapping; 17036ce19affSChao Yu } 17046ce19affSChao Yu 17056ce19affSChao Yu void f2fs_invalidate_compress_page(struct f2fs_sb_info *sbi, block_t blkaddr) 17066ce19affSChao Yu { 17076ce19affSChao Yu if (!sbi->compress_inode) 17086ce19affSChao Yu return; 17096ce19affSChao Yu invalidate_mapping_pages(COMPRESS_MAPPING(sbi), blkaddr, blkaddr); 17106ce19affSChao Yu } 17116ce19affSChao Yu 17126ce19affSChao Yu void f2fs_cache_compressed_page(struct f2fs_sb_info *sbi, struct page *page, 17136ce19affSChao Yu nid_t ino, block_t blkaddr) 17146ce19affSChao Yu { 17156ce19affSChao Yu struct page *cpage; 17166ce19affSChao Yu int ret; 17176ce19affSChao Yu 17186ce19affSChao Yu if (!test_opt(sbi, COMPRESS_CACHE)) 17196ce19affSChao Yu return; 17206ce19affSChao Yu 17216ce19affSChao Yu if (!f2fs_is_valid_blkaddr(sbi, blkaddr, DATA_GENERIC_ENHANCE_READ)) 17226ce19affSChao Yu return; 17236ce19affSChao Yu 17246ce19affSChao Yu if (!f2fs_available_free_memory(sbi, COMPRESS_PAGE)) 17256ce19affSChao Yu return; 17266ce19affSChao Yu 17276ce19affSChao Yu cpage = find_get_page(COMPRESS_MAPPING(sbi), blkaddr); 17286ce19affSChao Yu if (cpage) { 17296ce19affSChao Yu f2fs_put_page(cpage, 0); 17306ce19affSChao Yu return; 17316ce19affSChao Yu } 17326ce19affSChao Yu 17336ce19affSChao Yu cpage = alloc_page(__GFP_NOWARN | __GFP_IO); 17346ce19affSChao Yu if (!cpage) 17356ce19affSChao Yu return; 17366ce19affSChao Yu 17376ce19affSChao Yu ret = add_to_page_cache_lru(cpage, COMPRESS_MAPPING(sbi), 17386ce19affSChao Yu blkaddr, GFP_NOFS); 17396ce19affSChao Yu if (ret) { 17406ce19affSChao Yu f2fs_put_page(cpage, 0); 17416ce19affSChao Yu return; 17426ce19affSChao Yu } 17436ce19affSChao Yu 17446ce19affSChao Yu set_page_private_data(cpage, ino); 17456ce19affSChao Yu 17466ce19affSChao Yu if (!f2fs_is_valid_blkaddr(sbi, blkaddr, DATA_GENERIC_ENHANCE_READ)) 17476ce19affSChao Yu goto out; 17486ce19affSChao Yu 17496ce19affSChao Yu memcpy(page_address(cpage), page_address(page), PAGE_SIZE); 17506ce19affSChao Yu SetPageUptodate(cpage); 17516ce19affSChao Yu out: 17526ce19affSChao Yu f2fs_put_page(cpage, 1); 17536ce19affSChao Yu } 17546ce19affSChao Yu 17556ce19affSChao Yu bool f2fs_load_compressed_page(struct f2fs_sb_info *sbi, struct page *page, 17566ce19affSChao Yu block_t blkaddr) 17576ce19affSChao Yu { 17586ce19affSChao Yu struct page *cpage; 17596ce19affSChao Yu bool hitted = false; 17606ce19affSChao Yu 17616ce19affSChao Yu if (!test_opt(sbi, COMPRESS_CACHE)) 17626ce19affSChao Yu return false; 17636ce19affSChao Yu 17646ce19affSChao Yu cpage = f2fs_pagecache_get_page(COMPRESS_MAPPING(sbi), 17656ce19affSChao Yu blkaddr, FGP_LOCK | FGP_NOWAIT, GFP_NOFS); 17666ce19affSChao Yu if (cpage) { 17676ce19affSChao Yu if (PageUptodate(cpage)) { 17686ce19affSChao Yu atomic_inc(&sbi->compress_page_hit); 17696ce19affSChao Yu memcpy(page_address(page), 17706ce19affSChao Yu page_address(cpage), PAGE_SIZE); 17716ce19affSChao Yu hitted = true; 17726ce19affSChao Yu } 17736ce19affSChao Yu f2fs_put_page(cpage, 1); 17746ce19affSChao Yu } 17756ce19affSChao Yu 17766ce19affSChao Yu return hitted; 17776ce19affSChao Yu } 17786ce19affSChao Yu 17796ce19affSChao Yu void f2fs_invalidate_compress_pages(struct f2fs_sb_info *sbi, nid_t ino) 17806ce19affSChao Yu { 17816ce19affSChao Yu struct address_space *mapping = sbi->compress_inode->i_mapping; 17826ce19affSChao Yu struct pagevec pvec; 17836ce19affSChao Yu pgoff_t index = 0; 17846ce19affSChao Yu pgoff_t end = MAX_BLKADDR(sbi); 17856ce19affSChao Yu 17866ce19affSChao Yu if (!mapping->nrpages) 17876ce19affSChao Yu return; 17886ce19affSChao Yu 17896ce19affSChao Yu pagevec_init(&pvec); 17906ce19affSChao Yu 17916ce19affSChao Yu do { 17926ce19affSChao Yu unsigned int nr_pages; 17936ce19affSChao Yu int i; 17946ce19affSChao Yu 17956ce19affSChao Yu nr_pages = pagevec_lookup_range(&pvec, mapping, 17966ce19affSChao Yu &index, end - 1); 17976ce19affSChao Yu if (!nr_pages) 17986ce19affSChao Yu break; 17996ce19affSChao Yu 18006ce19affSChao Yu for (i = 0; i < nr_pages; i++) { 18016ce19affSChao Yu struct page *page = pvec.pages[i]; 18026ce19affSChao Yu 18036ce19affSChao Yu if (page->index > end) 18046ce19affSChao Yu break; 18056ce19affSChao Yu 18066ce19affSChao Yu lock_page(page); 18076ce19affSChao Yu if (page->mapping != mapping) { 18086ce19affSChao Yu unlock_page(page); 18096ce19affSChao Yu continue; 18106ce19affSChao Yu } 18116ce19affSChao Yu 18126ce19affSChao Yu if (ino != get_page_private_data(page)) { 18136ce19affSChao Yu unlock_page(page); 18146ce19affSChao Yu continue; 18156ce19affSChao Yu } 18166ce19affSChao Yu 18176ce19affSChao Yu generic_error_remove_page(mapping, page); 18186ce19affSChao Yu unlock_page(page); 18196ce19affSChao Yu } 18206ce19affSChao Yu pagevec_release(&pvec); 18216ce19affSChao Yu cond_resched(); 18226ce19affSChao Yu } while (index < end); 18236ce19affSChao Yu } 18246ce19affSChao Yu 18256ce19affSChao Yu int f2fs_init_compress_inode(struct f2fs_sb_info *sbi) 18266ce19affSChao Yu { 18276ce19affSChao Yu struct inode *inode; 18286ce19affSChao Yu 18296ce19affSChao Yu if (!test_opt(sbi, COMPRESS_CACHE)) 18306ce19affSChao Yu return 0; 18316ce19affSChao Yu 18326ce19affSChao Yu inode = f2fs_iget(sbi->sb, F2FS_COMPRESS_INO(sbi)); 18336ce19affSChao Yu if (IS_ERR(inode)) 18346ce19affSChao Yu return PTR_ERR(inode); 18356ce19affSChao Yu sbi->compress_inode = inode; 18366ce19affSChao Yu 18376ce19affSChao Yu sbi->compress_percent = COMPRESS_PERCENT; 18386ce19affSChao Yu sbi->compress_watermark = COMPRESS_WATERMARK; 18396ce19affSChao Yu 18406ce19affSChao Yu atomic_set(&sbi->compress_page_hit, 0); 18416ce19affSChao Yu 18426ce19affSChao Yu return 0; 18436ce19affSChao Yu } 18446ce19affSChao Yu 18456ce19affSChao Yu void f2fs_destroy_compress_inode(struct f2fs_sb_info *sbi) 18466ce19affSChao Yu { 18476ce19affSChao Yu if (!sbi->compress_inode) 18486ce19affSChao Yu return; 18496ce19affSChao Yu iput(sbi->compress_inode); 18506ce19affSChao Yu sbi->compress_inode = NULL; 18516ce19affSChao Yu } 18526ce19affSChao Yu 185331083031SChao Yu int f2fs_init_page_array_cache(struct f2fs_sb_info *sbi) 185431083031SChao Yu { 185531083031SChao Yu dev_t dev = sbi->sb->s_bdev->bd_dev; 185631083031SChao Yu char slab_name[32]; 185731083031SChao Yu 185831083031SChao Yu sprintf(slab_name, "f2fs_page_array_entry-%u:%u", MAJOR(dev), MINOR(dev)); 185931083031SChao Yu 186031083031SChao Yu sbi->page_array_slab_size = sizeof(struct page *) << 186131083031SChao Yu F2FS_OPTION(sbi).compress_log_size; 186231083031SChao Yu 186331083031SChao Yu sbi->page_array_slab = f2fs_kmem_cache_create(slab_name, 186431083031SChao Yu sbi->page_array_slab_size); 186531083031SChao Yu if (!sbi->page_array_slab) 186631083031SChao Yu return -ENOMEM; 186731083031SChao Yu return 0; 186831083031SChao Yu } 186931083031SChao Yu 187031083031SChao Yu void f2fs_destroy_page_array_cache(struct f2fs_sb_info *sbi) 187131083031SChao Yu { 187231083031SChao Yu kmem_cache_destroy(sbi->page_array_slab); 187331083031SChao Yu } 1874c68d6c88SChao Yu 1875c68d6c88SChao Yu static int __init f2fs_init_cic_cache(void) 1876c68d6c88SChao Yu { 1877c68d6c88SChao Yu cic_entry_slab = f2fs_kmem_cache_create("f2fs_cic_entry", 1878c68d6c88SChao Yu sizeof(struct compress_io_ctx)); 1879c68d6c88SChao Yu if (!cic_entry_slab) 1880c68d6c88SChao Yu return -ENOMEM; 1881c68d6c88SChao Yu return 0; 1882c68d6c88SChao Yu } 1883c68d6c88SChao Yu 1884c68d6c88SChao Yu static void f2fs_destroy_cic_cache(void) 1885c68d6c88SChao Yu { 1886c68d6c88SChao Yu kmem_cache_destroy(cic_entry_slab); 1887c68d6c88SChao Yu } 1888c68d6c88SChao Yu 1889c68d6c88SChao Yu static int __init f2fs_init_dic_cache(void) 1890c68d6c88SChao Yu { 1891c68d6c88SChao Yu dic_entry_slab = f2fs_kmem_cache_create("f2fs_dic_entry", 1892c68d6c88SChao Yu sizeof(struct decompress_io_ctx)); 1893c68d6c88SChao Yu if (!dic_entry_slab) 1894c68d6c88SChao Yu return -ENOMEM; 1895c68d6c88SChao Yu return 0; 1896c68d6c88SChao Yu } 1897c68d6c88SChao Yu 1898c68d6c88SChao Yu static void f2fs_destroy_dic_cache(void) 1899c68d6c88SChao Yu { 1900c68d6c88SChao Yu kmem_cache_destroy(dic_entry_slab); 1901c68d6c88SChao Yu } 1902c68d6c88SChao Yu 1903c68d6c88SChao Yu int __init f2fs_init_compress_cache(void) 1904c68d6c88SChao Yu { 1905c68d6c88SChao Yu int err; 1906c68d6c88SChao Yu 1907c68d6c88SChao Yu err = f2fs_init_cic_cache(); 1908c68d6c88SChao Yu if (err) 1909c68d6c88SChao Yu goto out; 1910c68d6c88SChao Yu err = f2fs_init_dic_cache(); 1911c68d6c88SChao Yu if (err) 1912c68d6c88SChao Yu goto free_cic; 1913c68d6c88SChao Yu return 0; 1914c68d6c88SChao Yu free_cic: 1915c68d6c88SChao Yu f2fs_destroy_cic_cache(); 1916c68d6c88SChao Yu out: 1917c68d6c88SChao Yu return -ENOMEM; 1918c68d6c88SChao Yu } 1919c68d6c88SChao Yu 1920c68d6c88SChao Yu void f2fs_destroy_compress_cache(void) 1921c68d6c88SChao Yu { 1922c68d6c88SChao Yu f2fs_destroy_dic_cache(); 1923c68d6c88SChao Yu f2fs_destroy_cic_cache(); 1924c68d6c88SChao Yu } 1925