mars/mars_buf.c

// (c) 2010 Thomas Schoebel-Theuer / 1&1 Internet AG

// Buf brick

//#define BRICK_DEBUGGING
//#define MARS_DEBUGGING

#include <linux/kernel.h>
#include <linux/module.h>
#include <linux/string.h>
#include <linux/bio.h>
#include <linux/blkdev.h>
#include <linux/delay.h>

#include "mars.h"

///////////////////////// own type definitions ////////////////////////

#include "mars_buf.h"

///////////////////////// own helper functions ////////////////////////

static inline int buf_hash(struct buf_brick *brick, loff_t pos)
{
	return (pos >> brick->backing_order) % MARS_BUF_HASH_MAX;
}

static struct buf_head *hash_find(struct buf_brick *brick, loff_t pos)
{
	int hash = buf_hash(brick, pos);
	struct list_head *start = &brick->cache_anchors[hash];
	struct list_head *tmp;
	struct buf_head *res;
	for (tmp = start->next; ; tmp = tmp->next) {
		if (tmp == start)
			return NULL;
		res = container_of(tmp, struct buf_head, bf_hash_head);
		if (res->bf_pos == pos)
			break;
	}
	return res;
}

static inline void hash_insert(struct buf_brick *brick, struct buf_head *elem)
{
	int hash = buf_hash(brick, elem->bf_pos);
	struct list_head *start = &brick->cache_anchors[hash];
	list_add(&elem->bf_hash_head, start);
}

static inline void free_buf(struct buf_brick *brick, struct buf_head *bf)
{
	free_pages((unsigned long)bf->bf_data, brick->backing_order);
	kfree(bf);
}

/* brick->buf_lock must be held
 */
static inline void __prune_cache(struct buf_brick *brick, int max_count, unsigned long *flags)
{
#if 1
	return;
#endif
	while (brick->alloc_count > max_count) {
		struct buf_head *bf;
		if (list_empty(&brick->free_anchor))
			break;
		bf = container_of(brick->free_anchor.next, struct buf_head, bf_lru_head);
		list_del_init(&bf->bf_lru_head);
		brick->current_count--;
		brick->alloc_count--;
		
		traced_unlock(&brick->buf_lock, *flags);
		free_buf(brick, bf);
		traced_lock(&brick->buf_lock, *flags);
	}
}

static inline void __lru_free(struct buf_brick *brick)
{
	while (brick->current_count > brick->max_count) {
		struct buf_head *bf;
		if (list_empty(&brick->lru_anchor))
			break;
		bf = container_of(brick->lru_anchor.prev, struct buf_head, bf_lru_head);
		list_del_init(&bf->bf_lru_head);
		list_del_init(&bf->bf_hash_head);
		brick->current_count--;
		list_add(&bf->bf_lru_head, &brick->free_anchor);
	}
}


static inline int get_info(struct buf_brick *brick)
{
	struct buf_input *input = brick->inputs[0];
	int status = GENERIC_INPUT_CALL(input, mars_get_info, &brick->base_info);
	if (status >= 0) {
		brick->got_info = 1;
	}
	return status;
}

/* Convert from arbitrary/odd kernel address/length to struct page,
 * create bio from it, round up/down to full sectors.
 * return the length (may be smaller or even larger than requested)
 */
