L2M_MACRO/DriverApi/driverapi.cpp

#include "stdafx.h"

#include <Windows.h>
#include <winioctl.h>
#include ".\..\common\common.h"
#include ".\..\common\driverapi.h"

#include <stdio.h>      /* printf, scanf, NULL */
#include <stdlib.h>

#include <setupapi.h>

#pragma comment(lib,"setupapi.lib")

// <20><><EFBFBD><EFBFBD> <20><><EFBFBD><EFBFBD><EFBFBD>Ǵ<EFBFBD> GUID<49><44> <20><><EFBFBD><EFBFBD><EFBFBD>ϴ<EFBFBD> DeviceStack<63><6B> <20><><EFBFBD><EFBFBD><EFBFBD><EFBFBD> <20>˷<EFBFBD><CBB7>ݴϴ<DDB4>
int GetDeviceStackNameCount(struct _GUID * pGuid)
{
	SP_INTERFACE_DEVICE_DATA interfaceData;
	int index = 0;
	HDEVINFO Info = SetupDiGetClassDevs(pGuid, 0, 0, DIGCF_PRESENT | DIGCF_INTERFACEDEVICE);

	if (Info == (HDEVINFO)-1)
		return 0; // <20>ý<EFBFBD><C3BD><EFBFBD><EFBFBD><EFBFBD> <20>̷<EFBFBD> <20><><EFBFBD><EFBFBD><EFBFBD><EFBFBD> <20><><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD> <20><><EFBFBD>Ѵ<EFBFBD>

	interfaceData.cbSize = sizeof(SP_INTERFACE_DEVICE_DATA);

	while (1)
	{
		BOOL bl;
		bl = SetupDiEnumDeviceInterfaces(Info, 0, pGuid, index, &interfaceData);
		if (bl == FALSE)
			break;
		index++;
	}

	SetupDiDestroyDeviceInfoList(Info);

	return index;
}

// <20><><EFBFBD><EFBFBD> <20><><EFBFBD><EFBFBD><EFBFBD>Ǵ<EFBFBD> GUID<49><44> <20><><EFBFBD><EFBFBD><EFBFBD>ϴ<EFBFBD> DeviceStack<63><6B> <20≯<EFBFBD><CCB8><EFBFBD><EFBFBD><EFBFBD> <20>˷<EFBFBD><CBB7>ݴϴ<DDB4>
BOOL GetDeviceStackName(struct _GUID * pGuid, TCHAR ** ppDeviceName, int index)
// index : <20><><EFBFBD><EFBFBD> Interface GUID<49><44> <20><><EFBFBD><EFBFBD> <20><>ġ<EFBFBD><C4A1> <20><><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ΰ<EFBFBD><CEB0><EFBFBD>, <20><><EFBFBD><EFBFBD> <20><><EFBFBD><EFBFBD> <20>˻<EFBFBD><CBBB>Ǵ<EFBFBD> <20><>ġ<EFBFBD><C4A1> <20><><EFBFBD><EFBFBD><EFBFBD>Ϸ<EFBFBD><CFB7><EFBFBD> <20>̰<EFBFBD><CCB0><EFBFBD> 0<><30> <20><><EFBFBD><EFBFBD><EFBFBD>մϴ<D5B4>
{
	DWORD size;
	BOOL bl;
	SP_INTERFACE_DEVICE_DATA interfaceData;
	PSP_INTERFACE_DEVICE_DETAIL_DATA pData;
	HDEVINFO Info = SetupDiGetClassDevs(pGuid, 0, 0, DIGCF_PRESENT | DIGCF_INTERFACEDEVICE);
	TCHAR *pDeviceName;
	*ppDeviceName = (TCHAR *)0;

	if (Info == (HANDLE)-1)
		return FALSE;

	interfaceData.cbSize = sizeof(SP_INTERFACE_DEVICE_DATA);

	bl = SetupDiEnumDeviceInterfaces(Info, 0, pGuid, index, &interfaceData);
	if (bl == FALSE)
		return bl;

	SetupDiGetDeviceInterfaceDetail(Info, &interfaceData, 0, 0, &size, 0);
	pData = (PSP_INTERFACE_DEVICE_DETAIL_DATA)malloc(size);
	pData->cbSize = sizeof(SP_INTERFACE_DEVICE_DETAIL_DATA);
	SetupDiGetDeviceInterfaceDetail(Info, &interfaceData, pData, size, 0, 0);

	pDeviceName = (TCHAR *)malloc((wcslen(pData->DevicePath) + 1) * sizeof(TCHAR));
	memset(pDeviceName, 0, wcslen(pData->DevicePath) * sizeof(TCHAR));
	wcscpy(pDeviceName, pData->DevicePath);
	free(pData);

	SetupDiDestroyDeviceInfoList(Info);
	*ppDeviceName = pDeviceName;
	return TRUE;
}

