Skip to content

Commit c3d0f64

Browse files
committed
Prefer to use YAML.safe_load
1 parent 36a339c commit c3d0f64

File tree

1 file changed

+12
-2
lines changed

1 file changed

+12
-2
lines changed

lib/yaml/dbm.rb

Lines changed: 12 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -55,7 +55,13 @@ def []=( key, val )
5555
def fetch( keystr, ifnone = nil )
5656
begin
5757
val = super( keystr )
58-
return YAML.load( val ) if String === val
58+
if String === val
59+
if YAML.respond_to?(:safe_load)
60+
return YAML.safe_load( val )
61+
else
62+
return YAML.load( val )
63+
end
64+
end
5965
rescue IndexError
6066
end
6167
if block_given?
@@ -101,7 +107,11 @@ def values_at( *keys )
101107
def delete( key )
102108
v = super( key )
103109
if String === v
104-
v = YAML.load( v )
110+
if YAML.respond_to?(:safe_load)
111+
v = YAML.safe_load( v )
112+
else
113+
v = YAML.load( v )
114+
end
105115
end
106116
v
107117
end

0 commit comments

Comments
 (0)