ntdsextract2 1.4.13

Display contents of Active Directory database files (ntds.dit)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59

use std::{
    collections::HashMap,
    ops::{Deref, Index},
};

use libesedb::Table;

use super::{Column, ColumnIndex};

pub struct ColumnsOfTable {
    ids: Vec<Column>,
    names: HashMap<String, ColumnIndex>,
}

impl TryFrom<&Table<'_>> for ColumnsOfTable {
    type Error = std::io::Error;

    fn try_from(table: &Table) -> Result<Self, Self::Error> {
        let mut ids = Vec::new();
        let mut names = HashMap::new();
        for (column, index) in table.iter_columns()?.zip(0..) {
            let column = Column::new(column?, index.into())?;
            names.insert(column.name().to_owned(), index.into());
            ids.push(column);
        }
        Ok(Self { ids, names })
    }
}

impl ColumnsOfTable {
    pub fn iter(&self) -> impl Iterator<Item = &Column> {
        self.ids.iter()
    }
}

impl Index<ColumnIndex> for ColumnsOfTable {
    type Output = Column;

    fn index(&self, index: ColumnIndex) -> &Self::Output {
        self.ids.index(*index as usize)
    }
}

impl Index<&ColumnIndex> for ColumnsOfTable {
    type Output = Column;

    fn index(&self, index: &ColumnIndex) -> &Self::Output {
        self.ids.index(*(index.deref()) as usize)
    }
}

impl Index<&str> for ColumnsOfTable {
    type Output = Column;

    fn index(&self, name: &str) -> &Self::Output {
        let index = self.names[name];
        self.ids.index(*(index.deref()) as usize)
    }
}