Not longer assuming every connection will have VirtualChannels

Improved the situation with #294
GoSecure · Nov 26, 2021 · fcf60e1 · fcf60e1
1 parent 9401b53
commit fcf60e1
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/pyrdp/mitm/SlowPathMITM.py b/pyrdp/mitm/SlowPathMITM.py
@@ -109,5 +109,6 @@ def onDemandActive(self, pdu: DemandActivePDU):
             supported[Order.TS_NEG_DRAWNINEGRID_INDEX] = 0
             orders.orderSupport = supported
 
-        pdu.parsedCapabilitySets[CapabilityType.CAPSTYPE_VIRTUALCHANNEL].flags = \
-            VirtualChannelCompressionFlag.VCCAPS_NO_COMPR
+        if CapabilityType.CAPSTYPE_VIRTUALCHANNEL in pdu.parsedCapabilitySets:
+            pdu.parsedCapabilitySets[CapabilityType.CAPSTYPE_VIRTUALCHANNEL].flags = \
+                VirtualChannelCompressionFlag.VCCAPS_NO_COMPR