//
// Copyright 2020 Staysail Systems, Inc. <info@staysail.tech>
//
// This software is supplied under the terms of the MIT License, a
// copy of which should be located in the distribution where this
// file was obtained (LICENSE.txt).  A copy of the license may also be
// found online at https://opensource.org/licenses/MIT.
//

#include <testutil.h>

#include <nng/protocol/pair0/pair.h>

#include <acutest.h>

static void
test_ws_url_path_filters(void)
{
	nng_socket s1;
	nng_socket s2;
	char       addr[NNG_MAXADDRLEN];

	TEST_NNG_PASS(nng_pair0_open(&s1));
	TEST_NNG_PASS(nng_pair0_open(&s2));

	testutil_scratch_addr("ws", sizeof(addr), addr);
	TEST_NNG_PASS(nng_listen(s1, addr, NULL, 0));

	// Now try we just remove the last character for now.
	// This will make the path different.
	addr[strlen(addr) - 1] = '\0';
	TEST_NNG_FAIL(nng_dial(s2, addr, NULL, 0), NNG_ECONNREFUSED);

	TEST_NNG_PASS(nng_close(s1));
	TEST_NNG_PASS(nng_close(s2));
}

static void
test_wild_card_port(void)
{
	nng_socket s1;
	nng_socket s2;
	nng_socket s3;
	nng_socket s4;
	nng_socket s5;
	nng_socket s6;

	nng_listener l1;
	nng_listener l2;
	nng_listener l3;
	int          port1;
	int          port2;
	int          port3;
	char         ws_url[128];
	TEST_NNG_PASS(nng_pair0_open(&s1));
	TEST_NNG_PASS(nng_pair0_open(&s2));
	TEST_NNG_PASS(nng_pair0_open(&s3));
	TEST_NNG_PASS(nng_pair0_open(&s4));
	TEST_NNG_PASS(nng_pair0_open(&s5));
	TEST_NNG_PASS(nng_pair0_open(&s6));
	TEST_NNG_PASS(nng_listen(s1, "ws://127.0.0.1:0/one", &l1, 0));
	TEST_NNG_PASS(
	    nng_listener_getopt_int(l1, NNG_OPT_TCP_BOUND_PORT, &port1));
	TEST_CHECK(port1 != 0);
	snprintf(ws_url, sizeof(ws_url), "ws4://127.0.0.1:%d/two", port1);
	TEST_NNG_PASS(nng_listen(s2, ws_url, &l2, 0));
	TEST_NNG_PASS(
	    nng_listener_getopt_int(l2, NNG_OPT_TCP_BOUND_PORT, &port2));
	TEST_CHECK(port1 != 0);
	TEST_CHECK(port1 == port2);
	// Now try a different wild card port.
	TEST_NNG_PASS(nng_listen(s3, "ws4://127.0.0.1:0/three", &l3, 0));
	TEST_NNG_PASS(
	    nng_listener_getopt_int(l3, NNG_OPT_TCP_BOUND_PORT, &port3));
	TEST_CHECK(port3 != 0);
	TEST_CHECK(port3 != port1);

	// Let's make sure can dial to each.
	snprintf(ws_url, sizeof(ws_url), "ws://127.0.0.1:%d/one", port1);
	TEST_NNG_PASS(nng_dial(s4, ws_url, NULL, 0));
	snprintf(ws_url, sizeof(ws_url), "ws://127.0.0.1:%d/two", port2);
	TEST_NNG_PASS(nng_dial(s6, ws_url, NULL, 0));
	snprintf(ws_url, sizeof(ws_url), "ws://127.0.0.1:%d/three", port3);
	TEST_NNG_PASS(nng_dial(s6, ws_url, NULL, 0));

	TEST_NNG_PASS(nng_close(s1));
	TEST_NNG_PASS(nng_close(s2));
	TEST_NNG_PASS(nng_close(s3));
	TEST_NNG_PASS(nng_close(s4));
	TEST_NNG_PASS(nng_close(s5));
	TEST_NNG_PASS(nng_close(s6));
}

static void
test_wild_card_host(void)
{
	nng_socket s1;
	nng_socket s2;
	char       addr[NNG_MAXADDRLEN];
	uint16_t   port;

	TEST_NNG_PASS(nng_pair0_open(&s1));
	TEST_NNG_PASS(nng_pair0_open(&s2));

	port = testutil_next_port();

	// we use ws4 to ensure 127.0.0.1 binding
	snprintf(addr, sizeof(addr), "ws4://*:%u/test", port);
	TEST_NNG_PASS(nng_listen(s1, addr, NULL, 0));
	nng_msleep(100);

	snprintf(addr, sizeof(addr), "ws://127.0.0.1:%u/test", port);
	TEST_NNG_PASS(nng_dial(s2, addr, NULL, 0));

	TEST_NNG_PASS(nng_close(s1));
	TEST_NNG_PASS(nng_close(s2));
}

static void
test_empty_host(void)
{
	nng_socket s1;
	nng_socket s2;
	char       addr[NNG_MAXADDRLEN];
	uint16_t   port;

	TEST_NNG_PASS(nng_pair0_open(&s1));
	TEST_NNG_PASS(nng_pair0_open(&s2));

	port = testutil_next_port();

	// we use ws4 to ensure 127.0.0.1 binding
	snprintf(addr, sizeof(addr), "ws4://:%u/test", port);
	TEST_NNG_PASS(nng_listen(s1, addr, NULL, 0));
	nng_msleep(100);

	snprintf(addr, sizeof(addr), "ws://127.0.0.1:%u/test", port);
	TEST_NNG_PASS(nng_dial(s2, addr, NULL, 0));

	TEST_NNG_PASS(nng_close(s1));
	TEST_NNG_PASS(nng_close(s2));
}

TEST_LIST = {
	{ "ws url path filters", test_ws_url_path_filters },
	{ "ws wild card port", test_wild_card_port },
	{ "ws wild card host", test_wild_card_host },
	{ "ws empty host", test_empty_host },
	{ NULL, NULL },
};