[linux-block.git] / fs / btrfs / compression.h

/* SPDX-License-Identifier: GPL-2.0 */
/*
 * Copyright (C) 2008 Oracle.  All rights reserved.
 */

#ifndef BTRFS_COMPRESSION_H
#define BTRFS_COMPRESSION_H

#include <linux/sizes.h>
#include "bio.h"

struct btrfs_inode;

/*
 * We want to make sure that amount of RAM required to uncompress an extent is
 * reasonable, so we limit the total size in ram of a compressed extent to
 * 128k.  This is a crucial number because it also controls how easily we can
 * spread reads across cpus for decompression.
 *
 * We also want to make sure the amount of IO required to do a random read is
 * reasonably small, so we limit the size of a compressed extent to 128k.
 */

/* Maximum length of compressed data stored on disk */
#define BTRFS_MAX_COMPRESSED		(SZ_128K)
#define BTRFS_MAX_COMPRESSED_PAGES	(BTRFS_MAX_COMPRESSED / PAGE_SIZE)
static_assert((BTRFS_MAX_COMPRESSED % PAGE_SIZE) == 0);

/* Maximum size of data before compression */
#define BTRFS_MAX_UNCOMPRESSED		(SZ_128K)

#define	BTRFS_ZLIB_DEFAULT_LEVEL		3

struct compressed_bio {
	/* Number of compressed pages in the array */
	unsigned int nr_pages;

	/* the pages with the compressed data on them */
	struct page **compressed_pages;

	/* starting offset in the inode for our pages */
	u64 start;

	/* Number of bytes in the inode we're working on */
	unsigned int len;

	/* Number of bytes on disk */
	unsigned int compressed_len;

	/* The compression algorithm for this bio */
	u8 compress_type;

	/* Whether this is a write for writeback. */
	bool writeback;

	union {
		/* For reads, this is the bio we are copying the data into */
		struct btrfs_bio *orig_bbio;
		struct work_struct write_end_work;
	};

	/* Must be last. */
	struct btrfs_bio bbio;
};

static inline unsigned int btrfs_compress_type(unsigned int type_level)
{
	return (type_level & 0xF);
}

static inline unsigned int btrfs_compress_level(unsigned int type_level)
{
	return ((type_level & 0xF0) >> 4);
}

int __init btrfs_init_compress(void);
void __cold btrfs_exit_compress(void);

int btrfs_compress_pages(unsigned int type_level, struct address_space *mapping,
			 u64 start, struct page **pages,
			 unsigned long *out_pages,
			 unsigned long *total_in,
			 unsigned long *total_out);
int btrfs_decompress(int type, const u8 *data_in, struct page *dest_page,
		     unsigned long start_byte, size_t srclen, size_t destlen);
int btrfs_decompress_buf2page(const char *buf, u32 buf_len,
			      struct compressed_bio *cb, u32 decompressed);

void btrfs_submit_compressed_write(struct btrfs_inode *inode, u64 start,
				  unsigned int len, u64 disk_start,
				  unsigned int compressed_len,
				  struct page **compressed_pages,
				  unsigned int nr_pages,
				  blk_opf_t write_flags,
				  bool writeback);
void btrfs_submit_compressed_read(struct btrfs_bio *bbio, int mirror_num);

unsigned int btrfs_compress_str2level(unsigned int type, const char *str);

enum btrfs_compression_type {
	BTRFS_COMPRESS_NONE  = 0,
	BTRFS_COMPRESS_ZLIB  = 1,
	BTRFS_COMPRESS_LZO   = 2,
	BTRFS_COMPRESS_ZSTD  = 3,
	BTRFS_NR_COMPRESS_TYPES = 4,
};

struct workspace_manager {
	struct list_head idle_ws;
	spinlock_t ws_lock;
	/* Number of free workspaces */
	int free_ws;
	/* Total number of allocated workspaces */
	atomic_t total_ws;
	/* Waiters for a free workspace */
	wait_queue_head_t ws_wait;
};

struct list_head *btrfs_get_workspace(int type, unsigned int level);
void btrfs_put_workspace(int type, struct list_head *ws);

