Commit | Line | Data |
---|---|---|
0c0686ee NC |
1 | /* |
2 | * Copyright (C) 2010 Nils Carlson | |
3 | * | |
4 | * This library is free software; you can redistribute it and/or | |
5 | * modify it under the terms of the GNU Lesser General Public | |
6 | * License as published by the Free Software Foundation; either | |
7 | * version 2.1 of the License, or (at your option) any later version. | |
8 | * | |
9 | * This library is distributed in the hope that it will be useful, | |
10 | * but WITHOUT ANY WARRANTY; without even the implied warranty of | |
11 | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU | |
12 | * Lesser General Public License for more details. | |
13 | * | |
14 | * You should have received a copy of the GNU Lesser General Public | |
15 | * License along with this library; if not, write to the Free Software | |
16 | * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA | |
17 | * | |
18 | */ | |
19 | ||
20 | #include <errno.h> | |
21 | #include <ust/tracepoint.h> | |
22 | #include <ust/core.h> | |
23 | #include <ust/kcompat/kcompat.h> | |
24 | #include "usterr.h" | |
25 | ||
26 | #define _LGPL_SOURCE | |
27 | #include <urcu-bp.h> | |
28 | ||
29 | /* libraries that contain trace_events (struct trace_event_lib) */ | |
0222e121 | 30 | static CDS_LIST_HEAD(libs); |
0c0686ee NC |
31 | |
32 | static DEFINE_MUTEX(trace_events_mutex); | |
33 | ||
34 | void lock_trace_events(void) | |
35 | { | |
36 | pthread_mutex_lock(&trace_events_mutex); | |
37 | } | |
38 | ||
39 | void unlock_trace_events(void) | |
40 | { | |
41 | pthread_mutex_unlock(&trace_events_mutex); | |
42 | } | |
43 | ||
44 | ||
45 | int lib_get_iter_trace_events(struct trace_event_iter *iter) | |
46 | { | |
47 | struct trace_event_lib *iter_lib; | |
48 | int found = 0; | |
49 | ||
0222e121 | 50 | cds_list_for_each_entry(iter_lib, &libs, list) { |
0c0686ee NC |
51 | if (iter_lib < iter->lib) |
52 | continue; | |
53 | else if (iter_lib > iter->lib) | |
54 | iter->trace_event = NULL; | |
55 | found = trace_event_get_iter_range(&iter->trace_event, | |
56 | iter_lib->trace_events_start, | |
57 | iter_lib->trace_events_start + iter_lib->trace_events_count); | |
58 | if (found) { | |
59 | iter->lib = iter_lib; | |
60 | break; | |
61 | } | |
62 | } | |
63 | return found; | |
64 | } | |
65 | ||
66 | /** | |
67 | * trace_event_get_iter_range - Get a next trace_event iterator given a range. | |
68 | * @trace_event: current trace_events (in), next trace_event (out) | |
69 | * @begin: beginning of the range | |
70 | * @end: end of the range | |
71 | * | |
72 | * Returns whether a next trace_event has been found (1) or not (0). | |
73 | * Will return the first trace_event in the range if the input trace_event is NULL. | |
74 | */ | |
75 | int trace_event_get_iter_range(struct trace_event **trace_event, struct trace_event *begin, | |
76 | struct trace_event *end) | |
77 | { | |
78 | if (!*trace_event && begin != end) { | |
79 | *trace_event = begin; | |
80 | return 1; | |
81 | } | |
82 | if (*trace_event >= begin && *trace_event < end) | |
83 | return 1; | |
84 | return 0; | |
85 | } | |
86 | ||
87 | static void trace_event_get_iter(struct trace_event_iter *iter) | |
88 | { | |
89 | int found = 0; | |
90 | ||
91 | found = lib_get_iter_trace_events(iter); | |
e2b46575 | 92 | |
0c0686ee NC |
93 | if (!found) |
94 | trace_event_iter_reset(iter); | |
95 | } | |
96 | ||
97 | void trace_event_iter_start(struct trace_event_iter *iter) | |
98 | { | |
99 | trace_event_get_iter(iter); | |
100 | } | |
101 | ||
102 | void trace_event_iter_next(struct trace_event_iter *iter) | |
103 | { | |
104 | iter->trace_event++; | |
105 | /* | |
106 | * iter->trace_event may be invalid because we blindly incremented it. | |
107 | * Make sure it is valid by marshalling on the trace_events, getting the | |
108 | * trace_events from following modules if necessary. | |
109 | */ | |
110 | trace_event_get_iter(iter); | |
111 | } | |
112 | ||
113 | void trace_event_iter_reset(struct trace_event_iter *iter) | |
114 | { | |
115 | iter->lib = NULL; | |
116 | iter->trace_event = NULL; | |
117 | } | |
118 | ||
119 | int trace_event_register_lib(struct trace_event *trace_events_start, | |
120 | int trace_events_count) | |
121 | { | |
122 | struct trace_event_lib *pl; | |
123 | ||
124 | pl = (struct trace_event_lib *) malloc(sizeof(struct trace_event_lib)); | |
125 | ||
126 | pl->trace_events_start = trace_events_start; | |
127 | pl->trace_events_count = trace_events_count; | |
128 | ||
129 | /* FIXME: maybe protect this with its own mutex? */ | |
130 | pthread_mutex_lock(&trace_events_mutex); | |
0222e121 | 131 | cds_list_add(&pl->list, &libs); |
0c0686ee NC |
132 | pthread_mutex_unlock(&trace_events_mutex); |
133 | ||
134 | DBG("just registered a trace_events section from %p and having %d trace_events", trace_events_start, trace_events_count); | |
135 | ||
136 | return 0; | |
137 | } | |
138 | ||
139 | int trace_event_unregister_lib(struct trace_event *trace_events_start) | |
140 | { | |
141 | struct trace_event_lib *lib; | |
142 | ||
143 | pthread_mutex_lock(&trace_events_mutex); | |
144 | ||
0222e121 | 145 | cds_list_for_each_entry(lib, &libs, list) { |
0c0686ee NC |
146 | if(lib->trace_events_start == trace_events_start) { |
147 | struct trace_event_lib *lib2free = lib; | |
0222e121 | 148 | cds_list_del(&lib->list); |
0c0686ee NC |
149 | free(lib2free); |
150 | break; | |
151 | } | |
152 | } | |
153 | ||
154 | pthread_mutex_unlock(&trace_events_mutex); | |
155 | ||
156 | return 0; | |
157 | } |