/*
 * Copyright 2010, Axel Dörfler, axeld@pinc-software.de.
 * This file may be used under the terms of the MIT License.
 */


#include <arpa/inet.h>
#include <netinet/in.h>
#include <sys/socket.h>
#include <stdio.h>
#include <string.h>
#include <unistd.h>


static bool sReceiveMode = false;


int
main(int argc, char** argv)
{
	const char* peerAddress = "192.168.0.1";
	if (argc > 1)
		peerAddress = argv[1];

	if (argc > 2 && !strcmp(argv[2], "recv"))
		sReceiveMode = true;

	int fd = socket(AF_INET, SOCK_DGRAM, 0);
	if (fd < 0)
		return 1;

#if 1
	sockaddr_in addr;
	addr.sin_family = AF_INET;
	addr.sin_addr.s_addr = INADDR_ANY;
	addr.sin_port = htons(sReceiveMode ? 8888 : 0);
	if (bind(fd, (sockaddr*)&addr, sizeof(addr)) != 0)
		perror("bind");

	sockaddr ours;
	socklen_t ourLength = sizeof(sockaddr);
	if (getsockname(fd, &ours, &ourLength) != 0)
		perror("getsockname");
	else {
		printf("After bind: local address %s (%u)\n",
			inet_ntoa(((sockaddr_in&)ours).sin_addr),
			ntohs(((sockaddr_in&)ours).sin_port));
	}
#endif

	if (!sReceiveMode) {
		sockaddr_in peer;
		peer.sin_family = AF_INET;
		peer.sin_addr.s_addr = inet_addr(peerAddress);
		peer.sin_port = htons(sReceiveMode ? 0 : 8888);
		if (connect(fd, (sockaddr*)&peer, sizeof(peer)) != 0)
			perror("connect");

		ourLength = sizeof(sockaddr);
		if (getsockname(fd, &ours, &ourLength) != 0)
			perror("getsockname");
		else {
			printf("After connect: local address %s (%u)\n",
				inet_ntoa(((sockaddr_in&)ours).sin_addr),
				ntohs(((sockaddr_in&)ours).sin_port));
		}
	}

	char buffer[256];
	if (sReceiveMode) {
		ssize_t bytesReceived = recv(fd, buffer, sizeof(buffer), 0);
		if (bytesReceived > 0) {
			buffer[bytesReceived] = 0;
			printf("received: %s\n", buffer);
		} else
			perror("recv");
	} else {
		if (argc > 2)
			send(fd, argv[2], strlen(argv[2]), 0);
		else {
			fgets(buffer, sizeof(buffer), stdin);
			send(fd, buffer, strlen(buffer), 0);
		}
	}

	close(fd);
	return 0;
}