100 lines
2.6 KiB
JavaScript
100 lines
2.6 KiB
JavaScript
'use strict';
|
|
|
|
var Buffer = require('safe-buffer').Buffer,
|
|
Stream = require('stream').Stream,
|
|
url = require('url'),
|
|
util = require('util'),
|
|
Base = require('./base'),
|
|
Headers = require('./headers'),
|
|
HttpParser = require('../http_parser');
|
|
|
|
var PORTS = { 'ws:': 80, 'wss:': 443 };
|
|
|
|
var Proxy = function(client, origin, options) {
|
|
this._client = client;
|
|
this._http = new HttpParser('response');
|
|
this._origin = (typeof client.url === 'object') ? client.url : url.parse(client.url);
|
|
this._url = (typeof origin === 'object') ? origin : url.parse(origin);
|
|
this._options = options || {};
|
|
this._state = 0;
|
|
|
|
this.readable = this.writable = true;
|
|
this._paused = false;
|
|
|
|
this._headers = new Headers();
|
|
this._headers.set('Host', this._origin.host);
|
|
this._headers.set('Connection', 'keep-alive');
|
|
this._headers.set('Proxy-Connection', 'keep-alive');
|
|
|
|
var auth = this._url.auth && Buffer.from(this._url.auth, 'utf8').toString('base64');
|
|
if (auth) this._headers.set('Proxy-Authorization', 'Basic ' + auth);
|
|
};
|
|
util.inherits(Proxy, Stream);
|
|
|
|
var instance = {
|
|
setHeader: function(name, value) {
|
|
if (this._state !== 0) return false;
|
|
this._headers.set(name, value);
|
|
return true;
|
|
},
|
|
|
|
start: function() {
|
|
if (this._state !== 0) return false;
|
|
this._state = 1;
|
|
|
|
var origin = this._origin,
|
|
port = origin.port || PORTS[origin.protocol],
|
|
start = 'CONNECT ' + origin.hostname + ':' + port + ' HTTP/1.1';
|
|
|
|
var headers = [start, this._headers.toString(), ''];
|
|
|
|
this.emit('data', Buffer.from(headers.join('\r\n'), 'utf8'));
|
|
return true;
|
|
},
|
|
|
|
pause: function() {
|
|
this._paused = true;
|
|
},
|
|
|
|
resume: function() {
|
|
this._paused = false;
|
|
this.emit('drain');
|
|
},
|
|
|
|
write: function(chunk) {
|
|
if (!this.writable) return false;
|
|
|
|
this._http.parse(chunk);
|
|
if (!this._http.isComplete()) return !this._paused;
|
|
|
|
this.statusCode = this._http.statusCode;
|
|
this.headers = this._http.headers;
|
|
|
|
if (this.statusCode === 200) {
|
|
this.emit('connect', new Base.ConnectEvent());
|
|
} else {
|
|
var message = "Can't establish a connection to the server at " + this._origin.href;
|
|
this.emit('error', new Error(message));
|
|
}
|
|
this.end();
|
|
return !this._paused;
|
|
},
|
|
|
|
end: function(chunk) {
|
|
if (!this.writable) return;
|
|
if (chunk !== undefined) this.write(chunk);
|
|
this.readable = this.writable = false;
|
|
this.emit('close');
|
|
this.emit('end');
|
|
},
|
|
|
|
destroy: function() {
|
|
this.end();
|
|
}
|
|
};
|
|
|
|
for (var key in instance)
|
|
Proxy.prototype[key] = instance[key];
|
|
|
|
module.exports = Proxy;
|