// Copyright Joyent, Inc. and other Node contributors.
//
// Permission is hereby granted, free of charge, to any person obtaining a
// copy of this software and associated documentation files (the
// "Software"), to deal in the Software without restriction, including
// without limitation the rights to use, copy, modify, merge, publish,
// distribute, sublicense, and/or sell copies of the Software, and to permit
// persons to whom the Software is furnished to do so, subject to the
// following conditions:
//
// The above copyright notice and this permission notice shall be included
// in all copies or substantial portions of the Software.
//
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
// OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
// MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN
// NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM,
// DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR
// OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE
// USE OR OTHER DEALINGS IN THE SOFTWARE.

'use strict';
const common = require('../common');

if (!common.hasCrypto)
  common.skip('missing crypto');

const assert = require('assert');
const crypto = require('crypto');

crypto.DEFAULT_ENCODING = 'buffer';

// bump, we register a lot of exit listeners
process.setMaxListeners(256);

const expectedErrorRegexp = /^TypeError: size must be a number >= 0$/;
[crypto.randomBytes, crypto.pseudoRandomBytes].forEach(function(f) {
  [-1, undefined, null, false, true, {}, []].forEach(function(value) {
    assert.throws(function() { f(value); }, expectedErrorRegexp);
    assert.throws(function() { f(value, common.mustNotCall()); },
                  expectedErrorRegexp);
  });

  [0, 1, 2, 4, 16, 256, 1024].forEach(function(len) {
    f(len, common.mustCall(function(ex, buf) {
      assert.strictEqual(ex, null);
      assert.strictEqual(buf.length, len);
      assert.ok(Buffer.isBuffer(buf));
    }));
  });
});

{
  const buf = Buffer.alloc(10);
  const before = buf.toString('hex');
  const after = crypto.randomFillSync(buf).toString('hex');
  assert.notStrictEqual(before, after);
}

{
  const buf = new Uint8Array(new Array(10).fill(0));
  const before = Buffer.from(buf).toString('hex');
  crypto.randomFillSync(buf);
  const after = Buffer.from(buf).toString('hex');
  assert.notStrictEqual(before, after);
}

{
  const buf = Buffer.alloc(10);
  const before = buf.toString('hex');
  crypto.randomFill(buf, common.mustCall((err, buf) => {
    assert.ifError(err);
    const after = buf.toString('hex');
    assert.notStrictEqual(before, after);
  }));
}

{
  const buf = new Uint8Array(new Array(10).fill(0));
  const before = Buffer.from(buf).toString('hex');
  crypto.randomFill(buf, common.mustCall((err, buf) => {
    assert.ifError(err);
    const after = Buffer.from(buf).toString('hex');
    assert.notStrictEqual(before, after);
  }));
}

{
  const buf = Buffer.alloc(10);
  const before = buf.toString('hex');
  crypto.randomFillSync(buf, 5, 5);
  const after = buf.toString('hex');
  assert.notStrictEqual(before, after);
  assert.deepStrictEqual(before.slice(0, 5), after.slice(0, 5));
}

{
  const buf = new Uint8Array(new Array(10).fill(0));
  const before = Buffer.from(buf).toString('hex');
  crypto.randomFillSync(buf, 5, 5);
  const after = Buffer.from(buf).toString('hex');
  assert.notStrictEqual(before, after);
  assert.deepStrictEqual(before.slice(0, 5), after.slice(0, 5));
}

{
  const buf = Buffer.alloc(10);
  const before = buf.toString('hex');
  crypto.randomFillSync(buf, 5);
  const after = buf.toString('hex');
  assert.notStrictEqual(before, after);
  assert.deepStrictEqual(before.slice(0, 5), after.slice(0, 5));
}

{
  const buf = Buffer.alloc(10);
  const before = buf.toString('hex');
  crypto.randomFill(buf, 5, 5, common.mustCall((err, buf) => {
    assert.ifError(err);
    const after = buf.toString('hex');
    assert.notStrictEqual(before, after);
    assert.deepStrictEqual(before.slice(0, 5), after.slice(0, 5));
  }));
}

{
  const buf = new Uint8Array(new Array(10).fill(0));
  const before = Buffer.from(buf).toString('hex');
  crypto.randomFill(buf, 5, 5, common.mustCall((err, buf) => {
    assert.ifError(err);
    const after = Buffer.from(buf).toString('hex');
    assert.notStrictEqual(before, after);
    assert.deepStrictEqual(before.slice(0, 5), after.slice(0, 5));
  }));
}

