Line data    Source code

       1              : use lief_ffi as ffi;
       2              : 
       3              : use crate::common::FromFFI;
       4              : use std::marker::PhantomData;
       5              : 
       6              : pub struct CodeIntegrity<'a> {
       7              :     ptr: cxx::UniquePtr<ffi::PE_CodeIntegrity>,
       8              :     _owner: PhantomData<&'a ffi::PE_LoadConfiguration>,
       9              : }
      10              : 
      11              : impl std::fmt::Debug for CodeIntegrity<'_> {
      12            0 :     fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
      13            0 :         f.debug_struct("CodeIntegrity")
      14            0 :             .field("flags", &self.flags())
      15            0 :             .field("catalog", &self.catalog())
      16            0 :             .field("catalog_offset", &self.catalog_offset())
      17            0 :             .field("reserved", &self.reserved())
      18            0 :             .finish()
      19            0 :     }
      20              : }
      21              : 
      22              : impl CodeIntegrity<'_> {
      23              :     /// Flags to indicate if CI information is available, etc.
      24            0 :     pub fn flags(&self) -> u16 {
      25            0 :         self.ptr.flags()
      26            0 :     }
      27              : 
      28              :     /// 0xFFFF means not available
      29            0 :     pub fn catalog(&self) -> u16 {
      30            0 :         self.ptr.catalog()
      31            0 :     }
      32            0 :     pub fn catalog_offset(&self) -> u32 {
      33            0 :         self.ptr.catalog_offset()
      34            0 :     }
      35              : 
      36              :     /// Additional bitmask to be defined later
      37            0 :     pub fn reserved(&self) -> u32 {
      38            0 :         self.ptr.reserved()
      39            0 :     }
      40              : }
      41              : 
      42              : impl<'a> FromFFI<ffi::PE_CodeIntegrity> for CodeIntegrity<'a> {
      43            0 :     fn from_ffi(ptr: cxx::UniquePtr<ffi::PE_CodeIntegrity>) -> Self {
      44            0 :         CodeIntegrity {
      45            0 :             ptr,
      46            0 :             _owner: PhantomData,
      47            0 :         }
      48            0 :     }
      49              : }