From 81aea62bf85aee5184c6b510aa0ebcbf49e313e4 Mon Sep 17 00:00:00 2001
From: Filip Skokan <panva.ip@gmail.com>
Date: Mon, 5 Jun 2023 17:45:47 +0100
Subject: [PATCH] benchmark: add crypto.create*Key
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

PR-URL: https://github.com/nodejs/node/pull/48284
Reviewed-By: Tobias Nießen <tniessen@tnie.de>
Reviewed-By: Luigi Pinca <luigipinca@gmail.com>
Reviewed-By: Yagiz Nizipli <yagiz@nizipli.com>
Reviewed-By: Rafael Gonzaga <rafael.nunu@hotmail.com>
Reviewed-By: James M Snell <jasnell@gmail.com>
---
 benchmark/crypto/create-keyobject.js | 85 ++++++++++++++++++++++++++++
 1 file changed, 85 insertions(+)
 create mode 100644 benchmark/crypto/create-keyobject.js

diff --git a/benchmark/crypto/create-keyobject.js b/benchmark/crypto/create-keyobject.js
new file mode 100644
index 00000000000000..1e98ac347d7efc
--- /dev/null
+++ b/benchmark/crypto/create-keyobject.js
@@ -0,0 +1,85 @@
+'use strict';
+
+const common = require('../common.js');
+const crypto = require('crypto');
+const fs = require('fs');
+const path = require('path');
+const fixtures_keydir = path.resolve(__dirname, '../../test/fixtures/keys/');
+
+function readKey(name) {
+  return fs.readFileSync(`${fixtures_keydir}/${name}.pem`, 'utf8');
+}
+
+function readKeyPair(publicKeyName, privateKeyName) {
+  return {
+    publicKey: readKey(publicKeyName),
+    privateKey: readKey(privateKeyName),
+  };
+}
+
+const keyFixtures = {
+  ec: readKeyPair('ec_p256_public', 'ec_p256_private'),
+  rsa: readKeyPair('rsa_public_2048', 'rsa_private_2048'),
+  ed25519: readKeyPair('ed25519_public', 'ed25519_private'),
+};
+
+const bench = common.createBenchmark(main, {
+  keyType: ['rsa', 'ec', 'ed25519'],
+  keyFormat: ['pkcs8', 'spki', 'der-pkcs8', 'der-spki', 'jwk-public', 'jwk-private'],
+  n: [1e3],
+});
+
+function measure(n, fn, input) {
+  bench.start();
+  for (let i = 0; i < n; ++i) {
+    fn(input);
+  }
+  bench.end(n);
+}
+
+function main({ n, keyFormat, keyType }) {
+  const keyPair = {
+    publicKey: crypto.createPublicKey(keyFixtures[keyType].publicKey),
+    privateKey: crypto.createPrivateKey(keyFixtures[keyType].privateKey),
+  };
+
+  let key, fn;
+  switch (keyFormat) {
+    case 'spki':
+      key = keyPair.publicKey.export({ format: 'pem', type: 'spki' });
+      fn = crypto.createPublicKey;
+      break;
+    case 'pkcs8':
+      key = keyPair.privateKey.export({ format: 'pem', type: 'pkcs8' });
+      fn = crypto.createPrivateKey;
+      break;
+    case 'der-spki': {
+      const options = { format: 'der', type: 'spki' };
+      key = { ...options, key: keyPair.publicKey.export(options) };
+      fn = crypto.createPublicKey;
+      break;
+    }
+    case 'der-pkcs8': {
+      const options = { format: 'der', type: 'pkcs8' };
+      key = { ...options, key: keyPair.privateKey.export(options) };
+      fn = crypto.createPrivateKey;
+      break;
+    }
+    case 'jwk-public': {
+      const options = { format: 'jwk' };
+      key = { ...options, key: keyPair.publicKey.export(options) };
+      fn = crypto.createPublicKey;
+      break;
+    }
+    case 'jwk-private': {
+      const options = { format: 'jwk' };
+      key = { ...options, key: keyPair.privateKey.export(options) };
+      fn = crypto.createPrivateKey;
+      break;
+    }
+    default:
+      throw new Error('not implemented');
+  }
+
+  measure(n, fn, key);
+}