ppatch 0.1.0

A library to search/replace binary data via patterns
Documentation 
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
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101

use crate::{Pattern, PatternSearchType, Replacer};
use std::fmt;
use std::ops;

pub struct PatternReplaceResultIterator<'a, I, T> {
    iter: I,
    replacer: Replacer<'a, T>,
}

impl<'a, I, T> PatternReplaceResultIterator<'a, I, T>
where
    T: From<u8>
        + fmt::Binary
        + num::PrimInt
        + num::Unsigned
        + Default
        + ops::ShlAssign<u32>
        + PartialEq
        + num::PrimInt<FromStrRadixErr = std::num::ParseIntError>
        + ops::BitOrAssign
        + ops::BitAndAssign,
{
    pub fn new(iter: I, pattern: &'a Pattern<T>) -> Self {
        Self {
            iter,
            replacer: Replacer::new(pattern),
        }
    }
}

impl<'a, I, E, T> Iterator for PatternReplaceResultIterator<'a, I, T>
where
    I: Iterator<Item = std::result::Result<PatternSearchType<T>, E>>,
    E: 'static + std::error::Error,
    T: From<u8>
        + fmt::Binary
        + num::PrimInt
        + num::Unsigned
        + Default
        + ops::ShlAssign<u32>
        + PartialEq
        + num::PrimInt<FromStrRadixErr = std::num::ParseIntError>
        + ops::BitOrAssign
        + ops::BitAndAssign,
{
    type Item = crate::Result<T>;

    fn next(&mut self) -> Option<Self::Item> {
        if let Some(result) = self.replacer.handle_existing_data() {
            return Some(Ok(result));
        }

        match self.iter.next() {
            Some(result) => match result {
                Ok(search_type) => Some(self.replacer.handle_next(search_type)),
                Err(error) => Some(Err(crate::Error::IteratorError {
                    source: error.into(),
                })),
            },
            None => None,
        }
    }
}

pub trait PatternReplaceResultExt<'a, E, T>:
    Iterator<Item = Result<PatternSearchType<T>, E>> + Sized
where
    E: std::error::Error,
    T: From<u8>
        + fmt::Binary
        + num::PrimInt
        + num::Unsigned
        + Default
        + ops::ShlAssign<u32>
        + PartialEq
        + num::PrimInt<FromStrRadixErr = std::num::ParseIntError>
        + ops::BitOrAssign
        + ops::BitAndAssign,
{
    fn replace_pattern(self, pattern: &'a Pattern<T>) -> PatternReplaceResultIterator<'a, Self, T>;
}

impl<'a, I, E, T> PatternReplaceResultExt<'a, E, T> for I
where
    I: Iterator<Item = Result<PatternSearchType<T>, E>>,
    E: std::error::Error,
    T: From<u8>
        + fmt::Binary
        + num::PrimInt
        + num::Unsigned
        + Default
        + ops::ShlAssign<u32>
        + PartialEq
        + num::PrimInt<FromStrRadixErr = std::num::ParseIntError>
        + ops::BitOrAssign
        + ops::BitAndAssign,
{
    fn replace_pattern(self, pattern: &'a Pattern<T>) -> PatternReplaceResultIterator<'a, Self, T> {
        PatternReplaceResultIterator::new(self, pattern)
    }
}