www.aleph1.co.uk Git - yaffs-website/blob - node_modules/websocket-driver/lib/websocket/driver/draft76.js

   1 'use strict';
   2
   3 var Base    = require('./base'),
   4     Draft75 = require('./draft75'),
   5     crypto  = require('crypto'),
   6     util    = require('util');
   7
   8
   9 var numberFromKey = function(key) {
  10   return parseInt(key.match(/[0-9]/g).join(''), 10);
  11 };
  12
  13 var spacesInKey = function(key) {
  14   return key.match(/ /g).length;
  15 };
  16
  17
  18 var Draft76 = function(request, url, options) {
  19   Draft75.apply(this, arguments);
  20   this._stage  = -1;
  21   this._body   = [];
  22   this.version = 'hixie-76';
  23
  24   this._headers.clear();
  25
  26   this._headers.set('Upgrade', 'WebSocket');
  27   this._headers.set('Connection', 'Upgrade');
  28   this._headers.set('Sec-WebSocket-Origin', this._request.headers.origin);
  29   this._headers.set('Sec-WebSocket-Location', this.url);
  30 };
  31 util.inherits(Draft76, Draft75);
  32
  33 var instance = {
  34   BODY_SIZE: 8,
  35
  36   start: function() {
  37     if (!Draft75.prototype.start.call(this)) return false;
  38     this._started = true;
  39     this._sendHandshakeBody();
  40     return true;
  41   },
  42
  43   close: function() {
  44     if (this.readyState === 3) return false;
  45     this._write(new Buffer([0xFF, 0x00]));
  46     this.readyState = 3;
  47     this.emit('close', new Base.CloseEvent(null, null));
  48     return true;
  49   },
  50
  51   _handshakeResponse: function() {
  52     var headers = this._request.headers,
  53
  54         key1    = headers['sec-websocket-key1'],
  55         number1 = numberFromKey(key1),
  56         spaces1 = spacesInKey(key1),
  57
  58         key2    = headers['sec-websocket-key2'],
  59         number2 = numberFromKey(key2),
  60         spaces2 = spacesInKey(key2);
  61
  62     if (number1 % spaces1 !== 0 || number2 % spaces2 !== 0) {
  63       this.emit('error', new Error('Client sent invalid Sec-WebSocket-Key headers'));
  64       this.close();
  65       return null;
  66     }
  67
  68     this._keyValues = [number1 / spaces1, number2 / spaces2];
  69
  70     var start   = 'HTTP/1.1 101 WebSocket Protocol Handshake',
  71         headers = [start, this._headers.toString(), ''];
  72
  73     return new Buffer(headers.join('\r\n'), 'binary');
  74   },
  75
  76   _handshakeSignature: function() {
  77     if (this._body.length < this.BODY_SIZE) return null;
  78
  79     var md5    = crypto.createHash('md5'),
  80         buffer = new Buffer(8 + this.BODY_SIZE);
  81
  82     buffer.writeUInt32BE(this._keyValues[0], 0);
  83     buffer.writeUInt32BE(this._keyValues[1], 4);
  84     new Buffer(this._body).copy(buffer, 8, 0, this.BODY_SIZE);
  85
  86     md5.update(buffer);
  87     return new Buffer(md5.digest('binary'), 'binary');
  88   },
  89
  90   _sendHandshakeBody: function() {
  91     if (!this._started) return;
  92     var signature = this._handshakeSignature();
  93     if (!signature) return;
  94
  95     this._write(signature);
  96     this._stage = 0;
  97     this._open();
  98
  99     if (this._body.length > this.BODY_SIZE)
 100       this.parse(this._body.slice(this.BODY_SIZE));
 101   },
 102
 103   _parseLeadingByte: function(octet) {
 104     if (octet !== 0xFF)
 105       return Draft75.prototype._parseLeadingByte.call(this, octet);
 106
 107     this._closing = true;
 108     this._length  = 0;
 109     this._stage   = 1;
 110   }
 111 };
 112
 113 for (var key in instance)
 114   Draft76.prototype[key] = instance[key];
 115
 116 module.exports = Draft76;