// Universal Plug and Play
// UPnP is used in this demo to programmatically manipulate
// port forwarding entries on the user's router. 
#include <MikeNet.h>
#include <windows.h>
#include <iostream>

using namespace std;

// Check for errors and deal with them
void CheckError()
{
	if(mnGetErrorFlag() == true)
	{
		system("CLS");
		cout << "An unexpected error has occurred. This indicates that Universal Plug and Play is disabled or is not functioning properly on your computer or network.\n";
		cout << "\nDetails of this error are as follows: \n";

		cout << "Full error message: " << mnGetErrorFull() << "\n\n";
		cout << "Operation: " << mnGetErrorOperation() << '\n';
		cout << "Command: " << mnGetErrorCommand() << '\n';
		cout << "Code: " << mnGetErrorCode() << '\n';
		cout << "Line: " << mnGetErrorLine() << '\n';
		cout << "File: " << mnGetErrorFile() << '\n';

		cout << "\n";
		system("PAUSE");
		exit(0);
	}
}

// Update NAT non blocking
void DoUpdate(bool ReloadInfo, char * Title)
{
	if(ReloadInfo == true)
	{
		// Start update NAT action without blocking
		cout << "Updating NAT...\n";
		long long int Action = mnUpdateNAT(false);
		CheckError();

		// Wait for update to finish
		int iResult;
		do
		{
			iResult = mnPollUPNP(Action);
			Sleep(1);
		}
		while(iResult == 2); // Loop until action is completed

		CheckError();
	}

	// Display information loaded by mnUpdateNAT
	system("CLS");
	cout << "Port map information " << Title << '\n';
	for(unsigned int n = 0;n<mnGetPortMapAmount();n++)
	{
		cout << n << ": Description: " << mnGetPortMapDescription(n) << ", " << mnGetPortMapInternalIP(n) << ", " << mnGetPortMapInternalPort(n) <<'\n';
	}

	system("PAUSE");
}

void main()
{
	// Disable EM_MESSAGE_BOX and enable EM_SAVE
	// so that our application can deal with the error
	mnToggleErrorMode(EM_SAVE);
	mnToggleErrorMode(EM_MESSAGE_BOX);

	// 1 = Delete using port map ID
	// 2 = Delete using protocol and external port
	const int iDeleteMethod = 1;

	// Stores ourt own port map's information
	char sProtocol[] = "TCP";
	long lExternalPort = 2525;

	// Start UPNP and NAT
	// NAT must be started after UPNP
	mnStartUPNP();
	CheckError();

	mnStartNAT(true);
	CheckError();

	// Display current port maps
	DoUpdate(false, "before adding, editing or deleting");

	// Add our own port map
	mnAddPortMap(sProtocol,lExternalPort,4242,"192.168.1.5",false,"DarkNet (demo port forwarding entry)",true);
	CheckError();
	DoUpdate(false, "after creating a new port map");

	unsigned int iPortMap;

	// Check that our own port map was added correctly
	int iPMExist = mnPortMapExist(sProtocol,lExternalPort);

	if(iPMExist == 1)
	{
		// Find our port map's ID
		iPortMap = mnFindPortMap(sProtocol,lExternalPort);
	}
	else
	{
		CheckError();
		cout << "Port map does not exist! (" << iPMExist << ")\n";
		return;
	}

	// Change setting of our port map
	mnSetPortMapInternalIP(iPortMap,"192.168.1.2",true);
	CheckError();

	mnSetPortMapInternalPort(iPortMap,1000,true);
	CheckError();

	mnSetPortMapDescription(iPortMap,"CHANGED DarkNet (demo)",true);
	CheckError();

	DoUpdate(false, "after editing our new port map");

	// Delete using port map ID
	if(iDeleteMethod == 1)
	{
		mnDeletePortMapA(iPortMap,true);
		CheckError();
	}
	// Delete using protocol and external port
	if(iDeleteMethod == 2)
	{
		mnDeletePortMapB(sProtocol,lExternalPort,true);
		CheckError();
	}

	DoUpdate(false, "after deleting our new port map");

	// Unload NAT and UPNP
	// UPNP must be unloaded last
	mnFinishNAT(true);
	CheckError();

	mnFinishUPNP(true);
	CheckError();

	system("CLS");
	cout << "Finished, ";
	system("PAUSE");
}