// Demonstrates the core windows firewall commands



#include <MikeNet.h>
#include <iostream>
#include <windows.h>
using namespace std;

void main()
{
	// Load windows firewall module
	mnStartFirewall();

	// Output full information about firewall profile currently in use
	unsigned int cp = mnGetCurrentProfile();

	cout << "FIREWALL INFORMATION\n";
	cout << "Firewall enabled: " << mnGetFirewallEnabled(cp) << '\n';
	cout << "Exceptions allowed: " << mnGetExceptionsAllowed(cp) << '\n';
	cout << "Notifications enabled: " << mnGetNotificationsEnabled(cp) << '\n';
	cout << "Unicast responses to multicast broadcast enabled: " << mnGetUnicastToMulticastEnabled(cp) << '\n';
	cout << "Remote admin IP version: " << mnGetRemoteAdminIPVersion(cp) << '\n';
	cout << "Remote admin scope: " << mnGetRemoteAdminScope(cp) << '\n';
	cout << "Remote admin addresses: " << mnGetRemoteAdminAddresses(cp) << '\n';
	cout << "Remote admin enabled: " << mnGetRemoteAdminEnabled(cp) << '\n';
	cout << "ICMP allow outbound destination unreachable: " << mnGetICMPAllowOutboundDestinationUnreachable(cp) << '\n';
	cout << "ICMP allow redirect: " << mnGetICMPAllowRedirect(cp) << '\n';
	cout << "ICMP allow inbound echo request: " << mnGetICMPAllowInboundEchoRequest(cp) << '\n';
	cout << "ICMP allow outbound time exceeded: " << mnGetICMPAllowOutboundTimeExceeded(cp) << '\n';
	cout << "ICMP allow outbound parameter problem: " << mnGetICMPAllowOutboundParameterProblem(cp) << '\n';
	cout << "ICMP allow outbound source quench: " << mnGetICMPAllowOutboundSourceQuench(cp) << '\n';
	cout << "ICMP allow inbound router request: " << mnGetICMPAllowInboundRouterRequest(cp) << '\n';
	cout << "ICMP allow inbound timestamp request: " << mnGetICMPAllowInboundTimestampRequest(cp) << '\n';
	cout << "ICMP allow inbound mask request: " << mnGetICMPAllowInboundMaskRequest(cp) << '\n';
	cout << "ICMP allow outbound packet too big: " << mnGetICMPAllowOutboundPacketTooBig(cp) << '\n';
	system("PAUSE");
	cout << "\n\n";

	cout << "SERVICES INFORMATION\n";
	cout << mnGetServiceAmount(cp) << " services found\n";
	cout << "\n";
	for(unsigned int n = 0;n<mnGetServiceAmount(cp);n++)
	{
		cout << "Service name: " << mnGetServiceName(cp, n) << '\n';
		cout << "Service type: " << mnGetServiceType(cp, n) << '\n';
		cout << "Service customized: " << mnGetServiceCustomized(cp, n) << '\n';
		cout << "Service IP version: " << mnGetServiceIPVersion(cp, n) << '\n';
		cout << "Service scope: " << mnGetServiceScope(cp, n) << '\n';
		cout << "Service remote addresses: " << mnGetServiceRemoteAddresses(cp, n) << '\n';
		cout << "Service enabled: " << mnGetServiceEnabled(cp, n) << '\n';
		cout << '\n';
	}
	system("PAUSE");
	cout << "\n\n";

	cout << "AUTHORIZED APPLICATION INFORMATION\n";
	cout << mnGetApplicationAmount(cp) << " authorized applications found\n";
	for(unsigned int n = 0;n<mnGetApplicationAmount(cp);n++)
	{
		cout << "Application name: " << mnGetApplicationName(cp, n) << '\n';
		cout << "Application file name: " << mnGetApplicationFileName(cp, n) << '\n';
		cout << "Application IP version: " << mnGetApplicationIPVersion(cp, n) << '\n';
		cout << "Application scope: " << mnGetApplicationScope(cp, n) << '\n';
		cout << "Application remote addresses: " << mnGetApplicationRemoteAddresses(cp, n) << '\n';
		cout << "Application enabled: " << mnGetApplicationEnabled(cp, n) << '\n';
		cout << '\n';
	}

	system("PAUSE");
	cout << "\n\n";

	// Turn firewall 'off and on' or 'on and off'
	unsigned int Seconds = 3;
	bool FirewallEnabled = mnGetFirewallEnabled(cp);
	if(FirewallEnabled == true)
	{
		cout << "Your firewall is enabled, would you like to disable it for "<<Seconds<<" seconds? (1 = Yes, 0 = No)\n";
	}
	else
	{
		cout << "Your firewall is disabled, would you like to enable it for "<<Seconds<<" seconds? (1 = Yes, 0 = No)\n";
	}

	bool bDo;
	cin >> bDo;

	if(bDo == true)
	{
		cout << "Taking action...\n";
		mnSetFirewallEnabled(cp, !FirewallEnabled); // bool is flipped
		cout << "Waiting "<<Seconds<<" seconds...\n";
		Sleep(Seconds * 1000);
		cout << "Reversing action...\n";
		mnSetFirewallEnabled(cp, FirewallEnabled);
		cout << "Action reversed\n";
	}
	
	system("PAUSE");
	cout << "\n\n";	

	// Add application to windows firewall
	const unsigned int FileNameSize = 1024;
	char FileName[FileNameSize];
	GetModuleFileName(NULL,FileName,FileNameSize);

	cout << "Adding application to windows firewall...\n";
	mnAddApplication(cp,"MikeNet windows firewall test",FileName,2,0,"*",false);
	
	// Check that operation completed successfully
	if(mnApplicationExist(cp,FileName) == 1)
	{
		cout << "Application added successfully\n";

		// Remove application from windows firewall
		cout << "Removing application from windows firewall\n";
		mnDeleteApplicationA(cp,FileName);
		
		if(mnApplicationExist(cp,FileName) == 0)
		{
			cout << "Application removed successfully\n";
		}
		else
		{
			cout << "Application was not removed successfully\n";
		}
	}
	else
	{
		cout << "Application was not added successfully\n";
	}

	system("PAUSE");

	// Unload windows firewall module
	mnFinishFirewall();
}