2 * This file is part of DisOrder.
3 * Copyright (C) 2004, 2007, 2008 Richard Kettlewell
5 * This program is free software; you can redistribute it and/or modify
6 * it under the terms of the GNU General Public License as published by
7 * the Free Software Foundation; either version 2 of the License, or
8 * (at your option) any later version.
10 * This program is distributed in the hope that it will be useful, but
11 * WITHOUT ANY WARRANTY; without even the implied warranty of
12 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
13 * General Public License for more details.
15 * You should have received a copy of the GNU General Public License
16 * along with this program; if not, write to the Free Software
17 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307
21 * @brief Socket address support */
25 #include <sys/types.h>
26 #include <sys/socket.h>
27 #include <netinet/in.h>
28 #include <arpa/inet.h>
36 /** @brief Convert a pair of strings to an address
37 * @param a Pointer to string list
38 * @param pref Hints structure for getaddrinfo, or NULL
39 * @param namep Where to store address description, or NULL
40 * @return Address info structure or NULL on error
42 * This converts one or two strings into an address specification suitable
43 * for passing to socket(), bind() etc.
45 * If there is only one string then it is assumed to be the service
46 * name (port number). If there are two then the first is the host
47 * name and the second the service name.
49 * @p namep is used to return a description of the address suitable
50 * for use in log messages.
52 * If an error occurs a message is logged and a null pointer returned.
54 struct addrinfo *get_address(const struct stringlist *a,
55 const struct addrinfo *pref,
63 byte_xasprintf(&name, "host * service %s", a->s[0]);
64 if((rc = getaddrinfo(0, a->s[0], pref, &res))) {
65 error(0, "getaddrinfo %s: %s", a->s[0], gai_strerror(rc));
70 byte_xasprintf(&name, "host %s service %s", a->s[0], a->s[1]);
71 if((rc = getaddrinfo(a->s[0], a->s[1], pref, &res))) {
72 error(0, "getaddrinfo %s %s: %s", a->s[0], a->s[1], gai_strerror(rc));
77 error(0, "invalid network address specification (n=%d)", a->n);
80 if(!res || (pref && res->ai_socktype != pref->ai_socktype)) {
81 error(0, "getaddrinfo didn't give us a suitable socket address");
91 /** @brief Comparison function for address information
93 * Suitable for qsort().
95 int addrinfocmp(const struct addrinfo *a,
96 const struct addrinfo *b) {
97 const struct sockaddr_in *ina, *inb;
98 const struct sockaddr_in6 *in6a, *in6b;
100 if(a->ai_family != b->ai_family) return a->ai_family - b->ai_family;
101 if(a->ai_socktype != b->ai_socktype) return a->ai_socktype - b->ai_socktype;
102 if(a->ai_protocol != b->ai_protocol) return a->ai_protocol - b->ai_protocol;
103 switch(a->ai_family) {
105 ina = (const struct sockaddr_in *)a->ai_addr;
106 inb = (const struct sockaddr_in *)b->ai_addr;
107 if(ina->sin_port != inb->sin_port) return ina->sin_port - inb->sin_port;
108 return ina->sin_addr.s_addr - inb->sin_addr.s_addr;
111 in6a = (const struct sockaddr_in6 *)a->ai_addr;
112 in6b = (const struct sockaddr_in6 *)b->ai_addr;
113 if(in6a->sin6_port != in6b->sin6_port)
114 return in6a->sin6_port - in6b->sin6_port;
115 return memcmp(&in6a->sin6_addr, &in6b->sin6_addr,
116 sizeof (struct in6_addr));
118 error(0, "unsupported protocol family %d", a->ai_protocol);
119 return memcmp(a->ai_addr, b->ai_addr, a->ai_addrlen); /* kludge */
123 /** @brief Return nonzero if @p sin4 is an IPv4 multicast address */
124 static inline int multicast4(const struct sockaddr_in *sin4) {
125 return IN_MULTICAST(ntohl(sin4->sin_addr.s_addr));
128 /** @brief Return nonzero if @p sin6 is an IPv6 multicast address */
129 static inline int multicast6(const struct sockaddr_in6 *sin6) {
130 return IN6_IS_ADDR_MULTICAST(&sin6->sin6_addr);
133 /** @brief Return true if @p sa represents a multicast address */
134 int multicast(const struct sockaddr *sa) {
135 switch(sa->sa_family) {
137 return multicast4((const struct sockaddr_in *)sa);
139 return multicast6((const struct sockaddr_in6 *)sa);
145 /** @brief Format an IPv4 address */
146 static inline char *format_sockaddr4(const struct sockaddr_in *sin4) {
147 char buffer[1024], *r;
150 byte_xasprintf(&r, "%s port %u",
151 inet_ntop(sin4->sin_family, &sin4->sin_addr,
152 buffer, sizeof buffer),
153 ntohs(sin4->sin_port));
155 byte_xasprintf(&r, "%s",
156 inet_ntop(sin4->sin_family, &sin4->sin_addr,
157 buffer, sizeof buffer));
161 /** @brief Format an IPv6 address */
162 static inline char *format_sockaddr6(const struct sockaddr_in6 *sin6) {
163 char buffer[1024], *r;
166 byte_xasprintf(&r, "%s port %u",
167 inet_ntop(sin6->sin6_family, &sin6->sin6_addr,
168 buffer, sizeof buffer),
169 ntohs(sin6->sin6_port));
171 byte_xasprintf(&r, "%s",
172 inet_ntop(sin6->sin6_family, &sin6->sin6_addr,
173 buffer, sizeof buffer));
177 /** @brief Format a UNIX socket address */
178 static inline char *format_sockaddrun(const struct sockaddr_un *sun) {
179 return xstrdup(sun->sun_path);
182 /** @brief Construct a text description a sockaddr
183 * @param sa Socket address
184 * @return Human-readable form of address
186 char *format_sockaddr(const struct sockaddr *sa) {
187 switch(sa->sa_family) {
189 return format_sockaddr4((const struct sockaddr_in *)sa);
191 return format_sockaddr6((const struct sockaddr_in6 *)sa);
193 return format_sockaddrun((const struct sockaddr_un *)sa);