Argument checks.

frostwire · Jan 26, 2015 · 44313a5 · 44313a5
1 parent a20ce7f
commit 44313a5
Showing 1 changed file with 10 additions and 3 deletions.
diff --git a/src/com/frostwire/jlibtorrent/FileSliceTracker.java b/src/com/frostwire/jlibtorrent/FileSliceTracker.java
@@ -34,12 +34,19 @@ public int getNumSlices() {
         return slices.size();
     }
 
-    public boolean isComplete(long offset) {
-        return Boolean.TRUE.equals(slices.get(offset).second);
+    public boolean isComplete(long offset) throws IllegalArgumentException {
+        Pair<FileSlice, Boolean> p = slices.get(offset);
+        if (p == null) {
+            throw new IllegalArgumentException("offset is not contained in the internal structure");
+        }
+        return Boolean.TRUE.equals(p.second);
     }
 
-    public void setComplete(long offset, boolean complete) {
+    public void setComplete(long offset, boolean complete) throws IllegalArgumentException {
         Pair<FileSlice, Boolean> p = slices.get(offset);
+        if (p == null) {
+            throw new IllegalArgumentException("offset is not contained in the internal structure");
+        }
         slices.put(offset, new Pair<FileSlice, Boolean>(p.first, Boolean.valueOf(complete)));
     }
 }