Merge pull request #1184 from HUGEIT/master

fix #1152 crash by putting mutex on sendRetryRequest()
This commit is contained in:
Adhiraj Singh
2022-01-25 19:36:00 +05:30
committed by GitHub

View File

@@ -3,7 +3,7 @@ import { proto } from '../../WAProto'
import { KEY_BUNDLE_TYPE } from '../Defaults' import { KEY_BUNDLE_TYPE } from '../Defaults'
import { Chat, GroupMetadata, MessageUserReceipt, ParticipantAction, SocketConfig, WAMessageStubType } from '../Types' import { Chat, GroupMetadata, MessageUserReceipt, ParticipantAction, SocketConfig, WAMessageStubType } from '../Types'
import { decodeMessageStanza, downloadAndProcessHistorySyncNotification, encodeBigEndian, generateSignalPubKey, toNumber, xmppPreKey, xmppSignedPreKey } from '../Utils' import { decodeMessageStanza, downloadAndProcessHistorySyncNotification, encodeBigEndian, generateSignalPubKey, toNumber, xmppPreKey, xmppSignedPreKey } from '../Utils'
import { makeKeyedMutex } from '../Utils/make-mutex' import { makeKeyedMutex, makeMutex } from '../Utils/make-mutex'
import { areJidsSameUser, BinaryNode, BinaryNodeAttributes, getAllBinaryNodeChildren, getBinaryNodeChildren, isJidGroup, jidDecode, jidEncode, jidNormalizedUser } from '../WABinary' import { areJidsSameUser, BinaryNode, BinaryNodeAttributes, getAllBinaryNodeChildren, getBinaryNodeChildren, isJidGroup, jidDecode, jidEncode, jidNormalizedUser } from '../WABinary'
import { makeChatsSocket } from './chats' import { makeChatsSocket } from './chats'
import { extractGroupMetadata } from './groups' import { extractGroupMetadata } from './groups'
@@ -38,9 +38,12 @@ export const makeMessagesRecvSocket = (config: SocketConfig) => {
resyncMainAppState, resyncMainAppState,
} = sock } = sock
/** the mutex ensures that the notifications (receipts, messages etc.) are processed in order */ /** this mutex ensures that the notifications (receipts, messages etc.) are processed in order */
const processingMutex = makeKeyedMutex() const processingMutex = makeKeyedMutex()
/** this mutex ensures that each retryRequest will wait for the previous one to finish */
const retryMutex = makeMutex()
const msgRetryMap = config.msgRetryCounterMap || { } const msgRetryMap = config.msgRetryCounterMap || { }
const historyCache = new Set<string>() const historyCache = new Set<string>()
@@ -354,7 +357,9 @@ export const makeMessagesRecvSocket = (config: SocketConfig) => {
{ msgId: msg.key.id, params: msg.messageStubParameters }, { msgId: msg.key.id, params: msg.messageStubParameters },
'failure in decrypting message' 'failure in decrypting message'
) )
await sendRetryRequest(stanza) retryMutex.mutex(
async () => await sendRetryRequest(stanza)
)
} else { } else {
await sendMessageAck(stanza, { class: 'receipt' }) await sendMessageAck(stanza, { class: 'receipt' })
// no type in the receipt => message delivered // no type in the receipt => message delivered