// Universal Plug and Play
// UPnP is used in this demo to programmatically manipulate
// port forwarding entries on the user's router. 


using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Threading;

namespace UPNP
{
    class Program
    {
        // Check for errors and deal with them
        static void CheckError()
        {
	        if(mn.GetErrorFlag() == true)
	        {
                Console.Clear();
                Console.WriteLine("An unexpected error has occurred. This indicates that Universal Plug and Play is disabled or is not functioning properly on your computer or network.");
		        Console.WriteLine();
                Console.WriteLine("Details of this error are as follows:");

		        Console.WriteLine("Full error message: " + mn.GetErrorFull());
                Console.WriteLine();
		        Console.WriteLine("Operation: " + mn.GetErrorOperation());
		        Console.WriteLine("Command: " + mn.GetErrorCommand());
		        Console.WriteLine("Code: " + mn.GetErrorCode().ToString());
		        Console.WriteLine("Line: " + mn.GetErrorLine().ToString());
                Console.WriteLine("File: " + mn.GetErrorFile());

                Console.WriteLine();
                Console.Read();
	        }
        }


        // Update NAT non blocking
        static void DoUpdate(bool ReloadInfo, string Title)
        {
	        if(ReloadInfo == true)
	        {
		        // Start update NAT action without blocking
		        Console.WriteLine("Updating NAT...");
		        long Action = mn.UpdateNAT(false);
                CheckError();

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

	        // Display information loaded by mnUpdateNAT
	        Console.Clear();
	        Console.WriteLine("Port map information " + Title);
	        for(uint n = 0;n<mn.GetPortMapAmount();n++)
	        {
		        Console.WriteLine(n + ": Description: " + mn.GetPortMapDescription(n) + ", " + mn.GetPortMapInternalIP(n) + ", " + mn.GetPortMapInternalPort(n));
	        }

            Console.ReadKey();
        }

        static void Main()
        {
            // Disable EM_MESSAGE_BOX (2) and enable EM_SAVE (1)
            // so that our application can deal with the error
            mn.ToggleErrorMode(1);
            mn.ToggleErrorMode(2);

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

	        // Stores ourt own port map's information
	        string sProtocol = "TCP";
            int lExternalPort = 2525;

	        // Start UPNP and NAT
	        // NAT must be started after UPNP
            mn.StartUPNP();
            CheckError();

            mn.StartNAT(true);
            CheckError();

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

	        // Add our own port map
            mn.AddPortMap(sProtocol, lExternalPort, 4242, "192.168.1.5", false, "DarkNet (demo port forwarding entry)", true);
            CheckError();

            DoUpdate(false, "after creating a new port map");

	        uint iPortMap;

	        // Check that our own port map was added correctly
	        int iPMExist = mn.PortMapExist(sProtocol,lExternalPort);
            CheckError();

	        if(iPMExist == 1)
	        {
			    // Find our port map's ID
			    iPortMap = mn.FindPortMap(sProtocol,lExternalPort);
	        }
	        else
	        {
		        Console.WriteLine("Port map does not exist! (" + iPMExist + ")");
		        return;
	        }

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

            mn.SetPortMapInternalPort(iPortMap, 1000, true);
            CheckError();

            mn.SetPortMapDescription(iPortMap, "CHANGED DarkNet (demo)", true);
            CheckError();

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

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

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

	        // Unload NAT and UPNP
	        // UPNP must be unloaded last
            mn.FinishNAT(true);
            CheckError();

            mn.FinishUPNP(true);
            CheckError();

	        Console.Clear();
	        Console.WriteLine("Finished, press any key to exit");
	        Console.ReadKey();
        }
    }
}