62 lines
1.5 KiB
Ruby
62 lines
1.5 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
module Gitlab
|
|
module Config
|
|
module Loader
|
|
class Yaml
|
|
DataTooLargeError = Class.new(Loader::FormatError)
|
|
NotHashError = Class.new(Loader::FormatError)
|
|
|
|
include Gitlab::Utils::StrongMemoize
|
|
|
|
MAX_YAML_SIZE = 1.megabyte
|
|
MAX_YAML_DEPTH = 100
|
|
|
|
def initialize(config, additional_permitted_classes: [])
|
|
@config = YAML.safe_load(config,
|
|
permitted_classes: [Symbol, *additional_permitted_classes],
|
|
permitted_symbols: [],
|
|
aliases: true
|
|
)
|
|
rescue Psych::Exception => e
|
|
raise Loader::FormatError, e.message
|
|
end
|
|
|
|
def valid?
|
|
hash? && !too_big?
|
|
end
|
|
|
|
def load_raw!
|
|
raise DataTooLargeError, 'The parsed YAML is too big' if too_big?
|
|
raise NotHashError, 'Invalid configuration format' unless hash?
|
|
|
|
@config
|
|
end
|
|
|
|
def load!
|
|
@symbolized_config ||= load_raw!.deep_symbolize_keys
|
|
end
|
|
|
|
private
|
|
|
|
def hash?
|
|
@config.is_a?(Hash)
|
|
end
|
|
|
|
def too_big?
|
|
return false unless Feature.enabled?(:ci_yaml_limit_size, default_enabled: true)
|
|
|
|
!deep_size.valid?
|
|
end
|
|
|
|
def deep_size
|
|
strong_memoize(:deep_size) do
|
|
Gitlab::Utils::DeepSize.new(@config,
|
|
max_size: MAX_YAML_SIZE,
|
|
max_depth: MAX_YAML_DEPTH)
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|