vlib: introduce vlib_get_main_by_index(), vlib_get_n_threads()
[vpp.git] / src / plugins / perfmon / perfmon.c
1 /*
2  * Copyright (c) 2020 Cisco and/or its affiliates.
3  * Licensed under the Apache License, Version 2.0 (the "License");
4  * you may not use this file except in compliance with the License.
5  * You may obtain a copy of the License at:
6  *
7  *     http://www.apache.org/licenses/LICENSE-2.0
8  *
9  * Unless required by applicable law or agreed to in writing, software
10  * distributed under the License is distributed on an "AS IS" BASIS,
11  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12  * See the License for the specific language governing permissions and
13  * limitations under the License.
14  */
15
16 #include <vnet/vnet.h>
17
18 #include <vlibapi/api.h>
19 #include <vlibmemory/api.h>
20 #include <vnet/plugin/plugin.h>
21 #include <vpp/app/version.h>
22 #include <linux/limits.h>
23 #include <sys/ioctl.h>
24
25 #include <perfmon/perfmon.h>
26
27 perfmon_main_t perfmon_main;
28
29 VLIB_PLUGIN_REGISTER () = {
30   .version = VPP_BUILD_VER,
31   .description = "Performance Monitor",
32 };
33
34 VLIB_REGISTER_LOG_CLASS (if_default_log, static) = {
35   .class_name = "perfmon",
36 };
37
38 #define log_debug(fmt, ...)                                                   \
39   vlib_log_debug (if_default_log.class, fmt, __VA_ARGS__)
40 #define log_warn(fmt, ...)                                                    \
41   vlib_log_warn (if_default_log.class, fmt, __VA_ARGS__)
42 #define log_err(fmt, ...) vlib_log_err (if_default_log.class, fmt, __VA_ARGS__)
43
44 void
45 perfmon_reset (vlib_main_t *vm)
46 {
47   perfmon_main_t *pm = &perfmon_main;
48   uword page_size = clib_mem_get_page_size ();
49
50   if (pm->is_running)
51     for (int i = 0; i < vlib_get_n_threads (); i++)
52       vlib_node_set_dispatch_wrapper (vlib_get_main_by_index (i), 0);
53
54   for (int i = 0; i < vec_len (pm->fds_to_close); i++)
55     close (pm->fds_to_close[i]);
56   vec_free (pm->fds_to_close);
57   vec_free (pm->group_fds);
58   if (pm->default_instance_type)
59     {
60       perfmon_instance_type_t *it = pm->default_instance_type;
61       for (int i = 0; i < vec_len (it->instances); i++)
62         vec_free (it->instances[i].name);
63       vec_free (it->instances);
64       vec_free (pm->default_instance_type);
65     }
66
67   for (int i = 0; i < vec_len (pm->thread_runtimes); i++)
68     {
69       perfmon_thread_runtime_t *tr = vec_elt_at_index (pm->thread_runtimes, i);
70       vec_free (tr->node_stats);
71       for (int j = 0; j < PERF_MAX_EVENTS; j++)
72         if (tr->mmap_pages[j])
73           munmap (tr->mmap_pages, page_size);
74     }
75   vec_free (pm->thread_runtimes);
76
77   pm->is_running = 0;
78   pm->active_instance_type = 0;
79   pm->active_bundle = 0;
80 }
81
82 clib_error_t *
83 perfmon_set (vlib_main_t *vm, perfmon_bundle_t *b)
84 {
85   clib_error_t *err = 0;
86   perfmon_main_t *pm = &perfmon_main;
87   perfmon_source_t *s;
88   int is_node = 0;
89   int n_nodes = vec_len (vm->node_main.nodes);
90   uword page_size = clib_mem_get_page_size ();
91   u32 instance_type = 0;
92   perfmon_event_t *e;
93   perfmon_instance_type_t *it = 0;
94
95   perfmon_reset (vm);
96
97   s = b->src;
98   ASSERT (b->n_events);
99
100   if (b->type == PERFMON_BUNDLE_TYPE_NODE)
101     is_node = 1;
102
103   if (s->instances_by_type == 0)
104     {
105       vec_add2 (pm->default_instance_type, it, 1);
106       it->name = is_node ? "Thread/Node" : "Thread";
107       for (int i = 0; i < vlib_get_n_threads (); i++)
108         {
109           vlib_worker_thread_t *w = vlib_worker_threads + i;
110           perfmon_instance_t *in;
111           vec_add2 (it->instances, in, 1);
112           in->cpu = w->cpu_id;
113           in->pid = w->lwp;
114           in->name = (char *) format (0, "%s (%u)%c", w->name, i, 0);
115         }
116       if (is_node)
117         vec_validate (pm->thread_runtimes, vlib_get_n_threads () - 1);
118     }
119   else
120     {
121       e = s->events + b->events[0];
122
123       if (e->type_from_instance)
124         {
125           instance_type = e->instance_type;
126           for (int i = 1; i < b->n_events; i++)
127             {
128               e = s->events + b->events[i];
129               ASSERT (e->type_from_instance == 1 &&
130                       e->instance_type == instance_type);
131             }
132         }
133       it = vec_elt_at_index (s->instances_by_type, instance_type);
134     }
135
136   pm->active_instance_type = it;
137
138   for (int i = 0; i < vec_len (it->instances); i++)
139     {
140       perfmon_instance_t *in = vec_elt_at_index (it->instances, i);
141
142       vec_validate (pm->group_fds, i);
143       pm->group_fds[i] = -1;
144
145       for (int j = 0; j < b->n_events; j++)
146         {
147           int fd;
148           perfmon_event_t *e = s->events + b->events[j];
149           struct perf_event_attr pe = {
150             .size = sizeof (struct perf_event_attr),
151             .type = e->type_from_instance ? in->type : e->type,
152             .config = e->config,
153             .exclude_kernel = e->exclude_kernel,
154             .read_format =
155               (PERF_FORMAT_GROUP | PERF_FORMAT_TOTAL_TIME_ENABLED |
156                PERF_FORMAT_TOTAL_TIME_RUNNING),
157             .disabled = 1,
158           };
159
160           log_debug ("perf_event_open pe.type=%u pe.config=0x%x pid=%d "
161                      "cpu=%d group_fd=%d",
162                      pe.type, pe.config, in->pid, in->cpu, pm->group_fds[i]);
163           fd = syscall (__NR_perf_event_open, &pe, in->pid, in->cpu,
164                         pm->group_fds[i], 0);
165
166           if (fd == -1)
167             {
168               err = clib_error_return_unix (0, "perf_event_open");
169               goto error;
170             }
171
172           vec_add1 (pm->fds_to_close, fd);
173
174           if (pm->group_fds[i] == -1)
175             pm->group_fds[i] = fd;
176
177           if (is_node)
178             {
179               perfmon_thread_runtime_t *tr;
180               tr = vec_elt_at_index (pm->thread_runtimes, i);
181               tr->mmap_pages[j] =
182                 mmap (0, page_size, PROT_READ, MAP_SHARED, fd, 0);
183
184               if (tr->mmap_pages[j] == MAP_FAILED)
185                 {
186                   err = clib_error_return_unix (0, "mmap");
187                   goto error;
188                 }
189             }
190         }
191
192       if (is_node)
193         {
194           perfmon_thread_runtime_t *rt;
195           rt = vec_elt_at_index (pm->thread_runtimes, i);
196           rt->n_events = b->n_events;
197           rt->n_nodes = n_nodes;
198           vec_validate_aligned (rt->node_stats, n_nodes - 1,
199                                 CLIB_CACHE_LINE_BYTES);
200         }
201     }
202
203   pm->active_bundle = b;
204
205 error:
206   if (err)
207     {
208       log_err ("%U", format_clib_error, err);
209       perfmon_reset (vm);
210     }
211   return err;
212 }
213
214 clib_error_t *
215 perfmon_start (vlib_main_t *vm)
216 {
217   perfmon_main_t *pm = &perfmon_main;
218   int n_groups = vec_len (pm->group_fds);
219
220   if (n_groups == 0)
221     return clib_error_return (0, "no bundle configured");
222
223   if (pm->is_running == 1)
224     return clib_error_return (0, "already running");
225
226   for (int i = 0; i < n_groups; i++)
227     {
228       if (ioctl (pm->group_fds[i], PERF_EVENT_IOC_ENABLE,
229                  PERF_IOC_FLAG_GROUP) == -1)
230         {
231           perfmon_reset (vm);
232           return clib_error_return_unix (0, "ioctl(PERF_EVENT_IOC_ENABLE)");
233         }
234     }
235   if (pm->active_bundle->type == PERFMON_BUNDLE_TYPE_NODE)
236     {
237       for (int i = 0; i < vlib_get_n_threads (); i++)
238         vlib_node_set_dispatch_wrapper (vlib_get_main_by_index (i),
239                                         perfmon_dispatch_wrapper);
240     }
241   pm->sample_time = vlib_time_now (vm);
242   pm->is_running = 1;
243   return 0;
244 }
245
246 clib_error_t *
247 perfmon_stop (vlib_main_t *vm)
248 {
249   perfmon_main_t *pm = &perfmon_main;
250   int n_groups = vec_len (pm->group_fds);
251
252   if (pm->is_running != 1)
253     return clib_error_return (0, "not running");
254
255   if (pm->active_bundle->type == PERFMON_BUNDLE_TYPE_NODE)
256     {
257       for (int i = 0; i < vlib_get_n_threads (); i++)
258         vlib_node_set_dispatch_wrapper (vlib_get_main_by_index (i), 0);
259     }
260
261   for (int i = 0; i < n_groups; i++)
262     {
263       if (ioctl (pm->group_fds[i], PERF_EVENT_IOC_DISABLE,
264                  PERF_IOC_FLAG_GROUP) == -1)
265         {
266           perfmon_reset (vm);
267           return clib_error_return_unix (0, "ioctl(PERF_EVENT_IOC_DISABLE)");
268         }
269     }
270
271   pm->is_running = 0;
272   pm->sample_time = vlib_time_now (vm) - pm->sample_time;
273   return 0;
274 }
275
276 static clib_error_t *
277 perfmon_init (vlib_main_t *vm)
278 {
279   perfmon_main_t *pm = &perfmon_main;
280   perfmon_source_t *s = pm->sources;
281   perfmon_bundle_t *b = pm->bundles;
282
283   pm->source_by_name = hash_create_string (0, sizeof (uword));
284   while (s)
285     {
286       clib_error_t *err;
287       if (hash_get_mem (pm->source_by_name, s->name) != 0)
288         clib_panic ("duplicate source name '%s'", s->name);
289       if (s->init_fn && ((err = (s->init_fn) (vm, s))))
290         {
291           log_warn ("skipping source '%s' - %U", s->name, format_clib_error,
292                     err);
293           clib_error_free (err);
294           s = s->next;
295           continue;
296         }
297
298       hash_set_mem (pm->source_by_name, s->name, s);
299       log_debug ("source '%s' regisrtered", s->name);
300       s = s->next;
301     }
302
303   pm->bundle_by_name = hash_create_string (0, sizeof (uword));
304   while (b)
305     {
306       clib_error_t *err;
307       uword *p;
308       if (hash_get_mem (pm->bundle_by_name, b->name) != 0)
309         clib_panic ("duplicate bundle name '%s'", b->name);
310
311       if ((p = hash_get_mem (pm->source_by_name, b->source)) == 0)
312         {
313           log_debug ("missing source '%s', skipping bundle '%s'", b->source,
314                      b->name);
315           b = b->next;
316           continue;
317         }
318
319       b->src = (perfmon_source_t *) p[0];
320       if (b->init_fn && ((err = (b->init_fn) (vm, b))))
321         {
322           log_warn ("skipping bundle '%s' - %U", b->name, format_clib_error,
323                     err);
324           clib_error_free (err);
325           b = b->next;
326           continue;
327         }
328
329       hash_set_mem (pm->bundle_by_name, b->name, b);
330       log_debug ("bundle '%s' regisrtered", b->name);
331
332       b = b->next;
333     }
334
335   return 0;
336 }
337
338 VLIB_INIT_FUNCTION (perfmon_init);