08a41f27cfed5d334919eb0e168529e8b18d7659
[sip-router] / src / modules / lrkproxy / lrkproxy_funcs.c
1 /*
2  * Copyright (C) 2001-2003 FhG Fokus
3  * Copyright (C) 2014-2015 Sipwise GmbH, http://www.sipwise.com
4  * Copyright (C) 2020 Mojtaba Esfandiari.S, Nasim-Telecom
5  *
6  * This file is part of Kamailio, a free SIP server.
7  *
8  * Kamailio is free software; you can redistribute it and/or modify
9  * it under the terms of the GNU General Public License as published by
10  * the Free Software Foundation; either version 2 of the License, or
11  * (at your option) any later version
12  *
13  * Kamailio is distributed in the hope that it will be useful,
14  * but WITHOUT ANY WARRANTY; without even the implied warranty of
15  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
16  * GNU General Public License for more details.
17  *
18  * You should have received a copy of the GNU General Public License
19  * along with this program; if not, write to the Free Software
20  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA
21  */
22
23 #include <stdio.h>
24 #include <stdlib.h>
25 #include <sys/types.h>
26 #include <unistd.h>
27 #include "lrkproxy_funcs.h"
28 #include "../../core/dprint.h"
29 #include "../../core/config.h"
30 #include "../../core/ut.h"
31 #include "../../core/forward.h"
32 #include "../../core/resolve.h"
33 #include "../../core/globals.h"
34 #include "../../core/udp_server.h"
35 #include "../../core/pt.h"
36 #include "../../core/parser/msg_parser.h"
37 #include "../../core/trim.h"
38 #include "../../core/parser/parse_from.h"
39 #include "../../core/parser/contact/parse_contact.h"
40 #include "../../core/parser/parse_uri.h"
41 #include "../../core/parser/parse_content.h"
42 #include "../../core/parser/parser_f.h"
43 #include "../../core/parser/sdp/sdp_helpr_funcs.h"
44
45 #define READ(val) \
46         (*(val + 0) + (*(val + 1) << 8) + (*(val + 2) << 16) + (*(val + 3) << 24))
47 #define advance(_ptr,_n,_str,_error) \
48         do{\
49                 if ((_ptr)+(_n)>(_str).s+(_str).len)\
50                         goto _error;\
51                 (_ptr) = (_ptr) + (_n);\
52         }while(0);
53 #define one_of_16( _x , _t ) \
54         (_x==_t[0]||_x==_t[15]||_x==_t[8]||_x==_t[2]||_x==_t[3]||_x==_t[4]\
55         ||_x==_t[5]||_x==_t[6]||_x==_t[7]||_x==_t[1]||_x==_t[9]||_x==_t[10]\
56         ||_x==_t[11]||_x==_t[12]||_x==_t[13]||_x==_t[14])
57 #define one_of_8( _x , _t ) \
58         (_x==_t[0]||_x==_t[7]||_x==_t[1]||_x==_t[2]||_x==_t[3]||_x==_t[4]\
59         ||_x==_t[5]||_x==_t[6])
60
61
62
63 /**
64  * return:
65  * -1: error
66  *  1: text or sdp
67  *  2: multipart
68  *  3: trickle ice sdp fragment
69  */
70 int check_content_type(struct sip_msg *msg)
71 {
72     static unsigned int appl[16] = {
73             0x6c707061/*appl*/,0x6c707041/*Appl*/,0x6c705061/*aPpl*/,
74             0x6c705041/*APpl*/,0x6c507061/*apPl*/,0x6c507041/*ApPl*/,
75             0x6c505061/*aPPl*/,0x6c505041/*APPl*/,0x4c707061/*appL*/,
76             0x4c707041/*AppL*/,0x4c705061/*aPpL*/,0x4c705041/*APpL*/,
77             0x4c507061/*apPL*/,0x4c507041/*ApPL*/,0x4c505061/*aPPL*/,
78             0x4c505041/*APPL*/};
79     static unsigned int icat[16] = {
80             0x74616369/*icat*/,0x74616349/*Icat*/,0x74614369/*iCat*/,
81             0x74614349/*ICat*/,0x74416369/*icAt*/,0x74416349/*IcAt*/,
82             0x74414369/*iCAt*/,0x74414349/*ICAt*/,0x54616369/*icaT*/,
83             0x54616349/*IcaT*/,0x54614369/*iCaT*/,0x54614349/*ICaT*/,
84             0x54416369/*icAT*/,0x54416349/*IcAT*/,0x54414369/*iCAT*/,
85             0x54414349/*ICAT*/};
86     static unsigned int ion_[8] = {
87             0x006e6f69/*ion_*/,0x006e6f49/*Ion_*/,0x006e4f69/*iOn_*/,
88             0x006e4f49/*IOn_*/,0x004e6f69/*ioN_*/,0x004e6f49/*IoN_*/,
89             0x004e4f69/*iON_*/,0x004e4f49/*ION_*/};
90     static unsigned int sdp_[8] = {
91             0x00706473/*sdp_*/,0x00706453/*Sdp_*/,0x00704473/*sDp_*/,
92             0x00704453/*SDp_*/,0x00506473/*sdP_*/,0x00506453/*SdP_*/,
93             0x00504473/*sDP_*/,0x00504453/*SDP_*/};
94     str           str_type;
95     unsigned int  x;
96     char          *p;
97
98     if (!msg->content_type)
99     {
100                 LM_WARN("the header Content-TYPE is absent!"
101                                 "let's assume the content is text/plain ;-)\n");
102         return 1;
103     }
104
105     trim_len(str_type.len,str_type.s,msg->content_type->body);
106     if (str_type.len>=15 && (*str_type.s=='m' || *str_type.s=='M')
107         && strncasecmp(str_type.s, "multipart/mixed", 15) == 0) {
108         return 2;
109     }
110     p = str_type.s;
111     advance(p,4,str_type,error_1);
112     x = READ(p-4);
113     if (!one_of_16(x,appl))
114         goto other;
115     advance(p,4,str_type,error_1);
116     x = READ(p-4);
117     if (!one_of_16(x,icat))
118         goto other;
119     advance(p,3,str_type,error_1);
120     x = READ(p-3) & 0x00ffffff;
121     if (!one_of_8(x,ion_))
122         goto other;
123
124     /* skip spaces and tabs if any */
125     while (*p==' ' || *p=='\t')
126         advance(p,1,str_type,error_1);
127     if (*p!='/')
128     {
129                 LM_ERR("no / found after primary type\n");
130         goto error;
131     }
132     advance(p,1,str_type,error_1);
133     while ((*p==' ' || *p=='\t') && p+1<str_type.s+str_type.len)
134         advance(p,1,str_type,error_1);
135
136     advance(p,3,str_type,error_1);
137     x = READ(p-3) & 0x00ffffff;
138     if (!one_of_8(x,sdp_)) {
139         if (strncasecmp(p-3, "trickle-ice-sdpfrag", 19) == 0)
140             return 3;
141         goto other;
142     }
143
144     if (*p==';'||*p==' '||*p=='\t'||*p=='\n'||*p=='\r'||*p==0) {
145                 LM_DBG("type <%.*s> found valid\n", (int)(p-str_type.s), str_type.s);
146         return 1;
147     } else {
148                 LM_ERR("bad end for type!\n");
149         return -1;
150     }
151
152     error_1:
153             LM_ERR("body ended :-(!\n");
154     error:
155     return -1;
156     other:
157             LM_ERR("invalid type for a message\n");
158     return -1;
159 }
160
161
162 /*
163  * Get message body and check Content-Type header field
164  */
165 int extract_body(struct sip_msg *msg, str *body )
166 {
167     char c;
168     int ret;
169     str mpdel;
170     char *rest, *p1, *p2;
171     struct hdr_field hf;
172     unsigned int mime;
173
174     body->s = get_body(msg);
175     if (body->s==0) {
176                 LM_ERR("failed to get the message body\n");
177         goto error;
178     }
179
180     /*
181      * Better use the content-len value - no need of any explicit
182      * parcing as get_body() parsed all headers and Conten-Length
183      * body header is automaticaly parsed when found.
184      */
185     if (msg->content_length==0) {
186                 LM_ERR("failed to get the content length in message\n");
187         goto error;
188     }
189
190     body->len = get_content_length(msg);
191     if (body->len==0) {
192                 LM_ERR("message body has length zero\n");
193         goto error;
194     }
195
196     if (body->len + body->s > msg->buf + msg->len) {
197                 LM_ERR("content-length exceeds packet-length by %d\n",
198                        (int)((body->len + body->s) - (msg->buf + msg->len)));
199         goto error;
200     }
201
202     /* no need for parse_headers(msg, EOH), get_body will
203      * parse everything */
204     /*is the content type correct?*/
205     if((ret = check_content_type(msg))==-1)
206     {
207                 LM_ERR("content type mismatching\n");
208         goto error;
209     }
210
211     if(ret!=2)
212         goto done;
213
214     /* multipart body */
215     if(get_mixed_part_delimiter(&msg->content_type->body,&mpdel) < 0) {
216         goto error;
217     }
218     p1 = find_sdp_line_delimiter(body->s, body->s+body->len, mpdel);
219     if (p1 == NULL) {
220                 LM_ERR("empty multipart content\n");
221         return -1;
222     }
223     p2=p1;
224     c = 0;
225     for(;;)
226     {
227         p1 = p2;
228         if (p1 == NULL || p1 >= body->s+body->len)
229             break; /* No parts left */
230         p2 = find_next_sdp_line_delimiter(p1, body->s+body->len,
231                                           mpdel, body->s+body->len);
232         /* p2 is text limit for application parsing */
233         rest = eat_line(p1 + mpdel.len + 2, p2 - p1 - mpdel.len - 2);
234         if ( rest > p2 ) {
235                     LM_ERR("Unparsable <%.*s>\n", (int)(p1-p1), p1);
236             return -1;
237         }
238         while( rest<p2 ) {
239             memset(&hf,0, sizeof(struct hdr_field));
240             rest = get_sdp_hdr_field(rest, p2, &hf);
241             if(hf.type==HDR_EOH_T)
242                 break;
243             if(hf.type==HDR_ERROR_T)
244                 return -1;
245             if(hf.type==HDR_CONTENTTYPE_T) {
246                 if(decode_mime_type(hf.body.s, hf.body.s + hf.body.len,
247                                     &mime)==NULL)
248                     return -1;
249                 if (((((unsigned int)mime)>>16) == TYPE_APPLICATION)
250                     && ((mime&0x00ff) == SUBTYPE_SDP)) {
251                     c = 1;
252                 }
253             }
254         } /* end of while */
255         if(c==1)
256         {
257             if (rest < p2 && *rest == '\r') rest++;
258             if (rest < p2 && *rest == '\n') rest++;
259             if (rest < p2 && p2[-1] == '\n') p2--;
260             if (rest < p2 && p2[-1] == '\r') p2--;
261             body->s = rest;
262             body->len = p2-rest;
263             goto done;
264         }
265     }
266
267     error:
268     return -1;
269
270     done:
271     /*LM_DBG("DEBUG:extract_body:=|%.*s|\n",body->len,body->s);*/
272     return ret; /* mirrors return type of check_content_type */
273 }
274
275 /*
276  * Some helper functions taken verbatim from tm module.
277  */
278
279 /*
280  * Extract Call-ID value
281  * assumes the callid header is already parsed
282  * (so make sure it is, before calling this function or
283  *  it might fail even if the message _has_ a callid)
284  */
285 int
286 get_callid(struct sip_msg* _m, str* _cid)
287 {
288
289     if ((parse_headers(_m, HDR_CALLID_F, 0) == -1)) {
290                 LM_ERR("failed to parse call-id header\n");
291         return -1;
292     }
293
294     if (_m->callid == NULL) {
295                 LM_ERR("call-id not found\n");
296         return -1;
297     }
298
299     _cid->s = _m->callid->body.s;
300     _cid->len = _m->callid->body.len;
301     trim(_cid);
302     return 0;
303 }
304
305 /*
306  * Extract tag from To header field of a response
307  */
308 int
309 get_to_tag(struct sip_msg* _m, str* _tag)
310 {
311
312     if (parse_to_header(_m) < 0) {
313                 LM_ERR("To header field missing\n");
314         return -1;
315     }
316
317     if (get_to(_m)->tag_value.len) {
318         _tag->s = get_to(_m)->tag_value.s;
319         _tag->len = get_to(_m)->tag_value.len;
320     } else {
321         _tag->s = NULL; /* fixes gcc 4.0 warnings */
322         _tag->len = 0;
323     }
324
325     return 0;
326 }
327
328 /*
329  * Extract tag from From header field of a request
330  */
331 int
332 get_from_tag(struct sip_msg* _m, str* _tag)
333 {
334
335     if (parse_from_header(_m)<0) {
336                 LM_ERR("failed to parse From header\n");
337         return -1;
338     }
339
340     if (get_from(_m)->tag_value.len) {
341         _tag->s = get_from(_m)->tag_value.s;
342         _tag->len = get_from(_m)->tag_value.len;
343     } else {
344         _tag->s = NULL; /* fixes gcc 4.0 warnings */
345         _tag->len = 0;
346     }
347
348     return 0;
349 }
350
351 /*
352  * Extract URI from the Contact header field
353  */
354 int
355 get_contact_uri(struct sip_msg* _m, struct sip_uri *uri, contact_t** _c)
356 {
357
358     if ((parse_headers(_m, HDR_CONTACT_F, 0) == -1) || !_m->contact)
359         return -1;
360     if (!_m->contact->parsed && parse_contact(_m->contact) < 0) {
361                 LM_ERR("failed to parse Contact body\n");
362         return -1;
363     }
364     *_c = ((contact_body_t*)_m->contact->parsed)->contacts;
365     if (*_c == NULL)
366         /* no contacts found */
367         return -1;
368
369     if (parse_uri((*_c)->uri.s, (*_c)->uri.len, uri) < 0 || uri->host.len <= 0) {
370                 LM_ERR("failed to parse Contact URI [%.*s]\n",
371                        (*_c)->uri.len, ((*_c)->uri.s)?(*_c)->uri.s:"");
372         return -1;
373     }
374     return 0;
375 }
376
377 /*
378  * Extract branch from Via header
379  */
380 int
381 get_via_branch(struct sip_msg* msg, int vianum, str* _branch)
382 {
383     struct via_body *via;
384     struct via_param *p;
385
386     if (parse_via_header(msg, vianum, &via) < 0)
387         return -1;
388
389     for (p = via->param_lst; p; p = p->next)
390     {
391         if (p->name.len == strlen("branch")
392             && strncasecmp(p->name.s, "branch", strlen("branch")) == 0) {
393             _branch->s = p->value.s;
394             _branch->len = p->value.len;
395             return 0;
396         }
397     }
398     return -1;
399 }
400
401 int get_sdp_ipaddr_media(struct sip_msg *msg, str *ip_addr) {
402     sdp_session_cell_t *sdp_session;
403     sdp_stream_cell_t *sdp_stream;
404     sdp_info_t *sdp = (sdp_info_t *) msg->body;
405     if (!sdp) {
406                 LM_INFO("sdp null\n");
407         return -1;
408     }
409
410
411     int sdp_session_num = 0;
412     sdp_session = get_sdp_session(msg, sdp_session_num);
413
414     if (!sdp_session) {
415                 LM_INFO("can not get the sdp session\n");
416         return 0;
417     }
418
419     if (sdp_session->ip_addr.s && sdp_session->ip_addr.len > 0) {
420                 LM_INFO("sdp_session->ip_addr:%.*s\n", sdp_session->ip_addr.len, sdp_session->ip_addr.s);
421         ip_addr->s = sdp_session->ip_addr.s;
422         ip_addr->len = sdp_session->ip_addr.len;
423         trim(ip_addr);
424     }
425     else {
426         int sdp_stream_num = 0;
427         sdp_stream = get_sdp_stream(msg, sdp_session_num, sdp_stream_num);
428         if (!sdp_stream) {
429                     LM_INFO("can not get the sdp stream\n");
430             return 0;
431         }
432         if (sdp_stream->ip_addr.s && sdp_stream->ip_addr.len > 0) {
433             LM_INFO("sdp_stream->ip_addr:%.*s\n", sdp_stream->ip_addr.len, sdp_stream->ip_addr.s);
434             ip_addr->s = sdp_stream->ip_addr.s;
435             ip_addr->len = sdp_stream->ip_addr.len;
436             trim(ip_addr);
437         }
438     }
439
440     
441     return 0;
442 }
443
444
445 int get_sdp_port_media(struct sip_msg *msg, str *port){
446 //    sdp_session_cell_t *sdp_session;
447     sdp_stream_cell_t *sdp_stream;
448     int sdp_session_num = 0;
449
450     sdp_info_t *sdp = (sdp_info_t *)msg->body;
451     if(!sdp) {
452                 LM_INFO("sdp null\n");
453         return -1;
454     }
455
456 //    sdp_session = get_sdp_session(msg, sdp_session_num);
457 //    if(!sdp_session) {
458 //                LM_INFO("can not get the sdp session\n");
459 //        return 0;
460 //    } else {
461 //                LM_INFO("NEW_IP_ADDRESS:>%.*s>\n", sdp_session->ip_addr.len, sdp_session->ip_addr.s);
462 //        lrk_sdp_info->ip_addr.s = sdp_session->ip_addr;
463         int sdp_stream_num = 0;
464         sdp_stream = get_sdp_stream(msg, sdp_session_num, sdp_stream_num);
465         if (!sdp_stream) {
466                     LM_INFO("can not get the sdp stream\n");
467             return -1;
468         } else {
469 //                    LM_INFO ("PORT:<%.*s>\n", sdp_stream->port.len, sdp_stream->port.s);
470 //            str2int(&sdp_stream->port, lrk_sdp_info->port)
471             port->s = sdp_stream->port.s;
472             port->len = sdp_stream->port.len;
473             trim(port);
474         }
475 //    }
476     return 0;
477
478 }
479