abide_dev_utils 0.11.2 → 0.12.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 85fb0e453df008a48284f03fe59bd1577b733498ce030039e215bf1863e1767c
-  data.tar.gz: 9a89ac46e8506ea059044006a5c50d8132847fa2a8f9a6b726174ed51ab6421c
+  metadata.gz: 1e93a77cf4142b06ee4d41863ef2d0b737d21d37306b57082aa316f2731de5e9
+  data.tar.gz: 9b49e05aa575884d54efed8fd70b5662712845c083a8b59caeccf406a19f5633
 SHA512:
-  metadata.gz: 9f221ac5c9c15db1414005ca4017dda7a9f848d8b7586d4b8a3ede9ecaf5fc8e35187ab2f172e127cad92a1d483befe31fa17278279b54b33abcc1d25f7a21a6
-  data.tar.gz: b9f3e5553a03faed668600c94b44759ee4554165e20722b7c6d5108add327ba23a6986e707034647b3eed400e6217b032eb058e537a74f26d11a004cf309bcfd
+  metadata.gz: 51316ad752020f3b2297acbdc4bb328c5275d03e499721ee12381aaac5da3a6e77ecb9efeb0ffa4b3b6bb512154dbb4b36bf249f4816944a0110d3ceeb871b2b
+  data.tar.gz: 40b83d7be09ccb1349650617548276f6f34e266fe2e590e8c1b28b8e9c44f4499c93652abf9e254d77469fec9c98e9394a1ae2f4b2d3f0921827c3ce302dfb7e

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    abide_dev_utils (0.11.2)
+    abide_dev_utils (0.12.2)
       amatch (~> 0.4)
       cmdparse (~> 3.0)
       facterdb (>= 1.18)

data/lib/abide_dev_utils/cem/generate/reference.rb

@@ -120,7 +120,11 @@ module AbideDevUtils
           private
 
           def heading_builder
-            @md.add_h2("#{@control.number} - #{@control.title}")
+            if @framework == 'stig'
+              @md.add_h2(@control.id)
+            else
+              @md.add_h2("#{@control.number} - #{@control.title}")
+            end
           end
 
           def control_has_valid_params?
@@ -216,12 +220,13 @@ module AbideDevUtils
             end
             return if out_str.empty?
 
-            out_str.unshift("    #{@control.title.dump}:")
+            @control.title.nil? ? out_str.unshift("    #{@control.id.dump}:") : out_str.unshift("    #{@control.title.dump}:")
             out_str.unshift('  control_configs:')
             out_str.unshift("#{@module_name}::config:")
             @md.add_ul('Hiera Configuration Example:')
             @md.add_code_block(out_str.join("\n"), language: 'yaml')
           rescue StandardError => e
