added encryptPrivate, decryptPublic methods

Author: rzcoder

src/NodeRSA.js

@@ -221,16 +221,52 @@ module.exports = (function () {
     };
 
     /**
-     * Encrypting data method
+     * Encrypting data method with public key
      *
      * @param buffer {string|number|object|array|Buffer} - data for encrypting. Object and array will convert to JSON string.
      * @param encoding {string} - optional. Encoding for output result, may be 'buffer', 'binary', 'hex' or 'base64'. Default 'buffer'.
      * @param source_encoding {string} - optional. Encoding for given string. Default utf8.
      * @returns {string|Buffer}
      */
     NodeRSA.prototype.encrypt = function (buffer, encoding, source_encoding) {
+        return this.$$encryptKey(false, buffer, encoding, source_encoding);
+    };
+
+    /**
+     * Decrypting data method with private key
+     *
+     * @param buffer {Buffer} - buffer for decrypting
+     * @param encoding - encoding for result string, can also take 'json' or 'buffer' for the automatic conversion of this type
+     * @returns {Buffer|object|string}
+     */
+    NodeRSA.prototype.decrypt = function (buffer, encoding) {
+        return this.$$decryptKey(false, buffer, encoding);
+    };
+
+    /**
+     * Encrypting data method with private key
+     *
+     * Parameters same as `encrypt` method
+     */
+    NodeRSA.prototype.encryptPrivate = function (buffer, encoding, source_encoding) {
+        return this.$$encryptKey(true, buffer, encoding, source_encoding);
+    };
+
+    /**
+     * Decrypting data method with public key
+     *
+     * Parameters same as `decrypt` method
+     */
+    NodeRSA.prototype.decryptPublic = function (buffer, encoding) {
+        return this.$$decryptKey(true, buffer, encoding);
+    };
+
+    /**
+     * Encrypting data method with custom key
+     */
+    NodeRSA.prototype.$$encryptKey = function (usePrivate, buffer, encoding, source_encoding) {
         try {
-            var res = this.keyPair.encrypt(this.$getDataForEncrypt(buffer, source_encoding));
+            var res = this.keyPair.encrypt(this.$getDataForEncrypt(buffer, source_encoding), usePrivate);
 
             if (encoding == 'buffer' || !encoding) {
                 return res;
@@ -243,16 +279,12 @@ module.exports = (function () {
     };
 
     /**
-     * Decrypting data method
-     *
-     * @param buffer {Buffer} - buffer for decrypting
-     * @param encoding - encoding for result string, can also take 'json' or 'buffer' for the automatic conversion of this type
-     * @returns {Buffer|object|string}
+     * Decrypting data method with custom key
      */
-    NodeRSA.prototype.decrypt = function (buffer, encoding) {
+    NodeRSA.prototype.$$decryptKey = function (usePublic, buffer, encoding) {
         try {
             buffer = _.isString(buffer) ? new Buffer(buffer, 'base64') : buffer;
-            var res = this.keyPair.decrypt(buffer);
+            var res = this.keyPair.decrypt(buffer, usePublic);
 
             if (res === null) {
                 throw Error('Key decrypt method returns null.');

src/libs/rsa.js

@@ -206,7 +206,7 @@ module.exports.Key = (function() {
      * @param buffer {Buffer}
      * @returns {Buffer}
      */
-    RSAKey.prototype.encrypt = function (buffer) {
+    RSAKey.prototype.encrypt = function (buffer, usePrivate) {
         var buffers = [];
         var results = [];
         var bufferSize = buffer.length;
@@ -225,7 +225,7 @@ module.exports.Key = (function() {
             var buf = buffers[i];
 
             var m = new BigInteger(this.encryptionScheme.encPad(buf));
-            var c = this.$doPublic(m);
+            var c = usePrivate ? this.$doPrivate(m) : this.$doPublic(m);
 
             if (c === null) {
                 return null;
@@ -243,7 +243,7 @@ module.exports.Key = (function() {
      * @param buffer {Buffer}
      * @returns {Buffer}
      */
-    RSAKey.prototype.decrypt = function (buffer) {
+    RSAKey.prototype.decrypt = function (buffer, usePublic) {
         if (buffer.length % this.encryptedDataLength > 0) {
             throw Error('Incorrect data or key');
         }
@@ -258,7 +258,8 @@ module.exports.Key = (function() {
             length = offset + this.encryptedDataLength;
 
             var c = new BigInteger(buffer.slice(offset, Math.min(length, buffer.length)));
-            var m = this.$doPrivate(c);
+            var m = usePublic ? this.$doPublic(c) : this.$doPrivate(c);
+
             result.push(this.encryptionScheme.encUnPad(m.toBuffer(this.encryptedDataLength)));
         }
 

test/tests.js

@@ -361,15 +361,15 @@ describe("NodeRSA", function(){
                                         var key = null;
                                         var suit = dataBundle[i];
 
-                                        it("should encrypt " + i, function () {
+                                        it("`encrypt()` should encrypt " + i, function () {
                                             key = generatedKeys[Math.round(Math.random() * 1000) % generatedKeys.length];
                                             key.setOptions({encryptionScheme: scheme});
                                             encrypted[i] = key.encrypt(suit.data);
                                             assert(Buffer.isBuffer(encrypted[i]));
                                             assert(encrypted[i].length > 0);
                                         });
 
-                                        it("should decrypt " + i, function () {
+                                        it("`decrypt()` should decrypt " + i, function () {
                                             decrypted[i] = key.decrypt(encrypted[i], _.isArray(suit.encoding) ? suit.encoding[0] : suit.encoding);
                                             if (Buffer.isBuffer(decrypted[i])) {
                                                 assert.equal(suit.data.toString('hex'), decrypted[i].toString('hex'));
@@ -379,6 +379,30 @@ describe("NodeRSA", function(){
                                         });
                                     })(i);
                                 }
+
+                                for (var i in dataBundle) {
+                                    (function (i) {
+                                        var key = null;
+                                        var suit = dataBundle[i];
+
+                                        it("`encryptPrivate()` should encrypt " + i, function () {
+                                            key = generatedKeys[Math.round(Math.random() * 1000) % generatedKeys.length];
+                                            key.setOptions({encryptionScheme: scheme});
+                                            encrypted[i] = key.encryptPrivate(suit.data);
+                                            assert(Buffer.isBuffer(encrypted[i]));
+                                            assert(encrypted[i].length > 0);
+                                        });
+
+                                        it("`decryptPublic()` should decrypt " + i, function () {
+                                            decrypted[i] = key.decryptPublic(encrypted[i], _.isArray(suit.encoding) ? suit.encoding[0] : suit.encoding);
+                                            if (Buffer.isBuffer(decrypted[i])) {
+                                                assert.equal(suit.data.toString('hex'), decrypted[i].toString('hex'));
+                                            } else {
+                                                assert(_.isEqual(suit.data, decrypted[i]));
+                                            }
+                                        });
+                                    })(i);
+                                }
                             });
 
                             describe("Bad cases", function () {