From c7852c6ec0f423cd30b26cd6d8220cbe5eb6413d Mon Sep 17 00:00:00 2001
From: abhishek <abhishek@e0a8ed71-7df4-0310-8962-fdc924857419>
Date: Tue, 24 May 2016 19:23:20 +0000
Subject: [PATCH] Adds Pong response, closes #383

---
 nselib/bitcoin.lua | 34 +++++++++++++++++++++++++++++++++-
 1 file changed, 33 insertions(+), 1 deletion(-)

diff --git a/nselib/bitcoin.lua b/nselib/bitcoin.lua
index 750e6ebe5..f495cd621 100644
--- a/nselib/bitcoin.lua
+++ b/nselib/bitcoin.lua
@@ -188,6 +188,26 @@ Request = {
       return bin.pack("<IAII", 0xD9B4BEF9, "verack\0\0\0\0\0\0", 0, 0xe2e0f65d)
     end,
 
+   },
+ 
+  -- The pong message is sent in response to a ping message.
+  Pong = {
+    new = function(self)
+      local o = {}
+      setmetatable(o, self)
+      self.__index = self
+      return o
+    end,
+
+    __tostring = function(self)
+      local magic = 0xD9B4BEF9
+      local cmd = "pong\0\0\0\0\0\0\0\0"
+      local len = 0
+      local chksum = 0xe2e0f65d
+
+      return bin.pack("<IAII", magic, cmd, len, chksum)
+    end,
+ 
   }
 
 }
@@ -384,12 +404,24 @@ Response = {
     local pos, magic, cmd, len, checksum = bin.unpack("<IA12II", header)
     local data = ""
 
-    -- the verack has no payload
+    -- the verack and ping has no payload
     if ( 0 ~= len ) then
       status, data = socket:receive_buf(match.numbytes(len), true)
       if ( not(status) ) then
         return false, "Failed to read the packet header"
       end
+    else
+      -- The ping message is sent primarily to confirm that the TCP/IP connection is still valid.
+      if( cmd == "ping\0\0\0\0\0\0\0\0" ) then
+        local req = Request.Pong:new()
+
+        local status, err = socket:send(tostring(req))
+        if ( not(status) ) then
+          return false, "Failed to send \"Pong\" reply to server"
+        else
+          return Response.recvPacket(socket, version)
+        end
+      end
     end
     return Response.decode(header .. data, version)
   end,