source: lib/ftutil.c @ 50d0a72

Last change on this file since 50d0a72 was 2e8523b, checked in by dx <dx@…>, at 2016-12-31T20:40:09Z

Use NI_MAXHOST rather than HOST_NAME_MAX for host lengths.

This constant is always available and meant to be used with
getnameinfo().

This fixes the build on Debian GNU/kFreeBSD.

  • Property mode set to 100644
File size: 4.9 KB
Line 
1/***************************************************************************\
2*                                                                           *
3*  BitlBee - An IRC to IM gateway                                           *
4*  Utility functions for file transfer                                      *
5*                                                                           *
6*  Copyright 2008 Uli Meis <a.sporto+bee@gmail.com>                         *
7*                                                                           *
8*  This program is free software; you can redistribute it and/or modify     *
9*  it under the terms of the GNU General Public License as published by     *
10*  the Free Software Foundation; either version 2 of the License, or        *
11*  (at your option) any later version.                                      *
12*                                                                           *
13*  This program is distributed in the hope that it will be useful,          *
14*  but WITHOUT ANY WARRANTY; without even the implied warranty of           *
15*  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the            *
16*  GNU General Public License for more details.                             *
17*                                                                           *
18*  You should have received a copy of the GNU General Public License along  *
19*  with this program; if not, write to the Free Software Foundation, Inc.,  *
20*  51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.              *
21*                                                                           *
22\***************************************************************************/
23
24#define BITLBEE_CORE
25#include "bitlbee.h"
26#include <poll.h>
27#include <netinet/tcp.h>
28#include "lib/ftutil.h"
29
30#define ASSERTSOCKOP(op, msg) \
31        if ((op) == -1) { \
32                g_snprintf(errmsg, sizeof(errmsg), msg ": %s", strerror(errno)); \
33                return -1; }
34
35/*
36 * Creates a listening socket and returns it in saddr_ptr.
37 */
38int ft_listen(struct sockaddr_storage *saddr_ptr, char *host, char *port, int copy_fd, int for_bitlbee_client,
39              char **errptr)
40{
41        int fd, gret, saddrlen;
42        struct addrinfo hints, *rp;
43        socklen_t ssize = sizeof(struct sockaddr_storage);
44        struct sockaddr_storage saddrs = {0}, *saddr = &saddrs;
45        static char errmsg[1024];
46        char *ftlisten = global.conf->ft_listen;
47
48        if (errptr) {
49                *errptr = errmsg;
50        }
51
52        strcpy(port, "0");
53
54        /* Format is <IP-A>[:<Port-A>];<IP-B>[:<Port-B>] where
55         * A is for connections with the bitlbee client (DCC)
56         * and B is for connections with IM peers.
57         */
58        if (ftlisten) {
59                char *scolon = strchr(ftlisten, ';');
60                char *colon;
61
62                if (scolon) {
63                        if (for_bitlbee_client) {
64                                *scolon = '\0';
65                                strncpy(host, ftlisten, NI_MAXHOST);
66                                *scolon = ';';
67                        } else {
68                                strncpy(host, scolon + 1, NI_MAXHOST);
69                        }
70                } else {
71                        strncpy(host, ftlisten, NI_MAXHOST);
72                }
73
74                if ((colon = strchr(host, ':'))) {
75                        *colon = '\0';
76                        strncpy(port, colon + 1, 5);
77                }
78        } else if (copy_fd >= 0 && getsockname(copy_fd, (struct sockaddr*) &saddrs, &ssize) == 0 &&
79                   (saddrs.ss_family == AF_INET || saddrs.ss_family == AF_INET6) &&
80                   getnameinfo((struct sockaddr*) &saddrs, ssize, host, NI_MAXHOST,
81                               NULL, 0, NI_NUMERICHOST) == 0) {
82                /* We just took our local address on copy_fd, which is likely to be a
83                   sensible address from which we can do a file transfer now - the
84                   most sensible we can get easily. */
85        } else {
86                ASSERTSOCKOP(gethostname(host, NI_MAXHOST), "gethostname()");
87        }
88
89        memset(&hints, 0, sizeof(struct addrinfo));
90        hints.ai_socktype = SOCK_STREAM;
91        hints.ai_flags = AI_NUMERICSERV;
92
93        if ((gret = getaddrinfo(host, port, &hints, &rp)) != 0) {
94                sprintf(errmsg, "getaddrinfo() failed: %s", gai_strerror(gret));
95                return -1;
96        }
97
98        saddrlen = rp->ai_addrlen;
99
100        memcpy(saddr, rp->ai_addr, saddrlen);
101
102        freeaddrinfo(rp);
103
104        ASSERTSOCKOP(fd = socket(saddr->ss_family, SOCK_STREAM, 0), "Opening socket");
105        ASSERTSOCKOP(bind(fd, ( struct sockaddr *) saddr, saddrlen), "Binding socket");
106        ASSERTSOCKOP(listen(fd, 1), "Making socket listen");
107
108        if (!inet_ntop(saddr->ss_family, saddr->ss_family == AF_INET ?
109                       ( void * ) &(( struct sockaddr_in * ) saddr)->sin_addr.s_addr :
110                       ( void * ) &(( struct sockaddr_in6 * ) saddr)->sin6_addr.s6_addr,
111                       host, NI_MAXHOST)) {
112                strcpy(errmsg, "inet_ntop failed on listening socket");
113                return -1;
114        }
115
116        ssize = sizeof(struct sockaddr_storage);
117        ASSERTSOCKOP(getsockname(fd, ( struct sockaddr *) saddr, &ssize), "Getting socket name");
118
119        if (saddr->ss_family == AF_INET) {
120                g_snprintf(port, 6, "%d", ntohs(((struct sockaddr_in *) saddr)->sin_port));
121        } else {
122                g_snprintf(port, 6, "%d", ntohs(((struct sockaddr_in6 *) saddr)->sin6_port));
123        }
124
125        if (saddr_ptr) {
126                memcpy(saddr_ptr, saddr, saddrlen);
127        }
128
129        /* I hate static-length strings.. */
130        host[NI_MAXHOST - 1] = '\0';
131        port[5] = '\0';
132
133        return fd;
134}
Note: See TracBrowser for help on using the repository browser.