tools/lib/traceevent/plugins/plugin_sched_switch.c

   1 // SPDX-License-Identifier: LGPL-2.1
   2 /*
   3  * Copyright (C) 2009, 2010 Red Hat Inc, Steven Rostedt <srostedt@redhat.com>
   4  */
   5 #include <stdio.h>
   6 #include <stdlib.h>
   7 #include <string.h>
   8
   9 #include "event-parse.h"
  10 #include "trace-seq.h"
  11
  12 static void write_state(struct trace_seq *s, int val)
  13 {
  14         const char states[] = "SDTtZXxW";
  15         int found = 0;
  16         int i;
  17
  18         for (i = 0; i < (sizeof(states) - 1); i++) {
  19                 if (!(val & (1 << i)))
  20                         continue;
  21
  22                 if (found)
  23                         trace_seq_putc(s, '|');
  24
  25                 found = 1;
  26                 trace_seq_putc(s, states[i]);
  27         }
  28
  29         if (!found)
  30                 trace_seq_putc(s, 'R');
  31 }
  32
  33 static void write_and_save_comm(struct tep_format_field *field,
  34                                 struct tep_record *record,
  35                                 struct trace_seq *s, int pid)
  36 {
  37         const char *comm;
  38         int len;
  39
  40         comm = (char *)(record->data + field->offset);
  41         len = s->len;
  42         trace_seq_printf(s, "%.*s",
  43                          field->size, comm);
  44
  45         /* make sure the comm has a \0 at the end. */
  46         trace_seq_terminate(s);
  47         comm = &s->buffer[len];
  48
  49         /* Help out the comm to ids. This will handle dups */
  50         tep_register_comm(field->event->tep, comm, pid);
  51 }
  52
  53 static int sched_wakeup_handler(struct trace_seq *s,
  54                                 struct tep_record *record,
  55                                 struct tep_event *event, void *context)
  56 {
  57         struct tep_format_field *field;
  58         unsigned long long val;
  59
  60         if (tep_get_field_val(s, event, "pid", record, &val, 1))
  61                 return trace_seq_putc(s, '!');
  62
  63         field = tep_find_any_field(event, "comm");
  64         if (field) {
  65                 write_and_save_comm(field, record, s, val);
  66                 trace_seq_putc(s, ':');
  67         }
  68         trace_seq_printf(s, "%lld", val);
  69
  70         if (tep_get_field_val(s, event, "prio", record, &val, 0) == 0)
  71                 trace_seq_printf(s, " [%lld]", val);
  72
  73         if (tep_get_field_val(s, event, "success", record, &val, 1) == 0)
  74                 trace_seq_printf(s, " success=%lld", val);
  75
  76         if (tep_get_field_val(s, event, "target_cpu", record, &val, 0) == 0)
  77                 trace_seq_printf(s, " CPU:%03llu", val);
  78
  79         return 0;
  80 }
  81
  82 static int sched_switch_handler(struct trace_seq *s,
  83                                 struct tep_record *record,
  84                                 struct tep_event *event, void *context)
  85 {
  86         struct tep_format_field *field;
  87         unsigned long long val;
  88
  89         if (tep_get_field_val(s, event, "prev_pid", record, &val, 1))
  90                 return trace_seq_putc(s, '!');
  91
  92         field = tep_find_any_field(event, "prev_comm");
  93         if (field) {
  94                 write_and_save_comm(field, record, s, val);
  95                 trace_seq_putc(s, ':');
  96         }
  97         trace_seq_printf(s, "%lld ", val);
  98
  99         if (tep_get_field_val(s, event, "prev_prio", record, &val, 0) == 0)
 100                 trace_seq_printf(s, "[%d] ", (int) val);
 101
 102         if (tep_get_field_val(s,  event, "prev_state", record, &val, 0) == 0)
 103                 write_state(s, val);
 104
 105         trace_seq_puts(s, " ==> ");
 106
 107         if (tep_get_field_val(s, event, "next_pid", record, &val, 1))
 108                 return trace_seq_putc(s, '!');
 109
 110         field = tep_find_any_field(event, "next_comm");
 111         if (field) {
 112                 write_and_save_comm(field, record, s, val);
 113                 trace_seq_putc(s, ':');
 114         }
 115         trace_seq_printf(s, "%lld", val);
 116
 117         if (tep_get_field_val(s, event, "next_prio", record, &val, 0) == 0)
 118                 trace_seq_printf(s, " [%d]", (int) val);
 119
 120         return 0;
 121 }
 122
 123 int TEP_PLUGIN_LOADER(struct tep_handle *tep)
 124 {
 125         tep_register_event_handler(tep, -1, "sched", "sched_switch",
 126                                    sched_switch_handler, NULL);
 127
 128         tep_register_event_handler(tep, -1, "sched", "sched_wakeup",
 129                                    sched_wakeup_handler, NULL);
 130
 131         tep_register_event_handler(tep, -1, "sched", "sched_wakeup_new",
 132                                    sched_wakeup_handler, NULL);
 133         return 0;
 134 }
 135
 136 void TEP_PLUGIN_UNLOADER(struct tep_handle *tep)
 137 {
 138         tep_unregister_event_handler(tep, -1, "sched", "sched_switch",
 139                                      sched_switch_handler, NULL);
 140
 141         tep_unregister_event_handler(tep, -1, "sched", "sched_wakeup",
 142                                      sched_wakeup_handler, NULL);
 143
 144         tep_unregister_event_handler(tep, -1, "sched", "sched_wakeup_new",
 145                                      sched_wakeup_handler, NULL);
 146 }