From d6a00207dcbae8fb446ccda83bda35c73fc6f9c6 Mon Sep 17 00:00:00 2001 From: Pat Riehecky Date: Fri, 24 May 2024 08:11:01 -0500 Subject: [PATCH] feat(processor.extensions) Add support for CPU flag detection Signed-off-by: Pat Riehecky --- .../facts/linux/processors/extensions.rb | 16 ++++++++++++ lib/facter/framework/core/file_loader.rb | 1 + lib/facter/resolvers/processors.rb | 22 ++++++++++++++++ spec/facter/resolvers/processors_spec.rb | 25 +++++++++++++++++++ 4 files changed, 64 insertions(+) create mode 100644 lib/facter/facts/linux/processors/extensions.rb diff --git a/lib/facter/facts/linux/processors/extensions.rb b/lib/facter/facts/linux/processors/extensions.rb new file mode 100644 index 0000000000..9512ddd1cd --- /dev/null +++ b/lib/facter/facts/linux/processors/extensions.rb @@ -0,0 +1,16 @@ +# frozen_string_literal: true + +module Facts + module Linux + module Processors + class Extensions + FACT_NAME = 'processors.extensions' + + def call_the_resolver + fact_value = Facter::Resolvers::Linux::Processors.resolve(:extensions) + Facter::ResolvedFact.new(FACT_NAME, fact_value) + end + end + end + end +end diff --git a/lib/facter/framework/core/file_loader.rb b/lib/facter/framework/core/file_loader.rb index 22e3d17a88..42dc651a0b 100644 --- a/lib/facter/framework/core/file_loader.rb +++ b/lib/facter/framework/core/file_loader.rb @@ -482,6 +482,7 @@ require_relative '../../facts/linux/processor' require_relative '../../facts/linux/processors/cores' require_relative '../../facts/linux/processors/count' + require_relative '../../facts/linux/processors/extensions' require_relative '../../facts/linux/processors/isa' require_relative '../../facts/linux/processors/models' require_relative '../../facts/linux/processors/physicalcount' diff --git a/lib/facter/resolvers/processors.rb b/lib/facter/resolvers/processors.rb index 20aa58af0b..601499cedb 100644 --- a/lib/facter/resolvers/processors.rb +++ b/lib/facter/resolvers/processors.rb @@ -1,5 +1,7 @@ # frozen_string_literal: true +require 'set' + module Facter module Resolvers module Linux @@ -10,6 +12,7 @@ class Processors < BaseResolver class << self # :count + # :extensions # :models # :physical_count # :speed @@ -32,6 +35,7 @@ def read_cpuinfo(fact_name) end def read_processors(cpuinfo_output) + @fact_list[:extensions] = Set[Facter::Resolvers::Uname.resolve(:processor)] @fact_list[:processors] = 0 @fact_list[:models] = [] @fact_list[:physical_processors] = [] @@ -41,7 +45,10 @@ def read_processors(cpuinfo_output) construct_models_list(tokens) count_physical_processors(tokens) build_speed(tokens) + check_extensions(tokens) end + @fact_list[:extensions] = @fact_list[:extensions].to_a + @fact_list[:extensions].sort! end def count_processors(tokens) @@ -82,6 +89,21 @@ def build_speed_for_x86(tokens) speed = tokens.last.strip.match(/^(\d+).*/)[1] @fact_list[:speed] = speed.to_i * MHZ_TO_HZ end + + def check_extensions(tokens) + return unless tokens.first.strip == 'flags' + + flags = tokens.last.split(' ') + + # TODO: As we gain support for other arches, change the guard + # so we only check the flags for the corosponding arches + return unless @fact_list[:extensions].include?('x86_64') + + @fact_list[:extensions].add('x86_64-v1') if (%w[cmov cx8 fpu fxsr lm mmx syscall sse2] - flags).empty? + @fact_list[:extensions].add('x86_64-v2') if (%w[cx16 lahf_lm popcnt sse4_1 sse4_2 ssse3] - flags).empty? + @fact_list[:extensions].add('x86_64-v3') if (%w[abm avx avx2 bmi1 bmi2 f16c fma movbe xsave] - flags).empty? + @fact_list[:extensions].add('x86_64-v4') if (%w[avx512f avx512bw avx512cd avx512dq avx512vl] - flags).empty? + end end end end diff --git a/spec/facter/resolvers/processors_spec.rb b/spec/facter/resolvers/processors_spec.rb index 972d25e369..2f39a47658 100644 --- a/spec/facter/resolvers/processors_spec.rb +++ b/spec/facter/resolvers/processors_spec.rb @@ -3,6 +3,7 @@ describe Facter::Resolvers::Linux::Processors do after do Facter::Resolvers::Linux::Processors.invalidate_cache + Facter::Resolvers::Uname.invalidate_cache end context 'when on x86 architecture' do @@ -21,6 +22,9 @@ end let(:speed) { 2_294_000_000 } + let(:extensions) do + %w[x86_64 x86_64-v1 x86_64-v2 x86_64-v3] + end it 'returns number of processors' do result = Facter::Resolvers::Linux::Processors.resolve(:processors) @@ -45,6 +49,16 @@ expect(result).to eq(speed) end + + it 'returns extensions supported' do + allow(Facter::Resolvers::Uname).to receive(:resolve) + .with(:processor) + .and_return('x86_64') + + result = Facter::Resolvers::Linux::Processors.resolve(:extensions) + + expect(result).to eq(extensions) + end end context 'when cpuinfo file is readable but no physical id' do @@ -75,6 +89,7 @@ after do Facter::Resolvers::Linux::Processors.invalidate_cache + Facter::Resolvers::Uname.invalidate_cache end let(:physical_processors) { 2 } @@ -135,6 +150,11 @@ .and_return('1') end + after do + Facter::Resolvers::Linux::Processors.invalidate_cache + Facter::Resolvers::Uname.invalidate_cache + end + let(:speed) { 2_926_000_000 } let(:physical_processors) { 2 } @@ -186,6 +206,11 @@ .and_return('0') end + after do + Facter::Resolvers::Linux::Processors.invalidate_cache + Facter::Resolvers::Uname.invalidate_cache + end + let(:physical_processors) { 1 } it 'returns physical_devices_count' do