source: trunk/src/sh_ipvx.c@ 525

Last change on this file since 525 was 509, checked in by katerina, 8 years ago

Fix for ticket #402 (memory leak in server).
File size: 12.9 KB
Line 
1/* SAMHAIN file system integrity testing */
2/* Copyright (C) 2010 Rainer Wichmann */
3/* */
4/* This program is free software; you can redistribute it */
5/* and/or modify */
6/* it under the terms of the GNU General Public License as */
7/* published by */
8/* the Free Software Foundation; either version 2 of the License, or */
9/* (at your option) any later version. */
10/* */
11/* This program is distributed in the hope that it will be useful, */
12/* but WITHOUT ANY WARRANTY; without even the implied warranty of */
13/* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the */
14/* GNU General Public License for more details. */
15/* */
16/* You should have received a copy of the GNU General Public License */
17/* along with this program; if not, write to the Free Software */
18/* Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA. */
19
20#include "config_xor.h"
21
22#include <sys/types.h>
23#include <sys/socket.h>
24#include <netdb.h>
25#include <netinet/in.h>
26#include <arpa/inet.h>
27#include <string.h>
28#include <ctype.h>
29
30#undef FIL__
31#define FIL__ _("sh_ipvx.c")
32
33#include "samhain.h"
34#define SH_NEED_GETHOSTBYXXX
35#include "sh_static.h"
36#include "sh_pthread.h"
37#include "sh_utils.h"
38#include "sh_ipvx.h"
39
40static int sh_ipvx_is_ipv4 (const char * addr)
41{
42 int j;
43 int count = 0;
44 int len = sl_strlen(addr);
45 unsigned int a, b, c, d;
46
47 if (len > 15) /* 3*4 + 3 dots */
48 return (1 == 0);
49
50 for (j = 0; j < len; ++j) {
51 if ( (addr[j] < '0' || addr[j] > '9') && addr[j] != '.')
52 return (1 == 0);
53 if (addr[j] == '.') ++count;
54 }
55
56 if (count != 3)
57 return (1 == 0);
58
59 if (sscanf(addr, "%3u.%3u.%3u.%3u", &a, &b, &c, &d) != 4)
60 return( 1 == 0 );
61
62 if ((a|b|c|d) > 255)
63 return( 1 == 0 );
64
65 return (1 == 1);
66}
67
68#if defined(USE_IPVX)
69static int sh_ipvx_is_ipv6 (const char * addr)
70{
71 int j, k = 0;
72 char c;
73 int len = sl_strlen(addr);
74
75 for (j = 0; j < len; ++j) {
76 c = addr[j];
77 if (( c < '0' || c > '9' ) &&
78 ( c < 'a' || c > 'f' ) &&
79 ( c < 'A' || c > 'F' ) &&
80 ( c != ':') && ( c != '.'))
81 return (1 == 0);
82 else if (c == ':')
83 ++k;
84 else if (c == '.' && k < 2)
85 return (1 == 0); /* ::ffff:ipv4 */
86 }
87 if (k < 2 || k > 7)
88 return (1 == 0);
89
90 return (1 == 1);
91}
92#endif
93
94
95int sh_ipvx_is_numeric (const char * addr)
96{
97#if defined(USE_IPVX)
98 if (!sh_ipvx_is_ipv4(addr))
99 return sh_ipvx_is_ipv6(addr);
100 else
101 return (1 == 1);
102#else
103 return sh_ipvx_is_ipv4(addr);
104#endif
105}
106
107int sh_ipvx_isany (struct sh_sockaddr * a)
108{
109#if defined(HOST_IS_CYGWIN)
110 /*
111 * Cygwin implementation gives 'missing braces around initializer'
112 * warning, thus replace it with correct version.
113 */
114#undef IN6ADDR_ANY_INIT
115#define IN6ADDR_ANY_INIT { { { 0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0 } } }
116#endif
117
118#if defined(USE_IPVX)
119 struct in6_addr anyaddr = IN6ADDR_ANY_INIT;
120#endif
121
122 switch (a->ss_family)
123 {
124 case AF_INET:
125 if ((a->sin).sin_addr.s_addr == INADDR_ANY)
126 return 1;
127 break;
128#if defined(USE_IPVX)
129 case AF_INET6:
130 if (0 == memcmp(&((a->sin6).sin6_addr.s6_addr), &anyaddr, 16))
131 return 1;
132 break;
133#endif
134 }
135 return 0;
136}
137
138int sh_ipvx_cmp (struct sh_sockaddr * a, struct sh_sockaddr * b)
139{
140 if (a->ss_family != b->ss_family)
141 return 1;
142
143 switch (a->ss_family)
144 {
145 case AF_INET:
146 return memcmp(&((a->sin).sin_addr.s_addr), &((b->sin).sin_addr.s_addr), 4);
147 break;
148#if defined(USE_IPVX)
149 case AF_INET6:
150 return memcmp(&((a->sin6).sin6_addr.s6_addr), &((b->sin6).sin6_addr.s6_addr), 16);
151 break;
152#endif
153 }
154 return 1;
155}
156
157int sh_ipvx_ntoa (char * name, size_t name_size, struct sh_sockaddr * ss)
158{
159#if defined(USE_IPVX)
160 int len = (ss->ss_family == AF_INET) ?
161 sizeof(struct sockaddr_in) :
162 sizeof(struct sockaddr_in6);
163
164 int ret = getnameinfo(sh_ipvx_sockaddr_cast(ss), len,
165 name, name_size, NULL, 0, NI_NUMERICHOST);
166
167 if (ret != 0 && name_size > 0)
168 {
169 name[name_size-1] = '\0';
170
171 if (!sh_ipvx_is_numeric(name))
172 {
173 if (name_size > 7) {
174 name[0] = '0'; name[1] = '.'; name[2] = '0'; name[3] = '.';
175 name[4] = '0'; name[5] = '.'; name[6] = '0'; name[7] = '\0';
176 } else {
177 name[0] = '\0';
178 }
179 }
180 }
181 return ret;
182#else
183 char * p = inet_ntoa((ss->sin).sin_addr);
184 sl_strlcpy(name, p, name_size);
185 return 0;
186#endif
187}
188
189struct sockaddr * sh_ipvx_sockaddr_cast (struct sh_sockaddr * ss)
190{
191#if defined(USE_IPVX)
192 if (ss->ss_family == AF_INET6)
193 return (struct sockaddr *) &(ss->sin6);
194#endif
195 return (struct sockaddr *) &(ss->sin);
196}
197
198char * sh_ipvx_print_sockaddr (struct sockaddr * sa, int sa_family)
199{
200 struct sh_sockaddr ss;
201 static char ipbuf[SH_IP_BUF];
202
203 sh_ipvx_save(&ss, sa_family, sa);
204 sh_ipvx_ntoa (ipbuf, sizeof(ipbuf), &ss);
205 return ipbuf;
206}
207
208void sh_ipvx_save(struct sh_sockaddr * ss, int sa_family, struct sockaddr * sa)
209{
210 /* memset(ss, '\0', sizeof(struct sh_sockaddr)); */
211
212 switch (sa_family)
213 {
214 case AF_INET:
215 ss->ss_family = AF_INET;
216 memcpy(&(ss->sin), (struct sockaddr_in*) sa, sizeof(struct sockaddr_in));
217 break;
218#if defined(USE_IPVX)
219 case AF_INET6:
220 ss->ss_family = AF_INET6;
221 memcpy(&(ss->sin6), (struct sockaddr_in6*) sa, sizeof(struct sockaddr_in6));
222 break;
223#endif
224 default:
225 break;
226 }
227 return;
228}
229
230int sh_ipvx_set_port(struct sh_sockaddr * ss, int port)
231{
232#if defined(USE_IPVX)
233
234 switch (ss->ss_family)
235 {
236 case AF_INET:
237 (ss->sin).sin_family = AF_INET;
238 (ss->sin).sin_port = htons (port);
239 break;
240 case AF_INET6:
241 (ss->sin6).sin6_family = AF_INET6;
242 (ss->sin6).sin6_port = htons (port);
243 break;
244 }
245 return 0;
246#else
247 (ss->sin).sin_family = AF_INET;
248 (ss->sin).sin_port = htons (port);
249 return 0;
250#endif
251}
252
253int sh_ipvx_get_port(struct sockaddr * sa, int sa_family)
254{
255 int port = 0;
256#if defined(USE_IPVX)
257
258 switch (sa_family)
259 {
260 case AF_INET:
261 port = ntohs(((struct sockaddr_in *)sa)->sin_port);
262 break;
263 case AF_INET6:
264 port = ntohs(((struct sockaddr_in6 *)sa)->sin6_port);
265 break;
266 }
267#else
268 (void) sa_family;
269 port = ntohs(((struct sockaddr_in *)sa)->sin_port);
270#endif
271 return port;
272}
273
274int sh_ipvx_aton (const char * name, struct sh_sockaddr * ss)
275{
276#if defined(USE_IPVX)
277 int ret;
278 struct addrinfo hints;
279 struct addrinfo *res;
280
281 memset(&hints, '\0', sizeof(hints));
282 hints.ai_family = PF_UNSPEC;
283 hints.ai_flags = AI_NUMERICHOST;
284 ret = getaddrinfo(name, NULL, &hints, &res);
285
286 if (ret)
287 return 0;
288
289 memset(ss, '\0', sizeof(struct sh_sockaddr));
290 switch(res->ai_family)
291 {
292 case AF_INET:
293 memcpy(&(ss->sin), res->ai_addr, sizeof(struct sockaddr_in));
294 ss->ss_family = AF_INET;
295 break;
296 case AF_INET6:
297 memcpy(&(ss->sin6), res->ai_addr, sizeof(struct sockaddr_in6));
298 ss->ss_family = AF_INET6;
299 break;
300 default:
301 return 0;
302 break;
303 }
304 freeaddrinfo(res);
305 return 1;
306#else
307 int ret = inet_aton(name, &((ss->sin).sin_addr));
308 ss->ss_family = AF_INET;
309 return ret;
310#endif
311}
312
313#if !defined(USE_IPVX)
314static const char * sh_ipvx_h_name (struct hostent * host_entry)
315{
316 char ** p;
317 if (strchr(host_entry->h_name, '.')) {
318 return host_entry->h_name;
319 } else {
320 for (p = host_entry->h_aliases; *p; ++p) {
321 if (strchr(*p, '.'))
322 return *p;
323 }
324 }
325 return host_entry->h_name;
326}
327#endif
328
329static char * sh_tolower (char * s)
330{
331 char * ret = s;
332 if (s)
333 {
334 for (; *s; ++s)
335 {
336 *s = tolower((unsigned char) *s);
337 }
338 }
339 return ret;
340}
341
342void * sh_dummy_341_out;
343
344char * sh_ipvx_canonical(const char * hostname, char * numeric, size_t nlen)
345{
346 volatile int flag = 0;
347 char *out = NULL;
348#if defined(USE_IPVX)
349 struct addrinfo hints;
350 struct addrinfo *res;
351 struct sockaddr *sa;
352 int salen;
353 int err;
354 struct sh_sockaddr ss;
355 const char * host;
356 char hostbuf[SH_BUFSIZE];
357
358 numeric[0] = '\0';
359
360 sh_dummy_341_out = (void *) &out;
361
362 if (sh_ipvx_is_numeric(hostname))
363 {
364 sh_ipvx_aton (hostname, &ss);
365 if (0 == getnameinfo(sh_ipvx_sockaddr_cast(&ss), SH_SS_LEN(ss),
366 hostbuf, sizeof(hostbuf), NULL, 0, NI_NAMEREQD))
367 host = hostbuf;
368 else
369 host = hostname;
370 }
371 else
372 {
373 host = hostname;
374 }
375
376 memset(&hints, 0, sizeof(hints));
377 hints.ai_family = PF_UNSPEC;
378#if defined(AI_CANONNAME)
379 hints.ai_flags = AI_CANONNAME;
380#endif
381
382 err = getaddrinfo(host, NULL, &hints, &res);
383 if (err == 0)
384 {
385 struct addrinfo * res_orig = res;
386
387#if defined(AI_CANONNAME)
388 if (res->ai_canonname && strlen(res->ai_canonname) > 0)
389 {
390 out = sh_util_strdup(res->ai_canonname);
391 sh_tolower (out);
392 if (strchr(out, '.'))
393 flag = 1;
394 }
395#endif
396
397 sa = res->ai_addr;
398 salen = res->ai_addrlen;
399 getnameinfo(sa, salen,
400 numeric, nlen, NULL, 0, NI_NUMERICHOST);
401
402 if (!flag)
403 out = SH_ALLOC(SH_PATHBUF);
404
405 while (res && !flag)
406 {
407 sa = res->ai_addr;
408 salen = res->ai_addrlen;
409
410 getnameinfo(sa, salen,
411 out, SH_PATHBUF, NULL, 0, 0);
412 sh_tolower (out);
413 if (strchr(out, '.'))
414 flag = 1;
415
416 res = res->ai_next;
417 }
418
419 freeaddrinfo(res_orig);
420 }
421#else
422 struct hostent *he;
423 struct sh_sockaddr ss;
424 volatile int isNum = 0;
425 struct sockaddr_in *sin;
426
427 numeric[0] = '\0';
428
429 sh_dummy_341_out = (void *) &out;
430
431 if (sh_ipvx_is_numeric(hostname))
432 {
433 sh_ipvx_aton (hostname, &ss);
434 isNum = 1;
435 }
436
437
438 SH_MUTEX_LOCK(mutex_resolv);
439
440 if (isNum == 0)
441 {
442 he = sh_gethostbyname(hostname);
443 }
444 else
445 {
446 sin = (struct sockaddr_in *) sh_ipvx_sockaddr_cast(&ss);
447 he = sh_gethostbyaddr(&(sin->sin_addr), sizeof(sin->sin_addr), AF_INET);
448 }
449
450 if (he != NULL)
451 {
452 out = sh_util_strdup(sh_ipvx_h_name(he));
453 sh_tolower (out);
454 sl_strlcpy (numeric,
455 inet_ntoa (*(struct in_addr *) he->h_addr),
456 nlen);
457 flag = 1;
458 }
459 SH_MUTEX_UNLOCK(mutex_resolv);
460#endif
461
462 if (flag)
463 return out;
464
465 if (out)
466 SH_FREE(out);
467 if (numeric[0] == '\0')
468 sl_strlcpy (numeric, _("0.0.0.0"), nlen);
469 return NULL;
470}
471
472char * sh_ipvx_addrtoname(struct sh_sockaddr * ss)
473{
474#if defined(USE_IPVX)
475 char namebuf[SH_BUFSIZE];
476
477 if (getnameinfo(sh_ipvx_sockaddr_cast(ss), SH_SSP_LEN(ss),
478 namebuf, sizeof(namebuf), NULL, 0, NI_NAMEREQD) != 0)
479 {
480 return NULL;
481 }
482 return sh_util_strdup(namebuf);
483#else
484 struct sockaddr_in *sin;
485 struct hostent *he;
486
487 sin = (struct sockaddr_in *) sh_ipvx_sockaddr_cast(ss);
488
489 he = sh_gethostbyaddr(&(sin->sin_addr), sizeof(sin->sin_addr), AF_INET);
490
491 if (he && he->h_name)
492 {
493 return sh_util_strdup(he->h_name);
494 }
495
496 return NULL;
497#endif
498}
499
500int sh_ipvx_reverse_check_ok (char * peer, int port, struct sh_sockaddr * ss)
501{
502#if defined(USE_IPVX)
503 struct addrinfo *res;
504 struct addrinfo hints;
505 char sport[32];
506 struct addrinfo *p;
507
508 sl_snprintf(sport, sizeof(sport), "%d", port);
509
510 memset(&hints, '\0', sizeof(hints));
511 hints.ai_socktype = SOCK_STREAM;
512 hints.ai_flags = AI_ADDRCONFIG;
513
514 if (getaddrinfo(peer, sport, &hints, &res) != 0)
515 {
516 return 0;
517 }
518
519 p = res;
520 while (p != NULL)
521 {
522 if (ss->ss_family == p->ai_family)
523 {
524 struct sh_sockaddr pp;
525
526 char dst1[SH_IP_BUF];
527 char dst2[SH_IP_BUF];
528
529 sh_ipvx_save(&pp, p->ai_family, p->ai_addr);
530
531 sh_ipvx_ntoa (dst1, sizeof(dst1), &pp);
532 sh_ipvx_ntoa (dst2, sizeof(dst2), ss);
533
534 if (0 == sl_strcmp(dst1, dst2))
535 {
536 return 1;
537 }
538 }
539 p = p->ai_next;
540 }
541 freeaddrinfo(res);
542#else
543 struct hostent * he;
544 char ** p;
545 struct sockaddr_in * sin = (struct sockaddr_in *) sh_ipvx_sockaddr_cast(ss);
546
547 (void) port;
548
549 he = sh_gethostbyname(peer);
550 if (he != NULL)
551 {
552 for (p = he->h_addr_list; *p; ++p)
553 {
554 if (0 == memcmp (*p, &(sin->sin_addr), sizeof(in_addr_t)) )
555 return 1;
556 }
557 }
558#endif
559 return 0;
560}
561
562#ifdef SH_CUTEST
563#include <stdlib.h>
564#include "CuTest.h"
565
566void Test_ipvx (CuTest *tc) {
567
568 int result;
569
570 result = sh_ipvx_is_ipv4("123456789");
571 CuAssertTrue(tc, result == FALSE);
572
573 result = sh_ipvx_is_ipv4("123456789123...");
574 CuAssertTrue(tc, result == FALSE);
575
576 result = sh_ipvx_is_ipv4("123.456.789.123");
577 CuAssertTrue(tc, result == FALSE);
578
579 result = sh_ipvx_is_ipv4("123.156.189.254");
580 CuAssertTrue(tc, result == TRUE);
581
582 result = sh_ipvx_is_ipv4("255.255.255.255");
583 CuAssertTrue(tc, result == TRUE);
584
585 result = sh_ipvx_is_ipv4("0.0.0.0");
586 CuAssertTrue(tc, result == TRUE);
587
588 result = sh_ipvx_is_ipv4("0022.156.189.254");
589 CuAssertTrue(tc, result == FALSE);
590
591 result = sh_ipvx_is_ipv4("999999999.1.1.2");
592 CuAssertTrue(tc, result == FALSE);
593
594}
595#endif
596
Note: See TracBrowser for help on using the repository browser.