cryptpad/scripts/tests/test-mailbox.js

// SPDX-FileCopyrightText: 2023 XWiki CryptPad Team <contact@cryptpad.org> and contributors
//
// SPDX-License-Identifier: AGPL-3.0-or-later

var Client = require("../../lib/client/");
var Crypto = require("../../www/components/chainpad-crypto");
var Mailbox = Crypto.Mailbox;
var Nacl = require("tweetnacl/nacl-fast");
var nThen = require("nthen");
var Pinpad = require("../../www/common/pinpad");
var Rpc = require("../../www/common/rpc");
var Hash = require("../../www/common/common-hash");
var CpNetflux = require("../../www/components/chainpad-netflux");
var Util = require("../../lib/common-util");

// you need more than 100 messages in the history, and you need a lastKnownHash between "50" and "length - 50"

var createMailbox = function (config, _cb) {
    var cb = Util.once(Util.mkAsync(_cb));

    var webchannel;
    var user = config.user;
    user.messages = [];

    CpNetflux.start({
        network: config.network,
        channel: config.channel,
        crypto: config.crypto,
        owners: [ config.edPublic ],

        noChainPad: true,

        lastKnownHash: config.lastKnownHash,
        onChannelError: function (err) {
            cb(err);
        },
        onConnect: function (wc /*, sendMessage */) {
            webchannel = wc;
        },
        onMessage: function (msg /*, user, vKey, isCp, hash, author */) {
            user.messages.push(msg);
        },
        onReady: function () {
            cb(void 0, webchannel);
        },
    });
};

process.on('unhandledRejection', function (err) {
    console.error(err);
});

var state = {};

var makeCurveKeys = function () {
    var pair = Nacl.box.keyPair();
    return {
        curvePrivate: Nacl.util.encodeBase64(pair.secretKey),
        curvePublic: Nacl.util.encodeBase64(pair.publicKey),
    };
};

var makeEdKeys = function () {
    var keys = Nacl.sign.keyPair.fromSeed(Nacl.randomBytes(Nacl.sign.seedLength));
    return {
        edPrivate: Nacl.util.encodeBase64(keys.secretKey),
        edPublic: Nacl.util.encodeBase64(keys.publicKey),
    };
};

var edKeys = makeEdKeys();
var curveKeys = makeCurveKeys();
var mailboxChannel = Hash.createChannelId();

var createUser = function (config, cb) {
    // config should contain keys for a team rpc (ed)
        // teamEdKeys
        // rosterHash

    var user;
    nThen(function (w) {
        Client.create(w(function (err, client) {
            if (err) {
                w.abort();
                return void cb(err);
            }
            user = client;
            user.destroy = Util.mkEvent(true);
            user.destroy.reg(function () {
                user.network.disconnect();
            });
        }));
    }).nThen(function (w) {
        // make all the parameters you'll need

        var network = user.network = user.config.network;
        user.edKeys = edKeys;
        user.curveKeys = curveKeys;

        user.mailbox = Mailbox.createEncryptor(user.curveKeys);
        user.mailboxChannel = mailboxChannel;

        // create an anon rpc for alice
        Rpc.createAnonymous(network, w(function (err, rpc) {
            if (err) {
                w.abort();
                user.shutdown();
                return void console.error('ANON_RPC_CONNECT_ERR');
            }
            user.anonRpc = rpc;
            user.destroy.reg(function () {
                user.anonRpc.destroy();
            });
        }));

        Pinpad.create(network, user.edKeys, w(function (err, rpc) {
            if (err) {
                w.abort();
                user.shutdown();
                console.error(err);
                return console.log('RPC_CONNECT_ERR');
            }
            user.rpc = rpc;
            user.destroy.reg(function () {
                user.rpc.destroy();
            });
        }));
    }).nThen(function (w) {
        // create and subscribe to your mailbox
        createMailbox({
            user: user,


            lastKnownHash: config.lastKnownHash,

            network: user.network,
            channel: user.mailboxChannel,
            crypto: user.mailbox,
            edPublic: user.edKeys.edPublic,
        }, w(function (err /*, wc*/) {
            if (err) {
                w.abort();
                //console.error("Mailbox creation error");
                cb(err);
                //process.exit(1);
            }
            //wc.leave();
        }));
    }).nThen(function () {
        user.cleanup = function (/* cb */) {
            //console.log("Destroying user");
            // TODO remove your mailbox
            user.destroy.fire();
        };

        cb(void 0, user);
    });
};

var alice;

nThen(function  (w) {
    createUser({
        //sharedConfig
    }, w(function (err, _alice) {
        if (err) {
            w.abort();
            return void console.log(err);
        }
        alice = _alice;
        alice.name = 'alice';
    }));
    /*
    createUser(sharedConfig, w(function (err, _bob) {
        if (err) {
            w.abort();
            return void console.log(err);
        }
        bob = _bob;
        bob.name = 'bob';
    }));*/
}).nThen(function (w) {
    var i = 0;
    var next = w();

    state.hashes = [];

    var send = function () {
        if (i++ >= 160) { return next(); }

        var msg = alice.mailbox.encrypt(JSON.stringify({
            pewpew: 'bangbang',
        }), alice.curveKeys.curvePublic);

        var hash = msg.slice(0, 64);
        state.hashes.push(hash);

        alice.anonRpc.send('WRITE_PRIVATE_MESSAGE', [
            alice.mailboxChannel,
            msg
            //Nacl.util.encodeBase64(Nacl.randomBytes(128))
        ], w(function (err) {
            if (err) { throw new Error(err); }
            console.log('message %s written successfully', i);
            setTimeout(send, 15);
        }));
    };
    send();
}).nThen(function (w) {
    console.log("Connecting with second user");
    createUser({
        lastKnownHash: state.hashes[55],
    }, w(function (err, _alice) {
        if (err) {
            w.abort();
            console.log("lastKnownHash: ", state.hashes[55]);
            console.log(err);
            process.exit(1);
            //return void console.log(err);
        }
        var user = state.alice2 = _alice;

        if (user.messages.length === 105) {
            process.exit(0);
        }
        //console.log(user.messages, user.messages.length);
        process.exit(1);
    }));
}).nThen(function () {
    

}).nThen(function () {
    alice.cleanup();
    //bob.cleanup();
});