static int make_bio(struct buf_brick *brick, struct bio **_bio, void *data, loff_t pos, int len)
{
	unsigned long long sector;
	int sector_offset;
	int page_offset;
	int page_len;
	int bvec_count;
	int status;
	int i;
	struct page *page;
	struct bio *bio = NULL;
	struct block_device *bdev;

	if (unlikely(!brick->got_info)) {
		struct request_queue *q;
		status = get_info(brick);
		if (status < 0)
			goto out;
		bdev = brick->base_info.backing_file->f_mapping->host->i_sb->s_bdev;
		q = bdev_get_queue(bdev);
		brick->bvec_max = queue_max_hw_sectors(q) >> (PAGE_SHIFT - 9);
	} else {
		bdev = brick->base_info.backing_file->f_mapping->host->i_sb->s_bdev;
	}

	if (unlikely(len <= 0)) {
		MARS_ERR("bad bio len %d\n", len);
		status = -EINVAL;
		goto out;
	}

	sector = pos >> 9;                     // TODO: make dynamic
	sector_offset = pos & ((1 << 9) - 1);  // TODO: make dynamic

	// round down to start of first sector
	data -= sector_offset;
	len += sector_offset;
	pos -= sector_offset;

	// round up to full sector
	len = (((len - 1) >> 9) + 1) << 9; // TODO: make dynamic

	// map onto pages. TODO: allow higher-order pages (performance!)
	page_offset = pos & (PAGE_SIZE - 1);
	page_len = len + page_offset;
	bvec_count = (page_len - 1) / PAGE_SIZE + 1;
	if (bvec_count > brick->bvec_max)
		bvec_count = brick->bvec_max;

	bio = bio_alloc(GFP_MARS, bvec_count);
	status = -ENOMEM;
	if (!bio)
		goto out;

	status = 0;
	for (i = 0; i < bvec_count && len > 0; i++) {
		int myrest = PAGE_SIZE - page_offset;
		int mylen = len;

		if (mylen > myrest)
			mylen = myrest;

		page = virt_to_page(data);

		bio->bi_io_vec[i].bv_page = page;
		bio->bi_io_vec[i].bv_len = mylen;
		bio->bi_io_vec[i].bv_offset = page_offset;

		data += mylen;
		len -= mylen;
		status += mylen;
		page_offset = 0;
	}

	if (unlikely(len)) {
		bio_put(bio);
		bio = NULL;
		MARS_ERR("computation of bvec_count %d was wrong, diff=%d\n", bvec_count, len);
		status = -EIO;
		goto out;
	}

	bio->bi_vcnt = i;
	bio->bi_idx = 0;
	bio->bi_size = status;
	bio->bi_sector = sector;
	bio->bi_bdev = bdev;
	bio->bi_private = NULL; // must be filled in later
	bio->bi_end_io = NULL; // must be filled in later
	bio->bi_rw = 0; // must be filled in later
	// ignore rounding-down on return
	if (status >= sector_offset)
		status -= sector_offset;

out:
	*_bio = bio;
	return status;
}

////////////////// own brick / input / output operations //////////////////

static int buf_get_info(struct buf_output *output, struct mars_info *info)
{
	struct buf_input *input = output->brick->inputs[0];
	return GENERIC_INPUT_CALL(input, mars_get_info, info);
}

