From bd4d585b7a9d9ecab5e8c85c797220f87b64a46e Mon Sep 17 00:00:00 2001
From: isaacs <i@izs.me>
Date: Tue, 19 Feb 2013 16:57:59 -0800
Subject: [PATCH] bench: Add bench-crypto

---
 Makefile                          |   3 +
 benchmark/crypto/cipher-stream.js | 103 ++++++++++++++++++++++++++++++
 2 files changed, 106 insertions(+)
 create mode 100644 benchmark/crypto/cipher-stream.js

diff --git a/Makefile b/Makefile
index 4e8655425d..a3eae2806e 100644
--- a/Makefile
+++ b/Makefile
@@ -324,6 +324,9 @@ tools/wrk/wrk:
 bench-net: all
 	@$(NODE) benchmark/common.js net
 
+bench-crypto: all
+	@$(NODE) benchmark/common.js crypto
+
 bench-tls: all
 	@$(NODE) benchmark/common.js tls
 
diff --git a/benchmark/crypto/cipher-stream.js b/benchmark/crypto/cipher-stream.js
new file mode 100644
index 0000000000..2a48a7e3f3
--- /dev/null
+++ b/benchmark/crypto/cipher-stream.js
@@ -0,0 +1,103 @@
+var common = require('../common.js');
+
+var bench = common.createBenchmark(main, {
+  writes: [500],
+  cipher: [ 'AES192', 'AES256' ],
+  type: ['asc', 'utf', 'buf'],
+  len: [2, 1024, 102400, 1024 * 1024],
+  api: ['legacy', 'stream']
+});
+
+function main(conf) {
+  var api = conf.api;
+  if (api === 'stream' && process.version.match(/^v0\.[0-8]\./)) {
+    console.error('Crypto streams not available until v0.10');
+    // use the legacy, just so that we can compare them.
+    api = 'legacy';
+  }
+
+  var dur = conf.dur;
+
+  var crypto = require('crypto');
+  var assert = require('assert');
+  var alice = crypto.getDiffieHellman('modp5');
+  var bob = crypto.getDiffieHellman('modp5');
+
+  alice.generateKeys();
+  bob.generateKeys();
+
+
+  var pubEnc = /^v0\.[0-8]/.test(process.version) ? 'binary' : null;
+  var alice_secret = alice.computeSecret(bob.getPublicKey(), pubEnc, 'hex');
+  var bob_secret = bob.computeSecret(alice.getPublicKey(), pubEnc, 'hex');
+
+  // alice_secret and bob_secret should be the same
+  assert(alice_secret == bob_secret);
+
+  var alice_cipher = crypto.createCipher(conf.cipher, alice_secret);
+  var bob_cipher = crypto.createDecipher(conf.cipher, bob_secret);
+
+  var message;
+  var encoding;
+  switch (conf.type) {
+    case 'asc':
+      message = new Array(conf.len + 1).join('a');
+      encoding = 'ascii';
+      break;
+    case 'utf':
+      message = new Array(conf.len / 2 + 1).join('ü');
+      encoding = 'utf8';
+      break;
+    case 'buf':
+      message = new Buffer(conf.len);
+      message.fill('b');
+      break;
+    default:
+      throw new Error('unknown message type: ' + conf.type);
+  }
+
+  var fn = api === 'stream' ? streamWrite : legacyWrite;
+
+  // write data as fast as possible to alice, and have bob decrypt.
+  // use old API for comparison to v0.8
+  bench.start();
+  fn(alice_cipher, bob_cipher, message, encoding, conf.writes);
+}
+
+function streamWrite(alice, bob, message, encoding, writes) {
+  var written = 0;
+  bob.on('data', function(c) {
+    written += c.length;
+  });
+
+  bob.on('end', function() {
+    // Gbits
+    var bits = written * 8;
+    var gbits = written / (1024 * 1024 * 1024);
+    bench.end(gbits);
+  });
+
+  alice.pipe(bob);
+
+  while (writes-- > 0)
+    alice.write(message, encoding);
+
+  alice.end();
+}
+
+function legacyWrite(alice, bob, message, encoding, writes) {
+  var written = 0;
+  for (var i = 0; i < writes; i++) {
+    var enc = alice.update(message, encoding);
+    var dec = bob.update(enc);
+    written += dec.length;
+  }
+  var enc = alice.final();
+  var dec = bob.update(enc);
+  written += dec.length;
+  dec = bob.final();
+  written += dec.length;
+  var bits = written * 8;
+  var gbits = written / (1024 * 1024 * 1024);
+  bench.end(gbits);
+}