/*
 *	BIRD Library -- IP address functions Tests
 *
 *	(c) 2015 CZ.NIC z.s.p.o.
 *
 *	Can be freely distributed and used under the terms of the GNU GPL.
 */

#include "test/birdtest.h"
#include "test/birdtest_support.h"	/* REMOVE ME */

#include "lib/ip.h"

#define IP4_MAX_LEN	16

static u32
build_ip4(u8 a, u8 b, u8 c, u8 d)
{
  return ((u32)a << 24) + ((u32)b << 16) + ((u32)c << 8) + (u32)d;
}

static u32
ip4_pton_(const char *s)
{
  ip4_addr ip;
  ip4_pton(s,&ip);
  return ip4_to_u32(ip);
}

static int
t_ip4_pton(void)
{
  struct in_out {
    char in[IP4_MAX_LEN];
    u32 out;
  } in_out[] = {
      {
	  .in  = "192.168.1.128",
	  .out = build_ip4(192, 168, 1, 128),
      },
      {
	  .in  = "255.255.255.255",
	  .out = build_ip4(255, 255, 255, 255),
      },
      {
	  .in  = "0.0.0.0",
	  .out = build_ip4(0, 0, 0, 0),
      },
  };

  bt_assert_out_fn_in(ip4_pton_, in_out, "'%s'", NULL);

  return bt_test_suite_success;
}

static void
ip6_pton_(const char *s, u32 (*addr)[4])
{
  static ip6_addr ip;
  ip6_pton(s, &ip);
  int i;
  for (i = 0; i < 4; i++)
    (*addr)[i] = ip.addr[i];
}

static int
t_ip6_pton(void)
{
  struct in_out {
    char *in;
    u32 out[4];
  } in_out[] = {
      {
	  .in  = "2001:0db8:0000:0000:0000:0000:1428:57ab",
	  .out = {0x20010DB8, 0x00000000, 0x00000000, 0x142857AB},
      },
      {
	  .in  = "2001:0db8:0000:0000:0000::1428:57ab",
	  .out = {0x20010DB8, 0x00000000, 0x00000000, 0x142857AB},
      },
      {
	  .in  = "2001:0db8::1428:57ab",
	  .out = {0x20010DB8, 0x00000000, 0x00000000, 0x142857AB},
      },
      {
	  .in  = "2001:db8::1428:57ab",
	  .out = {0x20010DB8, 0x00000000, 0x00000000, 0x142857AB},
      },
      {
	  .in  = "::1",
	  .out = {0x00000000, 0x00000000, 0x00000000, 0x00000001},
      },
      {
	  .in  = "::",
	  .out = {0x00000000, 0x00000000, 0x00000000, 0x00000000},
      },
      {
	  .in  = "2605:2700:0:3::4713:93e3",
	  .out = {0x26052700, 0x00000003, 0x00000000, 0x471393E3},
      },
  };

  bt_assert_fn_in_out(ip6_pton_, in_out, "'%s'", NULL);

  return bt_test_suite_success;
}

char *
ip4_ntop_(ip4_addr a, char (*b)[IP4_MAX_LEN])
{
  return ip4_ntop(a, (char *) b);
}

static int
t_ip4_ntop(void)
{
  struct in_out {
    ip4_addr in;
    char out[IP4_MAX_LEN];
  } in_out[] = {
      {
	  .in  = ip4_from_u32(build_ip4(192, 168, 1, 128)),
	  .out = "192.168.1.128",
      },
      {
	  .in  = ip4_from_u32(build_ip4(255, 255, 255, 255)),
	  .out = "255.255.255.255",
      },
      {
	  .in  = ip4_from_u32(build_ip4(0, 0, 0, 1)),
	  .out = "0.0.0.1",
      },

  };

  bt_assert_fn_in_out(ip4_ntop_, in_out, NULL, "'%s'");

  return bt_test_suite_success;
}

char *
ip6_ntop_(ip6_addr a, char (*b)[INET6_ADDRSTRLEN])
{
  return ip6_ntop(a, (char *) b);
}

static int
t_ip6_ntop(void)
{
  struct in_out {
    ip6_addr in;
    char out[INET6_ADDRSTRLEN];
  } in_out[] = {
      {
	  .in  = { .addr = {0x20010DB8, 0x00000000, 0x00000000, 0x142857AB}},
	  .out = "2001:db8::1428:57ab",
      },
      {
	  .in  = { .addr = {0x26052700, 0x00000003, 0x00000000, 0x471393E3}},
	  .out = "2605:2700:0:3::4713:93e3",
      },
  };

  bt_assert_fn_in_out(ip6_ntop_, in_out, NULL, "'%s'");

  return bt_test_suite_success;
}

int
main(int argc, char *argv[])
{
  bt_init(argc, argv);

  bt_test_suite(t_ip4_pton, "Converting IPv4 string to ip4_addr struct");
  bt_test_suite(t_ip6_pton, "Converting IPv6 string to ip6_addr struct");
  bt_test_suite(t_ip4_ntop, "Converting ip4_addr struct to IPv4 string");
  bt_test_suite(t_ip6_ntop, "Converting ip6_addr struct to IPv6 string");

  return bt_end();
}