franpcode/Baileys
1
0
Fork 0
mirror of https://github.com/FranP-code/Baileys.git synced 2025-10-13 00:32:22 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update Utils.ts

Browse Source
This commit is contained in:
Adhiraj
2020-07-11 20:56:24 +05:30
parent 44bd50cf4d
commit 9061f347af
1 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
src/WAClient/Utils.ts
View File

@@ -31,6 +31,9 @@ export function getNotificationType(message: WAMessage): [string, MessageType?]
} }
/** generates all the keys required to encrypt/decrypt & sign a media message */ /** generates all the keys required to encrypt/decrypt & sign a media message */
export function getMediaKeys(buffer, mediaType: MessageType) { export function getMediaKeys(buffer, mediaType: MessageType) {
if (typeof buffer === 'string') {
buffer = Buffer.from (buffer.replace('data:;base64,', ''), 'base64')
}
// expand using HKDF to 112 bytes, also pass in the relevant app info // expand using HKDF to 112 bytes, also pass in the relevant app info
const expandedMediaKey = hkdf(buffer, 112, HKDFInfoKeys[mediaType]) const expandedMediaKey = hkdf(buffer, 112, HKDFInfoKeys[mediaType])
return { return {
@@ -131,7 +134,7 @@ export async function decodeMediaMessage(message: WAMessageContent, filename: st
if (sign.equals(mac)) { if (sign.equals(mac)) {
return aesDecryptWithIV(file, mediaKeys.cipherKey, mediaKeys.iv) // decrypt media return aesDecryptWithIV(file, mediaKeys.cipherKey, mediaKeys.iv) // decrypt media
} else { } else {
throw new Error('HMAC sign does not match') throw new Error('')
} }
} }
const allTypes = [type, ...Object.keys(HKDFInfoKeys)] const allTypes = [type, ...Object.keys(HKDFInfoKeys)]
@@ -148,5 +151,5 @@ export async function decodeMediaMessage(message: WAMessageContent, filename: st
if (i === 0) { console.log (`decryption of ${type} media with original HKDF key failed`) } if (i === 0) { console.log (`decryption of ${type} media with original HKDF key failed`) }
} }
} }
throw new Error('HMAC sign does not match') throw new Error('HMAC sign does not match for: ' + buffer.toString('utf-8'))
} }