~ubuntu-branches/ubuntu/vivid/curl/vivid

« back to all changes in this revision

Viewing changes to lib/curl_addrinfo.c

  • Committer: Bazaar Package Importer
  • Author(s): Andreas Schuldei
  • Date: 2009-04-02 23:35:45 UTC
  • mto: (1.2.1 upstream) (3.2.3 sid)
  • mto: This revision was merged to the branch mainline in revision 38.
  • Revision ID: james.westby@ubuntu.com-20090402233545-geixkwhe3izccjt7
Tags: upstream-7.19.4
ImportĀ upstreamĀ versionĀ 7.19.4

Show diffs side-by-side

added added

removed removed

Lines of Context:
 
1
/***************************************************************************
 
2
 *                                  _   _ ____  _
 
3
 *  Project                     ___| | | |  _ \| |
 
4
 *                             / __| | | | |_) | |
 
5
 *                            | (__| |_| |  _ <| |___
 
6
 *                             \___|\___/|_| \_\_____|
 
7
 *
 
8
 * Copyright (C) 1998 - 2008, Daniel Stenberg, <daniel@haxx.se>, et al.
 
9
 *
 
10
 * This software is licensed as described in the file COPYING, which
 
11
 * you should have received as part of this distribution. The terms
 
12
 * are also available at http://curl.haxx.se/docs/copyright.html.
 
13
 *
 
14
 * You may opt to use, copy, modify, merge, publish, distribute and/or sell
 
15
 * copies of the Software, and permit persons to whom the Software is
 
16
 * furnished to do so, under the terms of the COPYING file.
 
17
 *
 
18
 * This software is distributed on an "AS IS" basis, WITHOUT WARRANTY OF ANY
 
19
 * KIND, either express or implied.
 
20
 *
 
21
 * $Id: curl_addrinfo.c,v 1.5 2008-11-06 17:19:57 yangtse Exp $
 
22
 ***************************************************************************/
 
23
 
 
24
#include "setup.h"
 
25
 
 
26
#include <curl/curl.h>
 
27
 
 
28
#ifdef NEED_MALLOC_H
 
29
#  include <malloc.h>
 
30
#endif
 
31
#ifdef HAVE_SYS_SOCKET_H
 
32
#  include <sys/socket.h>
 
33
#endif
 
34
#ifdef HAVE_NETINET_IN_H
 
35
#  include <netinet/in.h>
 
36
#endif
 
37
#ifdef HAVE_NETDB_H
 
38
#  include <netdb.h>
 
39
#endif
 
40
#ifdef HAVE_ARPA_INET_H
 
41
#  include <arpa/inet.h>
 
42
#endif
 
43
 
 
44
#ifdef  VMS
 
45
#  include <in.h>
 
46
#  include <inet.h>
 
47
#  include <stdlib.h>
 
48
#endif
 
49
 
 
50
#if defined(NETWARE) && defined(__NOVELL_LIBC__)
 
51
#  undef  in_addr_t
 
52
#  define in_addr_t unsigned long
 
53
#endif
 
54
 
 
55
#include "curl_addrinfo.h"
 
56
 
 
57
#define _MPRINTF_REPLACE /* use our functions only */
 
58
#include <curl/mprintf.h>
 
59
 
 
60
#include "memory.h"
 
61
/* The last #include file should be: */
 
62
#include "memdebug.h"
 
63
 
 
64
 
 
65
/*
 
66
 * Curl_freeaddrinfo()
 
67
 *
 
68
 * This is used to free a linked list of Curl_addrinfo structs along
 
69
 * with all its associated allocated storage. This function should be
 
70
 * called once for each successful call to Curl_getaddrinfo_ex() or to
 
71
 * any function call which actually allocates a Curl_addrinfo struct.
 
72
 */
 
73
 
 
74
void
 
75
Curl_freeaddrinfo(Curl_addrinfo *cahead)
 
