From e2afbd6a8ac19eb9b7fa8505747513d5650b0c78 Mon Sep 17 00:00:00 2001
From: Michael Tremer <michael.tremer@ipfire.org>
Date: Fri, 2 Apr 2010 00:47:11 +0200
Subject: [PATCH] tracker: Add scrape functionality.

---
 www/webapp/__init__.py |  1 +
 www/webapp/handlers.py | 28 +++++++++++++++++++++++-----
 www/webapp/torrent.py  | 28 ++++++++++++++++++++++++----
 3 files changed, 48 insertions(+), 9 deletions(-)

diff --git a/www/webapp/__init__.py b/www/webapp/__init__.py
index b40c84fc..98e60883 100644
--- a/www/webapp/__init__.py
+++ b/www/webapp/__init__.py
@@ -102,6 +102,7 @@ class Application(tornado.web.Application):
 			(r"/[A-Za-z]{2}/?", MainHandler),
 			(r"/[A-Za-z]{2}/index", DownloadTorrentHandler),
 			(r"/a.*", TrackerAnnounceHandler),
+			(r"/scrape", TrackerScrapeHandler),
 		] + static_handlers)
 
 		# ipfire.org
diff --git a/www/webapp/handlers.py b/www/webapp/handlers.py
index c93fe9a9..53d47ede 100644
--- a/www/webapp/handlers.py
+++ b/www/webapp/handlers.py
@@ -312,13 +312,22 @@ class RSSHandler(BaseHandler):
 
 
 class TrackerBaseHandler(tornado.web.RequestHandler):
-	def get_hexencoded_argument(self, name):
+	def get_hexencoded_argument(self, name, all=False):
 		try:
-			info_hash = self.request.arguments[name][0]
+			arguments = self.request.arguments[name]
 		except KeyError:
 			return None
 
-		return decode_hex(info_hash)
+		arguments_new = []
+		for argument in arguments:
+			arguments_new.append(decode_hex(argument))
+
+		arguments = arguments_new
+
+		if all:
+			return arguments
+
+		return arguments[0]
 
 	def send_tracker_error(self, error_message):
 		self.write(bencode({"failure reason" : error_message }))
@@ -333,7 +342,6 @@ class TrackerAnnounceHandler(TrackerBaseHandler):
 			self.send_tracker_error("Your client forgot to send your torrent's info_hash.")
 			return
 
-		compact = self.get_argument("compact", "0")
 		peer = {
 			"id" : self.get_hexencoded_argument("peer_id"),
 			"ip" : self.get_argument("ip", None),
@@ -372,14 +380,24 @@ class TrackerAnnounceHandler(TrackerBaseHandler):
 
 		tracker.update(hash=info_hash, **peer)
 
+		no_peer_id = self.get_argument("no_peer_id", False)
 		numwant = self.get_argument("numwant", tracker.numwant)
 
 		self.write(bencode({
 			"tracker id" : tracker.id,
 			"interval" : tracker.interval,
 			"min interval" : tracker.min_interval,
-			"peers" : tracker.get_peers(info_hash, limit=numwant, random=True),
+			"peers" : tracker.get_peers(info_hash, limit=numwant,
+				random=True, no_peer_id=no_peer_id),
 			"complete" : tracker.complete(info_hash),
 			"incomplete" : tracker.incomplete(info_hash),
 		}))
 		self.finish()
+
+
+class TrackerScrapeHandler(TrackerBaseHandler):
+	def get(self):
+		info_hashes = self.get_hexencoded_argument("info_hash", all=True)
+
+		self.write(bencode(tracker.scrape(hashes=info_hashes)))
+		self.finish()
diff --git a/www/webapp/torrent.py b/www/webapp/torrent.py
index 5beeb67b..b005dc74 100644
--- a/www/webapp/torrent.py
+++ b/www/webapp/torrent.py
@@ -32,7 +32,7 @@ class Tracker(object):
 			user="tracker",
 		)
 
-	def _fetch(self, hash, limit=None, random=False, completed=False):
+	def _fetch(self, hash, limit=None, random=False, completed=False, no_peer_id=False):
 		query = "SELECT * FROM peers WHERE last_update >= %d" % self.since
 
 		if hash:
@@ -52,11 +52,15 @@ class Tracker(object):
 			if not peer.ip or not peer.port:
 				continue
 
-			peers.append({
-				"peer id" : str(peer.id),
+			peer_dict = {
 				"ip" : str(peer.ip),
 				"port" : int(peer.port),
-			})
+			}
+
+			if not no_peer_id:
+				peer_dict["peer id"] = str(peer.id),
+
+			peers.append(peer_dict)
 
 		return peers
 
@@ -87,6 +91,22 @@ class Tracker(object):
 	def event_completed(self, hash, peer_id):
 		self.db.execute("UPDATE hashes SET completed=completed+1 WHERE hash = '%s'" % hash)
 
+	def scrape(self, hashes=[]):
+		ret = {}
+		for hash in self.db.query("SELECT hash, completed FROM hashes"):
+			hash, completed = hash.hash, hash.completed
+
+			if hashes and hash not in hashes:
+				continue
+
+			ret[hash] = {
+				"complete" : self.complete(hash),
+				"downloaded" : completed or 0,
+				"incomplete" : self.incomplete(hash),
+			}
+
+		return ret
+
 	def update(self, hash, id, ip=None, port=None, downloaded=None, uploaded=None, left=None):
 		args = [ "last_update = '%s'" % self.now ]
 
-- 
2.47.3