handle status other than STATUS_PENDING from FsRtlCheckOplock()

dokan-dev · Jul 5, 2016 · 0c64b84 · 0c64b84
1 parent 9527353
commit 0c64b84
Show file tree

Hide file tree

Showing 5 changed files with 38 additions and 12 deletions.
diff --git a/sys/cleanup.c b/sys/cleanup.c
@@ -119,8 +119,12 @@ Return Value:
     //  if FsRtlCheckOplock returns STATUS_PENDING the IRP has been posted
     //  to service an oplock break and we need to leave now.
     //
-    if (status == STATUS_PENDING) {
-      DDbgPrint("   FsRtlCheckOplock returned STATUS_PENDING\n");
+    if (status != STATUS_SUCCESS) {
+      if (status == STATUS_PENDING) {
+        DDbgPrint("   FsRtlCheckOplock returned STATUS_PENDING\n");
+      } else {
+        DokanFreeEventContext(eventContext);
+      }
       __leave;
     }
 

diff --git a/sys/fileinfo.c b/sys/fileinfo.c
@@ -467,8 +467,12 @@ DokanDispatchSetInformation(__in PDEVICE_OBJECT DeviceObject, __in PIRP Irp) {
     //  if FsRtlCheckOplock returns STATUS_PENDING the IRP has been posted
     //  to service an oplock break and we need to leave now.
     //
-    if (status == STATUS_PENDING) {
-      DDbgPrint("   FsRtlCheckOplock returned STATUS_PENDING\n");
+    if (status != STATUS_SUCCESS) {
+      if (status == STATUS_PENDING) {
+        DDbgPrint("   FsRtlCheckOplock returned STATUS_PENDING\n");
+      } else {
+        DokanFreeEventContext(eventContext);
+      }
       __leave;
     }
 

diff --git a/sys/flush.c b/sys/flush.c
@@ -85,8 +85,12 @@ DokanDispatchFlush(__in PDEVICE_OBJECT DeviceObject, __in PIRP Irp) {
     //  if FsRtlCheckOplock returns STATUS_PENDING the IRP has been posted
     //  to service an oplock break and we need to leave now.
     //
-    if (status == STATUS_PENDING) {
-      DDbgPrint("   FsRtlCheckOplock returned STATUS_PENDING\n");
+    if (status != STATUS_SUCCESS) {
+      if (status == STATUS_PENDING) {
+        DDbgPrint("   FsRtlCheckOplock returned STATUS_PENDING\n");
+      } else {
+        DokanFreeEventContext(eventContext);
+      }
       __leave;
     }
 

diff --git a/sys/read.c b/sys/read.c
@@ -203,8 +203,12 @@ Return Value:
       //  if FsRtlCheckOplock returns STATUS_PENDING the IRP has been posted
       //  to service an oplock break and we need to leave now.
       //
-      if (status == STATUS_PENDING) {
-        DDbgPrint("   FsRtlCheckOplock returned STATUS_PENDING\n");
+      if (status != STATUS_SUCCESS) {
+        if (status == STATUS_PENDING) {
+          DDbgPrint("   FsRtlCheckOplock returned STATUS_PENDING\n");
+        } else {
+          DokanFreeEventContext(eventContext);
+        }
         __leave;
       }
 

diff --git a/sys/write.c b/sys/write.c
@@ -196,8 +196,12 @@ DokanDispatchWrite(__in PDEVICE_OBJECT DeviceObject, __in PIRP Irp) {
         //  if FsRtlCheckOplock returns STATUS_PENDING the IRP has been posted
         //  to service an oplock break and we need to leave now.
         //
-        if (status == STATUS_PENDING) {
-          DDbgPrint("   FsRtlCheckOplock returned STATUS_PENDING\n");
+        if (status != STATUS_SUCCESS) {
+          if (status == STATUS_PENDING) {
+            DDbgPrint("   FsRtlCheckOplock returned STATUS_PENDING\n");
+          } else {
+            DokanFreeEventContext(eventContext);
+          }
           __leave;
         }
       }
@@ -248,8 +252,14 @@ DokanDispatchWrite(__in PDEVICE_OBJECT DeviceObject, __in PIRP Irp) {
         //  if FsRtlCheckOplock returns STATUS_PENDING the IRP has been posted
         //  to service an oplock break and we need to leave now.
         //
-        if (status == STATUS_PENDING) {
-          DDbgPrint("   FsRtlCheckOplock returned STATUS_PENDING\n");
+        if (status != STATUS_SUCCESS) {
+          if (status == STATUS_PENDING) {
+            DDbgPrint("   FsRtlCheckOplock returned STATUS_PENDING\n");
+          } else {
+            DokanFreeEventContext(requestContext);
+            Irp->Tail.Overlay.DriverContext[DRIVER_CONTEXT_EVENT] = 0;
+            DokanFreeEventContext(eventContext);
+          }
           __leave;
         }
       }