static int buf_buf_get(struct buf_output *output, struct mars_buf_object *mbuf)
{
	struct buf_brick *brick = output->brick;
	struct buf_mars_buf_aspect *mbuf_a;
	struct buf_head *bf;
	loff_t base_pos;
	int base_offset;
	int max_len;
	int buf_count;
	unsigned long flags;
	int status = -EILSEQ;

	might_sleep();

	buf_count = atomic_read(&mbuf->buf_count);
	if (unlikely(buf_count != 0)) {
		MARS_ERR("buf_count %d is not zero. probably calling buf_get() on the wrong mars_buf_object instance?\n", buf_count);
	}
	atomic_set(&mbuf->buf_count, 1);

	mbuf_a = buf_mars_buf_get_aspect(output, mbuf);
	if (unlikely(!mbuf_a))
		goto done;
	
	base_pos = mbuf->buf_pos & ~(loff_t)(brick->backing_size - 1);
	base_offset = (mbuf->buf_pos - base_pos);
	if (unlikely(base_offset < 0 || base_offset >= brick->backing_size)) {
		MARS_ERR("bad base_offset %d\n", base_offset);
	}

	max_len = brick->backing_size - base_offset;
	if (mbuf->buf_len > max_len)
		mbuf->buf_len = max_len;

	traced_lock(&brick->buf_lock, flags);
	bf = hash_find(brick, base_pos);
	if (!bf) {
		MARS_DBG("buf_get() hash nothing found\n");
		if (unlikely(list_empty(&brick->free_anchor))) {
			struct buf_head *test_bf;
			MARS_DBG("buf_get() alloc new buf_head\n");

			traced_unlock(&brick->buf_lock, flags);

			status = -ENOMEM;
			bf = kzalloc(sizeof(struct buf_head), GFP_MARS);
			if (!bf)
				goto done;

			bf->bf_data = (void*)__get_free_pages(GFP_MARS, brick->backing_order);
			if (!bf->bf_data)
				goto err_free;

			bf->bf_brick = brick;
			atomic_set(&bf->bf_bio_count, 0);
			//INIT_LIST_HEAD(&bf->bf_mbuf_anchor);
			INIT_LIST_HEAD(&bf->bf_lru_head);
			INIT_LIST_HEAD(&bf->bf_hash_head);
			INIT_LIST_HEAD(&bf->bf_io_pending_anchor);
			INIT_LIST_HEAD(&bf->bf_again_write_pending_anchor);

			traced_lock(&brick->buf_lock, flags);

			brick->alloc_count++;
			/* during the open lock, somebody might have raced
			 * against us at the same base_pos...
			 */
			test_bf = hash_find(brick, base_pos);
			if (unlikely(test_bf)) {
				free_buf(brick, bf);
				bf = test_bf;
			}
		} else {
			bf = container_of(brick->free_anchor.next, struct buf_head, bf_lru_head);
		}
		MARS_DBG("buf_get() bf=%p\n", bf);
		bf->bf_pos = base_pos;
		bf->bf_flags = 0;
		atomic_set(&bf->bf_count, 0);

		hash_insert(brick, bf);
		brick->current_count++;
	}

	mbuf_a->bfa_bf = bf;
	atomic_inc(&bf->bf_count);
	MARS_DBG("bf=%p initial bf_count=%d\n", bf, atomic_read(&bf->bf_count));

	list_del_init(&bf->bf_lru_head);
	mbuf->buf_flags = bf->bf_flags;

	traced_unlock(&brick->buf_lock, flags);

	mbuf->buf_data = bf->bf_data + base_offset;

	return mbuf->buf_len;

err_free:
	kfree(bf);
done:
	return status;
}

static void __bf_put(struct buf_head *bf)
{
	struct buf_brick *brick;
	unsigned long flags;
	int bf_count = atomic_read(&bf->bf_count);

	MARS_DBG("bf=%p bf_count=%d\n", bf, bf_count);
	if (bf_count <= 0) {
		MARS_ERR("bf_count UNDERRUN %d\n", bf_count);
		atomic_set(&bf->bf_count, 1);
	}

	if (!atomic_dec_and_test(&bf->bf_count))
		return;

	MARS_DBG("ZERO_COUNT\n");

	brick = bf->bf_brick;

	traced_lock(&brick->buf_lock, flags);

        /* NOTE: this may race against the above atomic_dec_and_test().
	 * But in worst case, nothing happens.
	 * So this race is ok.
	 */
	bf_count = atomic_read(&bf->bf_count);
	if (likely(bf_count <= 0)) {
		struct list_head *where = &brick->lru_anchor;
		if (unlikely(bf_count < 0)) {
			atomic_set(&bf->bf_count, 0);
			MARS_ERR("bf_count UNDERRUN %d\n", bf_count);
		}
#if 1
		if (unlikely(!list_empty(&bf->bf_io_pending_anchor))) {
			MARS_ERR("bf_io_pending_anchor is not empty!\n");
		}
		if (unlikely(!list_empty(&bf->bf_again_write_pending_anchor))) {
			MARS_ERR("bf_again_write_pending_anchor is not empty!\n");
		}
#endif
		if (unlikely(!(bf->bf_flags & MARS_BUF_UPTODATE))) {
			list_del_init(&bf->bf_hash_head);
			brick->current_count--;
			where = &brick->free_anchor;
		}
		list_del(&bf->bf_lru_head);
		list_add(&bf->bf_lru_head, where);
	} // else no harm can happen

	// lru freeing (this is completeley independent from bf)
	__lru_free(brick);
	__prune_cache(brick, brick->max_count, &flags);

	traced_unlock(&brick->buf_lock, flags);
}

