session-desktop/libtextsecure/test/message_receiver_test.js

/*
 * vim: ts=4:sw=4:expandtab
 */

describe('MessageReceiver', function() {
    textsecure.storage.impl = new SignalProtocolStore();
    var WebSocket = window.WebSocket;
    var number = '+19999999999';
    var deviceId = 1;
    var signalingKey = libsignal.crypto.getRandomBytes(32 + 20);
    before(function() {
        window.WebSocket = MockSocket;
        textsecure.storage.user.setNumberAndDeviceId(number, deviceId, 'name');
        textsecure.storage.put("password", "password");
        textsecure.storage.put("signaling_key", signalingKey);
    });
    after (function() { window.WebSocket = WebSocket;  });

    describe('connecting', function() {
        var blob = null;
        var attrs = {
            type: textsecure.protobuf.Envelope.Type.CIPHERTEXT,
            source: number,
            sourceDevice: deviceId,
            timestamp: Date.now(),
        };
        var websocketmessage = new textsecure.protobuf.WebSocketMessage({
            type: textsecure.protobuf.WebSocketMessage.Type.REQUEST,
            request: { verb: 'PUT', path: '/messages' }
        });

        before(function(done) {
            var signal = new textsecure.protobuf.Envelope(attrs).toArrayBuffer();
            var data = new textsecure.protobuf.DataMessage({ body: 'hello' });

            var signaling_key = signalingKey;
            var aes_key = signaling_key.slice(0, 32);
            var mac_key = signaling_key.slice(32, 32 + 20);

            window.crypto.subtle.importKey('raw', aes_key, {name: 'AES-CBC'}, false, ['encrypt']).then(function(key) {
                var iv = libsignal.crypto.getRandomBytes(16);
                window.crypto.subtle.encrypt({name: 'AES-CBC', iv: new Uint8Array(iv)}, key, signal).then(function(ciphertext) {
                    window.crypto.subtle.importKey('raw', mac_key, {name: 'HMAC', hash: {name: 'SHA-256'}}, false, ['sign']).then(function(key) {
                        window.crypto.subtle.sign( {name: 'HMAC', hash: 'SHA-256'}, key, signal).then(function(mac) {
                            var version = new Uint8Array([1]);
                            var message = dcodeIO.ByteBuffer.concat([version, iv, ciphertext, mac ]);
                            websocketmessage.request.body = message.toArrayBuffer();
                            console.log(new Uint8Array(message.toArrayBuffer()));
                            done();
                        });
                    });
                });
            });
        });

        it('connects', function(done) {
            var mockServer = new MockServer('ws://localhost:8080/v1/websocket/?login='+ encodeURIComponent(number) +'.1&password=password');

            mockServer.on('connection', function(server) {
                server.send(new Blob([ websocketmessage.toArrayBuffer() ]));
            });

            window.addEventListener('textsecure:message', function(ev) {
                var signal = ev.proto;
                for (var key in attrs) {
                    assert.strictEqual(attrs[key], signal[key]);
                }
                assert.strictEqual(signal.message.body, 'hello');
                server.close();
                done();
            });
            var messageReceiver = new textsecure.MessageReceiver('ws://localhost:8080', window);
        });
    });
});