141 lines · 3.9 KB · C
RawDownload
/* SPDX-License-Identifier: GPL-2.0 */#ifndef BLKTRACE_H#define BLKTRACE_H #include <linux/blkdev.h>#include <linux/relay.h>#include <linux/compat.h>#include <uapi/linux/blktrace_api.h>#include <linux/list.h> #if defined(CONFIG_BLK_DEV_IO_TRACE) #include <linux/sysfs.h> struct blk_trace {	int trace_state;	struct rchan *rchan;	unsigned long __percpu *sequence;	unsigned char __percpu *msg_data;	u16 act_mask;	u64 start_lba;	u64 end_lba;	u32 pid;	u32 dev;	struct dentry *dir;	struct dentry *dropped_file;	struct dentry *msg_file;	struct list_head running_list;	atomic_t dropped;}; struct blkcg; extern int blk_trace_ioctl(struct block_device *, unsigned, char __user *);extern void blk_trace_shutdown(struct request_queue *);extern __printf(3, 4)void __trace_note_message(struct blk_trace *, struct blkcg *blkcg, const char *fmt, ...); /** * blk_add_trace_msg - Add a (simple) message to the blktrace stream * @q:		queue the io is for * @fmt:	format to print message in * args...	Variable argument list for format * * Description: *     Records a (simple) message onto the blktrace stream. * *     NOTE: BLK_TN_MAX_MSG characters are output at most. *     NOTE: Can not use 'static inline' due to presence of var args... * **/#define blk_add_cgroup_trace_msg(q, cg, fmt, ...)			\	do {								\		struct blk_trace *bt;					\									\		rcu_read_lock();					\		bt = rcu_dereference((q)->blk_trace);			\		if (unlikely(bt))					\			__trace_note_message(bt, cg, fmt, ##__VA_ARGS__);\		rcu_read_unlock();					\	} while (0)#define blk_add_trace_msg(q, fmt, ...)					\	blk_add_cgroup_trace_msg(q, NULL, fmt, ##__VA_ARGS__)#define BLK_TN_MAX_MSG		128 static inline bool blk_trace_note_message_enabled(struct request_queue *q){	struct blk_trace *bt;	bool ret; 	rcu_read_lock();	bt = rcu_dereference(q->blk_trace);	ret = bt && (bt->act_mask & BLK_TC_NOTIFY);	rcu_read_unlock();	return ret;} extern void blk_add_driver_data(struct request_queue *q, struct request *rq,				void *data, size_t len);extern int blk_trace_setup(struct request_queue *q, char *name, dev_t dev,			   struct block_device *bdev,			   char __user *arg);extern int blk_trace_startstop(struct request_queue *q, int start);extern int blk_trace_remove(struct request_queue *q);extern void blk_trace_remove_sysfs(struct device *dev);extern int blk_trace_init_sysfs(struct device *dev); extern struct attribute_group blk_trace_attr_group; #else /* !CONFIG_BLK_DEV_IO_TRACE */# define blk_trace_ioctl(bdev, cmd, arg)		(-ENOTTY)# define blk_trace_shutdown(q)				do { } while (0)# define blk_add_driver_data(q, rq, data, len)		do {} while (0)# define blk_trace_setup(q, name, dev, bdev, arg)	(-ENOTTY)# define blk_trace_startstop(q, start)			(-ENOTTY)# define blk_trace_remove(q)				(-ENOTTY)# define blk_add_trace_msg(q, fmt, ...)			do { } while (0)# define blk_add_cgroup_trace_msg(q, cg, fmt, ...)	do { } while (0)# define blk_trace_remove_sysfs(dev)			do { } while (0)# define blk_trace_note_message_enabled(q)		(false)static inline int blk_trace_init_sysfs(struct device *dev){	return 0;} #endif /* CONFIG_BLK_DEV_IO_TRACE */ #ifdef CONFIG_COMPAT struct compat_blk_user_trace_setup {	char name[BLKTRACE_BDEV_SIZE];	u16 act_mask;	u32 buf_size;	u32 buf_nr;	compat_u64 start_lba;	compat_u64 end_lba;	u32 pid;};#define BLKTRACESETUP32 _IOWR(0x12, 115, struct compat_blk_user_trace_setup) #endif extern void blk_fill_rwbs(char *rwbs, unsigned int op, int bytes); static inline sector_t blk_rq_trace_sector(struct request *rq){	/*	 * Tracing should ignore starting sector for passthrough requests and	 * requests where starting sector didn't get set.	 */	if (blk_rq_is_passthrough(rq) || blk_rq_pos(rq) == (sector_t)-1)		return 0;	return blk_rq_pos(rq);} static inline unsigned int blk_rq_trace_nr_sectors(struct request *rq){	return blk_rq_is_passthrough(rq) ? 0 : blk_rq_sectors(rq);} #endif