1 /* This file is part of the Linux Trace Toolkit viewer
2 * Copyright (C) 2009 Benjamin Poirier <benjamin.poirier@polymtl.ca>
4 * This program is free software; you can redistribute it and/or modify
5 * it under the terms of the GNU General Public License Version 2 as
6 * published by the Free Software Foundation;
8 * This program is distributed in the hope that it will be useful,
9 * but WITHOUT ANY WARRANTY; without even the implied warranty of
10 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
11 * GNU General Public License for more details.
13 * You should have received a copy of the GNU General Public License
14 * along with this program; if not, write to the Free Software
15 * Foundation, Inc., 59 Temple Place - Suite 330, Boston,
23 #include "data_structures.h"
24 #include "event_processing_lttng_common.h"
28 #define g_info(format...) g_log (G_LOG_DOMAIN, G_LOG_LEVEL_INFO, format)
33 * Initialize the GQuarks needed to register the event hooks for
38 LTT_CHANNEL_NET
= g_quark_from_static_string("net");
40 LTT_EVENT_DEV_XMIT_EXTENDED
= g_quark_from_static_string("dev_xmit_extended");
41 LTT_EVENT_DEV_RECEIVE
= g_quark_from_static_string("dev_receive");
42 LTT_EVENT_TCPV4_RCV_EXTENDED
= g_quark_from_static_string("tcpv4_rcv_extended");
43 LTT_EVENT_UDPV4_RCV_EXTENDED
= g_quark_from_static_string("udpv4_rcv_extended");
45 LTT_FIELD_SKB
= g_quark_from_static_string("skb");
46 LTT_FIELD_PROTOCOL
= g_quark_from_static_string("protocol");
47 LTT_FIELD_NETWORK_PROTOCOL
=
48 g_quark_from_static_string("network_protocol");
49 LTT_FIELD_TRANSPORT_PROTOCOL
=
50 g_quark_from_static_string("transport_protocol");
51 LTT_FIELD_SADDR
= g_quark_from_static_string("saddr");
52 LTT_FIELD_DADDR
= g_quark_from_static_string("daddr");
53 LTT_FIELD_TOT_LEN
= g_quark_from_static_string("tot_len");
54 LTT_FIELD_IHL
= g_quark_from_static_string("ihl");
55 LTT_FIELD_SOURCE
= g_quark_from_static_string("source");
56 LTT_FIELD_DEST
= g_quark_from_static_string("dest");
57 LTT_FIELD_SEQ
= g_quark_from_static_string("seq");
58 LTT_FIELD_ACK_SEQ
= g_quark_from_static_string("ack_seq");
59 LTT_FIELD_DOFF
= g_quark_from_static_string("doff");
60 LTT_FIELD_ACK
= g_quark_from_static_string("ack");
61 LTT_FIELD_RST
= g_quark_from_static_string("rst");
62 LTT_FIELD_SYN
= g_quark_from_static_string("syn");
63 LTT_FIELD_FIN
= g_quark_from_static_string("fin");
64 LTT_FIELD_UNICAST
= g_quark_from_static_string("unicast");
65 LTT_FIELD_ULEN
= g_quark_from_static_string("ulen");
66 LTT_FIELD_DATA_START
= g_quark_from_static_string("data_start");
70 /* Fill hookListList and add event hooks
72 * Note: possibilité de remettre le code avec lttv_trace_find_marker_ids (voir
76 * hookListList: LttvTraceHook hookListList[traceNum][hookNum]
77 * traceSetContext: LTTV traceset
78 * hookFunction: call back function when event is encountered
79 * hookData: data that will be made accessible to hookFunction in
81 * eventTypes: types of events for which to register hooks
83 void registerHooks(GArray
* hookListList
, LttvTracesetContext
* const
84 traceSetContext
, LttvHook hookFunction
, gpointer hookData
, const bool
88 unsigned int traceNb
= lttv_traceset_number(traceSetContext
->ts
);
93 bool eventTypes
[TYPE_COUNT
];
94 } eventHookInfoList
[] = {
96 .channelName
= LTT_CHANNEL_NET
,
97 .eventName
= LTT_EVENT_DEV_XMIT_EXTENDED
,
98 .fields
= FIELD_ARRAY(LTT_FIELD_SKB
, LTT_FIELD_NETWORK_PROTOCOL
,
99 LTT_FIELD_TRANSPORT_PROTOCOL
, LTT_FIELD_SADDR
,
100 LTT_FIELD_DADDR
, LTT_FIELD_TOT_LEN
, LTT_FIELD_IHL
,
101 LTT_FIELD_SOURCE
, LTT_FIELD_DEST
, LTT_FIELD_SEQ
,
102 LTT_FIELD_ACK_SEQ
, LTT_FIELD_DOFF
, LTT_FIELD_ACK
,
103 LTT_FIELD_RST
, LTT_FIELD_SYN
, LTT_FIELD_FIN
),
104 .eventTypes
[TCP
]= true,
105 .eventTypes
[UDP
]= true,
107 .channelName
= LTT_CHANNEL_NET
,
108 .eventName
= LTT_EVENT_DEV_RECEIVE
,
109 .fields
= FIELD_ARRAY(LTT_FIELD_SKB
, LTT_FIELD_PROTOCOL
),
110 .eventTypes
[TCP
]= true,
111 .eventTypes
[UDP
]= true,
113 .channelName
= LTT_CHANNEL_NET
,
114 .eventName
= LTT_EVENT_TCPV4_RCV_EXTENDED
,
115 .fields
= FIELD_ARRAY(LTT_FIELD_SKB
, LTT_FIELD_SADDR
,
116 LTT_FIELD_DADDR
, LTT_FIELD_TOT_LEN
, LTT_FIELD_IHL
,
117 LTT_FIELD_SOURCE
, LTT_FIELD_DEST
, LTT_FIELD_SEQ
,
118 LTT_FIELD_ACK_SEQ
, LTT_FIELD_DOFF
, LTT_FIELD_ACK
,
119 LTT_FIELD_RST
, LTT_FIELD_SYN
, LTT_FIELD_FIN
),
120 .eventTypes
[TCP
]= true,
121 .eventTypes
[UDP
]= false,
123 .channelName
= LTT_CHANNEL_NET
,
124 .eventName
= LTT_EVENT_UDPV4_RCV_EXTENDED
,
125 .fields
= FIELD_ARRAY(LTT_FIELD_SKB
, LTT_FIELD_SADDR
,
126 LTT_FIELD_DADDR
, LTT_FIELD_UNICAST
, LTT_FIELD_ULEN
,
127 LTT_FIELD_DATA_START
),
128 .eventTypes
[TCP
]= false,
129 .eventTypes
[UDP
]= true,
131 }; // This is called a compound literal
132 unsigned int hookNb
= sizeof(eventHookInfoList
) / sizeof(*eventHookInfoList
);
134 for(i
= 0; i
< traceNb
; i
++)
136 LttvTraceContext
* tc
;
140 tc
= traceSetContext
->traces
[i
];
141 hookList
= g_array_new(FALSE
, FALSE
, sizeof(LttvTraceHook
));
142 g_array_append_val(hookListList
, hookList
);
145 for (j
= 0; j
< hookNb
; j
++)
151 for (k
= 0; k
< TYPE_COUNT
; k
++)
153 if (eventTypes
[k
] && eventHookInfoList
[j
].eventTypes
[k
] == false)
164 old_len
= hookList
->len
;
165 retval
= lttv_trace_find_hook(tc
->t
,
166 eventHookInfoList
[j
].channelName
,
167 eventHookInfoList
[j
].eventName
, eventHookInfoList
[j
].fields
,
168 hookFunction
, hookData
, &hookList
);
171 g_warning("Trace %d contains no %s.%s marker\n", i
,
172 g_quark_to_string(eventHookInfoList
[j
].channelName
),
173 g_quark_to_string(eventHookInfoList
[j
].eventName
));
177 g_assert(hookList
->len
- old_len
== 1);
181 // Add the hooks to each tracefile's event_by_id hook list
182 for(j
= 0; j
< tc
->tracefiles
->len
; j
++)
184 LttvTracefileContext
* tfc
;
186 tfc
= g_array_index(tc
->tracefiles
, LttvTracefileContext
*, j
);
188 for(k
= 0; k
< hookList
->len
; k
++)
190 LttvTraceHook
* traceHook
;
192 traceHook
= &g_array_index(hookList
, LttvTraceHook
, k
);
193 if (traceHook
->hook_data
!= hookData
)
195 g_assert_not_reached();
197 if (traceHook
->mdata
== tfc
->tf
->mdata
)
199 lttv_hooks_add(lttv_hooks_by_id_find(tfc
->event_by_id
,
200 traceHook
->id
), traceHook
->h
, traceHook
,
209 /* Remove event hooks and free hookListList
212 * hookListList: LttvTraceHook hookListList[traceNum][hookNum]
213 * traceSetContext: LTTV traceset
215 void unregisterHooks(GArray
* hookListList
, LttvTracesetContext
* const
218 unsigned int i
, j
, k
;
219 unsigned int traceNb
= lttv_traceset_number(traceSetContext
->ts
);
221 for(i
= 0; i
< traceNb
; i
++)
223 LttvTraceContext
* tc
;
226 tc
= traceSetContext
->traces
[i
];
227 hookList
= g_array_index(hookListList
, GArray
*, i
);
229 // Remove the hooks from each tracefile's event_by_id hook list
230 for(j
= 0; j
< tc
->tracefiles
->len
; j
++)
232 LttvTracefileContext
* tfc
;
234 tfc
= g_array_index(tc
->tracefiles
, LttvTracefileContext
*, j
);
236 for(k
= 0; k
< hookList
->len
; k
++)
238 LttvTraceHook
* traceHook
;
240 traceHook
= &g_array_index(hookList
, LttvTraceHook
, k
);
241 if (traceHook
->mdata
== tfc
->tf
->mdata
)
243 lttv_hooks_remove_data(lttv_hooks_by_id_find(tfc
->event_by_id
,
244 traceHook
->id
), traceHook
->h
, traceHook
);
249 g_array_free(hookList
, TRUE
);
251 g_array_free(hookListList
, TRUE
);