Line data    Source code

       1              : use super::Command;
       2              : use lief_ffi as ffi;
       3              : 
       4              : use crate::common::FromFFI;
       5              : use crate::to_slice;
       6              : use std::marker::PhantomData;
       7              : 
       8              : /// Structure that represents the `LC_DYLIB_CODE_SIGN_DRS` command
       9              : pub struct CodeSignatureDir<'a> {
      10              :     ptr: cxx::UniquePtr<ffi::MachO_CodeSignatureDir>,
      11              :     _owner: PhantomData<&'a ffi::MachO_Binary>,
      12              : }
      13              : 
      14              : impl CodeSignatureDir<'_> {
      15              :     /// Offset in the binary where the signature starts
      16           20 :     pub fn data_offset(&self) -> u32 {
      17           20 :         self.ptr.data_offset()
      18           20 :     }
      19              : 
      20              :     /// Size of the raw signature
      21           20 :     pub fn data_size(&self) -> u32 {
      22           20 :         self.ptr.data_size()
      23           20 :     }
      24              : 
      25              :     /// Content of the signature as a slice of bytes
      26           10 :     pub fn content(&self) -> &[u8] {
      27           10 :         to_slice!(self.ptr.content());
      28           10 :     }
      29              : }
      30              : 
      31              : impl std::fmt::Debug for CodeSignatureDir<'_> {
      32           20 :     fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
      33           20 :         let base = self as &dyn Command;
      34           20 :         f.debug_struct("CodeSignatureDir")
      35           20 :             .field("base", &base)
      36           20 :             .field("data_offset", &self.data_offset())
      37           20 :             .field("data_size", &self.data_size())
      38           20 :             .finish()
      39           20 :     }
      40              : }
      41              : 
      42              : impl FromFFI<ffi::MachO_CodeSignatureDir> for CodeSignatureDir<'_> {
      43           20 :     fn from_ffi(cmd: cxx::UniquePtr<ffi::MachO_CodeSignatureDir>) -> Self {
      44           20 :         Self {
      45           20 :             ptr: cmd,
      46           20 :             _owner: PhantomData,
      47           20 :         }
      48           20 :     }
      49              : }
      50              : 
      51              : impl Command for CodeSignatureDir<'_> {
      52           80 :     fn get_base(&self) -> &ffi::MachO_Command {
      53           80 :         self.ptr.as_ref().unwrap().as_ref()
      54           80 :     }
      55              : }