node tcp + http speed test

index.js

var net = require('net')
var through = require('through2')

var server = process.argv[2]
var path = process.argv[3]
var host = process.argv[4]

var socket = net.connect(80, server)

var req = ["GET " + path + " HTTP/1.1\r\n",
"User-Agent: curl/7.30.0\r\n",
"Host: " + host + "\r\n",
"Accept: */*\r\n\r\n"].join('')

console.log('connecting to', server)
console.log(req)

var start = Date.now()
var resp = new Buffer(0)
var contentLength

var concat = through(function(ch, enc, next) {
  if (typeof contentLength === 'undefined') {
    contentLength = false
    ch.toString().split('\r\r')[0].split('\r').map(function(header) {
      if (header.toLowerCase().indexOf('content-length') > -1) contentLength = +(header.split(': ')[1])
    })
  }
  resp = Buffer.concat([resp, ch])
  var end = resp.slice(resp.length - 5, resp.length)
  if (end.toString() === '0\r\n\r\n') socket.end()
  if (contentLength && resp.length >= contentLength) socket.end()
  console.log('got response chunk', ~~(ch.length / 1024) + 'kb')
  next()
})

socket.pipe(concat)
socket.write(new Buffer(req))
socket.on('end', function() {
  var elapsed = (Date.now() - start) / 1000
  console.log({elapsed: elapsed + 's', downloaded: ~~(resp.length / 1024) + 'kb', speed: ~~(resp.length / elapsed / 1024) + 'kb/s' })
})