moonlight-stream/moonlight-common-c
1
0
Fork 0
mirror of https://github.com/moonlight-stream/moonlight-common-c.git synced 2025-07-01 23:35:58 +00:00
Code Issues Packages Projects Releases Wiki Activity

Improve encrypted RTSP message validation

Browse Source
This commit is contained in:
Cameron Gutman 2024-02-03 14:39:12 -06:00
parent 955f13a18d
commit 7ab34e709a
1 changed files with 26 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
src/RtspConnection.c
View File

@ -157,19 +157,44 @@ static bool unsealRtspMessage(char* rawMessage, int rawMessageLen, PRTSP_MESSAGE
int decryptedMessageLen;
bool success;
// If the server just closed the connection without responding with anything,
// there's no point in proceeding any further trying to parse it.
if (rawMessageLen == 0) {
return false;
}
if (encryptedRtspEnabled) {
PENC_RTSP_HEADER encryptedMessage;
uint32_t seq;
uint32_t typeAndLen;
uint32_t len;
uint8_t iv[12] = { 0 };
if (rawMessageLen <= (int)sizeof(ENC_RTSP_HEADER)) {
Limelog("RTSP encrypted header too small\n");
return false;
}
encryptedMessage = (PENC_RTSP_HEADER)rawMessage;
seq = BE32(encryptedMessage->sequenceNumber);
typeAndLen = BE32(encryptedMessage->typeAndLength);
if (!(typeAndLen & ENCRYPTED_RTSP_BIT)) {
Limelog("Rejecting unencrypted RTSP message\n");
return false;
}
len = typeAndLen & ~ENCRYPTED_RTSP_BIT;
if (len + sizeof(ENC_RTSP_HEADER) > rawMessageLen) {
Limelog("Rejecting partial encrypted RTSP message\n");
return false;
}
else if (len + sizeof(ENC_RTSP_HEADER) < rawMessageLen) {
Limelog("Rejecting encrypted RTSP message with excess data\n");
return false;
}
// Populate the IV in little endian byte order
seq = BE32(encryptedMessage->sequenceNumber);
iv[3] = (uint8_t)(seq >> 24);
iv[2] = (uint8_t)(seq >> 16);
iv[1] = (uint8_t)(seq >> 8);