Simplify fake getname/addrinfo() functions, possibly fixing freeing a NULL pointer.
[tinc] / lib / fake-getaddrinfo.c
1 /*
2  * fake library for ssh
3  *
4  * This file includes getaddrinfo(), freeaddrinfo() and gai_strerror().
5  * These funtions are defined in rfc2133.
6  *
7  * But these functions are not implemented correctly. The minimum subset
8  * is implemented for ssh use only. For exapmle, this routine assumes
9  * that ai_family is AF_INET. Don't use it for another purpose.
10  */
11
12 #include "system.h"
13
14 #include "ipv4.h"
15 #include "ipv6.h"
16 #include "fake-getaddrinfo.h"
17
18 #ifndef HAVE_GAI_STRERROR
19 char *gai_strerror(int ecode)
20 {
21         switch (ecode) {
22                 case EAI_NODATA:
23                         return "No address associated with hostname";
24                 case EAI_MEMORY:
25                         return "Memory allocation failure";
26                 default:
27                         return "Unknown error";
28         }
29 }    
30 #endif /* !HAVE_GAI_STRERROR */
31
32 #ifndef HAVE_FREEADDRINFO
33 void freeaddrinfo(struct addrinfo *ai)
34 {
35         struct addrinfo *next;
36
37         while(ai) {
38                 next = ai->ai_next;
39                 free(ai);
40                 ai = next;
41         }
42 }
43 #endif /* !HAVE_FREEADDRINFO */
44
45 #ifndef HAVE_GETADDRINFO
46 static struct addrinfo *malloc_ai(uint16_t port, uint32_t addr)
47 {
48         struct addrinfo *ai;
49
50         ai = xmalloc_and_zero(sizeof(struct addrinfo) + sizeof(struct sockaddr_in));
51         
52         ai->ai_addr = (struct sockaddr *)(ai + 1);
53         ai->ai_addrlen = sizeof(struct sockaddr_in);
54         ai->ai_addr->sa_family = ai->ai_family = AF_INET;
55
56         ((struct sockaddr_in *)(ai)->ai_addr)->sin_port = port;
57         ((struct sockaddr_in *)(ai)->ai_addr)->sin_addr.s_addr = addr;
58         
59         return ai;
60 }
61
62 int getaddrinfo(const char *hostname, const char *servname, const struct addrinfo *hints, struct addrinfo **res)
63 {
64         struct addrinfo *prev = NULL;
65         struct hostent *hp;
66         struct in_addr in = {0};
67         int i;
68         uint16_t port = 0;
69
70         if (servname)
71                 port = htons(atoi(servname));
72
73         if (hints && hints->ai_flags & AI_PASSIVE) {
74                 *res = malloc_ai(port, htonl(0x00000000));
75                 return 0;
76         }
77                 
78         if (!hostname) {
79                 *res = malloc_ai(port, htonl(0x7f000001));
80                 return 0;
81         }
82         
83         hp = gethostbyname(hostname);
84
85         if(!hp || !hp->h_addr_list[0])
86                 return EAI_NODATA;
87
88         for (i = 0; hp->h_addr_list[i]; i++) {
89                 *res = malloc_ai(port, ((struct in_addr *)hp->h_addr_list[i])->s_addr);
90
91                 if(prev)
92                         prev->ai_next = *res;
93
94                 prev = *res;
95         }
96
97         return 0;
98 }
99 #endif /* !HAVE_GETADDRINFO */