Do not attempt to intercept unexisting files

GoSecure · Nov 15, 2022 · e7022cf · e7022cf
1 parent 7361859
commit e7022cf
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/pyrdp/mitm/DeviceRedirectionMITM.py b/pyrdp/mitm/DeviceRedirectionMITM.py
@@ -225,8 +225,9 @@ def handleCreateResponse(self, request: DeviceCreateRequestPDU, response: Device
 
         isFileRead = request.desiredAccess & (FileAccessMask.GENERIC_READ | FileAccessMask.FILE_READ_DATA) != 0
         isDirectory = request.createOptions & CreateOption.FILE_NON_DIRECTORY_FILE == 0
+        fileExists = (response.ioStatus != NTSTATUS.STATUS_NO_SUCH_FILE)
 
-        if isFileRead and not isDirectory:
+        if fileExists and isFileRead and not isDirectory:
             mapping = FileMapping.generate(request.path, self.config.fileDir, self.deviceRoot(response.deviceID), self.log)
             key = (response.deviceID, response.fileID)
             self.mappings[key] = mapping