fuchsia / third_party / qemu / ff0507c239a246fd7215b31c5658fc6a3ee1e4c5 / . / include / qemu / hbitmap.h

/* | |

* Hierarchical Bitmap Data Type | |

* | |

* Copyright Red Hat, Inc., 2012 | |

* | |

* Author: Paolo Bonzini <pbonzini@redhat.com> | |

* | |

* This work is licensed under the terms of the GNU GPL, version 2 or | |

* later. See the COPYING file in the top-level directory. | |

*/ | |

#ifndef HBITMAP_H | |

#define HBITMAP_H | |

#include "bitops.h" | |

#include "host-utils.h" | |

typedef struct HBitmap HBitmap; | |

typedef struct HBitmapIter HBitmapIter; | |

#define BITS_PER_LEVEL (BITS_PER_LONG == 32 ? 5 : 6) | |

/* For 32-bit, the largest that fits in a 4 GiB address space. | |

* For 64-bit, the number of sectors in 1 PiB. Good luck, in | |

* either case... :) | |

*/ | |

#define HBITMAP_LOG_MAX_SIZE (BITS_PER_LONG == 32 ? 34 : 41) | |

/* We need to place a sentinel in level 0 to speed up iteration. Thus, | |

* we do this instead of HBITMAP_LOG_MAX_SIZE / BITS_PER_LEVEL. The | |

* difference is that it allocates an extra level when HBITMAP_LOG_MAX_SIZE | |

* is an exact multiple of BITS_PER_LEVEL. | |

*/ | |

#define HBITMAP_LEVELS ((HBITMAP_LOG_MAX_SIZE / BITS_PER_LEVEL) + 1) | |

struct HBitmapIter { | |

const HBitmap *hb; | |

/* Copied from hb for access in the inline functions (hb is opaque). */ | |

int granularity; | |

/* Entry offset into the last-level array of longs. */ | |

size_t pos; | |

/* The currently-active path in the tree. Each item of cur[i] stores | |

* the bits (i.e. the subtrees) yet to be processed under that node. | |

*/ | |

unsigned long cur[HBITMAP_LEVELS]; | |

}; | |

/** | |

* hbitmap_alloc: | |

* @size: Number of bits in the bitmap. | |

* @granularity: Granularity of the bitmap. Aligned groups of 2^@granularity | |

* bits will be represented by a single bit. Each operation on a | |

* range of bits first rounds the bits to determine which group they land | |

* in, and then affect the entire set; iteration will only visit the first | |

* bit of each group. | |

* | |

* Allocate a new HBitmap. | |

*/ | |

HBitmap *hbitmap_alloc(uint64_t size, int granularity); | |

/** | |

* hbitmap_truncate: | |

* @hb: The bitmap to change the size of. | |

* @size: The number of elements to change the bitmap to accommodate. | |

* | |

* truncate or grow an existing bitmap to accommodate a new number of elements. | |

* This may invalidate existing HBitmapIterators. | |

*/ | |

void hbitmap_truncate(HBitmap *hb, uint64_t size); | |

/** | |

* hbitmap_merge: | |

* | |

* Store result of merging @a and @b into @result. | |

* @result is allowed to be equal to @a or @b. | |

* | |

* Return true if the merge was successful, | |

* false if it was not attempted. | |

*/ | |

bool hbitmap_merge(const HBitmap *a, const HBitmap *b, HBitmap *result); | |

/** | |

* hbitmap_can_merge: | |

* | |

* hbitmap_can_merge(a, b) && hbitmap_can_merge(a, result) is sufficient and | |

* necessary for hbitmap_merge will not fail. | |

* | |

*/ | |

bool hbitmap_can_merge(const HBitmap *a, const HBitmap *b); | |

/** | |

* hbitmap_empty: | |

* @hb: HBitmap to operate on. | |

* | |

* Return whether the bitmap is empty. | |

*/ | |

bool hbitmap_empty(const HBitmap *hb); | |

/** | |

* hbitmap_granularity: | |

* @hb: HBitmap to operate on. | |

* | |

* Return the granularity of the HBitmap. | |

*/ | |

int hbitmap_granularity(const HBitmap *hb); | |

/** | |

* hbitmap_count: | |

* @hb: HBitmap to operate on. | |

* | |

* Return the number of bits set in the HBitmap. | |

*/ | |

uint64_t hbitmap_count(const HBitmap *hb); | |

