Add a check to ensure the infohash is present in the scrape reply.

rakshasa · Nov 7, 2011 · ba24b8e · ba24b8e
1 parent 22b83bc
commit ba24b8e
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/tracker/tracker_http.cc b/src/tracker/tracker_http.cc
@@ -249,6 +249,10 @@ TrackerHttp::receive_done() {
   if (b.has_key_map("files")) {
     // Add better validation here...
     const Object& files = b.get_key("files");
+
+    if (!files.has_key_map(m_parent->info()->hash().str()))
+      return receive_failed("Tracker scrape replay did not contain infohash.");
+
     const Object& stats = files.get_key(m_parent->info()->hash().str());
 
     if (stats.has_key_value("complete"))