From 02cad1645aa995eb88cb5b26814b1aef77e9f50a Mon Sep 17 00:00:00 2001
From: mruddy <mruddy@users.noreply.github.com>
Date: Wed, 24 Jun 2015 23:10:31 -0400
Subject: [PATCH] fix root cause for bitcore.Transaction.sighash.sign not
 setting nhashtype. added unit tests. fixed an existing lint in a file that i
 touched.

---
 lib/crypto/signature.js  |  1 +
 test/crypto/signature.js | 32 +++++++++++++++++++++++++++++++-
 2 files changed, 32 insertions(+), 1 deletion(-)

diff --git a/lib/crypto/signature.js b/lib/crypto/signature.js
index 5b0c2a0..a179afe 100644
--- a/lib/crypto/signature.js
+++ b/lib/crypto/signature.js
@@ -27,6 +27,7 @@ Signature.prototype.set = function(obj) {
   this.i = typeof obj.i !== 'undefined' ? obj.i : this.i; //public key recovery parameter in range [0, 3]
   this.compressed = typeof obj.compressed !== 'undefined' ?
     obj.compressed : this.compressed; //whether the recovered pubkey is compressed
+  this.nhashtype = obj.nhashtype || this.nhashtype || undefined;
   return this;
 };
 
diff --git a/test/crypto/signature.js b/test/crypto/signature.js
index 74e3ba7..359c96d 100644
--- a/test/crypto/signature.js
+++ b/test/crypto/signature.js
@@ -34,6 +34,17 @@ describe('Signature', function() {
       }));
     });
 
+    it('should set nhashtype', function() {
+      var sig = Signature().set({
+        nhashtype: Signature.SIGHASH_ALL
+      });
+      sig.nhashtype.should.equal(Signature.SIGHASH_ALL);
+      sig.set({
+        nhashtype: Signature.SIGHASH_ALL | Signature.SIGHASH_ANYONECANPAY
+      });
+      sig.nhashtype.should.equal(Signature.SIGHASH_ALL | Signature.SIGHASH_ANYONECANPAY);
+    });
+
   });
 
   describe('#fromCompact', function() {
@@ -97,6 +108,25 @@ describe('Signature', function() {
 
   });
 
+  describe('#toTxFormat', function() {
+
+    it('should parse this known signature and rebuild it with updated zero-padded sighash types', function() {
+      var original = '30450221008bab1f0a2ff2f9cb8992173d8ad73c229d31ea8e10b0f4d4ae1a0d8ed76021fa02200993a6ec81755b9111762fc2cf8e3ede73047515622792110867d12654275e7201';
+      var buf = new Buffer(original, 'hex');
+      var sig = Signature.fromTxFormat(buf);
+      sig.nhashtype.should.equal(Signature.SIGHASH_ALL);
+      sig.set({
+        nhashtype: Signature.SIGHASH_ALL | Signature.SIGHASH_ANYONECANPAY
+      });
+      sig.toTxFormat().toString('hex').should.equal(original.slice(0, -2) + '81');
+      sig.set({
+        nhashtype: Signature.SIGHASH_SINGLE
+      });
+      sig.toTxFormat().toString('hex').should.equal(original.slice(0, -2) + '03');
+    });
+
+  });
+
   describe('#fromTxFormat', function() {
 
     it('should convert from this known tx-format buffer', function() {
@@ -108,7 +138,7 @@ describe('Signature', function() {
     });
 
     it('should parse this known signature and rebuild it', function() {
-      var hex = "3044022007415aa37ce7eaa6146001ac8bdefca0ddcba0e37c5dc08c4ac99392124ebac802207d382307fd53f65778b07b9c63b6e196edeadf0be719130c5db21ff1e700d67501";
+      var hex = '3044022007415aa37ce7eaa6146001ac8bdefca0ddcba0e37c5dc08c4ac99392124ebac802207d382307fd53f65778b07b9c63b6e196edeadf0be719130c5db21ff1e700d67501';
       var buf = new Buffer(hex, 'hex');
       var sig = Signature.fromTxFormat(buf);
       sig.toTxFormat().toString('hex').should.equal(hex);