/*
 * Copyright (c) 2014 embedded brains GmbH.  All rights reserved.
 *
 *  embedded brains GmbH
 *  Dornierstr. 4
 *  82178 Puchheim
 *  Germany
 *  <rtems@embedded-brains.de>
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in the
 *    documentation and/or other materials provided with the distribution.
 *
 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
 * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
 * SUCH DAMAGE.
 */

#include <sys/select.h>
#include <sys/socket.h>

#include <netinet/in.h>

#include <assert.h>
#include <errno.h>
#include <netdb.h>
#include <stdbool.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>

#include <rtems.h>
#include <rtems/mdns.h>

#define TEST_NAME "LIBBSD FOOBAR CLIENT"

static CacheEntity rr_cache[64];

static const char * const qc_results[] = {
	"rmv",
	"add",
	"addnocache",
	"forceresponse",
	"dnssec",
	"nodnssec",
	"suppressed"
};

typedef struct service_info_query {
	struct service_info_query *next;
	ServiceInfoQuery query;
	ServiceInfo info;
} service_info_query;

static service_info_query *service_info_query_list;

static mDNSBool
eval_service_name(const domainname *fqdn, domainlabel *name)
{
	domainname type;
	domainname domain;
	mDNSBool ok = DeconstructServiceName(fqdn, name, &type, &domain);

	if (ok) {
		char name_str[MAX_DOMAIN_LABEL + 1];
		char type_str[MAX_ESCAPED_DOMAIN_NAME];
		char domain_str[MAX_ESCAPED_DOMAIN_NAME];

		ConvertDomainLabelToCString_unescaped(name, name_str);
		ConvertDomainNameToCString(&type, type_str);
		ConvertDomainNameToCString(&domain, domain_str);

		printf("name = '%s', type = '%s', domain = '%s'",
		    name_str, type_str, domain_str);
	} else {
		printf("?");
	}

	return ok;
}

static void
foobar_action(const struct sockaddr_storage *addr)
{
	static const char foobar[] = "FooBar!";

	int sd;
	int rv;

	sd = socket(addr->ss_family, SOCK_STREAM, 0);
	assert(sd >= 0);

	rv = connect(sd, (struct sockaddr *) addr, addr->ss_len);
	if (rv == 0) {
		char buf[sizeof(foobar)];
		ssize_t n;

		n = read(sd, &buf[0], sizeof(buf));
		if (n == (ssize_t) sizeof(buf)) {
			assert(memcmp(&buf[0], &foobar[0], sizeof(buf)) == 0);
			printf("foobar client: read: %s\n", &buf[0]);
		} else if (n >= 0) {
			printf("foobar client: partial read: %zi\n", n);
		} else {
			printf("foobar client: read error: %s\n",
			    strerror(errno));
		}
	} else {
		printf("foobar client: connect error: %s\n", strerror(errno));
	}

	rv = close(sd);
	assert(rv == 0);
}

static void
foobar_service_info(mDNS *const m, ServiceInfoQuery *query)
{
	const ServiceInfo *i = query->info;
	struct sockaddr_storage addr;
	char host[NI_MAXHOST];
	domainlabel name;
	int rv;

	memset(&addr, 0, sizeof(addr));

	if (i->ip.type == mDNSAddrType_IPv4) {
		struct sockaddr_in *s = (struct sockaddr_in *) &addr;

		s->sin_len = sizeof(*s);
		s->sin_family = AF_INET;
		s->sin_port = i->port.NotAnInteger;
		s->sin_addr.s_addr = i->ip.ip.v4.NotAnInteger;
	} else {
		struct sockaddr_in6 *sin6 = (struct sockaddr_in6 *) &addr;

		sin6->sin6_len = sizeof(*sin6);
		sin6->sin6_family = AF_INET6;
		sin6->sin6_port = i->port.NotAnInteger;
		sin6->sin6_addr = *(struct in6_addr *) &i->ip.ip.v6;
	}

	rv = getnameinfo((struct sockaddr *) &addr, addr.ss_len, host,
	    sizeof(host), NULL, 0, NI_NUMERICHOST);
	assert(rv == 0);

	printf("foobar client: service info: ");
	eval_service_name(&i->name, &name);
	printf(", addr = %s, port = %i\n", &host[0], mDNSVal16(i->port));

	foobar_action(&addr);
}

static void
foobar_manage_service(mDNS *m, const ResourceRecord *answer,
    const domainlabel *name, bool add)
{
	if (add) {
		mStatus status;
		service_info_query *siq =
		    calloc(1, sizeof(*siq));

		assert(siq != NULL);

		siq->next = service_info_query_list;
		service_info_query_list = siq;

		siq->info.name = answer->rdata->u.name;
		siq->info.InterfaceID = answer->InterfaceID;

		status = mDNS_StartResolveService(m, &siq->query, &siq->info,
		    foobar_service_info, NULL);
		assert(status == mStatus_NoError);
	} else {
		service_info_query *cur = service_info_query_list;
		service_info_query *prev = cur;

		while (cur != NULL && !SameDomainLabel(&name->c[0],
		    &cur->info.name.c[0])) {
			prev = cur;
			cur = cur->next;
		}

		if (cur != NULL) {
			if (prev != cur) {
				prev->next = cur->next;
			} else {
				service_info_query_list = cur->next;
			}

			mDNS_StopResolveService(m, &cur->query);
			free(cur);
		}
	}
}

static void
foobar_browse(mDNS *m, DNSQuestion *question,
    const ResourceRecord *answer, QC_result res)
{
	(void) question;

	printf("foobar client: %s: ", qc_results[res]);

	if (answer->rrtype == kDNSType_PTR &&
	    (res == QC_rmv || res == QC_add)) {
		domainlabel name;
		mDNSBool ok = eval_service_name(&answer->rdata->u.name, &name);

		if (ok) {
			foobar_manage_service(m, answer, &name, res == QC_add);
		}

		printf("\n");
	}
}

static void
foobar_register(DNSQuestion *question)
{
	mStatus status;
	domainname type;
	domainname domain;

	MakeDomainNameFromDNSNameString(&type, "_foobar._tcp");
	MakeDomainNameFromDNSNameString(&domain, "local.");

	status = mDNS_StartBrowse(rtems_mdns_get_instance(), question, &type,
	    &domain, mDNSNULL, mDNSInterface_Any, 0, mDNSfalse, mDNSfalse,
	    foobar_browse, NULL);
	assert(status == mStatus_NoError);
}

static void
test_main(void)
{
	const char name[] = "foobarclient";
	int rv;
	DNSQuestion question;
	rtems_status_code sc;

	rv = sethostname(&name[0], sizeof(name) - 1);
	assert(rv == 0);

	sc = rtems_mdns_initialize(254, &rr_cache[0], RTEMS_ARRAY_SIZE(rr_cache));
	assert(sc == RTEMS_SUCCESSFUL);

	foobar_register(&question);

	rtems_task_delete(RTEMS_SELF);
	assert(0);
}

#define DEFAULT_NETWORK_DHCPCD_ENABLE
#define DEFAULT_NETWORK_DHCPCD_NO_DHCP_DISCOVERY
#define DEFAULT_NETWORK_SHELL

#include <rtems/bsd/test/default-network-init.h>