| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935 |
- # The contents of this file are subject to the BitTorrent Open Source License
- # Version 1.1 (the License). You may not copy or use this file, in either
- # source code or executable form, except in compliance with the License. You
- # may obtain a copy of the License at http://www.bittorrent.com/license/.
- #
- # Software distributed under the License is distributed on an AS IS basis,
- # WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License
- # for the specific language governing rights and limitations under the
- # License.
- # Written by Bram Cohen and John Hoffman
- import sys
- import os
- import signal
- import re
- import cPickle
- import logging
- import datetime
- from urlparse import urlparse
- from traceback import print_exc
- from time import time, gmtime, strftime, localtime
- from random import shuffle
- from types import StringType, IntType, LongType, ListType, DictType
- from binascii import b2a_hex
- from cStringIO import StringIO
- from BTL.translation import _
- from BTL.obsoletepythonsupport import *
- from BitTorrent import platform
- from BTL import BTFailure
- from BTL.platform import decode_from_filesystem, efs2
- from BTL.defer import DeferredEvent, ThreadedDeferred
- from BTL.yielddefer import wrap_task
- from BitTorrent.configfile import parse_configuration_and_args
- #from BitTorrent.parseargs import parseargs, printHelp
- from BitTorrent.RawServer_twisted import RawServer
- from BitTorrent.HTTPHandler import HTTPHandler
- from BTL.parsedir import parsedir
- from BitTorrent.NatCheck import NatCheck
- from BTL.bencode import bencode, bdecode, Bencached
- from urllib import quote, unquote
- from BTL.exceptions import str_exc
- from BitTorrent import version
- from BitTorrent.prefs import Preferences
- from BitTorrent.defaultargs import get_defaults
- from BitTorrent.UI import Size
- import socket
- import threading
- import traceback
- NOISY = False
- # code duplication because ow.
- MAX_INCOMPLETE = 100
- if os.name == 'nt':
- from BitTorrent.platform import win_version_num
- # starting in XP SP2 the incomplete outgoing connection limit was set to 10
- if win_version_num >= (2, 5, 1, 2, 0):
- MAX_INCOMPLETE = 10
- def statefiletemplate(x):
- if type(x) != DictType:
- raise ValueError
- for cname, cinfo in x.iteritems():
- if cname == 'peers':
- for y in cinfo.itervalues(): # The 'peers' key is a dictionary of SHA hashes (torrent ids)
- if type(y) != DictType: # ... for the active torrents, and each is a dictionary
- raise ValueError
- for peerid, info in y.iteritems(): # ... of client ids interested in that torrent
- if (len(peerid) != 20):
- raise ValueError
- if type(info) != DictType: # ... each of which is also a dictionary
- raise ValueError # ... which has an IP, a Port, and a Bytes Left count for that client for that torrent
- if type(info.get('ip', '')) != StringType:
- raise ValueError
- port = info.get('port')
- if type(port) not in (IntType, LongType) or port < 0:
- raise ValueError
- left = info.get('left')
- if type(left) not in (IntType, LongType) or left < 0:
- raise ValueError
- elif cname == 'completed':
- if (type(cinfo) != DictType): # The 'completed' key is a dictionary of SHA hashes (torrent ids)
- raise ValueError # ... for keeping track of the total completions per torrent
- for y in cinfo.itervalues(): # ... each torrent has an integer value
- if type(y) not in (IntType,LongType):
- raise ValueError # ... for the number of reported completions for that torrent
- elif cname == 'allowed':
- if (type(cinfo) != DictType): # a list of info_hashes and included data
- raise ValueError
- if x.has_key('allowed_dir_files'):
- adlist = [z[1] for z in x['allowed_dir_files'].itervalues()]
- for y in cinfo.iterkeys(): # and each should have a corresponding key here
- if not y in adlist:
- raise ValueError
- elif cname == 'allowed_dir_files':
- if (type(cinfo) != DictType): # a list of files, their attributes and info hashes
- raise ValueError
- dirkeys = {}
- for y in cinfo.itervalues(): # each entry should have a corresponding info_hash
- if not y[1]:
- continue
- if not x['allowed'].has_key(y[1]):
- raise ValueError
- if dirkeys.has_key(y[1]): # and each should have a unique info_hash
- raise ValueError
- dirkeys[y[1]] = 1
- alas = _("your file may exist elsewhere in the universe\nbut alas, not here\n")
- def isotime():
- #return strftime('%Y-%m-%d %H:%M UTC', gmtime(secs))
- return datetime.datetime.utcnow().isoformat()
- http_via_filter = re.compile(' for ([0-9.]+)\Z')
- def _get_forwarded_ip(headers):
- if headers.has_key('x_forwarded_for'):
- header = headers['x_forwarded_for']
- try:
- x,y = header.split(',')
- except:
- return header
- if not is_local_ip(x):
- return x
- return y
- if headers.has_key('client_ip'):
- return headers['client_ip']
- if headers.has_key('via'):
- x = http_via_filter.search(headers['via'])
- try:
- return x.group(1)
- except:
- pass
- if headers.has_key('from'):
- return headers['from']
- return None
- def get_forwarded_ip(headers):
- x = _get_forwarded_ip(headers)
- if x is None or not is_valid_ipv4(x) or is_local_ip(x):
- return None
- return x
- def compact_peer_info(ip, port):
- try:
- s = ( ''.join([chr(int(i)) for i in ip.split('.')])
- + chr((port & 0xFF00) >> 8) + chr(port & 0xFF) )
- if len(s) != 6:
- s = ''
- except:
- s = '' # not a valid IP, must be a domain name
- return s
- def is_valid_ipv4(ip):
- a = ip.split('.')
- if len(a) != 4:
- return False
- try:
- for x in a:
- chr(int(x))
- return True
- except:
- return False
- def is_local_ip(ip):
- try:
- v = [int(x) for x in ip.split('.')]
- if v[0] == 10 or v[0] == 127 or v[:2] in ([192, 168], [169, 254]):
- return 1
- if v[0] == 172 and v[1] >= 16 and v[1] <= 31:
- return 1
- except ValueError:
- return 0
- default_headers = {'Content-Type': 'text/plain', 'Pragma': 'no-cache'}
- class Tracker(object):
- def __init__(self, config, rawserver):
- self.config = config
- self.response_size = config['response_size']
- self.max_give = config['max_give']
- self.dfile = efs2(config['dfile'])
- self.natcheck = config['nat_check']
- favicon = config['favicon']
- self.favicon = None
- if favicon:
- try:
- h = open(favicon,'r')
- self.favicon = h.read()
- h.close()
- except:
- errorfunc(logging.WARNING,
- _("specified favicon file -- %s -- does not exist.") %
- favicon)
- self.rawserver = rawserver
- self.cached = {} # format: infohash: [[time1, l1, s1], [time2, l2, s2], [time3, l3, s3]]
- self.cached_t = {} # format: infohash: [time, cache]
- self.times = {}
- self.state = {}
- self.seedcount = {}
- self.save_pending = False
- self.parse_pending = False
- self.only_local_override_ip = config['only_local_override_ip']
- if self.only_local_override_ip == 2:
- self.only_local_override_ip = not config['nat_check']
- if os.path.exists(self.dfile):
- try:
- h = open(self.dfile, 'rb')
- ds = h.read()
- h.close()
- try:
- tempstate = cPickle.loads(ds)
- except:
- tempstate = bdecode(ds) # backwards-compatibility.
- if not tempstate.has_key('peers'):
- tempstate = {'peers': tempstate}
- statefiletemplate(tempstate)
- self.state = tempstate
- except:
- errorfunc(logging.WARNING,
- _("statefile %s corrupt; resetting") % self.dfile)
- self.downloads = self.state.setdefault('peers', {})
- self.completed = self.state.setdefault('completed', {})
- self.becache = {} # format: infohash: [[l1, s1], [l2, s2], [l3, s3]]
- for infohash, ds in self.downloads.iteritems():
- self.seedcount[infohash] = 0
- for x, y in ds.iteritems():
- if not y.get('nat', -1):
- ip = y.get('given_ip')
- if not (ip and self.allow_local_override(y['ip'], ip)):
- ip = y['ip']
- self.natcheckOK(infohash, x, ip, y['port'], y['left'])
- if not y['left']:
- self.seedcount[infohash] += 1
- for infohash in self.downloads:
- self.times[infohash] = {}
- for peerid in self.downloads[infohash]:
- self.times[infohash][peerid] = 0
- self.reannounce_interval = config['reannounce_interval']
- self.save_dfile_interval = config['save_dfile_interval']
- self.show_names = config['show_names']
- rawserver.add_task(self.save_dfile_interval, self.save_dfile)
- self.prevtime = time()
- self.timeout_downloaders_interval = config['timeout_downloaders_interval']
- rawserver.add_task(self.timeout_downloaders_interval, self.expire_downloaders)
- self.logfile = None
- self.log = None
- if (config['logfile'] != '') and (config['logfile'] != '-'):
- try:
- self.logfile = config['logfile']
- self.log = open(self.logfile, 'a')
- sys.stdout = self.log
- print _("# Log Started: "), isotime()
- except:
- print _("**warning** could not redirect stdout to log file: "), sys.exc_info()[0]
- if config['hupmonitor']:
- def huphandler(signum, frame, self = self):
- try:
- self.log.close ()
- self.log = open(self.logfile, 'a')
- sys.stdout = self.log
- print _("# Log reopened: "), isotime()
- except:
- print _("***warning*** could not reopen logfile")
- signal.signal(signal.SIGHUP, huphandler)
- self.allow_get = config['allow_get']
- if config['allowed_dir'] != '':
- self.allowed_dir = config['allowed_dir']
- self.parse_dir_interval = config['parse_dir_interval']
- self.allowed = self.state.setdefault('allowed', {})
- self.allowed_dir_files = self.state.setdefault('allowed_dir_files', {})
- self.allowed_dir_blocked = {}
- self.parse_allowed()
- else:
- try:
- del self.state['allowed']
- except:
- pass
- try:
- del self.state['allowed_dir_files']
- except:
- pass
- self.allowed = None
- self.uq_broken = unquote('+') != ' '
- self.keep_dead = config['keep_dead']
- def allow_local_override(self, ip, given_ip):
- return is_valid_ipv4(given_ip) and (
- not self.only_local_override_ip or is_local_ip(ip) )
- def get_infopage(self):
- try:
- if not self.config['show_infopage']:
- return (404, 'Not Found', default_headers, alas)
- red = self.config['infopage_redirect']
- if red != '':
- return (302, 'Found', {'Content-Type': 'text/html', 'Location': red},
- '<A HREF="'+red+'">Click Here</A>')
- s = StringIO()
- s.write('<!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.1//EN" "http://www.w3.org/TR/xhtml11/DTD/xhtml11.dtd">\n' \
- '<html><head><title>BitTorrent download info</title>\n')
- if self.favicon is not None:
- s.write('<link rel="shortcut icon" href="/favicon.ico">\n')
- s.write('</head>\n<body>\n' \
- '<h3>BitTorrent download info</h3>\n'\
- '<ul>\n'
- '<li><strong>tracker version:</strong> %s</li>\n' \
- '<li><strong>server time:</strong> %s</li>\n' \
- '</ul>\n' % (version, isotime()))
- if self.allowed is not None:
- if self.show_names:
- names = [ (value[1].name, infohash)
- for infohash, value in self.allowed.iteritems()]
- else:
- names = [(None, infohash) for infohash in self.allowed]
- else:
- names = [ (None, infohash) for infohash in self.downloads]
- if not names:
- s.write('<p>not tracking any files yet...</p>\n')
- else:
- names.sort()
- tn = 0
- tc = 0
- td = 0
- tt = 0 # Total transferred
- ts = 0 # Total size
- nf = 0 # Number of files displayed
- if self.allowed is not None and self.show_names:
- s.write('<table summary="files" border="1">\n' \
- '<tr><th>info hash</th><th>torrent name</th><th align="right">size</th><th align="right">complete</th><th align="right">downloading</th><th align="right">downloaded</th><th align="right">transferred</th></tr>\n')
- else:
- s.write('<table summary="files">\n' \
- '<tr><th>info hash</th><th align="right">complete</th><th align="right">downloading</th><th align="right">downloaded</th></tr>\n')
- for name, infohash in names:
- l = self.downloads[infohash]
- n = self.completed.get(infohash, 0)
- tn = tn + n
- c = self.seedcount[infohash]
- tc = tc + c
- d = len(l) - c
- td = td + d
- nf = nf + 1
- if self.allowed is not None and self.show_names:
- if self.allowed.has_key(infohash):
- sz = self.allowed[infohash][1].total_bytes # size
- ts = ts + sz
- szt = sz * n # Transferred for this torrent
- tt = tt + szt
- if self.allow_get == 1:
- linkname = '<a href="/file?info_hash=' + quote(infohash) + '">' + name + '</a>'
- else:
- linkname = name
- s.write('<tr><td><code>%s</code></td><td>%s</td><td align="right">%s</td><td align="right">%i</td><td align="right">%i</td><td align="right">%i</td><td align="right">%s</td></tr>\n' \
- % (b2a_hex(infohash), linkname, size_format(sz), c, d, n, size_format(szt)))
- else:
- s.write('<tr><td><code>%s</code></td><td align="right"><code>%i</code></td><td align="right"><code>%i</code></td><td align="right"><code>%i</code></td></tr>\n' \
- % (b2a_hex(infohash), c, d, n))
- ttn = 0
- for i in self.completed.itervalues():
- ttn = ttn + i
- if self.allowed is not None and self.show_names:
- s.write('<tr><td align="right" colspan="2">%i files</td><td align="right">%s</td><td align="right">%i</td><td align="right">%i</td><td align="right">%i/%i</td><td align="right">%s</td></tr>\n'
- % (nf, size_format(ts), tc, td, tn, ttn, size_format(tt)))
- else:
- s.write('<tr><td align="right">%i files</td><td align="right">%i</td><td align="right">%i</td><td align="right">%i/%i</td></tr>\n'
- % (nf, tc, td, tn, ttn))
- s.write('</table>\n' \
- '<ul>\n' \
- '<li><em>info hash:</em> SHA1 hash of the "info" section of the metainfo (*.torrent)</li>\n' \
- '<li><em>complete:</em> number of connected clients with the complete file</li>\n' \
- '<li><em>downloading:</em> number of connected clients still downloading</li>\n' \
- '<li><em>downloaded:</em> reported complete downloads (total: current/all)</li>\n' \
- '<li><em>transferred:</em> torrent size * total downloaded (does not include partial transfers)</li>\n' \
- '</ul>\n')
- s.write('</body>\n' \
- '</html>\n')
- return (200, 'OK',
- {'Content-Type': 'text/html; charset=iso-8859-1'},
- s.getvalue())
- except:
- print_exc()
- return (500, 'Internal Server Error',
- {'Content-Type': 'text/html; charset=iso-8859-1'},
- 'Server Error')
- def scrapedata(self, infohash, return_name = True):
- l = self.downloads[infohash]
- n = self.completed.get(infohash, 0)
- c = self.seedcount[infohash]
- d = len(l) - c
- f = {'complete': c, 'incomplete': d, 'downloaded': n}
- if return_name and self.show_names and self.allowed is not None:
- f['name'] = self.allowed[infohash]['name']
- return (f)
- def get_scrape(self, paramslist):
- fs = {}
- if paramslist.has_key('info_hash'):
- if self.config['scrape_allowed'] not in ['specific', 'full']:
- return (400, 'Not Authorized', default_headers,
- bencode({'failure_reason':
- "specific scrape function is not available with this tracker."}))
- for infohash in paramslist['info_hash']:
- if self.allowed is not None and infohash not in self.allowed:
- continue
- if infohash in self.downloads:
- fs[infohash] = self.scrapedata(infohash)
- else:
- if self.config['scrape_allowed'] != 'full':
- return (400, 'Not Authorized', default_headers,
- bencode({'failure reason':
- "full scrape function is not available with this tracker."}))
- #bencode({'failure reason':
- #_("full scrape function is not available with this tracker.")}))
- if self.allowed is not None:
- hashes = self.allowed
- else:
- hashes = self.downloads
- for infohash in hashes:
- fs[infohash] = self.scrapedata(infohash)
- return (200, 'OK', {'Content-Type': 'text/plain'}, bencode({'files': fs}))
- def get_file(self, infohash):
- if not self.allow_get:
- return (400, 'Not Authorized',
- default_headers,
- _("get function is not available with this tracker."))
- if not self.allowed.has_key(infohash):
- return (404, 'Not Found', default_headers, alas)
- fname = self.allowed[infohash]['file']
- fpath = self.allowed[infohash]['path']
- return (200, 'OK', {'Content-Type': 'application/x-bittorrent',
- 'Content-Disposition': 'attachment; filename=' + fname},
- open(fpath, 'rb').read())
- def check_allowed(self, infohash, paramslist):
- if self.allowed is not None:
- if not self.allowed.has_key(infohash):
- return (200, 'Not Authorized', default_headers,
- bencode({'failure reason':
- "Requested download is not authorized for use with this tracker."}))
- #_("Requested download is not authorized for use with this tracker.")}))
- if self.config['allowed_controls']:
- if self.allowed[infohash].has_key('failure reason'):
- return (200, 'Not Authorized', default_headers,
- bencode({'failure reason': self.allowed[infohash]['failure reason']}))
- return None
- def add_data(self, infohash, event, ip, paramslist):
- peers = self.downloads.setdefault(infohash, {})
- ts = self.times.setdefault(infohash, {})
- self.completed.setdefault(infohash, 0)
- self.seedcount.setdefault(infohash, 0)
- def params(key, default = None, l = paramslist):
- if l.has_key(key):
- return l[key][0]
- return default
- myid = params('peer_id','')
- if len(myid) != 20:
- raise ValueError, 'id not of length 20'
- if event not in ['started', 'completed', 'stopped', 'snooped', None]:
- raise ValueError, 'invalid event'
- port = int(params('port',''))
- if port < 0 or port > 65535:
- raise ValueError, 'invalid port'
- left = int(params('left',''))
- if left < 0:
- raise ValueError, 'invalid amount left'
- peer = peers.get(myid)
- mykey = params('key')
- auth = not peer or peer.get('key', -1) == mykey or peer.get('ip') == ip
- gip = params('ip')
- local_override = gip and self.allow_local_override(ip, gip)
- if local_override:
- ip1 = gip
- else:
- ip1 = ip
- if not auth and local_override and self.only_local_override_ip:
- auth = True
- if params('numwant') is not None:
- rsize = min(int(params('numwant')), self.max_give)
- else:
- rsize = self.response_size
- if event == 'stopped':
- if peer and auth:
- self.delete_peer(infohash,myid)
- elif not peer:
- ts[myid] = time()
- peer = {'ip': ip, 'port': port, 'left': left}
- if mykey:
- peer['key'] = mykey
- if gip:
- peer['given ip'] = gip
- if port:
- if not self.natcheck or (local_override and self.only_local_override_ip):
- peer['nat'] = 0
- self.natcheckOK(infohash,myid,ip1,port,left)
- else:
- NatCheck(self.connectback_result,infohash,myid,ip1,port,self.rawserver)
- else:
- peer['nat'] = 2**30
- if event == 'completed':
- self.completed[infohash] += 1
- if not left:
- self.seedcount[infohash] += 1
- peers[myid] = peer
- else:
- if not auth:
- return rsize # return w/o changing stats
- ts[myid] = time()
- if not left and peer['left']:
- self.completed[infohash] += 1
- self.seedcount[infohash] += 1
- if not peer.get('nat', -1):
- for bc in self.becache[infohash]:
- bc[1][myid] = bc[0][myid]
- del bc[0][myid]
- if peer['left']:
- peer['left'] = left
- recheck = False
- if ip != peer['ip']:
- peer['ip'] = ip
- recheck = True
- if gip != peer.get('given ip'):
- if gip:
- peer['given ip'] = gip
- elif peer.has_key('given ip'):
- del peer['given ip']
- if local_override:
- if self.only_local_override_ip:
- self.natcheckOK(infohash,myid,ip1,port,left)
- else:
- recheck = True
- if port and self.natcheck:
- if recheck:
- if peer.has_key('nat'):
- if not peer['nat']:
- l = self.becache[infohash]
- y = not peer['left']
- for x in l:
- del x[y][myid]
- del peer['nat'] # restart NAT testing
- else:
- natted = peer.get('nat', -1)
- if natted and natted < self.natcheck:
- recheck = True
- if recheck:
- NatCheck(self.connectback_result,infohash,myid,ip1,port,self.rawserver)
- return rsize
- def peerlist(self, infohash, stopped, is_seed, return_type, rsize):
- data = {} # return data
- seeds = self.seedcount[infohash]
- data['complete'] = seeds
- data['incomplete'] = len(self.downloads[infohash]) - seeds
- if ( self.allowed is not None and self.config['allowed_controls'] and
- self.allowed[infohash].has_key('warning message') ):
- data['warning message'] = self.allowed[infohash]['warning message']
- data['interval'] = self.reannounce_interval
- if stopped or not rsize: # save some bandwidth
- data['peers'] = []
- return data
- bc = self.becache.setdefault(infohash,[[{}, {}], [{}, {}], [{}, {}]])
- len_l = len(bc[0][0])
- len_s = len(bc[0][1])
- if not (len_l+len_s): # caches are empty!
- data['peers'] = []
- return data
- l_get_size = int(float(rsize)*(len_l)/(len_l+len_s))
- cache = self.cached.setdefault(infohash,[None,None,None])[return_type]
- if cache:
- if cache[0] + self.config['min_time_between_cache_refreshes'] < time():
- cache = None
- else:
- if ( (is_seed and len(cache[1]) < rsize)
- or len(cache[1]) < l_get_size or not cache[1] ):
- cache = None
- if not cache:
- vv = [[],[],[]]
- cache = [ time(),
- bc[return_type][0].values()+vv[return_type],
- bc[return_type][1].values() ]
- shuffle(cache[1])
- shuffle(cache[2])
- self.cached[infohash][return_type] = cache
- for rr in xrange(len(self.cached[infohash])):
- if rr != return_type:
- try:
- self.cached[infohash][rr][1].extend(vv[rr])
- except:
- pass
- if len(cache[1]) < l_get_size:
- peerdata = cache[1]
- if not is_seed:
- peerdata.extend(cache[2])
- cache[1] = []
- cache[2] = []
- else:
- if not is_seed:
- peerdata = cache[2][l_get_size-rsize:]
- del cache[2][l_get_size-rsize:]
- rsize -= len(peerdata)
- else:
- peerdata = []
- if rsize:
- peerdata.extend(cache[1][-rsize:])
- del cache[1][-rsize:]
- if return_type == 2:
- peerdata = ''.join(peerdata)
- data['peers'] = peerdata
- return data
- def get(self, connection, path, headers):
- ip = connection.get_ip()
- nip = get_forwarded_ip(headers)
- if nip and not self.only_local_override_ip:
- ip = nip
- paramslist = {}
- def params(key, default = None, l = paramslist):
- if l.has_key(key):
- return l[key][0]
- return default
- try:
- (scheme, netloc, path, pars, query, fragment) = urlparse(path)
- if self.uq_broken == 1:
- path = path.replace('+',' ')
- query = query.replace('+',' ')
- path = unquote(path)[1:]
- for s in query.split('&'):
- if s != '':
- i = s.index('=')
- kw = unquote(s[:i])
- paramslist.setdefault(kw, [])
- paramslist[kw] += [unquote(s[i+1:])]
- if path == '' or path == 'index.html':
- return self.get_infopage()
- if path == 'scrape':
- return self.get_scrape(paramslist)
- if (path == 'file'):
- return self.get_file(params('info_hash'))
- if path == 'favicon.ico' and self.favicon is not None:
- return (200, 'OK', {'Content-Type' : 'image/x-icon'}, self.favicon)
- if path != 'announce':
- return (404, 'Not Found', default_headers, alas)
- # main tracker function
- infohash = params('info_hash')
- if not infohash:
- raise ValueError, 'no info hash'
- notallowed = self.check_allowed(infohash, paramslist)
- if notallowed:
- if NOISY:
- self._print_event( "get: NOT ALLOWED: info_hash=%s, %s" %
- (infohash.encode('hex'). str(notallowed)) )
- return notallowed
- event = params('event')
- rsize = self.add_data(infohash, event, ip, paramslist)
- except ValueError, e:
- print e
- if NOISY:
- self._print_exc( "get: ",e )
- return (400, 'Bad Request',
- {'Content-Type': 'text/plain'},
- 'you sent me garbage - ' + str_exc(e))
- if params('compact'):
- return_type = 2
- elif params('no_peer_id'):
- return_type = 1
- else:
- return_type = 0
- data = self.peerlist(infohash, event=='stopped', not params('left'),
- return_type, rsize)
- if paramslist.has_key('scrape'):
- data['scrape'] = self.scrapedata(infohash, False)
- return (200, 'OK', default_headers, bencode(data))
- def natcheckOK(self, infohash, peerid, ip, port, not_seed):
- bc = self.becache.setdefault(infohash,[[{}, {}], [{}, {}], [{}, {}]])
- bc[0][not not_seed][peerid] = Bencached(bencode({'ip': ip, 'port': port,
- 'peer id': peerid}))
- bc[1][not not_seed][peerid] = Bencached(bencode({'ip': ip, 'port': port}))
- bc[2][not not_seed][peerid] = compact_peer_info(ip, port)
- def natchecklog(self, peerid, ip, port, result):
- print isotime(), '"!natcheck-%s:%i" %s %i 0 - -' % (
- ip, port, quote(peerid), result)
- def connectback_result(self, result, downloadid, peerid, ip, port):
- record = self.downloads.get(downloadid, {}).get(peerid)
- if ( record is None
- or (record['ip'] != ip and record.get('given ip') != ip)
- or record['port'] != port ):
- if self.config['log_nat_checks']:
- self.natchecklog(peerid, ip, port, 404)
- return
- if self.config['log_nat_checks']:
- if result:
- x = 200
- else:
- x = 503
- self.natchecklog(peerid, ip, port, x)
- if not record.has_key('nat'):
- record['nat'] = int(not result)
- if result:
- self.natcheckOK(downloadid,peerid,ip,port,record['left'])
- elif result and record['nat']:
- record['nat'] = 0
- self.natcheckOK(downloadid,peerid,ip,port,record['left'])
- elif not result:
- record['nat'] += 1
- def save_dfile(self):
- if self.save_pending:
- return
- self.save_pending = True
- # if this is taking all the time, threading it won't help anyway because
- # of the GIL
- #state = bencode(self.state)
- state = cPickle.dumps(self.state) # pickle handles Unicode.
- df = ThreadedDeferred(wrap_task(self.rawserver.external_add_task),
- self._save_dfile, state)
- def cb(r):
- self.save_pending = False
- if NOISY:
- self._print_event( "save_dfile: Completed" )
- def eb(etup):
- self.save_pending = False
- self._print_exc( "save_dfile: ", etup )
- df.addCallbacks(cb, eb)
- def _save_dfile(self, state):
- exc_info = None
- try:
- h = open(self.dfile, 'wb')
- h.write(state)
- h.close()
- except:
- exc_info = sys.exc_info()
- self.rawserver.external_add_task(self.save_dfile_interval, self.save_dfile)
- if exc_info:
- raise exc_info[0], exc_info[1], exc_info[2]
- def parse_allowed(self):
- if self.parse_pending:
- return
- self.parse_pending = True
- df = ThreadedDeferred(wrap_task(self.rawserver.external_add_task),
- self._parse_allowed, daemon=True)
- def eb(etup):
- self.parse_pending = False
- self._print_exc("parse_dir: ", etup)
- df.addCallbacks(self._parse_allowed_finished, eb)
- def _parse_allowed(self):
- def errfunc(message, exc_info=None):
- # logging broken .torrent files would be useful but could confuse
- # programs parsing log files
- m = "parse_dir: %s" % message
- if exc_info:
- self._print_exc(m, exc_info)
- else:
- self._print_event(m)
- pass
- r = parsedir(self.allowed_dir, self.allowed, self.allowed_dir_files,
- self.allowed_dir_blocked, errfunc, include_metainfo = False)
- # register the call to parse a dir.
- self.rawserver.external_add_task(self.parse_dir_interval,
- self.parse_allowed)
- return r
- def _parse_allowed_finished(self, r):
- self.parse_pending = False
- ( self.allowed, self.allowed_dir_files, self.allowed_dir_blocked,
- added, removed ) = r
- if NOISY:
- self._print_event("_parse_allowed_finished: removals: %s" %
- str(removed))
- for infohash in added:
- self.downloads.setdefault(infohash, {})
- self.completed.setdefault(infohash, 0)
- self.seedcount.setdefault(infohash, 0)
- self.state['allowed'] = self.allowed
- self.state['allowed_dir_files'] = self.allowed_dir_files
- def delete_peer(self, infohash, peerid):
- dls = self.downloads[infohash]
- peer = dls[peerid]
- if not peer['left']:
- self.seedcount[infohash] -= 1
- if not peer.get('nat', -1):
- l = self.becache[infohash]
- y = not peer['left']
- for x in l:
- del x[y][peerid]
- del self.times[infohash][peerid]
- del dls[peerid]
- def expire_downloaders(self):
- for infohash, peertimes in self.times.iteritems():
- items = peertimes.items()
- for myid, t in items:
- if t < self.prevtime:
- self.delete_peer(infohash, myid)
- self.prevtime = time()
- if self.keep_dead != 1:
- items = self.downloads.items()
- for key, peers in items:
- if len(peers) == 0 and (self.allowed is None or
- key not in self.allowed):
- del self.times[key]
- del self.downloads[key]
- del self.seedcount[key]
- self.rawserver.add_task(self.timeout_downloaders_interval,
- self.expire_downloaders)
- def _print_event(self, message):
- print datetime.datetime.utcnow().isoformat(), message
- def _print_exc(self, note, etup):
- print datetime.datetime.utcnow().isoformat(), note, ':'
- traceback.print_exception(*etup)
- def track(args):
- assert type(args) == list and \
- len([x for x in args if type(x)==str])==len(args)
- config = {}
- defaults = get_defaults('bittorrent-tracker') # hard-coded defaults.
- try:
- config, files = parse_configuration_and_args(defaults,
- 'bittorrent-tracker', args, 0, 0 )
- except ValueError, e:
- print _("error: ") + str_exc(e)
- print _("run with -? for parameter explanations")
- return
- except BTFailure, e:
- print _("error: ") + str_exc(e)
- print _("run with -? for parameter explanations")
- return
- if config['dfile']=="":
- config['dfile'] = decode_from_filesystem(
- os.path.join(platform.get_temp_dir(), efs2(u"dfile") +
- str(os.getpid())))
- config = Preferences().initWithDict(config)
- ef = lambda e: errorfunc(logging.WARNING, e)
- platform.write_pid_file(config['pid'], ef)
- t = None
- try:
- r = RawServer(config)
- t = Tracker(config, r)
- try:
- #DEBUG
- print "track: create_serversocket, port=", config['port']
- #END
- s = r.create_serversocket(config['port'], config['bind'])
- handler = HTTPHandler(t.get, config['min_time_between_log_flushes'])
- r.start_listening(s, handler)
- except socket.error, e:
- print ("Unable to open port %d. Use a different port?" %
- config['port'])
- return
- r.listen_forever()
- finally:
- if t: t.save_dfile()
- print _("# Shutting down: ") + isotime()
- def size_format(s):
- return str(Size(s))
- def errorfunc( level, text ):
- print "%s: %s" % (logging.getLevelName(level), text)
|