/* SPDX-License-Identifier: GPL-2.0 */
/*
 * Copyright (c) 2014-2026 Christoph Hellwig.
 *
 * Support for exportfs-based layout grants for direct block device access.
 */
#ifndef LINUX_EXPORTFS_BLOCK_H
#define LINUX_EXPORTFS_BLOCK_H 1

#include <linux/blkdev.h>
#include <linux/exportfs.h>
#include <linux/fs.h>

struct inode;
struct iomap;
struct super_block;

/*
 * There are the two types of block-style layout support:
 *  - In-band implies a device identified by a unique cookie inside the actual
 *    device address space checked by the ->get_uuid method as used by the pNFS
 *    block layout.  This is a bit dangerous and deprecated.
 *  - Out of band implies identification by out of band unique identifiers
 *    specified by the storage protocol, which is much safer and used by the
 *    pNFS SCSI/NVMe layouts.
 */
typedef unsigned int __bitwise expfs_block_layouts_t;
#define EXPFS_BLOCK_FLAG(__bit) \
	((__force expfs_block_layouts_t)(1u << __bit))
#define EXPFS_BLOCK_IN_BAND_ID		EXPFS_BLOCK_FLAG(0)
#define EXPFS_BLOCK_OUT_OF_BAND_ID	EXPFS_BLOCK_FLAG(1)

struct exportfs_block_ops {
	/*
	 * Returns the EXPFS_BLOCK_* bitmap of supported layout types.
	 */
	expfs_block_layouts_t (*layouts_supported)(struct super_block *sb);

	/*
	 * Get the in-band device unique signature exposed to clients.
	 */
	int (*get_uuid)(struct super_block *sb, u8 *buf, u32 *len, u64 *offset);

	/*
	 * Map blocks for direct block access.
	 * If @write is %true, also allocate the blocks for the range if needed.
	 */
	int (*map_blocks)(struct inode *inode, loff_t offset, u64 len,
			struct iomap *iomap, bool write,
			u32 *device_generation);

	/*
	 * Commit blocks previously handed out by ->map_blocks and written to by
	 * the client.
	 */
	int (*commit_blocks)(struct inode *inode, struct iomap *iomaps,
			int nr_iomaps, loff_t new_size);
};

static inline bool
exportfs_bdev_supports_out_of_band_id(struct block_device *bdev)
{
	return bdev->bd_disk->fops->pr_ops &&
		bdev->bd_disk->fops->get_unique_id;
}

#ifdef CONFIG_EXPORTFS_BLOCK_OPS
static inline expfs_block_layouts_t
exportfs_layouts_supported(struct super_block *sb)
{
	const struct exportfs_block_ops *bops = sb->s_export_op->block_ops;

	if (!bops ||
	    !bops->layouts_supported ||
	    WARN_ON_ONCE(!bops->map_blocks) ||
	    WARN_ON_ONCE(!bops->commit_blocks))
		return 0;
	return bops->layouts_supported(sb);
}
#else
static inline expfs_block_layouts_t
exportfs_layouts_supported(struct super_block *sb)
{
	return 0;
}
#endif /* CONFIG_EXPORTFS_BLOCK_OPS */

#endif /* LINUX_EXPORTFS_BLOCK_H */