// SPDX-License-Identifier: GPL-2.0-or-later
/*
 * Copyright (c) 2022 Huawei Device Co., Ltd.
 *
 * Description: Demo example of NewIP tcp server.
 *
 * Author: Yang Yanjun <yangyanjun@huawei.com>
 *
 * Data: 2022-09-06
 */
#include <stdio.h>
#include <unistd.h>
#include <string.h>
#include <arpa/inet.h>
#include <sys/socket.h>

#define __USE_GNU
#include <sched.h>
#include <pthread.h>

#include "nip_uapi.h"
#include "nip_lib.h"
#include "newip_route.h"

static void *recv_send(void *args)
{
	int cfd;
	ssize_t ret;
	char buf[BUFLEN] = {0};

	memcpy(&cfd, args, sizeof(int));
	for (int i = 0; i < PKTCNT; i++) {
		ssize_t recv_num = recv(cfd, buf, PKTLEN, MSG_WAITALL);

		if (recv_num < 0) {
			perror("recv");
			goto END;
		} else if (recv_num == 0) { /* no data */
			;
		} else {
			printf("Received -- %s --:%zd\n", buf, recv_num);
			ret = send(cfd, buf, recv_num, 0);
			if (ret < 0) {
				perror("send");
				goto END;
			}
			printf("Sending  -- %s --:%zd\n", buf, recv_num);
		}
	}
END:	close(cfd);
	return NULL;
}

int main(int argc, char **argv)
{
	pthread_t th;
	int fd, cfd, addr_len;
	struct sockaddr_nin si_local;
	struct sockaddr_nin si_remote;

	fd = socket(AF_NINET, SOCK_STREAM, IPPROTO_TCP);
	if (fd < 0) {
		perror("socket");
		return -1;
	}

	memset((char *)&si_local, 0, sizeof(si_local));
	si_local.sin_family = AF_NINET;
	si_local.sin_port = htons(TCP_SERVER_PORT);
	// 2-byte address of the server: 0xDE00
	si_local.sin_addr.NIP_ADDR_FIELD8[INDEX_0] = 0xDE;
	si_local.sin_addr.NIP_ADDR_FIELD8[INDEX_1] = 0x00;
	si_local.sin_addr.bitlen = NIP_ADDR_BIT_LEN_16; // 2-byte: 16bit

	if (bind(fd, (const struct sockaddr *)&si_local, sizeof(si_local)) < 0) {
		perror("bind");
		goto END;
	}
	printf("bind success, addr=0x%02x%02x, port=%d\n",
	       si_local.sin_addr.NIP_ADDR_FIELD8[INDEX_0],
	       si_local.sin_addr.NIP_ADDR_FIELD8[INDEX_1], TCP_SERVER_PORT);

	if (listen(fd, LISTEN_MAX) < 0) {
		perror("listen");
		goto END;
	}

	addr_len = sizeof(si_remote);
	memset(&si_remote, 0, sizeof(si_remote));
	cfd = accept(fd, (struct sockaddr *)&si_remote, (socklen_t *)&addr_len);
	pthread_create(&th, NULL, recv_send, &cfd);
	/* Wait for the thread to end and synchronize operations between threads */
	pthread_join(th, NULL);
END:	close(fd);
	return 0;
}