#include <stdafx.h>
#include "context.h"
#include <libcommon/string.h>
#include <windows.h>
#include <nsis/pluginapi.h>
#include <string>
#include <vector>

Context g_context;

namespace
{

std::wstring PopString()
{
	//
	// NSIS functions popstring() and popstringn() require that you definitely size the buffer
	// before popping the string. Let's do it ourselves instead.
	//

	if (!g_stacktop || !*g_stacktop)
	{
		throw std::runtime_error("NSIS variable stack is corrupted");
	}

	stack_t *th = *g_stacktop;

	std::wstring copy(th->text);

	*g_stacktop = th->next;
	GlobalFree((HGLOBAL)th);

	return copy;
}

EXTERN_C IMAGE_DOS_HEADER __ImageBase;

void PinDll()
{
	//
	// Apparently NSIS loads and unloads the plugin module for EVERY call it makes to the plugin.
	// This makes it kind of difficult to maintain state.
	//
	// We can work around this by incrementing the module reference count.
	// When NSIS calls FreeLibrary() the reference count decrements and becomes one.
	//

	wchar_t self[MAX_PATH];

	if (0 == GetModuleFileNameW((HINSTANCE)&__ImageBase, self, _countof(self)))
	{
		throw std::runtime_error("Failed to pin plugin module");
	}

	LoadLibraryW(self);
}

} // anonymous namespace

//
// EstablishBaseline
//
// Invoke with the output from "tapinstall hwids tap0901"
// e.g.: driverlogic::EstablishBaseline $1
//
enum class EstablishBaselineStatus
{
	GENERAL_ERROR = 0,
	NO_INTERFACES_PRESENT,
	SOME_INTERFACES_PRESENT,
	MULLVAD_INTERFACE_PRESENT
};

void __declspec(dllexport) NSISCALL EstablishBaseline
(
	HWND hwndParent,
	int string_size,
	LPTSTR variables,
	stack_t **stacktop,
	extra_parameters *extra,
	...
)
{
	EXDLL_INIT();

	try
	{
		PinDll();

		auto status = EstablishBaselineStatus::GENERAL_ERROR;

		switch (g_context.establishBaseline(PopString()))
		{
			case Context::BaselineStatus::NO_INTERFACES_PRESENT:
			{
				status = EstablishBaselineStatus::NO_INTERFACES_PRESENT;
				break;
			}
			case Context::BaselineStatus::SOME_INTERFACES_PRESENT:
			{
				status = EstablishBaselineStatus::SOME_INTERFACES_PRESENT;
				break;
			}
			case Context::BaselineStatus::MULLVAD_INTERFACE_PRESENT:
			{
				status = EstablishBaselineStatus::MULLVAD_INTERFACE_PRESENT;
				break;
			}
			default:
			{
				throw std::runtime_error("Missing case handler in switch clause");
			}
		}

		pushstring(L"");
		pushint(status);
	}
	catch (std::exception &err)
	{
		pushstring(common::string::ToWide(err.what()).c_str());
		pushint(EstablishBaselineStatus::GENERAL_ERROR);
	}
}

//
// IdentifyNewInterface
//
// Invoke with the output from "tapinstall hwids tap0901"
// e.g.: driverlogic::IdentifyNewInterface $1
//
enum class IdentifyNewInterfaceStatus
{
	GENERAL_ERROR = 0,
	SUCCESS
};

void __declspec(dllexport) NSISCALL IdentifyNewInterface
(
	HWND hwndParent,
	int string_size,
	LPTSTR variables,
	stack_t **stacktop,
	extra_parameters *extra,
	...
)
{
	EXDLL_INIT();

	try
	{
		g_context.recordCurrentState(PopString());

		auto nic = g_context.getNewAdapter();

		pushstring(nic.alias.c_str());
		pushint(IdentifyNewInterfaceStatus::SUCCESS);
	}
	catch (std::exception &err)
	{
		pushstring(common::string::ToWide(err.what()).c_str());
		pushint(IdentifyNewInterfaceStatus::GENERAL_ERROR);
	}
}