datastores: Audit ast_channel_datastore_remove usage.
[asterisk/asterisk.git] / funcs / func_frame_trace.c
1 /*
2  * Asterisk -- An open source telephony toolkit.
3  *
4  * Copyright (C) 2010, Digium, Inc.
5  *
6  * David Vossel <dvossel@digium.com>
7  *
8  * See http://www.asterisk.org for more information about
9  * the Asterisk project. Please do not directly contact
10  * any of the maintainers of this project for assistance;
11  * the project provides a web site, mailing lists and IRC
12  * channels for your use.
13  *
14  * This program is free software, distributed under the terms of
15  * the GNU General Public License Version 2. See the LICENSE file
16  * at the top of the source tree.
17  */
18
19 /*! \file
20  *
21  * \brief Trace internal ast_frames on a channel.
22  *
23  * \author David Vossel <dvossel@digium.com>
24  *
25  * \ingroup functions
26  */
27
28 /*** MODULEINFO
29         <support_level>extended</support_level>
30  ***/
31
32 #include "asterisk.h"
33
34 ASTERISK_FILE_VERSION(__FILE__, "$Revision$")
35
36 #include "asterisk/module.h"
37 #include "asterisk/channel.h"
38 #include "asterisk/pbx.h"
39 #include "asterisk/framehook.h"
40
41 /*** DOCUMENTATION
42         <function name="FRAME_TRACE" language="en_US">
43                 <synopsis>
44                         View internal ast_frames as they are read and written on a channel.
45                 </synopsis>
46                 <syntax>
47                         <parameter name="filter list type" required="true">
48                                 <para>A filter can be applied to the trace to limit what frames are viewed.  This
49                                 filter can either be a <literal>white</literal> or <literal>black</literal> list
50                                 of frame types.  When no filter type is present, <literal>white</literal> is
51                                 used.  If no arguments are provided at all, all frames will be output.
52                                 </para>
53
54                                 <para>Below are the different types of frames that can be filtered.</para>
55                                 <enumlist>
56                                         <enum name = "DTMF_BEGIN" />
57                                         <enum name = "DTMF_END" />
58                                         <enum name = "VOICE" />
59                                         <enum name = "VIDEO" />
60                                         <enum name = "CONTROL" />
61                                         <enum name = "NULL" />
62                                         <enum name = "IAX" />
63                                         <enum name = "TEXT" />
64                                         <enum name = "IMAGE" />
65                                         <enum name = "HTML" />
66                                         <enum name = "CNG" />
67                                         <enum name = "MODEM" />
68                                 </enumlist>
69                         </parameter>
70                 </syntax>
71                 <description>
72                         <para>Examples:</para>
73                         <para>exten => 1,1,Set(FRAME_TRACE(white)=DTMF_BEGIN,DTMF_END); view only DTMF frames. </para>
74                         <para>exten => 1,1,Set(FRAME_TRACE()=DTMF_BEGIN,DTMF_END); view only DTMF frames. </para>
75                         <para>exten => 1,1,Set(FRAME_TRACE(black)=DTMF_BEGIN,DTMF_END); view everything except DTMF frames. </para>
76                 </description>
77         </function>
78  ***/
79
80 static void print_frame(struct ast_frame *frame);
81 static struct {
82         enum ast_frame_type type;
83         const char *str;
84 } frametype2str[] = {
85         { AST_FRAME_DTMF_BEGIN,   "DTMF_BEGIN" },
86         { AST_FRAME_DTMF_END,   "DTMF_END" },
87         { AST_FRAME_VOICE,   "VOICE" },
88         { AST_FRAME_VIDEO,   "VIDEO" },
89         { AST_FRAME_CONTROL,   "CONTROL" },
90         { AST_FRAME_NULL,   "NULL" },
91         { AST_FRAME_IAX,   "IAX" },
92         { AST_FRAME_TEXT,   "TEXT" },
93         { AST_FRAME_IMAGE,   "IMAGE" },
94         { AST_FRAME_HTML,   "HTML" },
95         { AST_FRAME_CNG,   "CNG" },
96         { AST_FRAME_MODEM,   "MODEM" },
97 };
98
99 struct frame_trace_data {
100         int list_type; /* 0 = white, 1 = black */
101         int values[ARRAY_LEN(frametype2str)];
102 };
103
104 static void datastore_destroy_cb(void *data) {
105         ast_free(data);
106 }
107
108 static const struct ast_datastore_info frame_trace_datastore = {
109         .type = "frametrace",
110         .destroy = datastore_destroy_cb
111 };
112
113 static void hook_destroy_cb(void *framedata)
114 {
115         ast_free(framedata);
116 }
117
118 static struct ast_frame *hook_event_cb(struct ast_channel *chan, struct ast_frame *frame, enum ast_framehook_event event, void *data)
119 {
120         int i;
121         int show_frame = 0;
122         struct frame_trace_data *framedata = data;
123         if (!frame) {
124                 return frame;
125         }
126
127         if ((event != AST_FRAMEHOOK_EVENT_WRITE) && (event != AST_FRAMEHOOK_EVENT_READ)) {
128                 return frame;
129         }
130
131         for (i = 0; i < ARRAY_LEN(frametype2str); i++) {
132                 if (frame->frametype == frametype2str[i].type) {
133                         if ((framedata->list_type == 0) && (framedata->values[i])) { /* white list */
134                                 show_frame = 1;
135                         } else if ((framedata->list_type == 1) && (!framedata->values[i])){ /* black list */
136                                 show_frame = 1;
137                         }
138                         break;
139                 }
140         }
141
142         if (show_frame) {
143                 ast_verbose("%s on Channel %s\n", event == AST_FRAMEHOOK_EVENT_READ ? "<--Read" : "--> Write", ast_channel_name(chan));
144                 print_frame(frame);
145         }
146         return frame;
147 }
148
149 static int frame_trace_helper(struct ast_channel *chan, const char *cmd, char *data, const char *value)
150 {
151         struct frame_trace_data *framedata;
152         struct ast_datastore *datastore = NULL;
153         struct ast_framehook_interface interface = {
154                 .version = AST_FRAMEHOOK_INTERFACE_VERSION,
155                 .event_cb = hook_event_cb,
156                 .destroy_cb = hook_destroy_cb,
157         };
158         int i = 0;
159
160         if (!chan) {
161                 ast_log(LOG_WARNING, "No channel was provided to %s function.\n", cmd);
162                 return -1;
163         }
164
165         if (!(framedata = ast_calloc(1, sizeof(*framedata)))) {
166                 return 0;
167         }
168
169         interface.data = framedata;
170
171         if (!strcasecmp(data, "black")) {
172                 framedata->list_type = 1;
173         }
174         for (i = 0; i < ARRAY_LEN(frametype2str); i++) {
175                 if (strcasestr(value, frametype2str[i].str)) {
176                         framedata->values[i] = 1;
177                 }
178         }
179
180         ast_channel_lock(chan);
181         i = ast_framehook_attach(chan, &interface);
182         if (i >= 0) {
183                 int *id;
184                 if ((datastore = ast_channel_datastore_find(chan, &frame_trace_datastore, NULL))) {
185                         id = datastore->data;
186                         ast_framehook_detach(chan, *id);
187                         ast_channel_datastore_remove(chan, datastore);
188                         ast_datastore_free(datastore);
189                 }
190
191                 if (!(datastore = ast_datastore_alloc(&frame_trace_datastore, NULL))) {
192                         ast_framehook_detach(chan, i);
193                         ast_channel_unlock(chan);
194                         return 0;
195                 }
196
197                 if (!(id = ast_calloc(1, sizeof(int)))) {
198                         ast_datastore_free(datastore);
199                         ast_framehook_detach(chan, i);
200                         ast_channel_unlock(chan);
201                         return 0;
202                 }
203
204                 *id = i; /* Store off the id. The channel is still locked so it is safe to access this ptr. */
205                 datastore->data = id;
206                 ast_channel_datastore_add(chan, datastore);
207         }
208         ast_channel_unlock(chan);
209
210         return 0;
211 }
212
213 static void print_frame(struct ast_frame *frame)
214 {
215         switch (frame->frametype) {
216         case AST_FRAME_DTMF_END:
217                 ast_verbose("FrameType: DTMF END\n");
218                 ast_verbose("Digit: 0x%02X '%c'\n", (unsigned)frame->subclass.integer,
219                         frame->subclass.integer < ' ' ? ' ' : frame->subclass.integer);
220                 break;
221         case AST_FRAME_VOICE:
222                 ast_verbose("FrameType: VOICE\n");
223                 ast_verbose("Codec: %s\n", ast_format_get_name(frame->subclass.format));
224                 ast_verbose("MS: %ld\n", frame->len);
225                 ast_verbose("Samples: %d\n", frame->samples);
226                 ast_verbose("Bytes: %d\n", frame->datalen);
227                 break;
228         case AST_FRAME_VIDEO:
229                 ast_verbose("FrameType: VIDEO\n");
230                 ast_verbose("Codec: %s\n", ast_format_get_name(frame->subclass.format));
231                 ast_verbose("MS: %ld\n", frame->len);
232                 ast_verbose("Samples: %d\n", frame->samples);
233                 ast_verbose("Bytes: %d\n", frame->datalen);
234                 break;
235         case AST_FRAME_CONTROL:
236                 ast_verbose("FrameType: CONTROL\n");
237                 switch ((enum ast_control_frame_type) frame->subclass.integer) {
238                 case AST_CONTROL_HANGUP:
239                         ast_verbose("SubClass: HANGUP\n");
240                         break;
241                 case AST_CONTROL_RING:
242                         ast_verbose("SubClass: RING\n");
243                         break;
244                 case AST_CONTROL_RINGING:
245                         ast_verbose("SubClass: RINGING\n");
246                         break;
247                 case AST_CONTROL_ANSWER:
248                         ast_verbose("SubClass: ANSWER\n");
249                         break;
250                 case AST_CONTROL_BUSY:
251                         ast_verbose("SubClass: BUSY\n");
252                         break;
253                 case AST_CONTROL_TAKEOFFHOOK:
254                         ast_verbose("SubClass: TAKEOFFHOOK\n");
255                         break;
256                 case AST_CONTROL_OFFHOOK:
257                         ast_verbose("SubClass: OFFHOOK\n");
258                         break;
259                 case AST_CONTROL_CONGESTION:
260                         ast_verbose("SubClass: CONGESTION\n");
261                         break;
262                 case AST_CONTROL_FLASH:
263                         ast_verbose("SubClass: FLASH\n");
264                         break;
265                 case AST_CONTROL_WINK:
266                         ast_verbose("SubClass: WINK\n");
267                         break;
268                 case AST_CONTROL_OPTION:
269                         ast_verbose("SubClass: OPTION\n");
270                         break;
271                 case AST_CONTROL_RADIO_KEY:
272                         ast_verbose("SubClass: RADIO KEY\n");
273                         break;
274                 case AST_CONTROL_RADIO_UNKEY:
275                         ast_verbose("SubClass: RADIO UNKEY\n");
276                         break;
277                 case AST_CONTROL_PROGRESS:
278                         ast_verbose("SubClass: PROGRESS\n");
279                         break;
280                 case AST_CONTROL_PROCEEDING:
281                         ast_verbose("SubClass: PROCEEDING\n");
282                         break;
283                 case AST_CONTROL_HOLD:
284                         ast_verbose("SubClass: HOLD\n");
285                         break;
286                 case AST_CONTROL_UNHOLD:
287                         ast_verbose("SubClass: UNHOLD\n");
288                         break;
289                 case AST_CONTROL_VIDUPDATE:
290                         ast_verbose("SubClass: VIDUPDATE\n");
291                         break;
292                 case _XXX_AST_CONTROL_T38:
293                         ast_verbose("SubClass: XXX T38\n");
294                         break;
295                 case AST_CONTROL_SRCUPDATE:
296                         ast_verbose("SubClass: SRCUPDATE\n");
297                         break;
298                 case AST_CONTROL_TRANSFER:
299                         ast_verbose("SubClass: TRANSFER\n");
300                         break;
301                 case AST_CONTROL_CONNECTED_LINE:
302                         ast_verbose("SubClass: CONNECTED LINE\n");
303                         break;
304                 case AST_CONTROL_REDIRECTING:
305                         ast_verbose("SubClass: REDIRECTING\n");
306                         break;
307                 case AST_CONTROL_T38_PARAMETERS:
308                         ast_verbose("SubClass: T38 PARAMETERS\n");
309                         break;
310                 case AST_CONTROL_CC:
311                         ast_verbose("SubClass: CC\n");
312                         break;
313                 case AST_CONTROL_SRCCHANGE:
314                         ast_verbose("SubClass: SRCCHANGE\n");
315                         break;
316                 case AST_CONTROL_READ_ACTION:
317                         ast_verbose("SubClass: READ ACTION\n");
318                         break;
319                 case AST_CONTROL_AOC:
320                         ast_verbose("SubClass: AOC\n");
321                         break;
322                 case AST_CONTROL_MCID:
323                         ast_verbose("SubClass: MCID\n");
324                         break;
325                 case AST_CONTROL_INCOMPLETE:
326                         ast_verbose("SubClass: INCOMPLETE\n");
327                         break;
328                 case AST_CONTROL_END_OF_Q:
329                         ast_verbose("SubClass: END_OF_Q\n");
330                         break;
331                 case AST_CONTROL_UPDATE_RTP_PEER:
332                         ast_verbose("SubClass: UPDATE_RTP_PEER\n");
333                         break;
334                 case AST_CONTROL_PVT_CAUSE_CODE:
335                         ast_verbose("SubClass: PVT_CAUSE_CODE\n");
336                         break;
337                 case AST_CONTROL_STREAM_STOP:
338                         ast_verbose("SubClass: STREAM_STOP\n");
339                         break;
340                 case AST_CONTROL_STREAM_SUSPEND:
341                         ast_verbose("SubClass: STREAM_SUSPEND\n");
342                         break;
343                 case AST_CONTROL_STREAM_RESTART:
344                         ast_verbose("SubClass: STREAM_RESTART\n");
345                         break;
346                 case AST_CONTROL_STREAM_REVERSE:
347                         ast_verbose("SubClass: STREAM_REVERSE\n");
348                         break;
349                 case AST_CONTROL_STREAM_FORWARD:
350                         ast_verbose("SubClass: STREAM_FORWARD\n");
351                         break;
352                 case AST_CONTROL_RECORD_CANCEL:
353                         ast_verbose("SubClass: RECORD_CANCEL\n");
354                         break;
355                 case AST_CONTROL_RECORD_STOP:
356                         ast_verbose("SubClass: RECORD_STOP\n");
357                         break;
358                 case AST_CONTROL_RECORD_SUSPEND:
359                         ast_verbose("SubClass: RECORD_SUSPEND\n");
360                         break;
361                 case AST_CONTROL_RECORD_MUTE:
362                         ast_verbose("SubClass: RECORD_MUTE\n");
363                         break;
364                 }
365
366                 if (frame->subclass.integer == -1) {
367                         ast_verbose("SubClass: %d\n", frame->subclass.integer);
368                 }
369                 ast_verbose("Bytes: %d\n", frame->datalen);
370                 break;
371         case AST_FRAME_NULL:
372                 ast_verbose("FrameType: NULL\n");
373                 break;
374         case AST_FRAME_IAX:
375                 ast_verbose("FrameType: IAX\n");
376                 break;
377         case AST_FRAME_TEXT:
378                 ast_verbose("FrameType: TXT\n");
379                 break;
380         case AST_FRAME_IMAGE:
381                 ast_verbose("FrameType: IMAGE\n");
382                 break;
383         case AST_FRAME_HTML:
384                 ast_verbose("FrameType: HTML\n");
385                 break;
386         case AST_FRAME_CNG:
387                 ast_verbose("FrameType: CNG\n");
388                 break;
389         case AST_FRAME_MODEM:
390                 ast_verbose("FrameType: MODEM\n");
391                 break;
392         case AST_FRAME_DTMF_BEGIN:
393                 ast_verbose("FrameType: DTMF BEGIN\n");
394                 ast_verbose("Digit: 0x%02X '%c'\n", (unsigned)frame->subclass.integer,
395                         frame->subclass.integer < ' ' ? ' ' : frame->subclass.integer);
396                 break;
397         case AST_FRAME_BRIDGE_ACTION:
398                 ast_verbose("FrameType: Bridge\n");
399                 ast_verbose("SubClass: %d\n", frame->subclass.integer);
400                 break;
401         case AST_FRAME_BRIDGE_ACTION_SYNC:
402                 ast_verbose("Frametype: Synchronous Bridge\n");
403                 ast_verbose("Subclass: %d\n", frame->subclass.integer);
404                 break;
405         }
406
407         ast_verbose("Src: %s\n", ast_strlen_zero(frame->src) ? "NOT PRESENT" : frame->src);
408         ast_verbose("\n");
409 }
410
411 static struct ast_custom_function frame_trace_function = {
412         .name = "FRAME_TRACE",
413         .write = frame_trace_helper,
414 };
415
416 static int unload_module(void)
417 {
418         return ast_custom_function_unregister(&frame_trace_function);
419 }
420
421 static int load_module(void)
422 {
423         int res = ast_custom_function_register(&frame_trace_function);
424         return res ? AST_MODULE_LOAD_DECLINE : AST_MODULE_LOAD_SUCCESS;
425 }
426
427 AST_MODULE_INFO_STANDARD_EXTENDED(ASTERISK_GPL_KEY, "Frame Trace for internal ast_frame debugging.");
428
429