add NaN + Infinity checking to Fixnum+Bignum

- this might not be necessary...

1 files changed, 7 insertions, 1 deletions

diff --git a/lib/ffi_yajl/ffi/encoder.rb b/lib/ffi_yajl/ffi/encoder.rb
index 6e44796..f6aa122 100644
--- a/lib/ffi_yajl/ffi/encoder.rb
+++ b/lib/ffi_yajl/ffi/encoder.rb
@@ -104,8 +104,11 @@ end
 
 class Fixnum
   def ffi_yajl(yajl_gen, state)
+    str = self.to_s
+    if str == "NaN" || str == "Infinity" || str == "-Infinity"
+      raise ::FFI_Yajl::EncodeError.new("'#{str}' is an invalid number")
+    end
     if state[:processing_key]
-      str = self.to_s
       if ( status = FFI_Yajl.yajl_gen_string(yajl_gen, str, str.bytesize) ) != 0
         FFI_Yajl::Encoder.raise_error_for_status(status)
       end
@@ -120,6 +123,9 @@ end
 class Bignum
   def ffi_yajl(yajl_gen, state)
     str = self.to_s
+    if str == "NaN" || str == "Infinity" || str == "-Infinity"
+      raise ::FFI_Yajl::EncodeError.new("'#{str}' is an invalid number")
+    end
     if state[:processing_key]
       if ( status = FFI_Yajl.yajl_gen_string(yajl_gen, str, str.bytesize) ) != 0
         FFI_Yajl::Encoder.raise_error_for_status(status)