/*
 * Copyright (c) 2014 Grzegorz Kostka (kostka.grzegorz@gmail.com)
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 *
 * - Redistributions of source code must retain the above copyright
 *   notice, this list of conditions and the following disclaimer.
 * - 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.
 * - The name of the author may not be used to endorse or promote products
 *   derived from this software without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``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 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 <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <unistd.h>
#include <errno.h>
#include <stdbool.h>
#include <getopt.h>

#ifdef WIN32
#include <winsock2.h>
#include <ws2tcpip.h>
#include <windows.h>
static int inet_pton(int af, const char *src, void *dst);

#else
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <sys/types.h>
#endif

static int winsock_init(void);
static void winsock_fini(void);

/**@brief   Default server addres.*/
static char *server_addr = "127.0.0.1";

/**@brief   Default connection port.*/
static int connection_port = 1234;

/**@brief   Call op*/
static char *op_code;

static const char *usage = "                                    \n\
Welcome in lwext4_client.                                       \n\
Copyright (c) 2013 Grzegorz Kostka (kostka.grzegorz@gmail.com)  \n\
Usage:                                                          \n\
    --call (-c) - call opt                                      \n\
    --port (-p) - server port                                   \n\
    --addr (-a) - server ip address                             \n\
\n";

static int client_connect(void)
{
	int fd = 0;
	struct sockaddr_in serv_addr;

	if (winsock_init() < 0) {
		printf("winsock_init error\n");
		exit(-1);
	}

	memset(&serv_addr, '0', sizeof(serv_addr));
	fd = socket(AF_INET, SOCK_STREAM, 0);
	if (fd < 0) {
		printf("socket() error: %s\n", strerror(errno));
		exit(-1);
	}

	serv_addr.sin_family = AF_INET;
	serv_addr.sin_port = htons(connection_port);

	if (!inet_pton(AF_INET, server_addr, &serv_addr.sin_addr)) {
		printf("inet_pton() error\n");
		exit(-1);
	}

	if (connect(fd, (struct sockaddr *)&serv_addr, sizeof(serv_addr))) {
		printf("connect() error: %s\n", strerror(errno));
		exit(-1);
	}

	return fd;
}

static bool parse_opt(int argc, char **argv)
{
	int option_index = 0;
	int c;

	static struct option long_options[] = {
	    {"call", required_argument, 0, 'c'},
	    {"port", required_argument, 0, 'p'},
	    {"addr", required_argument, 0, 'a'},
	    {0, 0, 0, 0}};

	while (-1 != (c = getopt_long(argc, argv, "c:p:a:", long_options,
				      &option_index))) {

		switch (c) {
		case 'a':
			server_addr = optarg;
			break;
		case 'p':
			connection_port = atoi(optarg);
			break;
		case 'c':
			op_code = optarg;
			break;
		default:
			printf("%s", usage);
			return false;
		}
	}
	return true;
}

int main(int argc, char *argv[])
{
	int sockfd;
	int n;
	int rc;
	char recvBuff[1024];

	if (!parse_opt(argc, argv))
		return -1;

	sockfd = client_connect();

	n = send(sockfd, op_code, strlen(op_code), 0);
	if (n < 0) {
		printf("\tWrite error: %s fd = %d\n", strerror(errno), sockfd);
		return -1;
	}

	n = recv(sockfd, (void *)&rc, sizeof(rc), 0);
	if (n < 0) {
		printf("\tWrite error: %s fd = %d\n", strerror(errno), sockfd);
		return -1;
	}

	printf("rc: %d %s\n", rc, strerror(rc));
	if (rc)
		printf("\t%s\n", op_code);

	winsock_fini();
	return rc;
}

static int winsock_init(void)
{
#if WIN32
	int rc;
	static WSADATA wsaData;
	rc = WSAStartup(MAKEWORD(2, 2), &wsaData);
	if (rc != 0) {
		return -1;
	}
#endif
	return 0;
}

static void winsock_fini(void)
{
#if WIN32
	WSACleanup();
#endif
}

#if WIN32
static int inet_pton(int af, const char *src, void *dst)
{
	struct sockaddr_storage ss;
	int size = sizeof(ss);
	char src_copy[INET6_ADDRSTRLEN + 1];

	ZeroMemory(&ss, sizeof(ss));
	/* stupid non-const API */
	strncpy(src_copy, src, INET6_ADDRSTRLEN + 1);
	src_copy[INET6_ADDRSTRLEN] = 0;

	if (WSAStringToAddress(src_copy, af, NULL, (struct sockaddr *)&ss,
			       &size) == 0) {
		switch (af) {
		case AF_INET:
			*(struct in_addr *)dst =
			    ((struct sockaddr_in *)&ss)->sin_addr;
			return 1;
		case AF_INET6:
			*(struct in6_addr *)dst =
			    ((struct sockaddr_in6 *)&ss)->sin6_addr;
			return 1;
		}
	}
	return 0;
}
#endif