// Copyright 2015 Joyent, Inc.

module.exports = {

bufferSplit: bufferSplit,
addRSAMissing: addRSAMissing,
calculateDSAPublic: calculateDSAPublic,
calculateED25519Public: calculateED25519Public,
calculateX25519Public: calculateX25519Public,
mpNormalize: mpNormalize,
mpDenormalize: mpDenormalize,
ecNormalize: ecNormalize,
countZeros: countZeros,
assertCompatible: assertCompatible,
isCompatible: isCompatible,
opensslKeyDeriv: opensslKeyDeriv,
opensshCipherInfo: opensshCipherInfo,
publicFromPrivateECDSA: publicFromPrivateECDSA,
zeroPadToLength: zeroPadToLength,
writeBitString: writeBitString,
readBitString: readBitString,
pbkdf2: pbkdf2

};

var assert = require('assert-plus'); var Buffer = require('safer-buffer').Buffer; var PrivateKey = require('./private-key'); var Key = require('./key'); var crypto = require('crypto'); var algs = require('./algs'); var asn1 = require('asn1');

var ec = require('ecc-jsbn/lib/ec'); var jsbn = require('jsbn').BigInteger; var nacl = require('tweetnacl');

var MAX_CLASS_DEPTH = 3;

function isCompatible(obj, klass, needVer) {

if (obj === null || typeof (obj) !== 'object')
        return (false);
if (needVer === undefined)
        needVer = klass.prototype._sshpkApiVersion;
if (obj instanceof klass &&
    klass.prototype._sshpkApiVersion[0] == needVer[0])
        return (true);
var proto = Object.getPrototypeOf(obj);
var depth = 0;
while (proto.constructor.name !== klass.name) {
        proto = Object.getPrototypeOf(proto);
        if (!proto || ++depth > MAX_CLASS_DEPTH)
                return (false);
}
if (proto.constructor.name !== klass.name)
        return (false);
var ver = proto._sshpkApiVersion;
if (ver === undefined)
        ver = klass._oldVersionDetect(obj);
if (ver[0] != needVer[0] || ver[1] < needVer[1])
        return (false);
return (true);

}

function assertCompatible(obj, klass, needVer, name) {

if (name === undefined)
        name = 'object';
assert.ok(obj, name + ' must not be null');
assert.object(obj, name + ' must be an object');
if (needVer === undefined)
        needVer = klass.prototype._sshpkApiVersion;
if (obj instanceof klass &&
    klass.prototype._sshpkApiVersion[0] == needVer[0])
        return;
var proto = Object.getPrototypeOf(obj);
var depth = 0;
while (proto.constructor.name !== klass.name) {
        proto = Object.getPrototypeOf(proto);
        assert.ok(proto && ++depth <= MAX_CLASS_DEPTH,
            name + ' must be a ' + klass.name + ' instance');
}
assert.strictEqual(proto.constructor.name, klass.name,
    name + ' must be a ' + klass.name + ' instance');
var ver = proto._sshpkApiVersion;
if (ver === undefined)
        ver = klass._oldVersionDetect(obj);
assert.ok(ver[0] == needVer[0] && ver[1] >= needVer[1],
    name + ' must be compatible with ' + klass.name + ' klass ' +
    'version ' + needVer[0] + '.' + needVer[1]);

}

var CIPHER_LEN = {

'des-ede3-cbc': { key: 24, iv: 8 },
'aes-128-cbc': { key: 16, iv: 16 },
'aes-256-cbc': { key: 32, iv: 16 }

}; var PKCS5_SALT_LEN = 8;

function opensslKeyDeriv(cipher, salt, passphrase, count) {

assert.buffer(salt, 'salt');
assert.buffer(passphrase, 'passphrase');
assert.number(count, 'iteration count');

var clen = CIPHER_LEN[cipher];
assert.object(clen, 'supported cipher');

salt = salt.slice(0, PKCS5_SALT_LEN);

var D, D_prev, bufs;
var material = Buffer.alloc(0);
while (material.length < clen.key + clen.iv) {
        bufs = [];
        if (D_prev)
                bufs.push(D_prev);
        bufs.push(passphrase);
        bufs.push(salt);
        D = Buffer.concat(bufs);
        for (var j = 0; j < count; ++j)
                D = crypto.createHash('md5').update(D).digest();
        material = Buffer.concat([material, D]);
        D_prev = D;
}

return ({
    key: material.slice(0, clen.key),
    iv: material.slice(clen.key, clen.key + clen.iv)
});

}

