test: add CryptoKey transferring tests
PR-URL: https://github.com/nodejs/node/pull/45811 Reviewed-By: Antoine du Hamel <duhamelantoine1995@gmail.com> Reviewed-By: James M Snell <jasnell@gmail.com>
This commit is contained in:
parent
464e4d293a
commit
992fba4b49
@ -4,7 +4,12 @@ if (!common.hasCrypto)
|
||||
common.skip('missing crypto');
|
||||
|
||||
const assert = require('assert');
|
||||
const { createSecretKey, generateKeyPairSync, randomBytes } = require('crypto');
|
||||
const {
|
||||
generateKeySync,
|
||||
generateKeyPairSync,
|
||||
KeyObject,
|
||||
} = require('crypto');
|
||||
const { subtle } = globalThis.crypto;
|
||||
const { createContext } = require('vm');
|
||||
const {
|
||||
MessageChannel,
|
||||
@ -15,6 +20,9 @@ const {
|
||||
|
||||
function keyToString(key) {
|
||||
let ret;
|
||||
if (key instanceof CryptoKey) {
|
||||
key = KeyObject.from(key);
|
||||
}
|
||||
if (key.type === 'secret') {
|
||||
ret = key.export().toString('hex');
|
||||
} else {
|
||||
@ -33,17 +41,20 @@ if (process.env.HAS_STARTED_WORKER) {
|
||||
// Don't use isMainThread to allow running this test inside a worker.
|
||||
process.env.HAS_STARTED_WORKER = 1;
|
||||
|
||||
// The main thread generates keys and passes them to worker threads.
|
||||
const secretKey = createSecretKey(randomBytes(32));
|
||||
const { publicKey, privateKey } = generateKeyPairSync('rsa', {
|
||||
(async () => {
|
||||
// The main thread generates keys and passes them to worker threads.
|
||||
const secretKey = generateKeySync('aes', { length: 128 });
|
||||
const { publicKey, privateKey } = generateKeyPairSync('rsa', {
|
||||
modulusLength: 1024
|
||||
});
|
||||
});
|
||||
const cryptoKey = await subtle.generateKey(
|
||||
{ name: 'AES-CBC', length: 128 }, false, ['encrypt']);
|
||||
|
||||
// Get immutable representations of all keys.
|
||||
const keys = [secretKey, publicKey, privateKey]
|
||||
// Get immutable representations of all keys.
|
||||
const keys = [secretKey, publicKey, privateKey, cryptoKey]
|
||||
.map((key) => [key, keyToString(key)]);
|
||||
|
||||
for (const [key, repr] of keys) {
|
||||
for (const [key, repr] of keys) {
|
||||
{
|
||||
// Test 1: No context change.
|
||||
const { port1, port2 } = new MessageChannel();
|
||||
@ -84,4 +95,5 @@ for (const [key, repr] of keys) {
|
||||
port1.postMessage({ key });
|
||||
port1.close();
|
||||
}
|
||||
}
|
||||
}
|
||||
})().then(common.mustCall());
|
||||
|
@ -5,13 +5,30 @@ if (!common.hasCrypto)
|
||||
|
||||
// Issue https://github.com/nodejs/node/issues/35263
|
||||
// Description: Test that passing keyobject to worker thread does not crash.
|
||||
const { createSecretKey } = require('crypto');
|
||||
const {
|
||||
generateKeySync,
|
||||
generateKeyPairSync,
|
||||
} = require('crypto');
|
||||
const { subtle } = globalThis.crypto;
|
||||
|
||||
const assert = require('assert');
|
||||
|
||||
const { Worker, isMainThread, workerData } = require('worker_threads');
|
||||
|
||||
if (isMainThread) {
|
||||
const key = createSecretKey(Buffer.from('hello'));
|
||||
(async () => {
|
||||
const secretKey = generateKeySync('aes', { length: 128 });
|
||||
const { publicKey, privateKey } = generateKeyPairSync('rsa', {
|
||||
modulusLength: 1024
|
||||
});
|
||||
const cryptoKey = await subtle.generateKey(
|
||||
{ name: 'AES-CBC', length: 128 }, false, ['encrypt']);
|
||||
|
||||
for (const key of [secretKey, publicKey, privateKey, cryptoKey]) {
|
||||
new Worker(__filename, { workerData: key });
|
||||
}
|
||||
})().then(common.mustCall());
|
||||
} else {
|
||||
console.log(workerData);
|
||||
assert.notDeepStrictEqual(workerData, {});
|
||||
}
|
||||
|
Loading…
x
Reference in New Issue
Block a user