基于文件过滤驱动的文件重定向

转自：http://mzf2008.blog.163.com/blog/static/3559978620114153254458/

 

NTSTATUS
SfCreate (
    IN PDEVICE_OBJECT DeviceObject,
    IN PIRP Irp
    )
{
 PSFILTER_DEVICE_EXTENSION deviceExtension; 
 PDEVICE_OBJECT attachedDevice; 
 PIO_STACK_LOCATION irpSp = IoGetCurrentIrpStackLocation(Irp); 
 NTSTATUS status; 
 PFILE_OBJECT FileObject = irpSp->FileObject; 
 
 UNICODE_STRING sourceName; 
 UNICODE_STRING destinationName;

 //如果是控制设备就结束此IRP
 if(DeviceObject == gSFilterControlDeviceObject) 
 { 
  KdPrint(("if(DeviceObject == gSFilterControlDeviceObject)/n"));
  Irp->IoStatus.Status = STATUS_INVALID_DEVICE_REQUEST; 
  Irp->IoStatus.Information = 0; 
  status = Irp->IoStatus.Status; 
  IoCompleteRequest(Irp, IO_NO_INCREMENT); 
  return status; 
 } 
 
 ASSERT(IS_MY_DEVICE_OBJECT(DeviceObject));

 deviceExtension = DeviceObject->DeviceExtension; 
 attachedDevice = deviceExtension->AttachedToDeviceObject;

 RtlInitUnicodeString( &sourceName, L"//1.txt" );

 if (FileObject && RtlCompareUnicodeString( &sourceName, &FileObject->FileName, TRUE ) == 0) 
 {
  // 释放掉FileObject->FileName.Buffer中的内存, 重新分配之
  ExFreePool(FileObject->FileName.Buffer); 
  FileObject->FileName.Length = 0; 
  FileObject->FileName.MaximumLength = 256;
  FileObject->FileName.Buffer=(PWCHAR)ExAllocatePool(NonPagedPool, 256); 
  
  // 设定目标文件的全路径, 可以跨卷
  RtlInitUnicodeString( &destinationName, L"//Device//HarddiskVolume1//2.txt" ); 
  RtlCopyUnicodeString(&FileObject->FileName, &destinationName); 
  
  FileObject->FileName.Buffer[FileObject->FileName.Length/2] = UNICODE_NULL;

  // 设定IoStatus值 为 STATUS_REPARSE;
  Irp->IoStatus.Status = STATUS_REPARSE; 
  Irp->IoStatus.Information = IO_REPARSE; 
  IoCompleteRequest( Irp, IO_NO_INCREMENT ); 
  
  return STATUS_REPARSE;  
 } 
 else 
 { 
  IoSkipCurrentIrpStackLocation(Irp); 
  return IoCallDriver(attachedDevice,Irp); 
 } 
}

 

1.在未加载驱动的时候, 创建两个空白文件, 大小为0KB, 分别为1.txt 和 2.txt

 

2.加载驱动, 并打开1.txt 进行编辑.

 

3.保存文件后, 1.txt的大小没有变化, 而2.txt 的大小变成了1KB

 

4.关闭驱动后, 查看两个文本, 2.txt的内容是我们刚刚写入的, 文件重定向成功~

 

总结如下:

在做文件重定向时,确实应该在IRP_MJ_CREATE里进行重定向,而且应该这样做: 
1,先释放原FileObject->FileName;

2,重新分配一个UNICODE_STRING,并将其Buffer设置为你想打开的文件全路径, 类似 "//Device//HarddiskVolume1//2.txt" ;

3,Irp->IoStatus.Status=STATUS_REPARSE; 
Irp->IoStatus.Informiation=IO_REPARSE; 
IoCompleteRequeset(Irp,IO_NO_INCEMENT); 
return STATUS_REPARSE;