struct btrfs_compress_op {
	struct workspace_manager *workspace_manager;
	/* Maximum level supported by the compression algorithm */
	unsigned int max_level;
	unsigned int default_level;
};

/* The heuristic workspaces are managed via the 0th workspace manager */
#define BTRFS_NR_WORKSPACE_MANAGERS	BTRFS_NR_COMPRESS_TYPES

extern const struct btrfs_compress_op btrfs_heuristic_compress;
extern const struct btrfs_compress_op btrfs_zlib_compress;
extern const struct btrfs_compress_op btrfs_lzo_compress;
extern const struct btrfs_compress_op btrfs_zstd_compress;

const char* btrfs_compress_type2str(enum btrfs_compression_type type);
bool btrfs_compress_is_valid_type(const char *str, size_t len);

int btrfs_compress_heuristic(struct inode *inode, u64 start, u64 end);

int zlib_compress_pages(struct list_head *ws, struct address_space *mapping,
		u64 start, struct page **pages, unsigned long *out_pages,
		unsigned long *total_in, unsigned long *total_out);
int zlib_decompress_bio(struct list_head *ws, struct compressed_bio *cb);
int zlib_decompress(struct list_head *ws, const u8 *data_in,
		struct page *dest_page, unsigned long start_byte, size_t srclen,
		size_t destlen);
struct list_head *zlib_alloc_workspace(unsigned int level);
void zlib_free_workspace(struct list_head *ws);
struct list_head *zlib_get_workspace(unsigned int level);

int lzo_compress_pages(struct list_head *ws, struct address_space *mapping,
		u64 start, struct page **pages, unsigned long *out_pages,
		unsigned long *total_in, unsigned long *total_out);
int lzo_decompress_bio(struct list_head *ws, struct compressed_bio *cb);
int lzo_decompress(struct list_head *ws, const u8 *data_in,
		struct page *dest_page, unsigned long start_byte, size_t srclen,
		size_t destlen);
struct list_head *lzo_alloc_workspace(unsigned int level);
void lzo_free_workspace(struct list_head *ws);

int zstd_compress_pages(struct list_head *ws, struct address_space *mapping,
		u64 start, struct page **pages, unsigned long *out_pages,
		unsigned long *total_in, unsigned long *total_out);
int zstd_decompress_bio(struct list_head *ws, struct compressed_bio *cb);
int zstd_decompress(struct list_head *ws, const u8 *data_in,
		struct page *dest_page, unsigned long start_byte, size_t srclen,
		size_t destlen);
void zstd_init_workspace_manager(void);
void zstd_cleanup_workspace_manager(void);
struct list_head *zstd_alloc_workspace(unsigned int level);
void zstd_free_workspace(struct list_head *ws);
struct list_head *zstd_get_workspace(unsigned int level);
void zstd_put_workspace(struct list_head *ws);

#endif
Commit	Line	Data
9888c340	1	/* SPDX-License-Identifier: GPL-2.0 */
c8b97818 CM	2	/*
c8b97818 CM	3	* Copyright (C) 2008 Oracle. All rights reserved.
c8b97818 CM	4	*/
c8b97818 CM	5
9888c340 DS	6	#ifndef BTRFS_COMPRESSION_H
9888c340 DS	7	#define BTRFS_COMPRESSION_H
c8b97818	8
d5c1d68f	9	#include <linux/sizes.h>
544fe4a9	10	#include "bio.h"
d5c1d68f	11
c7ee1819 NB	12	struct btrfs_inode;
c7ee1819 NB	13
ff763866 DS	14	/*
	15	* We want to make sure that amount of RAM required to uncompress an extent is
	16	* reasonable, so we limit the total size in ram of a compressed extent to
	17	* 128k. This is a crucial number because it also controls how easily we can
	18	* spread reads across cpus for decompression.
	19	*
	20	* We also want to make sure the amount of IO required to do a random read is
	21	* reasonably small, so we limit the size of a compressed extent to 128k.
	22	*/
	23
	24	/* Maximum length of compressed data stored on disk */
	25	#define BTRFS_MAX_COMPRESSED (SZ_128K)
