/*
  Copyright (C) 2007 Arnaldo Carvalho de Melo <acme@redhat.com>

  This program is free software; you can redistribute it and/or modify it
  under the terms of version 2 of the GNU General Public License as
  published by the Free Software Foundation.
*/

#include <stdio.h>
#include <stdlib.h>
#include <signal.h>
#include <ctype.h>
#include <errno.h>
#include <string.h>
#include <netdb.h>
#include <unistd.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <sys/time.h>
#include <sys/resource.h>
#include <netinet/in.h>
#include <netinet/tcp.h>
#include <arpa/inet.h>

#define NR_DATA_ENTRIES 15
#define SIZE_DATA_ENTRY 2
#define SIZE_RESPONSE 2

static void tcp_nodelay(int fd)
{
	int value = 1;
	if (setsockopt(fd, SOL_TCP, TCP_NODELAY, &value, sizeof(value)) != 0)
		perror("setsockopt(TCP_NODELAY):");
}

static void __tcp_cork(int fd, int cork)
{
	int value = cork;
	if (setsockopt(fd, SOL_TCP, TCP_CORK, &value, sizeof(value)) != 0)
		perror("setsockopt(TCP_CORK):");
}

static void tcp_cork(int fd)
{
	__tcp_cork(fd, 1);
}

static void tcp_uncork(int fd)
{
	__tcp_cork(fd, 0);
}

static char response[SIZE_RESPONSE];

static int nread(int fd, char *bf, int len)
{
	do {
		int n = read(fd, bf, len);

		if (n <= 0)
			return -1;

		len -= n;
		bf += n;
	} while (len != 0);
	return 0;
}

static int exchange_packet(int fd, const int cork)
{
	char data[NR_DATA_ENTRIES * SIZE_DATA_ENTRY];

	if (nread(fd, data, sizeof(data)) != 0)
		return -1;

	if (cork)
		tcp_cork(fd);

	if (write(fd, response, SIZE_RESPONSE) != SIZE_RESPONSE) {
		fprintf(stderr, "server: write failed!\n");
		return -1;
	}

	if (cork)
		tcp_uncork(fd);

	return 0;
}

int main(int argc, char **argv)
{
	struct addrinfo *host;
	struct addrinfo hints = {
		.ai_family   = AF_INET,
		.ai_socktype = SOCK_STREAM,
		.ai_protocol = IPPROTO_TCP,
		.ai_flags    = AI_PASSIVE,
	};
	const char *port;
	int rc = 1, fd, nr_logical_packets, client, cork = 0, no_delay = 0;

	if (argc < 3) {
		fprintf(stderr, "usage: %s <port> <nr_logical_packets> "
				"[no_delay|cork]\n", argv[0]);
		goto out;
	}

	if (argc > 4) {
		cork = strcmp(argv[4], "cork") == 0;
		no_delay = strcmp(argv[4], "no_delay") == 0;
	}

	port = argv[1];
	nr_logical_packets = atoi(argv[2]);

	rc = getaddrinfo(NULL, port, &hints, &host);
	if (rc != 0) {
		fprintf(stderr, "error using getaddrinfo: %s\n", gai_strerror(rc));
		goto out;
	}
	
	fd = socket(host->ai_family, host->ai_socktype, host->ai_protocol);
	if (fd < 0) {
		perror("socket: ");
		goto out_freeaddrinfo;
	}

	if (bind(fd, host->ai_addr, host->ai_addrlen) < 0) {
		perror("bind: ");
		goto out_close_server;
	}

	memset(response, 'Z', SIZE_RESPONSE);

	listen(fd, 1);
	while (1) {
		char peer[1024];

		puts("server: waiting for connection");

		client = accept(fd, host->ai_addr, &host->ai_addrlen);
		if (client < 0) {
			perror("accept: ");
			goto out_close_server;
		}

		rc = getnameinfo(host->ai_addr, host->ai_addrlen,
				 peer, sizeof(peer), NULL, 0, 0);
		if (rc != 0) {
			fprintf(stderr, "error using getnameinfo: %s\n",
				gai_strerror(rc));
			continue;
		}

		printf("server: accept from %s\n", peer);

		if (no_delay)
			tcp_nodelay(client);

		while (nr_logical_packets--)
			if (exchange_packet(client, cork) != 0)
				break;

		close(client);
	}
out_close_server:
	close(fd);
out_freeaddrinfo:
	freeaddrinfo(host);
out:
	return rc;
}