sqozz/TorrentIndexer
1
0
Fork 0
Code Issues Pull requests Releases Wiki Activity

reject torrents that are not announced by our tracker

Browse source
This commit is contained in:
Stefan Rupp 2017-12-29 06:30:58 +01:00
parent 2c220afced
commit ef3224ecf3
1 changed files with 30 additions and 11 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

37
indexer.py
View file

@ -131,10 +131,33 @@ def createNewTorrent(reuqest):
bcoded = bencoder.decode(content) bcoded = bencoder.decode(content)
info_hash = sha1(bencoder.encode(bcoded[b'info'])).hexdigest() info_hash = sha1(bencoder.encode(bcoded[b'info'])).hexdigest()
#TODO: Validate the input serverside before writing it to the database
name = request.form["name"]
category = request.form["category"]
subcategory = request.form["subcategory"]
description = request.form["description"]
audioquality_description = request.form["audioquality_description"]
videoquality_description = request.form["videoquality_description"]
newTFile = TorrentFile(info_hash, name, category, subcategory, description, audioquality_description, videoquality_description)
try:
announce = bcoded[b'announce-list']
except KeyError:
announce = (bcoded[b'announce'], )
is_ours = False
for a in announce:
if a in [b'udp://tracker.lootbox.cf:6969/announce', b'udp://tracker.lootbox.cf:6969/announce/',b'udp://tracker.lootbox.cf/announce',b'udp://tracker.lootbox.cf/announce/']:
is_ours = True
break
if not is_ours:
newTFile.errors = ["Rejecting torrent <a href=\"/search?h={}\">{}</a>, as it does not use our tracker".format(info_hash, info_hash[:-20])]
return newTFile
with open("torrentFiles/" + info_hash, "wb") as torrent_file: with open("torrentFiles/" + info_hash, "wb") as torrent_file:
torrent_file.write(content) torrent_file.write(content)
bcoded = bencoder.decode(content)
size = ((len(bcoded[b'info'][b'pieces']) / 20) * bcoded[b'info'][b'piece length']) / 1024 / 1024 size = ((len(bcoded[b'info'][b'pieces']) / 20) * bcoded[b'info'][b'piece length']) / 1024 / 1024
print("=== CREATE NEW TORRENT FILE ===") print("=== CREATE NEW TORRENT FILE ===")
@ -144,14 +167,6 @@ def createNewTorrent(reuqest):
print( "Subcategory: " + request.form["subcategory"] ) print( "Subcategory: " + request.form["subcategory"] )
print( "Description: " + request.form["description"] ) print( "Description: " + request.form["description"] )
#TODO: Validate the input serverside before writing it to the database
name = request.form["name"]
category = request.form["category"]
subcategory = request.form["subcategory"]
description = request.form["description"]
audioquality_description = request.form["audioquality_description"]
videoquality_description = request.form["videoquality_description"]
newTFile = TorrentFile(info_hash, name, category, subcategory, description, audioquality_description, videoquality_description)
try: try:
connection = sqlite3.connect("torrentdb.sqlite") connection = sqlite3.connect("torrentdb.sqlite")
newTFile.writeToDb(connection.cursor()) newTFile.writeToDb(connection.cursor())
@ -162,14 +177,18 @@ def createNewTorrent(reuqest):
print(e) print(e)
newTFile.errors = ["Torrent <a href=\"/search?h={}\">{}</a> already exists".format(info_hash, info_hash[:-20])] newTFile.errors = ["Torrent <a href=\"/search?h={}\">{}</a> already exists".format(info_hash, info_hash[:-20])]
except Exception as e: except Exception as e:
print(e)
newTFile.errors = ["Unknown error in creation"] newTFile.errors = ["Unknown error in creation"]
return newTFile return newTFile
class Metadata(): class Metadata():
def __init__(self, fileid): def __init__(self, fileid):
try:
with open("torrentFiles/" + fileid, "rb") as f: with open("torrentFiles/" + fileid, "rb") as f:
torrent = f.read() torrent = f.read()
except FileNotFoundError:
return
self.fileid = fileid self.fileid = fileid
self.bcoded = bencoder.decode(torrent) self.bcoded = bencoder.decode(torrent)
self.created_by = self.bcoded.get(b'created by', b"") self.created_by = self.bcoded.get(b'created by', b"")