xmlsec 0.3.0

Wrapper for xmlsec1 library
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

//!
//! XmlSec Extensions over LibXML2 Wrapper
//!
use crate::XmlSecResult;

use crate::XmlDocument;
use crate::XmlXPathContext;

use libxml::bindings; // FIXME requires common bindings generation over libxml2

use std::ffi::CString;
use std::os::raw::c_uchar;
use std::ptr::null_mut;


/// Extensions on XmlDocument for the usage by or with XmlSec.
pub trait XmlSecDocumentExt
{
    /// Specifies the name of an ID attribute on a set of nodes selected by an xpath search. See examples for usage.
    fn specify_idattr(&self, search: &str, idattr_name: &str, namespaces: Option<&[(&str, &str)]>) -> XmlSecResult<()>;
}


impl XmlSecDocumentExt for XmlDocument
{
    fn specify_idattr(&self, search: &str, idattr_name: &str, namespaces: Option<&[(&str, &str)]>) -> XmlSecResult<()>
    {
        let xpath = XmlXPathContext::new(self)
            .expect("Should not have failed to build xpath context XML document");

        if let Some(nss) = namespaces
        {
            for (prefix, href) in nss {
                xpath.register_namespace(prefix, href).ok();
            }
        }

        let result = match xpath.evaluate(search)
        {
            Ok(s)  => { s },
            Err(_) => { return Err(format!("Should not have failed while xpath searching for '{}'", search).into()); }
        };

        let subjnodes = result.get_nodes_as_vec();

        if subjnodes.is_empty()
        {
            return Err(
                format!("Could not find any nodes to specify ID attribute that go by the XPath: {}", search).into()
            );
        }

        for node in &subjnodes
        {
            if let Some(attrnode) = node.get_property_node(idattr_name)
            {
                let docptr  = self.doc_ptr();
                let attrptr = attrnode.node_ptr() as *mut bindings::_xmlAttr;

                let id     = attrnode.get_content();
                let cid    = CString::new(id.clone()).unwrap();
                let cidptr = cid.as_ptr() as *mut c_uchar;

                let existing = unsafe { bindings::xmlGetID(docptr, cidptr) };

                if existing.is_null() {
                    unsafe { bindings::xmlAddID(null_mut(), docptr, cidptr, attrptr) };
                } else if existing != attrptr {
                    return Err(format!("Error: duplicate ID attribute: {}", id).into());
                }
            }
        }

        Ok(())
    }
}