请教，关于minifilter中StreamContext的用法

unsettled

请教，关于minifilter中StreamContext的用法 [复制链接]

我想在PostCreate中设置上下文，在PreRead/PostRead或PreWrite/PostWrite能得到该上下文.

顺序：
1.在FLT_CONTEXT_REGISTRATION中定义。如:
     { FLT_STREAM_CONTEXT,
       0,
       CleanupStreamContext,
       sizeof(FILE_CONTEXT),
       CONTEXT_TAG },
2.InstanceSetup中，没做FltAllocateContext和FltSetStreamContext
3.FltGetStreamContext函数，

1. 
   
2. Comments
   
3. FltGetStreamContext retrieves a context that was set for a file stream by a given minifilter driver instance.
   
4. 
   
5. FltGetStreamContext increments the reference count on the context that the Context parameter points to. When this context pointer is no longer needed, the caller must decrement its reference count by calling FltReleaseContext. Thus every successful call to FltGetStreamContext must be matched by a subsequent call to FltReleaseContext.
   
6. 
   
7. 不好意思，英语不太好，FltGetStreamContext成功调用，是否需要FltReleaseContext或FltReferenceContext，因为别的函数可能需要用到它(看4或5)
   

复制代码

4.PostCreate根据需要跟踪的文件，设置FltAllocateContext及FltSetStreamContext。
我参照ctx例子，程序大概的意思如下：

1. 
   
2. typedef struct _FILE_CONTEXT
   
3. {   
   
4.       BOOLEAN EncryptFlagExist;
   
5. 
   
6. } FILE_CONTEXT, *PFILE_CONTEXT;
   
7. 
   
8. PFILE_CONTEXT fileCtxPtr ;
   
9. NTSTATUS status ;
   
10. status = FltGetStreamContext( FltObjects->Instance,
    
11.                                   FltObjects->FileObject,
    
12.                                   &streamContext );
    
13. 
    
14. if (!NT_SUCCESS(status)&&(status==STATUS_NOT_FOUND))
    
15. {
    
16.    status = FltAllocateContext( Globals.Filter,
    
17.         FLT_STREAM_CONTEXT,
    
18.                                         sizeof(FILE_CONTEXT),
    
19.         PagedPool,
    
20.         &fileCtxPtr );
    
21.    if (!NT_SUCCESS( status ))
    
22.    {
    
23.     KdPrint(("[Ctx]: Failed to create stream context"));
    
24.    }
    
25.    else
    
26.   {
    
27.       fileCtxPtr->EncryptFlagExist= IsEncrypted;
    
28.       status=FltSetStreamContext (FltObjects->Instance,
    
29.                 FltObjects->FileObject,
    
30.                 FLT_SET_CONTEXT_REPLACE_IF_EXISTS,
    
31.                 fileCtxPtr ,
    
32.                 NULL);
    
33.   if (!NT_SUCCESS(status)) FltReleaseContext( fileCtxPtr ); [color=#6666CC]//是否需要这句,因为在PreRead或PreWrite需要用到它[/color]
    
34.   }
    
35. }
    
36. else
    
37. {
    
38.     //FltReleaseContext( fileCtxPtr ); [color=#6666CC]//同样，是否需要这句，因为在PreRead或PreWrite需要用到它[/color]
    
39. }
    
40. 
    

复制代码

5.在PreRead中，程序如下：

1. 
   
2. PFILE_CONTEXT fileCtxPtr ;
   
3. NTSTATUS status ;
   
4. status = FltGetStreamContext( FltObjects->Instance,
   
5.                                       FltObjects->FileObject,
   
6.                                       &fileCtx );
   
7. if (!NT_SUCCESS(status)) {
   
8.             KdPrint(("PreRead:  Error getting stream context, status=%x\n",status) );
   
9.       //需要FltReleaseContext(fileCtx)还是FltReferenceContext(fileCtx)吗？
   
10.    return FLT_PREOP_SUCCESS_NO_CALLBACK;
    
11. }
    
12. else
    
13. {
    
14.    //使用fileCtx 的值
    
15. .....
    
16.       *CompletionContext = fileCtx;
    
17. 
    
18.     //之后需要FltReleaseContext(fileCtx)还是FltReferenceContext(fileCtx)吗？
    
19. 
    
20.     return FLT_PREOP_SUCCESS_WITH_CALLBACK;
    
21. }
    
22. 
    

复制代码

6.在PreClose或PreCleanup中，清除该PFILE_CONTEXT如何做？

1. 
   
2. VOID CleanupFileContext(
   
3.     __in PFLT_CONTEXT Context,
   
4.     __in FLT_CONTEXT_TYPE ContextType
   
5.     )
   
6. {
   
7. 这样做吗？
   
8. PFILE_CONTEXT fileCtx=Context;
   
9. FltReleaseContext(fileCtx);
   
10. 
    
11. 还是要
    
12. if (ContextType==FLT_STREAM_CONTEXT)
    
13. FltDeleteContext(Context);
    
14. }
    

复制代码