Prevent a crash in res_pjsip_dtmf_info.c
[asterisk/asterisk.git] / res / res_pjsip_dtmf_info.c
1 /*
2  * Asterisk -- An open source telephony toolkit.
3  *
4  * Copyright (C) 2013, Digium, Inc.
5  *
6  * Jason Parker <jparker@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 /*** MODULEINFO
20         <depend>pjproject</depend>
21         <depend>res_pjsip</depend>
22         <depend>res_pjsip_session</depend>
23         <support_level>core</support_level>
24  ***/
25
26 #include "asterisk.h"
27
28 #include <pjsip.h>
29 #include <pjsip_ua.h>
30
31 #include "asterisk/res_pjsip.h"
32 #include "asterisk/res_pjsip_session.h"
33 #include "asterisk/module.h"
34
35 static int is_media_type(pjsip_rx_data *rdata, char *subtype)
36 {
37         return rdata->msg_info.ctype
38                 && !pj_strcmp2(&rdata->msg_info.ctype->media.type, "application")
39                 && !pj_strcmp2(&rdata->msg_info.ctype->media.subtype, subtype);
40 }
41
42 static void send_response(struct ast_sip_session *session,
43                           struct pjsip_rx_data *rdata, int code)
44 {
45         pjsip_tx_data *tdata;
46         pjsip_dialog *dlg = session->inv_session->dlg;
47
48         if (pjsip_dlg_create_response(dlg, rdata, code,
49                                       NULL, &tdata) == PJ_SUCCESS) {
50                 struct pjsip_transaction *tsx = pjsip_rdata_get_tsx(rdata);
51                 pjsip_dlg_send_response(dlg, tsx, tdata);
52         }
53 }
54
55 static char get_event(const char *c)
56 {
57         unsigned int event;
58
59         if (*c == '!' || *c == '*' || *c == '#' ||
60             ('A' <= *c && *c <= 'D') ||
61             ('a' <= *c && *c <= 'd')) {
62                 return *c;
63         }
64
65         if ((sscanf(c, "%30u", &event) != 1) || event > 16) {
66                 return '\0';
67         }
68
69         if (event < 10) {
70                 return *c;
71         }
72
73         switch (event) {
74         case 10: return '*';
75         case 11: return '#';
76         case 16: return '!';
77         }
78
79         return 'A' + (event - 12);
80 }
81
82 static int dtmf_info_incoming_request(struct ast_sip_session *session, struct pjsip_rx_data *rdata)
83 {
84         pjsip_msg_body *body = rdata->msg_info.msg->body;
85         char buf[body ? body->len : 0];
86         char *cur = buf;
87         char *line;
88
89         char event = '\0';
90         unsigned int duration = 100;
91
92         char is_dtmf = is_media_type(rdata, "dtmf");
93
94         if (!is_dtmf && !is_media_type(rdata, "dtmf-relay")) {
95                 return 0;
96         }
97
98         if (!body || !body->len) {
99                 /* need to return 200 OK on empty body */
100                 send_response(session, rdata, 200);
101                 return 0;
102         }
103
104         body->print_body(body, buf, body->len);
105
106         if (is_dtmf) {
107                 /* directly use what is in the message body */
108                 event = get_event(cur);
109         } else { /* content type = application/dtmf-relay */
110                 while ((line = strsep(&cur, "\r\n"))) {
111                         char *c;
112
113                         if (!(c = strchr(line, '='))) {
114                                 continue;
115                         }
116
117                         *c++ = '\0';
118                         c = ast_skip_blanks(c);
119
120                         if (!strcasecmp(line, "signal")) {
121                                 if (!(event = get_event(c))) {
122                                         break;
123                                 }
124                         } else if (!strcasecmp(line, "duration")) {
125                                 sscanf(c, "%30u", &duration);
126                         }
127                 }
128         }
129
130         if (event == '!') {
131                 struct ast_frame f = { AST_FRAME_CONTROL, { AST_CONTROL_FLASH, } };
132                 ast_queue_frame(session->channel, &f);
133         } else if (event != '\0') {
134                 struct ast_frame f = { AST_FRAME_DTMF, };
135                 f.len = duration;
136                 f.subclass.integer = event;
137                 ast_queue_frame(session->channel, &f);
138         } else {
139                 ast_log(LOG_ERROR, "Invalid DTMF event signal in INFO message.\n");
140         }
141
142         send_response(session, rdata, event ? 200 : 500);
143         return event ? 0 : -1;
144 }
145
146 static struct ast_sip_session_supplement dtmf_info_supplement = {
147         .method = "INFO",
148         .incoming_request = dtmf_info_incoming_request,
149 };
150
151 static int load_module(void)
152 {
153         ast_sip_session_register_supplement(&dtmf_info_supplement);
154         return AST_MODULE_LOAD_SUCCESS;
155 }
156
157 static int unload_module(void)
158 {
159         ast_sip_session_unregister_supplement(&dtmf_info_supplement);
160         return 0;
161 }
162
163 AST_MODULE_INFO(ASTERISK_GPL_KEY, AST_MODFLAG_LOAD_ORDER, "PJSIP DTMF INFO Support",
164                 .load = load_module,
165                 .unload = unload_module,
166                 .load_pri = AST_MODPRI_APP_DEPEND,
167                );