/** | |

* hbitmap_set: | |

* @hb: HBitmap to operate on. | |

* @start: First bit to set (0-based). | |

* @count: Number of bits to set. | |

* | |

* Set a consecutive range of bits in an HBitmap. | |

*/ | |

void hbitmap_set(HBitmap *hb, uint64_t start, uint64_t count); | |

/** | |

* hbitmap_reset: | |

* @hb: HBitmap to operate on. | |

* @start: First bit to reset (0-based). | |

* @count: Number of bits to reset. | |

* | |

* Reset a consecutive range of bits in an HBitmap. | |

* @start and @count must be aligned to bitmap granularity. The only exception | |

* is resetting the tail of the bitmap: @count may be equal to hb->orig_size - | |

* @start, in this case @count may be not aligned. The sum of @start + @count is | |

* allowed to be greater than hb->orig_size, but only if @start < hb->orig_size | |

* and @start + @count = ALIGN_UP(hb->orig_size, granularity). | |

*/ | |

void hbitmap_reset(HBitmap *hb, uint64_t start, uint64_t count); | |

/** | |

* hbitmap_reset_all: | |

* @hb: HBitmap to operate on. | |

* | |

* Reset all bits in an HBitmap. | |

*/ | |

void hbitmap_reset_all(HBitmap *hb); | |

/** | |

* hbitmap_get: | |

* @hb: HBitmap to operate on. | |

* @item: Bit to query (0-based). | |

* | |

* Return whether the @item-th bit in an HBitmap is set. | |

*/ | |

bool hbitmap_get(const HBitmap *hb, uint64_t item); | |

/** | |

* hbitmap_is_serializable: | |

* @hb: HBitmap which should be (de-)serialized. | |

* | |

* Returns whether the bitmap can actually be (de-)serialized. Other | |

* (de-)serialization functions may only be invoked if this function returns | |

* true. | |

* | |

* Calling (de-)serialization functions does not affect a bitmap's | |

* (de-)serializability. | |

*/ | |

bool hbitmap_is_serializable(const HBitmap *hb); | |

/** | |

* hbitmap_serialization_align: | |

* @hb: HBitmap to operate on. | |

* | |

* Required alignment of serialization chunks, used by other serialization | |

* functions. For every chunk: | |

* 1. Chunk start should be aligned to this granularity. | |

* 2. Chunk size should be aligned too, except for last chunk (for which | |

* start + count == hb->size) | |

*/ | |

uint64_t hbitmap_serialization_align(const HBitmap *hb); | |

/** | |

* hbitmap_serialization_size: | |

* @hb: HBitmap to operate on. | |

* @start: Starting bit | |

* @count: Number of bits | |

* | |

* Return number of bytes hbitmap_(de)serialize_part needs | |

*/ | |

uint64_t hbitmap_serialization_size(const HBitmap *hb, | |

uint64_t start, uint64_t count); | |

/** | |

* hbitmap_serialize_part | |

* @hb: HBitmap to operate on. | |

* @buf: Buffer to store serialized bitmap. | |

* @start: First bit to store. | |

* @count: Number of bits to store. | |

* | |

* Stores HBitmap data corresponding to given region. The format of saved data | |

* is linear sequence of bits, so it can be used by hbitmap_deserialize_part | |

* independently of endianness and size of HBitmap level array elements | |

*/ | |

void hbitmap_serialize_part(const HBitmap *hb, uint8_t *buf, | |

uint64_t start, uint64_t count); | |

/** | |

* hbitmap_deserialize_part | |

* @hb: HBitmap to operate on. | |

* @buf: Buffer to restore bitmap data from. | |

* @start: First bit to restore. | |

* @count: Number of bits to restore. | |

* @finish: Whether to call hbitmap_deserialize_finish automatically. | |

* | |

* Restores HBitmap data corresponding to given region. The format is the same | |

* as for hbitmap_serialize_part. | |

* | |

* If @finish is false, caller must call hbitmap_serialize_finish before using | |

* the bitmap. | |

*/ | |

void hbitmap_deserialize_part(HBitmap *hb, uint8_t *buf, | |

uint64_t start, uint64_t count, | |

bool finish); | |

/** | |

* hbitmap_deserialize_zeroes | |

* @hb: HBitmap to operate on. | |

* @start: First bit to restore. | |

* @count: Number of bits to restore. | |

* @finish: Whether to call hbitmap_deserialize_finish automatically. | |

* | |

* Fills the bitmap with zeroes. | |

* | |

* If @finish is false, caller must call hbitmap_serialize_finish before using | |

* the bitmap. | |

*/ | |

