56 lines
2.2 KiB
Diff
56 lines
2.2 KiB
Diff
From 11555828d2cf289b350baba5ad1f462f10b80b76 Mon Sep 17 00:00:00 2001
|
|
From: akallabeth <akallabeth@posteo.net>
|
|
Date: Thu, 13 Oct 2022 08:47:51 +0200
|
|
Subject: [PATCH] Fixed missing input buffer length check in urbdrc
|
|
|
|
(cherry picked from commit 497df00f741dd4fc89292aaef2db7368aee45d0d)
|
|
---
|
|
channels/urbdrc/client/data_transfer.c | 20 ++++++++++++++++++++
|
|
1 file changed, 20 insertions(+)
|
|
|
|
diff --git a/channels/urbdrc/client/data_transfer.c b/channels/urbdrc/client/data_transfer.c
|
|
index d8725c02cf3..aabeef84752 100644
|
|
--- a/channels/urbdrc/client/data_transfer.c
|
|
+++ b/channels/urbdrc/client/data_transfer.c
|
|
@@ -247,6 +247,10 @@ static UINT urbdrc_process_io_control(IUDEVICE* pdev, URBDRC_CHANNEL_CALLBACK* c
|
|
|
|
Stream_Read_UINT32(s, OutputBufferSize);
|
|
Stream_Read_UINT32(s, RequestId);
|
|
+
|
|
+ if (OutputBufferSize > UINT32_MAX - 4)
|
|
+ return ERROR_INVALID_DATA;
|
|
+
|
|
InterfaceId = ((STREAM_ID_PROXY << 30) | pdev->get_ReqCompletion(pdev));
|
|
out = urb_create_iocompletion(InterfaceId, MessageId, RequestId, OutputBufferSize + 4);
|
|
|
|
@@ -726,6 +730,15 @@ static UINT urb_bulk_or_interrupt_transfer(IUDEVICE* pdev, URBDRC_CHANNEL_CALLBA
|
|
Stream_Read_UINT32(s, TransferFlags); /** TransferFlags */
|
|
Stream_Read_UINT32(s, OutputBufferSize);
|
|
EndpointAddress = (PipeHandle & 0x000000ff);
|
|
+
|
|
+ if (transferDir == USBD_TRANSFER_DIRECTION_OUT)
|
|
+ {
|
|
+ if (!Stream_CheckAndLogRequiredLength(TAG, s, OutputBufferSize))
|
|
+ {
|
|
+ return ERROR_INVALID_DATA;
|
|
+ }
|
|
+ }
|
|
+
|
|
/** process TS_URB_BULK_OR_INTERRUPT_TRANSFER */
|
|
return pdev->bulk_or_interrupt_transfer(
|
|
pdev, callback, MessageId, RequestId, EndpointAddress, TransferFlags, noAck,
|
|
@@ -810,6 +823,13 @@ static UINT urb_isoch_transfer(IUDEVICE* pdev, URBDRC_CHANNEL_CALLBACK* callback
|
|
packetDescriptorData = Stream_Pointer(s);
|
|
Stream_Seek(s, NumberOfPackets * 12);
|
|
Stream_Read_UINT32(s, OutputBufferSize);
|
|
+
|
|
+ if (transferDir == USBD_TRANSFER_DIRECTION_OUT)
|
|
+ {
|
|
+ if (!Stream_CheckAndLogRequiredLength(TAG, s, OutputBufferSize))
|
|
+ return ERROR_INVALID_DATA;
|
|
+ }
|
|
+
|
|
return pdev->isoch_transfer(
|
|
pdev, callback, MessageId, RequestId, EndpointAddress, TransferFlags, StartFrame,
|
|
ErrorCount, noAck, packetDescriptorData, NumberOfPackets, OutputBufferSize,
|