source: mediastreamer2/src/mtu.c @ 971:4f4d02a4d0bf

Last change on this file since 971:4f4d02a4d0bf was 971:4f4d02a4d0bf, checked in by Simon Morlat <simon.morlat@…>, 3 years ago

fix possible buffer overflow with mtu discovery.
File size: 6.9 KB
Line 
1/*
2mediastreamer2 library - modular sound and video processing and streaming
3Copyright (C) 2006  Simon MORLAT (simon.morlat@linphone.org)
4
5This program is free software; you can redistribute it and/or
6modify it under the terms of the GNU General Public License
7as published by the Free Software Foundation; either version 2
8of the License, or (at your option) any later version.
9
10This program is distributed in the hope that it will be useful,
11but WITHOUT ANY WARRANTY; without even the implied warranty of
12MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
13GNU General Public License for more details.
14
15You should have received a copy of the GNU General Public License
16along with this program; if not, write to the Free Software
17Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
18*/
19
20
21/* mtu.c : discover the mtu automatically */
22
23#include "mediastreamer2/mscommon.h"
24
25
26#if defined(WIN32) && !defined(_WIN32_WCE)
27
28HINSTANCE m_IcmpInst = NULL;
29
30typedef struct ip_option_information {
31    UCHAR   Ttl;
32    UCHAR   Tos;
33    UCHAR   Flags;
34    UCHAR   OptionsSize;
35    PUCHAR  OptionsData;
36} IP_OPTION_INFORMATION, * PIP_OPTION_INFORMATION;
37
38typedef BOOL (WINAPI *ICMPCLOSEHANDLE)(HANDLE IcmpHandle);
39typedef HANDLE (WINAPI *ICMPCREATEFILE)(VOID);
40typedef DWORD (WINAPI *ICMPSENDECHO)(HANDLE IcmpHandle,ULONG DestinationAddress, LPVOID RequestData, WORD RequestSize, PIP_OPTION_INFORMATION RequestOptions, LPVOID ReplyBuffer, DWORD ReplySize, DWORD Timeout);
41
42ICMPCLOSEHANDLE pIcmpCloseHandle = NULL;
43ICMPCREATEFILE pIcmpCreateFile = NULL;
44ICMPSENDECHO pIcmpSendEcho = NULL;
45
46#define IP_FLAG_DF      0x2         // Don't fragment this packet.
47#define IP_OPT_ROUTER_ALERT 0x94  // Router Alert Option
48
49#define IP_STATUS_BASE              11000
50#define IP_PACKET_TOO_BIG           (IP_STATUS_BASE + 9)
51#define IP_REQ_TIMED_OUT            (IP_STATUS_BASE + 10)
52
53static int mtus[] = {
54  1500,   // Ethernet, Point-to-Point (default)
55  1492,   // IEEE 802.3
56  1006,   // SLIP, ARPANET
57  576,    // X.25 Networks
58  544,    // DEC IP Portal
59  512,    // NETBIOS
60  508,    // IEEE 802/Source-Rt Bridge, ARCNET
61  296,    // Point-to-Point (low delay)
62  68,     // Official minimum
63  0
64};
65
66int ms_discover_mtu(const char *host)
67{
68  int i;
69
70        struct addrinfo hints,*ai=NULL;
71        char port[10];
72  char ipaddr[INET6_ADDRSTRLEN];
73  int err;
74
75  HANDLE hIcmp;
76  unsigned long target_addr;
77
78  struct ip_option_information ip_opts;
79  unsigned char reply_buffer[10000];
80
81  if (!m_IcmpInst)
82        {
83                m_IcmpInst = LoadLibrary("icmp.dll");
84                if (m_IcmpInst)
85                {
86                        pIcmpCloseHandle = (ICMPCLOSEHANDLE)GetProcAddress(m_IcmpInst, "IcmpCloseHandle");
87                        pIcmpCreateFile  = (ICMPCREATEFILE) GetProcAddress(m_IcmpInst, "IcmpCreateFile");
88                        pIcmpSendEcho =    (ICMPSENDECHO)   GetProcAddress(m_IcmpInst, "IcmpSendEcho");
89                }
90        }
91
92  hIcmp = pIcmpCreateFile();
93
94        memset(&hints,0,sizeof(hints));
95        hints.ai_family = PF_INET;
96        hints.ai_socktype = SOCK_DGRAM;
97       
98        snprintf(port,sizeof(port),"0");
99        err=getaddrinfo(host,port,&hints,&ai);
100        if (err!=0){
101    pIcmpCloseHandle( hIcmp );
102                ms_error("getaddrinfo(): error\n");
103                return -1;
104        }
105  getnameinfo (ai->ai_addr, ai->ai_addrlen, ipaddr, sizeof (ipaddr), port,
106               sizeof (port), NI_NUMERICHOST | NI_NUMERICSERV);
107        freeaddrinfo(ai);
108
109  target_addr=inet_addr(ipaddr);
110
111
112  /* Prepare the IP options */
113  memset(&ip_opts,0,sizeof(ip_opts));
114  ip_opts.Ttl=30;
115  ip_opts.Flags = IP_FLAG_DF | IP_OPT_ROUTER_ALERT;
116
117
118  // ignore icmpbuff data contents
119  for (i=0;mtus[i]!=0;i++)
120  {
121    char icmpbuff[2048];
122    char *icmp_data = icmpbuff;
123
124    int status = -1;
125    if (pIcmpSendEcho)
126      status=pIcmpSendEcho(hIcmp,
127                          target_addr,
128                          (LPVOID)icmp_data,
129                          mtus[i]-60, /* icmp_data_size */
130                          &ip_opts,
131                          reply_buffer,
132                          sizeof(reply_buffer),
133                          3000L); // 3 seconds
134    if (status || GetLastError() == IP_REQ_TIMED_OUT)
135    {
136      pIcmpCloseHandle( hIcmp );
137      return mtus[i];
138    }
139  }
140
141  pIcmpCloseHandle( hIcmp );
142
143  return -1;
144}
145
146#elif defined(__linux)
147
148#include <sys/types.h>
149#include <sys/socket.h>
150#include <string.h>
151#include <errno.h>
152#include <netinet/in.h>
153#include <netinet/ip.h>
154#include <netdb.h>
155
156#ifndef IP_MTU
157#define IP_MTU 14
158#endif
159
160int ms_discover_mtu(const char *host){
161        int sock;
162        int err,mtu=0,new_mtu;
163        socklen_t optlen;
164        char port[10];
165        struct addrinfo hints,*ai=NULL;
166        int rand_port;
167        int retry=0;
168        struct timeval tv;
169
170        memset(&hints,0,sizeof(hints));
171        hints.ai_family = PF_INET;
172        hints.ai_socktype = SOCK_DGRAM;
173       
174        gettimeofday(&tv,NULL); 
175        srandom(tv.tv_usec);
176        rand_port=random() & 0xFFFF;
177        if (rand_port<1000) rand_port+=1000;
178        snprintf(port,sizeof(port),"%i",rand_port);
179        err=getaddrinfo(host,port,&hints,&ai);
180        if (err!=0){
181                ms_error("getaddrinfo(): %s\n",gai_strerror(err));
182                return -1;
183        }
184        sock=socket(PF_INET,SOCK_DGRAM,0);
185        if(sock < 0)
186        {
187                ms_error("socket(): %s",strerror(errno));
188                return sock;
189        }
190        mtu=IP_PMTUDISC_DO;
191        optlen=sizeof(mtu);
192        err=setsockopt(sock,IPPROTO_IP,IP_MTU_DISCOVER,&mtu,optlen);
193        if (err!=0){
194                ms_error("setsockopt(): %s",strerror(errno));
195                err = close(sock);
196                if (err!=0)
197                        ms_error("close(): %s", strerror(errno));
198                return -1;
199        }
200        err=connect(sock,ai->ai_addr,ai->ai_addrlen);
201        freeaddrinfo(ai);
202        if (err!=0){
203                ms_error("connect(): %s",strerror(errno));
204                err = close(sock);
205                if (err !=0)
206                        ms_error("close(): %s", strerror(errno));
207                return -1;
208        }
209        mtu=1500-28;//was the size of the inital buf
210        do{
211                int send_returned;
212                char *buf=ms_malloc(mtu);//avoid mtu greater than the beginning
213                if(buf == NULL)
214                {
215                        ms_error("malloc(): %s",strerror(errno));
216                        err = close(sock);
217                        if (err!=0)
218                                ms_error("close(): %s", strerror(errno));
219                        return -1;
220                }
221                send_returned = send(sock,buf,mtu,0);
222                ms_free(buf);
223                if( send_returned < 0)//mtu actually change...
224                {
225                        ms_error("send(): %s",strerror(errno));
226                        err = close(sock);
227                        if (err!=0)
228                                ms_error("close(): %s", strerror(errno));
229                        return -1;
230                }
231                usleep(500000);/*wait for an icmp message come back */
232                err=getsockopt(sock,IPPROTO_IP,IP_MTU,&new_mtu,&optlen);
233                if (err!=0){
234                        ms_error("getsockopt(): %s",strerror(errno));
235                        err = close(sock);
236                        if (err!=0)
237                                ms_error("close(): %s", strerror(errno));
238                        return -1;
239                }else{
240                        ms_message("Partial MTU discovered : %i",new_mtu);
241                        if (new_mtu==mtu) break;
242                        else mtu=new_mtu;
243                }
244                retry++;
245        }while(retry<10);
246       
247        ms_message("mtu to %s is %i",host,mtu);
248
249        err = close(sock);
250        if (err!=0)
251                ms_error("close() %s", strerror(errno));
252        return mtu;
253}
254
255#else
256
257int ms_discover_mtu(const char*host){
258        ms_warning("mtu discovery not implemented.");
259        return -1;
260}
261
262#endif
263
264
265void ms_set_mtu(int mtu){
266        /*60= IPv6+UDP+RTP overhead */
267        if (mtu>60){
268                if (mtu>1500) mtu=1500;/*limit to 1500, the mediastreamer2 buffer are not large enough anyway*/
269                ms_set_payload_max_size(mtu-60);
270        }else ms_set_payload_max_size(0);
271}
Note: See TracBrowser for help on using the repository browser.