feat: include platform in creds

This commit is contained in:
Adhiraj Singh
2022-04-13 12:49:46 +05:30
parent 564c9f2e86
commit 7db8796fed
3 changed files with 37 additions and 22 deletions

View File

@@ -490,7 +490,10 @@ export const makeSocket = ({
} }
} }
logger.info({ jid: updatedCreds.me!.id }, 'registered connection, restart server') logger.info(
{ me: updatedCreds.me, platform: updatedCreds.platform },
'registered connection, restart server'
)
ev.emit('creds.update', updatedCreds) ev.emit('creds.update', updatedCreds)
ev.emit('connection.update', { isNewLogin: true, qr: undefined }) ev.emit('connection.update', { isNewLogin: true, qr: undefined })

View File

@@ -44,6 +44,7 @@ export type AuthenticationCreds = SignalCreds & {
nextPreKeyId: number nextPreKeyId: number
lastAccountSyncTimestamp?: number lastAccountSyncTimestamp?: number
platform?: string
accountSettings: AccountSettings accountSettings: AccountSettings
} }

View File

@@ -3,7 +3,7 @@ import { createHash } from 'crypto'
import { proto } from '../../WAProto' import { proto } from '../../WAProto'
import { KEY_BUNDLE_TYPE } from '../Defaults' import { KEY_BUNDLE_TYPE } from '../Defaults'
import type { AuthenticationCreds, SignalCreds, SocketConfig } from '../Types' import type { AuthenticationCreds, SignalCreds, SocketConfig } from '../Types'
import { Binary, BinaryNode, getAllBinaryNodeChildren, jidDecode, S_WHATSAPP_NET } from '../WABinary' import { Binary, BinaryNode, getBinaryNodeChild, jidDecode, S_WHATSAPP_NET } from '../WABinary'
import { Curve, hmacSign } from './crypto' import { Curve, hmacSign } from './crypto'
import { encodeBigEndian } from './generics' import { encodeBigEndian } from './generics'
import { createSignalIdentity } from './signal' import { createSignalIdentity } from './signal'
@@ -97,21 +97,28 @@ export const configureSuccessfulPairing = (
stanza: BinaryNode, stanza: BinaryNode,
{ advSecretKey, signedIdentityKey, signalIdentities }: Pick<AuthenticationCreds, 'advSecretKey' | 'signedIdentityKey' | 'signalIdentities'> { advSecretKey, signedIdentityKey, signalIdentities }: Pick<AuthenticationCreds, 'advSecretKey' | 'signedIdentityKey' | 'signalIdentities'>
) => { ) => {
const [pair] = getAllBinaryNodeChildren(stanza)
const pairContent = Array.isArray(pair.content) ? pair.content : []
const msgId = stanza.attrs.id const msgId = stanza.attrs.id
const deviceIdentity = pairContent.find(m => m.tag === 'device-identity')?.content
const businessName = pairContent.find(m => m.tag === 'biz')?.attrs?.name
const verifiedName = businessName || ''
const jid = pairContent.find(m => m.tag === 'device')?.attrs?.jid
const { details, hmac } = proto.ADVSignedDeviceIdentityHMAC.decode(deviceIdentity as Buffer) const pairSuccessNode = getBinaryNodeChild(stanza, 'pair-success')
const deviceIdentityNode = getBinaryNodeChild(pairSuccessNode, 'device-identity')
const platformNode = getBinaryNodeChild(pairSuccessNode, 'platform')
const deviceNode = getBinaryNodeChild(pairSuccessNode, 'device')
const businessNode = getBinaryNodeChild(pairSuccessNode, 'biz')
if(!deviceIdentityNode || !deviceNode) {
throw new Boom('Missing device-identity or device in pair success node', { data: stanza })
}
const bizName = businessNode?.attrs.name
const jid = deviceNode.attrs.jid
const { details, hmac } = proto.ADVSignedDeviceIdentityHMAC.decode(deviceIdentityNode.content as Buffer)
const advSign = hmacSign(details, Buffer.from(advSecretKey, 'base64')) const advSign = hmacSign(details, Buffer.from(advSecretKey, 'base64'))
if(Buffer.compare(hmac, advSign) !== 0) { if(Buffer.compare(hmac, advSign) !== 0) {
throw new Boom('Invalid pairing') throw new Boom('Invalid account signature')
} }
const account = proto.ADVSignedDeviceIdentity.decode(details) const account = proto.ADVSignedDeviceIdentity.decode(details)
@@ -122,17 +129,16 @@ export const configureSuccessfulPairing = (
throw new Boom('Failed to verify account signature') throw new Boom('Failed to verify account signature')
} }
const identity = createSignalIdentity(jid, accountSignatureKey)
const deviceIdentity = proto.ADVDeviceIdentity.decode(account.details)
const keyIndex = deviceIdentity.keyIndex
const deviceMsg = Binary.build(new Uint8Array([6, 1]), account.details, signedIdentityKey.public, account.accountSignatureKey).readByteArray() const deviceMsg = Binary.build(new Uint8Array([6, 1]), account.details, signedIdentityKey.public, account.accountSignatureKey).readByteArray()
account.deviceSignature = Curve.sign(signedIdentityKey.private, deviceMsg) account.deviceSignature = Curve.sign(signedIdentityKey.private, deviceMsg)
const identity = createSignalIdentity(jid, accountSignatureKey) delete account.accountSignatureKey
const accountEnc = proto.ADVSignedDeviceIdentity.encode(account).finish()
const keyIndex = proto.ADVDeviceIdentity.decode(account.details).keyIndex
const accountEnc = proto.ADVSignedDeviceIdentity.encode({
...account.toJSON(),
accountSignatureKey: undefined
}).finish()
const reply: BinaryNode = { const reply: BinaryNode = {
tag: 'iq', tag: 'iq',
@@ -148,7 +154,7 @@ export const configureSuccessfulPairing = (
content: [ content: [
{ {
tag: 'device-identity', tag: 'device-identity',
attrs: { 'key-index': `${keyIndex}` }, attrs: { 'key-index': deviceIdentity.keyIndex.toString() },
content: accountEnc content: accountEnc
} }
] ]
@@ -158,9 +164,14 @@ export const configureSuccessfulPairing = (
const authUpdate: Partial<AuthenticationCreds> = { const authUpdate: Partial<AuthenticationCreds> = {
account, account,
me: { id: jid, verifiedName }, me: { id: jid, name: bizName },
signalIdentities: [...(signalIdentities || []), identity] signalIdentities: [
...(signalIdentities || []),
identity
],
platform: platformNode?.attrs.name
} }
return { return {
creds: authUpdate, creds: authUpdate,
reply reply