[Truffle] Extreme bounds checking for Float#round.

jruby · Dec 18, 2014 · a851478 · a851478
1 parent d56d246
commit a851478
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/core/src/main/java/org/jruby/truffle/nodes/core/FloatNodes.java b/core/src/main/java/org/jruby/truffle/nodes/core/FloatNodes.java
@@ -575,6 +575,14 @@ public RoundNode(RoundNode prev) {
         public Object round(double n) {
             // Algorithm copied from JRuby - not shared as we want to branch profile it
 
+            if (Double.isInfinite(n)) {
+                throw new RaiseException(getContext().getCoreLibrary().floatDomainError("Infinity", this));
+            }
+
+            if (Double.isNaN(n)) {
+                throw new RaiseException(getContext().getCoreLibrary().floatDomainError("NaN", this));
+            }
+
             double f = n;
 
             if (f > 0.0) {