123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120 |
- /*!
- * ws: a node.js websocket client
- * Copyright(c) 2011 Einar Otto Stangvik <einaros@gmail.com>
- * MIT Licensed
- */
-
- var events = require('events')
- , util = require('util')
- , EventEmitter = events.EventEmitter;
-
- /**
- * Hixie Sender implementation
- */
-
- function Sender(socket) {
- events.EventEmitter.call(this);
-
- this.socket = socket;
- this.continuationFrame = false;
- this.isClosed = false;
- }
-
- module.exports = Sender;
-
- /**
- * Inherits from EventEmitter.
- */
-
- util.inherits(Sender, events.EventEmitter);
-
- /**
- * Frames and writes data.
- *
- * @api public
- */
-
- Sender.prototype.send = function(data, options, cb) {
- if (this.isClosed) return;
-
- var isString = typeof data == 'string'
- , length = isString ? Buffer.byteLength(data) : data.length
- , lengthbytes = (length > 127) ? 2 : 1 // assume less than 2**14 bytes
- , writeStartMarker = this.continuationFrame == false
- , writeEndMarker = !options || !(typeof options.fin != 'undefined' && !options.fin)
- , buffer = new Buffer((writeStartMarker ? ((options && options.binary) ? (1 + lengthbytes) : 1) : 0) + length + ((writeEndMarker && !(options && options.binary)) ? 1 : 0))
- , offset = writeStartMarker ? 1 : 0;
-
- if (writeStartMarker) {
- if (options && options.binary) {
- buffer.write('\x80', 'binary');
- // assume length less than 2**14 bytes
- if (lengthbytes > 1)
- buffer.write(String.fromCharCode(128+length/128), offset++, 'binary');
- buffer.write(String.fromCharCode(length&0x7f), offset++, 'binary');
- } else
- buffer.write('\x00', 'binary');
- }
-
- if (isString) buffer.write(data, offset, 'utf8');
- else data.copy(buffer, offset, 0);
-
- if (writeEndMarker) {
- if (options && options.binary) {
- // sending binary, not writing end marker
- } else
- buffer.write('\xff', offset + length, 'binary');
- this.continuationFrame = false;
- }
- else this.continuationFrame = true;
-
- try {
- this.socket.write(buffer, 'binary', cb);
- } catch (e) {
- this.error(e.toString());
- }
- };
-
- /**
- * Sends a close instruction to the remote party.
- *
- * @api public
- */
-
- Sender.prototype.close = function(code, data, mask, cb) {
- if (this.isClosed) return;
- this.isClosed = true;
- try {
- if (this.continuationFrame) this.socket.write(new Buffer([0xff], 'binary'));
- this.socket.write(new Buffer([0xff, 0x00]), 'binary', cb);
- } catch (e) {
- this.error(e.toString());
- }
- };
-
- /**
- * Sends a ping message to the remote party. Not available for hixie.
- *
- * @api public
- */
-
- Sender.prototype.ping = function(data, options) {};
-
- /**
- * Sends a pong message to the remote party. Not available for hixie.
- *
- * @api public
- */
-
- Sender.prototype.pong = function(data, options) {};
-
- /**
- * Handles an error
- *
- * @api private
- */
-
- Sender.prototype.error = function (reason) {
- this.emit('error', reason);
- return this;
- };
|