76
{
 
77
  Curl_addrinfo *ca, *canext;
 
78
 
 
79
  for(ca = cahead; ca != NULL; ca = canext) {
 
80
 
 
81
    if(ca->ai_addr)
 
82
      free(ca->ai_addr);
 
83
 
 
84
    if(ca->ai_canonname)
 
85
      free(ca->ai_canonname);
 
86
 
 
87
    canext = ca->ai_next;
 
88
 
 
89
    free(ca);
 
90
  }
 
91
}
 
92
 
 
93
 
 
94
#ifdef HAVE_GETADDRINFO
 
95
/*
 
96
 * Curl_getaddrinfo_ex()
 
97
 *
 
98
 * This is a wrapper function around system's getaddrinfo(), with
 
99
 * the only difference that instead of returning a linked list of
 
100
 * addrinfo structs this one returns a linked list of Curl_addrinfo
 
101
 * ones. The memory allocated by this function *MUST* be free'd with
 
102
 * Curl_freeaddrinfo().  For each successful call to this function
 
103
 * there must be an associated call later to Curl_freeaddrinfo().
 
104
 *
 
105
 * There should be no single call to system's getaddrinfo() in the
 
106
 * whole library, any such call should be 'routed' through this one.
 
107
 */
 
108
 
 
109
int
 
110
Curl_getaddrinfo_ex(const char *nodename,
 
111
                    const char *servname,
 
112
                    const struct addrinfo *hints,
 
113
                    Curl_addrinfo **result)
 
114
{
 
115
  const struct addrinfo *ainext;
 
116
  const struct addrinfo *ai;
 
117
  struct addrinfo *aihead;
 
118
  Curl_addrinfo *cafirst = NULL;
 
119
  Curl_addrinfo *calast = NULL;
 
120
  Curl_addrinfo *ca;
 
121
  int error;
 
122
 
 
123
  *result = NULL; /* assume failure */
 
124
 
 
125
  error = getaddrinfo(nodename, servname, hints, &aihead);
 
126
  if(error)
 
127
    return error;
 
128
 
 
129
  for(ai = aihead; ai != NULL; ai = ainext) {
 
130
 
 
131
    if((ca = malloc(sizeof(Curl_addrinfo))) == NULL) {
 
132
      error = EAI_MEMORY;
 
133
      break;
 
134
    }
 
135
 
 
136
    /* copy each structure member individually, member ordering, */
 
137
    /* size, or padding might be different for each structure.   */
 
138
 
 
139
    ca->ai_flags     = ai->ai_flags;
 
140
    ca->ai_family    = ai->ai_family;
 
141
    ca->ai_socktype  = ai->ai_socktype;
 
142
    ca->ai_protocol  = ai->ai_protocol;
 
143
    ca->ai_addrlen   = 0;
 
144
    ca->ai_addr      = NULL;
 
145
    ca->ai_canonname = NULL;
 
146
    ca->ai_next      = NULL;
 
147
 
 
148
    if((ai->ai_addrlen > 0) && (ai->ai_addr != NULL)) {
 
149
      ca->ai_addrlen  = ai->ai_addrlen;
 
150
      if((ca->ai_addr = malloc(ca->ai_addrlen)) == NULL) {
 
151
        error = EAI_MEMORY;
 
152
        free(ca);
 
153
        break;
 
154
      }
 
155
      memcpy(ca->ai_addr, ai->ai_addr, ca->ai_addrlen);
 
156
    }
 
157
 
 
158
    if(ai->ai_canonname != NULL) {
 
159
      if((ca->ai_canonname = strdup(ai->ai_canonname)) == NULL) {
 
160
        error = EAI_MEMORY;
 
161
        if(ca->ai_addr)
 
162
          free(ca->ai_addr);
 
163
        free(ca);
 
164
        break;
 
165
      }
 
166
    }
 
167
 
 
168
    /* if the return list is empty, this becomes the first element */
 
169
    if(!cafirst)
 
170
      cafirst = ca;
 
171
 
 
172
    /* add this element last in the return list */
 
173
    if(calast)
 
174
      calast->ai_next = ca;
 
175
    calast = ca;
 
176
 
 
177
    /* fetch next element fom the addrinfo list */
 
178
    ainext = ai->ai_next;
 
179
  }
 
180
 
 
181
  /* destroy the addrinfo list */
 
182
  if(aihead)
 
183
    freeaddrinfo(aihead);
 
184
 
 
185
  /* if we failed, also destroy the Curl_addrinfo list */
 
186
  if(error) {
 
187
    Curl_freeaddrinfo(cafirst);
 
188
    cafirst = NULL;
 
189
  }
 
190
 
 
191
  *result = cafirst;
 
192
 
 
193
  /* This is not a CURLcode */
 
194
  return error;
 
195
}
 
