CAddrMan::Deserialize handle corrupt serializations better.

PIVX-Project · Jul 28, 2021 · e986ed0 · e986ed0
1 parent d2a8baf
commit e986ed0
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/src/addrman.h b/src/addrman.h
@@ -365,6 +365,14 @@ class CAddrMan
             nUBuckets ^= (1 << 30);
         }
 
+        if (nNew > ADDRMAN_NEW_BUCKET_COUNT * ADDRMAN_BUCKET_SIZE) {
+            throw std::ios_base::failure("Corrupt CAddrMan serialization, nNew exceeds limit.");
+        }
+
+        if (nTried > ADDRMAN_TRIED_BUCKET_COUNT * ADDRMAN_BUCKET_SIZE) {
+            throw std::ios_base::failure("Corrupt CAddrMan serialization, nTried exceeds limit.");
+        }
+
         // Deserialize entries from the new table.
         for (int n = 0; n < nNew; n++) {
             CAddrInfo& info = mapInfo[n];