static void _buf_buf_put(struct buf_mars_buf_aspect *mbuf_a)
{
	struct mars_buf_object *mbuf = mbuf_a->object;
	struct buf_head *bf;

#if 1
	int test;
	test = atomic_read(&mbuf->buf_count);
	if (test <= 0) {
		atomic_set(&mbuf->buf_count, 1);
		MARS_ERR("bad buf_count %d\n", test);
	}
#endif
	if (!atomic_dec_and_test(&mbuf->buf_count))
		return;

	bf = mbuf_a->bfa_bf;
	MARS_DBG("buf_buf_put() mbuf=%p mbuf_a=%p bf=%p\n", mbuf, mbuf_a, bf);
	
	__bf_put(bf);

	buf_free_mars_buf(mbuf);
}

static void buf_buf_put(struct buf_output *output, struct mars_buf_object *mbuf)
{
	struct buf_mars_buf_aspect *mbuf_a;
	mbuf_a = buf_mars_buf_get_aspect(output, mbuf);
	if (unlikely(!mbuf_a)) {
		MARS_FAT("cannot get aspect\n");
		return;
	}
	_buf_buf_put(mbuf_a);
}

static void _buf_endio(struct mars_buf_object *mbuf)
{
	int error = mbuf->cb_error;
	struct bio *bio = mbuf->orig_bio;
	MARS_DBG("_buf_endio() mbuf=%p bio=%p\n", mbuf, bio);
	if (bio) {
		if (error < 0) {
			MARS_ERR("_buf_endio() error=%d bi_size=%d\n", error, bio->bi_size);
		}
		if (error > 0)
			error = 0;
		bio_endio(bio, error);
		bio_put(bio);
	} else {
		//...
	}
}

static void _buf_bio_callback(struct bio *bio, int code);

static int _buf_make_bios(struct buf_brick *brick, struct buf_head *bf, void *start_data, loff_t start_pos, int start_len, int rw)
{
	struct buf_input *input;
	LIST_HEAD(tmp);
	int status = EINVAL;
	int iters = 0;

#if 1
	loff_t bf_end = bf->bf_pos + brick->backing_size;
	loff_t end_pos;
	if (start_pos < bf->bf_pos || start_pos >= bf_end) {
		MARS_ERR("bad start_pos %llu (%llu ... %llu)\n", start_pos, bf->bf_pos, bf_end);
		goto done;
	}
	end_pos = start_pos + start_len;
	if (end_pos <= bf->bf_pos || end_pos > bf_end) {
		MARS_ERR("bad end_pos %llu (%llu ... %llu)\n", end_pos, bf->bf_pos, bf_end);
		goto done;
	}
	if (!start_data) {
		MARS_ERR("bad start_data\n");
		goto done;
	}
	if (start_len <= 0) {
		MARS_ERR("bad start_len %d\n", start_len);
		goto done;
	}
#endif
	status = -ENOMEM;
	while (start_len > 0) {
		struct mars_buf_object *mbuf;
		struct buf_mars_buf_aspect *mbuf_a;
		struct bio *bio = NULL;
		int len;

		mbuf = buf_alloc_mars_buf(brick->outputs[0], &brick->mbuf_object_layout);
		if (unlikely(!mbuf))
			break;

		mbuf_a = buf_mars_buf_get_aspect(brick->outputs[0], mbuf);
		if (unlikely(!mbuf_a)) {
			buf_free_mars_buf(mbuf);
			break;
		}

		list_add(&mbuf_a->tmp_head, &tmp);
		mbuf_a->bfa_bf = bf;

		len = make_bio(brick, &bio, start_data, start_pos, start_len);
		if (unlikely(len <= 0 || !bio)) {
			buf_free_mars_buf(mbuf);
			break;
		}

		bio->bi_private = mbuf_a;
		bio->bi_end_io = _buf_bio_callback;
		bio->bi_rw = rw;
		mbuf->cb_buf_endio = _buf_endio;

		mars_buf_attach_bio(mbuf, bio);

		start_data += len;
		start_pos += len;
		start_len -= len;
		iters++;
	}
	if (!start_len)
		status = 0;
#if 1
	if (iters != 1) {
		MARS_INF("start_pos=%lld start_len=%d iters=%d, status=%d\n", start_pos, start_len, iters, status);
	}
	iters = 0;
#endif

	input = brick->inputs[0];
	while (!list_empty(&tmp)) {
		struct mars_buf_object *mbuf;
		struct buf_mars_buf_aspect *mbuf_a;
		mbuf_a = container_of(tmp.next, struct buf_mars_buf_aspect, tmp_head);
		mbuf = mbuf_a->object;
		list_del_init(&mbuf_a->tmp_head);
		iters++;

		if (status < 0) { // clean up
			mbuf->cb_error = status;
			mbuf->cb_buf_endio(mbuf);
#if 0
			if (mbuf->orig_bio)
				bio_put(mbuf->orig_bio);
#endif
			buf_free_mars_buf(mbuf);
			continue;
		}

		/* Remember the number of bios we are submitting.
		 */
		atomic_inc(&bf->bf_bio_count);

		MARS_DBG("starting buf IO mbuf=%p bio=%p bf=%p bf_count=%d bf_bio_count=%d\n", mbuf, mbuf->orig_bio, bf, atomic_read(&bf->bf_count), atomic_read(&bf->bf_bio_count));
#if 1
		GENERIC_INPUT_CALL(input, mars_buf_io, mbuf, rw);
#else
		// fake IO for testing
		mbuf->cb_error = status;
		mbuf->cb_buf_endio(mbuf);
#if 0
		if (mbuf->orig_bio)
			bio_put(mbuf->orig_bio);
#endif
		buf_free_mars_buf(mbuf);
#endif
	}
#if 1
	if (iters != 1) {
		MARS_INF("start_pos=%lld start_len=%d iters=%d, status=%d\n", start_pos, start_len, iters, status);
	}
	iters = 0;
#endif
done:
	return status;
}

