首页
社区
课程
招聘
[旧帖] 进程监控 0.00雪花
发表于: 2012-5-25 09:40 4164

[旧帖] 进程监控 0.00雪花

2012-5-25 09:40
4164
做进程监控实验的时候。。参考PRMonitor的效果
http://bbs.pediy.com/showthread.php?t=55092
用下面的模型做

在用户程序中建立EVENT对象。
使用DeviceIoControl将EVENT的HANDLE传递到驱动程序中。
在驱动程序中建立KEVENT对象,使用OBJECTREFENCE函数将EVENT对象链接起来。
下面就是在驱动中SETEVENT,在用户程序中WaitforSingleObject了。

我按照这个思路折腾了2天都没弄明白。。

这个模式求一份简单的可编译参考源码,最好有R3,RO,别是看雪PRMonitor源码。。

[课程]FART 脱壳王!加量不加价!FART作者讲授!

收藏
免费 0
支持
分享
最新回复 (3)
雪    币: 1753
活跃值: (840)
能力值: ( LV8,RANK:120 )
在线值:
发帖
回帖
粉丝
2
这样可否·!??

// 自定义函数的声明
NTSTATUS DispatchCreateClose(PDEVICE_OBJECT pDevObj, PIRP pIrp);
void DriverUnload(PDRIVER_OBJECT pDriverObj);
NTSTATUS DispatchIoctl(PDEVICE_OBJECT pDevObj, PIRP pIrp);
VOID ProcessCallback(IN HANDLE  hParentId, IN HANDLE  hProcessId, IN BOOLEAN bCreate);

// 驱动内部名称、符号连接名称、事件对象名称
#define DEVICE_NAME    L"\\Device\\devNTProcDrv"
#define LINK_NAME    L"\\DosDevices\\slNTProcDrv"
#define EVENT_NAME    L"\\BaseNamedObjects\\NTProcDrvProcessEvent"

typedef struct _DEVICE_EXTENSION  // 设备对象的私有存储
{
    HANDLE  hProcessHandle;  // 事件对象句柄
    PKEVENT ProcessEvent;  // 用户和内核通信的事件对象指针

    HANDLE  hPParentId;    // 在回调函数中保存进程信息,当用户程序请求时,传递过去
    HANDLE  hPProcessId;
    BOOLEAN bPCreate;
} DEVICE_EXTENSION, *PDEVICE_EXTENSION;

PDEVICE_OBJECT g_pDeviceObject;

// 驱动程序加载时调用DriverEntry例程
extern "C" NTSTATUS DriverEntry(PDRIVER_OBJECT pDriverObj, PUNICODE_STRING pRegistryString)
{
  NTSTATUS status = STATUS_SUCCESS;

  // 初始化各个派遣例程
  pDriverObj->MajorFunction[IRP_MJ_CREATE] = DispatchCreateClose;
  pDriverObj->MajorFunction[IRP_MJ_CLOSE] = DispatchCreateClose;
  pDriverObj->MajorFunction[IRP_MJ_DEVICE_CONTROL] = DispatchIoctl;
  pDriverObj->DriverUnload = DriverUnload;

    // 创建、初始化设备对象
  // 设备名称
  UNICODE_STRING ustrDevName;
  RtlInitUnicodeString(&ustrDevName, DEVICE_NAME);
  // 创建设备对象
  PDEVICE_OBJECT pDevObj;
  status = IoCreateDevice(pDriverObj,
        sizeof(DEVICE_EXTENSION), // 为设备扩展结构申请空间
        &ustrDevName,
        FILE_DEVICE_UNKNOWN,
        0,
        FALSE,
        &pDevObj);
  if(!NT_SUCCESS(status))
  {
    return status;
  }
  PDEVICE_EXTENSION pDevExt = (PDEVICE_EXTENSION)pDevObj->DeviceExtension;

    // 创建符号连接名称
  // 符号连接名称
  UNICODE_STRING ustrLinkName;
  RtlInitUnicodeString(&ustrLinkName, LINK_NAME);
  // 创建关联
  status = IoCreateSymbolicLink(&ustrLinkName, &ustrDevName);  
  if(!NT_SUCCESS(status))
  {
    IoDeleteDevice(pDevObj);  
    return status;
  }

  // 保存到设备对象的指针,下面在进程回调函数中还要使用
  g_pDeviceObject = pDevObj;

  // 为了用户模式进程能够监视,创建事件对象
  UNICODE_STRING  uszProcessEventString;
    RtlInitUnicodeString(&uszProcessEventString, EVENT_NAME);
    pDevExt->ProcessEvent = IoCreateNotificationEvent(&uszProcessEventString, &pDevExt->hProcessHandle);

    // 设置它为非受信状态
    KeClearEvent(pDevExt->ProcessEvent);
    // 设置回调例程
    status = PsSetCreateProcessNotifyRoutine(ProcessCallback, FALSE);
  
  return status;
}

