session-desktop/libtextsecure/axolotl_wrapper.js

/*
 * vim: ts=4:sw=4:expandtab
 */
;(function() {
    'use strict';
    window.textsecure = window.textsecure || {};
    window.textsecure.storage = window.textsecure.storage || {};

    textsecure.storage.axolotl = new AxolotlStore();
    var axolotlInstance = axolotl.protocol(textsecure.storage.axolotl);

    /*
     * jobQueue manages multiple queues indexed by device to serialize
     * session io ops on the database.
     */
    var jobQueue = {};
    function queueJobForNumber(number, runJob) {
        var runPrevious = jobQueue[number] || Promise.resolve();
        var runCurrent = jobQueue[number] = runPrevious.then(runJob, runJob);
        runCurrent.then(function() {
            if (jobQueue[number] === runCurrent) {
                delete jobQueue[number];
            }
        });

        return runCurrent;
    }

    window.textsecure = window.textsecure || {};
    window.textsecure.protocol_wrapper = {
        decryptWhisperMessage: function(fromAddress, blob) {
            return queueJobForNumber(fromAddress, function() {
                return axolotlInstance.decryptWhisperMessage(fromAddress, blob.toArrayBuffer());
            });
        },
        closeOpenSessionForDevice: function(encodedNumber) {
            return queueJobForNumber(encodedNumber, function() {
                return axolotlInstance.closeOpenSessionForDevice(encodedNumber);
            });
        },
        encryptMessageFor: function(deviceObject, pushMessageContent) {
            return queueJobForNumber(deviceObject.encodedNumber, function() {
                return axolotlInstance.encryptMessageFor(deviceObject, pushMessageContent);
            });
        },
        startWorker: function() {
            axolotlInstance.startWorker('/js/libaxolotl-worker.js');
        },
        stopWorker: function() {
            axolotlInstance.stopWorker();
        },
        createIdentityKeyRecvSocket: function() {
            return axolotlInstance.createIdentityKeyRecvSocket();
        },
        hasOpenSession: function(encodedNumber) {
            return queueJobForNumber(encodedNumber, function() {
                return axolotlInstance.hasOpenSession(encodedNumber);
            });
        },
        getRegistrationId: function(encodedNumber) {
            return queueJobForNumber(encodedNumber, function() {
                return axolotlInstance.getRegistrationId(encodedNumber);
            });
        },
        handlePreKeyWhisperMessage: function(from, blob) {
            blob.mark();
            var version = blob.readUint8();
            if ((version & 0xF) > 3 || (version >> 4) < 3) {
                // min version > 3 or max version < 3
                throw new Error("Incompatible version byte");
            }
            return queueJobForNumber(from, function() {
                return axolotlInstance.handlePreKeyWhisperMessage(from, blob).catch(function(e) {
                    if (e.message === 'Unknown identity key') {
                        blob.reset(); // restore the version byte.

                        // create an error that the UI will pick up and ask the
                        // user if they want to re-negotiate
                        throw new textsecure.IncomingIdentityKeyError(from, blob.toArrayBuffer(), e.identityKey);
                    }
                    throw e;
                });
            });
        }
    };
})();