]> bbs.cooldavid.org Git - net-next-2.6.git/blame - kernel/trace/trace.h
x86, bts: provide in-kernel branch-trace interface
[net-next-2.6.git] / kernel / trace / trace.h
CommitLineData
bc0c38d1
SR
1#ifndef _LINUX_KERNEL_TRACE_H
2#define _LINUX_KERNEL_TRACE_H
3
4#include <linux/fs.h>
5#include <asm/atomic.h>
6#include <linux/sched.h>
7#include <linux/clocksource.h>
3928a8a2 8#include <linux/ring_buffer.h>
bd8ac686 9#include <linux/mmiotrace.h>
d13744cd 10#include <linux/ftrace.h>
3f5ec136 11#include <trace/boot.h>
bc0c38d1 12
72829bc3
TG
13enum trace_type {
14 __TRACE_FIRST_TYPE = 0,
15
16 TRACE_FN,
17 TRACE_CTX,
18 TRACE_WAKE,
dd0e545f 19 TRACE_CONT,
72829bc3 20 TRACE_STACK,
dd0e545f 21 TRACE_PRINT,
72829bc3 22 TRACE_SPECIAL,
bd8ac686
PP
23 TRACE_MMIO_RW,
24 TRACE_MMIO_MAP,
9f029e83 25 TRACE_BRANCH,
74239072
FW
26 TRACE_BOOT_CALL,
27 TRACE_BOOT_RET,
287b6e68
FW
28 TRACE_GRAPH_RET,
29 TRACE_GRAPH_ENT,
02b67518 30 TRACE_USER_STACK,
1e9b51c2 31 TRACE_BTS,
f3f47a67 32 TRACE_POWER,
72829bc3
TG
33
34 __TRACE_LAST_TYPE
35};
36
777e208d
SR
37/*
38 * The trace entry - the most basic unit of tracing. This is what
39 * is printed in the end as a single line in the trace output, such as:
40 *
41 * bash-15816 [01] 235.197585: idle_cpu <- irq_enter
42 */
43struct trace_entry {
44 unsigned char type;
45 unsigned char cpu;
46 unsigned char flags;
47 unsigned char preempt_count;
48 int pid;
02b67518 49 int tgid;
777e208d
SR
50};
51
bc0c38d1
SR
52/*
53 * Function trace entry - function address and parent function addres:
54 */
55struct ftrace_entry {
777e208d 56 struct trace_entry ent;
bc0c38d1
SR
57 unsigned long ip;
58 unsigned long parent_ip;
59};
15e6cb36 60
287b6e68
FW
61/* Function call entry */
62struct ftrace_graph_ent_entry {
63 struct trace_entry ent;
64 struct ftrace_graph_ent graph_ent;
65};
66
15e6cb36 67/* Function return entry */
287b6e68
FW
68struct ftrace_graph_ret_entry {
69 struct trace_entry ent;
70 struct ftrace_graph_ret ret;
15e6cb36 71};
d13744cd 72extern struct tracer boot_tracer;
bc0c38d1
SR
73
74/*
75 * Context switch trace entry - which task (and prio) we switched from/to:
76 */
77struct ctx_switch_entry {
777e208d 78 struct trace_entry ent;
bc0c38d1
SR
79 unsigned int prev_pid;
80 unsigned char prev_prio;
81 unsigned char prev_state;
82 unsigned int next_pid;
83 unsigned char next_prio;
bac524d3 84 unsigned char next_state;
80b5e940 85 unsigned int next_cpu;
bc0c38d1
SR
86};
87
f0a920d5
IM
88/*
89 * Special (free-form) trace entry:
90 */
91struct special_entry {
777e208d 92 struct trace_entry ent;
f0a920d5
IM
93 unsigned long arg1;
94 unsigned long arg2;
95 unsigned long arg3;
96};
97
86387f7e
IM
98/*
99 * Stack-trace entry:
100 */
101
74f4e369 102#define FTRACE_STACK_ENTRIES 8
86387f7e
IM
103
104struct stack_entry {
777e208d 105 struct trace_entry ent;
86387f7e
IM
106 unsigned long caller[FTRACE_STACK_ENTRIES];
107};
108
02b67518
TE
109struct userstack_entry {
110 struct trace_entry ent;
111 unsigned long caller[FTRACE_STACK_ENTRIES];
112};
113
dd0e545f
SR
114/*
115 * ftrace_printk entry:
116 */
117struct print_entry {
777e208d 118 struct trace_entry ent;
dd0e545f 119 unsigned long ip;
1fd8f2a3 120 int depth;
dd0e545f
SR
121 char buf[];
122};
123
777e208d
SR
124#define TRACE_OLD_SIZE 88
125
126struct trace_field_cont {
127 unsigned char type;
128 /* Temporary till we get rid of this completely */
129 char buf[TRACE_OLD_SIZE - 1];
130};
131
132struct trace_mmiotrace_rw {
133 struct trace_entry ent;
134 struct mmiotrace_rw rw;
135};
136
137struct trace_mmiotrace_map {
138 struct trace_entry ent;
139 struct mmiotrace_map map;
140};
141
74239072 142struct trace_boot_call {
777e208d 143 struct trace_entry ent;
74239072
FW
144 struct boot_trace_call boot_call;
145};
146
147struct trace_boot_ret {
148 struct trace_entry ent;
149 struct boot_trace_ret boot_ret;
777e208d
SR
150};
151
52f232cb
SR
152#define TRACE_FUNC_SIZE 30
153#define TRACE_FILE_SIZE 20
9f029e83 154struct trace_branch {
52f232cb
SR
155 struct trace_entry ent;
156 unsigned line;
157 char func[TRACE_FUNC_SIZE+1];
158 char file[TRACE_FILE_SIZE+1];
159 char correct;
160};
161
1e9b51c2
MM
162struct bts_entry {
163 struct trace_entry ent;
164 unsigned long from;
165 unsigned long to;
166};
167
f3f47a67
AV
168struct trace_power {
169 struct trace_entry ent;
170 struct power_trace state_data;
171};
172
fc5e27ae
PP
173/*
174 * trace_flag_type is an enumeration that holds different
175 * states when a trace occurs. These are:
9244489a
SR
176 * IRQS_OFF - interrupts were disabled
177 * IRQS_NOSUPPORT - arch does not support irqs_disabled_flags
178 * NEED_RESCED - reschedule is requested
179 * HARDIRQ - inside an interrupt handler
180 * SOFTIRQ - inside a softirq handler
181 * CONT - multiple entries hold the trace item
fc5e27ae
PP
182 */
183enum trace_flag_type {
184 TRACE_FLAG_IRQS_OFF = 0x01,
9244489a
SR
185 TRACE_FLAG_IRQS_NOSUPPORT = 0x02,
186 TRACE_FLAG_NEED_RESCHED = 0x04,
187 TRACE_FLAG_HARDIRQ = 0x08,
188 TRACE_FLAG_SOFTIRQ = 0x10,
189 TRACE_FLAG_CONT = 0x20,
fc5e27ae
PP
190};
191
5bf9a1ee 192#define TRACE_BUF_SIZE 1024
bc0c38d1
SR
193
194/*
195 * The CPU trace array - it consists of thousands of trace entries
196 * plus some other descriptor data: (for example which task started
197 * the trace, etc.)
198 */
199struct trace_array_cpu {
bc0c38d1 200 atomic_t disabled;
4e3c3333 201
c7aafc54 202 /* these fields get copied into max-trace: */
c7aafc54 203 unsigned long trace_idx;
53d0aa77 204 unsigned long overrun;
bc0c38d1
SR
205 unsigned long saved_latency;
206 unsigned long critical_start;
207 unsigned long critical_end;
208 unsigned long critical_sequence;
209 unsigned long nice;
210 unsigned long policy;
211 unsigned long rt_priority;
212 cycle_t preempt_timestamp;
213 pid_t pid;
214 uid_t uid;
215 char comm[TASK_COMM_LEN];
216};
217
218struct trace_iterator;
219
220/*
221 * The trace array - an array of per-CPU trace arrays. This is the
222 * highest level data structure that individual tracers deal with.
223 * They have on/off state as well:
224 */
225struct trace_array {
3928a8a2 226 struct ring_buffer *buffer;
bc0c38d1 227 unsigned long entries;
bc0c38d1
SR
228 int cpu;
229 cycle_t time_start;
b3806b43 230 struct task_struct *waiter;
bc0c38d1
SR
231 struct trace_array_cpu *data[NR_CPUS];
232};
233
7104f300
SR
234#define FTRACE_CMP_TYPE(var, type) \
235 __builtin_types_compatible_p(typeof(var), type *)
236
237#undef IF_ASSIGN
238#define IF_ASSIGN(var, entry, etype, id) \
239 if (FTRACE_CMP_TYPE(var, etype)) { \
240 var = (typeof(var))(entry); \
241 WARN_ON(id && (entry)->type != id); \
242 break; \
243 }
244
245/* Will cause compile errors if type is not found. */
246extern void __ftrace_bad_type(void);
247
248/*
249 * The trace_assign_type is a verifier that the entry type is
250 * the same as the type being assigned. To add new types simply
251 * add a line with the following format:
252 *
253 * IF_ASSIGN(var, ent, type, id);
254 *
255 * Where "type" is the trace type that includes the trace_entry
256 * as the "ent" item. And "id" is the trace identifier that is
257 * used in the trace_type enum.
258 *
259 * If the type can have more than one id, then use zero.
260 */
261#define trace_assign_type(var, ent) \
262 do { \
263 IF_ASSIGN(var, ent, struct ftrace_entry, TRACE_FN); \
264 IF_ASSIGN(var, ent, struct ctx_switch_entry, 0); \
265 IF_ASSIGN(var, ent, struct trace_field_cont, TRACE_CONT); \
266 IF_ASSIGN(var, ent, struct stack_entry, TRACE_STACK); \
02b67518 267 IF_ASSIGN(var, ent, struct userstack_entry, TRACE_USER_STACK);\
7104f300
SR
268 IF_ASSIGN(var, ent, struct print_entry, TRACE_PRINT); \
269 IF_ASSIGN(var, ent, struct special_entry, 0); \
270 IF_ASSIGN(var, ent, struct trace_mmiotrace_rw, \
271 TRACE_MMIO_RW); \
272 IF_ASSIGN(var, ent, struct trace_mmiotrace_map, \
273 TRACE_MMIO_MAP); \
74239072
FW
274 IF_ASSIGN(var, ent, struct trace_boot_call, TRACE_BOOT_CALL);\
275 IF_ASSIGN(var, ent, struct trace_boot_ret, TRACE_BOOT_RET);\
9f029e83 276 IF_ASSIGN(var, ent, struct trace_branch, TRACE_BRANCH); \
287b6e68
FW
277 IF_ASSIGN(var, ent, struct ftrace_graph_ent_entry, \
278 TRACE_GRAPH_ENT); \
279 IF_ASSIGN(var, ent, struct ftrace_graph_ret_entry, \
280 TRACE_GRAPH_RET); \
1e9b51c2 281 IF_ASSIGN(var, ent, struct bts_entry, TRACE_BTS);\
f3f47a67 282 IF_ASSIGN(var, ent, struct trace_power, TRACE_POWER); \
7104f300
SR
283 __ftrace_bad_type(); \
284 } while (0)
2c4f035f
FW
285
286/* Return values for print_line callback */
287enum print_line_t {
288 TRACE_TYPE_PARTIAL_LINE = 0, /* Retry after flushing the seq */
289 TRACE_TYPE_HANDLED = 1,
290 TRACE_TYPE_UNHANDLED = 2 /* Relay to other output functions */
291};
292
adf9f195
FW
293
294/*
295 * An option specific to a tracer. This is a boolean value.
296 * The bit is the bit index that sets its value on the
297 * flags value in struct tracer_flags.
298 */
299struct tracer_opt {
300 const char *name; /* Will appear on the trace_options file */
301 u32 bit; /* Mask assigned in val field in tracer_flags */
302};
303
304/*
305 * The set of specific options for a tracer. Your tracer
306 * have to set the initial value of the flags val.
307 */
308struct tracer_flags {
309 u32 val;
310 struct tracer_opt *opts;
311};
312
313/* Makes more easy to define a tracer opt */
314#define TRACER_OPT(s, b) .name = #s, .bit = b
315
bc0c38d1
SR
316/*
317 * A specific tracer, represented by methods that operate on a trace array:
318 */
319struct tracer {
320 const char *name;
1c80025a
FW
321 /* Your tracer should raise a warning if init fails */
322 int (*init)(struct trace_array *tr);
bc0c38d1 323 void (*reset)(struct trace_array *tr);
9036990d
SR
324 void (*start)(struct trace_array *tr);
325 void (*stop)(struct trace_array *tr);
bc0c38d1 326 void (*open)(struct trace_iterator *iter);
107bad8b 327 void (*pipe_open)(struct trace_iterator *iter);
bc0c38d1 328 void (*close)(struct trace_iterator *iter);
107bad8b
SR
329 ssize_t (*read)(struct trace_iterator *iter,
330 struct file *filp, char __user *ubuf,
331 size_t cnt, loff_t *ppos);
60a11774
SR
332#ifdef CONFIG_FTRACE_STARTUP_TEST
333 int (*selftest)(struct tracer *trace,
334 struct trace_array *tr);
335#endif
8bba1bf5 336 void (*print_header)(struct seq_file *m);
2c4f035f 337 enum print_line_t (*print_line)(struct trace_iterator *iter);
adf9f195
FW
338 /* If you handled the flag setting, return 0 */
339 int (*set_flag)(u32 old_flags, u32 bit, int set);
bc0c38d1
SR
340 struct tracer *next;
341 int print_max;
adf9f195 342 struct tracer_flags *flags;
bc0c38d1
SR
343};
344
214023c3
SR
345struct trace_seq {
346 unsigned char buffer[PAGE_SIZE];
347 unsigned int len;
6c6c2796 348 unsigned int readpos;
214023c3
SR
349};
350
bc0c38d1
SR
351/*
352 * Trace iterator - used by printout routines who present trace
353 * results to users and which routines might sleep, etc:
354 */
355struct trace_iterator {
356 struct trace_array *tr;
357 struct tracer *trace;
107bad8b 358 void *private;
3928a8a2 359 struct ring_buffer_iter *buffer_iter[NR_CPUS];
4e3c3333 360
53d0aa77
SR
361 /* The below is zeroed out in pipe_read */
362 struct trace_seq seq;
bc0c38d1 363 struct trace_entry *ent;
4e3c3333 364 int cpu;
3928a8a2 365 u64 ts;
4e3c3333 366
bc0c38d1
SR
367 unsigned long iter_flags;
368 loff_t pos;
4c11d7ae 369 long idx;
a309720c
SR
370
371 cpumask_t started;
bc0c38d1
SR
372};
373
9036990d 374int tracing_is_enabled(void);
45dcd8b8 375void trace_wake_up(void);
3928a8a2 376void tracing_reset(struct trace_array *tr, int cpu);
bc0c38d1
SR
377int tracing_open_generic(struct inode *inode, struct file *filp);
378struct dentry *tracing_init_dentry(void);
d618b3e6
IM
379void init_tracer_sysprof_debugfs(struct dentry *d_tracer);
380
45dcd8b8
PP
381struct trace_entry *tracing_get_trace_entry(struct trace_array *tr,
382 struct trace_array_cpu *data);
383void tracing_generic_entry_update(struct trace_entry *entry,
38697053
SR
384 unsigned long flags,
385 int pc);
45dcd8b8 386
bc0c38d1
SR
387void ftrace(struct trace_array *tr,
388 struct trace_array_cpu *data,
389 unsigned long ip,
390 unsigned long parent_ip,
38697053 391 unsigned long flags, int pc);
bc0c38d1
SR
392void tracing_sched_switch_trace(struct trace_array *tr,
393 struct trace_array_cpu *data,
394 struct task_struct *prev,
395 struct task_struct *next,
38697053 396 unsigned long flags, int pc);
bc0c38d1 397void tracing_record_cmdline(struct task_struct *tsk);
57422797
IM
398
399void tracing_sched_wakeup_trace(struct trace_array *tr,
400 struct trace_array_cpu *data,
401 struct task_struct *wakee,
402 struct task_struct *cur,
38697053 403 unsigned long flags, int pc);
f0a920d5
IM
404void trace_special(struct trace_array *tr,
405 struct trace_array_cpu *data,
406 unsigned long arg1,
407 unsigned long arg2,
38697053 408 unsigned long arg3, int pc);
6fb44b71
SR
409void trace_function(struct trace_array *tr,
410 struct trace_array_cpu *data,
411 unsigned long ip,
412 unsigned long parent_ip,
38697053 413 unsigned long flags, int pc);
bc0c38d1 414
287b6e68 415void trace_graph_return(struct ftrace_graph_ret *trace);
e49dc19c 416int trace_graph_entry(struct ftrace_graph_ent *trace);
1e9b51c2
MM
417void trace_bts(struct trace_array *tr,
418 unsigned long from,
419 unsigned long to);
420
41bc8144
SR
421void tracing_start_cmdline_record(void);
422void tracing_stop_cmdline_record(void);
e168e051
SR
423void tracing_sched_switch_assign_trace(struct trace_array *tr);
424void tracing_stop_sched_switch_record(void);
425void tracing_start_sched_switch_record(void);
bc0c38d1
SR
426int register_tracer(struct tracer *type);
427void unregister_tracer(struct tracer *type);
428
429extern unsigned long nsecs_to_usecs(unsigned long nsecs);
430
431extern unsigned long tracing_max_latency;
432extern unsigned long tracing_thresh;
433
434void update_max_tr(struct trace_array *tr, struct task_struct *tsk, int cpu);
435void update_max_tr_single(struct trace_array *tr,
436 struct task_struct *tsk, int cpu);
437
e309b41d 438extern cycle_t ftrace_now(int cpu);
bc0c38d1 439
606576ce 440#ifdef CONFIG_FUNCTION_TRACER
001b6767
SR
441void tracing_start_function_trace(void);
442void tracing_stop_function_trace(void);
443#else
444# define tracing_start_function_trace() do { } while (0)
445# define tracing_stop_function_trace() do { } while (0)
446#endif
447
bc0c38d1
SR
448#ifdef CONFIG_CONTEXT_SWITCH_TRACER
449typedef void
450(*tracer_switch_func_t)(void *private,
5b82a1b0 451 void *__rq,
bc0c38d1
SR
452 struct task_struct *prev,
453 struct task_struct *next);
454
455struct tracer_switch_ops {
456 tracer_switch_func_t func;
457 void *private;
458 struct tracer_switch_ops *next;
459};
460
660c7f9b 461char *trace_find_cmdline(int pid);
bc0c38d1
SR
462#endif /* CONFIG_CONTEXT_SWITCH_TRACER */
463
464#ifdef CONFIG_DYNAMIC_FTRACE
465extern unsigned long ftrace_update_tot_cnt;
d05cdb25
SR
466#define DYN_FTRACE_TEST_NAME trace_selftest_dynamic_test_func
467extern int DYN_FTRACE_TEST_NAME(void);
bc0c38d1
SR
468#endif
469
60a11774 470#ifdef CONFIG_FTRACE_STARTUP_TEST
60a11774
SR
471extern int trace_selftest_startup_function(struct tracer *trace,
472 struct trace_array *tr);
60a11774
SR
473extern int trace_selftest_startup_irqsoff(struct tracer *trace,
474 struct trace_array *tr);
60a11774
SR
475extern int trace_selftest_startup_preemptoff(struct tracer *trace,
476 struct trace_array *tr);
60a11774
SR
477extern int trace_selftest_startup_preemptirqsoff(struct tracer *trace,
478 struct trace_array *tr);
60a11774
SR
479extern int trace_selftest_startup_wakeup(struct tracer *trace,
480 struct trace_array *tr);
fb1b6d8b
SN
481extern int trace_selftest_startup_nop(struct tracer *trace,
482 struct trace_array *tr);
60a11774
SR
483extern int trace_selftest_startup_sched_switch(struct tracer *trace,
484 struct trace_array *tr);
a6dd24f8
IM
485extern int trace_selftest_startup_sysprof(struct tracer *trace,
486 struct trace_array *tr);
80e5ea45
SR
487extern int trace_selftest_startup_branch(struct tracer *trace,
488 struct trace_array *tr);
60a11774
SR
489#endif /* CONFIG_FTRACE_STARTUP_TEST */
490
c7aafc54 491extern void *head_page(struct trace_array_cpu *data);
72829bc3 492extern int trace_seq_printf(struct trace_seq *s, const char *fmt, ...);
fc5e27ae
PP
493extern void trace_seq_print_cont(struct trace_seq *s,
494 struct trace_iterator *iter);
15e6cb36
FW
495
496extern int
497seq_print_ip_sym(struct trace_seq *s, unsigned long ip,
498 unsigned long sym_flags);
6c6c2796
PP
499extern ssize_t trace_seq_to_user(struct trace_seq *s, char __user *ubuf,
500 size_t cnt);
72829bc3 501extern long ns2usecs(cycle_t nsec);
1fd8f2a3
FW
502extern int
503trace_vprintk(unsigned long ip, int depth, const char *fmt, va_list args);
c7aafc54 504
4e655519
IM
505extern unsigned long trace_flags;
506
15e6cb36 507/* Standard output formatting function used for function return traces */
fb52607a
FW
508#ifdef CONFIG_FUNCTION_GRAPH_TRACER
509extern enum print_line_t print_graph_function(struct trace_iterator *iter);
ea4e2bc4
SR
510
511#ifdef CONFIG_DYNAMIC_FTRACE
512/* TODO: make this variable */
513#define FTRACE_GRAPH_MAX_FUNCS 32
514extern int ftrace_graph_count;
515extern unsigned long ftrace_graph_funcs[FTRACE_GRAPH_MAX_FUNCS];
516
517static inline int ftrace_graph_addr(unsigned long addr)
518{
519 int i;
520
521 if (!ftrace_graph_count || test_tsk_trace_graph(current))
522 return 1;
523
524 for (i = 0; i < ftrace_graph_count; i++) {
525 if (addr == ftrace_graph_funcs[i])
526 return 1;
527 }
528
529 return 0;
530}
15e6cb36 531#else
ea4e2bc4
SR
532static inline int ftrace_trace_addr(unsigned long addr)
533{
6b253930
IM
534 return 1;
535}
536static inline int ftrace_graph_addr(unsigned long addr)
537{
538 return 1;
ea4e2bc4
SR
539}
540#endif /* CONFIG_DYNAMIC_FTRACE */
541
542#else /* CONFIG_FUNCTION_GRAPH_TRACER */
15e6cb36 543static inline enum print_line_t
fb52607a 544print_graph_function(struct trace_iterator *iter)
15e6cb36
FW
545{
546 return TRACE_TYPE_UNHANDLED;
547}
ea4e2bc4 548#endif /* CONFIG_FUNCTION_GRAPH_TRACER */
15e6cb36 549
978f3a45 550extern struct pid *ftrace_pid_trace;
804a6851
SR
551
552static inline int ftrace_trace_task(struct task_struct *task)
553{
77d683f3 554 if (!ftrace_pid_trace)
804a6851
SR
555 return 1;
556
557 return test_tsk_trace_trace(task);
558}
559
4fcdae83
SR
560/*
561 * trace_iterator_flags is an enumeration that defines bit
562 * positions into trace_flags that controls the output.
563 *
564 * NOTE: These bits must match the trace_options array in
565 * trace.c.
566 */
4e655519
IM
567enum trace_iterator_flags {
568 TRACE_ITER_PRINT_PARENT = 0x01,
569 TRACE_ITER_SYM_OFFSET = 0x02,
570 TRACE_ITER_SYM_ADDR = 0x04,
571 TRACE_ITER_VERBOSE = 0x08,
572 TRACE_ITER_RAW = 0x10,
573 TRACE_ITER_HEX = 0x20,
574 TRACE_ITER_BIN = 0x40,
575 TRACE_ITER_BLOCK = 0x80,
576 TRACE_ITER_STACKTRACE = 0x100,
4ac3ba41 577 TRACE_ITER_SCHED_TREE = 0x200,
f09ce573 578 TRACE_ITER_PRINTK = 0x400,
b2a866f9 579 TRACE_ITER_PREEMPTONLY = 0x800,
9f029e83 580 TRACE_ITER_BRANCH = 0x1000,
12ef7d44 581 TRACE_ITER_ANNOTATE = 0x2000,
b54d3de9
TE
582 TRACE_ITER_USERSTACKTRACE = 0x4000,
583 TRACE_ITER_SYM_USEROBJ = 0x8000
4e655519
IM
584};
585
15e6cb36
FW
586/*
587 * TRACE_ITER_SYM_MASK masks the options in trace_flags that
588 * control the output of kernel symbols.
589 */
590#define TRACE_ITER_SYM_MASK \
591 (TRACE_ITER_PRINT_PARENT|TRACE_ITER_SYM_OFFSET|TRACE_ITER_SYM_ADDR)
592
43a15386
FW
593extern struct tracer nop_trace;
594
8f0a056f
SR
595/**
596 * ftrace_preempt_disable - disable preemption scheduler safe
597 *
598 * When tracing can happen inside the scheduler, there exists
599 * cases that the tracing might happen before the need_resched
600 * flag is checked. If this happens and the tracer calls
601 * preempt_enable (after a disable), a schedule might take place
602 * causing an infinite recursion.
603 *
604 * To prevent this, we read the need_recshed flag before
605 * disabling preemption. When we want to enable preemption we
606 * check the flag, if it is set, then we call preempt_enable_no_resched.
607 * Otherwise, we call preempt_enable.
608 *
609 * The rational for doing the above is that if need resched is set
610 * and we have yet to reschedule, we are either in an atomic location
611 * (where we do not need to check for scheduling) or we are inside
612 * the scheduler and do not want to resched.
613 */
614static inline int ftrace_preempt_disable(void)
615{
616 int resched;
617
618 resched = need_resched();
619 preempt_disable_notrace();
620
621 return resched;
622}
623
624/**
625 * ftrace_preempt_enable - enable preemption scheduler safe
626 * @resched: the return value from ftrace_preempt_disable
627 *
628 * This is a scheduler safe way to enable preemption and not miss
629 * any preemption checks. The disabled saved the state of preemption.
630 * If resched is set, then we were either inside an atomic or
631 * are inside the scheduler (we would have already scheduled
632 * otherwise). In this case, we do not want to call normal
633 * preempt_enable, but preempt_enable_no_resched instead.
634 */
635static inline void ftrace_preempt_enable(int resched)
636{
637 if (resched)
638 preempt_enable_no_resched_notrace();
639 else
640 preempt_enable_notrace();
641}
642
2ed84eeb 643#ifdef CONFIG_BRANCH_TRACER
9f029e83
SR
644extern int enable_branch_tracing(struct trace_array *tr);
645extern void disable_branch_tracing(void);
646static inline int trace_branch_enable(struct trace_array *tr)
52f232cb 647{
9f029e83
SR
648 if (trace_flags & TRACE_ITER_BRANCH)
649 return enable_branch_tracing(tr);
52f232cb
SR
650 return 0;
651}
9f029e83 652static inline void trace_branch_disable(void)
52f232cb
SR
653{
654 /* due to races, always disable */
9f029e83 655 disable_branch_tracing();
52f232cb
SR
656}
657#else
9f029e83 658static inline int trace_branch_enable(struct trace_array *tr)
52f232cb
SR
659{
660 return 0;
661}
9f029e83 662static inline void trace_branch_disable(void)
52f232cb
SR
663{
664}
2ed84eeb 665#endif /* CONFIG_BRANCH_TRACER */
52f232cb 666
bc0c38d1 667#endif /* _LINUX_KERNEL_TRACE_H */