/* SPDX-License-Identifier: GPL-2.0 */
#ifndef _LINUX_BADBLOCKS_H
#define _LINUX_BADBLOCKS_H

#include <linux/seqlock.h>
#include <linux/device.h>
#include <linux/kernel.h>
#include <linux/stddef.h>
#include <linux/types.h>

#define BB_LEN_MASK (0x00000000000001FFULL)
#define BB_OFFSET_MASK (0x7FFFFFFFFFFFFE00ULL)
#define BB_ACK_MASK (0x8000000000000000ULL)
#define BB_MAX_LEN 512
#define BB_OFFSET(x) (((x) & BB_OFFSET_MASK) >> 9)
#define BB_LEN(x) (((x) & BB_LEN_MASK) + 1)
#define BB_ACK(x) (!!((x) & BB_ACK_MASK))
#define BB_END(x) (BB_OFFSET(x) + BB_LEN(x))
#define BB_MAKE(a, l, ack) (((a)<<9) | ((l)-1) | ((u64)(!!(ack)) << 63))

/* Bad block numbers are stored sorted in a single page.
 * 64bits is used for each block or extent.
 * 54 bits are sector number, 9 bits are extent size,
 * 1 bit is an 'acknowledged' flag.
 */
#define MAX_BADBLOCKS (PAGE_SIZE/8)

struct badblocks {
 struct device *dev; /* set by devm_init_badblocks */
 int count;  /* count of bad blocks */
 int unacked_exist; /* there probably are unacknowledged
     * bad blocks.  This is only cleared
     * when a read discovers none
 */
 int shift;  /* shift from sectors to block size
     * a -ve shift means badblocks are
 * disabled.*/
 u64 *page;  /* badblock list */
 int changed;
 seqlock_t lock;
 sector_t sector;
 sector_t size;  /* in sectors */
};

struct badblocks_context {
 sector_t start;
 sector_t len;
 int  ack;
};

int badblocks_check(struct badblocks *bb, sector_t s, sector_t sectors,
      sector_t *first_bad, sector_t *bad_sectors);
bool badblocks_set(struct badblocks *bb, sector_t s, sector_t sectors,
     int acknowledged);
bool badblocks_clear(struct badblocks *bb, sector_t s, sector_t sectors);
void ack_all_badblocks(struct badblocks *bb);
ssize_t badblocks_show(struct badblocks *bb, char *page, int unack);
ssize_t badblocks_store(struct badblocks *bb, const char *page, size_t len,
   int unack);
int badblocks_init(struct badblocks *bb, int enable);
void badblocks_exit(struct badblocks *bb);
struct device;
int devm_init_badblocks(struct device *dev, struct badblocks *bb);
static inline void devm_exit_badblocks(struct device *dev, struct badblocks *bb)
{
 if (bb->dev != dev) {
  dev_WARN_ONCE(dev, 1, "%s: badblocks instance not associated\n",
    __func__);
  return;
 }
 badblocks_exit(bb);
}

static inline int badblocks_full(struct badblocks *bb)
{
 return (bb->count >= MAX_BADBLOCKS);
}

static inline int badblocks_empty(struct badblocks *bb)
{
 return (bb->count == 0);
}

static inline void set_changed(struct badblocks *bb)
{
 if (bb->changed != 1)
  bb->changed = 1;
}

static inline void clear_changed(struct badblocks *bb)
{
 if (bb->changed != 0)
  bb->changed = 0;
}

#endif