196
#endif /* HAVE_GETADDRINFO */
 
197
 
 
198
 
 
199
/*
 
200
 * Curl_he2ai()
 
201
 *
 
202
 * This function returns a pointer to the first element of a newly allocated
 
203
 * Curl_addrinfo struct linked list filled with the data of a given hostent.
 
204
 * Curl_addrinfo is meant to work like the addrinfo struct does for a IPv6
 
205
 * stack, but usable also for IPv4, all hosts and environments.
 
206
 *
 
207
 * The memory allocated by this function *MUST* be free'd later on calling
 
208
 * Curl_freeaddrinfo().  For each successful call to this function there
 
209
 * must be an associated call later to Curl_freeaddrinfo().
 
210
 *
 
211
 *   Curl_addrinfo defined in "lib/curl_addrinfo.h"
 
212
 *
 
213
 *     struct Curl_addrinfo {
 
214
 *       int                   ai_flags;
 
215
 *       int                   ai_family;
 
216
 *       int                   ai_socktype;
 
217
 *       int                   ai_protocol;
 
218
 *       socklen_t             ai_addrlen;   * Follow rfc3493 struct addrinfo *
 
219
 *       char                 *ai_canonname;
 
220
 *       struct sockaddr      *ai_addr;
 
221
 *       struct Curl_addrinfo *ai_next;
 
222
 *     };
 
223
 *     typedef struct Curl_addrinfo Curl_addrinfo;
 
224
 *
 
225
 *   hostent defined in <netdb.h>
 
226
 *
 
227
 *     struct hostent {
 
228
 *       char    *h_name;
 
229
 *       char    **h_aliases;
 
230
 *       int     h_addrtype;
 
231
 *       int     h_length;
 
232
 *       char    **h_addr_list;
 
233
 *     };
 
234
 *
 
235
 *   for backward compatibility:
 
236
 *
 
237
 *     #define h_addr  h_addr_list[0]
 
238
 */
 
239
 
 
240
Curl_addrinfo *
 
241
Curl_he2ai(const struct hostent *he, int port)
 
