#ifndef _BCACHE_IO_H
#define _BCACHE_IO_H

#include "io_types.h"

#include <linux/lz4.h>
#include <linux/zlib.h>

#define COMPRESSION_WORKSPACE_SIZE					\
	max_t(size_t, zlib_deflate_workspacesize(MAX_WBITS, DEF_MEM_LEVEL),\
	max_t(size_t, zlib_inflate_workspacesize(),			\
		      LZ4HC_MEM_COMPRESS))

#define to_bbio(_bio)		container_of((_bio), struct bbio, bio)

#define to_wbio(_bio)			\
	container_of((_bio), struct bch_write_bio, bio.bio)

void bch_bio_free_pages_pool(struct cache_set *, struct bio *);
void bch_bio_alloc_pages_pool(struct cache_set *, struct bio *, size_t);

enum bch_write_flags {
	BCH_WRITE_ALLOC_NOWAIT		= (1 << 0),
	BCH_WRITE_DISCARD		= (1 << 1),
	BCH_WRITE_CACHED		= (1 << 2),
	BCH_WRITE_FLUSH			= (1 << 3),
	BCH_WRITE_DISCARD_ON_ERROR	= (1 << 4),
	BCH_WRITE_DATA_COMPRESSED	= (1 << 5),

	/* Internal: */
	BCH_WRITE_JOURNAL_SEQ_PTR	= (1 << 6),
	BCH_WRITE_DONE			= (1 << 7),
	BCH_WRITE_LOOPED		= (1 << 8),
};

static inline u64 *op_journal_seq(struct bch_write_op *op)
{
	return (op->flags & BCH_WRITE_JOURNAL_SEQ_PTR)
		? op->journal_seq_p : &op->journal_seq;
}

void bch_write_op_init(struct bch_write_op *, struct cache_set *,
		       struct bch_write_bio *,
		       struct disk_reservation, struct write_point *,
		       struct bkey_s_c,
		       struct extent_insert_hook *, u64 *, unsigned);
void bch_write(struct closure *);

void bch_replace_init(struct bch_replace_info *, struct bkey_s_c);

struct cache_promote_op;

struct extent_pick_ptr;

void bch_read_extent_iter(struct cache_set *, struct bch_read_bio *,
			  struct bvec_iter, struct bkey_s_c k,
			  struct extent_pick_ptr *, unsigned);

static inline void bch_read_extent(struct cache_set *c,
				   struct bch_read_bio *orig,
				   struct bkey_s_c k,
				   struct extent_pick_ptr *pick,
				   unsigned flags)
{
	bch_read_extent_iter(c, orig, orig->bio.bi_iter,
			     k, pick, flags);
}

enum bch_read_flags {
	BCH_READ_FORCE_BOUNCE		= 1 << 0,
	BCH_READ_RETRY_IF_STALE		= 1 << 1,
	BCH_READ_PROMOTE		= 1 << 2,
	BCH_READ_IS_LAST		= 1 << 3,
	BCH_READ_MAY_REUSE_BIO		= 1 << 4,
};

void bch_read(struct cache_set *, struct bch_read_bio *, u64);

void bch_bbio_endio(struct bbio *);

void bch_generic_make_request(struct bio *, struct cache_set *);
void bch_bio_submit_work(struct work_struct *);
void bch_submit_bbio(struct bbio *, struct cache *,
		     const struct bch_extent_ptr *, bool);
void bch_submit_bbio_replicas(struct bch_write_bio *, struct cache_set *,
			      const struct bkey_i *, unsigned, bool);

int bch_discard(struct cache_set *, struct bpos, struct bpos,
		u64, struct disk_reservation *,
		struct extent_insert_hook *, u64 *);

void __cache_promote(struct cache_set *, struct bbio *,
		     struct bkey_s_c, struct bkey_s_c, unsigned);
bool cache_promote(struct cache_set *, struct bbio *, struct bkey_s_c);

void bch_read_retry_work(struct work_struct *);
void bch_wake_delayed_writes(unsigned long data);

void bch_bio_decompress_work(struct work_struct *);

#endif /* _BCACHE_IO_H */