mirror of
https://github.com/taobataoma/meanTorrent.git
synced 2026-01-16 12:22:26 +01:00
919 lines
26 KiB
JavaScript
919 lines
26 KiB
JavaScript
'use strict';
|
|
|
|
/**
|
|
* Module dependencies
|
|
*/
|
|
var path = require('path'),
|
|
config = require(path.resolve('./config/config')),
|
|
common = require(path.resolve('./config/lib/common')),
|
|
mongoose = require('mongoose'),
|
|
errorHandler = require(path.resolve('./modules/core/server/controllers/errors.server.controller')),
|
|
User = mongoose.model('User'),
|
|
Torrent = mongoose.model('Torrent'),
|
|
Peer = mongoose.model('Peer'),
|
|
Complete = mongoose.model('Complete'),
|
|
moment = require('moment'),
|
|
async = require('async'),
|
|
querystring = require('querystring'),
|
|
url = require('url'),
|
|
sprintf = require('sprintf-js').sprintf,
|
|
traceLogCreate = require(path.resolve('./config/lib/tracelog')).create,
|
|
scoreUpdate = require(path.resolve('./config/lib/score')).update;
|
|
|
|
var traceConfig = config.meanTorrentConfig.trace;
|
|
var scoreConfig = config.meanTorrentConfig.score;
|
|
var hnrConfig = config.meanTorrentConfig.hitAndRun;
|
|
var signConfig = config.meanTorrentConfig.sign;
|
|
var announceConfig = config.meanTorrentConfig.announce;
|
|
|
|
var mtDebug = require(path.resolve('./config/lib/debug'));
|
|
|
|
const FAILURE_REASONS = {
|
|
100: 'Invalid request type: client request was not a HTTP GET',
|
|
101: 'Missing info_hash',
|
|
102: 'Missing peer_id',
|
|
103: 'Missing port',
|
|
104: 'Missing passkey',
|
|
|
|
150: 'Invalid infohash: infohash is not 20 bytes long',
|
|
151: 'Invalid peerid: peerid is not 20 bytes long',
|
|
152: 'Invalid numwant. Client requested more peers than allowed by tracker',
|
|
153: 'Passkey length error (length=32)',
|
|
154: 'Invalid passkey, if you changed you passkey, please re-download the torrent file from ' + config.meanTorrentConfig.announce.baseUrl,
|
|
|
|
160: 'Invalid torrent info_hash',
|
|
161: 'No torrent with that info_hash has been found',
|
|
162: 'ip length error',
|
|
|
|
170: 'your account status is banned',
|
|
171: 'your account status is inactive',
|
|
172: 'your client is not allowed, here is the blacklist: ' + config.meanTorrentConfig.announce.clientBlackListUrl,
|
|
173: 'this torrent status is not reviewed by administrators, try again later',
|
|
174: 'this torrent is only for VIP members',
|
|
175: 'your account status is idle',
|
|
|
|
180: 'You already are downloading the same torrent. You may only leech from one location at a time',
|
|
181: 'You cannot seed the same torrent from more than 3 locations',
|
|
182: 'save peer failed',
|
|
183: 'save torrent failed',
|
|
184: 'save passkeyuser failed',
|
|
185: 'get completeTorrent failed',
|
|
186: 'create completeTorrent failed',
|
|
|
|
190: 'You have more H&R warning, can not download any torrent now!',
|
|
191: 'not find this torrent complete data',
|
|
|
|
200: 'Your total ratio is less than %.2f, can not download anything now, you can continue seed and upgrade your ratio',
|
|
201: 'You can not download from your own seeding',
|
|
|
|
600: 'This tracker only supports compact mode',
|
|
900: 'Generic error'
|
|
};
|
|
|
|
const EVENT_NONE = 0;
|
|
const EVENT_COMPLETED = 1;
|
|
const EVENT_STARTED = 2;
|
|
const EVENT_STOPPED = 3;
|
|
|
|
const WANT_DEFAULT = 50;
|
|
|
|
const PEERSTATE_SEEDER = 'seeder';
|
|
const PEERSTATE_LEECHER = 'leecher';
|
|
|
|
const PEER_COMPACT_SIZE = 6;
|
|
const ANNOUNCE_INTERVAL = Math.floor(config.meanTorrentConfig.announce.announceInterval / 1000);
|
|
const ANNOUNCE_GHOST = 2;
|
|
|
|
const PARAMS_INTEGER = [
|
|
'port', 'uploaded', 'downloaded', 'left', 'compact', 'numwant'
|
|
];
|
|
|
|
const PARAMS_STRING = [
|
|
'event'
|
|
];
|
|
|
|
/**
|
|
* event
|
|
* @param e
|
|
* @returns {number}
|
|
*/
|
|
function event(e) {
|
|
switch (e) {
|
|
case 'completed':
|
|
return EVENT_COMPLETED;
|
|
case 'started':
|
|
return EVENT_STARTED;
|
|
case 'stopped':
|
|
return EVENT_STOPPED;
|
|
}
|
|
return EVENT_NONE;
|
|
}
|
|
|
|
/**
|
|
* Failure
|
|
* @param code
|
|
* @param reason
|
|
* @constructor
|
|
*/
|
|
function Failure(code, reason) {
|
|
this.code = code;
|
|
this.reason = reason;
|
|
if (reason === undefined && typeof FAILURE_REASONS[this.code] !== 'undefined')
|
|
this.reason = FAILURE_REASONS[this.code];
|
|
else if (this.code == null)
|
|
this.code = 900;
|
|
}
|
|
|
|
/**
|
|
* Failure.prototype
|
|
* @type {{bencode: Function}}
|
|
*/
|
|
Failure.prototype = {
|
|
bencode: function () {
|
|
return 'd14:failure reason' + this.reason.length + ':' + this.reason + '12:failure codei' + this.code + 'ee';
|
|
}
|
|
};
|
|
|
|
/**
|
|
* info api
|
|
* @param req
|
|
* @param res
|
|
*/
|
|
exports.announce = function (req, res) {
|
|
req.torrent = undefined;
|
|
req.currentPeer = undefined;
|
|
req.completeTorrent = undefined;
|
|
req.selfpeer = [];
|
|
req.seeder = false;
|
|
|
|
mtDebug.debugGreen('------------ Announce request ----------------');
|
|
//mtDebug.debugGreen(req.url);
|
|
|
|
var s = req.url.split('?');
|
|
var query = common.querystringParse(s[1]);
|
|
var passkey = req.params.passkey || query.passkey || undefined;
|
|
|
|
async.waterfall([
|
|
/*---------------------------------------------------------------
|
|
validateQueryCheck
|
|
---------------------------------------------------------------*/
|
|
function (done) {
|
|
var i = 0;
|
|
var p;
|
|
|
|
if (req.method !== 'GET') {
|
|
done(100);
|
|
} else if (typeof query.info_hash === 'undefined') {
|
|
done(101);
|
|
} else if (typeof query.peer_id === 'undefined') {
|
|
done(102);
|
|
} else if (typeof query.port === 'undefined') {
|
|
done(103);
|
|
} else if (query.info_hash.length !== 20) {
|
|
done(150);
|
|
} else if (query.peer_id.length !== 20) {
|
|
done(151);
|
|
} else if (typeof query.compact === 'undefined' || query.compact !== '1') {
|
|
done(600);
|
|
} else {
|
|
for (i = 0; i < PARAMS_INTEGER.length; i++) {
|
|
p = PARAMS_INTEGER[i];
|
|
if (typeof query[p] !== 'undefined')
|
|
query[p] = parseInt(query[p].toString(), 10);
|
|
}
|
|
|
|
for (i = 0; i < PARAMS_STRING.length; i++) {
|
|
p = PARAMS_STRING[i];
|
|
if (typeof query[p] !== 'undefined')
|
|
query[p] = query[p].toString();
|
|
}
|
|
|
|
query.info_hash = common.binaryToHex(query.info_hash);
|
|
req.seeder = (query.left === 0) ? true : false;
|
|
|
|
done(null);
|
|
}
|
|
},
|
|
|
|
/*---------------------------------------------------------------
|
|
validatePasskeyCheck
|
|
---------------------------------------------------------------*/
|
|
function (done) {
|
|
if (typeof passkey === 'undefined') {
|
|
done(104);
|
|
} else if (passkey.length !== 32) {
|
|
done(153);
|
|
} else if (req.passkeyuser === undefined) {
|
|
done(154);
|
|
} else {
|
|
done(null);
|
|
}
|
|
},
|
|
|
|
/*---------------------------------------------------------------
|
|
validateUserCheck
|
|
check normal,banned,sealed
|
|
---------------------------------------------------------------*/
|
|
function (done) {
|
|
switch (req.passkeyuser.status) {
|
|
case 'banned':
|
|
done(170);
|
|
break;
|
|
case 'idle':
|
|
done(175);
|
|
break;
|
|
case 'inactive':
|
|
done(171);
|
|
break;
|
|
default:
|
|
done(null);
|
|
}
|
|
},
|
|
|
|
/*---------------------------------------------------------------
|
|
validateClientCheck
|
|
check client blacklist
|
|
---------------------------------------------------------------*/
|
|
function (done) {
|
|
var ua = req.get('User-Agent');
|
|
var inlist = false;
|
|
if (ua) {
|
|
config.meanTorrentConfig.clientBlackList.forEach(function (client) {
|
|
if (ua.toUpperCase().indexOf(client.name.toUpperCase()) >= 0) {
|
|
inlist = true;
|
|
}
|
|
});
|
|
}
|
|
if (inlist) {
|
|
done(172);
|
|
} else {
|
|
done(null);
|
|
}
|
|
},
|
|
|
|
/*---------------------------------------------------------------
|
|
getTorrentItemData
|
|
torrent data include peers
|
|
if peer is ghost, delete it
|
|
if not found current peer with peer_id, create it(because maybe deleted)
|
|
---------------------------------------------------------------*/
|
|
function (done) {
|
|
Torrent.findOne({
|
|
info_hash: query.info_hash
|
|
})
|
|
.populate('user')
|
|
.populate('_peers')
|
|
.exec(function (err, t) {
|
|
if (err) {
|
|
done(160);
|
|
} else if (!t) {
|
|
done(161);
|
|
} else if (t.torrent_status === 'new') {
|
|
done(173);
|
|
} else if (t.torrent_vip && !req.passkeyuser.isVip) {
|
|
done(174);
|
|
} else {
|
|
req.torrent = t;
|
|
|
|
/*
|
|
find myself peers
|
|
if the peer is ghost, deleted it
|
|
*/
|
|
if (req.torrent._peers.length > 0) {
|
|
for (var i = req.torrent._peers.length; i > 0; i--) {
|
|
var p = req.torrent._peers[i - 1];
|
|
if (p.user.equals(req.passkeyuser._id)) {
|
|
var diff = moment(Date.now()).diff(moment(p.last_announce_at || p.startedat), 'seconds');
|
|
if (diff > ANNOUNCE_INTERVAL * ANNOUNCE_GHOST) {
|
|
removePeer(p);
|
|
} else {
|
|
req.selfpeer.push(p);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
getCurrentPeer();
|
|
done(null);
|
|
}
|
|
});
|
|
},
|
|
|
|
/*---------------------------------------------------------------
|
|
find complete torrent data
|
|
if not find and torrent is h&r, then create complete record
|
|
---------------------------------------------------------------*/
|
|
function (done) {
|
|
Complete.findOne({
|
|
torrent: req.torrent._id,
|
|
user: req.passkeyuser._id
|
|
})
|
|
.populate('torrent')
|
|
.exec(function (err, t) {
|
|
if (err) {
|
|
done(185);
|
|
} else {
|
|
if (!t) {
|
|
if (req.torrent.torrent_hnr) {
|
|
var comp = new Complete();
|
|
comp.torrent = req.torrent._id;
|
|
comp.user = req.passkeyuser._id;
|
|
comp.complete = req.seeder ? true : false;
|
|
|
|
comp.save(function (err) {
|
|
if (err) {
|
|
done(186);
|
|
} else {
|
|
req.completeTorrent = comp;
|
|
done(null);
|
|
}
|
|
});
|
|
} else {
|
|
done(null);
|
|
}
|
|
} else {
|
|
req.completeTorrent = t;
|
|
done(null);
|
|
}
|
|
}
|
|
});
|
|
},
|
|
|
|
/*---------------------------------------------------------------
|
|
check N&R can download
|
|
if user has too more H&R warning numbers, can not download any torrent
|
|
but can continue download the warning status torrent
|
|
---------------------------------------------------------------*/
|
|
function (done) {
|
|
if (!req.seeder && event(query.event) !== EVENT_STOPPED) {
|
|
if (req.passkeyuser.hnr_warning >= hnrConfig.forbiddenDownloadMinWarningNumber) {
|
|
if (!req.torrent.torrent_hnr) {
|
|
done(190);
|
|
} else {
|
|
if (!req.completeTorrent) {
|
|
done(191);
|
|
} else {
|
|
if (!req.completeTorrent.hnr_warning) {
|
|
done(190);
|
|
} else {
|
|
done(null);
|
|
}
|
|
}
|
|
}
|
|
} else {
|
|
done(null);
|
|
}
|
|
} else {
|
|
done(null);
|
|
}
|
|
},
|
|
|
|
/*---------------------------------------------------------------
|
|
announce download check
|
|
ratio check, setting in announce.downloadCheck
|
|
---------------------------------------------------------------*/
|
|
function (done) {
|
|
if (!req.seeder && event(query.event) !== EVENT_STOPPED) {
|
|
if (req.passkeyuser.ratio !== -1 && req.passkeyuser.ratio < announceConfig.downloadCheck.ratio) {
|
|
var checkTimeBegin = moment(req.passkeyuser.created).add(announceConfig.downloadCheck.checkAfterSignupDays, 'd');
|
|
if (checkTimeBegin < moment(Date.now())) {
|
|
var reason = sprintf(FAILURE_REASONS[200], announceConfig.downloadCheck.ratio);
|
|
done(200, reason);
|
|
} else {
|
|
done(null);
|
|
}
|
|
} else {
|
|
done(null);
|
|
}
|
|
} else {
|
|
done(null);
|
|
}
|
|
},
|
|
|
|
/*---------------------------------------------------------------
|
|
announce check download from own seeding
|
|
own seeding check, setting in announce.downloadCheck
|
|
---------------------------------------------------------------*/
|
|
function (done) {
|
|
if (!req.seeder && event(query.event) !== EVENT_STOPPED) {
|
|
if (getSelfSeederCount() > 0 && !announceConfig.downloadCheck.canDownloadOwnSeeding) {
|
|
done(201);
|
|
} else {
|
|
done(null);
|
|
}
|
|
} else {
|
|
done(null);
|
|
}
|
|
},
|
|
|
|
/*---------------------------------------------------------------
|
|
onEventStarted
|
|
if downloading, check download peer num only 1, torrent leechers +1
|
|
if seeding, check seed peer num less 3, torrent seeds +1
|
|
if no peer founded, create new peer
|
|
---------------------------------------------------------------*/
|
|
function (done) {
|
|
if (event(query.event) === EVENT_STARTED) {
|
|
mtDebug.debugGreen('---------------EVENT_STARTED----------------');
|
|
|
|
if (getSelfLeecherCount() >= 1 && !req.seeder) {
|
|
mtDebug.debugGreen(req.selfpeer);
|
|
done(180);
|
|
} else if (getSelfSeederCount() >= 3 && req.seeder) {
|
|
mtDebug.debugGreen(req.selfpeer);
|
|
done(181);
|
|
} else {
|
|
if (req.currentPeer === undefined) {
|
|
createCurrentPeer();
|
|
}
|
|
|
|
done(null);
|
|
}
|
|
} else {
|
|
done(null);
|
|
}
|
|
},
|
|
|
|
/*---------------------------------------------------------------
|
|
onEventStopped
|
|
delete peers
|
|
if not seeder, torrent leechers -1
|
|
if seeder, torrent seeds -1
|
|
---------------------------------------------------------------*/
|
|
function (done) {
|
|
if (event(query.event) === EVENT_STOPPED) {
|
|
mtDebug.debugGreen('---------------EVENT_STOPPED----------------');
|
|
|
|
if (req.currentPeer !== undefined) {
|
|
req.selfpeer.splice(req.selfpeer.indexOf(req.currentPeer), 1);
|
|
removePeer(req.currentPeer);
|
|
}
|
|
}
|
|
done(null);
|
|
},
|
|
|
|
/*---------------------------------------------------------------
|
|
onEventCompleted
|
|
torrent leechers -1
|
|
torrent finished +1
|
|
torrent seeds +1, auto change to seeder?
|
|
---------------------------------------------------------------*/
|
|
function (done) {
|
|
if (event(query.event) === EVENT_COMPLETED) {
|
|
mtDebug.debugGreen('---------------EVENT_COMPLETED----------------');
|
|
|
|
if (req.currentPeer === undefined) {
|
|
createCurrentPeer();
|
|
} else {
|
|
req.currentPeer.update({
|
|
$set: {peer_status: PEERSTATE_SEEDER, finishedat: Date.now()}
|
|
}).exec();
|
|
|
|
req.torrent.update({
|
|
$inc: {torrent_seeds: 1, torrent_finished: 1, torrent_leechers: -1}
|
|
}).exec();
|
|
|
|
req.passkeyuser.update({
|
|
$inc: {seeded: 1, finished: 1, leeched: -1}
|
|
}).exec();
|
|
|
|
//update completeTorrent complete status
|
|
if (req.completeTorrent) {
|
|
req.completeTorrent.update({
|
|
$set: {complete: true}
|
|
}).exec();
|
|
}
|
|
}
|
|
}
|
|
done(null);
|
|
},
|
|
|
|
/*---------------------------------------------------------------
|
|
writeUpDownData
|
|
uploaded,downloaded
|
|
update complete data if completeTorrent is exist
|
|
if has upload and download data size, write data size,
|
|
write time of seed(complete) whether or not u/d is 0
|
|
---------------------------------------------------------------*/
|
|
function (done) {
|
|
mtDebug.debugGreen('---------------WRITE_UP_DOWN_DATA----------------');
|
|
//refresh user`s vip status and ratio
|
|
req.passkeyuser.globalUpdateMethod();
|
|
//active torrent update method to update torrent isSaling status
|
|
if (req.torrent.isSaling) {
|
|
req.torrent.globalUpdateMethod();
|
|
}
|
|
|
|
var udr = getUDRatio();
|
|
|
|
if (req.currentPeer !== undefined) {
|
|
var curru = query.uploaded - req.currentPeer.peer_uploaded;
|
|
var currd = query.downloaded - req.currentPeer.peer_downloaded;
|
|
|
|
var u = Math.round(curru * udr.ur);
|
|
var d = Math.round(currd * udr.dr);
|
|
|
|
if (curru > 0 || currd > 0) {
|
|
if (req.passkeyuser.isVip) {
|
|
u = u * config.meanTorrentConfig.torrentSalesValue.vip.Ur;
|
|
d = d * config.meanTorrentConfig.torrentSalesValue.vip.Dr;
|
|
}
|
|
req.passkeyuser.update({
|
|
$inc: {uploaded: u, downloaded: d}
|
|
}).exec();
|
|
|
|
//write complete data to completeTorrent and refresh completed ratio
|
|
if (req.completeTorrent) {
|
|
req.completeTorrent.update({
|
|
$inc: {
|
|
total_uploaded: curru,
|
|
total_downloaded: currd
|
|
}
|
|
}).exec();
|
|
req.completeTorrent.globalUpdateMethod();
|
|
}
|
|
|
|
//write peer speed
|
|
req.currentPeer.update({
|
|
$set: {
|
|
peer_uspeed: curru / config.meanTorrentConfig.announce.announceInterval * 1000,
|
|
peer_dspeed: currd / config.meanTorrentConfig.announce.announceInterval * 1000
|
|
}
|
|
}).exec();
|
|
}
|
|
|
|
//only add seed time for completed torrent
|
|
if (req.completeTorrent && req.completeTorrent.complete) {
|
|
req.completeTorrent.update({
|
|
$inc: {
|
|
total_seed_time: config.meanTorrentConfig.announce.announceInterval
|
|
}
|
|
}).exec();
|
|
}
|
|
//update warning status
|
|
if (req.completeTorrent) {
|
|
req.completeTorrent.countHnRWarning(req.passkeyuser);
|
|
}
|
|
|
|
//create trace log
|
|
if (curru > 0 || currd > 0) {
|
|
traceLogCreate(req, traceConfig.action.userAnnounceData, {
|
|
user: req.passkeyuser._id,
|
|
torrent: req.torrent._id,
|
|
curr_uploaded: curru,
|
|
curr_downloaded: currd,
|
|
write_uploaded: u,
|
|
write_downloaded: d,
|
|
isVip: req.passkeyuser.isVip,
|
|
agent: req.get('User-Agent'),
|
|
ip: req.headers['x-forwarded-for'] || req.connection.remoteAddress,
|
|
port: query.port
|
|
});
|
|
//update score
|
|
var action = scoreConfig.action.seedAnnounce;
|
|
if (curru > 0 && action.enable) {
|
|
var unitScore = Math.round(Math.sqrt(req.torrent.torrent_size / action.additionSize) * 100) / 100;
|
|
var upScore = Math.round((curru / action.perlSize) * 100) / 100;
|
|
var score = unitScore * action.value * upScore;
|
|
scoreUpdate(req, req.passkeyuser, action, score);
|
|
}
|
|
}
|
|
}
|
|
|
|
if (event(query.event) !== EVENT_STOPPED) {
|
|
if (req.currentPeer === undefined) {
|
|
createCurrentPeer();
|
|
}
|
|
req.currentPeer.update({
|
|
$set: {peer_uploaded: query.uploaded, peer_downloaded: query.downloaded, peer_left: query.left, last_announce_at: Date.now()}
|
|
}).exec();
|
|
}
|
|
|
|
done(null);
|
|
},
|
|
|
|
/*---------------------------------------------------------------
|
|
sendPeers
|
|
compact mode
|
|
---------------------------------------------------------------*/
|
|
function (done) {
|
|
var want = WANT_DEFAULT;
|
|
if (typeof query.numwant !== 'undefined' && query.numwant > 0)
|
|
want = query.numwant;
|
|
|
|
var peerBuffer = new Buffer(want * PEER_COMPACT_SIZE);
|
|
var len = writePeers(peerBuffer, want, req.torrent._peers);
|
|
peerBuffer = peerBuffer.slice(0, len);
|
|
|
|
var resp = 'd8:intervali' + ANNOUNCE_INTERVAL + 'e8:completei' + req.torrent.torrent_seeds + 'e10:incompletei' + req.torrent.torrent_leechers + 'e10:downloadedi' + req.torrent.torrent_finished + 'e5:peers' + len + ':';
|
|
mtDebug.debugGreen(resp);
|
|
|
|
res.writeHead(200, {
|
|
'Content-Length': resp.length + peerBuffer.length + 1,
|
|
'Content-Type': 'text/plain'
|
|
});
|
|
|
|
res.write(resp);
|
|
res.write(peerBuffer);
|
|
res.end('e');
|
|
|
|
done(null, 'done');
|
|
}
|
|
], function (err, reason) {
|
|
if (err) {
|
|
sendError(new Failure(err, reason));
|
|
}
|
|
});
|
|
|
|
/**
|
|
* getSelfCurrentPeer
|
|
* @returns {boolean}
|
|
*/
|
|
function getCurrentPeer() {
|
|
req.selfpeer.forEach(function (p) {
|
|
if (p.peer_id === query.peer_id) {
|
|
req.currentPeer = p;
|
|
}
|
|
});
|
|
}
|
|
|
|
/**
|
|
* createCurrentPeer
|
|
*/
|
|
function createCurrentPeer() {
|
|
var peer = new Peer();
|
|
peer.user = req.passkeyuser;
|
|
peer.torrent = req.torrent;
|
|
peer.peer_id = query.peer_id;
|
|
//peer.peer_ip = req.connection.remoteAddress;
|
|
peer.peer_ip = req.headers['x-forwarded-for'] || req.connection.remoteAddress;
|
|
peer.peer_port = query.port;
|
|
peer.peer_status = req.seeder ? PEERSTATE_SEEDER : PEERSTATE_LEECHER;
|
|
peer.user_agent = req.get('User-Agent');
|
|
peer.save();
|
|
|
|
req.selfpeer.push(peer);
|
|
|
|
if (req.seeder) {
|
|
req.torrent.update({
|
|
$inc: {torrent_seeds: 1}
|
|
}).exec();
|
|
req.passkeyuser.update({
|
|
$inc: {seeded: 1}
|
|
}).exec();
|
|
} else {
|
|
req.torrent.update({
|
|
$inc: {torrent_leechers: 1}
|
|
}).exec();
|
|
req.passkeyuser.update({
|
|
$inc: {leeched: 1}
|
|
}).exec();
|
|
}
|
|
|
|
req.torrent.update({
|
|
$addToSet: {_peers: peer}
|
|
}).exec();
|
|
|
|
//save ip to user
|
|
req.passkeyuser.addLeechedIp(peer.peer_ip);
|
|
req.passkeyuser.addClientAgent(peer.user_agent);
|
|
|
|
req.currentPeer = peer;
|
|
mtDebug.debugGreen('---------------createCurrentPeer()----------------');
|
|
}
|
|
|
|
/**
|
|
* removePeer
|
|
* @param p
|
|
*/
|
|
function removePeer(p) {
|
|
if (p.peer_status === PEERSTATE_LEECHER) {
|
|
req.torrent.update({
|
|
$inc: {torrent_leechers: -1}
|
|
}).exec();
|
|
req.passkeyuser.update({
|
|
$inc: {leeched: -1}
|
|
}).exec();
|
|
} else if (p.peer_status === PEERSTATE_SEEDER) {
|
|
req.torrent.update({
|
|
$inc: {torrent_seeds: -1}
|
|
}).exec();
|
|
req.passkeyuser.update({
|
|
$inc: {seeded: -1}
|
|
}).exec();
|
|
|
|
}
|
|
|
|
req.torrent.update({
|
|
$pull: {_peers: p._id}
|
|
}).exec();
|
|
|
|
p.remove();
|
|
mtDebug.debugGreen('---------------removePeer()----------------');
|
|
}
|
|
|
|
/**
|
|
* getSelfLeecherCount
|
|
* @returns {number}
|
|
*/
|
|
function getSelfLeecherCount() {
|
|
if (req.selfpeer.length === 0) {
|
|
return 0;
|
|
} else {
|
|
var i = 0;
|
|
|
|
req.selfpeer.forEach(function (p) {
|
|
if (p.peer_status === PEERSTATE_LEECHER) {
|
|
i++;
|
|
}
|
|
});
|
|
|
|
return i;
|
|
}
|
|
}
|
|
|
|
/**
|
|
* getSelfSeederCount
|
|
* @returns {number}
|
|
*/
|
|
function getSelfSeederCount() {
|
|
if (req.selfpeer.length === 0) {
|
|
return 0;
|
|
} else {
|
|
var i = 0;
|
|
|
|
req.selfpeer.forEach(function (p) {
|
|
if (p.peer_status === PEERSTATE_SEEDER) {
|
|
i++;
|
|
}
|
|
});
|
|
|
|
return i;
|
|
}
|
|
}
|
|
|
|
/**
|
|
* getUDRatio
|
|
* @returns {{}}
|
|
*/
|
|
function getUDRatio() {
|
|
var udr = {};
|
|
var sale = req.torrent.torrent_sale_status;
|
|
|
|
if (config.meanTorrentConfig.torrentSalesValue.global !== undefined) {
|
|
sale = config.meanTorrentConfig.torrentSalesValue.global;
|
|
}
|
|
|
|
switch (sale) {
|
|
case 'U1/FREE':
|
|
udr.ur = 1;
|
|
udr.dr = 0;
|
|
break;
|
|
case 'U1/D.3':
|
|
udr.ur = 1;
|
|
udr.dr = 0.3;
|
|
break;
|
|
case 'U1/D.5':
|
|
udr.ur = 1;
|
|
udr.dr = 0.5;
|
|
break;
|
|
case 'U1/D.8':
|
|
udr.ur = 1;
|
|
udr.dr = 0.8;
|
|
break;
|
|
case 'U2/FREE':
|
|
udr.ur = 2;
|
|
udr.dr = 0;
|
|
break;
|
|
case 'U2/D.3':
|
|
udr.ur = 2;
|
|
udr.dr = 0.3;
|
|
break;
|
|
case 'U2/D.5':
|
|
udr.ur = 2;
|
|
udr.dr = 0.5;
|
|
break;
|
|
case 'U2/D.8':
|
|
udr.ur = 2;
|
|
udr.dr = 0.8;
|
|
break;
|
|
case 'U2/D1':
|
|
udr.ur = 2;
|
|
udr.dr = 1;
|
|
break;
|
|
case 'U3/FREE':
|
|
udr.ur = 3;
|
|
udr.dr = 0;
|
|
break;
|
|
case 'U3/D.5':
|
|
udr.ur = 3;
|
|
udr.dr = 0.5;
|
|
break;
|
|
case 'U3/D.8':
|
|
udr.ur = 3;
|
|
udr.dr = 0.8;
|
|
break;
|
|
case 'U3/D1':
|
|
udr.ur = 3;
|
|
udr.dr = 1;
|
|
break;
|
|
default: /* U1D1 */
|
|
udr.ur = 1;
|
|
udr.dr = 1;
|
|
}
|
|
return udr;
|
|
}
|
|
|
|
/**
|
|
* sendError
|
|
* @param failure
|
|
*/
|
|
function sendError(failure) {
|
|
var respc = failure.bencode();
|
|
mtDebug.debugGreen(respc);
|
|
res.writeHead(500, {
|
|
'Content-Length': respc.length,
|
|
'Content-Type': 'text/plain'
|
|
});
|
|
|
|
res.end(respc);
|
|
}
|
|
|
|
/**
|
|
* writePeers
|
|
* @param buf
|
|
* @param count
|
|
* @param peers
|
|
* @returns {number}
|
|
*/
|
|
function writePeers(buf, count, peers) {
|
|
var c = 0;
|
|
var p;
|
|
|
|
var m = Math.min(peers.length, count);
|
|
for (var i = 0; i < m; i++) {
|
|
var index = Math.floor(Math.random() * peers.length);
|
|
p = peers[index];
|
|
//if (p !== undefined && !p.user.equals(req.passkeyuser._id)) {
|
|
if (p !== undefined) {
|
|
var b = compact(p);
|
|
if (b) {
|
|
b.copy(buf, c++ * PEER_COMPACT_SIZE);
|
|
}
|
|
}
|
|
}
|
|
|
|
return c * PEER_COMPACT_SIZE;
|
|
}
|
|
|
|
/**
|
|
* compact
|
|
* @param p
|
|
* @returns {*}
|
|
*/
|
|
function compact(p) {
|
|
var b = new Buffer(PEER_COMPACT_SIZE);
|
|
|
|
var parts = p.peer_ip.split('.');
|
|
if (parts.length !== 4) {
|
|
return null;
|
|
} else {
|
|
for (var i = 0; i < 4; i++)
|
|
b[i] = parseInt(parts[i], 10);
|
|
|
|
b[4] = (p.peer_port >> 8) & 0xff;
|
|
b[5] = p.peer_port & 0xff;
|
|
|
|
return b;
|
|
}
|
|
}
|
|
};
|
|
|
|
/**
|
|
* userByPasskey
|
|
* @param req
|
|
* @param res
|
|
* @param next
|
|
* @param pk
|
|
* @returns {*}
|
|
*/
|
|
exports.userByPasskey = function (req, res, next, pk) {
|
|
User.findOne({passkey: pk})
|
|
.exec(function (err, u) {
|
|
if (u) {
|
|
req.passkeyuser = u;
|
|
|
|
if ((moment(Date.now()) - moment(req.passkeyuser.last_signed)) > signConfig.accountIdleForTime) {
|
|
req.passkeyuser.update({
|
|
$set: {status: 'idle'}
|
|
}).exec();
|
|
req.passkeyuser.status = 'idle';
|
|
}
|
|
|
|
} else {
|
|
req.passkeyuser = undefined;
|
|
}
|
|
next();
|
|
});
|
|
};
|