@trenskow/signed-stream 0.1.8 → 0.1.10

package/lib/sign-stream.js

@@ -12,12 +12,13 @@ import CryptoStream from './crypto-stream.js';
 
 export default class SignStream extends CryptoStream {
 
-	constructor({ algorithm, privateKey }) {
+	constructor({ algorithm, privateKey, errorFactory }) {
 
 		const sign = createSign(algorithm);
 
 		super({
-			digester: sign
+			digester: sign,
+			errorFactory
 		});
 
 		this._sign = sign;
@@ -26,9 +27,22 @@ export default class SignStream extends CryptoStream {
 	};
 
 	_final(callback) {
-		super._final(callback);
+		super._final(() => {
 
-		this._resolve(this._sign.sign(this._privateKey));
+			let result;
+
+			try {
+				result = this._sign.sign(this._privateKey);
+			} catch (error) {
+				this._reject(error);
+				return callback(error);
+			}
+
+			this._resolve(result);
+
+			callback();
+
+		});
 
 	}
 

package/lib/verify-stream.js

@@ -12,7 +12,7 @@ import CryptoStream from './crypto-stream.js';
 
 export default class VerifyStream extends CryptoStream {
 
-	constructor({ algorithm, publicKey, signature, encoding }) {
+	constructor({ algorithm, publicKey, signature, encoding, errorFactory }) {
 
 		const verify = createVerify(algorithm);
 
@@ -23,14 +23,31 @@ export default class VerifyStream extends CryptoStream {
 		this._verify = verify;
 		this._publicKey = publicKey;
 		this._signature = Buffer.from(signature, encoding);
+		this._errorFactory = errorFactory;
 
 	};
 
 	_final(callback) {
-		super._final(callback);
+		super._final(() => {
 
-		this._resolve(this._verify.verify(this._publicKey, this._signature));
+			let valid = false;
 
+			try {
+				valid = this._verify.verify(this._publicKey, this._signature);
+			} catch (error) {
+				this._reject(error);
+				return callback(error);
+			}
+
+			this._resolve(valid);
+
+			if (!valid & this._errorFactory) {
+				callback(this._errorFactory());
+			} else {
+				callback();
+			}
+
+		});
 	}
 
 	valid() {

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "@trenskow/signed-stream",
-	"version": "0.1.8",
+	"version": "0.1.10",
 	"description": "A package for cryptographically signing or verifying a stream.",
 	"main": "index.js",
 	"type": "module",