57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112

# File 'lib/ronin/extensions/ip_addr.rb', line 57

def IPAddr.each(cidr_or_glob,&block)
  unless (cidr_or_glob.include?('*') || cidr_or_glob.include?('-'))
    IPAddr.new(cidr_or_glob).each(&block)
    return nil
  end

  if cidr_or_glob.include?('::')
    prefix = if cidr_or_glob =~ /^::/
               '::'
             else
               ''
             end

    separator = '::'
    base = 16

    format = lambda { |address|
      prefix + address.map { |i| '%.2x' % i }.join('::')
    }
  else
    separator = '.'
    base = 10

    format = lambda { |address| address.join('.') }
  end

  ranges = cidr_or_glob.split(separator).map { |segment|
    if segment == '*'
      (1..254)
    elsif segment.include?('-')
      start, stop = segment.split('-',2).map { |i| i.to_i(base) }

      (start..stop)
    elsif !(segment.empty?)
      segment.to_i(base)
    end
  }.compact

  expand_range = lambda { |address,remaining|
    if remaining.empty?
      block.call(format.call(address))
    else
      n = remaining.first
      remaining = remaining[1..-1]

      if n.kind_of?(Range)
        n.each { |i| expand_range.call(address + [i], remaining) }
      else
        expand_range.call(address + [n], remaining)
      end
    end
  }

  expand_range.call([], ranges)
  return nil
end