scbc_repos/SATHelper/SATHelper/TestWizardDLL.cpp

#include "stdafx.h"
#include "TestWizardDLL.h"

namespace TW {
	HMODULE g_hdll = NULL;

	OPEN_XML OpenXml = NULL;
	LOAD_XML LoadXml = NULL;
	ADD_ITEM AddItem = NULL;
	DEL_ITEM DelItem = NULL;
	REMOVEALL_ITEM RemoveAllItem = NULL;
	SAVE_XML Save2Xml = NULL;
	OPEN_DEVICE OpenDevice = NULL;
	CLOSE_DEVICE CloseDevice = NULL;
	SEND_KEY SendKey = NULL;
	SEND_KEY SendCmd = NULL;
	GETSIGNALSNAME GetSignalsName = NULL;
	OPEN_SWITCH OpenSwitch = NULL;
	CLOSE_SWITCH CloseSwitch = NULL;
	SEND_SWITCH SendSwitch = NULL;


	BOOL LoadLibrary() 
	{
		if (g_hdll == NULL) {
			g_hdll = (HMODULE)::LoadLibrary(_T("TestWizard.dll"));
			if (!g_hdll)
				return FALSE;
		}

		OpenXml = (OPEN_XML)GetProcAddress(g_hdll, "OpenXML");
		if (!OpenXml)
		{
			UnloadLibrary();
			return FALSE;
		}

		LoadXml = (LOAD_XML)GetProcAddress(g_hdll, "LoadXML");
		if (!LoadXml)
		{
			UnloadLibrary();
			return FALSE;
		}

		AddItem = (ADD_ITEM)GetProcAddress(g_hdll, "AddItem");
		if (!AddItem)
		{
			UnloadLibrary();
			return FALSE;
		}

		DelItem = (DEL_ITEM)GetProcAddress(g_hdll, "DelItem");
		if (!DelItem)
		{
			UnloadLibrary();
			return FALSE;
		}

		RemoveAllItem = (REMOVEALL_ITEM)GetProcAddress(g_hdll, "RemoveAllItem");
		if (!RemoveAllItem)
		{
			UnloadLibrary();
			return FALSE;
		}

		Save2Xml = (SAVE_XML)GetProcAddress(g_hdll, "Save2XML");
		if (!Save2Xml)
		{
			UnloadLibrary();
			return FALSE;
		}

		OpenDevice = (OPEN_DEVICE)GetProcAddress(g_hdll, "OpenDevice");
		if (!OpenDevice)
		{
			UnloadLibrary();
			return FALSE;
		}

		CloseDevice = (CLOSE_DEVICE)GetProcAddress(g_hdll, "CloseDevice");
		if (!CloseDevice)
		{
			UnloadLibrary();
			return FALSE;
		}

		SendKey = (SEND_KEY)GetProcAddress(g_hdll, "SendKey");
		if (!SendKey)
		{
			UnloadLibrary();
			return FALSE;
		}

		SendCmd = (SEND_CMD)GetProcAddress(g_hdll, "SendCmd");
		if (!SendCmd)
		{
			UnloadLibrary();
			return FALSE;
		}

		GetSignalsName = (GETSIGNALSNAME)GetProcAddress(g_hdll, "GetSignalsName");
		if (!GetSignalsName)
		{
			UnloadLibrary();
			return FALSE;
		}

		// usb switch
		OpenSwitch = (OPEN_SWITCH)GetProcAddress(g_hdll, "OpenUsbSwitch");
		if (!OpenSwitch)
		{
			UnloadLibrary();
			return FALSE;
		}

		CloseSwitch = (CLOSE_SWITCH)GetProcAddress(g_hdll, "CloseUsbSwitch");
		if (!CloseSwitch)
		{
			UnloadLibrary();
			return FALSE;
		}

		SendSwitch = (SEND_SWITCH)GetProcAddress(g_hdll, "SendSwitch");
		if (!SendSwitch)
		{
			UnloadLibrary();
			return FALSE;
		}

		return TRUE;
	}

	void UnloadLibrary()
	{
		if (g_hdll) {
			if (FreeLibrary(g_hdll)) {
				g_hdll = NULL;

				OpenXml = NULL;
				LoadXml = NULL;
				AddItem = NULL;
				DelItem = NULL;
				RemoveAllItem = NULL;
				Save2Xml = NULL;
				OpenDevice = NULL;
				CloseDevice = NULL;
				SendKey = NULL;
				SendCmd = NULL;
				GetSignalsName = NULL;
				OpenSwitch = NULL;
				SendSwitch = NULL;
				CloseSwitch = NULL;
			}
		}
	}
}