/*
 * Copyright (C) Roland Jax 2012-2014 <roland.jax@liwest.at>
 *
 * This file is part of ebusd.
 *
 * ebusd is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * ebusd is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with ebusd. If not, see http://www.gnu.org/licenses/.
 */

#include "tcpsocket.h"
#include <cstdlib>
#include <sys/types.h>
#include <sys/socket.h>
#include <unistd.h>
#include <fcntl.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <netdb.h>
#include <string.h>

TCPSocket::TCPSocket(int sfd, struct sockaddr_in* address) : m_sfd(sfd)
{
	char ip[17];
	inet_ntop(AF_INET, (struct in_addr*)&(address->sin_addr.s_addr), ip, sizeof(ip)-1);
	m_ip = ip;
	m_port = ntohs(address->sin_port);
}

bool TCPSocket::isValid()
{
	if (fcntl(m_sfd, F_GETFL) == -1)
		return false;
	else
		return true;
}


TCPSocket* TCPClient::connect(const std::string& server, const int& port)
{
	struct sockaddr_in address;
	int ret;

	memset((char*) &address, 0, sizeof(address));

	if (inet_addr(server.c_str()) == INADDR_NONE) {
		struct hostent* he;

		he = gethostbyname(server.c_str());
		if (he == NULL)
			return NULL;

		memcpy(&address.sin_addr, he->h_addr_list[0], he->h_length);
	} else {
		ret = inet_aton(server.c_str(), &address.sin_addr);
		if (ret == 0)
			return NULL;
	}

	address.sin_family = AF_INET;
	address.sin_port = htons(port);

	int sfd = socket(AF_INET, SOCK_STREAM, 0);
	if (sfd < 0)
		return NULL;

	ret = ::connect(sfd, (struct sockaddr*) &address, sizeof(address));
	if (ret < 0)
		return NULL;

	return new TCPSocket(sfd, &address);
}


int TCPServer::start()
{
	if (m_listening == true)
		return 0;

	m_lfd = socket(AF_INET, SOCK_STREAM, 0);
	struct sockaddr_in address;

	memset(&address, 0, sizeof(address));

	address.sin_family = AF_INET;
	address.sin_port = htons(m_port);

	if (m_address.size() > 0)
		inet_pton(AF_INET, m_address.c_str(), &(address.sin_addr));
	else
		address.sin_addr.s_addr = INADDR_ANY;

	int optval = 1;
	setsockopt(m_lfd, SOL_SOCKET, SO_REUSEADDR, &optval, sizeof(optval));

	int result = bind(m_lfd, (struct sockaddr*) &address, sizeof(address));
	if (result != 0)
		return result;

	result = listen(m_lfd, 5);
	if (result != 0)
		return result;

	m_listening = true;
	return result;
}

TCPSocket* TCPServer::newSocket()
{
	if (m_listening == false)
		return NULL;

	struct sockaddr_in address;
	socklen_t len = sizeof(address);

	memset(&address, 0, sizeof(address));

	int sfd = accept(m_lfd, (struct sockaddr*) &address, &len);
	if (sfd < 0)
		return NULL;

	return new TCPSocket(sfd, &address);
}