/*
 * Copyright (c) 2017 Intel Corporation
 *
 * SPDX-License-Identifier: Apache-2.0
 */

#include "tcp_client.h"
#include "config.h"

#include <net/net_core.h>
#include <net/net_if.h>
#include <net/nbuf.h>

#include <misc/printk.h>

static
int set_addr(struct sockaddr *sock_addr, const char *addr, uint16_t server_port)
{
	void *ptr = NULL;
	int rc;

#ifdef CONFIG_NET_IPV6
	net_sin6(sock_addr)->sin6_port = htons(server_port);
	sock_addr->family = AF_INET6;
	ptr = &(net_sin6(sock_addr)->sin6_addr);
	rc = net_addr_pton(AF_INET6, addr, ptr);
#else
	net_sin(sock_addr)->sin_port = htons(server_port);
	sock_addr->family = AF_INET;
	ptr = &(net_sin(sock_addr)->sin_addr);
	rc = net_addr_pton(AF_INET, addr, ptr);
#endif
	if (rc) {
		printk("Invalid IP address: %s\n", addr);
	}

	return rc;
}

static
int if_addr_add(struct sockaddr *local_sock)
{
	void *p = NULL;

#ifdef CONFIG_NET_IPV6
	p = net_if_ipv6_addr_add(net_if_get_default(),
				 &net_sin6(local_sock)->sin6_addr,
				 NET_ADDR_MANUAL, 0);
#else
	p = net_if_ipv4_addr_add(net_if_get_default(),
				 &net_sin(local_sock)->sin_addr,
				 NET_ADDR_MANUAL, 0);
#endif
	if (p) {
		return 0;
	}

	return -EINVAL;
}

int tcp_set_local_addr(struct tcp_client_ctx *ctx, const char *local_addr)
{
	int  rc;

	rc = set_addr(&ctx->local_sock, local_addr, 0);
	if (rc) {
		printk("set_addr (local) error\n");
		goto lb_exit;
	}

	rc = if_addr_add(&ctx->local_sock);
	if (rc) {
		printk("if_addr_add error\n");
	}

lb_exit:
	return rc;
}

static
void recv_cb(struct net_context *net_ctx, struct net_buf *rx, int status,
	     void *data)
{
	struct tcp_client_ctx *ctx = (struct tcp_client_ctx *)data;

	ARG_UNUSED(net_ctx);

	if (status) {
		return;
	}

	if (rx == NULL || net_nbuf_appdatalen(rx) == 0) {
		goto lb_exit;
	}

	/* receive_cb must take ownership of the rx buffer */
	if (ctx->receive_cb) {
		ctx->receive_cb(ctx, rx);
		return;
	}

lb_exit:
	net_buf_unref(rx);
}

int tcp_connect(struct tcp_client_ctx *ctx, const char *server_addr,
		uint16_t server_port)
{
#if CONFIG_NET_IPV6
	socklen_t addr_len = sizeof(struct sockaddr_in6);
	sa_family_t family = AF_INET6;
#else
	socklen_t addr_len = sizeof(struct sockaddr_in);
	sa_family_t family = AF_INET;
#endif
	struct sockaddr server_sock;
	int rc;

	rc = net_context_get(family, SOCK_STREAM, IPPROTO_TCP, &ctx->net_ctx);
	if (rc) {
		printk("net_context_get error\n");
		return rc;
	}

	rc = net_context_bind(ctx->net_ctx, &ctx->local_sock, addr_len);
	if (rc) {
		printk("net_context_bind error\n");
		goto lb_exit;
	}

	rc = set_addr(&server_sock, server_addr, server_port);
	if (rc) {
		printk("set_addr (server) error\n");
		goto lb_exit;
	}

	rc = net_context_connect(ctx->net_ctx, &server_sock, addr_len, NULL,
				 ctx->timeout, NULL);
	if (rc) {
		printk("net_context_connect error\n");
		goto lb_exit;
	}

	(void)net_context_recv(ctx->net_ctx, recv_cb, K_NO_WAIT, ctx);

	return 0;

lb_exit:
	net_context_put(ctx->net_ctx);

	return rc;
}

int tcp_disconnect(struct tcp_client_ctx *ctx)
{
	if (ctx->net_ctx) {
		net_context_put(ctx->net_ctx);
		ctx->net_ctx = NULL;
	}

	return 0;
}