Avoid integer multiplication overflow in getMemoryUsage (thanks FindB…

…ugs!)
jeanlyn · May 11, 2015 · 39434f9 · 39434f9
1 parent 1e3ad52
commit 39434f9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/core/src/main/java/org/apache/spark/shuffle/unsafe/UnsafeShuffleExternalSorter.java b/core/src/main/java/org/apache/spark/shuffle/unsafe/UnsafeShuffleExternalSorter.java
@@ -246,7 +246,7 @@ void spill() throws IOException {
   }
 
   private long getMemoryUsage() {
-    return sorter.getMemoryUsage() + (allocatedPages.size() * PAGE_SIZE);
+    return sorter.getMemoryUsage() + (allocatedPages.size() * (long) PAGE_SIZE);
   }
 
   private long freeMemory() {