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 * Initialize the GQuarks needed to register the event hooks for
33 LTT_CHANNEL_NET
= g_quark_from_static_string("net");
35 LTT_EVENT_DEV_XMIT_EXTENDED
= g_quark_from_static_string("dev_xmit_extended");
36 LTT_EVENT_DEV_RECEIVE
= g_quark_from_static_string("dev_receive");
37 LTT_EVENT_TCPV4_RCV_EXTENDED
= g_quark_from_static_string("tcpv4_rcv_extended");
38 LTT_EVENT_UDPV4_RCV_EXTENDED
= g_quark_from_static_string("udpv4_rcv_extended");
40 LTT_FIELD_SKB
= g_quark_from_static_string("skb");
41 LTT_FIELD_PROTOCOL
= g_quark_from_static_string("protocol");
42 LTT_FIELD_NETWORK_PROTOCOL
=
43 g_quark_from_static_string("network_protocol");
44 LTT_FIELD_TRANSPORT_PROTOCOL
=
45 g_quark_from_static_string("transport_protocol");
46 LTT_FIELD_SADDR
= g_quark_from_static_string("saddr");
47 LTT_FIELD_DADDR
= g_quark_from_static_string("daddr");
48 LTT_FIELD_TOT_LEN
= g_quark_from_static_string("tot_len");
49 LTT_FIELD_IHL
= g_quark_from_static_string("ihl");
50 LTT_FIELD_SOURCE
= g_quark_from_static_string("source");
51 LTT_FIELD_DEST
= g_quark_from_static_string("dest");
52 LTT_FIELD_SEQ
= g_quark_from_static_string("seq");
53 LTT_FIELD_ACK_SEQ
= g_quark_from_static_string("ack_seq");
54 LTT_FIELD_DOFF
= g_quark_from_static_string("doff");
55 LTT_FIELD_ACK
= g_quark_from_static_string("ack");
56 LTT_FIELD_RST
= g_quark_from_static_string("rst");
57 LTT_FIELD_SYN
= g_quark_from_static_string("syn");
58 LTT_FIELD_FIN
= g_quark_from_static_string("fin");
59 LTT_FIELD_UNICAST
= g_quark_from_static_string("unicast");
60 LTT_FIELD_ULEN
= g_quark_from_static_string("ulen");
61 LTT_FIELD_DATA_START
= g_quark_from_static_string("data_start");
65 /* Fill hookListList and add event hooks
67 * Note: possibilité de remettre le code avec lttv_trace_find_marker_ids (voir
71 * hookListList: LttvTraceHook hookListList[traceNum][hookNum]
72 * traceSetContext: LTTV traceset
73 * hookFunction: call back function when event is encountered
74 * hookData: data that will be made accessible to hookFunction in
76 * eventTypes: types of events for which to register hooks
78 void registerHooks(GArray
* hookListList
, LttvTracesetContext
* const
79 traceSetContext
, LttvHook hookFunction
, gpointer hookData
, const bool
83 unsigned int traceNb
= lttv_traceset_number(traceSetContext
->ts
);
88 bool eventTypes
[TYPE_COUNT
];
89 } eventHookInfoList
[] = {
91 .channelName
= LTT_CHANNEL_NET
,
92 .eventName
= LTT_EVENT_DEV_XMIT_EXTENDED
,
93 .fields
= FIELD_ARRAY(LTT_FIELD_SKB
, LTT_FIELD_NETWORK_PROTOCOL
,
94 LTT_FIELD_TRANSPORT_PROTOCOL
, LTT_FIELD_SADDR
,
95 LTT_FIELD_DADDR
, LTT_FIELD_TOT_LEN
, LTT_FIELD_IHL
,
96 LTT_FIELD_SOURCE
, LTT_FIELD_DEST
, LTT_FIELD_SEQ
,
97 LTT_FIELD_ACK_SEQ
, LTT_FIELD_DOFF
, LTT_FIELD_ACK
,
98 LTT_FIELD_RST
, LTT_FIELD_SYN
, LTT_FIELD_FIN
),
99 .eventTypes
[TCP
]= true,
100 .eventTypes
[UDP
]= true,
102 .channelName
= LTT_CHANNEL_NET
,
103 .eventName
= LTT_EVENT_DEV_RECEIVE
,
104 .fields
= FIELD_ARRAY(LTT_FIELD_SKB
, LTT_FIELD_PROTOCOL
),
105 .eventTypes
[TCP
]= true,
106 .eventTypes
[UDP
]= true,
108 .channelName
= LTT_CHANNEL_NET
,
109 .eventName
= LTT_EVENT_TCPV4_RCV_EXTENDED
,
110 .fields
= FIELD_ARRAY(LTT_FIELD_SKB
, LTT_FIELD_SADDR
,
111 LTT_FIELD_DADDR
, LTT_FIELD_TOT_LEN
, LTT_FIELD_IHL
,
112 LTT_FIELD_SOURCE
, LTT_FIELD_DEST
, LTT_FIELD_SEQ
,
113 LTT_FIELD_ACK_SEQ
, LTT_FIELD_DOFF
, LTT_FIELD_ACK
,
114 LTT_FIELD_RST
, LTT_FIELD_SYN
, LTT_FIELD_FIN
),
115 .eventTypes
[TCP
]= true,
116 .eventTypes
[UDP
]= false,
118 .channelName
= LTT_CHANNEL_NET
,
119 .eventName
= LTT_EVENT_UDPV4_RCV_EXTENDED
,
120 .fields
= FIELD_ARRAY(LTT_FIELD_SKB
, LTT_FIELD_SADDR
,
121 LTT_FIELD_DADDR
, LTT_FIELD_UNICAST
, LTT_FIELD_ULEN
,
122 LTT_FIELD_DATA_START
),
123 .eventTypes
[TCP
]= false,
124 .eventTypes
[UDP
]= true,
126 }; // This is called a compound literal
127 unsigned int hookNb
= sizeof(eventHookInfoList
) / sizeof(*eventHookInfoList
);
129 for(i
= 0; i
< traceNb
; i
++)
131 LttvTraceContext
* tc
;
135 tc
= traceSetContext
->traces
[i
];
136 hookList
= g_array_new(FALSE
, FALSE
, sizeof(LttvTraceHook
));
137 g_array_append_val(hookListList
, hookList
);
140 for (j
= 0; j
< hookNb
; j
++)
146 for (k
= 0; k
< TYPE_COUNT
; k
++)
148 if (eventTypes
[k
] && eventHookInfoList
[j
].eventTypes
[k
])
159 old_len
= hookList
->len
;
160 retval
= lttv_trace_find_hook(tc
->t
,
161 eventHookInfoList
[j
].channelName
,
162 eventHookInfoList
[j
].eventName
, eventHookInfoList
[j
].fields
,
163 hookFunction
, hookData
, &hookList
);
166 g_warning("Trace %d contains no %s.%s marker\n", i
,
167 g_quark_to_string(eventHookInfoList
[j
].channelName
),
168 g_quark_to_string(eventHookInfoList
[j
].eventName
));
172 g_assert(hookList
->len
- old_len
== 1);
176 // Add the hooks to each tracefile's event_by_id hook list
177 for(j
= 0; j
< tc
->tracefiles
->len
; j
++)
179 LttvTracefileContext
* tfc
;
181 tfc
= g_array_index(tc
->tracefiles
, LttvTracefileContext
*, j
);
183 for(k
= 0; k
< hookList
->len
; k
++)
185 LttvTraceHook
* traceHook
;
187 traceHook
= &g_array_index(hookList
, LttvTraceHook
, k
);
188 if (traceHook
->hook_data
!= hookData
)
190 g_assert_not_reached();
192 if (traceHook
->mdata
== tfc
->tf
->mdata
)
194 lttv_hooks_add(lttv_hooks_by_id_find(tfc
->event_by_id
,
195 traceHook
->id
), traceHook
->h
, traceHook
,
204 /* Remove event hooks and free hookListList
207 * hookListList: LttvTraceHook hookListList[traceNum][hookNum]
208 * traceSetContext: LTTV traceset
210 void unregisterHooks(GArray
* hookListList
, LttvTracesetContext
* const
213 unsigned int i
, j
, k
;
214 unsigned int traceNb
= lttv_traceset_number(traceSetContext
->ts
);
216 for(i
= 0; i
< traceNb
; i
++)
218 LttvTraceContext
* tc
;
221 tc
= traceSetContext
->traces
[i
];
222 hookList
= g_array_index(hookListList
, GArray
*, i
);
224 // Remove the hooks from each tracefile's event_by_id hook list
225 for(j
= 0; j
< tc
->tracefiles
->len
; j
++)
227 LttvTracefileContext
* tfc
;
229 tfc
= g_array_index(tc
->tracefiles
, LttvTracefileContext
*, j
);
231 for(k
= 0; k
< hookList
->len
; k
++)
233 LttvTraceHook
* traceHook
;
235 traceHook
= &g_array_index(hookList
, LttvTraceHook
, k
);
236 if (traceHook
->mdata
== tfc
->tf
->mdata
)
238 lttv_hooks_remove_data(lttv_hooks_by_id_find(tfc
->event_by_id
,
239 traceHook
->id
), traceHook
->h
, traceHook
);
244 g_array_free(hookList
, TRUE
);
246 g_array_free(hookListList
, TRUE
);