544fe4a9	26	#define BTRFS_MAX_COMPRESSED_PAGES (BTRFS_MAX_COMPRESSED / PAGE_SIZE)
a55e65b8 DS	27	static_assert((BTRFS_MAX_COMPRESSED % PAGE_SIZE) == 0);
a55e65b8 DS	28
ff763866 DS	29	/* Maximum size of data before compression */
	30	#define BTRFS_MAX_UNCOMPRESSED (SZ_128K)
	31
eae8d825 QW	32	#define BTRFS_ZLIB_DEFAULT_LEVEL 3
eae8d825 QW	33
e1ddce71	34	struct compressed_bio {
282ab3ff DS	35	/* Number of compressed pages in the array */
	36	unsigned int nr_pages;
	37
e1ddce71 AJ	38	/* the pages with the compressed data on them */
	39	struct page **compressed_pages;
	40
e1ddce71 AJ	41	/* starting offset in the inode for our pages */
	42	u64 start;
	43
282ab3ff DS	44	/* Number of bytes in the inode we're working on */
282ab3ff DS	45	unsigned int len;
e1ddce71	46
282ab3ff DS	47	/* Number of bytes on disk */
282ab3ff DS	48	unsigned int compressed_len;
e1ddce71	49
282ab3ff DS	50	/* The compression algorithm for this bio */
282ab3ff DS	51	u8 compress_type;
e1ddce71	52
7c0c7269 OS	53	/* Whether this is a write for writeback. */
	54	bool writeback;
	55
fed8a72d CH	56	union {
fed8a72d CH	57	/* For reads, this is the bio we are copying the data into */
b7d463a1	58	struct btrfs_bio *orig_bbio;
fed8a72d CH	59	struct work_struct write_end_work;
fed8a72d CH	60	};
544fe4a9 CH	61
	62	/* Must be last. */
	63	struct btrfs_bio bbio;
e1ddce71 AJ	64	};
e1ddce71 AJ	65
1972708a DZ	66	static inline unsigned int btrfs_compress_type(unsigned int type_level)
	67	{
	68	return (type_level & 0xF);
	69	}
	70
	71	static inline unsigned int btrfs_compress_level(unsigned int type_level)
	72	{
	73	return ((type_level & 0xF0) >> 4);
	74	}
	75
5565b8e0	76	int __init btrfs_init_compress(void);
e67c718b	77	void __cold btrfs_exit_compress(void);
261507a0	78
f51d2b59	79	int btrfs_compress_pages(unsigned int type_level, struct address_space *mapping,
38c31464	80	u64 start, struct page **pages,
261507a0 LZ	81	unsigned long *out_pages,
261507a0 LZ	82	unsigned long *total_in,
e5d74902	83	unsigned long *total_out);
3e09b5b2	84	int btrfs_decompress(int type, const u8 data_in, struct page dest_page,
261507a0	85	unsigned long start_byte, size_t srclen, size_t destlen);
1c3dc173 QW	86	int btrfs_decompress_buf2page(const char *buf, u32 buf_len,
1c3dc173 QW	87	struct compressed_bio *cb, u32 decompressed);
261507a0	88
544fe4a9	89	void btrfs_submit_compressed_write(struct btrfs_inode *inode, u64 start,
65b5355f AJ	90	unsigned int len, u64 disk_start,
65b5355f AJ	91	unsigned int compressed_len,
c8b97818	92	struct page **compressed_pages,
65b5355f	93	unsigned int nr_pages,
bf9486d6	94	blk_opf_t write_flags,
7c0c7269	95	bool writeback);
690834e4	96	void btrfs_submit_compressed_read(struct btrfs_bio *bbio, int mirror_num);
ebb8765b	97
d0ab62ce	98	unsigned int btrfs_compress_str2level(unsigned int type, const char *str);
f51d2b59	99
ebb8765b AJ	100	enum btrfs_compression_type {
	101	BTRFS_COMPRESS_NONE = 0,
	102	BTRFS_COMPRESS_ZLIB = 1,
	103	BTRFS_COMPRESS_LZO = 2,
5c1aab1d	104	BTRFS_COMPRESS_ZSTD = 3,
ce96b7ff	105	BTRFS_NR_COMPRESS_TYPES = 4,
ebb8765b AJ	106	};
ebb8765b AJ	107
92ee5530	108	struct workspace_manager {
92ee5530 DZ	109	struct list_head idle_ws;
	110	spinlock_t ws_lock;
	111	/* Number of free workspaces */
	112	int free_ws;
	113	/* Total number of allocated workspaces */
	114	atomic_t total_ws;
	115	/* Waiters for a free workspace */
	116	wait_queue_head_t ws_wait;
	117	};
	118
