ce531efab88a114c24a30b503288ef78459861fb
[pazpar2-moved-to-github.git] / src / getaddrinfo.c
1 /* $Id: getaddrinfo.c,v 1.6 2007-07-09 20:00:41 adam Exp $
2    Copyright (c) 2006-2007, Index Data.
3
4 This file is part of Pazpar2.
5
6 Pazpar2 is free software; you can redistribute it and/or modify it under
7 the terms of the GNU General Public License as published by the Free
8 Software Foundation; either version 2, or (at your option) any later
9 version.
10
11 Pazpar2 is distributed in the hope that it will be useful, but WITHOUT ANY
12 WARRANTY; without even the implied warranty of MERCHANTABILITY or
13 FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
14 for more details.
15
16 You should have received a copy of the GNU General Public License
17 along with Pazpar2; see the file LICENSE.  If not, write to the
18 Free Software Foundation, 59 Temple Place - Suite 330, Boston, MA
19 02111-1307, USA.
20  */
21
22 #if HAVE_CONFIG_H
23 #include "cconfig.h"
24 #endif
25
26 #include "sel_thread.h"
27
28 #include <unistd.h>
29 #include <stdlib.h>
30 #include <pthread.h>
31 #include <assert.h>
32 #include <sys/types.h>
33 #include <sys/socket.h>
34 #include <netdb.h>
35 #include <netinet/in.h>
36
37 #include <yaz/log.h>
38 #include <yaz/nmem.h>
39 #include <yaz/tcpip.h>
40
41 #include "pazpar2.h"
42 #include "connection.h"
43 #include "host.h"
44
45 struct work {
46     char *hostport;  /* hostport to be resolved in separate thread */
47     char *ipport;    /* result or NULL if it could not be resolved */
48     struct host *host; /* host that we're dealing with - mother thread */
49 };
50
51 static int log_level = YLOG_LOG;
52
53 void perform_getaddrinfo(struct work *w)
54 {
55     int res = 0;
56     char *port;
57 #if HAVE_GETADDRINFO
58     struct addrinfo *addrinfo, hints;
59 #else
60     struct hostent *hp;
61 #endif
62     char *hostport = xstrdup(w->hostport);
63     
64     if ((port = strchr(hostport, ':')))
65         *(port++) = '\0';
66     else
67         port = "210";
68     
69 #if HAVE_GETADDRINFO
70     hints.ai_flags = 0;
71     hints.ai_family = PF_INET;
72     hints.ai_socktype = SOCK_STREAM;
73     hints.ai_protocol = IPPROTO_TCP;
74     hints.ai_addrlen = 0;
75     hints.ai_addr = 0;
76     hints.ai_canonname = 0;
77     hints.ai_next = 0;
78     // This is not robust code. It assumes that getaddrinfo always
79     // returns AF_INET address.
80     if ((res = getaddrinfo(hostport, port, &hints, &addrinfo)))
81     {
82         yaz_log(YLOG_WARN, "Failed to resolve %s: %s", 
83                 w->hostport, gai_strerror(res));
84     }
85     else
86     {
87         char ipport[128];
88         unsigned char addrbuf[4];
89         assert(addrinfo->ai_family == PF_INET);
90         memcpy(addrbuf, 
91                &((struct sockaddr_in*)addrinfo->ai_addr)->sin_addr.s_addr, 4);
92         sprintf(ipport, "%u.%u.%u.%u:%s",
93                 addrbuf[0], addrbuf[1], addrbuf[2], addrbuf[3], port);
94         freeaddrinfo(addrinfo);
95         w->ipport = xstrdup(ipport);
96         yaz_log(log_level, "Resolved %s -> %s", hostport, ipport);
97     }
98 #else
99     hp = gethostbyname(hostport);
100     if (!hp)
101     {
102         yaz_log(YLOG_WARN|YLOG_ERRNO, "Failed to resolve %s", hostport);
103     }
104     else
105     {
106         char ipport[128];
107         unsigned char addrbuf[4];
108
109         memcpy(addrbuf, *hp->h_addr_list, 4 * sizeof(unsigned char));
110         sprintf(ipport, "%u.%u.%u.%u:%s",
111                 addrbuf[0], addrbuf[1], addrbuf[2], addrbuf[3], port);
112         w->ipport = xstrdup(ipport);
113         yaz_log(log_level, "Resolved %s -> %s", hostport, ipport);
114     }
115 #endif
116     xfree(hostport);
117 }
118
119 static void work_handler(void *vp)
120 {
121     struct work *w = vp;
122
123     int sec = 0;  /* >0 for debugging/testing purposes */
124     if (sec)
125     {
126         yaz_log(log_level, "waiting %d seconds", sec);
127         sleep(sec);
128     }
129     perform_getaddrinfo(w);
130 }
131
132 void iochan_handler(struct iochan *i, int event)
133 {
134     sel_thread_t p = iochan_getdata(i);
135
136     if (event & EVENT_INPUT)
137     {
138         struct work *w = sel_thread_result(p);
139         w->host->ipport = w->ipport;
140         connect_resolver_host(w->host);
141         xfree(w);
142     }
143 }
144
145 static sel_thread_t resolver_thread = 0;
146
147 static void getaddrinfo_start(void)
148 {
149     int fd;
150     sel_thread_t p = resolver_thread = 
151         sel_thread_create(work_handler, 0 /* work_destroy */, &fd,
152                           3 /* no of resolver threads */);
153     if (!p)
154     {
155         yaz_log(YLOG_FATAL|YLOG_ERRNO, "sel_create_create failed");
156         exit(1);
157     }
158     else
159     {
160         IOCHAN chan = iochan_create(fd, iochan_handler, EVENT_INPUT);
161         iochan_setdata(chan, p);
162         pazpar2_add_channel(chan);
163     }
164     yaz_log(log_level, "resolver start");
165     resolver_thread = p;
166 }
167
168 int host_getaddrinfo(struct host *host)
169 {
170     struct work *w = xmalloc(sizeof(*w));
171     int use_thread = 1; /* =0 to disable threading entirely */
172
173     w->hostport = host->hostport;
174     w->ipport = 0;
175     w->host = host;
176     if (use_thread)
177     {
178         if (resolver_thread == 0)
179             getaddrinfo_start();
180         assert(resolver_thread);
181         sel_thread_add(resolver_thread, w);
182     }
183     else
184     {
185         perform_getaddrinfo(w);
186         host->ipport = w->ipport;
187         xfree(w);
188         if (!host->ipport)
189             return -1;
190     }
191     return 0;
192 }
193
194 /*
195  * Local variables:
196  * c-basic-offset: 4
197  * indent-tabs-mode: nil
198  * End:
199  * vim: shiftwidth=4 tabstop=8 expandtab
200  */