1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17 package edu.internet2.middleware.shibboleth.common.attribute.encoding.provider;
18
19 import java.util.List;
20
21 import org.opensaml.Configuration;
22 import org.opensaml.saml2.core.Attribute;
23 import org.opensaml.saml2.core.AttributeValue;
24 import org.opensaml.xml.XMLObject;
25 import org.opensaml.xml.XMLObjectBuilder;
26 import org.opensaml.xml.schema.XSString;
27 import org.opensaml.xml.schema.impl.XSStringBuilder;
28 import org.opensaml.xml.util.Base64;
29 import org.slf4j.Logger;
30 import org.slf4j.LoggerFactory;
31
32 import edu.internet2.middleware.shibboleth.common.attribute.BaseAttribute;
33
34
35
36
37
38 public class SAML2Base64AttributeEncoder extends AbstractSAML2AttributeEncoder {
39
40
41 private final Logger log = LoggerFactory.getLogger(SAML2Base64AttributeEncoder.class);
42
43
44 private final XMLObjectBuilder<XSString> stringBuilder;
45
46
47 public SAML2Base64AttributeEncoder() {
48 super();
49 stringBuilder = (XSStringBuilder) Configuration.getBuilderFactory().getBuilder(XSString.TYPE_NAME);
50 }
51
52
53 public Attribute encode(BaseAttribute attribute) {
54 Attribute samlAttribute = attributeBuilder.buildObject();
55 populateAttribute(samlAttribute);
56
57 byte[] attributeValue;
58 XSString samlAttributeValue;
59 for (Object o : attribute.getValues()) {
60 if (o == null || !(o instanceof byte[])) {
61 log.debug("Skipping attribute value because it is either null or not a byte[]");
62 continue;
63 }
64
65 attributeValue = (byte[]) o;
66 samlAttributeValue = stringBuilder.buildObject(AttributeValue.DEFAULT_ELEMENT_NAME, XSString.TYPE_NAME);
67 samlAttributeValue.setValue(Base64.encodeBytes(attributeValue));
68 samlAttribute.getAttributeValues().add(samlAttributeValue);
69 }
70
71 List<XMLObject> attributeValues = samlAttribute.getAttributeValues();
72 if (attributeValues == null || attributeValues.isEmpty()) {
73 log.debug("Unable to encode {} attribute. It does not contain any values", attribute.getId());
74 return null;
75 }
76
77 return samlAttribute;
78 }
79
80 }