///////////////////////////////////////////////////////////////

HANDLE _HK_ClientDriverApiFindDriverName()
{
	BOOL bl;
	int count = 0;
	int index = 0;
	WCHAR DeviceName[MAX_PATH];
	WCHAR *pDeviceName = NULL;
	HANDLE hDevice = (HANDLE)-1;

	count = GetDeviceStackNameCount((struct _GUID *)&GUID_VUSBBUS);

	if (count == 0)
		goto exit;

	bl = GetDeviceStackName((struct _GUID *)&GUID_VUSBBUS, &pDeviceName, 0);

	if (bl == TRUE)
	{
		wcscpy(DeviceName, pDeviceName);
		free(pDeviceName);
		hDevice = CreateFile(DeviceName, FILE_READ_DATA | FILE_WRITE_DATA,
			0, NULL, OPEN_EXISTING, 0, NULL);
	}

exit:
	return hDevice;
}

BOOLEAN _HK_ClientDriverApiSetLEDEvent(HANDLE hDriverHandle, HANDLE hEvent)
{
	BOOLEAN bRet = FALSE;
	DWORD ret;

	if (hDriverHandle == (HANDLE)-1)
		goto exit;

	bRet = DeviceIoControl(hDriverHandle, IOCTL_BUSDRIVER_SENDKEYBOARD_LED_EVENT, &hEvent, sizeof(HANDLE), NULL, 0, &ret, NULL);

exit:
	return bRet;
}

BOOLEAN _HK_ClientDriverApiQueryIndicators(HANDLE hDriverHandle, HIDLED * pIndicators)
{
	BOOLEAN bRet = FALSE;
	DWORD ret;

	if (hDriverHandle == (HANDLE)-1)
		goto exit;

	bRet = DeviceIoControl(hDriverHandle, _IOCTL_KEYBOARD_QUERY_INDICATORS, NULL, 0,
		pIndicators, sizeof(KEYBOARD_HOOK_INDICATOR_PARAMETERS), &ret, NULL);

exit:
	return bRet;
}

BOOLEAN _HK_ClientDriverApiSimulateKey(HANDLE hDriverHandle, HIDKEYCODE * pKeyCode)
{
	BOOLEAN bRet = FALSE;
	DWORD ret;

	if (hDriverHandle == (HANDLE)-1)
		goto exit;

	bRet = DeviceIoControl(hDriverHandle, IOCTL_BUSDRIVER_SENDKEYBOARD, pKeyCode, sizeof(HIDKEYCODE), NULL, 0, &ret, NULL);

exit:
	return bRet;
}

BOOLEAN _HK_ClientDriverApiSimulateMouse(HANDLE hDriverHandle, HIDMOUSECODE * pMouseCode)
{
	BOOLEAN bRet = FALSE;
	DWORD ret;

	if (hDriverHandle == (HANDLE)-1)
		goto exit;

	bRet = DeviceIoControl(hDriverHandle, IOCTL_BUSDRIVER_SENDMOUSE, pMouseCode, sizeof(HIDMOUSECODE), NULL, 0, &ret, NULL);

exit:
	return bRet;
}

void _HK_ClientDriverApiInitialize()
{
}

void _HK_ClientDriverApiCloseHandle(HANDLE hDriverHandle)
{
	CloseHandle(hDriverHandle);
}