void hbitmap_deserialize_zeroes(HBitmap *hb, uint64_t start, uint64_t count, | |

bool finish); | |

/** | |

* hbitmap_deserialize_ones | |

* @hb: HBitmap to operate on. | |

* @start: First bit to restore. | |

* @count: Number of bits to restore. | |

* @finish: Whether to call hbitmap_deserialize_finish automatically. | |

* | |

* Fills the bitmap with ones. | |

* | |

* If @finish is false, caller must call hbitmap_serialize_finish before using | |

* the bitmap. | |

*/ | |

void hbitmap_deserialize_ones(HBitmap *hb, uint64_t start, uint64_t count, | |

bool finish); | |

/** | |

* hbitmap_deserialize_finish | |

* @hb: HBitmap to operate on. | |

* | |

* Repair HBitmap after calling hbitmap_deserialize_data. Actually, all HBitmap | |

* layers are restored here. | |

*/ | |

void hbitmap_deserialize_finish(HBitmap *hb); | |

/** | |

* hbitmap_sha256: | |

* @bitmap: HBitmap to operate on. | |

* | |

* Returns SHA256 hash of the last level. | |

*/ | |

char *hbitmap_sha256(const HBitmap *bitmap, Error **errp); | |

/** | |

* hbitmap_free: | |

* @hb: HBitmap to operate on. | |

* | |

* Free an HBitmap and all of its associated memory. | |

*/ | |

void hbitmap_free(HBitmap *hb); | |

/** | |

* hbitmap_iter_init: | |

* @hbi: HBitmapIter to initialize. | |

* @hb: HBitmap to iterate on. | |

* @first: First bit to visit (0-based, must be strictly less than the | |

* size of the bitmap). | |

* | |

* Set up @hbi to iterate on the HBitmap @hb. hbitmap_iter_next will return | |

* the lowest-numbered bit that is set in @hb, starting at @first. | |

* | |

* Concurrent setting of bits is acceptable, and will at worst cause the | |

* iteration to miss some of those bits. | |

* | |

* The concurrent resetting of bits is OK. | |

*/ | |

void hbitmap_iter_init(HBitmapIter *hbi, const HBitmap *hb, uint64_t first); | |

/* | |

* hbitmap_next_dirty: | |

* | |

* Find next dirty bit within selected range. If not found, return -1. | |

* | |

* @hb: The HBitmap to operate on | |

* @start: The bit to start from. | |

* @count: Number of bits to proceed. If @start+@count > bitmap size, the whole | |

* bitmap is looked through. You can use INT64_MAX as @count to search up to | |

* the bitmap end. | |

*/ | |

int64_t hbitmap_next_dirty(const HBitmap *hb, int64_t start, int64_t count); | |

/* hbitmap_next_zero: | |

* | |

* Find next not dirty bit within selected range. If not found, return -1. | |

* | |

* @hb: The HBitmap to operate on | |

* @start: The bit to start from. | |

* @count: Number of bits to proceed. If @start+@count > bitmap size, the whole | |

* bitmap is looked through. You can use INT64_MAX as @count to search up to | |

* the bitmap end. | |

*/ | |

int64_t hbitmap_next_zero(const HBitmap *hb, int64_t start, int64_t count); | |

/* hbitmap_next_dirty_area: | |

* @hb: The HBitmap to operate on | |

* @start: the offset to start from | |

* @end: end of requested area | |

* @max_dirty_count: limit for out parameter dirty_count | |

* @dirty_start: on success: start of found area | |

* @dirty_count: on success: length of found area | |

* | |

* If dirty area found within [@start, @end), returns true and sets | |

* @dirty_start and @dirty_count appropriately. @dirty_count will not exceed | |

* @max_dirty_count. | |

* If dirty area was not found, returns false and leaves @dirty_start and | |

* @dirty_count unchanged. | |

*/ | |

bool hbitmap_next_dirty_area(const HBitmap *hb, int64_t start, int64_t end, | |

int64_t max_dirty_count, | |

int64_t *dirty_start, int64_t *dirty_count); | |

/** | |

* hbitmap_iter_next: | |

* @hbi: HBitmapIter to operate on. | |

* | |

* Return the next bit that is set in @hbi's associated HBitmap, | |

* or -1 if all remaining bits are zero. | |

*/ | |

int64_t hbitmap_iter_next(HBitmapIter *hbi); | |

#endif |