]>
Commit | Line | Data |
---|---|---|
9b67da3d C |
1 | import { logger } from '../helpers/logger' |
2 | import * as express from 'express' | |
3 | import * as http from 'http' | |
4 | import * as bitTorrentTracker from 'bittorrent-tracker' | |
5 | import * as proxyAddr from 'proxy-addr' | |
6 | import { Server as WebSocketServer } from 'ws' | |
6dd9de95 | 7 | import { TRACKER_RATE_LIMITS } from '../initializers/constants' |
cc43831a | 8 | import { VideoFileModel } from '../models/video/video-file' |
09209296 | 9 | import { VideoStreamingPlaylistModel } from '../models/video/video-streaming-playlist' |
6dd9de95 | 10 | import { CONFIG } from '../initializers/config' |
9b67da3d C |
11 | |
12 | const TrackerServer = bitTorrentTracker.Server | |
13 | ||
14 | const trackerRouter = express.Router() | |
15 | ||
16 | let peersIps = {} | |
17 | let peersIpInfoHash = {} | |
18 | runPeersChecker() | |
19 | ||
20 | const trackerServer = new TrackerServer({ | |
21 | http: false, | |
22 | udp: false, | |
23 | ws: false, | |
24 | dht: false, | |
09209296 | 25 | filter: async function (infoHash, params, cb) { |
31b6ddf8 C |
26 | if (CONFIG.TRACKER.ENABLED === false) { |
27 | return cb(new Error('Tracker is disabled on this instance.')) | |
28 | } | |
29 | ||
9b67da3d C |
30 | let ip: string |
31 | ||
32 | if (params.type === 'ws') { | |
33 | ip = params.socket.ip | |
34 | } else { | |
35 | ip = params.httpReq.ip | |
36 | } | |
37 | ||
38 | const key = ip + '-' + infoHash | |
39 | ||
a1587156 C |
40 | peersIps[ip] = peersIps[ip] ? peersIps[ip] + 1 : 1 |
41 | peersIpInfoHash[key] = peersIpInfoHash[key] ? peersIpInfoHash[key] + 1 : 1 | |
9b67da3d | 42 | |
a1587156 C |
43 | if (CONFIG.TRACKER.REJECT_TOO_MANY_ANNOUNCES && peersIpInfoHash[key] > TRACKER_RATE_LIMITS.ANNOUNCES_PER_IP_PER_INFOHASH) { |
44 | return cb(new Error(`Too many requests (${peersIpInfoHash[key]} of ip ${ip} for torrent ${infoHash}`)) | |
9b67da3d C |
45 | } |
46 | ||
09209296 | 47 | try { |
31b6ddf8 C |
48 | if (CONFIG.TRACKER.PRIVATE === false) return cb() |
49 | ||
35f28e94 | 50 | const videoFileExists = await VideoFileModel.doesInfohashExistCached(infoHash) |
09209296 | 51 | if (videoFileExists === true) return cb() |
cc43831a | 52 | |
09209296 C |
53 | const playlistExists = await VideoStreamingPlaylistModel.doesInfohashExist(infoHash) |
54 | if (playlistExists === true) return cb() | |
55 | ||
56 | return cb(new Error(`Unknown infoHash ${infoHash}`)) | |
57 | } catch (err) { | |
58 | logger.error('Error in tracker filter.', { err }) | |
59 | return cb(err) | |
60 | } | |
9b67da3d C |
61 | } |
62 | }) | |
63 | ||
31b6ddf8 | 64 | if (CONFIG.TRACKER.ENABLED !== false) { |
9b67da3d | 65 | |
31b6ddf8 C |
66 | trackerServer.on('error', function (err) { |
67 | logger.error('Error in tracker.', { err }) | |
68 | }) | |
69 | ||
70 | trackerServer.on('warning', function (err) { | |
71 | logger.warn('Warning in tracker.', { err }) | |
72 | }) | |
73 | } | |
9b67da3d C |
74 | |
75 | const onHttpRequest = trackerServer.onHttpRequest.bind(trackerServer) | |
76 | trackerRouter.get('/tracker/announce', (req, res) => onHttpRequest(req, res, { action: 'announce' })) | |
77 | trackerRouter.get('/tracker/scrape', (req, res) => onHttpRequest(req, res, { action: 'scrape' })) | |
78 | ||
cef534ed | 79 | function createWebsocketTrackerServer (app: express.Application) { |
9b67da3d | 80 | const server = http.createServer(app) |
89ada4e2 C |
81 | const wss = new WebSocketServer({ noServer: true }) |
82 | ||
9b67da3d | 83 | wss.on('connection', function (ws, req) { |
89ada4e2 | 84 | ws['ip'] = proxyAddr(req, CONFIG.TRUST_PROXY) |
9b67da3d C |
85 | |
86 | trackerServer.onWebSocketConnection(ws) | |
87 | }) | |
88 | ||
a1587156 | 89 | server.on('upgrade', (request: express.Request, socket, head) => { |
4832e415 | 90 | if (request.url === '/tracker/socket') { |
89ada4e2 C |
91 | wss.handleUpgrade(request, socket, head, ws => wss.emit('connection', ws, request)) |
92 | } | |
93 | ||
94 | // Don't destroy socket, we have Socket.IO too | |
95 | }) | |
96 | ||
9b67da3d C |
97 | return server |
98 | } | |
99 | ||
100 | // --------------------------------------------------------------------------- | |
101 | ||
102 | export { | |
103 | trackerRouter, | |
cef534ed | 104 | createWebsocketTrackerServer |
9b67da3d C |
105 | } |
106 | ||
107 | // --------------------------------------------------------------------------- | |
108 | ||
109 | function runPeersChecker () { | |
110 | setInterval(() => { | |
111 | logger.debug('Checking peers.') | |
112 | ||
113 | for (const ip of Object.keys(peersIpInfoHash)) { | |
114 | if (peersIps[ip] > TRACKER_RATE_LIMITS.ANNOUNCES_PER_IP) { | |
115 | logger.warn('Peer %s made abnormal requests (%d).', ip, peersIps[ip]) | |
116 | } | |
117 | } | |
118 | ||
119 | peersIpInfoHash = {} | |
120 | peersIps = {} | |
121 | }, TRACKER_RATE_LIMITS.INTERVAL) | |
122 | } |