pcap_tests.c File Reference

#include "pcap.c"
#include <cgreen/cgreen.h>
#include <cgreen/mocks.h>
#include <gvm/base/hosts.h>

Include dependency graph for pcap_tests.c:

Go to the source code of this file.

Functions
	Describe (pcap)
	BeforeEach (pcap)
	AfterEach (pcap)
	__attribute__ ((weak))
int	__wrap_setsockopt (__attribute__((unused)) int sockfd, __attribute__((unused)) int level, __attribute__((unused)) int optname, __attribute__((unused)) const void *optval, __attribute__((unused)) socklen_t optlen)
	Ensure (pcap, routethrough_dst_is_localhost)
	Ensure (pcap, routethrough_dst_is_not_localhost)
	Ensure (pcap, routethrough_no_src_dst_given)
	Ensure (pcap, routethrough_src_globalsource_set)
	Ensure (pcap, routethrough_src_globalsource_not_set)
	Ensure (pcap, v6_islocalhost)
	Ensure (pcap, islocalhost)
static void	apply_ipv6_mask (struct in6_addr *network, struct in6_addr *dest, struct in6_addr *mask)
	Apply mask to dest addr.
	Ensure (pcap, ipv6_prefix_to_mask)
static TestSuite *	openvas_routethrough ()
int	main (int argc, char **argv)

Variables
bool	g_setsockopt_use_real = true

Function Documentation

__attribute__()

__attribute__

(

(weak)

)

Definition at line 21 of file pcap_tests.c.

{
  if (g_socket_use_real)
    return __real_socket (domain, type, protocol);
 
  return (int) mock (domain, type, protocol);
}

References __attribute__, and protocol.

__wrap_setsockopt()

int __wrap_setsockopt	(	__attribute__((unused)) int	sockfd,
		__attribute__((unused)) int	level,
		__attribute__((unused)) int	optname,
		__attribute__((unused)) const void *	optval,
		__attribute__((unused)) socklen_t	optlen )

Definition at line 58 of file pcap_tests.c.

{
  if (g_setsockopt_use_real)
    return __real_setsockopt (sockfd, level, optname, optval, optlen);
 
  return (int) mock (sockfd, level, optname, optval, optlen);
}

References __attribute__, and g_setsockopt_use_real.

AfterEach()

AfterEach

(

pcap

)

Definition at line 17 of file pcap_tests.c.

{
}

apply_ipv6_mask()

void apply_ipv6_mask ( struct in6_addr * network, struct in6_addr * dest, struct in6_addr * mask )

static

Apply mask to dest addr.

Parameters

[out]	network	Masked dest addr.
[in]	dest	Destination addr.
[in]	mask	Mask to apply.

Definition at line 248 of file pcap_tests.c.

{
  for (int i = 0; i < (int) sizeof (struct in6_addr); i++)
    network->s6_addr[i] = dest->s6_addr[i] & mask->s6_addr[i];
}

Referenced by Ensure().

Here is the caller graph for this function:

BeforeEach()

BeforeEach

(

pcap

)

Definition at line 13 of file pcap_tests.c.

{
  cgreen_mocks_are (loose_mocks);
}

Describe()

Describe

(

pcap

)

Ensure() [1/8]

Ensure	(	pcap	,
		ipv6_prefix_to_mask	)

Definition at line 255 of file pcap_tests.c.

{
  struct in6_addr dest;
  struct in6_addr result;
  struct in6_addr mask;
  struct in6_addr network;
  const uint8_t byte_options[9] = {0xFF, 0x00, 0x80, 0xC0, 0xE0,
                                   0xF0, 0xF8, 0xFC, 0xFE};
 
  // create dst addr
  const uint8_t addr_in[16] = {0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF,
                               0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF};
  memcpy (dest.s6_addr, addr_in, sizeof addr_in);
  // create expected result addr
  const uint8_t result_in[16] = {0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF,
                                 0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF,
                                 0xFF, 0xFF, 0xFF, 0xFF};
  memcpy (result.s6_addr, result_in, sizeof result_in);
 
  // check every possible bit mask
  for (int i = 128; i > 0; i--)
    {
      ipv6_prefix_to_mask (i, &mask);
      apply_ipv6_mask (&network, &dest, &mask);
      int byte_to_modify = i / 8;
      if (byte_to_modify != 16)
        result.s6_addr[byte_to_modify] = byte_options[(i % 8) + 1];
 
      assert_that (IN6_ARE_ADDR_EQUAL (&network, &result));
    }
}

References apply_ipv6_mask(), and ipv6_prefix_to_mask().

Here is the call graph for this function:

