source: trunk/third-party/libnatpmp/getgateway.c @ 6271

Last change on this file since 6271 was 6271, checked in by charles, 14 years ago

possible fix to the getgateway.c build error -- let's see what hudson thinks of it

File size: 5.7 KB
Line 
1/* $Id: getgateway.c,v 1.8 2008/06/30 14:15:40 nanard Exp $ */
2/* libnatpmp
3 * Copyright (c) 2007-2008, Thomas BERNARD <miniupnp@free.fr>
4 *
5 * Permission to use, copy, modify, and/or distribute this software for any
6 * purpose with or without fee is hereby granted, provided that the above
7 * copyright notice and this permission notice appear in all copies.
8 *
9 * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
10 * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
11 * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
12 * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
13 * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
14 * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
15 * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE. */
16#include <stdio.h>
17#include <ctype.h>
18#include <netinet/in.h>
19#include <sys/param.h>
20/* There is no portable method to get the default route gateway.
21 * So below are three differents functions implementing this.
22 * Parsing /proc/net/route is for linux.
23 * sysctl is the way to access such informations on BSD systems.
24 * Many systems should provide route information through raw PF_ROUTE
25 * sockets. */
26#ifdef __linux__
27  #define USE_PROC_NET_ROUTE
28#elif defined(__APPLE__)
29  #define USE_SYSCTL_NET_ROUTE
30#elif defined(BSD)
31  /*#define USE_SYSCTL_NET_ROUTE*/
32  #define USE_SOCKET_ROUTE
33#elif (defined(sun) && defined(__SVR4))
34  #define USE_SOCKET_ROUTE
35#endif
36
37#ifdef USE_SYSCTL_NET_ROUTE
38#include <stdlib.h>
39#include <sys/sysctl.h>
40#include <sys/socket.h>
41#include <net/route.h>
42#endif
43#ifdef USE_SOCKET_ROUTE
44#include <unistd.h>
45#include <string.h>
46#include <sys/socket.h>
47#include <net/if.h>
48#include <net/route.h>
49#endif
50#include "getgateway.h"
51
52#define SUCCESS (0)
53#define FAILED  (-1)
54
55#ifdef USE_PROC_NET_ROUTE
56int getdefaultgateway(in_addr_t * addr)
57{
58        long d, g;
59        char buf[256];
60        int line = 0;
61        FILE * f;
62        char * p;
63        f = fopen("/proc/net/route", "r");
64        if(!f)
65                return FAILED;
66        while(fgets(buf, sizeof(buf), f)) {
67                if(line > 0) {
68                        p = buf;
69                        while(*p && !isspace(*p))
70                                p++;
71                        while(*p && isspace(*p))
72                                p++;
73                        if(sscanf(p, "%lx%lx", &d, &g)==2) {
74                                if(d == 0) { /* default */
75                                        *addr = g;
76                                        fclose(f);
77                                        return SUCCESS;
78                                }
79                        }
80                }
81                line++;
82        }
83        /* default route not found ! */
84        if(f)
85                fclose(f);
86        return FAILED;
87}
88#endif /* #ifdef USE_PROC_NET_ROUTE */
89
90
91#ifdef USE_SYSCTL_NET_ROUTE
92
93#define ROUNDUP(a) \
94        ((a) > 0 ? (1 + (((a) - 1) | (sizeof(long) - 1))) : sizeof(long))
95
96int getdefaultgateway(in_addr_t * addr)
97{
98#if 0
99        /* net.route.0.inet.dump.0.0 ? */
100        int mib[] = {CTL_NET, PF_ROUTE, 0, AF_INET,
101                     NET_RT_DUMP, 0, 0/*tableid*/};
102#endif
103        /* net.route.0.inet.flags.gateway */
104        int mib[] = {CTL_NET, PF_ROUTE, 0, AF_INET,
105                     NET_RT_FLAGS, RTF_GATEWAY};
106        size_t l;
107        char * buf, * p;
108        struct rt_msghdr * rt;
109        struct sockaddr * sa;
110        struct sockaddr * sa_tab[RTAX_MAX];
111        int i;
112        int r = FAILED;
113        if(sysctl(mib, sizeof(mib)/sizeof(int), 0, &l, 0, 0) < 0) {
114                return FAILED;
115        }
116        if(l>0) {
117                buf = malloc(l);
118                if(sysctl(mib, sizeof(mib)/sizeof(int), buf, &l, 0, 0) < 0) {
119                        return FAILED;
120                }
121                for(p=buf; p<buf+l; p+=rt->rtm_msglen) {
122                        rt = (struct rt_msghdr *)p;
123                        sa = (struct sockaddr *)(rt + 1);
124                        for(i=0; i<RTAX_MAX; i++) {
125                                if(rt->rtm_addrs & (1 << i)) {
126                                        sa_tab[i] = sa;
127                                        sa = (struct sockaddr *)((char *)sa + ROUNDUP(sa->sa_len));
128                                } else {
129                                        sa_tab[i] = NULL;
130                                }
131                        }
132                        if( ((rt->rtm_addrs & (RTA_DST|RTA_GATEWAY)) == (RTA_DST|RTA_GATEWAY))
133              && sa_tab[RTAX_DST]->sa_family == AF_INET
134              && sa_tab[RTAX_GATEWAY]->sa_family == AF_INET) {
135                                if(((struct sockaddr_in *)sa_tab[RTAX_DST])->sin_addr.s_addr == 0) {
136                                        *addr = ((struct sockaddr_in *)(sa_tab[RTAX_GATEWAY]))->sin_addr.s_addr;
137                                        r = SUCCESS;
138                                }
139                        }
140                }
141                free(buf);
142        }
143        return r;
144}
145#endif /* #ifdef USE_SYSCTL_NET_ROUTE */
146
147
148#ifdef USE_SOCKET_ROUTE
149/* Thanks to Darren Kenny for this code */
150#define NEXTADDR(w, u) \
151        if (rtm_addrs & (w)) {\
152            l = sizeof(struct sockaddr); memmove(cp, &(u), l); cp += l;\
153        }
154
155#define rtm m_rtmsg.m_rtm
156
157struct {
158  struct rt_msghdr m_rtm;
159  char       m_space[512];
160} m_rtmsg;
161
162int getdefaultgateway(in_addr_t *addr)
163{
164  int s, seq, l, rtm_addrs, i;
165  pid_t pid;
166  struct sockaddr so_dst, so_mask;
167  char *cp = m_rtmsg.m_space; 
168  struct sockaddr *gate = NULL, *sa;
169  struct rt_msghdr *msg_hdr;
170
171  pid = getpid();
172  seq = 0;
173  rtm_addrs = RTA_DST | RTA_NETMASK;
174
175  memset(&so_dst, 0, sizeof(so_dst));
176  memset(&so_mask, 0, sizeof(so_mask));
177  memset(&rtm, 0, sizeof(struct rt_msghdr));
178
179  rtm.rtm_type = RTM_GET;
180  rtm.rtm_flags = RTF_UP | RTF_GATEWAY;
181  rtm.rtm_version = RTM_VERSION;
182  rtm.rtm_seq = ++seq;
183  rtm.rtm_addrs = rtm_addrs; 
184
185  so_dst.sa_family = AF_INET;
186  so_mask.sa_family = AF_INET;
187
188  NEXTADDR(RTA_DST, so_dst);
189  NEXTADDR(RTA_NETMASK, so_mask);
190
191  rtm.rtm_msglen = l = cp - (char *)&m_rtmsg;
192
193  s = socket(PF_ROUTE, SOCK_RAW, 0);
194
195  if (write(s, (char *)&m_rtmsg, l) < 0) {
196      close(s);
197      return FAILED;
198  }
199
200  do {
201    l = read(s, (char *)&m_rtmsg, sizeof(m_rtmsg));
202  } while (l > 0 && (rtm.rtm_seq != seq || rtm.rtm_pid != pid));
203                       
204  close(s);
205
206  msg_hdr = &rtm;
207
208  cp = ((char *)(msg_hdr + 1));
209  if (msg_hdr->rtm_addrs) {
210    for (i = 1; i; i <<= 1)
211      if (i & msg_hdr->rtm_addrs) {
212        sa = (struct sockaddr *)cp;
213        if (i == RTA_GATEWAY )
214          gate = sa;
215
216        cp += sizeof(struct sockaddr);
217      }
218  } else {
219      return FAILED;
220  }
221
222
223  if (gate != NULL ) {
224      *addr = ((struct sockaddr_in *)gate)->sin_addr.s_addr;
225      return SUCCESS;
226  } else {
227      return FAILED;
228  }
229}
230#endif /* #ifdef USE_SOCKET_ROUTE */
231
Note: See TracBrowser for help on using the repository browser.