1 /* SPDX-License-Identifier: GPL-2.0 */
2 /*
3 * linux/include/linux/relay.h
4 *
5 * Copyright (C) 2002, 2003 - Tom Zanussi (zanussi@us.ibm.com), IBM Corp
6 * Copyright (C) 1999, 2000, 2001, 2002 - Karim Yaghmour (karim@opersys.com)
7 *
8 * CONFIG_RELAY definitions and declarations
9 */
10
11 #ifndef _LINUX_RELAY_H
12 #define _LINUX_RELAY_H
13
14 #include <linux/types.h>
15 #include <linux/sched.h>
16 #include <linux/timer.h>
17 #include <linux/wait.h>
18 #include <linux/list.h>
19 #include <linux/irq_work.h>
20 #include <linux/bug.h>
21 #include <linux/fs.h>
22 #include <linux/poll.h>
23 #include <linux/kref.h>
24 #include <linux/percpu.h>
25
26 /*
27 * Tracks changes to rchan/rchan_buf structs
28 */
29 #define RELAYFS_CHANNEL_VERSION 7
30
31 /*
32 * Per-cpu relay channel buffer
33 */
34 struct rchan_buf
35 {
36 void *start; /* start of channel buffer */
37 void *data; /* start of current sub-buffer */
38 size_t offset; /* current offset into sub-buffer */
39 size_t subbufs_produced; /* count of sub-buffers produced */
40 size_t subbufs_consumed; /* count of sub-buffers consumed */
41 struct rchan *chan; /* associated channel */
42 wait_queue_head_t read_wait; /* reader wait queue */
43 struct irq_work wakeup_work; /* reader wakeup */
44 struct dentry *dentry; /* channel file dentry */
45 struct kref kref; /* channel buffer refcount */
46 struct page **page_array; /* array of current buffer pages */
47 unsigned int page_count; /* number of current buffer pages */
48 unsigned int finalized; /* buffer has been finalized */
49 size_t *padding; /* padding counts per sub-buffer */
50 size_t prev_padding; /* temporary variable */
51 size_t bytes_consumed; /* bytes consumed in cur read subbuf */
52 size_t early_bytes; /* bytes consumed before VFS inited */
53 unsigned int cpu; /* this buf's cpu */
54 } ____cacheline_aligned;
55
56 /*
57 * Relay channel data structure
58 */
59 struct rchan
60 {
61 u32 version; /* the version of this struct */
62 size_t subbuf_size; /* sub-buffer size */
63 size_t n_subbufs; /* number of sub-buffers per buffer */
64 size_t alloc_size; /* total buffer size allocated */
65 struct rchan_callbacks *cb; /* client callbacks */
66 struct kref kref; /* channel refcount */
67 void *private_data; /* for user-defined data */
68 size_t last_toobig; /* tried to log event > subbuf size */
69 struct rchan_buf * __percpu *buf; /* per-cpu channel buffers */
70 int is_global; /* One global buffer ? */
71 struct list_head list; /* for channel list */
72 struct dentry *parent; /* parent dentry passed to open */
73 int has_base_filename; /* has a filename associated? */
74 char base_filename[NAME_MAX]; /* saved base filename */
75 };
76
77 /*
78 * Relay channel client callbacks
79 */
80 struct rchan_callbacks
81 {
82 /*
83 * subbuf_start - called on buffer-switch to a new sub-buffer
84 * @buf: the channel buffer containing the new sub-buffer
85 * @subbuf: the start of the new sub-buffer
86 * @prev_subbuf: the start of the previous sub-buffer
87 * @prev_padding: unused space at the end of previous sub-buffer
88 *
89 * The client should return 1 to continue logging, 0 to stop
90 * logging.
91 *
92 * NOTE: subbuf_start will also be invoked when the buffer is
93 * created, so that the first sub-buffer can be initialized
94 * if necessary. In this case, prev_subbuf will be NULL.
95 *
96 * NOTE: the client can reserve bytes at the beginning of the new
97 * sub-buffer by calling subbuf_start_reserve() in this callback.
98 */
99 int (*subbuf_start) (struct rchan_buf *buf,
100 void *subbuf,
101 void *prev_subbuf,
102 size_t prev_padding);
103
104 /*
105 * buf_mapped - relay buffer mmap notification
106 * @buf: the channel buffer
107 * @filp: relay file pointer
108 *
109 * Called when a relay file is successfully mmapped
110 */
111 void (*buf_mapped)(struct rchan_buf *buf,
112 struct file *filp);
113
114 /*
115 * buf_unmapped - relay buffer unmap notification
116 * @buf: the channel buffer
117 * @filp: relay file pointer
118 *
119 * Called when a relay file is successfully unmapped
120 */
121 void (*buf_unmapped)(struct rchan_buf *buf,
122 struct file *filp);
123 /*
124 * create_buf_file - create file to represent a relay channel buffer
125 * @filename: the name of the file to create
126 * @parent: the parent of the file to create
127 * @mode: the mode of the file to create
128 * @buf: the channel buffer
129 * @is_global: outparam - set non-zero if the buffer should be global
130 *
131 * Called during relay_open(), once for each per-cpu buffer,
132 * to allow the client to create a file to be used to
133 * represent the corresponding channel buffer. If the file is
134 * created outside of relay, the parent must also exist in
135 * that filesystem.
136 *
137 * The callback should return the dentry of the file created
138 * to represent the relay buffer.
139 *
140 * Setting the is_global outparam to a non-zero value will
141 * cause relay_open() to create a single global buffer rather
142 * than the default set of per-cpu buffers.
143 *
144 * See Documentation/filesystems/relay.rst for more info.
145 */
146 struct dentry *(*create_buf_file)(const char *filename,
147 struct dentry *parent,
148 umode_t mode,
149 struct rchan_buf *buf,
150 int *is_global);
151
152 /*
153 * remove_buf_file - remove file representing a relay channel buffer
154 * @dentry: the dentry of the file to remove
155 *
156 * Called during relay_close(), once for each per-cpu buffer,
157 * to allow the client to remove a file used to represent a
158 * channel buffer.
159 *
160 * The callback should return 0 if successful, negative if not.
161 */
162 int (*remove_buf_file)(struct dentry *dentry);
163 };
164
165 /*
166 * CONFIG_RELAY kernel API, kernel/relay.c
167 */
168
169 struct rchan *relay_open(const char *base_filename,
170 struct dentry *parent,
171 size_t subbuf_size,
172 size_t n_subbufs,
173 struct rchan_callbacks *cb,
174 void *private_data);
175 extern int relay_late_setup_files(struct rchan *chan,
176 const char *base_filename,
177 struct dentry *parent);
178 extern void relay_close(struct rchan *chan);
179 extern void relay_flush(struct rchan *chan);
180 extern void relay_subbufs_consumed(struct rchan *chan,
181 unsigned int cpu,
182 size_t consumed);
183 extern void relay_reset(struct rchan *chan);
184 extern int relay_buf_full(struct rchan_buf *buf);
185
186 extern size_t relay_switch_subbuf(struct rchan_buf *buf,
187 size_t length);
188
189 /**
190 * relay_write - write data into the channel
191 * @chan: relay channel
192 * @data: data to be written
193 * @length: number of bytes to write
194 *
195 * Writes data into the current cpu's channel buffer.
196 *
197 * Protects the buffer by disabling interrupts. Use this
198 * if you might be logging from interrupt context. Try
199 * __relay_write() if you know you won't be logging from
200 * interrupt context.
201 */
relay_write(struct rchan * chan,const void * data,size_t length)202 static inline void relay_write(struct rchan *chan,
203 const void *data,
204 size_t length)
205 {
206 unsigned long flags;
207 struct rchan_buf *buf;
208
209 local_irq_save(flags);
210 buf = *this_cpu_ptr(chan->buf);
211 if (unlikely(buf->offset + length > chan->subbuf_size))
212 length = relay_switch_subbuf(buf, length);
213 memcpy(buf->data + buf->offset, data, length);
214 buf->offset += length;
215 local_irq_restore(flags);
216 }
217
218 /**
219 * __relay_write - write data into the channel
220 * @chan: relay channel
221 * @data: data to be written
222 * @length: number of bytes to write
223 *
224 * Writes data into the current cpu's channel buffer.
225 *
226 * Protects the buffer by disabling preemption. Use
227 * relay_write() if you might be logging from interrupt
228 * context.
229 */
__relay_write(struct rchan * chan,const void * data,size_t length)230 static inline void __relay_write(struct rchan *chan,
231 const void *data,
232 size_t length)
233 {
234 struct rchan_buf *buf;
235
236 buf = *get_cpu_ptr(chan->buf);
237 if (unlikely(buf->offset + length > buf->chan->subbuf_size))
238 length = relay_switch_subbuf(buf, length);
239 memcpy(buf->data + buf->offset, data, length);
240 buf->offset += length;
241 put_cpu_ptr(chan->buf);
242 }
243
244 /**
245 * relay_reserve - reserve slot in channel buffer
246 * @chan: relay channel
247 * @length: number of bytes to reserve
248 *
249 * Returns pointer to reserved slot, NULL if full.
250 *
251 * Reserves a slot in the current cpu's channel buffer.
252 * Does not protect the buffer at all - caller must provide
253 * appropriate synchronization.
254 */
relay_reserve(struct rchan * chan,size_t length)255 static inline void *relay_reserve(struct rchan *chan, size_t length)
256 {
257 void *reserved = NULL;
258 struct rchan_buf *buf = *get_cpu_ptr(chan->buf);
259
260 if (unlikely(buf->offset + length > buf->chan->subbuf_size)) {
261 length = relay_switch_subbuf(buf, length);
262 if (!length)
263 goto end;
264 }
265 reserved = buf->data + buf->offset;
266 buf->offset += length;
267
268 end:
269 put_cpu_ptr(chan->buf);
270 return reserved;
271 }
272
273 /**
274 * subbuf_start_reserve - reserve bytes at the start of a sub-buffer
275 * @buf: relay channel buffer
276 * @length: number of bytes to reserve
277 *
278 * Helper function used to reserve bytes at the beginning of
279 * a sub-buffer in the subbuf_start() callback.
280 */
subbuf_start_reserve(struct rchan_buf * buf,size_t length)281 static inline void subbuf_start_reserve(struct rchan_buf *buf,
282 size_t length)
283 {
284 BUG_ON(length >= buf->chan->subbuf_size - 1);
285 buf->offset = length;
286 }
287
288 /*
289 * exported relay file operations, kernel/relay.c
290 */
291 extern const struct file_operations relay_file_operations;
292
293 #ifdef CONFIG_RELAY
294 int relay_prepare_cpu(unsigned int cpu);
295 #else
296 #define relay_prepare_cpu NULL
297 #endif
298
299 #endif /* _LINUX_RELAY_H */
300
301