/* See: RFC2898 */ function pbkdf2(hashAlg, salt, iterations, size, passphrase) {

var hkey = Buffer.alloc(salt.length + 4);
salt.copy(hkey);

var gen = 0, ts = [];
var i = 1;
while (gen < size) {
        var t = T(i++);
        gen += t.length;
        ts.push(t);
}
return (Buffer.concat(ts).slice(0, size));

function T(I) {
        hkey.writeUInt32BE(I, hkey.length - 4);

        var hmac = crypto.createHmac(hashAlg, passphrase);
        hmac.update(hkey);

        var Ti = hmac.digest();
        var Uc = Ti;
        var c = 1;
        while (c++ < iterations) {
                hmac = crypto.createHmac(hashAlg, passphrase);
                hmac.update(Uc);
                Uc = hmac.digest();
                for (var x = 0; x < Ti.length; ++x)
                        Ti[x] ^= Uc[x];
        }
        return (Ti);
}

}

/* Count leading zero bits on a buffer */ function countZeros(buf) {

var o = 0, obit = 8;
while (o < buf.length) {
        var mask = (1 << obit);
        if ((buf[o] & mask) === mask)
                break;
        obit--;
        if (obit < 0) {
                o++;
                obit = 8;
        }
}
return (o*8 + (8 - obit) - 1);

}

function bufferSplit(buf, chr) {

assert.buffer(buf);
assert.string(chr);

var parts = [];
var lastPart = 0;
var matches = 0;
for (var i = 0; i < buf.length; ++i) {
        if (buf[i] === chr.charCodeAt(matches))
                ++matches;
        else if (buf[i] === chr.charCodeAt(0))
                matches = 1;
        else
                matches = 0;

        if (matches >= chr.length) {
                var newPart = i + 1;
                parts.push(buf.slice(lastPart, newPart - matches));
                lastPart = newPart;
                matches = 0;
        }
}
if (lastPart <= buf.length)
        parts.push(buf.slice(lastPart, buf.length));

return (parts);

}

function ecNormalize(buf, addZero) {

assert.buffer(buf);
if (buf[0] === 0x00 && buf[1] === 0x04) {
        if (addZero)
                return (buf);
        return (buf.slice(1));
} else if (buf[0] === 0x04) {
        if (!addZero)
                return (buf);
} else {
        while (buf[0] === 0x00)
                buf = buf.slice(1);
        if (buf[0] === 0x02 || buf[0] === 0x03)
                throw (new Error('Compressed elliptic curve points ' +
                    'are not supported'));
        if (buf[0] !== 0x04)
                throw (new Error('Not a valid elliptic curve point'));
        if (!addZero)
                return (buf);
}
var b = Buffer.alloc(buf.length + 1);
b[0] = 0x0;
buf.copy(b, 1);
return (b);

}

function readBitString(der, tag) {

if (tag === undefined)
        tag = asn1.Ber.BitString;
var buf = der.readString(tag, true);
assert.strictEqual(buf[0], 0x00, 'bit strings with unused bits are ' +
    'not supported (0x' + buf[0].toString(16) + ')');
return (buf.slice(1));

}

function writeBitString(der, buf, tag) {

if (tag === undefined)
        tag = asn1.Ber.BitString;
var b = Buffer.alloc(buf.length + 1);
b[0] = 0x00;
buf.copy(b, 1);
der.writeBuffer(b, tag);

}

function mpNormalize(buf) {

assert.buffer(buf);
while (buf.length > 1 && buf[0] === 0x00 && (buf[1] & 0x80) === 0x00)
        buf = buf.slice(1);
if ((buf[0] & 0x80) === 0x80) {
        var b = Buffer.alloc(buf.length + 1);
        b[0] = 0x00;
        buf.copy(b, 1);
        buf = b;
}
return (buf);

}

