From 88664d2c8cc175ef7dd13770d35d8051b567d7b5 Mon Sep 17 00:00:00 2001
From: Jeff Garzik <jgarzik@bitpay.com>
Date: Fri, 2 Aug 2013 13:54:53 -0400
Subject: [PATCH] Move varint, varstr parsing to BinaryParser for wider sharing
 of code.

---
 Connection.js        | 45 +++++++++++---------------------------------
 util/BinaryParser.js | 23 ++++++++++++++++++++++
 2 files changed, 34 insertions(+), 34 deletions(-)

diff --git a/Connection.js b/Connection.js
index e54a8ec2c..e94aab68c 100644
--- a/Connection.js
+++ b/Connection.js
@@ -413,13 +413,13 @@ function spec(b) {
       data.addr_me = parser.buffer(26);
       data.addr_you = parser.buffer(26);
       data.nonce = parser.buffer(8);
-      data.subversion = Connection.parseVarStr(parser);
+      data.subversion = parser.varStr();
       data.start_height = parser.word32le();
       break;
 
     case 'inv':
     case 'getdata':
-      data.count = Connection.parseVarInt(parser);
+      data.count = parser.varInt();
 
       data.invs = [];
       for (i = 0; i < data.count; i++) {
@@ -438,7 +438,7 @@ function spec(b) {
       data.bits = parser.word32le();
       data.nonce = parser.word32le();
 
-      var txCount = Connection.parseVarInt(parser);
+      var txCount = parser.varInt();
 
       data.txs = [];
       for (i = 0; i < txCount; i++) {
@@ -465,7 +465,7 @@ function spec(b) {
 
       // TODO: Limit block locator size?
       // reference implementation limits to 500 results
-      var startCount = Connection.parseVarInt(parser);
+      var startCount = parser.varInt();
 
       data.starts = [];
       for (i = 0; i < startCount; i++) {
@@ -475,7 +475,7 @@ function spec(b) {
       break;
 
     case 'addr':
-      var addrCount = Connection.parseVarInt(parser);
+      var addrCount = parser.varInt();
 
       // Enforce a maximum number of addresses per message
       if (addrCount > 1000) {
@@ -495,8 +495,8 @@ function spec(b) {
       break;
 
     case 'alert':
-      data.payload = Connection.parseVarStr(parser);
-      data.signature = Connection.parseVarStr(parser);
+      data.payload = parser.varStr();
+      data.signature = parser.varStr();
       break;
 
     case 'ping':
@@ -521,29 +521,6 @@ function spec(b) {
     return data;
   };
 
-  Connection.parseVarInt = function (parser)
-  {
-    var firstByte = parser.word8();
-    switch (firstByte) {
-    case 0xFD:
-      return parser.word16le();
-
-    case 0xFE:
-      return parser.word32le();
-
-    case 0xFF:
-      return parser.word64le();
-
-    default:
-      return firstByte;
-    }
-  };
-
-  Connection.parseVarStr = function (parser) {
-    var len = Connection.parseVarInt(parser);
-    return parser.buffer(len);
-  };
-
   Connection.parseTx = function (parser) {
     if (Buffer.isBuffer(parser)) {
       parser = new Parser(parser);
@@ -553,25 +530,25 @@ function spec(b) {
 
     data.version = parser.word32le();
     
-    var txinCount = Connection.parseVarInt(parser, 'tx_in_count');
+    var txinCount = parser.varInt();
 
     data.ins = [];
     for (j = 0; j < txinCount; j++) {
       var txin = {};
       txin.o = parser.buffer(36);               // outpoint
-      sLen = Connection.parseVarInt(parser);    // script_len
+      sLen = parser.varInt();    // script_len
       txin.s = parser.buffer(sLen);             // script
       txin.q = parser.word32le();               // sequence
       data.ins.push(txin);
     }
 
-    var txoutCount = Connection.parseVarInt(parser);
+    var txoutCount = parser.varInt();
 
     data.outs = [];
     for (j = 0; j < txoutCount; j++) {
       var txout = {};
       txout.v = parser.buffer(8);               // value
-      sLen = Connection.parseVarInt(parser);    // script_len
+      sLen = parser.varInt();    // script_len
       txout.s = parser.buffer(sLen);            // script
       data.outs.push(txout);
     }
diff --git a/util/BinaryParser.js b/util/BinaryParser.js
index 87c1b068e..d0707e77b 100644
--- a/util/BinaryParser.js
+++ b/util/BinaryParser.js
@@ -120,6 +120,29 @@ function spec(b) {
     Parser.prototype.word8s = Parser.prototype.word8bs;
   });
 
+  Parser.prototype.varInt = function ()
+  {
+    var firstByte = this.word8();
+    switch (firstByte) {
+    case 0xFD:
+      return this.word16le();
+
+    case 0xFE:
+      return this.word32le();
+
+    case 0xFF:
+      return this.word64le();
+
+    default:
+      return firstByte;
+    }
+  };
+
+  Parser.prototype.varStr = function () {
+    var len = this.varInt();
+    return this.buffer(len);
+  };
+
   return Parser;
 };
 module.defineClass(spec);