kg: synchronous APIs for getPrime()

indutny · Oct 28, 2015 · ed36300 · ed36300
1 parent 071717a
commit ed36300
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 3 deletions.
diff --git a/lib/kg.js b/lib/kg.js
@@ -46,14 +46,20 @@ KeyGen.prototype.getPrime = function getPrime(bits, cb) {
         continue;
 
       e.emit('try', 'prime');
-      return cb(null, r);
-    } while (false);
+      if (cb)
+        return cb(null, r);
+      else
+        return r;
+    } while (!cb);
 
     if (aborted)
       return cb(new Error('aborted'));
     setImmediate(next);
   }
-  setImmediate(next);
+  if (cb)
+    setImmediate(next);
+  else
+    return next();
 
   return e;
 };

diff --git a/test/api-test.js b/test/api-test.js
@@ -26,6 +26,11 @@ describe('Self-Signed', function() {
     });
   });
 
+  it('should generate prime numbers synchronously', function() {
+    var p = kg.getPrime(256);
+    assert.equal(p.bitLength(), 256);
+  });
+
   it('should generate private key and cert', function(cb) {
     function find() {
       kg.getPrime(512, function(err, p) {