5907a9bb	119	struct list_head *btrfs_get_workspace(int type, unsigned int level);
a3bbd2a9	120	void btrfs_put_workspace(int type, struct list_head *ws);
92ee5530	121
261507a0	122	struct btrfs_compress_op {
be951045	123	struct workspace_manager *workspace_manager;
e18333a7 DS	124	/* Maximum level supported by the compression algorithm */
	125	unsigned int max_level;
	126	unsigned int default_level;
261507a0 LZ	127	};
261507a0 LZ	128
ca4ac360	129	/* The heuristic workspaces are managed via the 0th workspace manager */
ce96b7ff	130	#define BTRFS_NR_WORKSPACE_MANAGERS BTRFS_NR_COMPRESS_TYPES
ca4ac360 DZ	131
ca4ac360 DZ	132	extern const struct btrfs_compress_op btrfs_heuristic_compress;
e8c9f186 DS	133	extern const struct btrfs_compress_op btrfs_zlib_compress;
e8c9f186 DS	134	extern const struct btrfs_compress_op btrfs_lzo_compress;
5c1aab1d	135	extern const struct btrfs_compress_op btrfs_zstd_compress;
261507a0	136
e128f9c3	137	const char* btrfs_compress_type2str(enum btrfs_compression_type type);
aa53e3bf	138	bool btrfs_compress_is_valid_type(const char *str, size_t len);
e128f9c3	139
c2fcdcdf TT	140	int btrfs_compress_heuristic(struct inode *inode, u64 start, u64 end);
c2fcdcdf TT	141
cb4c9198 DS	142	int zlib_compress_pages(struct list_head ws, struct address_space mapping,
	143	u64 start, struct page *pages, unsigned long out_pages,
	144	unsigned long total_in, unsigned long total_out);
	145	int zlib_decompress_bio(struct list_head ws, struct compressed_bio cb);
3e09b5b2	146	int zlib_decompress(struct list_head ws, const u8 data_in,
cb4c9198 DS	147	struct page *dest_page, unsigned long start_byte, size_t srclen,
	148	size_t destlen);
	149	struct list_head *zlib_alloc_workspace(unsigned int level);
	150	void zlib_free_workspace(struct list_head *ws);
	151	struct list_head *zlib_get_workspace(unsigned int level);
	152
	153	int lzo_compress_pages(struct list_head ws, struct address_space mapping,
	154	u64 start, struct page *pages, unsigned long out_pages,
	155	unsigned long total_in, unsigned long total_out);
	156	int lzo_decompress_bio(struct list_head ws, struct compressed_bio cb);
3e09b5b2	157	int lzo_decompress(struct list_head ws, const u8 data_in,
cb4c9198 DS	158	struct page *dest_page, unsigned long start_byte, size_t srclen,
	159	size_t destlen);
	160	struct list_head *lzo_alloc_workspace(unsigned int level);
	161	void lzo_free_workspace(struct list_head *ws);
	162
	163	int zstd_compress_pages(struct list_head ws, struct address_space mapping,
	164	u64 start, struct page *pages, unsigned long out_pages,
	165	unsigned long total_in, unsigned long total_out);
	166	int zstd_decompress_bio(struct list_head ws, struct compressed_bio cb);
3e09b5b2	167	int zstd_decompress(struct list_head ws, const u8 data_in,
cb4c9198 DS	168	struct page *dest_page, unsigned long start_byte, size_t srclen,
	169	size_t destlen);
	170	void zstd_init_workspace_manager(void);
	171	void zstd_cleanup_workspace_manager(void);
	172	struct list_head *zstd_alloc_workspace(unsigned int level);
	173	void zstd_free_workspace(struct list_head *ws);
	174	struct list_head *zstd_get_workspace(unsigned int level);
	175	void zstd_put_workspace(struct list_head *ws);
	176
c8b97818	177	#endif