通过Firmware检测虚拟机环境

// 不支持XP 5.2 from Windows Server 2003 SP1, and higher
//
#define SystemFirmwareTableInformation 76
// 根据EnumSystemFirmwareTables与GetSystemFirmwareTable封装;
//
NTSTATUS
OSLibEnumSystemFirmwareTables(
__in ULONG FirmwareTableProviderSignature,
__out PSYSTEM_FIRMWARE_TABLE_INFORMATION* FirmWareTableInformation
)
{
    NTSTATUS Status = STATUS_SUCCESS;
    ULONG ReturnLength = 0;
    PSYSTEM_FIRMWARE_TABLE_INFORMATION FirmWareTable = NULL;

    do
    {
        FirmWareTable = (PSYSTEM_FIRMWARE_TABLE_INFORMATION)ExAllocatePoolWithTag(
            PagedPool,
            sizeof(SYSTEM_FIRMWARE_TABLE_INFORMATION),
            KLIB_KAPI_TAG
            );
        if (FirmWareTable == NULL)
        {
            Status = STATUS_INSUFFICIENT_RESOURCES;
            break;
        }

        while (TRUE)
        {
            FirmWareTable->ProviderSignature = FirmwareTableProviderSignature;
            FirmWareTable->TableID = 0;
            FirmWareTable->TableBufferLength = ReturnLength;
            FirmWareTable->Action = SystemFirmwareTable_Enumerate;

            Status = ZwQuerySystemInformation(
                SystemFirmwareTableInformation,
                FirmWareTable,
                ReturnLength + sizeof(SYSTEM_FIRMWARE_TABLE_INFORMATION),
                &ReturnLength
                );
            if (NT_SUCCESS(Status))
            {
                break;
            }

            if (Status == STATUS_BUFFER_TOO_SMALL || Status == STATUS_INFO_LENGTH_MISMATCH)
            {
                ReturnLength = FirmWareTable->TableBufferLength;
                ExFreePoolWithTag(FirmWareTable, KLIB_KAPI_TAG);
                FirmWareTable = NULL;

                FirmWareTable = (PSYSTEM_FIRMWARE_TABLE_INFORMATION)ExAllocatePoolWithTag(
                    PagedPool,
                    sizeof(SYSTEM_FIRMWARE_TABLE_INFORMATION) + ReturnLength,
                    KLIB_KAPI_TAG
                    );
                if (FirmWareTable == NULL)
                {
                    Status = STATUS_INSUFFICIENT_RESOURCES;
                    break;
                }
            }
            else
            {
                break;
            }

        }

    } while (FALSE);

    if (Status != STATUS_SUCCESS)
    {
        ExFreePoolWithTag(FirmWareTable, KLIB_KAPI_TAG);
        FirmWareTable = NULL;
    }
    else
    {
        *FirmWareTableInformation = FirmWareTable;
    }

    return Status;
}

NTSTATUS
OSLibGetSystemFirmwareTables(
__in ULONG FirmwareTableProviderSignature,
__in ULONG FirmwareTableID,
__out PSYSTEM_FIRMWARE_TABLE_INFORMATION* FirmWareTableInformation
)
{
    NTSTATUS Status = STATUS_SUCCESS;
    ULONG ReturnLength = 0;
    PSYSTEM_FIRMWARE_TABLE_INFORMATION FirmWareTable = NULL;

    do
    {
        FirmWareTable = (PSYSTEM_FIRMWARE_TABLE_INFORMATION)ExAllocatePoolWithTag(
            PagedPool,
            sizeof(SYSTEM_FIRMWARE_TABLE_INFORMA
  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值