Ensure() [2/8]

Ensure	(	pcap	,
		islocalhost	)

Definition at line 219 of file pcap_tests.c.

{
  /* IPv4 */
  struct in_addr addr;
 
  inet_pton (AF_INET, "127.0.0.1", &(addr.s_addr));
  assert_that (islocalhost (&addr), is_true);
  inet_pton (AF_INET, "0.0.0.0", &(addr.s_addr));
  assert_that (islocalhost (&addr), is_true);
  inet_pton (AF_INET, "127.100.5.99", &(addr.s_addr));
  assert_that (islocalhost (&addr), is_true);
 
  /* dependent on local environment */
  // // inet_pton (AF_INET, <some local interface address>, &(addr));
  // // assert_that (islocalhost (&addr), is_true);
 
  /* example.com */
  inet_pton (AF_INET, "93.184.216.34", &(addr.s_addr));
  assert_that (islocalhost (&addr), is_false);
}

References islocalhost().

Here is the call graph for this function:

Ensure() [3/8]

Ensure	(	pcap	,
		routethrough_dst_is_localhost	)

Definition at line 71 of file pcap_tests.c.

{
  /* setup */
  g_socket_use_real = false;
  gchar *interface = NULL;
  gchar *ipv4_str = "127.0.0.1";
  gvm_host_t *gvm_host = NULL;
  struct in6_addr dst6;
  struct in6_addr *dst6_p = &dst6;
  struct in_addr dst4;
  struct in_addr *dst4_p = &dst4;
  assert_that ((gvm_host = gvm_host_from_str (ipv4_str)), is_not_null);
  assert_that (gvm_host_get_addr6 ((gvm_host_t *) gvm_host, dst6_p),
               is_equal_to (0));
  assert_that (dst6_p, is_not_null);
  dst4.s_addr = dst6_p->s6_addr32[3];
 
  interface = routethrough (dst4_p, NULL);
  (void) interface;
 
  /* dependent on local environment */
  // assert_that ((interface = routethrough (dst4_p, NULL)), is_not_null);
  // assert_that (interface, is_equal_to_string ("lo"));
  g_socket_use_real = true;
 
  gvm_host_free (gvm_host);
}

References routethrough().

Here is the call graph for this function:

Ensure() [4/8]

Ensure	(	pcap	,
		routethrough_dst_is_not_localhost	)

Definition at line 101 of file pcap_tests.c.

{
  g_socket_use_real = false;
  /* setup */
  gchar *interface = NULL;
  gchar *ipv4_str = "93.184.216.34"; /* example.com */
  gvm_host_t *gvm_host = NULL;
  struct in6_addr dst6;
  struct in6_addr *dst6_p = &dst6;
  struct in_addr dst4;
  struct in_addr *dst4_p = &dst4;
  assert_that ((gvm_host = gvm_host_from_str (ipv4_str)), is_not_null);
  assert_that (gvm_host_get_addr6 ((gvm_host_t *) gvm_host, dst6_p),
               is_equal_to (0));
  assert_that (dst6_p, is_not_null);
  dst4.s_addr = dst6_p->s6_addr32[3];
 
  interface = routethrough (dst4_p, NULL);
  assert_that (interface, is_not_equal_to_string ("lo"));
  g_socket_use_real = true;
 
  gvm_host_free (gvm_host);
}

References routethrough().

Here is the call graph for this function:

Ensure() [5/8]

Ensure	(	pcap	,
		routethrough_no_src_dst_given	)

Definition at line 126 of file pcap_tests.c.

{
  gchar *interface = NULL;
  assert_that ((interface = routethrough (NULL, NULL)), is_null);
}

References routethrough().

Here is the call graph for this function:

Ensure() [6/8]

Ensure	(	pcap	,
		routethrough_src_globalsource_not_set	)

Definition at line 161 of file pcap_tests.c.

{
  g_socket_use_real = false;
 
  struct in_addr src = {.s_addr = 0}; /* ip src */
  gchar *interface = NULL;
  struct in_addr dst;
  inet_pton (AF_INET, "127.0.0.1", &(dst.s_addr));
 
  /* global source address not set */
  gvm_source_iface_init (NULL);
  /* dst not given */
  assert_that ((interface = routethrough (NULL, &src)), is_null);
  assert_that ((src.s_addr == INADDR_ANY));
  /* dst localhost given */
  src.s_addr = 0;
 
  interface = routethrough (&dst, &src);
  /* dependent on local environment */
  // assert_that ((interface = routethrough (&dst, &src)), is_not_null);
  // assert_that (interface, is_equal_to_string ("lo"));
  assert_that ((src.s_addr != INADDR_ANY));
  g_socket_use_real = true;
}