/* This is called from the bio layer.
 */
static void _buf_bio_callback(struct bio *bio, int code)
{
	struct buf_mars_buf_aspect *mbuf_a;
	struct buf_head *bf;
	struct buf_brick *brick;
	void  *start_data = NULL;
	loff_t start_pos = 0;
	int    start_len = 0;
	int old_flags;
	unsigned long flags;
	LIST_HEAD(tmp);
	int test;
#if 1
	int count = 0;
#endif

	mbuf_a = bio->bi_private;
	bf = mbuf_a->bfa_bf;

	MARS_DBG("_buf_bio_callback() mbuf=%p bio=%p bf=%p bf_count=%d bf_bio_count=%d code=%d\n", mbuf_a->object, bio, bf, atomic_read(&bf->bf_count), atomic_read(&bf->bf_bio_count), code);

	if (unlikely(code < 0)) {
		MARS_ERR("BIO ERROR %d (old=%d)\n", code, bf->bf_bio_status);
		// this can race, but we don't worry about the exact error code
		bf->bf_bio_status = code;
	}

	test = atomic_read(&bf->bf_bio_count);
	if (unlikely(test <= 0)) {
		MARS_ERR("bad bf_bio_count %d\n", test);
		atomic_set(&bf->bf_bio_count, 1);
	}
	if (!atomic_dec_and_test(&bf->bf_bio_count))
		return;

	MARS_DBG("_buf_bio_callback() ZERO_COUNT mbuf=%p bio=%p bf=%p code=%d\n", mbuf_a->object, bio, bf, code);

	brick = bf->bf_brick;

	// get an extra reference, to avoid freeing bf underneath during callbacks
	test = atomic_read(&bf->bf_count);
	if (unlikely(test <= 0)) {
		MARS_ERR("bad bf_count %d\n", test);
	}
	atomic_inc(&bf->bf_count);

	traced_lock(&brick->buf_lock, flags);

	// update flags. this must be done before the callbacks.
	old_flags = bf->bf_flags;
	if (!bf->bf_bio_status && (old_flags & MARS_BUF_READING)) {
		bf->bf_flags |= MARS_BUF_UPTODATE;
	}
	// clear the flags, callbacks must not see them. may be re-enabled later.
	bf->bf_flags &= ~(MARS_BUF_READING | MARS_BUF_WRITING);

	/* Remember current version of pending list.
	 * This is necessary because later the callbacks might
	 * change it underneath.
	 */
	if (!list_empty(&bf->bf_io_pending_anchor)) {
		struct list_head *next = bf->bf_io_pending_anchor.next;
		list_del_init(&bf->bf_io_pending_anchor);
		list_add_tail(&tmp, next);
	}

	/* Move pending jobs to work.
	 * This is in essence an automatic restart mechanism.
	 * do this before the callbacks, because they may start
	 * new IOs. If not done in the right order, this could violate
	 * IO ordering semantics.
	 */
	while (!list_empty(&bf->bf_again_write_pending_anchor)) {
		struct buf_mars_buf_aspect *mbuf_a = container_of(bf->bf_again_write_pending_anchor.next, struct buf_mars_buf_aspect, bfc_pending_head);
		struct mars_buf_object *mbuf = mbuf_a->object;
		if (mbuf_a->bfa_bf != bf) {
			MARS_ERR("bad pointers %p != %p\n", mbuf_a->bfa_bf, bf);
		}
#if 1
		if (!(++count % 100)) {
			MARS_ERR("endless loop 1\n");
		}
#endif
		list_del_init(&mbuf_a->bfc_pending_head);
		list_add_tail(&mbuf_a->bfc_pending_head, &bf->bf_io_pending_anchor);

		// re-enable flags
		bf->bf_flags |= MARS_BUF_WRITING;
		bf->bf_bio_status = 0;

		if (!start_len) {
			// first time: only flush the affected area
			start_data = mbuf->buf_data;
			start_pos = mbuf->buf_pos;
			start_len = mbuf->buf_len;
		} else if (start_data != mbuf->buf_data ||
			  start_pos != mbuf->buf_pos ||
			  start_len != mbuf->buf_len) {
			// another time: flush the whole buffer
			start_data = bf->bf_data;
			start_pos = bf->bf_pos;
			start_len = brick->backing_size;
		}
	}

	traced_unlock(&brick->buf_lock, flags);

	/* Signal success by calling all callbacks.
	 * Thanks to the tmp list, we can do this outside the spinlock.
	 */
	while (!list_empty(&tmp)) {
		struct buf_mars_buf_aspect *mbuf_a = container_of(tmp.next, struct buf_mars_buf_aspect, bfc_pending_head);
		struct mars_buf_object *mbuf = mbuf_a->object;
		int buf_count;

		if (mbuf_a->bfa_bf != bf) {
			MARS_ERR("bad pointers %p != %p\n", mbuf_a->bfa_bf, bf);
		}
#if 1
		if (!(++count % 100)) {
			MARS_ERR("endless loop 2\n");
		}
#endif
		buf_count = atomic_read(&mbuf->buf_count);
		if (buf_count <= 0) {
			MARS_ERR("bad buf_count %d\n", buf_count);
		}
		list_del_init(&mbuf_a->bfc_pending_head);

		// update infos for callbacks, they may inspect it.
		mbuf->buf_flags = bf->bf_flags;
		mbuf->cb_error = bf->bf_bio_status;
		mbuf_a->nr_io_pending--;

		mbuf->cb_buf_endio(mbuf);

		_buf_buf_put(mbuf_a);
	}

	if (start_len) {
		MARS_DBG("ATTENTION %d\n", start_len);
		_buf_make_bios(brick, bf, start_data, start_pos, start_len, WRITE);
	}
	// drop the extra reference from above
	__bf_put(bf);
}