+            require 'pry'; binding.pry
             err_msg = [
               "Failed to generate config example for control #{@control.id}",
               "Error: #{e.message}",

data/lib/abide_dev_utils/cem/mapping/mapper.rb

@@ -6,13 +6,13 @@ require 'abide_dev_utils/cem/hiera_data/mapping_data/mixins'
 module AbideDevUtils
   module CEM
     module Mapping
-      ALL_TYPES = %w[hiera_title_num number hiera_title vulnid title].freeze
+      ALL_TYPES = %w[hiera_title_num number hiera_title title vulnid ruleid].freeze
       FRAMEWORK_TYPES = {
         'cis' => %w[hiera_title_num number hiera_title title],
-        'stig' => %w[hiera_title_num number hiera_title vulnid title],
+        'stig' => %w[vulnid ruleid],
       }.freeze
       CIS_TYPES = %w[hiera_title_num number hiera_title title].freeze
-      STIG_TYPES = %w[hiera_title_num number hiera_title vulnid title].freeze
+      STIG_TYPES = %w[vulnid ruleid].freeze
 
       # Represents a single map data file
       class MapData
@@ -168,6 +168,8 @@ module AbideDevUtils
             'hiera_title'
           when %r{^V-[0-9]{6}$}
             'vulnid'
+          when %r{^SV-[0-9]+r[0-9]+_rule$}
+            'ruleid'
           else
             'title'
           end

data/lib/abide_dev_utils/cli/jira.rb

@@ -22,6 +22,7 @@ module Abide
         add_command(JiraGetIssueCommand.new)
         add_command(JiraNewIssueCommand.new)
         add_command(JiraFromCoverageCommand.new)
+        add_command(JiraFromXccdfCommand.new)
       end
     end
 
@@ -113,5 +114,26 @@ module Abide
         end
       end
     end
+
+    class JiraFromXccdfCommand < CmdParse::Command
+      CMD_NAME = 'from_xccdf'
+      CMD_SHORT = 'Creates a parent issue with subtasks from a xccdf file'
+      CMD_LONG = 'Creates a parent issue with subtasks for a benchmark and any uncovered controls'
+      def initialize
+        super(CMD_NAME, takes_commands: false)
+        short_desc(CMD_SHORT)
+        long_desc(CMD_LONG)
+        argument_desc(PATH: 'An XCCDF file from the abide puppet ticket coverage command', PROJECT: 'A Jira project')
+        options.on('-d', '--dry-run', 'Print to console instead of saving objects') { |_| @data[:dry_run] = true }
+      end
+
+      def execute(path, project)
+        Abide::CLI::VALIDATE.file(path)
+        @data[:dry_run] = false if @data[:dry_run].nil?
+        client = JIRA.client(options: {})
+        proj = JIRA.project(client, project)
+        JIRA.new_issues_from_xccdf(client, proj, path, dry_run: @data[:dry_run])
+      end
+    end
   end
 end

data/lib/abide_dev_utils/jira.rb

@@ -135,6 +135,36 @@ module AbideDevUtils
       end
     end
 
+    def self.new_issues_from_xccdf(client, project, xccdf_path, dry_run: false)
+      dr_prefix = dry_run ? 'DRY RUN: ' : ''
+      i_attrs = all_project_issues_attrs(project)
+
+      xccdf = AbideDevUtils::XCCDF::Benchmark.new(xccdf_path)
+
+      summaries = summaries_from_xccdf(xccdf)
+      summaries.each do |profile_summary, control_summaries|
+        if summary_exist?(profile_summary, i_attrs)
+          AbideDevUtils::Output.simple("#{dr_prefix}Skipping #{profile_summary} as it already exists")
+          next
+        end
+
+        parent = new_issue(client, project.attrs['key'], profile_summary, dry_run: dry_run)
+        AbideDevUtils::Output.simple("#{dr_prefix}Created parent issue #{profile_summary}")
+        parent_issue = issue(client, parent.attrs['key']) unless parent.respond_to?(:dummy)
+        AbideDevUtils::Output.simple("#{dr_prefix}Creating subtasks, this can take a while...")
+        progress = AbideDevUtils::Output.progress(title: "#{dr_prefix}Creating Subtasks", total: nil)
+        control_summaries.each do |control_summary|
+          next if summary_exist?(control_summary, i_attrs)
+
+          progress.title = "#{dr_prefix}#{control_summary}"
+          new_subtask(client, parent_issue, control_summary, dry_run: dry_run)
+          progress.increment
+        end
+        final_text = "#{dr_prefix}Created #{control_summaries.count} subtasks for #{profile_summary}"
+        puts "\r\033[K#{final_text}\n"
+      end
+    end
+
     # def self.new_issues_from_comply_report(client, project, report, dry_run: false)
     #   dr_prefix = dry_run ? 'DRY RUN: ' : ''
     #   i_attrs = all_project_issues_attrs(project)
@@ -177,6 +207,21 @@ module AbideDevUtils
       summaries.transform_keys { |k| "#{COV_PARENT_SUMMARY_PREFIX}#{benchmark}-#{k}"}
     end
 
+    def self.summaries_from_xccdf(xccdf)
+      summaries = {}
+      facter_os = xccdf.facter_benchmark.join('-')
+      xccdf.profiles.each do |profile|
+        summaries["#{COV_PARENT_SUMMARY_PREFIX}#{facter_os} - #{profile.level} #{profile.title}"] = profile.controls.collect do |control|
+          summary = "#{COV_CHILD_SUMMARY_PREFIX}#{control.vulnid} - #{control.title}"
+          if summary.length > 255
+            summary = summary[0..251] + '...'
+          end
+          summary
+        end
+      end
+      summaries
+    end
+
     # def self.summaries_from_comply_report(report)
     #   summaries = {}
     #   report.each do |_, v|

data/lib/abide_dev_utils/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module AbideDevUtils
-  VERSION = "0.11.2"
+  VERSION = "0.12.2"
 end

data/lib/abide_dev_utils/xccdf.rb

@@ -16,6 +16,8 @@ module AbideDevUtils
       case type.downcase
       when 'cis'
         Benchmark.new(xccdf_file).gen_map(**opts)
+      when 'stig'
+        Benchmark.new(xccdf_file).gen_map(**opts)
       else
         raise AbideDevUtils::Errors::UnsupportedXCCDFError, "XCCDF type #{type} is unsupported!"
       end
@@ -51,26 +53,31 @@ module AbideDevUtils
     module Common
       XPATHS = {
         benchmark: {
-          all: 'xccdf:Benchmark',
-          title: 'xccdf:Benchmark/xccdf:title',
-          version: 'xccdf:Benchmark/xccdf:version'
+          all: 'Benchmark',
+          title: 'Benchmark/title',
+          version: 'Benchmark/version'
         },
         cis: {
           profiles: {
-            all: 'xccdf:Benchmark/xccdf:Profile',
-            relative_title: './xccdf:title',
-            relative_select: './xccdf:select'
+            all: 'Benchmark/Profile',
+            relative_title: './title',
+            relative_select: './select'
           }
         }
       }.freeze
       CONTROL_PREFIX = /^[\d.]+_/.freeze
       UNDERSCORED = /(\s|\(|\)|-|\.)/.freeze
+      CIS_TITLE_MARKER = 'CIS'
       CIS_NEXT_GEN_WINDOWS = /[Nn]ext_[Gg]eneration_[Ww]indows_[Ss]ecurity/.freeze
       CIS_CONTROL_NUMBER = /([0-9.]+[0-9]+)/.freeze
       CIS_LEVEL_CODE = /(?:_|^)([Ll]evel_[0-9]|[Ll]1|[Ll]2|[NnBb][GgLl]|#{CIS_NEXT_GEN_WINDOWS})/.freeze
       CIS_CONTROL_PARTS = /#{CIS_CONTROL_NUMBER}#{CIS_LEVEL_CODE}?_+([A-Za-z].*)/.freeze
       CIS_PROFILE_PARTS = /#{CIS_LEVEL_CODE}[_-]+([A-Za-z].*)/.freeze
-      STIG_PROFILE_PARTS = /(STIG)/.freeze
+      STIG_TITLE_MARKER = 'Security Technical Implementation Guide'
+      STIG_CONTROL_PARTS = /(V-[0-9]+)/.freeze
+      STIG_PROFILE_PARTS = /(MAC-\d+)_([A-Za-z].+)/.freeze
+      PROFILE_PARTS = /#{CIS_PROFILE_PARTS}|#{STIG_PROFILE_PARTS}/.freeze
+      CONTROL_PARTS = /#{CIS_CONTROL_PARTS}|#{STIG_CONTROL_PARTS}/.freeze
 
       def xpath(path)
         @xml.xpath(path)
@@ -120,21 +127,41 @@ module AbideDevUtils
       end
 
       def profile_parts(profile)
-        return ['STIG', ''] if profile == 'STIG'
-
-        parts = control_profile_text(profile).match(CIS_PROFILE_PARTS)
+        parts = control_profile_text(profile).match(PROFILE_PARTS)
         raise AbideDevUtils::Errors::ProfilePartsError, profile if parts.nil?
 
-        parts[1].gsub!(/[Ll]evel_/, 'L')
-        parts[1..2]
+        if parts[1]
+          # CIS profile
+          parts[1].gsub!(/[Ll]evel_/, 'L')
+          parts[1..2]
+        elsif parts[3]
+          # STIG profile
+          parts[3..4]
+        else
+          raise AbideDevUtils::Errors::ProfilePartsError, profile
+        end
       end
 
-      def control_parts(control, parent_level: nil)
-        mdata = control_profile_text(control).match(CIS_CONTROL_PARTS)
+      def control_parts(control)
+        mdata = control_profile_text(control).match(CONTROL_PARTS)
         raise AbideDevUtils::Errors::ControlPartsError, control if mdata.nil?
 
-        mdata[2] = parent_level unless parent_level.nil?
-        mdata[1..3]
+        if mdata[1]
+          # CIS control
+          mdata[1..3]
+        elsif mdata[4]
+          # STIG control
+          vuln_id = mdata[4]
+          group = @benchmark.xpath("Group[@id='#{vuln_id}']")
+          if group.xpath('Rule').length != 1
+            raise AbideDevUtils::Errors::ControlPartsError, control
+          end
+          rule_id = group.xpath('Rule/@id').first.value
+          title = group.xpath('Rule/title').text
+          return [vuln_id, rule_id, title]
+        else
+          raise AbideDevUtils::Errors::ControlPartsError, control
+        end
       end
 
       def control_profile_text(item)
@@ -151,14 +178,6 @@ module AbideDevUtils
         end
       end
 
-      def sorted_control_classes(raw_select_list, sort_key: :number)
-        raw_select_list.map { |x| Control.new(x) }.sort_by(&sort_key)
-      end
-
-      def sorted_profile_classes(raw_profile_list, sort_key: :title)
-        raw_profile_list.map { |x| Profile.new(x) }.sort_by(&sort_key)
-      end
-
       def ==(other)
         diff_properties.map { |x| send(x) } == other.diff_properties.map { |x| other.send(x) }
       end
@@ -172,14 +191,17 @@ module AbideDevUtils
     class Benchmark
       include AbideDevUtils::XCCDF::Common
 
-      MAP_INDICES = %w[title hiera_title hiera_title_num number].freeze
+      CIS_MAP_INDICES = %w[title hiera_title hiera_title_num number].freeze
+      STIG_MAP_INDICES = %w[vulnid ruleid].freeze
 
-      attr_reader :xml, :title, :version, :diff_properties
+      attr_reader :xml, :title, :version, :diff_properties, :benchmark
 
       def initialize(path)
         @xml = parse(path)
-        @title = xpath('xccdf:Benchmark/xccdf:title').text
-        @version = xpath('xccdf:Benchmark/xccdf:version').text
+        @xml.remove_namespaces!
+        @benchmark = xpath('Benchmark')
+        @title = xpath('Benchmark/title').text
+        @version = xpath('Benchmark/version').text
         @diff_properties = %i[title version profiles]
       end
 
@@ -188,7 +210,7 @@ module AbideDevUtils
       end
 
       def profiles
-        @profiles ||= Profiles.new(xpath('xccdf:Benchmark/xccdf:Profile'))
+        @profiles ||= Profiles.new(xpath('Benchmark/Profile'), @benchmark)
       end
 
       def profile_levels
@@ -200,7 +222,7 @@ module AbideDevUtils
       end
 
       def controls
-        @controls ||= Controls.new(xpath('//xccdf:select'))
+        @controls ||= Controls.new(xpath('//select'))
       end
 
       def controls_by_profile_level(level_code)
@@ -212,15 +234,22 @@ module AbideDevUtils
       end
 
       def gen_map(dir: nil, type: 'cis', parent_key_prefix: '', version_output_dir: false, **_)
-        os, ver = facter_platform
+        case type
+        when 'cis'
+          os, ver = facter_platform
+          indicies = CIS_MAP_INDICES
+        when 'stig'
+          os, ver = facter_benchmark
+          indicies = STIG_MAP_INDICES
+        end
         output_path = [type, os, ver]
         output_path.unshift(File.expand_path(dir)) if dir
         output_path << version if version_output_dir
         mapping_dir = File.expand_path(File.join(output_path))
         parent_key_prefix = '' if parent_key_prefix.nil?
-        MAP_INDICES.each_with_object({}) do |idx, h|
+        indicies.each_with_object({}) do |idx, h|
           map_file_path = "#{mapping_dir}/#{idx}.yaml"
-          h[map_file_path] = map_indexed(index: idx, framework: type, key_prefix: parent_key_prefix)
+          h[map_file_path] = map_indexed(indicies: indicies, index: idx, framework: type, key_prefix: parent_key_prefix)
         end
       end
 
@@ -240,10 +269,10 @@ module AbideDevUtils
         }
       end
 
-      def map_indexed(index: 'title', framework: 'cis', key_prefix: '')
+      def map_indexed(indicies: [], index: 'title', framework: 'cis', key_prefix: '')
         c_map = profiles.each_with_object({}) do |profile, obj|
           obj[profile.level.downcase] = {} unless obj[profile.level.downcase].is_a?(Hash)
-          obj[profile.level.downcase][profile.title.downcase] = map_controls_hash(profile, index).sort_by { |k, _| k }.to_h
+          obj[profile.level.downcase][profile.title.downcase] = map_controls_hash(profile, indicies, index).sort_by { |k, _| k }.to_h
         end
 
         c_map['benchmark'] = { 'title' => title, 'version' => version }
@@ -252,8 +281,13 @@ module AbideDevUtils
         { mappings.join('::') => c_map }.to_yaml
       end
 
+      def facter_benchmark
+        id = xpath('Benchmark/@id').text
+        id.split('_')[0..-2]
+      end
+
       def facter_platform
-        cpe = xpath('xccdf:Benchmark/xccdf:platform')[0]['idref'].split(':')
+        cpe = xpath('Benchmark/platform')[0]['idref'].split(':')
         if cpe.length > 4
           product_name = cpe[4].split('_')
           product_version = cpe[5].split('.') unless cpe[5].nil?
@@ -283,8 +317,8 @@ module AbideDevUtils
         hash.to_yaml
       end
 
-      def resolve_control_reference(control)
-        xpath("//xccdf:Rule[@id='#{control.reference}']")
+      def resolve_cis_control_reference(control)
+        xpath("//Rule[@id='#{control.reference}']")
       end
 
       private
@@ -294,15 +328,15 @@ module AbideDevUtils
         when 'hiera_title_num'
           control.hiera_title(number_format: true)
         when 'title'
-          resolve_control_reference(control).xpath('./xccdf:title').text
+          resolve_cis_control_reference(control).xpath('./title').text
         else
           control.send(index.to_sym)
         end
       end
 
-      def map_controls_hash(profile, index)
+      def map_controls_hash(profile, indicies, index)
         profile.controls.each_with_object({}) do |ctrl, hsh|
-          control_array = MAP_INDICES.each_with_object([]) do |idx_sym, ary|
+          control_array = indicies.each_with_object([]) do |idx_sym, ary|
             next if idx_sym == index
 
             item = format_map_control_index(idx_sym, ctrl)
@@ -319,13 +353,9 @@ module AbideDevUtils
         end
       end
 
-      def sorted_profile_classes(raw_profile_list, sort_key: :level)
-        raw_profile_list.map { |x| Profile.new(x) }.sort_by(&sort_key)
-      end
-
       def find_profiles
         profs = {}
-        xpath('xccdf:Benchmark/xccdf:Profile').each do |profile|
+        xpath('Benchmark/Profile').each do |profile|
           level_code, name = profile_parts(profile['id'])
           profs[name] = {} unless profs.key?(name)
           profs[name][level_code] = profile
@@ -352,11 +382,66 @@ module AbideDevUtils
       end
     end
 
-    class ObjectContainer
+    class XccdfObject
+      include AbideDevUtils::XCCDF::Common
+
+      def initialize(benchmark)
+        @benchmark = benchmark
+        @benchmark_type = benchmark_type
+      end
+
+      def controls_class
+        case @benchmark_type
+        when :cis
+          CisControls
+        when :stig
+          StigControls
+        else
+          raise AbideDevUtils::Errors::UnsupportedXCCDFError
+        end
+      end
+
+      def control_sort_key
+        case @benchmark_type
+        when :cis
+          :number
+        when :stig
+          :vulnid
+        else
+          raise AbideDevUtils::Errors::UnsupportedXCCDFError
+        end
+      end
+
+      def control_class
+        case @benchmark_type
+        when :cis
+          CisControl
+        when :stig
+          StigControl
+        else
+          raise AbideDevUtils::Errors::UnsupportedXCCDFError
+        end
+      end
+
+      private
+
+      def benchmark_type
+        title = @benchmark.at_xpath('title').text
+        if title.include?(STIG_TITLE_MARKER)
+          return :stig
+        elsif title.include?(CIS_TITLE_MARKER)
+          return :cis
+        end
+        raise AbideDevUtils::Errors::UnsupportedXCCDFError, "XCCDF type is unsupported!"
+      end
+    end
+
+    class ObjectContainer < XccdfObject
       include AbideDevUtils::XCCDF::Common
 
-      def initialize(list, object_creation_method, *args, **kwargs)
-        @object_list = send(object_creation_method.to_sym, list, *args, **kwargs)
+      def initialize(list, object_creation_method, benchmark, *args, **kwargs)
+        super(benchmark)
+        @object_list = send(object_creation_method.to_sym, list, benchmark, *args, **kwargs)
         @searchable = []
       end
 
@@ -399,6 +484,14 @@ module AbideDevUtils
 
       private
 
+      def sorted_control_classes(raw_select_list, benchmark)
+        raw_select_list.map { |x| control_class.new(x, benchmark) }.sort_by(&control_sort_key)
+      end
+
+      def sorted_profile_classes(raw_profile_list, benchmark)
+        raw_profile_list.map { |x| Profile.new(x, benchmark) }.sort_by(&:title)
+      end
+
       def resolve_hash_key(obj)
         return obj.send(:raw_title) unless defined?(@hash_key)
 
@@ -419,8 +512,8 @@ module AbideDevUtils
     end
 
     class Profiles < ObjectContainer
-      def initialize(list)
-        super(list, :sorted_profile_classes)
+      def initialize(list, benchmark)
+        super(list, :sorted_profile_classes, benchmark)
         searchable! :level, :title
         index! :title
         hash_key! :level, :title
@@ -443,9 +536,34 @@ module AbideDevUtils
       end
     end
 
-    class Controls < ObjectContainer
-      def initialize(list)
-        super(list, :sorted_control_classes)
+    class StigControls < ObjectContainer
+      def initialize(list, benchmark)
+        super(list, :sorted_control_classes, benchmark)
+        searchable! :vulnid, :ruleid
+        index! :vulnid
+        hash_key! :vulnid
+      end
+
+      def vulnids
+        @vulnids ||= @object_list.map(&:vulnid).sort
+      end
+
+      def ruleids
+        @ruleids ||= @object_list.map(&:ruleid).sort
+      end
+
+      def include_vulnid?(item)
+        @object_list.map(&:vulnid).include?(item)
+      end
+
+      def include_ruleid?(item)
+        @object_list.map(&:ruleid).include?(item)
+      end
+    end
+
+    class CisControls < ObjectContainer
+      def initialize(list, benchmark)
+        super(list, :sorted_control_classes, benchmark)
         searchable! :level, :title, :number
         index! :number
         hash_key! :number
@@ -476,10 +594,11 @@ module AbideDevUtils
       end
     end
 
-    class XccdfElement
+    class XccdfElement < XccdfObject
       include AbideDevUtils::XCCDF::Common
 
-      def initialize(element)
+      def initialize(element, benchmark)
+        super(benchmark)
         @xml = element
         @element_type = self.class.name.split('::').last.downcase
         @raw_title = control_profile_text(element)
@@ -501,11 +620,12 @@ module AbideDevUtils
       end
 
       def reference
-        @reference ||= @element_type == 'control' ? @xml['idref'] : @xml['id']
+        @reference ||= @element_type.include?('control') ? @xml['idref'] : @xml['id']
       end
 
       def hiera_title(**opts)
-        send("normalize_#{@element_type}_name".to_sym, @xml, **opts)
+        e_type = @element_type.include?('control') ? 'control' : 'profile'
+        send("normalize_#{e_type}_name".to_sym, @xml, **opts)
       end
 
       private
@@ -527,19 +647,27 @@ module AbideDevUtils
     end
 
     class Profile < XccdfElement
-      def initialize(profile)
-        super(profile)
+      def initialize(profile, benchmark)
+        super(profile, benchmark)
         @level, @title = profile_parts(control_profile_text(profile))
-        @plain_text_title = @xml.xpath('./xccdf:title').text
-        @controls = Controls.new(xpath('./xccdf:select'))
+        @plain_text_title = @xml.xpath('./title').text
+        @controls = controls_class.new(xpath('./select'), benchmark)
         properties :title, :level, :plain_text_title, controls: :to_h
       end
     end
 
-    class Control < XccdfElement
-      def initialize(control, parent_level: nil)
-        super(control)
-        @number, @level, @title = control_parts(control_profile_text(control), parent_level: parent_level)
+    class StigControl < XccdfElement
+      def initialize(control, benchmark)
+        super(control, benchmark)
+        @vulnid, @ruleid, @title = control_parts(control_profile_text(control))
+        properties :vulnid, :ruleid, :title
+      end
+    end
+
+    class CisControl < XccdfElement
+      def initialize(control, benchmark)
+        super(control, benchmark)
+        @number, @level, @title = control_parts(control_profile_text(control))
         properties :number, :level, :title
       end
     end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: abide_dev_utils
 version: !ruby/object:Gem::Version
-  version: 0.11.2
+  version: 0.12.2
 platform: ruby
 authors:
 - abide-team
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2022-08-15 00:00:00.000000000 Z
+date: 2022-10-17 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: nokogiri