void DriverUnload(PDRIVER_OBJECT pDriverObj)
{  
  DbgPrint("Driver Unload\n");
  // 移除进程回调例程
    PsSetCreateProcessNotifyRoutine(ProcessCallback, TRUE);

  // 删除符号连接名称
  UNICODE_STRING strLink;
  RtlInitUnicodeString(&strLink, LINK_NAME);
  IoDeleteSymbolicLink(&strLink);

  // 删除设备对象
  IoDeleteDevice(pDriverObj->DeviceObject);
}

// 处理IRP_MJ_CREATE、IRP_MJ_CLOSE功能代码
NTSTATUS DispatchCreateClose(PDEVICE_OBJECT pDevObj, PIRP pIrp)
{
  pIrp->IoStatus.Status = STATUS_SUCCESS;
  // 完成此请求
  IoCompleteRequest(pIrp, IO_NO_INCREMENT);

  return STATUS_SUCCESS;
}

// I/O控制派遣例程
NTSTATUS DispatchIoctl(PDEVICE_OBJECT pDevObj, PIRP pIrp)
{
  DbgPrint(" ProcDrv: DispatchIoctl()... \n");

  // 假设失败
  NTSTATUS status = STATUS_INVALID_DEVICE_REQUEST;

  // 取得此IRP(pIrp)的I/O堆栈指针
  PIO_STACK_LOCATION pIrpStack = IoGetCurrentIrpStackLocation(pIrp);

  // 取得设备扩展结构指针
  PDEVICE_EXTENSION pDevExt = (PDEVICE_EXTENSION)pDevObj->DeviceExtension;

  // 取得I/O控制代码
  ULONG uIoControlCode = pIrpStack->Parameters.DeviceIoControl.IoControlCode;
  // 取得I/O缓冲区指针和它的长度
  PCALLBACK_INFO pCallbackInfo = (PCALLBACK_INFO)pIrp->AssociatedIrp.SystemBuffer;
  ULONG uInSize = pIrpStack->Parameters.DeviceIoControl.InputBufferLength;
  ULONG uOutSize = pIrpStack->Parameters.DeviceIoControl.OutputBufferLength;

  switch(uIoControlCode)
  {
  case IOCTL_NTPROCDRV_GET_PROCINFO:  // 向用户程序返回有事件发生的进程的信息
    {
      if(uOutSize >= sizeof(CALLBACK_INFO))
      {
        pCallbackInfo->hParentId  = pDevExt->hPParentId;
                pCallbackInfo->hProcessId = pDevExt->hPProcessId;
                pCallbackInfo->bCreate    = pDevExt->bPCreate;
                status = STATUS_SUCCESS;
      }
    }
    break;
  }

  if(status == STATUS_SUCCESS)
  {
    pIrp->IoStatus.Information = uOutSize;
  }
  else
  {
    pIrp->IoStatus.Information = 0;
  }

  // 完成请求
  pIrp->IoStatus.Status = status;
  IoCompleteRequest(pIrp, IO_NO_INCREMENT);

  return status;
}

// 进程回调函数
VOID ProcessCallback(IN HANDLE  hParentId, IN HANDLE  hProcessId, IN BOOLEAN bCreate)
{
  DbgPrint("Call ProcessCallback()\n");
  // 得到设备扩展结构的指针
    PDEVICE_EXTENSION pDevExt =  (PDEVICE_EXTENSION)g_pDeviceObject->DeviceExtension;

  // 安排当前值到设备扩展结构
  // 用户模式应用程序将使用DeviceIoControl调用把它取出
    pDevExt->hPParentId  = hParentId;
    pDevExt->hPProcessId = hProcessId;
    pDevExt->bPCreate    = bCreate;
   
    if(pDevExt->bPCreate)
    {
      DbgPrint("create: %d\n",pDevExt->hPProcessId);
    }
    else
    {
      DbgPrint("Kill: %d\n",pDevExt->hPProcessId);
    }

  // 触发这个事件,以便任何正在监听的用户程序知道有事情发生了。
  // 用户模式下的应用程序不能重置KM事件,所以我们要在这里触发它
    KeSetEvent(pDevExt->ProcessEvent, 0, FALSE);
    KeClearEvent(pDevExt->ProcessEvent);
}
2012-5-27 11:43
0
雪    币: 239
活跃值: (133)
能力值: ( LV5,RANK:60 )
在线值:
发帖
回帖
粉丝
3
我已经读过你那帖子了
顺便提个,如果在应用层创建了一个无名对象,传入R0,然后waitforsi。。

在R0中可以设置它让应用层的wait获得通知呢?
具体怎么用,给个参考或者推荐网址看看
2012-5-27 11:53
0
雪    币: 239
活跃值: (133)
能力值: ( LV5,RANK:60 )
在线值:
发帖
回帖
粉丝
4
其实我希望的是

1.R0创建事件  设置回调,等待进程创建关闭,                                                         4.  R0设置 事件                                                                       6.R0再次获得通知,是否执行

                                                 2. R3打开事件,等待事件触发                                              5.  R3获得通知,决定是否要执行,传入R0。。。
                                                                                               3. 某一进程创建                               

你的也有一定参考价值,我再创建的事件调试下。。
2012-5-27 12:01
0
游客
登录 | 注册 方可回帖
返回
//