function mpDenormalize(buf) {

assert.buffer(buf);
while (buf.length > 1 && buf[0] === 0x00)
        buf = buf.slice(1);
return (buf);

}

function zeroPadToLength(buf, len) {

assert.buffer(buf);
assert.number(len);
while (buf.length > len) {
        assert.equal(buf[0], 0x00);
        buf = buf.slice(1);
}
while (buf.length < len) {
        var b = Buffer.alloc(buf.length + 1);
        b[0] = 0x00;
        buf.copy(b, 1);
        buf = b;
}
return (buf);

}

function bigintToMpBuf(bigint) {

var buf = Buffer.from(bigint.toByteArray());
buf = mpNormalize(buf);
return (buf);

}

function calculateDSAPublic(g, p, x) {

assert.buffer(g);
assert.buffer(p);
assert.buffer(x);
g = new jsbn(g);
p = new jsbn(p);
x = new jsbn(x);
var y = g.modPow(x, p);
var ybuf = bigintToMpBuf(y);
return (ybuf);

}

function calculateED25519Public(k) {

assert.buffer(k);

var kp = nacl.sign.keyPair.fromSeed(new Uint8Array(k));
return (Buffer.from(kp.publicKey));

}

function calculateX25519Public(k) {

assert.buffer(k);

var kp = nacl.box.keyPair.fromSeed(new Uint8Array(k));
return (Buffer.from(kp.publicKey));

}

function addRSAMissing(key) {

assert.object(key);
assertCompatible(key, PrivateKey, [1, 1]);

var d = new jsbn(key.part.d.data);
var buf;

if (!key.part.dmodp) {
        var p = new jsbn(key.part.p.data);
        var dmodp = d.mod(p.subtract(1));

        buf = bigintToMpBuf(dmodp);
        key.part.dmodp = {name: 'dmodp', data: buf};
        key.parts.push(key.part.dmodp);
}
if (!key.part.dmodq) {
        var q = new jsbn(key.part.q.data);
        var dmodq = d.mod(q.subtract(1));

        buf = bigintToMpBuf(dmodq);
        key.part.dmodq = {name: 'dmodq', data: buf};
        key.parts.push(key.part.dmodq);
}

}

function publicFromPrivateECDSA(curveName, priv) {

assert.string(curveName, 'curveName');
assert.buffer(priv);
var params = algs.curves[curveName];
var p = new jsbn(params.p);
var a = new jsbn(params.a);
var b = new jsbn(params.b);
var curve = new ec.ECCurveFp(p, a, b);
var G = curve.decodePointHex(params.G.toString('hex'));

var d = new jsbn(mpNormalize(priv));
var pub = G.multiply(d);
pub = Buffer.from(curve.encodePointHex(pub), 'hex');

var parts = [];
parts.push({name: 'curve', data: Buffer.from(curveName)});
parts.push({name: 'Q', data: pub});

var key = new Key({type: 'ecdsa', curve: curve, parts: parts});
return (key);

}

function opensshCipherInfo(cipher) {

var inf = {};
switch (cipher) {
case '3des-cbc':
        inf.keySize = 24;
        inf.blockSize = 8;
        inf.opensslName = 'des-ede3-cbc';
        break;
case 'blowfish-cbc':
        inf.keySize = 16;
        inf.blockSize = 8;
        inf.opensslName = 'bf-cbc';
        break;
case 'aes128-cbc':
case 'aes128-ctr':
case 'aes128-gcm@openssh.com':
        inf.keySize = 16;
        inf.blockSize = 16;
        inf.opensslName = 'aes-128-' + cipher.slice(7, 10);
        break;
case 'aes192-cbc':
case 'aes192-ctr':
case 'aes192-gcm@openssh.com':
        inf.keySize = 24;
        inf.blockSize = 16;
        inf.opensslName = 'aes-192-' + cipher.slice(7, 10);
        break;
case 'aes256-cbc':
case 'aes256-ctr':
case 'aes256-gcm@openssh.com':
        inf.keySize = 32;
        inf.blockSize = 16;
        inf.opensslName = 'aes-256-' + cipher.slice(7, 10);
        break;
default:
        throw (new Error(
            'Unsupported openssl cipher "' + cipher + '"'));
}
return (inf);

}