ebusd/src/network.cpp

/*
 * Copyright (C) Roland Jax 2014 <roland.jax@liwest.at>
 *
 * This file is part of ebus-daemon.
 *
 * ebus-daemon 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.
 *
 * ebus-daemon 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 ebus-daemon. If not, see http://www.gnu.org/licenses/.
 */

#include "network.h"
#include "logger.h"
#include "appl.h"
#include <sstream>
#include <cstring>
#include <sys/select.h>

extern LogInstance& L;
extern Appl& A;

int Connection::m_count = -1;

void Connection::addResult(Message message)
{
	Message* tmp = new Message(Message(message));
	m_result.add(tmp);
}

void* Connection::run()
{
	m_running = true;
	
	int maxfd;
	fd_set checkfds;
	struct timeval timeout;

	FD_ZERO(&checkfds);
	FD_SET(m_notify.notifyFD(), &checkfds);
	FD_SET(m_socket->getFD(), &checkfds);
	
	(m_notify.notifyFD() > m_socket->getFD()) ?
		(maxfd = m_notify.notifyFD()) : (maxfd = m_socket->getFD());

	for (;;) {
		fd_set readfds;
		int ret;

		// set select timeout 10 secs
		timeout.tv_sec = 10;
		timeout.tv_usec = 0;		

		// set readfds to inital checkfds
		readfds = checkfds;

		ret = select(maxfd + 1, &readfds, NULL, NULL, &timeout);
		if (ret == 0) {
			continue;
		}
			
		// new data from notify
		if (FD_ISSET(m_notify.notifyFD(), &readfds))
			break;

		// new data from socket
		if (FD_ISSET(m_socket->getFD(), &readfds)) {
			char data[256];
			size_t datalen;

			if (m_socket->isValid() == true)
				datalen = m_socket->recv(data, sizeof(data)-1);
			else
				break;
			
			// removed closed socket
			if (datalen <= 0 || strncasecmp(data, "quit", 4) == 0)
				break;

			// send data
			data[datalen] = '\0';
			m_data->add(new Message(data, this));

			// wait for result
			L.log(net, debug, "[%08x] wait for result", getID());
			Message* message = m_result.remove();

			L.log(net, debug, "[%08x] result added", getID());
			std::string result(message->getData());
			
			if (m_socket->isValid() == true)
				m_socket->send(result.c_str(), result.size());
			else
				break;				
			
			delete message;

		}

	}

	delete m_socket;
	m_running = false;
	L.log(net, event, "[%08x] connection closed - active connections: %d", getID(), m_count);

	return NULL;
}



Network::Network(const bool localhost) : m_listening(false), m_running(false)
{
	// Start Listener
	if (localhost == true)
		m_Listener = new TCPListener(A.getParam<int>("p_port"), "127.0.0.1");
	else
		m_Listener = new TCPListener(A.getParam<int>("p_port"), "0.0.0.0");

	if (m_Listener && m_Listener->start() == 0)
		m_listening = true;

}

Network::~Network()
{
	while (m_connections.empty() == false) {
		Connection* connection = m_connections.back();
		m_connections.pop_back();
		connection->stop();
		connection->join();
		delete connection;
	}
	
	if (m_running == true)
		stop();
		
	delete m_Listener;
}

void* Network::run()
{
	if (m_listening == false)
		return NULL;

	m_running = true;

	int maxfd;
	fd_set checkfds;
	struct timeval timeout;

	FD_ZERO(&checkfds);
	FD_SET(m_notify.notifyFD(), &checkfds);
	FD_SET(m_Listener->getFD(), &checkfds);
	
	(m_notify.notifyFD() > m_Listener->getFD()) ?
		(maxfd = m_notify.notifyFD()) : (maxfd = m_Listener->getFD());

	for (;;) {
		fd_set readfds;
		int ret;

		// set select timeout 1 secs
		timeout.tv_sec = 1;
		timeout.tv_usec = 0;		

		// set readfds to inital checkfds
		readfds = checkfds;

		ret = select(maxfd + 1, &readfds, NULL, NULL, &timeout);
		if (ret == 0) {
			cleanConnections();
			continue;
		}
			
		// new data from notify
		if (FD_ISSET(m_notify.notifyFD(), &readfds)) {
			m_running = false;
			break;
		}

		// new data from socket
		if (FD_ISSET(m_Listener->getFD(), &readfds)) {
			TCPSocket* socket = m_Listener->newSocket();
			if (socket == NULL)
				continue;
				
			Connection* connection = new Connection(socket, m_queue);
			if (connection == NULL)
				continue;

			connection->start("netConnection");
			m_connections.push_back(connection);
			L.log(net, event, "[%08x] connection opened %s", connection->getID(), socket->getIP().c_str());	
		}

	}

	return NULL;
}

void Network::cleanConnections()
{
	std::list<Connection*>::iterator c_it;
	for (c_it = m_connections.begin(); c_it != m_connections.end(); c_it++) {
		if ((*c_it)->isRunning() == false) {
			Connection* connection = *c_it;
			c_it = m_connections.erase(c_it);
			delete connection;
			L.log(net, debug, "dead connection removed - %d", m_connections.size());
		}
	}
}