{
  const bufs = [
    Buffer.alloc(10),
    new Uint8Array(new Array(10).fill(0))
  ];
  const errMessages = {
    offsetNotNumber: /^TypeError: offset must be a number$/,
    offsetOutOfRange: /^RangeError: offset out of range$/,
    offsetNotUInt32: /^TypeError: offset must be a uint32$/,
    sizeNotNumber: /^TypeError: size must be a number$/,
    sizeNotUInt32: /^TypeError: size must be a uint32$/,
    bufferTooSmall: /^RangeError: buffer too small$/,
  };

  const max = require('buffer').kMaxLength + 1;

  for (const buf of bufs) {
    const len = Buffer.byteLength(buf);
    assert.strictEqual(len, 10, `Expected byteLength of 10, got ${len}`);

    assert.throws(() => {
      crypto.randomFillSync(buf, 'test');
    }, errMessages.offsetNotNumber);

    assert.throws(() => {
      crypto.randomFillSync(buf, NaN);
    }, errMessages.offsetNotNumber);

    assert.throws(() => {
      crypto.randomFill(buf, 'test', common.mustNotCall());
    }, errMessages.offsetNotNumber);

    assert.throws(() => {
      crypto.randomFill(buf, NaN, common.mustNotCall());
    }, errMessages.offsetNotNumber);

    assert.throws(() => {
      crypto.randomFillSync(buf, 11);
    }, errMessages.offsetOutOfRange);

    assert.throws(() => {
      crypto.randomFillSync(buf, max);
    }, errMessages.offsetOutOfRange);

    assert.throws(() => {
      crypto.randomFill(buf, 11, common.mustNotCall());
    }, errMessages.offsetOutOfRange);

    assert.throws(() => {
      crypto.randomFill(buf, max, common.mustNotCall());
    }, errMessages.offsetOutOfRange);

    assert.throws(() => {
      crypto.randomFillSync(buf, 0, 'test');
    }, errMessages.sizeNotNumber);

    assert.throws(() => {
      crypto.randomFillSync(buf, 0, NaN);
    }, errMessages.sizeNotNumber);

    assert.throws(() => {
      crypto.randomFill(buf, 0, 'test', common.mustNotCall());
    }, errMessages.sizeNotNumber);

    assert.throws(() => {
      crypto.randomFill(buf, 0, NaN, common.mustNotCall());
    }, errMessages.sizeNotNumber);

    {
      const size = (-1 >>> 0) + 1;

      assert.throws(() => {
        crypto.randomFillSync(buf, 0, -10);
      }, errMessages.sizeNotUInt32);

      assert.throws(() => {
        crypto.randomFillSync(buf, 0, size);
      }, errMessages.sizeNotUInt32);

      assert.throws(() => {
        crypto.randomFill(buf, 0, -10, common.mustNotCall());
      }, errMessages.sizeNotUInt32);

      assert.throws(() => {
        crypto.randomFill(buf, 0, size, common.mustNotCall());
      }, errMessages.sizeNotUInt32);
    }

    assert.throws(() => {
      crypto.randomFillSync(buf, -10);
    }, errMessages.offsetNotUInt32);

    assert.throws(() => {
      crypto.randomFill(buf, -10, common.mustNotCall());
    }, errMessages.offsetNotUInt32);

    assert.throws(() => {
      crypto.randomFillSync(buf, 1, 10);
    }, errMessages.bufferTooSmall);

    assert.throws(() => {
      crypto.randomFill(buf, 1, 10, common.mustNotCall());
    }, errMessages.bufferTooSmall);

    assert.throws(() => {
      crypto.randomFillSync(buf, 0, 12);
    }, errMessages.bufferTooSmall);

    assert.throws(() => {
      crypto.randomFill(buf, 0, 12, common.mustNotCall());
    }, errMessages.bufferTooSmall);

    {
      // Offset is too big
      const offset = (-1 >>> 0) + 1;
      assert.throws(() => {
        crypto.randomFillSync(buf, offset, 10);
      }, errMessages.offsetNotUInt32);

      assert.throws(() => {
        crypto.randomFill(buf, offset, 10, common.mustNotCall());
      }, errMessages.offsetNotUInt32);
    }
  }
}

// #5126, "FATAL ERROR: v8::Object::SetIndexedPropertiesToExternalArrayData()
// length exceeds max acceptable value"
assert.throws(function() {
  crypto.randomBytes((-1 >>> 0) + 1);
}, /^TypeError: size must be a number >= 0$/);