static void buf_buf_io(struct buf_output *output, struct mars_buf_object *mbuf, int rw)
{
	struct buf_brick *brick = output->brick;
	struct buf_mars_buf_aspect *mbuf_a;
	struct buf_head *bf;
	void  *start_data = NULL;
	loff_t start_pos = 0;
	int    start_len = 0;
	int buf_count;
	int bf_count;
	int status = -EINVAL;
	unsigned long flags;

	if (unlikely(!mbuf)) {
		MARS_ERR("internal problem: forgotten to supply mbuf\n");
		goto callback;
	}
	mbuf_a = buf_mars_buf_get_aspect(output, mbuf);
	if (unlikely(!mbuf_a)) {
		MARS_ERR("internal problem: mbuf aspect does not work\n");
		goto callback;
	}
	if (unlikely(mbuf_a->nr_io_pending++ > 0)) {
		MARS_ERR("sorry, you cannot start multiple IOs in parallel on the same mars_buf instance. buf_get() a different instance instead!\n");
		goto callback;
	}
	bf = mbuf_a->bfa_bf;
	if (unlikely(!bf)) {
		MARS_ERR("internal problem: forgotten bf\n");
		goto callback;
	}

	buf_count = atomic_read(&mbuf->buf_count);
	if (unlikely(buf_count <= 0)) {
		atomic_set(&mbuf->buf_count, 1);
		MARS_ERR("improper pairing of buf_get() / buf_put(): buf_count=%d\n", buf_count);
	}
	bf_count = atomic_read(&bf->bf_count);
	if (unlikely(bf_count <= 0)) {
		atomic_set(&bf->bf_count, 1);
		MARS_ERR("improper pairing of buf_get() / buf_put(): bf_count=%d\n", bf_count);
	}

	if (rw != READ) {
		if (unlikely(mbuf->buf_may_write == READ)) {
			MARS_ERR("sorry, forgotten to set buf_may_write\n");
			goto callback;
		}
		if (unlikely(!(bf->bf_flags & MARS_BUF_UPTODATE))) {
			MARS_ERR("sorry, writing is only allowed on UPTODATE buffers\n");
			goto callback;
		}
	}

	mbuf->buf_rw = rw;

	traced_lock(&brick->buf_lock, flags);

	if (rw) { // WRITE
		if (bf->bf_flags & MARS_BUF_READING) {
			MARS_ERR("bad bf_flags %d\n", bf->bf_flags);
		}
		if (!(bf->bf_flags & MARS_BUF_WRITING)) {
			// by definition, a writeout buffer is uptodate
			bf->bf_flags |= (MARS_BUF_WRITING | MARS_BUF_UPTODATE);
			bf->bf_bio_status = 0;
#if 1
			start_data = mbuf->buf_data;
			start_pos = mbuf->buf_pos;
			start_len = mbuf->buf_len;
#else // only for testing: write the full buffer
			start_data = (void*)((unsigned long)mbuf->buf_data & ~(unsigned long)(brick->backing_size - 1));
			start_pos = mbuf->buf_pos & ~(loff_t)(brick->backing_size - 1);
			start_len = brick->backing_size;
#endif
			list_add(&mbuf_a->bfc_pending_head, &bf->bf_io_pending_anchor);
		} else {
			list_add(&mbuf_a->bfc_pending_head, &bf->bf_again_write_pending_anchor);
			MARS_DBG("postponing %lld %d\n", mbuf->buf_pos, mbuf->buf_len);
		}
	} else { // READ
		if (bf->bf_flags & (MARS_BUF_UPTODATE | MARS_BUF_WRITING)) {
			goto already_done;
		}
		if (!(bf->bf_flags & MARS_BUF_READING)) {
			bf->bf_flags |= MARS_BUF_READING;
			bf->bf_bio_status = 0;

			// always read the whole buffer.
			start_data = (void*)((unsigned long)mbuf->buf_data & ~(unsigned long)(brick->backing_size - 1));
			start_pos = mbuf->buf_pos & ~(loff_t)(brick->backing_size - 1);
			start_len = brick->backing_size;
		}
		list_add(&mbuf_a->bfc_pending_head, &bf->bf_io_pending_anchor);
	}

	mbuf->buf_flags = bf->bf_flags;
	mbuf->cb_error = bf->bf_bio_status;

	/* Grab an extra reference.
	 * This will be released later in _buf_bio_callback() after
	 * calling the callbacks.
	 */
	atomic_inc(&mbuf->buf_count);

	traced_unlock(&brick->buf_lock, flags);

	if (!start_len) {
		// nothing to start, IO is already started.
		return;
	}

	status = _buf_make_bios(brick, bf, start_data, start_pos, start_len, rw);
	if (likely(status >= 0)) {
		/* No immediate callback, this time.
		 * Callbacks will be called later from _buf_bio_callback().
		 */
		return;
	}

	MARS_ERR("error %d during buf_buf_io()\n", status);
	buf_buf_put(output, mbuf);
	goto callback;

already_done:
	mbuf->buf_flags = bf->bf_flags;
	status = bf->bf_bio_status;

	traced_unlock(&brick->buf_lock, flags);

callback:
	mbuf->cb_error = status;
	mbuf->cb_buf_endio(mbuf);
}

