Nomchy/discord.js
1
0
Fork 0
mirror of https://github.com/discordjs/discord.js.git synced 2026-03-15 11:03:30 +01:00
Code Issues Packages Projects Releases Wiki Activity

Implement voice encryption support

Browse Source
This commit is contained in:
qeled
2016-03-18 06:32:35 +03:00
parent 27a0361afc
commit 4d6cca72f4
5 changed files with 97 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
src/Voice/VoiceConnection.js
View File

@@ -17,6 +17,9 @@ import StreamIntent from "./StreamIntent";
import EventEmitter from "events";
import unpipe from "unpipe";
const MODE_xsalsa20_poly1305 = "xsalsa20_poly1305";
const MODE_plain = "plain";
export default class VoiceConnection extends EventEmitter {
constructor(channel, client, session, token, server, endpoint) {
super();
@@ -39,6 +42,10 @@ export default class VoiceConnection extends EventEmitter {
this.KAI = null;
this.timestamp = 0;
this.sequence = 0;
this.mode = null;
this.secret = null;
this.volume = new VolumeTransformer();
this.init();
}
@@ -196,7 +203,7 @@ export default class VoiceConnection extends EventEmitter {
}
var buffer = self.encoder.opusBuffer(rawbuffer);
var packet = new VoicePacket(buffer, sequence, timestamp, self.vWSData.ssrc);
var packet = new VoicePacket(buffer, sequence, timestamp, self.vWSData.ssrc, self.secret);
return self.sendPacket(packet, callback);
} catch (e) {
@@ -314,6 +321,13 @@ export default class VoiceConnection extends EventEmitter {
}
discordPort = msg.readUIntLE(msg.length - 2, 2).toString(10);
var modes = self.vWSData.modes;
var mode = MODE_xsalsa20_poly1305;
if (modes.indexOf(MODE_xsalsa20_poly1305) < 0) {
mode = MODE_plain;
self.client.emit("debug", "Encrypted mode not reported as supported by the server, using 'plain'");
}
vWS.send(JSON.stringify({
"op": 1,
"d": {
@@ -321,7 +335,7 @@ export default class VoiceConnection extends EventEmitter {
"data": {
"address": discordIP,
"port": Number(discordPort),
"mode": self.vWSData.modes[0] //Plain
"mode": mode
}
}
}));
@@ -365,6 +379,13 @@ export default class VoiceConnection extends EventEmitter {
});
break;
case 4:
if (data.d.secret_key && data.d.secret_key.length > 0) {
const buffer = new ArrayBuffer(data.d.secret_key.length);
self.secret = new Uint8Array(buffer);
for (let i = 0; i < this.secret.length; i++) {
self.secret[i] = data.d.secret_key[i];
}
}
self.ready = true;
self.mode = data.d.mode;

23
src/Voice/VoicePacket.js
View File

@@ -1,10 +1,17 @@
"use strict";
export default class VoicePacket {
constructor(data, sequence, time, ssrc){
import nacl from "tweetnacl";
var audioBuffer = data,
returnBuffer = new Buffer(audioBuffer.length + 12);
const nonce = new Buffer(24);
nonce.fill(0);
export default class VoicePacket {
constructor(data, sequence, time, ssrc, secret){
var mac = secret ? 16 : 0;
var packetLength = data.length + 12 + mac;
var audioBuffer = data;
var returnBuffer = new Buffer(packetLength);
returnBuffer.fill(0);
returnBuffer[0] = 0x80;
@@ -14,7 +21,13 @@ export default class VoicePacket {
returnBuffer.writeUIntBE(time, 4, 4);
returnBuffer.writeUIntBE(ssrc, 8, 4);
for (var i=0; i<audioBuffer.length; i++) {
if (secret) {
// copy first 12 bytes
returnBuffer.copy(nonce, 0, 0, 12);
audioBuffer = nacl.secretbox(data, nonce, secret);
}
for (var i = 0; i < audioBuffer.length; i++) {
returnBuffer[i + 12] = audioBuffer[i];
}