References routethrough().

Here is the call graph for this function:

Ensure() [7/8]

Ensure	(	pcap	,
		routethrough_src_globalsource_set	)

Definition at line 133 of file pcap_tests.c.

{
  /* setup */
  g_socket_use_real = false;
  cgreen_mocks_are (learning_mocks);
 
  struct in_addr src = {.s_addr = 0}; /* ip src */
  gchar *interface = NULL;
  struct in_addr dst;
  inet_pton (AF_INET, "93.184.216.34", &(dst.s_addr));
 
  /* global source address set */
  gvm_source_iface_init ("lo"); // lo is set but not really used after being set
  /* dst not given */
  assert_that ((interface = routethrough (NULL, &src)), is_null);
  assert_that ((src.s_addr == INADDR_ANY));
  /* dst localhost given */
  src.s_addr = 0;
 
  interface = routethrough (&dst, &src);
  /* dependent on local environment */
  // assert_that ((interface = routethrough (&dst, &src)), is_not_null);
  assert_that (interface, is_not_equal_to_string ("lo"));
  assert_that ((src.s_addr != INADDR_ANY));
  g_socket_use_real = true;
}

References routethrough().

Here is the call graph for this function:

Ensure() [8/8]

Ensure	(	pcap	,
		v6_islocalhost	)

Definition at line 186 of file pcap_tests.c.

{
  /* IPv4 */
  struct in_addr addr;
  struct sockaddr_in sin;
  memset (&sin, 0, sizeof (struct sockaddr_in));
  sin.sin_family = AF_INET;
 
  /* example.com */
  inet_pton (AF_INET, "93.184.216.34", &(addr.s_addr));
 
  /* IPv6 */
  struct in6_addr addr_6;
 
  inet_pton (AF_INET6, "::FFFF:127.0.0.1", &(addr_6));
  assert_that (v6_islocalhost (&addr_6), is_true);
  inet_pton (AF_INET6, "::FFFF:0.0.0.0", &(addr_6));
  assert_that (v6_islocalhost (&addr_6), is_true);
  inet_pton (AF_INET6, "::FFFF:127.100.5.99", &(addr_6));
  assert_that (v6_islocalhost (&addr_6), is_true);
  /* loopback address */
  inet_pton (AF_INET6, "0:0:0:0:0:0:0:1", &(addr_6));
  assert_that (v6_islocalhost (&addr_6), is_true);
 
  /* dependent on local environment */
  // inet_pton (AF_INET6, <some local interface address>, &(addr_6));
  // assert_that (v6_islocalhost (&addr_6), is_true);
 
  /* example.com */
  inet_pton (AF_INET6, "2606:2800:220:1:248:1893:25c8:1946", &(addr_6));
  assert_that (v6_islocalhost (&addr_6), is_false);
}

References v6_islocalhost().

Here is the call graph for this function:

main()

int main	(	int	argc,
		char **	argv )

Definition at line 304 of file pcap_tests.c.

{
  int ret;
  TestSuite *suite;
 
  suite = create_test_suite ();
  add_suite (suite, openvas_routethrough ());
 
  if (argc > 1)
    ret = run_single_test (suite, argv[1], create_text_reporter ());
  else
    ret = run_test_suite (suite, create_text_reporter ());
 
  destroy_test_suite (suite);
 
  return ret;
}

References openvas_routethrough().

Here is the call graph for this function:

openvas_routethrough()

TestSuite * openvas_routethrough ( )

static

Definition at line 288 of file pcap_tests.c.

{
  TestSuite *suite = create_test_suite ();
  add_test_with_context (suite, pcap, routethrough_dst_is_localhost);
  add_test_with_context (suite, pcap, routethrough_dst_is_not_localhost);
  add_test_with_context (suite, pcap, routethrough_no_src_dst_given);
  add_test_with_context (suite, pcap, routethrough_src_globalsource_set);
  add_test_with_context (suite, pcap, routethrough_src_globalsource_not_set);
  add_test_with_context (suite, pcap, v6_islocalhost);
  add_test_with_context (suite, pcap, islocalhost);
  add_test_with_context (suite, pcap, ipv6_prefix_to_mask);
 
  return suite;
}

References ipv6_prefix_to_mask(), islocalhost(), and v6_islocalhost().

Referenced by main().

Here is the call graph for this function:

Here is the caller graph for this function:

Variable Documentation

g_setsockopt_use_real

bool g_setsockopt_use_real = true

Definition at line 50 of file pcap_tests.c.

Referenced by __wrap_setsockopt().