//////////////// object / aspect constructors / destructors ///////////////

static int buf_mars_buf_aspect_init_fn(struct generic_aspect *_ini, void *_init_data)
{
	struct buf_mars_buf_aspect *ini = (void*)_ini;
	ini->bfa_bf = NULL;
	INIT_LIST_HEAD(&ini->bfc_pending_head);
	INIT_LIST_HEAD(&ini->tmp_head);
	ini->nr_io_pending = 0;
	return 0;
}

MARS_MAKE_STATICS(buf);

////////////////////// brick constructors / destructors ////////////////////

static int buf_brick_construct(struct buf_brick *brick)
{
	int i;
	brick->backing_order = 5; // TODO: make this configurable
	brick->backing_size = PAGE_SIZE << brick->backing_order;
	brick->max_count = 32; // TODO: make this configurable
	brick->current_count = 0;
	brick->alloc_count = 0;
	spin_lock_init(&brick->buf_lock);
	INIT_LIST_HEAD(&brick->free_anchor);
	INIT_LIST_HEAD(&brick->lru_anchor);
	for (i = 0; i < MARS_BUF_HASH_MAX; i++) {
		INIT_LIST_HEAD(&brick->cache_anchors[i]);
	}
	return 0;
}

static int buf_output_construct(struct buf_output *output)
{
	return 0;
}

