]> CyberLeo.Net >> Repos - FreeBSD/FreeBSD.git/blob - lib/libpmc/pmu-events/jevents.c
bhyvectl(8): Normalize the man page date
[FreeBSD/FreeBSD.git] / lib / libpmc / pmu-events / jevents.c
1 /* Parse event JSON files */
2
3 /*
4  * Copyright (c) 2014, Intel Corporation
5  * All rights reserved.
6  *
7  * Redistribution and use in source and binary forms, with or without
8  * modification, are permitted provided that the following conditions are met:
9  *
10  * 1. Redistributions of source code must retain the above copyright notice,
11  * this list of conditions and the following disclaimer.
12  *
13  * 2. Redistributions in binary form must reproduce the above copyright
14  * notice, this list of conditions and the following disclaimer in the
15  * documentation and/or other materials provided with the distribution.
16  *
17  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
18  * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
19  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
20  * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
21  * COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT,
22  * INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
23  * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
24  * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
25  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
26  * STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
27  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED
28  * OF THE POSSIBILITY OF SUCH DAMAGE.
29  *
30  * $FreeBSD$
31  *
32 */
33
34 #include <sys/param.h>
35 #include <sys/resource.h>               /* getrlimit */
36 #include <sys/stat.h>
37 #include <sys/time.h>                   /* getrlimit */
38
39 #include <ctype.h>
40 #include <dirent.h>
41 #include <errno.h>
42 #include <fts.h>
43 #include <ftw.h>
44 #include <libgen.h>
45 #include <limits.h>
46 #include <stdarg.h>
47 #include <stdbool.h>
48 #include <stddef.h>
49 #include <stdio.h>
50 #include <stdlib.h>
51 #include <string.h>
52 #include <unistd.h>
53
54 #include "list.h"
55 #include "jsmn.h"
56 #include "json.h"
57 #include "jevents.h"
58
59 static int
60 nftw_ordered(const char *path, int (*fn)(const char *, const struct stat *, int,
61         struct FTW *), int nfds, int ftwflags);
62
63 int verbose;
64 static char *prog;
65
66 int
67 eprintf(int level, int var, const char *fmt, ...)
68 {
69
70         int ret;
71         va_list args;
72
73         if (var < level)
74                 return 0;
75
76         va_start(args, fmt);
77
78         ret = vfprintf(stderr, fmt, args);
79
80         va_end(args);
81
82         return ret;
83 }
84
85 __attribute__((weak)) char *
86 get_cpu_str(void)
87 {
88         return NULL;
89 }
90
91 static void
92 addfield(char *map, char **dst, const char *sep, const char *a, jsmntok_t *bt)
93 {
94         unsigned int len = strlen(a) + 1 + strlen(sep);
95         int olen = *dst ? strlen(*dst) : 0;
96         int blen = bt ? json_len(bt) : 0;
97         char *out;
98
99         out = realloc(*dst, len + olen + blen);
100         if (!out) {
101                 /* Don't add field in this case */
102                 return;
103         }
104         *dst = out;
105
106         if (!olen)
107                 *(*dst) = 0;
108         else
109                 strcat(*dst, sep);
110         strcat(*dst, a);
111         if (bt)
112                 strncat(*dst, map + bt->start, blen);
113 }
114
115 static void
116 fixname(char *s)
117 {
118         for (; *s; s++)
119                 *s = tolower(*s);
120 }
121
122 static void
123 fixdesc(char *s)
124 {
125         char *e = s + strlen(s);
126
127         /* Remove trailing dots that look ugly in perf list */
128         --e;
129         while (e >= s && isspace(*e))
130                 --e;
131         if (e >= s && *e == '.')
132                 *e = 0;
133 }
134
135 /* Add escapes for '\' so they are proper C strings. */
136 static char *
137 fixregex(char *s)
138 {
139         int len = 0;
140         int esc_count = 0;
141         char *fixed = NULL;
142         char *p, *q;
143
144         /* Count the number of '\' in string */
145         for (p = s; *p; p++) {
146                 ++len;
147                 if (*p == '\\')
148                         ++esc_count;
149         }
150
151         if (esc_count == 0)
152                 return s;
153
154         /* allocate space for a new string */
155         fixed = (char *) malloc(len + 1);
156         if (!fixed)
157                 return NULL;
158
159         /* copy over the characters */
160         q = fixed;
161         for (p = s; *p; p++) {
162                 if (*p == '\\') {
163                         *q = '\\';
164                         ++q;
165                 }
166                 *q = *p;
167                 ++q;
168         }
169         *q = '\0';
170         return fixed;
171 }
172
173 static struct msrmap {
174         const char *num;
175         const char *pname;
176 } msrmap[] = {
177         { "0x3F6", "ldlat=" },
178         { "0x1A6", "offcore_rsp=" },
179         { "0x1A7", "offcore_rsp=" },
180         { "0x3F7", "frontend=" },
181         { NULL, NULL }
182 };
183
184 static struct field {
185         const char *field;
186         const char *kernel;
187 } fields[] = {
188         { "UMask",      "umask=" },
189         { "CounterMask", "cmask=" },
190         { "Invert",     "inv=" },
191         { "AnyThread",  "any=" },
192         { "EdgeDetect", "edge=" },
193         { "SampleAfterValue", "period=" },
194         { "FCMask",     "fc_mask=" },
195         { "PortMask",   "ch_mask=" },
196         { "L3ThreadMask", "l3_thread_mask=" },
197         { "L3SliceMask", "l3_slice_mask=" },
198         { NULL, NULL }
199 };
200
201 static void
202 cut_comma(char *map, jsmntok_t *newval)
203 {
204         int i;
205
206         /* Cut off everything after comma */
207         for (i = newval->start; i < newval->end; i++) {
208                 if (map[i] == ',')
209                         newval->end = i;
210         }
211 }
212
213 static int
214 match_field(char *map, jsmntok_t *field, int nz, char **event, jsmntok_t *val)
215 {
216         struct field *f;
217         jsmntok_t newval = *val;
218
219         for (f = fields; f->field; f++)
220                 if (json_streq(map, field, f->field) && nz) {
221                         cut_comma(map, &newval);
222                         addfield(map, event, ",", f->kernel, &newval);
223                         return 1;
224                 }
225         return 0;
226 }
227
228 static struct msrmap *
229 lookup_msr(char *map, jsmntok_t *val)
230 {
231         jsmntok_t newval = *val;
232         static bool warned;
233         int i;
234
235         cut_comma(map, &newval);
236         for (i = 0; msrmap[i].num; i++)
237                 if (json_streq(map, &newval, msrmap[i].num))
238                         return &msrmap[i];
239         if (!warned) {
240                 warned = true;
241                 pr_err("%s: Unknown MSR in event file %.*s\n", prog,
242                         json_len(val), map + val->start);
243         }
244         return NULL;
245 }
246
247 static struct map {
248         const char *json;
249         const char *perf;
250 } unit_to_pmu[] = {
251         { "CBO", "uncore_cbox" },
252         { "QPI LL", "uncore_qpi" },
253         { "SBO", "uncore_sbox" },
254         { "iMPH-U", "uncore_arb" },
255         {}
256 };
257
258 static const char *
259 field_to_perf(struct map *table, char *map, jsmntok_t *val)
260 {
261         int i;
262
263         for (i = 0; table[i].json; i++) {
264                 if (json_streq(map, val, table[i].json))
265                         return table[i].perf;
266         }
267         return NULL;
268 }
269
270 #define EXPECT(e, t, m) do { if (!(e)) {                        \
271         jsmntok_t *loc = (t);                                   \
272         if (!(t)->start && (t) > tokens)                        \
273                 loc = (t) - 1;                                  \
274         pr_err("%s:%d: " m ", got %s\n", fn,                    \
275                json_line(map, loc),                             \
276                json_name(t));                                   \
277         err = -EIO;                                             \
278         goto out_free;                                          \
279 } } while (0)
280
281 static char *topic;
282
283 static char *
284 get_topic(void)
285 {
286         char *tp;
287         int i;
288
289         /* tp is free'd in process_one_file() */
290         i = asprintf(&tp, "%s", topic);
291         if (i < 0) {
292                 pr_info("%s: asprintf() error %s\n", prog);
293                 return NULL;
294         }
295
296         for (i = 0; i < (int) strlen(tp); i++) {
297                 char c = tp[i];
298
299                 if (c == '-')
300                         tp[i] = ' ';
301                 else if (c == '.') {
302                         tp[i] = '\0';
303                         break;
304                 }
305         }
306
307         return tp;
308 }
309
310 static int
311 add_topic(const char *bname)
312 {
313         free(topic);
314         topic = strdup(bname);
315         if (!topic) {
316                 pr_info("%s: strdup() error %s for file %s\n", prog,
317                                 strerror(errno), bname);
318                 return -ENOMEM;
319         }
320         return 0;
321 }
322
323 struct perf_entry_data {
324         FILE *outfp;
325         char *topic;
326 };
327
328 static int close_table;
329
330 static void
331 print_events_table_prefix(FILE *fp, const char *tblname)
332 {
333         fprintf(fp, "static struct pmu_event %s[] = {\n", tblname);
334         close_table = 1;
335 }
336
337 static int
338 print_events_table_entry(void *data, char *name, const char *event, char *desc,
339     char *long_desc, char *pmu, char *unit, char *perpkg, char *metric_expr,
340     char *metric_name, char *metric_group)
341 {
342         struct perf_entry_data *pd = data;
343         FILE *outfp = pd->outfp;
344         char *etopic = pd->topic;
345
346         /*
347          * TODO: Remove formatting chars after debugging to reduce
348          *       string lengths.
349          */
350         fprintf(outfp, "{\n");
351
352         if (name)
353                 fprintf(outfp, "\t.name = \"%s\",\n", name);
354         if (event)
355                 fprintf(outfp, "\t.event = \"%s\",\n", event);
356         fprintf(outfp, "\t.desc = \"%s\",\n", desc);
357         fprintf(outfp, "\t.topic = \"%s\",\n", etopic);
358         if (long_desc && long_desc[0])
359                 fprintf(outfp, "\t.long_desc = \"%s\",\n", long_desc);
360         if (pmu)
361                 fprintf(outfp, "\t.pmu = \"%s\",\n", pmu);
362         if (unit)
363                 fprintf(outfp, "\t.unit = \"%s\",\n", unit);
364         if (perpkg)
365                 fprintf(outfp, "\t.perpkg = \"%s\",\n", perpkg);
366         if (metric_expr)
367                 fprintf(outfp, "\t.metric_expr = \"%s\",\n", metric_expr);
368         if (metric_name)
369                 fprintf(outfp, "\t.metric_name = \"%s\",\n", metric_name);
370         if (metric_group)
371                 fprintf(outfp, "\t.metric_group = \"%s\",\n", metric_group);
372         fprintf(outfp, "},\n");
373
374         return 0;
375 }
376
377 struct event_struct {
378         struct list_head list;
379         char *name;
380         char *event;
381         char *desc;
382         char *long_desc;
383         char *pmu;
384         char *unit;
385         char *perpkg;
386         char *metric_expr;
387         char *metric_name;
388         char *metric_group;
389 };
390
391 #define ADD_EVENT_FIELD(field) do { if (field) {                \
392         es->field = strdup(field);                              \
393         if (!es->field)                                         \
394                 goto out_free;                                  \
395 } } while (0)
396
397 #define FREE_EVENT_FIELD(field) free(es->field)
398
399 #define TRY_FIXUP_FIELD(field) do { if (es->field && !*field) {\
400         *field = strdup(es->field);                             \
401         if (!*field)                                            \
402                 return -ENOMEM;                                 \
403 } } while (0)
404
405 #define FOR_ALL_EVENT_STRUCT_FIELDS(op) do {                    \
406         op(name);                                               \
407         op(event);                                              \
408         op(desc);                                               \
409         op(long_desc);                                          \
410         op(pmu);                                                \
411         op(unit);                                               \
412         op(perpkg);                                             \
413         op(metric_expr);                                        \
414         op(metric_name);                                        \
415         op(metric_group);                                       \
416 } while (0)
417
418 static LIST_HEAD(arch_std_events);
419
420 static void
421 free_arch_std_events(void)
422 {
423         struct event_struct *es, *next;
424
425         list_for_each_entry_safe(es, next, &arch_std_events, list) {
426                 FOR_ALL_EVENT_STRUCT_FIELDS(FREE_EVENT_FIELD);
427                 list_del(&es->list);
428                 free(es);
429         }
430 }
431
432 static int
433 save_arch_std_events(void *data __unused, char *name, const char *event,
434     char *desc, char *long_desc, char *pmu, char *unit, char *perpkg,
435     char *metric_expr, char *metric_name, char *metric_group)
436 {
437         struct event_struct *es;
438
439         es = malloc(sizeof(*es));
440         if (!es)
441                 return -ENOMEM;
442         memset(es, 0, sizeof(*es));
443         FOR_ALL_EVENT_STRUCT_FIELDS(ADD_EVENT_FIELD);
444         list_add_tail(&es->list, &arch_std_events);
445         return 0;
446 out_free:
447         FOR_ALL_EVENT_STRUCT_FIELDS(FREE_EVENT_FIELD);
448         free(es);
449         return -ENOMEM;
450 }
451
452 static void
453 print_events_table_suffix(FILE *outfp)
454 {
455         fprintf(outfp, "{\n");
456
457         fprintf(outfp, "\t.name = 0,\n");
458         fprintf(outfp, "\t.event = 0,\n");
459         fprintf(outfp, "\t.desc = 0,\n");
460
461         fprintf(outfp, "},\n");
462         fprintf(outfp, "};\n");
463         close_table = 0;
464 }
465
466 static struct fixed {
467         const char *name;
468         const char *event;
469 } fixed[] = {
470         { "inst_retired.any", "event=0xc0" },
471         { "inst_retired.any_p", "event=0xc0" },
472         { "cpu_clk_unhalted.ref", "event=0x0,umask=0x03" },
473         { "cpu_clk_unhalted.thread", "event=0x3c" },
474         { "cpu_clk_unhalted.thread_any", "event=0x3c,any=1" },
475         { NULL, NULL},
476 };
477
478 /*
479  * Handle different fixed counter encodings between JSON and perf.
480  */
481 static const char *
482 real_event(const char *name, char *event)
483 {
484         int i;
485
486         if (!name)
487                 return NULL;
488
489         for (i = 0; fixed[i].name; i++)
490                 if (!strcasecmp(name, fixed[i].name))
491                         return fixed[i].event;
492         return event;
493 }
494
495 static int
496 try_fixup(const char *fn, char *arch_std, char **event, char **desc,
497     char **name, char **long_desc, char **pmu, char **filter __unused,
498     char **perpkg, char **unit, char **metric_expr, char **metric_name,
499     char **metric_group, unsigned long long eventcode)
500 {
501         /* try to find matching event from arch standard values */
502         struct event_struct *es;
503
504         list_for_each_entry(es, &arch_std_events, list) {
505                 if (!strcmp(arch_std, es->name)) {
506                         if (!eventcode && es->event) {
507                                 /* allow EventCode to be overridden */
508                                 free(*event);
509                                 *event = NULL;
510                         }
511                         FOR_ALL_EVENT_STRUCT_FIELDS(TRY_FIXUP_FIELD);
512                         return 0;
513                 }
514         }
515
516         pr_err("%s: could not find matching %s for %s\n",
517                                         prog, arch_std, fn);
518         return -1;
519 }
520
521 /* Call func with each event in the json file */
522 int
523 json_events(const char *fn,
524           int (*func)(void *data, char *name, const char *event, char *desc,
525                       char *long_desc,
526                       char *pmu, char *unit, char *perpkg,
527                       char *metric_expr,
528                       char *metric_name, char *metric_group),
529           void *data)
530 {
531         int err;
532         size_t size;
533         jsmntok_t *tokens, *tok;
534         int i, j, len;
535         char *map;
536         char buf[128];
537
538         if (!fn)
539                 return -ENOENT;
540
541         tokens = parse_json(fn, &map, &size, &len);
542         if (!tokens)
543                 return -EIO;
544         EXPECT(tokens->type == JSMN_ARRAY, tokens, "expected top level array");
545         tok = tokens + 1;
546         for (i = 0; i < tokens->size; i++) {
547                 char *event = NULL, *desc = NULL, *name = NULL;
548                 char *long_desc = NULL;
549                 char *extra_desc = NULL;
550                 char *pmu = NULL;
551                 char *filter = NULL;
552                 char *perpkg = NULL;
553                 char *unit = NULL;
554                 char *metric_expr = NULL;
555                 char *metric_name = NULL;
556                 char *metric_group = NULL;
557                 char *arch_std = NULL;
558                 unsigned long long eventcode = 0;
559                 struct msrmap *msr = NULL;
560                 jsmntok_t *msrval = NULL;
561                 jsmntok_t *precise = NULL;
562                 jsmntok_t *obj = tok++;
563
564                 EXPECT(obj->type == JSMN_OBJECT, obj, "expected object");
565                 for (j = 0; j < obj->size; j += 2) {
566                         jsmntok_t *field, *val;
567                         int nz;
568                         char *s;
569
570                         field = tok + j;
571                         EXPECT(field->type == JSMN_STRING, tok + j,
572                                "Expected field name");
573                         val = tok + j + 1;
574                         EXPECT(val->type == JSMN_STRING, tok + j + 1,
575                                "Expected string value");
576
577                         nz = !json_streq(map, val, "0");
578                         if (match_field(map, field, nz, &event, val)) {
579                                 /* ok */
580                         } else if (json_streq(map, field, "EventCode")) {
581                                 char *code = NULL;
582                                 addfield(map, &code, "", "", val);
583                                 eventcode |= strtoul(code, NULL, 0);
584                                 free(code);
585                         } else if (json_streq(map, field, "ExtSel")) {
586                                 char *code = NULL;
587                                 addfield(map, &code, "", "", val);
588                                 eventcode |= strtoul(code, NULL, 0) << 21;
589                                 free(code);
590                         } else if (json_streq(map, field, "EventName")) {
591                                 addfield(map, &name, "", "", val);
592                         } else if (json_streq(map, field, "BriefDescription")) {
593                                 addfield(map, &desc, "", "", val);
594                                 fixdesc(desc);
595                         } else if (json_streq(map, field,
596                                              "PublicDescription")) {
597                                 addfield(map, &long_desc, "", "", val);
598                                 fixdesc(long_desc);
599                         } else if (json_streq(map, field, "PEBS") && nz) {
600                                 precise = val;
601                         } else if (json_streq(map, field, "MSRIndex") && nz) {
602                                 msr = lookup_msr(map, val);
603                         } else if (json_streq(map, field, "MSRValue")) {
604                                 msrval = val;
605                         } else if (json_streq(map, field, "Errata") &&
606                                    !json_streq(map, val, "null")) {
607                                 addfield(map, &extra_desc, ". ",
608                                         " Spec update: ", val);
609                         } else if (json_streq(map, field, "Data_LA") && nz) {
610                                 addfield(map, &extra_desc, ". ",
611                                         " Supports address when precise",
612                                         NULL);
613                         } else if (json_streq(map, field, "Unit")) {
614                                 const char *ppmu;
615
616                                 ppmu = field_to_perf(unit_to_pmu, map, val);
617                                 if (ppmu) {
618                                         pmu = strdup(ppmu);
619                                 } else {
620                                         if (!pmu)
621                                                 pmu = strdup("uncore_");
622                                         addfield(map, &pmu, "", "", val);
623                                         for (s = pmu; *s; s++)
624                                                 *s = tolower(*s);
625                                 }
626                                 addfield(map, &desc, ". ", "Unit: ", NULL);
627                                 addfield(map, &desc, "", pmu, NULL);
628                                 addfield(map, &desc, "", " ", NULL);
629                         } else if (json_streq(map, field, "Filter")) {
630                                 addfield(map, &filter, "", "", val);
631                         } else if (json_streq(map, field, "ScaleUnit")) {
632                                 addfield(map, &unit, "", "", val);
633                         } else if (json_streq(map, field, "PerPkg")) {
634                                 addfield(map, &perpkg, "", "", val);
635                         } else if (json_streq(map, field, "MetricName")) {
636                                 addfield(map, &metric_name, "", "", val);
637                         } else if (json_streq(map, field, "MetricGroup")) {
638                                 addfield(map, &metric_group, "", "", val);
639                         } else if (json_streq(map, field, "MetricExpr")) {
640                                 addfield(map, &metric_expr, "", "", val);
641                                 for (s = metric_expr; *s; s++)
642                                         *s = tolower(*s);
643                         } else if (json_streq(map, field, "ArchStdEvent")) {
644                                 addfield(map, &arch_std, "", "", val);
645                                 for (s = arch_std; *s; s++)
646                                         *s = tolower(*s);
647                         }
648                         /* ignore unknown fields */
649                 }
650                 if (precise && desc && !strstr(desc, "(Precise Event)")) {
651                         if (json_streq(map, precise, "2"))
652                                 addfield(map, &extra_desc, " ",
653                                                 "(Must be precise)", NULL);
654                         else
655                                 addfield(map, &extra_desc, " ",
656                                                 "(Precise event)", NULL);
657                 }
658                 snprintf(buf, sizeof(buf), "event=%#llx", eventcode);
659                 addfield(map, &event, ",", buf, NULL);
660                 if (desc && extra_desc)
661                         addfield(map, &desc, " ", extra_desc, NULL);
662                 if (long_desc && extra_desc)
663                         addfield(map, &long_desc, " ", extra_desc, NULL);
664                 if (filter)
665                         addfield(map, &event, ",", filter, NULL);
666                 if (msr != NULL)
667                         addfield(map, &event, ",", msr->pname, msrval);
668                 if (name)
669                         fixname(name);
670
671                 if (arch_std) {
672                         /*
673                          * An arch standard event is referenced, so try to
674                          * fixup any unassigned values.
675                          */
676                         err = try_fixup(fn, arch_std, &event, &desc, &name,
677                                         &long_desc, &pmu, &filter, &perpkg,
678                                         &unit, &metric_expr, &metric_name,
679                                         &metric_group, eventcode);
680                         if (err)
681                                 goto free_strings;
682                 }
683                 err = func(data, name, real_event(name, event), desc, long_desc,
684                            pmu, unit, perpkg, metric_expr, metric_name, metric_group);
685 free_strings:
686                 free(event);
687                 free(desc);
688                 free(name);
689                 free(long_desc);
690                 free(extra_desc);
691                 free(pmu);
692                 free(filter);
693                 free(perpkg);
694                 free(unit);
695                 free(metric_expr);
696                 free(metric_name);
697                 free(metric_group);
698                 free(arch_std);
699
700                 if (err)
701                         break;
702                 tok += j;
703         }
704         EXPECT(tok - tokens == len, tok, "unexpected objects at end");
705         err = 0;
706 out_free:
707         free_json(map, size, tokens);
708         return err;
709 }
710
711 static char *
712 file_name_to_table_name(const char *fname)
713 {
714         unsigned int i;
715         int n;
716         int c;
717         char *tblname;
718
719
720         /*
721          * Ensure tablename starts with alphabetic character.
722          * Derive rest of table name from basename of the JSON file,
723          * replacing hyphens and stripping out .json suffix.
724          */
725         n = asprintf(&tblname, "pme_%s", fname);
726         if (n < 0) {
727                 pr_info("%s: asprintf() error %s for file %s\n", prog,
728                                 strerror(errno), fname);
729                 return NULL;
730         }
731
732         for (i = 0; i < strlen(tblname); i++) {
733                 c = tblname[i];
734
735                 if (c == '-' || c == '/')
736                         tblname[i] = '_';
737                 else if (c == '.') {
738                         tblname[i] = '\0';
739                         break;
740                 } else if (!isalnum(c) && c != '_') {
741                         char *tmp = strdup(fname);
742                         pr_err("%s: Invalid character '%c' in file name %s\n",
743                                         prog, c, basename(tmp));
744                         free(tblname);
745                         free(tmp);
746                         tblname = NULL;
747                         break;
748                 }
749         }
750
751         return tblname;
752 }
753
754 static void
755 print_mapping_table_prefix(FILE *outfp)
756 {
757         fprintf(outfp, "struct pmu_events_map pmu_events_map[] = {\n");
758 }
759
760 static void
761 print_mapping_table_suffix(FILE *outfp)
762 {
763         /*
764          * Print the terminating, NULL entry.
765          */
766         fprintf(outfp, "{\n");
767         fprintf(outfp, "\t.cpuid = 0,\n");
768         fprintf(outfp, "\t.version = 0,\n");
769         fprintf(outfp, "\t.type = 0,\n");
770         fprintf(outfp, "\t.table = 0,\n");
771         fprintf(outfp, "},\n");
772
773         /* and finally, the closing curly bracket for the struct */
774         fprintf(outfp, "};\n");
775 }
776
777 static int
778 process_mapfile(FILE *outfp, char *fpath)
779 {
780         int n = 16384;
781         FILE *mapfp;
782         char *save = NULL;
783         char *line, *p;
784         int line_num;
785         char *tblname;
786
787         pr_info("%s: Processing mapfile %s\n", prog, fpath);
788
789         line = malloc(n);
790         if (!line)
791                 return -1;
792
793         mapfp = fopen(fpath, "r");
794         if (!mapfp) {
795                 pr_info("%s: Error %s opening %s\n", prog, strerror(errno),
796                                 fpath);
797                 free(line);
798                 return -1;
799         }
800
801         print_mapping_table_prefix(outfp);
802
803         /* Skip first line (header) */
804         p = fgets(line, n, mapfp);
805         if (!p)
806                 goto out;
807
808         line_num = 1;
809         while (1) {
810                 char *cpuid, *version, *type, *fname;
811
812                 line_num++;
813                 p = fgets(line, n, mapfp);
814                 if (!p)
815                         break;
816
817                 if (line[0] == '#' || line[0] == '\n')
818                         continue;
819
820                 if (line[strlen(line)-1] != '\n') {
821                         /* TODO Deal with lines longer than 16K */
822                         pr_info("%s: Mapfile %s: line %d too long, aborting\n",
823                                         prog, fpath, line_num);
824                         free(line);
825                         fclose(mapfp);
826                         return -1;
827                 }
828                 line[strlen(line)-1] = '\0';
829
830                 cpuid = fixregex(strtok_r(p, ",", &save));
831                 version = strtok_r(NULL, ",", &save);
832                 fname = strtok_r(NULL, ",", &save);
833                 type = strtok_r(NULL, ",", &save);
834
835                 tblname = file_name_to_table_name(fname);
836                 fprintf(outfp, "{\n");
837                 fprintf(outfp, "\t.cpuid = \"%s\",\n", cpuid);
838                 fprintf(outfp, "\t.version = \"%s\",\n", version);
839                 fprintf(outfp, "\t.type = \"%s\",\n", type);
840
841                 /*
842                  * CHECK: We can't use the type (eg "core") field in the
843                  * table name. For us to do that, we need to somehow tweak
844                  * the other caller of file_name_to_table(), process_json()
845                  * to determine the type. process_json() file has no way
846                  * of knowing these are "core" events unless file name has
847                  * core in it. If filename has core in it, we can safely
848                  * ignore the type field here also.
849                  */
850                 fprintf(outfp, "\t.table = %s\n", tblname);
851                 fprintf(outfp, "},\n");
852         }
853
854 out:
855         print_mapping_table_suffix(outfp);
856         free(line);
857         fclose(mapfp);
858         return 0;
859 }
860
861 /*
862  * If we fail to locate/process JSON and map files, create a NULL mapping
863  * table. This would at least allow perf to build even if we can't find/use
864  * the aliases.
865  */
866 static void
867 create_empty_mapping(const char *output_file)
868 {
869         FILE *outfp;
870
871         pr_info("%s: Creating empty pmu_events_map[] table\n", prog);
872
873         /* Truncate file to clear any partial writes to it */
874         outfp = fopen(output_file, "w");
875         if (!outfp) {
876                 perror("fopen()");
877                 _Exit(1);
878         }
879
880         fprintf(outfp, "#include \"pmu-events/pmu-events.h\"\n");
881         print_mapping_table_prefix(outfp);
882         print_mapping_table_suffix(outfp);
883         fclose(outfp);
884 }
885
886 static int
887 get_maxfds(void)
888 {
889         struct rlimit rlim;
890
891         if (getrlimit(RLIMIT_NOFILE, &rlim) == 0) {
892                 if (rlim.rlim_max == RLIM_INFINITY)
893                         return 512;
894                 return MIN(rlim.rlim_max / 2, 512);
895         }
896
897         return 512;
898 }
899
900 /*
901  * nftw() doesn't let us pass an argument to the processing function,
902  * so use a global variables.
903  */
904 static FILE *eventsfp;
905 static char *mapfile;
906
907 static int
908 is_leaf_dir(const char *fpath)
909 {
910         DIR *d;
911         struct dirent *dir;
912         int res = 1;
913
914         d = opendir(fpath);
915         if (!d)
916                 return 0;
917
918         while ((dir = readdir(d)) != NULL) {
919                 if (!strcmp(dir->d_name, ".") || !strcmp(dir->d_name, ".."))
920                         continue;
921
922                 if (dir->d_type == DT_DIR) {
923                         res = 0;
924                         break;
925                 } else if (dir->d_type == DT_UNKNOWN) {
926                         char path[PATH_MAX];
927                         struct stat st;
928
929                         snprintf(path, sizeof(path), "%s/%s", fpath,
930                             dir->d_name);
931                         if (stat(path, &st))
932                                 break;
933
934                         if (S_ISDIR(st.st_mode)) {
935                                 res = 0;
936                                 break;
937                         }
938                 }
939         }
940
941         closedir(d);
942
943         return res;
944 }
945
946 static int
947 is_json_file(const char *name)
948 {
949         const char *suffix;
950
951         if (strlen(name) < 5)
952                 return 0;
953
954         suffix = name + strlen(name) - 5;
955
956         if (strncmp(suffix, ".json", 5) == 0)
957                 return 1;
958         return 0;
959 }
960
961 static int
962 preprocess_arch_std_files(const char *fpath, const struct stat *sb,
963     int typeflag, struct FTW *ftwbuf)
964 {
965         int level = ftwbuf->level;
966         int is_file = typeflag == FTW_F;
967
968         if (level == 1 && is_file && is_json_file(fpath))
969                 return json_events(fpath, save_arch_std_events, (void *)(uintptr_t)sb);
970
971         return 0;
972 }
973
974 static int
975 process_one_file(const char *fpath, const struct stat *sb, int typeflag,
976     struct FTW *ftwbuf)
977 {
978         char *tblname;
979         const char *bname;
980         int is_dir  = typeflag == FTW_D;
981         int is_file = typeflag == FTW_F;
982         int level   = ftwbuf->level;
983         int err = 0;
984
985         if (level == 2 && is_dir) {
986                 /*
987                  * For level 2 directory, bname will include parent name,
988                  * like vendor/platform. So search back from platform dir
989                  * to find this.
990                  */
991                 bname = fpath + ftwbuf->base - 2;
992                 for (;;) {
993                         if (*bname == '/')
994                                 break;
995                         bname--;
996                 }
997                 bname++;
998         } else
999                 bname = fpath + ftwbuf->base;
1000
1001         pr_debug("%s %d %7jd %-20s %s\n",
1002                  is_file ? "f" : is_dir ? "d" : "x",
1003                  level, sb->st_size, bname, fpath);
1004
1005         /* base dir or too deep */
1006         if (level == 0 || level > 3)
1007                 return 0;
1008
1009
1010         /* model directory, reset topic */
1011         if ((level == 1 && is_dir && is_leaf_dir(fpath)) ||
1012             (level == 2 && is_dir)) {
1013                 if (close_table)
1014                         print_events_table_suffix(eventsfp);
1015
1016                 /*
1017                  * Drop file name suffix. Replace hyphens with underscores.
1018                  * Fail if file name contains any alphanum characters besides
1019                  * underscores.
1020                  */
1021                 tblname = file_name_to_table_name(bname);
1022                 if (!tblname) {
1023                         pr_info("%s: Error determining table name for %s\n", prog,
1024                                 bname);
1025                         return -1;
1026                 }
1027
1028                 print_events_table_prefix(eventsfp, tblname);
1029                 return 0;
1030         }
1031
1032         /*
1033          * Save the mapfile name for now. We will process mapfile
1034          * after processing all JSON files (so we can write out the
1035          * mapping table after all PMU events tables).
1036          *
1037          */
1038         if (level == 1 && is_file) {
1039                 if (!strcmp(bname, "mapfile.csv")) {
1040                         mapfile = strdup(fpath);
1041                         return 0;
1042                 }
1043
1044                 pr_info("%s: Ignoring file %s\n", prog, fpath);
1045                 return 0;
1046         }
1047
1048         /*
1049          * If the file name does not have a .json extension,
1050          * ignore it. It could be a readme.txt for instance.
1051          */
1052         if (is_file) {
1053                 if (!is_json_file(bname)) {
1054                         pr_info("%s: Ignoring file without .json suffix %s\n", prog,
1055                                 fpath);
1056                         return 0;
1057                 }
1058         }
1059
1060         if (level > 1 && add_topic(bname))
1061                 return -ENOMEM;
1062
1063         /*
1064          * Assume all other files are JSON files.
1065          *
1066          * If mapfile refers to 'power7_core.json', we create a table
1067          * named 'power7_core'. Any inconsistencies between the mapfile
1068          * and directory tree could result in build failure due to table
1069          * names not being found.
1070          *
1071          * Atleast for now, be strict with processing JSON file names.
1072          * i.e. if JSON file name cannot be mapped to C-style table name,
1073          * fail.
1074          */
1075         if (is_file) {
1076                 struct perf_entry_data data = {
1077                         .topic = get_topic(),
1078                         .outfp = eventsfp,
1079                 };
1080
1081                 err = json_events(fpath, print_events_table_entry, &data);
1082
1083                 free(data.topic);
1084         }
1085
1086         return err;
1087 }
1088
1089 /*
1090  * Starting in directory 'start_dirname', find the "mapfile.csv" and
1091  * the set of JSON files for the architecture 'arch'.
1092  *
1093  * From each JSON file, create a C-style "PMU events table" from the
1094  * JSON file (see struct pmu_event).
1095  *
1096  * From the mapfile, create a mapping between the CPU revisions and
1097  * PMU event tables (see struct pmu_events_map).
1098  *
1099  * Write out the PMU events tables and the mapping table to pmu-event.c.
1100  */
1101 int
1102 main(int argc, char *argv[])
1103 {
1104         int rc;
1105         int maxfds;
1106         char ldirname[PATH_MAX];
1107
1108         const char *arch;
1109         const char *output_file;
1110         const char *start_dirname;
1111         struct stat stbuf;
1112
1113         prog = basename(argv[0]);
1114         if (argc < 4) {
1115                 pr_err("Usage: %s <arch> <starting_dir> <output_file>\n", prog);
1116                 return 1;
1117         }
1118
1119         arch = argv[1];
1120         start_dirname = argv[2];
1121         output_file = argv[3];
1122
1123         if (argc > 4)
1124                 verbose = atoi(argv[4]);
1125
1126         eventsfp = fopen(output_file, "w");
1127         if (!eventsfp) {
1128                 pr_err("%s Unable to create required file %s (%s)\n",
1129                                 prog, output_file, strerror(errno));
1130                 return 2;
1131         }
1132
1133         snprintf(ldirname, sizeof(ldirname), "%s/%s", start_dirname, arch);
1134
1135         /* If architecture does not have any event lists, bail out */
1136         if (stat(ldirname, &stbuf) < 0) {
1137                 pr_info("%s: Arch %s has no PMU event lists\n", prog, arch);
1138                 goto empty_map;
1139         }
1140
1141         /* Include pmu-events.h first */
1142         fprintf(eventsfp, "#include \"pmu-events/pmu-events.h\"\n");
1143
1144         /*
1145          * The mapfile allows multiple CPUids to point to the same JSON file,
1146          * so, not sure if there is a need for symlinks within the pmu-events
1147          * directory.
1148          *
1149          * For now, treat symlinks of JSON files as regular files and create
1150          * separate tables for each symlink (presumably, each symlink refers
1151          * to specific version of the CPU).
1152          */
1153
1154         maxfds = get_maxfds();
1155         mapfile = NULL;
1156         rc = nftw_ordered(ldirname, preprocess_arch_std_files, maxfds, 0);
1157         if (rc && verbose) {
1158                 pr_info("%s: Error preprocessing arch standard files %s: %s\n",
1159                         prog, ldirname, strerror(errno));
1160                 goto empty_map;
1161         } else if (rc < 0) {
1162                 /* Make build fail */
1163                 free_arch_std_events();
1164                 return 1;
1165         } else if (rc) {
1166                 goto empty_map;
1167         }
1168
1169         rc = nftw_ordered(ldirname, process_one_file, maxfds, 0);
1170         if (rc && verbose) {
1171                 pr_info("%s: Error walking file tree %s\n", prog, ldirname);
1172                 goto empty_map;
1173         } else if (rc < 0) {
1174                 /* Make build fail */
1175                 free_arch_std_events();
1176                 return 1;
1177         } else if (rc) {
1178                 goto empty_map;
1179         }
1180
1181         if (close_table)
1182                 print_events_table_suffix(eventsfp);
1183
1184         if (!mapfile) {
1185                 pr_info("%s: No CPU->JSON mapping?\n", prog);
1186                 goto empty_map;
1187         }
1188
1189         if (process_mapfile(eventsfp, mapfile)) {
1190                 pr_info("%s: Error processing mapfile %s\n", prog, mapfile);
1191                 /* Make build fail */
1192                 return 1;
1193         }
1194
1195         return 0;
1196
1197 empty_map:
1198         fclose(eventsfp);
1199         create_empty_mapping(output_file);
1200         free_arch_std_events();
1201         return 0;
1202 }
1203
1204 static int
1205 fts_compare(const FTSENT * const *a, const FTSENT * const *b)
1206 {
1207         return (strcmp((*a)->fts_name, (*b)->fts_name));
1208 }
1209
1210 static int
1211 nftw_ordered(const char *path, int (*fn)(const char *, const struct stat *, int,
1212      struct FTW *), int nfds, int ftwflags)
1213 {
1214         char * const paths[2] = { (char *)path, NULL };
1215         struct FTW ftw;
1216         FTSENT *cur;
1217         FTS *ftsp;
1218         int error = 0, ftsflags, fnflag, postorder, sverrno;
1219
1220         /* XXX - nfds is currently unused */
1221         if (nfds < 1) {
1222                 errno = EINVAL;
1223                 return (-1);
1224         }
1225
1226         ftsflags = FTS_COMFOLLOW;
1227         if (!(ftwflags & FTW_CHDIR))
1228                 ftsflags |= FTS_NOCHDIR;
1229         if (ftwflags & FTW_MOUNT)
1230                 ftsflags |= FTS_XDEV;
1231         if (ftwflags & FTW_PHYS)
1232                 ftsflags |= FTS_PHYSICAL;
1233         else
1234                 ftsflags |= FTS_LOGICAL;
1235         postorder = (ftwflags & FTW_DEPTH) != 0;
1236         ftsp = fts_open(paths, ftsflags, fts_compare);
1237         if (ftsp == NULL)
1238                 return (-1);
1239         while ((cur = fts_read(ftsp)) != NULL) {
1240                 switch (cur->fts_info) {
1241                 case FTS_D:
1242                         if (postorder)
1243                                 continue;
1244                         fnflag = FTW_D;
1245                         break;
1246                 case FTS_DC:
1247                         continue;
1248                 case FTS_DNR:
1249                         fnflag = FTW_DNR;
1250                         break;
1251                 case FTS_DP:
1252                         if (!postorder)
1253                                 continue;
1254                         fnflag = FTW_DP;
1255                         break;
1256                 case FTS_F:
1257                 case FTS_DEFAULT:
1258                         fnflag = FTW_F;
1259                         break;
1260                 case FTS_NS:
1261                 case FTS_NSOK:
1262                         fnflag = FTW_NS;
1263                         break;
1264                 case FTS_SL:
1265                         fnflag = FTW_SL;
1266                         break;
1267                 case FTS_SLNONE:
1268                         fnflag = FTW_SLN;
1269                         break;
1270                 default:
1271                         error = -1;
1272                         goto done;
1273                 }
1274                 ftw.base = cur->fts_pathlen - cur->fts_namelen;
1275                 ftw.level = cur->fts_level;
1276                 error = fn(cur->fts_path, cur->fts_statp, fnflag, &ftw);
1277                 if (error != 0)
1278                         break;
1279         }
1280 done:
1281         sverrno = errno;
1282         if (fts_close(ftsp) != 0 && error == 0)
1283                 error = -1;
1284         else
1285                 errno = sverrno;
1286         return (error);
1287 }