diff --git a/lib/tcp.js b/lib/tcp.js index cd263d9..ef9c218 100755 --- a/lib/tcp.js +++ b/lib/tcp.js @@ -1,6 +1,7 @@ // Copyright 2013 Mark Cavage, Inc. All rights reserved. var net = require('net'); +var tls = require('tls'); var util = require('util'); var assert = require('assert-plus'); @@ -29,8 +30,10 @@ function createSocket(opts) { assert.string(opts.host, 'options.host'); assert.number(opts.port, 'options.port'); assert.object(opts.proxy, 'options.proxy'); + assert.optionalBool(opts.tls, 'options.tls'); - var s = net.connect({ + var transport = opts.tls ? tls : net; + var s = transport.connect({ host: opts.host, port: opts.port }); @@ -57,6 +60,7 @@ function TCPStream(opts) { this.host = opts.host || '127.0.0.1'; this.port = opts.port || 514; + this.tls = opts.tls || false; this.queue = []; @@ -79,7 +83,8 @@ function TCPStream(opts) { self.socket = createSocket({ host: self.host, port: self.port, - proxy: self + proxy: self, + tls: self.tls }); self.socket.on('close', setTimeout.bind(null, connect, 1000)); self.socket.on('error', setTimeout.bind(null, connect, 1000));