#ifndef KVM__QCOW_H
#define KVM__QCOW_H

#include "kvm/mutex.h"

#include <linux/types.h>
#include <stdbool.h>
#include <linux/rbtree.h>
#include <linux/list.h>

#define QCOW_MAGIC		(('Q' << 24) | ('F' << 16) | ('I' << 8) | 0xfb)

#define QCOW1_VERSION		1
#define QCOW2_VERSION		2

#define QCOW1_OFLAG_COMPRESSED	(1LL << 63)

#define QCOW1_OFLAG_MASK	QCOW1_OFLAG_COMPRESSED

#define QCOW2_OFLAG_COPIED	(1LL << 63)
#define QCOW2_OFLAG_COMPRESSED	(1LL << 62)
#define QCOW2_OFLAG_MASK	(QCOW2_OFLAG_COPIED|QCOW2_OFLAG_COMPRESSED)

#define MAX_CACHE_NODES         32

struct qcow_l2_table {
	u64                     offset;
	struct rb_node          node;
	struct list_head        list;
	u8			dirty;
	u64                     table[];
};

struct qcow_table {
	u32			table_size;
	u64			*l1_table;
};

struct qcow {
	pthread_mutex_t		mutex;
	void			*header;
	struct qcow_table	table;
	int			fd;

	/* Level2 caching data structures */
	struct rb_root          root;
	struct list_head        lru_list;
	int                     nr_cached;
};

struct qcow_header {
	u64			size; /* in bytes */
	u64			l1_table_offset;
	u32			l1_size;
	u8			cluster_bits;
	u8			l2_bits;
	u64			oflag_mask;
};

struct qcow1_header_disk {
	u32			magic;
	u32			version;

	u64			backing_file_offset;
	u32 			backing_file_size;
	u32			mtime;

	u64			size; /* in bytes */

	u8			cluster_bits;
	u8			l2_bits;
	u32			crypt_method;

	u64			l1_table_offset;
};

struct qcow2_header_disk {
	u32			magic;
	u32			version;

	u64			backing_file_offset;
	u32			backing_file_size;

	u32			cluster_bits;
	u64			size; /* in bytes */
	u32			crypt_method;

	u32			l1_size;
	u64			l1_table_offset;

	u64			refcount_table_offset;
	u32			refcount_table_clusters;

	u32			nb_snapshots;
	u64			snapshots_offset;
};

struct disk_image *qcow_probe(int fd, bool readonly);

#endif /* KVM__QCOW_H */