mirror of
https://github.com/cliffe/SecGen.git
synced 2026-02-21 11:18:06 +00:00
56 lines
1.8 KiB
Ruby
56 lines
1.8 KiB
Ruby
require 'nokogiri/class_resolver'
|
|
require 'nokogiri'
|
|
|
|
require_relative '../helpers/print.rb'
|
|
require_relative '../helpers/constants.rb'
|
|
|
|
@num_modules_processed = 0
|
|
@num_errors = 0
|
|
|
|
def check_modules(type, modules_dir, schema)
|
|
Print.std "Reading #{type} modules! ***************************"
|
|
modules = []
|
|
# Get a list of all the modules
|
|
Dir.chdir(modules_dir) do
|
|
modules = Dir["**/secgen_metadata.xml"].sort
|
|
end
|
|
|
|
modules.each { |mod|
|
|
# Print.verbose "Reading #{mod}"
|
|
@num_modules_processed = @num_modules_processed + 1
|
|
doc, xsd = nil
|
|
begin
|
|
doc = Nokogiri::XML(File.read("#{modules_dir}/#{mod}"))
|
|
rescue
|
|
Print.err "Failed to read module file (#{mod})"
|
|
end
|
|
|
|
# validate scenario XML against schema
|
|
begin
|
|
xsd = Nokogiri::XML::Schema(File.open(schema))
|
|
xsd.validate("#{modules_dir}/#{mod}").each do |error|
|
|
Print.err " Error in module file - #{mod}:"
|
|
Print.err " #{error.line}: #{error.message}"
|
|
@num_errors = @num_errors + 1
|
|
|
|
end
|
|
Print.info " Valid XML - #{mod}"
|
|
|
|
rescue Exception => e
|
|
Print.err "Failed to validate module XML (#{mod}): against schema (#{VULNERABILITY_SCHEMA_FILE})"
|
|
Print.err e.message
|
|
@num_errors = @num_errors + 1
|
|
end
|
|
}
|
|
end
|
|
|
|
check_modules("vulnerability", "#{ROOT_DIR}/modules/vulnerabilities", VULNERABILITY_SCHEMA_FILE)
|
|
check_modules("service", "#{ROOT_DIR}/modules/services", SERVICE_SCHEMA_FILE)
|
|
check_modules("utility", "#{ROOT_DIR}/modules/utilities", UTILITY_SCHEMA_FILE)
|
|
check_modules("generator", "#{ROOT_DIR}/modules/generators", GENERATOR_SCHEMA_FILE)
|
|
check_modules("encoder", "#{ROOT_DIR}/modules/encoders", ENCODER_SCHEMA_FILE)
|
|
check_modules("base", "#{ROOT_DIR}/modules/bases", BASE_SCHEMA_FILE)
|
|
|
|
Print.std "Tested validation of #{@num_modules_processed} modules"
|
|
Print.std "#{@num_errors} errors"
|