CODEWITHSUNDEEP
javascriptnode.jssockets
Andrew White
Andrew White

Reputation: 610

Getting TCP socket specific data on data response

I've written the following function:

  let responses = {}
  let socks = {}
  module.ping = function (port, address) {
    //console.log(`Ping function was called - ${address} ${port}`)
    if (socks[`${address}:${port}`]) {
      //console.log("Using existing socket")
      responses[`${address}:${port}`] = false
      sock = socks[`${address}:${port}`]
      sock.write('PING\n')
      console.log(`Sent PING to ${address} ${port}`)
    }
    else {
      sock = new net.Socket();
      responses[`${address}:${port}`] = false
      sock.connect(port, address, async function() {
        sock.write('PING\n')
        console.log(`Sent PING to ${address} ${port}`)
      });

      // Response listeners
      sock.on('data', function(data) {
        clean_data = data.toString().replace(/\n/g, '').replace(/\r/g, '')
        console.log(`[${sock.remoteAddress}:${sock.remotePort}] Received ${clean_data}`)
          if (clean_data == 'PONG') {
            //console.log(`[${sock.remoteAddress}:${sock.remotePort}] Received PONG`)
            //sock.end()
            //delete socks[`${address}:${port}`]
            responses[`${sock.remoteAddress}:${sock.remotePort}`] = true
          }
        }
      });
      sock.on('error', function(error) { 
        if (sock.remoteAddress) {
          responses[`${sock.remoteAddress}:${sock.remotePort}`] = false
        }
        sock.destroy()
        delete socks[`${address}:${port}`]
      });

      // Add to list of sockets
      if (sock) {
        socks[`${address}:${port}`] = sock 
      }
    }
  }

On the other end, I have a listening TCP server that simple responds with "PONG\n". When I try a single host I get the expected output:

module.ping(1337, 10.0.0.100)
await delay(5000) // Custom function
module.ping(1337, 10.0.0.100)

Sent PING to 10.0.0.100 1337

[10.0.0.100:1337] Received PONG

Sent PING to 10.0.0.100 1337

[10.0.0.100:1337] Received PONG

However when I attempt to hit multiple hosts:

module.ping(1337, 10.0.0.100)
module.ping(1337, 10.0.0.200)
await delay(5000) // Custom function
module.ping(1337, 10.0.0.100)
module.ping(1337, 10.0.0.200)

Sent PING to 10.0.0.100 1337

Sent PING to 10.0.0.200 1337

[10.0.0.200:1337] Received PONG

[10.0.0.200:1337] Received PONG

Sent PING to 10.0.0.100 1337

Sent PING to 10.0.0.200 1337

[10.0.0.200:1337] Received PONG

[10.0.0.200:1337] Received PONG

It seems to me that the event listener for "data" I've added has somehow bound to the incorrect address within my code, however I can't see where. If I add more hosts to the list, the last host with the PING sent is the one that all the PONGs are marked as part of.

Upvotes: 1

Views: 178

Answers (1)

jfriend00
jfriend00

Reputation: 707158

For starters, I see no declaration for the sock variable which means it's in some some higher scope and getting wrongly confused or trounced between different asynchronous operations that are both in progress.

Declare that variable locally so each use of it is a separate variable and one async operation won't overwrite the one you were using for another operation. ALL variables, not purposely intended to be higher scoped and shared MUST be declared locally.

I don't know if that is the only problem here, but it IS a problem here. For example, if you call .ping() twice in a row and have to create two new sockets, the second one will overwrite the sock variable before the .connect() succeeds causing you to send the PING to the wrong sock which is exactly what your diagnostic logs show.

Upvotes: 2

Related Questions