Policies are owned by policy model.

Author: isimluk

lib/openscap/xccdf/policy_model.rb

@@ -11,10 +11,13 @@
 
 require 'openscap/exceptions'
 require 'openscap/xccdf/benchmark'
+require 'openscap/xccdf/policy'
 
 module OpenSCAP
   module Xccdf
     class PolicyModel
+      attr_reader :raw
+
       def initialize(b)
         case b
         when OpenSCAP::Xccdf::Benchmark
@@ -26,13 +29,36 @@ def initialize(b)
         OpenSCAP.raise! if @raw.null?
       end
 
+      def policies
+        @policies ||= policies_init
+      end
+
       def destroy
         OpenSCAP.xccdf_policy_model_free @raw
         @raw = nil
       end
+
+      private
+
+      def policies_init
+        policies = {}
+        polit = OpenSCAP.xccdf_policy_model_get_policies raw
+        while OpenSCAP.xccdf_policy_iterator_has_more polit
+          policy_p = OpenSCAP.xccdf_policy_iterator_next polit
+          policy = OpenSCAP::Xccdf::Policy.new policy_p
+          policies[policy.id] = policy
+        end
+        OpenSCAP.xccdf_policy_iterator_free polit
+        policies
+      end
     end
   end
 
   attach_function :xccdf_policy_model_new, [:pointer], :pointer
   attach_function :xccdf_policy_model_free, [:pointer], :void
+
+  attach_function :xccdf_policy_model_get_policies, [:pointer], :pointer
+  attach_function :xccdf_policy_iterator_has_more, [:pointer], :bool
+  attach_function :xccdf_policy_iterator_next, [:pointer], :pointer
+  attach_function :xccdf_policy_iterator_free, [:pointer], :void
 end