242
{
 
243
  Curl_addrinfo *ai;
 
244
  Curl_addrinfo *prevai = NULL;
 
245
  Curl_addrinfo *firstai = NULL;
 
246
  struct sockaddr_in *addr;
 
247
#ifdef ENABLE_IPV6
 
248
  struct sockaddr_in6 *addr6;
 
249
#endif
 
250
  CURLcode result = CURLE_OK;
 
251
  int i;
 
252
  char *curr;
 
253
 
 
254
  if(!he)
 
255
    /* no input == no output! */
 
256
    return NULL;
 
257
 
 
258
  DEBUGASSERT((he->h_name != NULL) && (he->h_addr_list != NULL));
 
259
 
 
260
  for(i=0; (curr = he->h_addr_list[i]) != NULL; i++) {
 
261
 
 
262
    int ss_size;
 
263
#ifdef ENABLE_IPV6
 
264
    if (he->h_addrtype == AF_INET6)
 
265
      ss_size = sizeof (struct sockaddr_in6);
 
266
    else
 
267
#endif
 
268
      ss_size = sizeof (struct sockaddr_in);
 
269
 
 
270
    if((ai = calloc(1, sizeof(Curl_addrinfo))) == NULL) {
 
271
      result = CURLE_OUT_OF_MEMORY;
 
272
      break;
 
273
    }
 
274
    if((ai->ai_canonname = strdup(he->h_name)) == NULL) {
 
275
      result = CURLE_OUT_OF_MEMORY;
 
276
      free(ai);
 
277
      break;
 
278
    }
 
279
    if((ai->ai_addr = calloc(1, ss_size)) == NULL) {
 
280
      result = CURLE_OUT_OF_MEMORY;
 
281
      free(ai->ai_canonname);
 
282
      free(ai);
 
283
      break;
 
284
    }
 
285
 
 
286
    if(!firstai)
 
287
      /* store the pointer we want to return from this function */
 
288
      firstai = ai;
 
289
 
 
290
    if(prevai)
 
291
      /* make the previous entry point to this */
 
292
      prevai->ai_next = ai;
 
293
 
 
294
    ai->ai_family = he->h_addrtype;
 
295
 
 
296
    /* we return all names as STREAM, so when using this address for TFTP
 
297
       the type must be ignored and conn->socktype be used instead! */
 
298
    ai->ai_socktype = SOCK_STREAM;
 
299
 
 
300
    ai->ai_addrlen = ss_size;
 
301
 
 
302
    /* leave the rest of the struct filled with zero */
 
303
 
 
304
    switch (ai->ai_family) {
 
305
    case AF_INET:
 
306
      addr = (void *)ai->ai_addr; /* storage area for this info */
 
307
 
 
308
      memcpy(&addr->sin_addr, curr, sizeof(struct in_addr));
 
309
      addr->sin_family = (unsigned short)(he->h_addrtype);
 
310
      addr->sin_port = htons((unsigned short)port);
 
311
      break;
 
312
 
 
313
#ifdef ENABLE_IPV6
 
314
    case AF_INET6:
 
315
      addr6 = (void *)ai->ai_addr; /* storage area for this info */
 
316
 
 
317
      memcpy(&addr6->sin6_addr, curr, sizeof(struct in6_addr));
 
318
      addr6->sin6_family = (unsigned short)(he->h_addrtype);
 
319
      addr6->sin6_port = htons((unsigned short)port);
 
320
      break;
 
321
#endif
 
322
    }
 
323
 
 
324
    prevai = ai;
 
325
  }
 
326
 
 
327
  if(result != CURLE_OK) {
 
328
    Curl_freeaddrinfo(firstai);
 
329
    firstai = NULL;
 
330
  }
 
331
 
 
332
  return firstai;
 
333
}
 
334
 
 
335
 
 
336
struct namebuff {
 
337
  struct hostent hostentry;
 
338
  union {
 
339
    struct in_addr  ina4;
 
340
#ifdef ENABLE_IPV6
 
341
    struct in6_addr ina6;
 
342
#endif
 
343
  } addrentry;
 
344
  char *h_addr_list[2];
 
345
};
 
346
 
 
347
 
 
348
/*
 
349
 * Curl_ip2addr()
 
350
 *
 
351
 * This function takes an internet address, in binary form, as input parameter
 
352
 * along with its address family and the string version of the address, and it
 
353
 * returns a Curl_addrinfo chain filled in correctly with information for the
 
354
 * given address/host
 
355
 */
 
356
 
 
357
Curl_addrinfo *
 
358
Curl_ip2addr(int af, const void *inaddr, const char *hostname, int port)
 
