var forge = require('./forge');
require('./debug');
require('./tls');
require('./util');
var http = module.exports = forge.http = forge.http || {};
var cat = 'forge.http';
if(forge.debug) {
forge.debug.set('forge.http', 'clients', []);
}
var _normalize = function(name) {
return name.toLowerCase().replace(/(^.)|(-.)/g,
function(a) {return a.toUpperCase();});
};
var _getStorageId = function(client) {
return 'forge.http.' +
client.url.scheme + '.' +
client.url.host + '.' +
client.url.port;
};
var _loadCookies = function(client) {
if(client.persistCookies) {
try {
var cookies = forge.util.getItem(
client.socketPool.flashApi,
_getStorageId(client), 'cookies');
client.cookies = cookies || {};
} catch(ex) {
}
}
};
var _saveCookies = function(client) {
if(client.persistCookies) {
try {
forge.util.setItem(
client.socketPool.flashApi,
_getStorageId(client), 'cookies', client.cookies);
} catch(ex) {
}
}
_loadCookies(client);
};
var _clearCookies = function(client) {
if(client.persistCookies) {
try {
forge.util.clearItems(
client.socketPool.flashApi,
_getStorageId(client));
} catch(ex) {
}
}
};
var _doRequest = function(client, socket) {
if(socket.isConnected()) {
socket.options.request.connectTime = +new Date();
socket.connected({
type: 'connect',
id: socket.id
});
} else {
socket.options.request.connectTime = +new Date();
socket.connect({
host: client.url.host,
port: client.url.port,
policyPort: client.policyPort,
policyUrl: client.policyUrl
});
}
};
var _handleNextRequest = function(client, socket) {
socket.buffer.clear();
var pending = null;
while(pending === null && client.requests.length > 0) {
pending = client.requests.shift();
if(pending.request.aborted) {
pending = null;
}
}
if(pending === null) {
if(socket.options !== null) {
socket.options = null;
}
client.idle.push(socket);
} else {
socket.retries = 1;
socket.options = pending;
_doRequest(client, socket);
}
};
var _initSocket = function(client, socket, tlsOptions) {
socket.options = null;
socket.connected = function(e) {
if(socket.options === null) {
_handleNextRequest(client, socket);
} else {
var request = socket.options.request;
request.connectTime = +new Date() - request.connectTime;
e.socket = socket;
socket.options.connected(e);
if(request.aborted) {
socket.close();
} else {
var out = request.toString();
if(request.body) {
out += request.body;
}
request.time = +new Date();
socket.send(out);
request.time = +new Date() - request.time;
socket.options.response.time = +new Date();
socket.sending = true;
}
}
};
socket.closed = function(e) {
if(socket.sending) {
socket.sending = false;
if(socket.retries > 0) {
--socket.retries;
_doRequest(client, socket);
} else {
socket.error({
id: socket.id,
type: 'ioError',
message: 'Connection closed during send. Broken pipe.',
bytesAvailable: 0
});
}
} else {
var response = socket.options.response;
if(response.readBodyUntilClose) {
response.time = +new Date() - response.time;
response.bodyReceived = true;
socket.options.bodyReady({
request: socket.options.request,
response: response,
socket: socket
});
}
socket.options.closed(e);
_handleNextRequest(client, socket);
}
};
socket.data = function(e) {
socket.sending = false;
var request = socket.options.request;
if(request.aborted) {
socket.close();
} else {
var response = socket.options.response;
var bytes = socket.receive(e.bytesAvailable);
if(bytes !== null) {
socket.buffer.putBytes(bytes);
if(!response.headerReceived) {
response.readHeader(socket.buffer);
if(response.headerReceived) {
socket.options.headerReady({
request: socket.options.request,
response: response,
socket: socket
});
}
}
if(response.headerReceived && !response.bodyReceived) {
response.readBody(socket.buffer);
}
if(response.bodyReceived) {
socket.options.bodyReady({
request: socket.options.request,
response: response,
socket: socket
});
var value = response.getField('Connection') || '';
if(value.indexOf('close') != -1 ||
(response.version === 'HTTP/1.0' &&
response.getField('Keep-Alive') === null)) {
socket.close();
} else {
_handleNextRequest(client, socket);
}
}
}
}
};
socket.error = function(e) {
socket.options.error({
type: e.type,
message: e.message,
request: socket.options.request,
response: socket.options.response,
socket: socket
});
socket.close();
};
if(tlsOptions) {
socket = forge.tls.wrapSocket({
sessionId: null,
sessionCache: {},
caStore: tlsOptions.caStore,
cipherSuites: tlsOptions.cipherSuites,
socket: socket,
virtualHost: tlsOptions.virtualHost,
verify: tlsOptions.verify,
getCertificate: tlsOptions.getCertificate,
getPrivateKey: tlsOptions.getPrivateKey,
getSignature: tlsOptions.getSignature,
deflate: tlsOptions.deflate || null,
inflate: tlsOptions.inflate || null
});
socket.options = null;
socket.buffer = forge.util.createBuffer();
client.sockets.push(socket);
if(tlsOptions.prime) {
socket.connect({
host: client.url.host,
port: client.url.port,
policyPort: client.policyPort,
policyUrl: client.policyUrl
});
} else {
client.idle.push(socket);
}
} else {
socket.buffer = forge.util.createBuffer();
client.sockets.push(socket);
client.idle.push(socket);
}
};
var _hasCookieExpired = function(cookie) {
var rval = false;
if(cookie.maxAge !== -1) {
var now = _getUtcTime(new Date());
var expires = cookie.created + cookie.maxAge;
if(expires <= now) {
rval = true;
}
}
return rval;
};
var _writeCookies = function(client, request) {
var expired = [];
var url = client.url;
var cookies = client.cookies;
for(var name in cookies) {
var paths = cookies[name];
for(var p in paths) {
var cookie = paths[p];
if(_hasCookieExpired(cookie)) {
expired.push(cookie);
} else if(request.path.indexOf(cookie.path) === 0) {
request.addCookie(cookie);
}
}
}
for(var i = 0; i < expired.length; ++i) {
var cookie = expired[i];
client.removeCookie(cookie.name, cookie.path);
}
};
var _readCookies = function(client, response) {
var cookies = response.getCookies();
for(var i = 0; i < cookies.length; ++i) {
try {
client.setCookie(cookies[i]);
} catch(ex) {
}
}
};
http.createClient = function(options) {
var caStore = null;
if(options.caCerts) {
caStore = forge.pki.createCaStore(options.caCerts);
}
options.url = (options.url ||
window.location.protocol + '//' + window.location.host);
var url = http.parseUrl(options.url);
if(!url) {
var error = new Error('Invalid url.');
error.details = {url: options.url};
throw error;
}
options.connections = options.connections || 1;
var sp = options.socketPool;
var client = {
url: url,
socketPool: sp,
policyPort: options.policyPort,
policyUrl: options.policyUrl,
requests: [],
sockets: [],
idle: [],
secure: (url.scheme === 'https'),
cookies: {},
persistCookies: (typeof(options.persistCookies) === 'undefined') ?
true : options.persistCookies
};
if(forge.debug) {
forge.debug.get('forge.http', 'clients').push(client);
}
_loadCookies(client);
var _defaultCertificateVerify = function(c, verified, depth, certs) {
if(depth === 0 && verified === true) {
var cn = certs[depth].subject.getField('CN');
if(cn === null || client.url.host !== cn.value) {
verified = {
message: 'Certificate common name does not match url host.'
};
}
}
return verified;
};
var tlsOptions = null;
if(client.secure) {
tlsOptions = {
caStore: caStore,
cipherSuites: options.cipherSuites || null,
virtualHost: options.virtualHost || url.host,
verify: options.verify || _defaultCertificateVerify,
getCertificate: options.getCertificate || null,
getPrivateKey: options.getPrivateKey || null,
getSignature: options.getSignature || null,
prime: options.primeTlsSockets || false
};
if(sp.flashApi !== null) {
tlsOptions.deflate = function(bytes) {
return forge.util.deflate(sp.flashApi, bytes, true);
};
tlsOptions.inflate = function(bytes) {
return forge.util.inflate(sp.flashApi, bytes, true);
};
}
}
for(var i = 0; i < options.connections; ++i) {
_initSocket(client, sp.createSocket(), tlsOptions);
}
client.send = function(options) {
if(options.request.getField('Host') === null) {
options.request.setField('Host', client.url.fullHost);
}
var opts = {};
opts.request = options.request;
opts.connected = options.connected || function() {};
opts.closed = options.close || function() {};
opts.headerReady = function(e) {
_readCookies(client, e.response);
if(options.headerReady) {
options.headerReady(e);
}
};
opts.bodyReady = options.bodyReady || function() {};
opts.error = options.error || function() {};
opts.response = http.createResponse();
opts.response.time = 0;
opts.response.flashApi = client.socketPool.flashApi;
opts.request.flashApi = client.socketPool.flashApi;
opts.request.abort = function() {
opts.request.aborted = true;
opts.connected = function() {};
opts.closed = function() {};
opts.headerReady = function() {};
opts.bodyReady = function() {};
opts.error = function() {};
};
_writeCookies(client, opts.request);
if(client.idle.length === 0) {
client.requests.push(opts);
} else {
var socket = null;
var len = client.idle.length;
for(var i = 0; socket === null && i < len; ++i) {
socket = client.idle[i];
if(socket.isConnected()) {
client.idle.splice(i, 1);
} else {
socket = null;
}
}
if(socket === null) {
socket = client.idle.pop();
}
socket.options = opts;
_doRequest(client, socket);
}
};
client.destroy = function() {
client.requests = [];
for(var i = 0; i < client.sockets.length; ++i) {
client.sockets[i].close();
client.sockets[i].destroy();
}
client.socketPool = null;
client.sockets = [];
client.idle = [];
};
client.setCookie = function(cookie) {
var rval;
if(typeof(cookie.name) !== 'undefined') {
if(cookie.value === null || typeof(cookie.value) === 'undefined' ||
cookie.value === '') {
rval = client.removeCookie(cookie.name, cookie.path);
} else {
cookie.comment = cookie.comment || '';
cookie.maxAge = cookie.maxAge || 0;
cookie.secure = (typeof(cookie.secure) === 'undefined') ?
true : cookie.secure;
cookie.httpOnly = cookie.httpOnly || true;
cookie.path = cookie.path || '/';
cookie.domain = cookie.domain || null;
cookie.version = cookie.version || null;
cookie.created = _getUtcTime(new Date());
if(cookie.secure !== client.secure) {
var error = new Error('Http client url scheme is incompatible ' +
'with cookie secure flag.');
error.url = client.url;
error.cookie = cookie;
throw error;
}
if(!http.withinCookieDomain(client.url, cookie)) {
var error = new Error('Http client url scheme is incompatible ' +
'with cookie secure flag.');
error.url = client.url;
error.cookie = cookie;
throw error;
}
if(!(cookie.name in client.cookies)) {
client.cookies[cookie.name] = {};
}
client.cookies[cookie.name][cookie.path] = cookie;
rval = true;
_saveCookies(client);
}
}
return rval;
};
client.getCookie = function(name, path) {
var rval = null;
if(name in client.cookies) {
var paths = client.cookies[name];
if(path) {
if(path in paths) {
rval = paths[path];
}
} else {
for(var p in paths) {
rval = paths[p];
break;
}
}
}
return rval;
};
client.removeCookie = function(name, path) {
var rval = false;
if(name in client.cookies) {
if(path) {
var paths = client.cookies[name];
if(path in paths) {
rval = true;
delete client.cookies[name][path];
var empty = true;
for(var i in client.cookies[name]) {
empty = false;
break;
}
if(empty) {
delete client.cookies[name];
}
}
} else {
rval = true;
delete client.cookies[name];
}
}
if(rval) {
_saveCookies(client);
}
return rval;
};
client.clearCookies = function() {
client.cookies = {};
_clearCookies(client);
};
if(forge.log) {
forge.log.debug('forge.http', 'created client', options);
}
return client;
};
var _trimString = function(str) {
return str.replace(/^\s*/, '').replace(/\s*$/, '');
};
var _createHeader = function() {
var header = {
fields: {},
setField: function(name, value) {
header.fields[_normalize(name)] = [_trimString('' + value)];
},
appendField: function(name, value) {
name = _normalize(name);
if(!(name in header.fields)) {
header.fields[name] = [];
}
header.fields[name].push(_trimString('' + value));
},
getField: function(name, index) {
var rval = null;
name = _normalize(name);
if(name in header.fields) {
index = index || 0;
rval = header.fields[name][index];
}
return rval;
}
};
return header;
};
var _getUtcTime = function(d) {
var utc = +d + d.getTimezoneOffset() * 60000;
return Math.floor(+new Date() / 1000);
};
http.createRequest = function(options) {
options = options || {};
var request = _createHeader();
request.version = options.version || 'HTTP/1.1';
request.method = options.method || null;
request.path = options.path || null;
request.body = options.body || null;
request.bodyDeflated = false;
request.flashApi = null;
var headers = options.headers || [];
if(!forge.util.isArray(headers)) {
headers = [headers];
}
for(var i = 0; i < headers.length; ++i) {
for(var name in headers[i]) {
request.appendField(name, headers[i][name]);
}
}
request.addCookie = function(cookie) {
var value = '';
var field = request.getField('Cookie');
if(field !== null) {
value = field + '; ';
}
var now = _getUtcTime(new Date());
value += cookie.name + '=' + cookie.value;
request.setField('Cookie', value);
};
request.toString = function() {
if(request.getField('User-Agent') === null) {
request.setField('User-Agent', 'forge.http 1.0');
}
if(request.getField('Accept') === null) {
request.setField('Accept', '*/*');
}
if(request.getField('Connection') === null) {
request.setField('Connection', 'keep-alive');
request.setField('Keep-Alive', '115');
}
if(request.flashApi !== null &&
request.getField('Accept-Encoding') === null) {
request.setField('Accept-Encoding', 'deflate');
}
if(request.flashApi !== null && request.body !== null &&
request.getField('Content-Encoding') === null &&
!request.bodyDeflated && request.body.length > 100) {
request.body = forge.util.deflate(request.flashApi, request.body);
request.bodyDeflated = true;
request.setField('Content-Encoding', 'deflate');
request.setField('Content-Length', request.body.length);
} else if(request.body !== null) {
request.setField('Content-Length', request.body.length);
}
var rval =
request.method.toUpperCase() + ' ' + request.path + ' ' +
request.version + '\r\n';
for(var name in request.fields) {
var fields = request.fields[name];
for(var i = 0; i < fields.length; ++i) {
rval += name + ': ' + fields[i] + '\r\n';
}
}
rval += '\r\n';
return rval;
};
return request;
};
http.createResponse = function() {
var _first = true;
var _chunkSize = 0;
var _chunksFinished = false;
var response = _createHeader();
response.version = null;
response.code = 0;
response.message = null;
response.body = null;
response.headerReceived = false;
response.bodyReceived = false;
response.flashApi = null;
var _readCrlf = function(b) {
var line = null;
var i = b.data.indexOf('\r\n', b.read);
if(i != -1) {
line = b.getBytes(i - b.read);
b.getBytes(2);
}
return line;
};
var _parseHeader = function(line) {
var tmp = line.indexOf(':');
var name = line.substring(0, tmp++);
response.appendField(
name, (tmp < line.length) ? line.substring(tmp) : '');
};
response.readHeader = function(b) {
var line = '';
while(!response.headerReceived && line !== null) {
line = _readCrlf(b);
if(line !== null) {
if(_first) {
_first = false;
var tmp = line.split(' ');
if(tmp.length >= 3) {
response.version = tmp[0];
response.code = parseInt(tmp[1], 10);
response.message = tmp.slice(2).join(' ');
} else {
var error = new Error('Invalid http response header.');
error.details = {'line': line};
throw error;
}
} else if(line.length === 0) {
response.headerReceived = true;
} else {
_parseHeader(line);
}
}
}
return response.headerReceived;
};
var _readChunkedBody = function(b) {
var line = '';
while(line !== null && b.length() > 0) {
if(_chunkSize > 0) {
if(_chunkSize + 2 > b.length()) {
break;
}
response.body += b.getBytes(_chunkSize);
b.getBytes(2);
_chunkSize = 0;
} else if(!_chunksFinished) {
line = _readCrlf(b);
if(line !== null) {
_chunkSize = parseInt(line.split(';', 1)[0], 16);
_chunksFinished = (_chunkSize === 0);
}
} else {
line = _readCrlf(b);
while(line !== null) {
if(line.length > 0) {
_parseHeader(line);
line = _readCrlf(b);
} else {
response.bodyReceived = true;
line = null;
}
}
}
}
return response.bodyReceived;
};
response.readBody = function(b) {
var contentLength = response.getField('Content-Length');
var transferEncoding = response.getField('Transfer-Encoding');
if(contentLength !== null) {
contentLength = parseInt(contentLength);
}
if(contentLength !== null && contentLength >= 0) {
response.body = response.body || '';
response.body += b.getBytes(contentLength);
response.bodyReceived = (response.body.length === contentLength);
} else if(transferEncoding !== null) {
if(transferEncoding.indexOf('chunked') != -1) {
response.body = response.body || '';
_readChunkedBody(b);
} else {
var error = new Error('Unknown Transfer-Encoding.');
error.details = {'transferEncoding': transferEncoding};
throw error;
}
} else if((contentLength !== null && contentLength < 0) ||
(contentLength === null &&
response.getField('Content-Type') !== null)) {
response.body = response.body || '';
response.body += b.getBytes();
response.readBodyUntilClose = true;
} else {
response.body = null;
response.bodyReceived = true;
}
if(response.bodyReceived) {
response.time = +new Date() - response.time;
}
if(response.flashApi !== null &&
response.bodyReceived && response.body !== null &&
response.getField('Content-Encoding') === 'deflate') {
response.body = forge.util.inflate(
response.flashApi, response.body);
}
return response.bodyReceived;
};
response.getCookies = function() {
var rval = [];
if('Set-Cookie' in response.fields) {
var field = response.fields['Set-Cookie'];
var now = +new Date() / 1000;
var regex = /\s*([^=]*)=?([^;]*)(;|$)/g;
for(var i = 0; i < field.length; ++i) {
var fv = field[i];
var m;
regex.lastIndex = 0;
var first = true;
var cookie = {};
do {
m = regex.exec(fv);
if(m !== null) {
var name = _trimString(m[1]);
var value = _trimString(m[2]);
if(first) {
cookie.name = name;
cookie.value = value;
first = false;
} else {
name = name.toLowerCase();
switch(name) {
case 'expires':
value = value.replace(/-/g, ' ');
var secs = Date.parse(value) / 1000;
cookie.maxAge = Math.max(0, secs - now);
break;
case 'max-age':
cookie.maxAge = parseInt(value, 10);
break;
case 'secure':
cookie.secure = true;
break;
case 'httponly':
cookie.httpOnly = true;
break;
default:
if(name !== '') {
cookie[name] = value;
}
}
}
}
} while(m !== null && m[0] !== '');
rval.push(cookie);
}
}
return rval;
};
response.toString = function() {
var rval =
response.version + ' ' + response.code + ' ' + response.message + '\r\n';
for(var name in response.fields) {
var fields = response.fields[name];
for(var i = 0; i < fields.length; ++i) {
rval += name + ': ' + fields[i] + '\r\n';
}
}
rval += '\r\n';
return rval;
};
return response;
};
http.parseUrl = forge.util.parseUrl;
http.withinCookieDomain = function(url, cookie) {
var rval = false;
var domain = (cookie === null || typeof cookie === 'string') ?
cookie : cookie.domain;
if(domain === null) {
rval = true;
} else if(domain.charAt(0) === '.') {
if(typeof url === 'string') {
url = http.parseUrl(url);
}
var host = '.' + url.host;
var idx = host.lastIndexOf(domain);
if(idx !== -1 && (idx + domain.length === host.length)) {
rval = true;
}
}
return rval;
};