static int buf_brick_destruct(struct buf_brick *brick)
{
	unsigned long flags;

	traced_lock(&brick->buf_lock, flags);

	brick->max_count = 0;
	__lru_free(brick);
	__prune_cache(brick, 0, &flags);

	traced_unlock(&brick->buf_lock, flags);

	return 0;
}

///////////////////////// static structs ////////////////////////

static struct buf_brick_ops buf_brick_ops = {
};

static struct buf_output_ops buf_output_ops = {
	.make_object_layout = buf_make_object_layout,
	.mars_get_info = buf_get_info,
	.mars_buf_get = buf_buf_get,
	.mars_buf_put = buf_buf_put,
	.mars_buf_io = buf_buf_io,
};

static const struct buf_input_type buf_input_type = {
	.type_name = "buf_input",
	.input_size = sizeof(struct buf_input),
};

static const struct buf_input_type *buf_input_types[] = {
	&buf_input_type,
};

static const struct buf_output_type buf_output_type = {
	.type_name = "buf_output",
	.output_size = sizeof(struct buf_output),
	.master_ops = &buf_output_ops,
	.output_construct = &buf_output_construct,
	.aspect_types = buf_aspect_types,
	.layout_code = {
		[BRICK_OBJ_MARS_BUF] = LAYOUT_ALL,
	}
};

static const struct buf_output_type *buf_output_types[] = {
	&buf_output_type,
};

const struct buf_brick_type buf_brick_type = {
	.type_name = "buf_brick",
	.brick_size = sizeof(struct buf_brick),
	.max_inputs = 1,
	.max_outputs = 1,
	.master_ops = &buf_brick_ops,
	.default_input_types = buf_input_types,
	.default_output_types = buf_output_types,
	.brick_construct = &buf_brick_construct,
	.brick_destruct = &buf_brick_destruct,
};
EXPORT_SYMBOL_GPL(buf_brick_type);

////////////////// module init stuff /////////////////////////

static int __init init_buf(void)
{
	printk(MARS_INFO "init_buf()\n");
	return buf_register_brick_type();
}

static void __exit exit_buf(void)
{
	printk(MARS_INFO "exit_buf()\n");
	buf_unregister_brick_type();
}

MODULE_DESCRIPTION("MARS buf brick");
MODULE_AUTHOR("Thomas Schoebel-Theuer <tst@1und1.de>");
MODULE_LICENSE("GPL");

module_init(init_buf);
module_exit(exit_buf);