lostecho/edge-tts
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add some checks to ensure that it contains header length and drop struct

Browse Source
This commit is contained in:
rany2
2023-04-21 14:20:01 +03:00
parent bc781b2769
commit 6e733a306b
1 changed files with 15 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
src/edge_tts/communicate.py
View File

@@ -5,7 +5,6 @@ Communicate package.
import json import json
import re import re
import struct
import time import time
import uuid import uuid
from contextlib import nullcontext from contextlib import nullcontext
@@ -415,13 +414,22 @@ class Communicate:
"We received a binary message, but we are not expecting one." "We received a binary message, but we are not expecting one."
) )
if len(received.data) < 2:
raise UnexpectedResponse(
"We received a binary message, but it is missing the header length."
)
# See: https://github.com/microsoft/cognitive-services-speech-sdk-js/blob/d071d11d1e9f34d6f79d0ab6114c90eecb02ba1f/src/common.speech/WebsocketMessageFormatter.ts#L46 # See: https://github.com/microsoft/cognitive-services-speech-sdk-js/blob/d071d11d1e9f34d6f79d0ab6114c90eecb02ba1f/src/common.speech/WebsocketMessageFormatter.ts#L46
header_length = struct.unpack(">H", received.data[:2])[0] header_length = int.from_bytes(received.data[:2], "big")
if len(received.data) > header_length + 2: if len(received.data) < header_length + 2:
yield { raise UnexpectedResponse(
"type": "audio", "We received a binary message, but it is missing the audio data."
"data": received.data[header_length + 2 :], )
}
yield {
"type": "audio",
"data": received.data[header_length + 2 :],
}
audio_was_received = True audio_was_received = True
elif received.type == aiohttp.WSMsgType.ERROR: elif received.type == aiohttp.WSMsgType.ERROR:
raise WebSocketError( raise WebSocketError(