first commit

build/node_modules/ssh2-streams/test/test-ssh.js

@@ -0,0 +1,143 @@
+var SSH2Stream = require('../lib/ssh');
+var utils = require('../lib/utils');
+var parseKey = utils.parseKey;
+var genPubKey = utils.genPublicKey;
+
+var basename = require('path').basename;
+var assert_ = require('assert');
+var inherits = require('util').inherits;
+var inspect = require('util').inspect;
+var TransformStream = require('stream').Transform;
+var fs = require('fs');
+
+var group = basename(__filename, '.js') + '/';
+var t = -1;
+var SERVER_KEY = fs.readFileSync(__dirname + '/fixtures/ssh_host_rsa_key');
+var HOST_KEYS = { 'ssh-rsa': makeServerKey(SERVER_KEY) };
+
+function SimpleStream() {
+  TransformStream.call(this);
+  this.buffer = '';
+}
+inherits(SimpleStream, TransformStream);
+SimpleStream.prototype._transform = function(chunk, encoding, cb) {
+  this.buffer += chunk.toString('binary');
+  cb(null, chunk);
+};
+
+var tests = [
+  // client-side tests
+  { run: function() {
+      var algos = ['ssh-dss', 'ssh-rsa', 'ecdsa-sha2-nistp521'];
+      var client = new SSH2Stream({
+        algorithms: {
+          serverHostKey: algos
+        }
+      });
+      var clientBufStream = new SimpleStream();
+      var clientReady = false;
+      var server = new SSH2Stream({
+        server: true,
+        hostKeys: HOST_KEYS
+      });
+      var serverBufStream = new SimpleStream();
+      var serverReady = false;
+
+      function onNEWKEYS() {
+        if (this === client) {
+          assert(!clientReady, 'Already received client NEWKEYS event');
+          clientReady = true;
+        } else {
+          assert(!serverReady, 'Already received server NEWKEYS event');
+          serverReady = true;
+        }
+        if (clientReady && serverReady) {
+          var traffic = clientBufStream.buffer;
+          var algoList = algos.join(',');
+          var re = new RegExp('\x00\x00\x00'
+                              + hexByte(algoList.length)
+                              + algoList);
+          assert(re.test(traffic), 'Unexpected client algorithms');
+
+          traffic = serverBufStream.buffer;
+          assert(/\x00\x00\x00\x07ssh-rsa/.test(traffic),
+                 'Unexpected server algorithms');
+
+          next();
+        }
+      }
+
+      client.on('NEWKEYS', onNEWKEYS);
+      server.on('NEWKEYS', onNEWKEYS);
+
+      client.pipe(clientBufStream)
+            .pipe(server)
+            .pipe(serverBufStream)
+            .pipe(client);
+    },
+    what: 'Custom algorithms'
+  },
+  { run: function() {
+      var serverIdent = 'testing  \t';
+      var expectedFullIdent = 'SSH-2.0-' + serverIdent;
+
+      var client = new SSH2Stream({});
+      client.on('header', function(header) {
+        assert(header.identRaw === expectedFullIdent,
+               '\nSaw: ' + inspect(header.identRaw) + '\n'
+                 + 'Expected: ' + inspect(expectedFullIdent));
+        next();
+      });
+
+      var server = new SSH2Stream({
+        server: true,
+        hostKeys: HOST_KEYS,
+        ident: serverIdent
+      });
+
+      client.pipe(server).pipe(client);
+    },
+    what: 'Remote ident is not trimmed'
+  }
+];
+
+function makeServerKey(raw) {
+  var privateKey = parseKey(raw);
+  return {
+    privateKey: privateKey,
+    publicKey: genPubKey(privateKey)
+  };
+}
+
+function hexByte(n) {
+  return String.fromCharCode(n);
+}
+
+function assert(expression, msg) {
+  msg || (msg = 'failed assertion');
+  assert_(expression, makeMsg(tests[t].what, msg));
+}
+
+
+
+function next() {
+  if (Array.isArray(process._events.exit))
+    process._events.exit = process._events.exit[1];
+  if (++t === tests.length)
+    return;
+
+  var v = tests[t];
+  v.run.call(v);
+}
+
+function makeMsg(what, msg) {
+  return '[' + group + what + ']: ' + msg;
+}
+
+process.once('exit', function() {
+  assert_(t === tests.length,
+          makeMsg('_exit',
+                  'Only finished ' + t + '/' + tests.length + ' tests'));
+});
+
+next();