source: trunk/src/sh_ipvx.c @ 509

Last change on this file since 509 was 509, checked in by katerina, 5 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.