Ruby  2.0.0p247(2013-06-27revision41674)
ext/socket/getnameinfo.c
Go to the documentation of this file.
00001 /*
00002  * Copyright (C) 1995, 1996, 1997, and 1998 WIDE Project.
00003  * All rights reserved.
00004  *
00005  * Redistribution and use in source and binary forms, with or without
00006  * modification, are permitted provided that the following conditions
00007  * are met:
00008  * 1. Redistributions of source code must retain the above copyright
00009  *    notice, this list of conditions and the following disclaimer.
00010  * 2. Redistributions in binary form must reproduce the above copyright
00011  *    notice, this list of conditions and the following disclaimer in the
00012  *    documentation and/or other materials provided with the distribution.
00013  * 3. Neither the name of the project nor the names of its contributors
00014  *    may be used to endorse or promote products derived from this software
00015  *    without specific prior written permission.
00016  *
00017  * THIS SOFTWARE IS PROVIDED BY THE PROJECT AND CONTRIBUTORS ``AS IS'' AND
00018  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
00019  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
00020  * ARE DISCLAIMED.  IN NO EVENT SHALL THE PROJECT OR CONTRIBUTORS BE LIABLE
00021  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
00022  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
00023  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
00024  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
00025  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
00026  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
00027  * SUCH DAMAGE.
00028  */
00029 
00030 /*
00031  * Issues to be discussed:
00032  * - Thread safe-ness must be checked
00033  * - Return values.  There seems to be no standard for return value (RFC2133)
00034  *   but INRIA implementation returns EAI_xxx defined for getaddrinfo().
00035  */
00036 
00037 #include "ruby/config.h"
00038 #ifdef RUBY_EXTCONF_H
00039 #include RUBY_EXTCONF_H
00040 #endif
00041 #include <stdio.h>
00042 #include <sys/types.h>
00043 #ifndef _WIN32
00044 #if defined(__BEOS__) && !defined(__HAIKU__) && !defined(BONE)
00045 # include <net/socket.h>
00046 #else
00047 # include <sys/socket.h>
00048 #endif
00049 #include <netinet/in.h>
00050 #if defined(HAVE_ARPA_INET_H)
00051 #include <arpa/inet.h>
00052 #endif
00053 #if defined(HAVE_ARPA_NAMESER_H)
00054 #include <arpa/nameser.h>
00055 #endif
00056 #include <netdb.h>
00057 #if defined(HAVE_RESOLV_H)
00058 #include <resolv.h>
00059 #endif
00060 #endif
00061 #ifdef _WIN32
00062 #include <winsock2.h>
00063 #include <ws2tcpip.h>
00064 #define snprintf _snprintf
00065 #endif
00066 
00067 #include <string.h>
00068 #include <stddef.h>
00069 
00070 #ifdef SOCKS5
00071 #include <socks.h>
00072 #endif
00073 
00074 #include "addrinfo.h"
00075 #include "sockport.h"
00076 
00077 #define SUCCESS 0
00078 #define ANY 0
00079 #define YES 1
00080 #define NO  0
00081 
00082 struct sockinet {
00083         u_char  si_len;
00084         u_char  si_family;
00085         u_short si_port;
00086 };
00087 
00088 static struct afd {
00089         int a_af;
00090         int a_addrlen;
00091         int a_socklen;
00092         int a_off;
00093 } afdl [] = {
00094 #ifdef INET6
00095 #define N_INET6 0
00096         {PF_INET6, sizeof(struct in6_addr),
00097          sizeof(struct sockaddr_in6),
00098          offsetof(struct sockaddr_in6, sin6_addr)},
00099 #define N_INET  1
00100 #else
00101 #define N_INET  0
00102 #endif
00103         {PF_INET, sizeof(struct in_addr),
00104          sizeof(struct sockaddr_in),
00105          offsetof(struct sockaddr_in, sin_addr)},
00106         {0, 0, 0, 0},
00107 };
00108 
00109 #define ENI_NOSOCKET    0
00110 #define ENI_NOSERVNAME  1
00111 #define ENI_NOHOSTNAME  2
00112 #define ENI_MEMORY      3
00113 #define ENI_SYSTEM      4
00114 #define ENI_FAMILY      5
00115 #define ENI_SALEN       6
00116 
00117 #ifndef HAVE_INET_NTOP
00118 static const char *
00119 inet_ntop(int af, const void *addr, char *numaddr, size_t numaddr_len)
00120 {
00121 #ifdef HAVE_INET_NTOA
00122         struct in_addr in;
00123         memcpy(&in.s_addr, addr, sizeof(in.s_addr));
00124         snprintf(numaddr, numaddr_len, "%s", inet_ntoa(in));
00125 #else
00126         unsigned long x = ntohl(*(unsigned long*)addr);
00127         snprintf(numaddr, numaddr_len, "%d.%d.%d.%d",
00128                  (int) (x>>24) & 0xff, (int) (x>>16) & 0xff,
00129                  (int) (x>> 8) & 0xff, (int) (x>> 0) & 0xff);
00130 #endif
00131         return numaddr;
00132 }
00133 #endif
00134 
00135 int
00136 getnameinfo(const struct sockaddr *sa, socklen_t salen, char *host, socklen_t hostlen, char *serv, socklen_t servlen, int flags)
00137 {
00138         struct afd *afd;
00139         struct hostent *hp;
00140         u_short port;
00141         int family, len, i;
00142         char *addr, *p;
00143         u_long v4a;
00144 #ifdef INET6
00145         u_char pfx;
00146 #endif
00147         int h_error;
00148         char numserv[512];
00149         char numaddr[512];
00150 
00151         if (sa == NULL)
00152                 return ENI_NOSOCKET;
00153 
00154         len = SA_LEN(sa);
00155         if (len != salen) return ENI_SALEN;
00156 
00157         family = sa->sa_family;
00158         for (i = 0; afdl[i].a_af; i++)
00159                 if (afdl[i].a_af == family) {
00160                         afd = &afdl[i];
00161                         goto found;
00162                 }
00163         return ENI_FAMILY;
00164 
00165  found:
00166         if (len != afd->a_socklen) return ENI_SALEN;
00167 
00168         port = ((struct sockinet *)sa)->si_port; /* network byte order */
00169         addr = (char *)sa + afd->a_off;
00170 
00171         if (serv == NULL || servlen == 0) {
00172                 /* what we should do? */
00173         } else if (flags & NI_NUMERICSERV) {
00174                 snprintf(numserv, sizeof(numserv), "%d", ntohs(port));
00175                 if (strlen(numserv) + 1 > servlen)
00176                         return ENI_MEMORY;
00177                 strcpy(serv, numserv);
00178         } else {
00179 #if defined(HAVE_GETSERVBYPORT)
00180                 struct servent *sp = getservbyport(port, (flags & NI_DGRAM) ? "udp" : "tcp");
00181                 if (sp) {
00182                         if (strlen(sp->s_name) + 1 > servlen)
00183                                 return ENI_MEMORY;
00184                         strcpy(serv, sp->s_name);
00185                 } else
00186                         return ENI_NOSERVNAME;
00187 #else
00188                 return ENI_NOSERVNAME;
00189 #endif
00190         }
00191 
00192         switch (sa->sa_family) {
00193         case AF_INET:
00194                 v4a = ntohl(((struct sockaddr_in *)sa)->sin_addr.s_addr);
00195                 if (IN_MULTICAST(v4a) || IN_EXPERIMENTAL(v4a))
00196                         flags |= NI_NUMERICHOST;
00197                 v4a >>= IN_CLASSA_NSHIFT;
00198                 if (v4a == 0)
00199                         flags |= NI_NUMERICHOST;
00200                 break;
00201 #ifdef INET6
00202         case AF_INET6:
00203 #ifdef HAVE_ADDR8
00204                 pfx = ((struct sockaddr_in6 *)sa)->sin6_addr.s6_addr8[0];
00205 #else
00206                 pfx = ((struct sockaddr_in6 *)sa)->sin6_addr.s6_addr[0];
00207 #endif
00208                 if (pfx == 0 || pfx == 0xfe || pfx == 0xff)
00209                         flags |= NI_NUMERICHOST;
00210                 break;
00211 #endif
00212         }
00213         if (host == NULL || hostlen == 0) {
00214                 /* what should we do? */
00215         } else if (flags & NI_NUMERICHOST) {
00216                 if (inet_ntop(afd->a_af, addr, numaddr, sizeof(numaddr))
00217                     == NULL)
00218                         return ENI_SYSTEM;
00219                 if (strlen(numaddr) > hostlen)
00220                         return ENI_MEMORY;
00221                 strcpy(host, numaddr);
00222         } else {
00223 #ifdef INET6
00224                 hp = getipnodebyaddr(addr, afd->a_addrlen, afd->a_af, &h_error);
00225 #else
00226                 hp = gethostbyaddr(addr, afd->a_addrlen, afd->a_af);
00227                 h_error = h_errno;
00228 #endif
00229 
00230                 if (hp) {
00231                         if (flags & NI_NOFQDN) {
00232                                 p = strchr(hp->h_name, '.');
00233                                 if (p) *p = '\0';
00234                         }
00235                         if (strlen(hp->h_name) + 1 > hostlen) {
00236 #ifdef INET6
00237                                 freehostent(hp);
00238 #endif
00239                                 return ENI_MEMORY;
00240                         }
00241                         strcpy(host, hp->h_name);
00242 #ifdef INET6
00243                         freehostent(hp);
00244 #endif
00245                 } else {
00246                         if (flags & NI_NAMEREQD)
00247                                 return ENI_NOHOSTNAME;
00248                         if (inet_ntop(afd->a_af, addr, numaddr, sizeof(numaddr))
00249                             == NULL)
00250                                 return ENI_NOHOSTNAME;
00251                         if (strlen(numaddr) > hostlen)
00252                                 return ENI_MEMORY;
00253                         strcpy(host, numaddr);
00254                 }
00255         }
00256         return SUCCESS;
00257 }
00258