#include <cstdio>
#include <windows.h>
#include "Natupnp.h."

inline void ReleaseObj(IUnknown* pInterface){
	if(pInterface){
		pInterface->Release();
	}
}

void log(char*arg_0,...) {
	char V8[1024];
	sprintf(V8, "%s\n", arg_0);
	va_list args;
	va_start (args, arg_0);
	vprintf (V8, args);
	va_end (args);
}


int HandleCOMError(HRESULT hr) {
	if(hr == S_OK){
		//log ("Success");
		return 0;
	}
	if (hr == REGDB_E_CLASSNOTREG){
		log ("REGDB_E_CLASSNOTREG: A specified class is not registered in the registration database. Also can indicate that the type of server you requested in the CLSCTX enumeration is not registered or the values for the server types in the registry are corrupt.");
	} else if (hr == CLASS_E_NOAGGREGATION){
		log ("CLASS_E_NOAGGREGATION: This class cannot be created as part of an aggregate.");
	} else if (hr == E_NOINTERFACE){
		log ("E_NOINTERFACE: The specified class does not implement the requested interface, or the controlling IUnknown does not expose the requested interface.");
	} else if (hr == E_ABORT){
		log ("E_ABORT: The operation was aborted. ");
	} else if (hr == E_FAIL){
		log ("E_FAIL:  An unspecified error occurred.");
	} else if (hr == E_INVALIDARG){
		log ("E_INVALIDARG: One of the parameters is invalid.");
	} else if (hr == E_NOTIMPL){
		log ("E_NOTIMPL: A specified method is not implemented.");
	} else if (hr == E_OUTOFMEMORY){
		log ("E_OUTOFMEMORY: The method was unable to allocate required memory. ");
	} else if (hr == E_POINTER){
		log ("E_POINTER: A pointer passed as a parameter is not valid.");
	} else if (hr == E_UNEXPECTED){
		log ("E_UNEXPECTED: The method failed for unknown reasons. ");
	} else {
		log ("Error of value 0x%x", hr);
	}

	return 1; 
}

HRESULT hr;
IUPnPNAT * upnpnat;
IStaticPortMappingCollection * spmaps;
long int count = 0;

int main (int argc, char**args) {
	// COM initializze
	hr = CoInitialize(NULL);


	upnpnat = NULL;
	log ("creating UPnPNAT");

	hr = CoCreateInstance(CLSID_UPnPNAT, NULL, CLSCTX_INPROC_SERVER, IID_IUPnPNAT, (void **) &upnpnat);
	if (HandleCOMError(hr)) {
		CoUninitialize();
		return 0;
	}

	log("Getting static port mapping collection");

	spmaps = 0;

	hr = upnpnat->get_StaticPortMappingCollection(&spmaps);

	if (HandleCOMError(hr)) {
		ReleaseObj(upnpnat);
		CoUninitialize();
		return 0;
	}

	log("Getting static port mapping collection count");

	
	hr = spmaps->get_Count(&count);
	
	if (HandleCOMError(hr)) {
		ReleaseObj(upnpnat);
		CoUninitialize();
		return 0;
	}

	log("%i static port maps found", count);



	log ("releasing UPnPNAT");
	//terminate the device finder
    ReleaseObj(upnpnat);
    upnpnat = NULL;
	CoUninitialize();

	return 0;
}

creating UPnPNAT
Getting static port mapping collection 
Getting static port mapping collection count