Refactoring ssh-keygen function (#1533)

red/runtime/storage/localfilesystem/projects/index.js

@@ -23,6 +23,7 @@ var crypto = require('crypto');
 var storageSettings = require("../settings");
 var util = require("../util");
 var gitTools = require("./git");
+var sshTools = require("./ssh");
 
 var Projects = require("./Project");
 
@@ -89,6 +90,7 @@ function init(_settings, _runtime) {
                 projectsEnabled = false;
             } else {
                 Projects.init(settings,runtime);
+                sshTools.init(settings,runtime);
                 projectsDir = fspath.join(settings.userDir,"projects");
                 if (!settings.readOnly) {
                     return fs.ensureDir(projectsDir)

red/runtime/storage/localfilesystem/projects/ssh.js

@@ -0,0 +1,80 @@
+/**
+ * Copyright JS Foundation and other contributors, http://js.foundation
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ **/
+
+var exec = require('child_process').exec;
+var spawn = require('child_process').spawn;
+
+var sshkeygenCommand = "ssh-keygen";
+
+var log;
+
+function runSshKeygenCommand(args,cwd,env) {
+    // console.log("[run ssh-keygen] args:", args);
+    return new Promise(function(resolve, reject) {
+        var child = spawn(sshkeygenCommand, args, {cwd: cwd, detached: true, env: env});
+        var stdout = "";
+        var stderr = "";
+
+        child.stdout.on('data', function(data) {
+            stdout += data;
+        });
+        child.stderr.on('data', function(data) {
+            stderr += data;
+        });
+        child.on('close', function(code, signal) {
+            if (code !== 0) {
+                var err = new Error(stderr);
+                err.stdout = stdout;
+                err.stderr = stderr;
+                if (/passphrase is too short/.test(stderr)) {
+                    err.code = "key_passphrase_too_short";
+                } else if(/Key must at least be 1024 bits/.test(stderr)) {
+                    err.code = "key_length_too_short";
+                }
+                reject(err);
+            }
+            else {
+                resolve(stdout);
+            }
+        });
+    });
+}
+
+function init(_settings, _runtime) {
+    log = _runtime.log;
+}
+
+function generateKey(options) {
+    var args = ['-q', '-t', 'rsa'];
+    if (options.size) {
+        args.push('-b', options.size);
+    }
+    if (options.location) {
+        args.push('-f', options.location);
+    }
+    if (options.comment) {
+        args.push('-C', options.comment);
+    }
+    if (options.password) {
+        args.push('-N', options.password);
+    }
+    return runSshKeygenCommand(args,__dirname);
+}
+
+module.exports = {
+    init: init,
+    generateKey: generateKey,
+};

red/runtime/storage/localfilesystem/sshkeys.js

@@ -17,7 +17,7 @@
 var fs = require('fs-extra');
 var when = require('when');
 var fspath = require("path");
-var keygen = require('ssh-keygen');
+var sshTools = require("./projects/ssh");
 
 var settings;
 var runtime;
@@ -97,26 +97,12 @@ function generateSSHKey(username, options) {
             } else {
                 var comment = options.comment || "";
                 var password = options.password || "";
-                if (password.length > 0 && password.length < 5) {
-                    var e2 = new Error("SSH Key passphrase too short");
-                    e2.code = "key_passphrase_too_short";
-                    throw e2;
-                }
                 var size = options.size || 2048;
                 var sshKeyFileBasename = username + '_' + name;
                 var privateKeyFilePath = fspath.join(sshkeyDir, sshKeyFileBasename);
                 return generateSSHKeyPair(name, privateKeyFilePath, comment, password, size)
             }
         })
-        // .then(function(keyfile_name) {
-        //     return checkSSHKeyFileAndGetPublicKeyFileName(username, name)
-        //         .then(function() {
-        //             return keyfile_name;
-        //         })
-        //         .catch(function(err) {
-        //             throw new Error('Failed to generate ssh key files');
-        //         });
-        // });
 }
 
 function deleteSSHKey(username, name) {
@@ -162,27 +148,22 @@ function deleteSSHKeyFiles(username, name) {
     return Promise.all([
         fs.remove(privateKeyFilePath),
         fs.remove(publicKeyFilePath)
-    ]);
+    ])
+    .then(function() {
+        return true;
+    });
 }
 
 function generateSSHKeyPair(name, privateKeyPath, comment, password, size) {
     log.trace("ssh-keygen["+[name,privateKeyPath,comment,size,"hasPassword?"+!!password].join(",")+"]");
-    return new Promise(function(resolve, reject) {
-        keygen({
-            location: privateKeyPath,
-            comment: comment,
-            password: password,
-            size: size
-        }, function(err, out) {
-            if ( err ) {
-                err.code = "key_generation_failed";
-                reject(err);
-            }
-            else {
-                resolve(name);
-            }
-        });
-    });
+    return sshTools.generateKey({location: privateKeyPath, comment: comment, password: password, size: size})
+            .then(function(stdout) {
+                return name;
+            })
+            .catch(function(err) {
+                log.log('[SSHKey generation] error:', err);
+                throw err;
+            });
 }
 
 module.exports = {