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
use std::collections::HashSet;
use std::convert::AsRef;
use std::hash::Hash;
use std::io;
use lber::structures::{Tag, Enumerated, Sequence, Set, OctetString};
use lber::common::TagClass;
use futures::{future, Future};
use tokio_service::Service;
use ldap::{Ldap, LdapOp, next_req_controls};
use result::LdapResult;
#[derive(Clone, Debug, PartialEq)]
pub enum Mod<S: AsRef<str> + Eq + Hash> {
Add(S, HashSet<S>),
Delete(S, HashSet<S>),
Replace(S, HashSet<S>),
}
impl Ldap {
pub fn modify<S: AsRef<str> + Eq + Hash>(&self, dn: &str, mods: Vec<Mod<S>>) ->
Box<Future<Item=LdapResult, Error=io::Error>> {
let mut any_add_empty = false;
let req = Tag::Sequence(Sequence {
id: 6,
class: TagClass::Application,
inner: vec![
Tag::OctetString(OctetString {
inner: Vec::from(dn.as_bytes()),
.. Default::default()
}),
Tag::Sequence(Sequence {
inner: mods.into_iter().map(|m| {
let mut is_add = false;
let (num, attr, set) = match m {
Mod::Add(attr, set) => { is_add = true; (0, attr, set) },
Mod::Delete(attr, set) => (1, attr, set),
Mod::Replace(attr, set) => (2, attr, set),
};
if set.is_empty() && is_add {
any_add_empty = true;
}
let op = Tag::Enumerated(Enumerated {
inner: num,
.. Default::default()
});
let part_attr = Tag::Sequence(Sequence {
inner: vec![
Tag::OctetString(OctetString {
inner: Vec::from(attr.as_ref()),
.. Default::default()
}),
Tag::Set(Set {
inner: set.into_iter().map(|val| {
Tag::OctetString(OctetString {
inner: Vec::from(val.as_ref()),
.. Default::default()
})
}).collect(),
.. Default::default()
})
],
.. Default::default()
});
Tag::Sequence(Sequence {
inner: vec![op, part_attr],
.. Default::default()
})
}).collect(),
.. Default::default()
})
]
});
if any_add_empty {
return Box::new(future::err(io::Error::new(io::ErrorKind::Other, "empty value set for Add")));
}
let fut = self.call(LdapOp::Single(req, next_req_controls(self)))
.and_then(|response| {
let (mut result, controls) = (LdapResult::from(response.0), response.1);
result.ctrls = controls;
Ok(result)
});
Box::new(fut)
}
}