diff options
author | Aaron Patterson <aaron.patterson@gmail.com> | 2012-11-16 19:12:04 -0800 |
---|---|---|
committer | Aaron Patterson <aaron.patterson@gmail.com> | 2012-11-16 19:12:04 -0800 |
commit | 9d7be25e66a27d1a4c970921a54b19cb8ce70256 (patch) | |
tree | 1489cd5cfa75223a198ba07638b9f48c38448022 | |
parent | 4c724549f98317e098502a04162d518d25df7aea (diff) | |
download | psych-9d7be25e66a27d1a4c970921a54b19cb8ce70256.tar.gz |
* ext/psych/lib/psych/scalar_scanner.rb: avoid raising exceptions when
parsing Floats and Integers. Thanks riffraff [ruby-core:44426]
* test/psych/test_numeric.rb: associated test
-rw-r--r-- | CHANGELOG.rdoc | 6 | ||||
-rw-r--r-- | lib/psych/scalar_scanner.rb | 34 | ||||
-rw-r--r-- | test/psych/test_numeric.rb | 16 |
3 files changed, 43 insertions, 13 deletions
diff --git a/CHANGELOG.rdoc b/CHANGELOG.rdoc index b0c20fb..291ce97 100644 --- a/CHANGELOG.rdoc +++ b/CHANGELOG.rdoc @@ -1,3 +1,9 @@ +Sat Nov 17 12:03:41 2012 Aaron Patterson <aaron@tenderlovemaking.com> + + * ext/psych/lib/psych/scalar_scanner.rb: avoid raising exceptions when + parsing Floats and Integers. Thanks riffraff [ruby-core:44426] + * test/psych/test_numeric.rb: associated test + Sat Nov 17 11:26:36 2012 Aaron Patterson <aaron@tenderlovemaking.com> * ext/psych/lib/psych/core_ext.rb: move Kernel#y so that it can diff --git a/lib/psych/scalar_scanner.rb b/lib/psych/scalar_scanner.rb index 41e889a..d0beee3 100644 --- a/lib/psych/scalar_scanner.rb +++ b/lib/psych/scalar_scanner.rb @@ -8,11 +8,17 @@ module Psych TIME = /^\d{4}-\d{1,2}-\d{1,2}([Tt]|\s+)\d{1,2}:\d\d:\d\d(\.\d*)?(\s*Z|[-+]\d{1,2}(:\d\d)?)?/ # Taken from http://yaml.org/type/float.html - FLOAT = /^(?:[-+]?([0-9][0-9_,]*)?\.[0-9.]*([eE][-+][0-9]+)?(?# base 10) + FLOAT = /^(?:[-+]?([0-9][0-9_,]*)?\.[0-9]*([eE][-+][0-9]+)?(?# base 10) |[-+]?[0-9][0-9_,]*(:[0-5]?[0-9])+\.[0-9_]*(?# base 60) |[-+]?\.(inf|Inf|INF)(?# infinity) |\.(nan|NaN|NAN)(?# not a number))$/x + # Taken from http://yaml.org/type/int.html + INTEGER = /^(?:[-+]?0b[0-1_]+ (?# base 2) + |[-+]?0[0-7_]+ (?# base 8) + |[-+]?(?:0|[1-9][0-9_]*) (?# base 10) + |[-+]?0x[0-9a-fA-F_]+ (?# base 16))$/x + # Create a new scanner def initialize @string_cache = {} @@ -86,20 +92,15 @@ module Psych end i when FLOAT - begin - return Float(string.gsub(/[,_]/, '')) - rescue ArgumentError + if string == '.' + @string_cache[string] = true + string + else + Float(string.gsub(/[,_]/, '')) end - - @string_cache[string] = true - string else - if string.count('.') < 2 - begin - return Integer(string.gsub(/[,_]/, '')) - rescue ArgumentError - end - end + int = parse_int string.gsub(/[,_]/, '') + return int if int @string_cache[string] = true string @@ -107,6 +108,13 @@ module Psych end ### + # Parse and return an int from +string+ + def parse_int string + return unless INTEGER === string + Integer(string) + end + + ### # Parse and return a Time from +string+ def parse_time string date, time = *(string.split(/[ tT]/, 2)) diff --git a/test/psych/test_numeric.rb b/test/psych/test_numeric.rb index bae723a..200a9f0 100644 --- a/test/psych/test_numeric.rb +++ b/test/psych/test_numeric.rb @@ -7,6 +7,15 @@ module Psych # http://yaml.org/type/float.html # http://yaml.org/type/int.html class TestNumeric < TestCase + def setup + @old_debug = $DEBUG + $DEBUG = true + end + + def teardown + $DEBUG = @old_debug + end + def test_non_float_with_0 str = Psych.load('--- 090') assert_equal '090', str @@ -21,5 +30,12 @@ module Psych decimal = BigDecimal("12.34") assert_cycle decimal end + + def test_does_not_attempt_numeric + str = Psych.load('--- 4 roses') + assert_equal '4 roses', str + str = Psych.load('--- 1.1.1') + assert_equal '1.1.1', str + end end end |