include/linux/blktrace_api.h

   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
  14 struct 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
  31 struct blkcg;
  32
  33 extern int blk_trace_ioctl(struct block_device *, unsigned, char __user *);
  34 extern void blk_trace_shutdown(struct request_queue *);
  35 extern __printf(3, 4)
  36 void __trace_note_message(struct blk_trace *, struct blkcg *blkcg, const char *fmt, ...);
  37
  38 /**
  39  * blk_add_trace_msg - Add a (simple) message to the blktrace stream
  40  * @q:          queue the io is for
  41  * @fmt:        format to print message in
  42  * args...      Variable argument list for format
  43  *
  44  * Description:
  45  *     Records a (simple) message onto the blktrace stream.
  46  *
  47  *     NOTE: BLK_TN_MAX_MSG characters are output at most.
  48  *     NOTE: Can not use 'static inline' due to presence of var args...
  49  *
  50  **/
  51 #define blk_add_cgroup_trace_msg(q, cg, fmt, ...)                       \
  52         do {                                                            \
  53                 struct blk_trace *bt = (q)->blk_trace;                  \
  54                 if (unlikely(bt))                                       \
  55                         __trace_note_message(bt, cg, fmt, ##__VA_ARGS__);\
  56         } while (0)
  57 #define blk_add_trace_msg(q, fmt, ...)                                  \
  58         blk_add_cgroup_trace_msg(q, NULL, fmt, ##__VA_ARGS__)
  59 #define BLK_TN_MAX_MSG          128
  60
  61 static inline bool blk_trace_note_message_enabled(struct request_queue *q)
  62 {
  63         struct blk_trace *bt = q->blk_trace;
  64         if (likely(!bt))
  65                 return false;
  66         return bt->act_mask & BLK_TC_NOTIFY;
  67 }
  68
  69 extern void blk_add_driver_data(struct request_queue *q, struct request *rq,
  70                                 void *data, size_t len);
  71 extern int blk_trace_setup(struct request_queue *q, char *name, dev_t dev,
  72                            struct block_device *bdev,
  73                            char __user *arg);
  74 extern int blk_trace_startstop(struct request_queue *q, int start);
  75 extern int blk_trace_remove(struct request_queue *q);
  76 extern void blk_trace_remove_sysfs(struct device *dev);
  77 extern int blk_trace_init_sysfs(struct device *dev);
  78
  79 extern struct attribute_group blk_trace_attr_group;
  80
  81 #else /* !CONFIG_BLK_DEV_IO_TRACE */
  82 # define blk_trace_ioctl(bdev, cmd, arg)                (-ENOTTY)
  83 # define blk_trace_shutdown(q)                          do { } while (0)
  84 # define blk_add_driver_data(q, rq, data, len)          do {} while (0)
  85 # define blk_trace_setup(q, name, dev, bdev, arg)       (-ENOTTY)
  86 # define blk_trace_startstop(q, start)                  (-ENOTTY)
  87 # define blk_trace_remove(q)                            (-ENOTTY)
  88 # define blk_add_trace_msg(q, fmt, ...)                 do { } while (0)
  89 # define blk_add_cgroup_trace_msg(q, cg, fmt, ...)      do { } while (0)
  90 # define blk_trace_remove_sysfs(dev)                    do { } while (0)
  91 # define blk_trace_note_message_enabled(q)              (false)
  92 static inline int blk_trace_init_sysfs(struct device *dev)
  93 {
  94         return 0;
  95 }
  96
  97 #endif /* CONFIG_BLK_DEV_IO_TRACE */
  98
  99 #ifdef CONFIG_COMPAT
 100
 101 struct compat_blk_user_trace_setup {
 102         char name[BLKTRACE_BDEV_SIZE];
 103         u16 act_mask;
 104         u32 buf_size;
 105         u32 buf_nr;
 106         compat_u64 start_lba;
 107         compat_u64 end_lba;
 108         u32 pid;
 109 };
 110 #define BLKTRACESETUP32 _IOWR(0x12, 115, struct compat_blk_user_trace_setup)
 111
 112 #endif
 113
 114 extern void blk_fill_rwbs(char *rwbs, unsigned int op, int bytes);
 115
 116 static inline sector_t blk_rq_trace_sector(struct request *rq)
 117 {
 118         return blk_rq_is_passthrough(rq) ? 0 : blk_rq_pos(rq);
 119 }
 120
 121 static inline unsigned int blk_rq_trace_nr_sectors(struct request *rq)
 122 {
 123         return blk_rq_is_passthrough(rq) ? 0 : blk_rq_sectors(rq);
 124 }
 125
 126 #endif