sk-nat/Node.cpp

/*
 * Node.cpp
 *
 *  Created on: 10.01.2017
 *      Author: piotrek
 */

#include "Node.h"
#include "NATRouter.h"

Node * Node::findConnection(string ip)
{
	NATRouter *ptr;
	if (ip == "0.0.0.0")
		return NULL;

	vector<Node*>::iterator it = this->connectedNodes.begin();
	for (; it != this->connectedNodes.end(); ++it)
	{
		if ((*it)->getIp() == ip)
				return *it;

		/*
		 * jezeli wezel ma polaczenie do NAT-routera
		 * (i istnieje jego interfejs WAN to sprawdz
		 * czy nie ma z nim bezposredniego polaczenia)
		 */
		ptr = dynamic_cast<NATRouter*>(*it);

		if (ptr != NULL)
			if (ptr->getWanIp() == ip)
				return *it;
	}

	return NULL;
}

/* funkcja pobiera i usuwa pakiet z bufora "karty sieciowej" */
Packet Node::recv()
{
	Packet packet;
	if (!this->rcvBuffer.empty())
	{
		packet = this->rcvBuffer.front();
		this->rcvBuffer.pop();
	}

	return packet;
}

string Node::ipToString(unsigned int ip)
{
	ostringstream ss;
	ss << ((ip>>24)&0xFF) << "." << ((ip>>16)&0xFF) << "." << ((ip>>8)&0xFF) << "." << (ip&0xFF);

	return ss.str();
}

unsigned int Node::stringToIp(string ip)
{
	istringstream iss(ip);
	int offset = 24;
	unsigned int res=0;

	std::string token;
	while (getline(iss, token, '.')) {
	    if (!token.empty())
	    {
	        res |= atoi(token.c_str()) << offset;
	        offset -= 8;
	    }
	}

	return res;
}

Node::Node()
{
	this->setHostname("");
	this->setIp("0.0.0.0");
	this->setMask("0.0.0.0");
	this->setGatewayIp("0.0.0.0");
}

Node::~Node()
{
	this->connectedNodes.clear();
	while(!this->rcvBuffer.empty())
		this->rcvBuffer.pop();
}

Node::Node(string hostname) : Node()
{
	this->setHostname(hostname);
}

Node::Node(string hostname, string ip, string mask) : Node(hostname)
{
	this->setIp(ip);
	this->setMask(mask);
}

Node::Node(string hostname, string ip, string mask, string gatewayIp) : Node(hostname, ip, mask)
{
	this->setGatewayIp(gatewayIp);
}

/*
 * funkcja symuluje typową sytuację: mamy kartę sieciową, a funkcja connectNode jest podłączeniem wtyczki
 * z jednej strony
 * @param firstConnected - oznacza, czy pierwsza strona jest podlaczona (jezeli nie to sprawdzamy
 * czy polaczenie juz gdzies zostalo przypadkowo nawiazane, nawiazujemy polaczenie, a nastepnie
 * przekazujemy sterowanie drugiemu wezlowi zeby zrobil to samo)
 */
bool Node::connectNode(Node *node, bool firstConnected)
{
	if (node->getIp() == "0.0.0.0")		// jezeli wezel nie ma skonfigurowanej sieci, nie mozna go przylaczyc
		return false;

	if (!firstConnected)
	{
		vector<Node*>::iterator it = this->connectedNodes.begin();
		for (; it != this->connectedNodes.end(); ++it)	// sprawdzamy, czy połączenia już przypadkiem nie ma
		{
			if (*it == node)
				return false;
		}
	}
	this->connectedNodes.push_back(node); 			// podłączamy drugi węzeł

	if (!firstConnected)
		node->connectNode(this, true);				// to samo w drugą stronę

	return true;
}

/*
 * funkcja wysyla, czyli przekazuje kopie pakietu kolejnemu wezlowi - docelowemu lub bramie domyslnej
 * o ile jest skonfigurowana i istnieje do niej sciezka
 */
bool Node::send(Packet packet, bool isRouter)
{
	Node *node;
	node = this->findConnection(packet.getDstIp()); // znajdz bezposrednia trase
	if (!node)
		node = this->findConnection(this->getGatewayIp());

	if (!node)
		return false;	// nie ma zadnej trasy do wezla

	if (packet.getDstPort() == 0)
		return false;	// wypada zdefiniowac nadawce oraz docelowy port...

	// jezeli wezel nie jest routerem to ma ustawic pakietowi swoj srcIp oraz losowy port
	if (!isRouter)
	{
		packet.setSrcIp(this->getIp());
		if (packet.getSrcPort() == 0)
		{
			// ustaw port zrodlowy na port losowy z zakresu [32768,61000] - zob. empheral port (port emferyczny)
			srand(time(NULL));
			packet.setSrcPort(rand() % 32768 + 28233);
		}
	}
	node->putPacket(packet);

	return true;
}

/*
 * funkcja jest modelem karty sieciowej - odbiera dane z sieci i umieszcza je
 * w swoim buforze (czy tam systemie)
 */
void Node::putPacket(Packet packet)
{
	this->rcvBuffer.push(packet);
}

void Node::setHostname(string hostname)
{
	this->hostname = hostname;
}

void Node::setIp(string ip)
{
	this->netConf.ip = ip;
	this->setNetwork();
}

void Node::setMask(string mask)
{
	this->netConf.mask = mask;
	this->setNetwork();
}

void Node::setGatewayIp(string gatewayIp)
{
	// TODO mozna ewentualnie zrobic sprawdzenie czy gateway jest w podsieci
	this->netConf.gatewayIp = gatewayIp;
}

string Node::getHostname()
{
	return this->hostname;
}

string Node::getIp()
{
	return this->netConf.ip;
}

string Node::getMask()
{
	return this->netConf.mask;
}

string Node::getGatewayIp()
{
	return this->netConf.gatewayIp;
}

string Node::calculateNetwork(string ip, string mask)
{
	return ipToString(stringToIp(ip) & stringToIp(mask));
}

void Node::onRecv()
{
	// TESTOWO
	Packet p = this->recv();
	if (p.getSrcPort() != 0)
	{
		cout << endl << "<<<<<< " << this->getHostname() << " >>>>>>" << endl;
		cout << "Odebrano wiadomosc!" << endl;
		cout << "Src: " << p.getSrcIp() << ":" << p.getSrcPort() << endl;
		cout << "Dst: " << p.getDstIp() << ":" << p.getDstPort() << endl;
		cout << "---" << endl << p.getMsg() << endl << endl;
	}

}

void Node::setNetwork()
{
	this->netConf.network = this->calculateNetwork(this->getIp(), this->getMask());
}

string Node::getNetwork()
{
	return this->netConf.network;
}