inspec/lib/resources/mysql_conf.rb
Dominik Richter 40ed9799b7 feature: mysql config resource updated
Signed-off-by: Dominik Richter <dominik@vulcanosec.com>
2015-06-22 15:51:44 +02:00

84 lines
1.8 KiB
Ruby

# encoding: utf-8
# copyright: 2015, Dominik Richter
# license: All rights reserved
require 'utils/parseconfig'
class MysqlConfEntry
def initialize( path, params )
@params = params
@path = path
end
def method_missing name, *args
k = name.to_s
res = @params[k]
return true if res.nil? && @params.key?(k)
@params[k]
end
def to_s
group = ' '
group = "[#{@path.join('][')}] " unless @path.nil? or @path.empty?
"MySQL Config entry [#{@path.join(' ')}]"
end
end
class MysqlConf
def initialize( conf_path )
@runner = Specinfra::Runner
@conf_path = conf_path
@files_contents = {}
@content = nil
@params = nil
read_content
end
def content
@content ||= read_content
end
def params *opts
@params || read_content
res = @params
opts.each do |opt|
res = res[opt] unless res.nil?
end
MysqlConfEntry.new(opts, res)
end
def method_missing name
@params || read_content
@params[name.to_s]
end
def read_content
@content = ""
@params = {}
to_read = [@conf_path]
while !to_read.empty?
raw_conf = read_file(to_read[0])
@content += raw_conf
params = ParseConfig.new(raw_conf).params
@params.merge!(params)
to_read = to_read.drop(1)
# see if there is more stuff to include
include_files = raw_conf.scan(/^!include\s+(.*)\s*/).flatten.compact
include_dirs = raw_conf.scan(/^!includedir\s+(.*)\s*/).flatten.compact
include_dirs.map do |include_dir|
include_files += Dir.glob(File.join include_dir, '*')
end
to_read += include_files.find_all do |fp|
not @files_contents.key? fp
end
end
#
@content
end
def read_file(path)
@files_contents[path] ||= @runner.get_file_content(path).stdout
end
end