honey_freelist.h

Go to the documentation of this file.

 
/* Copyright 2014 Olly Betts
 *
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU General Public License as
 * published by the Free Software Foundation; either version 2 of the
 * License, or (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, see
 * <https://www.gnu.org/licenses/>.
 */
 
#ifndef XAPIAN_INCLUDED_HONEY_FREELIST_H
#define XAPIAN_INCLUDED_HONEY_FREELIST_H
 
#include "honey_defs.h"
#include "pack.h"
 
class HoneyTable;
 
class HoneyFLCursor {
  public:
    uint4 n = 0;
 
    unsigned c = 0;
 
    HoneyFLCursor() { }
 
    bool operator==(const HoneyFLCursor& o) const {
        return n == o.n && c == o.c;
    }
 
    bool operator!=(const HoneyFLCursor& o) const {
        return !(*this == o);
    }
 
    void swap(HoneyFLCursor& o) {
        std::swap(n, o.n);
        std::swap(c, o.c);
    }
 
    void pack(std::string& buf) {
        pack_uint(buf, n);
        pack_uint(buf, c / 4);
    }
 
    bool unpack(const char** p, const char* end) {
        bool r = unpack_uint(p, end, &n) && unpack_uint(p, end, &c);
        if (usual(r))
            c *= 4;
        return r;
    }
};
 
class HoneyFreeList {
    HoneyFreeList(const HoneyFreeList&);
 
    void operator=(const HoneyFreeList&);
 
    void read_block(const HoneyTable* B, uint4 n, uint8_t* p);
 
    void write_block(const HoneyTable* B, uint4 n, uint8_t* p, uint4 rev);
 
  protected:
    uint4 revision = 0;
 
    uint4 first_unused_block = 0;
 
    HoneyFLCursor fl, fl_end, flw;
 
    bool flw_appending = false;
 
  private:
    uint8_t* p = nullptr;
 
    uint8_t* pw = nullptr;
 
  public:
    HoneyFreeList() { }
 
    void reset() {
        revision = 0;
        first_unused_block = 0;
        flw_appending = false;
    }
 
    ~HoneyFreeList() { delete [] p; delete [] pw; }
 
    bool empty() const { return fl == fl_end; }
 
    uint4 get_block(const HoneyTable* B, uint4 block_size,
                    uint4* blk_to_free = NULL);
 
    uint4 walk(const HoneyTable* B, uint4 block_size, bool inclusive);
 
    void mark_block_unused(const HoneyTable* B, uint4 block_size, uint4 n);
 
    uint4 get_revision() const { return revision; }
    void set_revision(uint4 revision_) { revision = revision_; }
 
    uint4 get_first_unused_block() const { return first_unused_block; }
 
    // Used when compacting to a single file.
    void set_first_unused_block(uint4 base) { first_unused_block = base; }
 
    void commit(const HoneyTable* B, uint4 block_size);
 
    void pack(std::string& buf) {
        pack_uint(buf, revision);
        pack_uint(buf, first_unused_block);
        fl.pack(buf);
        flw.pack(buf);
    }
 
    bool unpack(const char** pstart, const char* end) {
        bool r = unpack_uint(pstart, end, &revision) &&
                 unpack_uint(pstart, end, &first_unused_block) &&
                 fl.unpack(pstart, end) &&
                 flw.unpack(pstart, end);
        if (r) {
            fl_end = flw;
            flw_appending = false;
        }
        return r;
    }
 
    bool unpack(const std::string& s) {
        const char* ptr = s.data();
        const char* end = ptr + s.size();
        return unpack(&ptr, end) && ptr == end;
    }
};
 
class HoneyFreeListChecker {
    // FIXME: uint_fast32_t is probably a good choice.
    typedef unsigned long elt_type;
 
    uint4 bitmap_size;
 
    elt_type* bitmap;
 
    // Prevent copying
    HoneyFreeListChecker(const HoneyFreeListChecker&);
    HoneyFreeListChecker& operator=(const HoneyFreeListChecker&);
 
  public:
    explicit HoneyFreeListChecker(const HoneyFreeList& fl);
 
    ~HoneyFreeListChecker() {
        delete [] bitmap;
    }
 
    bool mark_used(uint4 n) {
        const unsigned BITS_PER_ELT = sizeof(elt_type) * 8;
        elt_type mask = static_cast<elt_type>(1) << (n & (BITS_PER_ELT - 1));
        n /= BITS_PER_ELT;
        if (rare(n >= bitmap_size)) return false;
        if ((bitmap[n] & mask) == 0) return false;
        bitmap[n] &= ~mask;
        return true;
    }
 
    uint4 count_set_bits(uint4 * p_first_bad_blk) const;
};
 
#endif // XAPIAN_INCLUDED_HONEY_FREELIST_H