Merge pull request #375 from GoSecure/optional-virtual-channels

Not longer assuming every connection will have VirtualChannels
GoSecure · Nov 26, 2021 · 8062864 · 8062864
2 parents 4edbae0 + fcf60e1
commit 8062864
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