summaryrefslogtreecommitdiff
path: root/lib/rubygems/safe_marshal/visitors/visitor.rb
blob: c9a079dc0ea593b60a6acca1022b8ca0d867840a (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
# frozen_string_literal: true

module Gem::SafeMarshal::Visitors
  class Visitor
    def visit(target)
      send DISPATCH.fetch(target.class), target
    end

    private

    DISPATCH = Gem::SafeMarshal::Elements.constants.each_with_object({}) do |c, h|
      next if c == :Element

      klass = Gem::SafeMarshal::Elements.const_get(c)
      h[klass] = :"visit_#{klass.name.gsub("::", "_")}"
      h.default = :visit_unknown_element
    end.compare_by_identity.freeze
    private_constant :DISPATCH

    def visit_unknown_element(e)
      raise ArgumentError, "Attempting to visit unknown element #{e.inspect}"
    end

    def visit_Gem_SafeMarshal_Elements_Array(target)
      target.elements.each {|e| visit(e) }
    end

    def visit_Gem_SafeMarshal_Elements_Bignum(target); end
    def visit_Gem_SafeMarshal_Elements_False(target); end
    def visit_Gem_SafeMarshal_Elements_Float(target); end

    def visit_Gem_SafeMarshal_Elements_Hash(target)
      target.pairs.each do |k, v|
        visit(k)
        visit(v)
      end
    end

    def visit_Gem_SafeMarshal_Elements_HashWithDefaultValue(target)
      visit_Gem_SafeMarshal_Elements_Hash(target)
      visit(target.default)
    end

    def visit_Gem_SafeMarshal_Elements_Integer(target); end
    def visit_Gem_SafeMarshal_Elements_Nil(target); end

    def visit_Gem_SafeMarshal_Elements_Object(target)
      visit(target.name)
    end

    def visit_Gem_SafeMarshal_Elements_ObjectLink(target); end
    def visit_Gem_SafeMarshal_Elements_String(target); end
    def visit_Gem_SafeMarshal_Elements_Symbol(target); end
    def visit_Gem_SafeMarshal_Elements_SymbolLink(target); end
    def visit_Gem_SafeMarshal_Elements_True(target); end

    def visit_Gem_SafeMarshal_Elements_UserDefined(target)
      visit(target.name)
    end

    def visit_Gem_SafeMarshal_Elements_UserMarshal(target)
      visit(target.name)
      visit(target.data)
    end

    def visit_Gem_SafeMarshal_Elements_WithIvars(target)
      visit(target.object)
      target.ivars.each do |k, v|
        visit(k)
        visit(v)
      end
    end
  end
end