Return to getroute.c CVS log

Up to [ELWIX - Embedded LightWeight unIX -] / embedaddon / miniupnpd / bsd

Annotation of embedaddon/miniupnpd/bsd/getroute.c, revision 1.1

1.1     ! misho       1: /* $Id: getroute.c,v 1.3 2013/02/06 13:11:45 nanard Exp $ */
        !             2: /* MiniUPnP project
        !             3:  * http://miniupnp.free.fr/ or http://miniupnp.tuxfamily.org/
        !             4:  * (c) 2006-2013 Thomas Bernard
        !             5:  * This software is subject to the conditions detailed
        !             6:  * in the LICENCE file provided within the distribution */
        !             7: 
        !             8: #include <stdio.h>
        !             9: #include <syslog.h>
        !            10: #include <unistd.h>
        !            11: #include <string.h>
        !            12: #include <sys/socket.h>
        !            13: #include <net/if.h>
        !            14: #include <net/route.h>
        !            15: #include <netinet/in.h>
        !            16: #ifdef AF_LINK
        !            17: #include <net/if_dl.h>
        !            18: #endif
        !            19: 
        !            20: #include "../config.h"
        !            21: #include "../upnputils.h"
        !            22: 
        !            23: 
        !            24: int
        !            25: get_src_for_route_to(const struct sockaddr * dst,
        !            26:                      void * src, size_t * src_len,
        !            27:                      int * index)
        !            28: {
        !            29:        int found = 0;
        !            30:        int s;
        !            31:        int l, i;
        !            32:        char * p;
        !            33:        struct sockaddr * sa;
        !            34:        struct {
        !            35:          struct rt_msghdr m_rtm;
        !            36:          char       m_space[512];
        !            37:        } m_rtmsg;
        !            38: #define rtm m_rtmsg.m_rtm
        !            39: 
        !            40:        if(dst == NULL)
        !            41:                return -1;
        !            42: #ifdef __APPLE__
        !            43:        if(dst->sa_family == AF_INET6) {
        !            44:                syslog(LOG_ERR, "Sorry, get_src_for_route_to() is known to fail with IPV6 on OS X...");
        !            45:                return -1;
        !            46:        }
        !            47: #endif
        !            48:        s = socket(PF_ROUTE, SOCK_RAW, dst->sa_family);
        !            49:        if(s < 0) {
        !            50:                syslog(LOG_ERR, "socket(PF_ROUTE) failed : %m");
        !            51:                return -1;
        !            52:        }
        !            53:        memset(&rtm, 0, sizeof(rtm));
        !            54:        rtm.rtm_type = RTM_GET;
        !            55:        rtm.rtm_flags = RTF_UP;
        !            56:        rtm.rtm_version = RTM_VERSION;
        !            57:        rtm.rtm_seq = 1;
        !            58:        rtm.rtm_addrs = RTA_DST;        /* destination address */
        !            59:        memcpy(m_rtmsg.m_space, dst, sizeof(struct sockaddr));
        !            60:        rtm.rtm_msglen = sizeof(struct rt_msghdr) + sizeof(struct sockaddr);
        !            61:        if(write(s, &m_rtmsg, rtm.rtm_msglen) < 0) {
        !            62:                syslog(LOG_ERR, "write: %m");
        !            63:                close(s);
        !            64:                return -1;
        !            65:        }
        !            66: 
        !            67:        do {
        !            68:                l = read(s, &m_rtmsg, sizeof(m_rtmsg));
        !            69:                if(l<0) {
        !            70:                        syslog(LOG_ERR, "read: %m");
        !            71:                        close(s);
        !            72:                        return -1;
        !            73:                }
        !            74:                syslog(LOG_DEBUG, "read l=%d seq=%d pid=%d",
        !            75:                       l, rtm.rtm_seq, rtm.rtm_pid);
        !            76:        } while(l > 0 && (rtm.rtm_pid != getpid() || rtm.rtm_seq != 1));
        !            77:        close(s);
        !            78:        p = m_rtmsg.m_space;
        !            79:        if(rtm.rtm_addrs) {
        !            80:                for(i=1; i<0x8000; i <<= 1) {
        !            81:                        if(i & rtm.rtm_addrs) {
        !            82:                                char tmp[256] = { 0 };
        !            83:                                sa = (struct sockaddr *)p;
        !            84:                                sockaddr_to_string(sa, tmp, sizeof(tmp));
        !            85:                                syslog(LOG_DEBUG, "type=%d sa_len=%d sa_family=%d %s",
        !            86:                                       i, sa->sa_len, sa->sa_family, tmp);
        !            87:                                if((i == RTA_DST || i == RTA_GATEWAY) &&
        !            88:                                   (src_len && src)) {
        !            89:                                        size_t len = 0;
        !            90:                                        void * paddr = NULL;
        !            91:                                        if(sa->sa_family == AF_INET) {
        !            92:                                                paddr = &((struct sockaddr_in *)sa)->sin_addr;
        !            93:                                                len = sizeof(struct in_addr);
        !            94:                                        } else if(sa->sa_family == AF_INET6) {
        !            95:                                                paddr = &((struct sockaddr_in6 *)sa)->sin6_addr;
        !            96:                                                len = sizeof(struct in6_addr);
        !            97:                                        }
        !            98:                                        if(paddr) {
        !            99:                                                if(*src_len < len) {
        !           100:                                                        syslog(LOG_WARNING, "cannot copy src. %u<%u",
        !           101:                                                               (unsigned)*src_len, (unsigned)len);
        !           102:                                                        return -1;
        !           103:                                                }
        !           104:                                                memcpy(src, paddr, len);
        !           105:                                                *src_len = len;
        !           106:                                                found = 1;
        !           107:                                        }
        !           108:                                }
        !           109: #ifdef AF_LINK
        !           110:                                if(sa->sa_family == AF_LINK) {
        !           111:                                        struct sockaddr_dl * sdl = (struct sockaddr_dl *)sa;
        !           112:                                        if(index)
        !           113:                                                *index = sdl->sdl_index;
        !           114:                                }
        !           115: #endif
        !           116:                                p += sa->sa_len;
        !           117:                        }
        !           118:                }
        !           119:        }
        !           120:        return found ? 0 : -1;
        !           121: }
        !           122: