ATAPI（磁盘端口驱动）级文件保护简单实现

最新推荐文章于 2025-10-30 09:53:30 发布

原创最新推荐文章于 2025-10-30 09:53:30 发布 · 1.2k 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#磁盘 #object #io

windows底层核心編程专栏收录该内容

743 篇文章

订阅专栏

本文介绍了一个用于检查IRP中特定文件对象的函数IfIrpHasFobj，并展示了如何使用此函数来验证IRP是否关联了指定文件。此外，还提供了一个在代理ATAPI SCSI设备驱动时使用该函数的例子。

define IoGetIrpStackLocation( Irp , Level) (/
(Irp)->Tail.Overlay.CurrentStackLocation + Level )
BOOL IfIrpHasFobj(PIRP pIrp , LPCWSTR FileName)
{
ULONG i , j ;
PIO_STACK_LOCATION irpStack ;
PVOID NamePool ;

for ( i = 0 ; i < (ULONG)(pIrp->StackCount - pIrp->CurrentLocation + 1) ; i ++)
{
irpStack = IoGetIrpStackLocation(pIrp , i );
if (irpStack->FileObject)
{
if (irpStack->FileObject->FileName.Buffer)
{
NamePool = ExAllocatePool(NonPagedPool , irpStack->FileObject->FileName.Length + 2);
if (NamePool)
{
KDMSG(("%ws" , irpStack->FileObject->FileName.Buffer));
RtlCopyMemory(NamePool ,
irpStack->FileObject->FileName.Buffer ,
irpStack->FileObject->FileName.Length + 2);
_wcsupr((wchar_t*)NamePool);
if (wcsstr((wchar_t*)NamePool , FileName))
{
ExFreePool(NamePool);
return TRUE ;
}

ExFreePool(NamePool);
}

}
}

}
return FALSE ;
}

NTSTATUS ProxyAtapiScsi(PDEVICE_OBJECT devobj , PIRP pIrp)
{
NTSTATUS stat ;
if (IfIrpHasFobj(pIrp , L"123.TXT"))
{
pIrp->IoStatus.Status = 0xc0000001;
pIrp->IoStatus.Information = 0 ;

IoCompleteRequest(pIrp , IO_NO_INCREMENT);
return 0xc0000001;
}

__asm
{
push pIrp
push devobj
call OldAtapiScsiDispatch
mov stat ,eax
}
return stat ;
}