359
{
 
360
  Curl_addrinfo *ai;
 
361
 
 
362
#if defined(VMS) && \
 
363
    defined(__INITIAL_POINTER_SIZE) && (__INITIAL_POINTER_SIZE == 64)
 
364
#pragma pointer_size save
 
365
#pragma pointer_size short
 
366
#pragma message disable PTRMISMATCH
 
367
#endif
 
368
 
 
369
  struct hostent  *h;
 
370
  struct namebuff *buf;
 
371
  char  *addrentry;
 
372
  char  *hoststr;
 
373
  int    addrsize;
 
374
 
 
375
  DEBUGASSERT(inaddr && hostname);
 
376
 
 
377
  buf = malloc(sizeof(struct namebuff));
 
378
  if(!buf)
 
379
    return NULL;
 
380
 
 
381
  hoststr = strdup(hostname);
 
382
  if(!hoststr) {
 
383
    free(buf);
 
384
    return NULL;
 
385
  }
 
386
 
 
387
  switch(af) {
 
388
  case AF_INET:
 
389
    addrsize = sizeof(struct in_addr);
 
390
    addrentry = (void *)&buf->addrentry.ina4;
 
391
    memcpy(addrentry, inaddr, sizeof(struct in_addr));
 
392
    break;
 
393
#ifdef ENABLE_IPV6
 
394
  case AF_INET6:
 
395
    addrsize = sizeof(struct in6_addr);
 
396
    addrentry = (void *)&buf->addrentry.ina6;
 
397
    memcpy(addrentry, inaddr, sizeof(struct in6_addr));
 
398
    break;
 
399
#endif
 
400
  default:
 
401
    free(hoststr);
 
402
    free(buf);
 
403
    return NULL;
 
404
  }
 
405
 
 
406
  h = &buf->hostentry;
 
407
  h->h_name = hoststr;
 
408
  h->h_aliases = NULL;
 
409
  h->h_addrtype = (short)af;
 
410
  h->h_length = (short)addrsize;
 
411
  h->h_addr_list = &buf->h_addr_list[0];
 
412
  h->h_addr_list[0] = addrentry;
 
413
  h->h_addr_list[1] = NULL; /* terminate list of entries */
 
414
 
 
415
#if defined(VMS) && \
 
416
    defined(__INITIAL_POINTER_SIZE) && (__INITIAL_POINTER_SIZE == 64)
 
417
#pragma pointer_size restore
 
418
#pragma message enable PTRMISMATCH
 
419
#endif
 
420
 
 
421
  ai = Curl_he2ai(h, port);
 
422
 
 
423
  free(hoststr);
 
424
  free(buf);
 
425
 
 
426
  return ai;
 
427
}
 
428
 
 
429
 
 
430
#if defined(CURLDEBUG) && defined(HAVE_FREEADDRINFO)
 
431
/*
 
432
 * curl_dofreeaddrinfo()
 
433
 *
 
434
 * This is strictly for memory tracing and are using the same style as the
 
435
 * family otherwise present in memdebug.c. I put these ones here since they
 
436
 * require a bunch of structs I didn't wanna include in memdebug.c
 
437
 */
 
438
 
 
439
void
 
440
curl_dofreeaddrinfo(struct addrinfo *freethis,
 
441
                    int line, const char *source)
 
442
{
 
443
  (freeaddrinfo)(freethis);
 
444
  if(logfile)
 
445
    fprintf(logfile, "ADDR %s:%d freeaddrinfo(%p)\n",
 
446
            source, line, (void *)freethis);
 
447
}
 
448
#endif /* defined(CURLDEBUG) && defined(HAVE_FREEADDRINFO) */
 
449
 
 
450
 
 
451
#if defined(CURLDEBUG) && defined(HAVE_GETADDRINFO)
 
452
/*
 
453
 * curl_dogetaddrinfo()
 
454
 *
 
455
 * This is strictly for memory tracing and are using the same style as the
 
456
 * family otherwise present in memdebug.c. I put these ones here since they
 
457
 * require a bunch of structs I didn't wanna include in memdebug.c
 
458
 */
 
459
 
 
460
int
 
461
curl_dogetaddrinfo(const char *hostname,
 
462
                   const char *service,
 
463
                   const struct addrinfo *hints,
 
464
                   struct addrinfo **result,
 
465
                   int line, const char *source)
 
466
{
 
467
  int res=(getaddrinfo)(hostname, service, hints, result);
 
468
  if(0 == res) {
 
469
    /* success */
 
470
    if(logfile)
 
471
      fprintf(logfile, "ADDR %s:%d getaddrinfo() = %p\n",
 
472
              source, line, (void *)*result);
 
473
  }
 
474
  else {
 
475
    if(logfile)
 
476
      fprintf(logfile, "ADDR %s:%d getaddrinfo() failed\n",
 
477
              source, line);
 
478
  }
 
479
  return res;
 
480
}
 
481
#endif /* defined(CURLDEBUG) && defined(HAVE_GETADDRINFO) */
 
482