1#ifndef BLKTRACE_H
2#define BLKTRACE_H
3
4#include <linux/blkdev.h>
5#include <linux/relay.h>
6#include <linux/compat.h>
7#include <uapi/linux/blktrace_api.h>
8#include <linux/list.h>
9
10#if defined(CONFIG_BLK_DEV_IO_TRACE)
11
12#include <linux/sysfs.h>
13
14struct blk_trace {
15	int trace_state;
16	struct rchan *rchan;
17	unsigned long __percpu *sequence;
18	unsigned char __percpu *msg_data;
19	u16 act_mask;
20	u64 start_lba;
21	u64 end_lba;
22	u32 pid;
23	u32 dev;
24	struct dentry *dir;
25	struct dentry *dropped_file;
26	struct dentry *msg_file;
27	struct list_head running_list;
28	atomic_t dropped;
29};
30
31extern int blk_trace_ioctl(struct block_device *, unsigned, char __user *);
32extern void blk_trace_shutdown(struct request_queue *);
33extern int do_blk_trace_setup(struct request_queue *q, char *name,
34			      dev_t dev, struct block_device *bdev,
35			      struct blk_user_trace_setup *buts);
36extern __printf(2, 3)
37void __trace_note_message(struct blk_trace *, const char *fmt, ...);
38
39/**
40 * blk_add_trace_msg - Add a (simple) message to the blktrace stream
41 * @q:		queue the io is for
42 * @fmt:	format to print message in
43 * args...	Variable argument list for format
44 *
45 * Description:
46 *     Records a (simple) message onto the blktrace stream.
47 *
48 *     NOTE: BLK_TN_MAX_MSG characters are output at most.
49 *     NOTE: Can not use 'static inline' due to presence of var args...
50 *
51 **/
52#define blk_add_trace_msg(q, fmt, ...)					\
53	do {								\
54		struct blk_trace *bt = (q)->blk_trace;			\
55		if (unlikely(bt))					\
56			__trace_note_message(bt, fmt, ##__VA_ARGS__);	\
57	} while (0)
58#define BLK_TN_MAX_MSG		128
59
60extern void blk_add_driver_data(struct request_queue *q, struct request *rq,
61				void *data, size_t len);
62extern int blk_trace_setup(struct request_queue *q, char *name, dev_t dev,
63			   struct block_device *bdev,
64			   char __user *arg);
65extern int blk_trace_startstop(struct request_queue *q, int start);
66extern int blk_trace_remove(struct request_queue *q);
67extern void blk_trace_remove_sysfs(struct device *dev);
68extern int blk_trace_init_sysfs(struct device *dev);
69
70extern struct attribute_group blk_trace_attr_group;
71
72#else /* !CONFIG_BLK_DEV_IO_TRACE */
73# define blk_trace_ioctl(bdev, cmd, arg)		(-ENOTTY)
74# define blk_trace_shutdown(q)				do { } while (0)
75# define do_blk_trace_setup(q, name, dev, bdev, buts)	(-ENOTTY)
76# define blk_add_driver_data(q, rq, data, len)		do {} while (0)
77# define blk_trace_setup(q, name, dev, bdev, arg)	(-ENOTTY)
78# define blk_trace_startstop(q, start)			(-ENOTTY)
79# define blk_trace_remove(q)				(-ENOTTY)
80# define blk_add_trace_msg(q, fmt, ...)			do { } while (0)
81# define blk_trace_remove_sysfs(dev)			do { } while (0)
82static inline int blk_trace_init_sysfs(struct device *dev)
83{
84	return 0;
85}
86
87#endif /* CONFIG_BLK_DEV_IO_TRACE */
88
89#ifdef CONFIG_COMPAT
90
91struct compat_blk_user_trace_setup {
92	char name[BLKTRACE_BDEV_SIZE];
93	u16 act_mask;
94	u32 buf_size;
95	u32 buf_nr;
96	compat_u64 start_lba;
97	compat_u64 end_lba;
98	u32 pid;
99};
100#define BLKTRACESETUP32 _IOWR(0x12, 115, struct compat_blk_user_trace_setup)
101
102#endif
103
104#if defined(CONFIG_EVENT_TRACING) && defined(CONFIG_BLOCK)
105
106static inline int blk_cmd_buf_len(struct request *rq)
107{
108	return (rq->cmd_type == REQ_TYPE_BLOCK_PC) ? rq->cmd_len * 3 : 1;
109}
110
111extern void blk_dump_cmd(char *buf, struct request *rq);
112extern void blk_fill_rwbs(char *rwbs, u32 rw, int bytes);
113
114#endif /* CONFIG_EVENT_TRACING && CONFIG_BLOCK */
115
116#endif
117