1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18#define pr_fmt(fmt) "gcov: " fmt
19
20#include <linux/init.h>
21#include <linux/module.h>
22#include <linux/debugfs.h>
23#include <linux/fs.h>
24#include <linux/list.h>
25#include <linux/string.h>
26#include <linux/slab.h>
27#include <linux/mutex.h>
28#include <linux/seq_file.h>
29#include <linux/mm.h>
30#include "gcov.h"
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52struct gcov_node {
53 struct list_head list;
54 struct list_head children;
55 struct list_head all;
56 struct gcov_node *parent;
57 struct gcov_info **loaded_info;
58 struct gcov_info *unloaded_info;
59 struct dentry *dentry;
60 struct dentry **links;
61 int num_loaded;
62 char name[];
63};
64
65static const char objtree[] = OBJTREE;
66static const char srctree[] = SRCTREE;
67static struct gcov_node root_node;
68static LIST_HEAD(all_head);
69static DEFINE_MUTEX(node_lock);
70
71
72static int gcov_persist = 1;
73
74static int __init gcov_persist_setup(char *str)
75{
76 unsigned long val;
77
78 if (kstrtoul(str, 0, &val)) {
79 pr_warn("invalid gcov_persist parameter '%s'\n", str);
80 return 0;
81 }
82 gcov_persist = val;
83 pr_info("setting gcov_persist to %d\n", gcov_persist);
84
85 return 1;
86}
87__setup("gcov_persist=", gcov_persist_setup);
88
89#define ITER_STRIDE PAGE_SIZE
90
91
92
93
94
95
96
97
98struct gcov_iterator {
99 struct gcov_info *info;
100 size_t size;
101 loff_t pos;
102 char buffer[];
103};
104
105
106
107
108
109
110
111static struct gcov_iterator *gcov_iter_new(struct gcov_info *info)
112{
113 struct gcov_iterator *iter;
114 size_t size;
115
116
117 size = convert_to_gcda(NULL, info);
118
119 iter = kvmalloc(struct_size(iter, buffer, size), GFP_KERNEL);
120 if (!iter)
121 return NULL;
122
123 iter->info = info;
124 iter->size = size;
125 convert_to_gcda(iter->buffer, info);
126
127 return iter;
128}
129
130
131
132
133
134
135static void gcov_iter_free(struct gcov_iterator *iter)
136{
137 kvfree(iter);
138}
139
140
141
142
143
144static struct gcov_info *gcov_iter_get_info(struct gcov_iterator *iter)
145{
146 return iter->info;
147}
148
149
150
151
152
153static void gcov_iter_start(struct gcov_iterator *iter)
154{
155 iter->pos = 0;
156}
157
158
159
160
161
162
163
164static int gcov_iter_next(struct gcov_iterator *iter)
165{
166 if (iter->pos < iter->size)
167 iter->pos += ITER_STRIDE;
168
169 if (iter->pos >= iter->size)
170 return -EINVAL;
171
172 return 0;
173}
174
175
176
177
178
179
180
181
182static int gcov_iter_write(struct gcov_iterator *iter, struct seq_file *seq)
183{
184 size_t len;
185
186 if (iter->pos >= iter->size)
187 return -EINVAL;
188
189 len = ITER_STRIDE;
190 if (iter->pos + len > iter->size)
191 len = iter->size - iter->pos;
192
193 seq_write(seq, iter->buffer + iter->pos, len);
194
195 return 0;
196}
197
198
199
200
201
202
203
204static void *gcov_seq_start(struct seq_file *seq, loff_t *pos)
205{
206 loff_t i;
207
208 gcov_iter_start(seq->private);
209 for (i = 0; i < *pos; i++) {
210 if (gcov_iter_next(seq->private))
211 return NULL;
212 }
213 return seq->private;
214}
215
216
217static void *gcov_seq_next(struct seq_file *seq, void *data, loff_t *pos)
218{
219 struct gcov_iterator *iter = data;
220
221 (*pos)++;
222 if (gcov_iter_next(iter))
223 return NULL;
224
225 return iter;
226}
227
228
229static int gcov_seq_show(struct seq_file *seq, void *data)
230{
231 struct gcov_iterator *iter = data;
232
233 if (gcov_iter_write(iter, seq))
234 return -EINVAL;
235 return 0;
236}
237
238static void gcov_seq_stop(struct seq_file *seq, void *data)
239{
240
241}
242
243static const struct seq_operations gcov_seq_ops = {
244 .start = gcov_seq_start,
245 .next = gcov_seq_next,
246 .show = gcov_seq_show,
247 .stop = gcov_seq_stop,
248};
249
250
251
252
253
254
255static struct gcov_info *get_node_info(struct gcov_node *node)
256{
257 if (node->num_loaded > 0)
258 return node->loaded_info[0];
259
260 return node->unloaded_info;
261}
262
263
264
265
266
267static struct gcov_info *get_accumulated_info(struct gcov_node *node)
268{
269 struct gcov_info *info;
270 int i = 0;
271
272 if (node->unloaded_info)
273 info = gcov_info_dup(node->unloaded_info);
274 else
275 info = gcov_info_dup(node->loaded_info[i++]);
276 if (!info)
277 return NULL;
278 for (; i < node->num_loaded; i++)
279 gcov_info_add(info, node->loaded_info[i]);
280
281 return info;
282}
283
284
285
286
287
288static int gcov_seq_open(struct inode *inode, struct file *file)
289{
290 struct gcov_node *node = inode->i_private;
291 struct gcov_iterator *iter;
292 struct seq_file *seq;
293 struct gcov_info *info;
294 int rc = -ENOMEM;
295
296 mutex_lock(&node_lock);
297
298
299
300
301
302 info = get_accumulated_info(node);
303 if (!info)
304 goto out_unlock;
305 iter = gcov_iter_new(info);
306 if (!iter)
307 goto err_free_info;
308 rc = seq_open(file, &gcov_seq_ops);
309 if (rc)
310 goto err_free_iter_info;
311 seq = file->private_data;
312 seq->private = iter;
313out_unlock:
314 mutex_unlock(&node_lock);
315 return rc;
316
317err_free_iter_info:
318 gcov_iter_free(iter);
319err_free_info:
320 gcov_info_free(info);
321 goto out_unlock;
322}
323
324
325
326
327
328static int gcov_seq_release(struct inode *inode, struct file *file)
329{
330 struct gcov_iterator *iter;
331 struct gcov_info *info;
332 struct seq_file *seq;
333
334 seq = file->private_data;
335 iter = seq->private;
336 info = gcov_iter_get_info(iter);
337 gcov_iter_free(iter);
338 gcov_info_free(info);
339 seq_release(inode, file);
340
341 return 0;
342}
343
344
345
346
347
348static struct gcov_node *get_node_by_name(const char *name)
349{
350 struct gcov_node *node;
351 struct gcov_info *info;
352
353 list_for_each_entry(node, &all_head, all) {
354 info = get_node_info(node);
355 if (info && (strcmp(gcov_info_filename(info), name) == 0))
356 return node;
357 }
358
359 return NULL;
360}
361
362
363
364
365static void reset_node(struct gcov_node *node)
366{
367 int i;
368
369 if (node->unloaded_info)
370 gcov_info_reset(node->unloaded_info);
371 for (i = 0; i < node->num_loaded; i++)
372 gcov_info_reset(node->loaded_info[i]);
373}
374
375static void remove_node(struct gcov_node *node);
376
377
378
379
380
381
382static ssize_t gcov_seq_write(struct file *file, const char __user *addr,
383 size_t len, loff_t *pos)
384{
385 struct seq_file *seq;
386 struct gcov_info *info;
387 struct gcov_node *node;
388
389 seq = file->private_data;
390 info = gcov_iter_get_info(seq->private);
391 mutex_lock(&node_lock);
392 node = get_node_by_name(gcov_info_filename(info));
393 if (node) {
394
395 if (node->num_loaded == 0)
396 remove_node(node);
397 else
398 reset_node(node);
399 }
400
401 gcov_info_reset(info);
402 mutex_unlock(&node_lock);
403
404 return len;
405}
406
407
408
409
410
411
412
413static char *link_target(const char *dir, const char *path, const char *ext)
414{
415 char *target;
416 char *old_ext;
417 char *copy;
418
419 copy = kstrdup(path, GFP_KERNEL);
420 if (!copy)
421 return NULL;
422 old_ext = strrchr(copy, '.');
423 if (old_ext)
424 *old_ext = '\0';
425 if (dir)
426 target = kasprintf(GFP_KERNEL, "%s/%s.%s", dir, copy, ext);
427 else
428 target = kasprintf(GFP_KERNEL, "%s.%s", copy, ext);
429 kfree(copy);
430
431 return target;
432}
433
434
435
436
437
438
439
440static char *get_link_target(const char *filename, const struct gcov_link *ext)
441{
442 const char *rel;
443 char *result;
444
445 if (strncmp(filename, objtree, strlen(objtree)) == 0) {
446 rel = filename + strlen(objtree) + 1;
447 if (ext->dir == SRC_TREE)
448 result = link_target(srctree, rel, ext->ext);
449 else
450 result = link_target(objtree, rel, ext->ext);
451 } else {
452
453 result = link_target(NULL, filename, ext->ext);
454 }
455
456 return result;
457}
458
459#define SKEW_PREFIX ".tmp_"
460
461
462
463
464
465static const char *deskew(const char *basename)
466{
467 if (strncmp(basename, SKEW_PREFIX, sizeof(SKEW_PREFIX) - 1) == 0)
468 return basename + sizeof(SKEW_PREFIX) - 1;
469 return basename;
470}
471
472
473
474
475
476static void add_links(struct gcov_node *node, struct dentry *parent)
477{
478 const char *basename;
479 char *target;
480 int num;
481 int i;
482
483 for (num = 0; gcov_link[num].ext; num++)
484 ;
485 node->links = kcalloc(num, sizeof(struct dentry *), GFP_KERNEL);
486 if (!node->links)
487 return;
488 for (i = 0; i < num; i++) {
489 target = get_link_target(
490 gcov_info_filename(get_node_info(node)),
491 &gcov_link[i]);
492 if (!target)
493 goto out_err;
494 basename = kbasename(target);
495 if (basename == target)
496 goto out_err;
497 node->links[i] = debugfs_create_symlink(deskew(basename),
498 parent, target);
499 kfree(target);
500 }
501
502 return;
503out_err:
504 kfree(target);
505 while (i-- > 0)
506 debugfs_remove(node->links[i]);
507 kfree(node->links);
508 node->links = NULL;
509}
510
511static const struct file_operations gcov_data_fops = {
512 .open = gcov_seq_open,
513 .release = gcov_seq_release,
514 .read = seq_read,
515 .llseek = seq_lseek,
516 .write = gcov_seq_write,
517};
518
519
520static void init_node(struct gcov_node *node, struct gcov_info *info,
521 const char *name, struct gcov_node *parent)
522{
523 INIT_LIST_HEAD(&node->list);
524 INIT_LIST_HEAD(&node->children);
525 INIT_LIST_HEAD(&node->all);
526 if (node->loaded_info) {
527 node->loaded_info[0] = info;
528 node->num_loaded = 1;
529 }
530 node->parent = parent;
531 if (name)
532 strcpy(node->name, name);
533}
534
535
536
537
538
539static struct gcov_node *new_node(struct gcov_node *parent,
540 struct gcov_info *info, const char *name)
541{
542 struct gcov_node *node;
543
544 node = kzalloc(sizeof(struct gcov_node) + strlen(name) + 1, GFP_KERNEL);
545 if (!node)
546 goto err_nomem;
547 if (info) {
548 node->loaded_info = kcalloc(1, sizeof(struct gcov_info *),
549 GFP_KERNEL);
550 if (!node->loaded_info)
551 goto err_nomem;
552 }
553 init_node(node, info, name, parent);
554
555 if (info) {
556 node->dentry = debugfs_create_file(deskew(node->name), 0600,
557 parent->dentry, node, &gcov_data_fops);
558 } else
559 node->dentry = debugfs_create_dir(node->name, parent->dentry);
560 if (info)
561 add_links(node, parent->dentry);
562 list_add(&node->list, &parent->children);
563 list_add(&node->all, &all_head);
564
565 return node;
566
567err_nomem:
568 kfree(node);
569 pr_warn("out of memory\n");
570 return NULL;
571}
572
573
574static void remove_links(struct gcov_node *node)
575{
576 int i;
577
578 if (!node->links)
579 return;
580 for (i = 0; gcov_link[i].ext; i++)
581 debugfs_remove(node->links[i]);
582 kfree(node->links);
583 node->links = NULL;
584}
585
586
587
588
589
590static void release_node(struct gcov_node *node)
591{
592 list_del(&node->list);
593 list_del(&node->all);
594 debugfs_remove(node->dentry);
595 remove_links(node);
596 kfree(node->loaded_info);
597 if (node->unloaded_info)
598 gcov_info_free(node->unloaded_info);
599 kfree(node);
600}
601
602
603static void remove_node(struct gcov_node *node)
604{
605 struct gcov_node *parent;
606
607 while ((node != &root_node) && list_empty(&node->children)) {
608 parent = node->parent;
609 release_node(node);
610 node = parent;
611 }
612}
613
614
615
616
617
618static struct gcov_node *get_child_by_name(struct gcov_node *parent,
619 const char *name)
620{
621 struct gcov_node *node;
622
623 list_for_each_entry(node, &parent->children, list) {
624 if (strcmp(node->name, name) == 0)
625 return node;
626 }
627
628 return NULL;
629}
630
631
632
633
634
635static ssize_t reset_write(struct file *file, const char __user *addr,
636 size_t len, loff_t *pos)
637{
638 struct gcov_node *node;
639
640 mutex_lock(&node_lock);
641restart:
642 list_for_each_entry(node, &all_head, all) {
643 if (node->num_loaded > 0)
644 reset_node(node);
645 else if (list_empty(&node->children)) {
646 remove_node(node);
647
648 goto restart;
649 }
650 }
651 mutex_unlock(&node_lock);
652
653 return len;
654}
655
656
657static ssize_t reset_read(struct file *file, char __user *addr, size_t len,
658 loff_t *pos)
659{
660
661 return 0;
662}
663
664static const struct file_operations gcov_reset_fops = {
665 .write = reset_write,
666 .read = reset_read,
667 .llseek = noop_llseek,
668};
669
670
671
672
673
674static void add_node(struct gcov_info *info)
675{
676 char *filename;
677 char *curr;
678 char *next;
679 struct gcov_node *parent;
680 struct gcov_node *node;
681
682 filename = kstrdup(gcov_info_filename(info), GFP_KERNEL);
683 if (!filename)
684 return;
685 parent = &root_node;
686
687 for (curr = filename; (next = strchr(curr, '/')); curr = next + 1) {
688 if (curr == next)
689 continue;
690 *next = 0;
691 if (strcmp(curr, ".") == 0)
692 continue;
693 if (strcmp(curr, "..") == 0) {
694 if (!parent->parent)
695 goto err_remove;
696 parent = parent->parent;
697 continue;
698 }
699 node = get_child_by_name(parent, curr);
700 if (!node) {
701 node = new_node(parent, NULL, curr);
702 if (!node)
703 goto err_remove;
704 }
705 parent = node;
706 }
707
708 node = new_node(parent, info, curr);
709 if (!node)
710 goto err_remove;
711out:
712 kfree(filename);
713 return;
714
715err_remove:
716 remove_node(parent);
717 goto out;
718}
719
720
721
722
723
724static void add_info(struct gcov_node *node, struct gcov_info *info)
725{
726 struct gcov_info **loaded_info;
727 int num = node->num_loaded;
728
729
730
731
732
733
734 loaded_info = kcalloc(num + 1, sizeof(struct gcov_info *), GFP_KERNEL);
735 if (!loaded_info) {
736 pr_warn("could not add '%s' (out of memory)\n",
737 gcov_info_filename(info));
738 return;
739 }
740 memcpy(loaded_info, node->loaded_info,
741 num * sizeof(struct gcov_info *));
742 loaded_info[num] = info;
743
744 if (num == 0) {
745
746
747
748
749 if (!gcov_info_is_compatible(node->unloaded_info, info)) {
750 pr_warn("discarding saved data for %s "
751 "(incompatible version)\n",
752 gcov_info_filename(info));
753 gcov_info_free(node->unloaded_info);
754 node->unloaded_info = NULL;
755 }
756 } else {
757
758
759
760
761 if (!gcov_info_is_compatible(node->loaded_info[0], info)) {
762 pr_warn("could not add '%s' (incompatible "
763 "version)\n", gcov_info_filename(info));
764 kfree(loaded_info);
765 return;
766 }
767 }
768
769 kfree(node->loaded_info);
770 node->loaded_info = loaded_info;
771 node->num_loaded = num + 1;
772}
773
774
775
776
777static int get_info_index(struct gcov_node *node, struct gcov_info *info)
778{
779 int i;
780
781 for (i = 0; i < node->num_loaded; i++) {
782 if (node->loaded_info[i] == info)
783 return i;
784 }
785 return -ENOENT;
786}
787
788
789
790
791static void save_info(struct gcov_node *node, struct gcov_info *info)
792{
793 if (node->unloaded_info)
794 gcov_info_add(node->unloaded_info, info);
795 else {
796 node->unloaded_info = gcov_info_dup(info);
797 if (!node->unloaded_info) {
798 pr_warn("could not save data for '%s' "
799 "(out of memory)\n",
800 gcov_info_filename(info));
801 }
802 }
803}
804
805
806
807
808
809static void remove_info(struct gcov_node *node, struct gcov_info *info)
810{
811 int i;
812
813 i = get_info_index(node, info);
814 if (i < 0) {
815 pr_warn("could not remove '%s' (not found)\n",
816 gcov_info_filename(info));
817 return;
818 }
819 if (gcov_persist)
820 save_info(node, info);
821
822 node->loaded_info[i] = node->loaded_info[node->num_loaded - 1];
823 node->num_loaded--;
824 if (node->num_loaded > 0)
825 return;
826
827 kfree(node->loaded_info);
828 node->loaded_info = NULL;
829 node->num_loaded = 0;
830 if (!node->unloaded_info)
831 remove_node(node);
832}
833
834
835
836
837
838void gcov_event(enum gcov_action action, struct gcov_info *info)
839{
840 struct gcov_node *node;
841
842 mutex_lock(&node_lock);
843 node = get_node_by_name(gcov_info_filename(info));
844 switch (action) {
845 case GCOV_ADD:
846 if (node)
847 add_info(node, info);
848 else
849 add_node(info);
850 break;
851 case GCOV_REMOVE:
852 if (node)
853 remove_info(node, info);
854 else {
855 pr_warn("could not remove '%s' (not found)\n",
856 gcov_info_filename(info));
857 }
858 break;
859 }
860 mutex_unlock(&node_lock);
861}
862
863
864static __init int gcov_fs_init(void)
865{
866 init_node(&root_node, NULL, NULL, NULL);
867
868
869
870
871 root_node.dentry = debugfs_create_dir("gcov", NULL);
872
873
874
875
876 debugfs_create_file("reset", 0600, root_node.dentry, NULL,
877 &gcov_reset_fops);
878
879 gcov_enable_events();
880 return 0;
881}
882device_initcall(gcov_fs_init);
883