GitHub/dbot
3
0
Fork 1
mirror of https://github.com/reality/dbot.git synced 2024-12-25 04:02:39 +01:00
Code Issues Projects Releases Wiki Activity

fix crash on too big of a request [Close #332]

Browse Source
This commit is contained in:
reality 2013-03-25 18:23:35 +00:00
parent c376ab97ac
commit 6176dd5695
1 changed files with 11 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
modules/link/link.js
View File

@ -11,7 +11,9 @@ var link = function(dbot) {
this.urlRegex = /(\b(https?|ftp|file):\/\/[-A-Z0-9+&@#\/%?=~_|!:,.;]*[-A-Z0-9+&@#\/%=~_|])/ig; this.urlRegex = /(\b(https?|ftp|file):\/\/[-A-Z0-9+&@#\/%?=~_|!:,.;]*[-A-Z0-9+&@#\/%=~_|])/ig;
this.links = {}; this.links = {};
this.fetchTitle = function(event, link) { this.fetchTitle = function(event, link) {
request(link, function(error, response, body) { var limit = 25 * 1000,
size = 0,
page = request(link, function(error, response, body) {
if(!error && response.statusCode == 200) { if(!error && response.statusCode == 200) {
body = body.replace(/(\r\n|\n\r|\n)/gm, " "); body = body.replace(/(\r\n|\n\r|\n)/gm, " ");
var title = body.valMatch(/<title>(.*)<\/title>/, 2); var title = body.valMatch(/<title>(.*)<\/title>/, 2);
@ -20,6 +22,14 @@ var link = function(dbot) {
} }
} }
}); });
page.on('data', function(chunk) {
size += chunk.length;
if(size > limit) {
page.abort();
event.reply('request too big, fuck off');
}
});
}; };
var commands = { var commands = {