org.owasp.html.AttributePolicy Maven / Gradle / Ivy
// Copyright (c) 2011, Mike Samuel
// All rights reserved.
//
// Redistribution and use in source and binary forms, with or without
// modification, are permitted provided that the following conditions
// are met:
//
// Redistributions of source code must retain the above copyright
// notice, this list of conditions and the following disclaimer.
// Redistributions in binary form must reproduce the above copyright
// notice, this list of conditions and the following disclaimer in the
// documentation and/or other materials provided with the distribution.
// Neither the name of the OWASP nor the names of its contributors may
// be used to endorse or promote products derived from this software
// without specific prior written permission.
// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
// "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
// LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
// FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
// COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
// INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
// BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
// LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
// CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
// LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN
// ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
// POSSIBILITY OF SUCH DAMAGE.
package org.owasp.html;
import com.google.common.collect.ImmutableList;
import java.util.Collection;
import java.util.Set;
import java.util.LinkedHashSet;
import javax.annotation.CheckReturnValue;
import javax.annotation.Nullable;
import javax.annotation.concurrent.Immutable;
/**
* A policy that can be applied to an HTML attribute to decide whether or not to
* allow it in the output, possibly after transforming its value.
*
* @author Mike Samuel
* @see HtmlPolicyBuilder.AttributeBuilder#matching(AttributePolicy)
*/
@TCB public interface AttributePolicy {
/**
* @param elementName the lower-case element name.
* @param attributeName the lower-case attribute name.
* @param value the attribute value without quotes and with HTML entities
* decoded.
*
* @return {@code null} to disallow the attribute or the adjusted value if
* allowed.
*/
public @Nullable String apply(
String elementName, String attributeName, String value);
/** Utilities for working with attribute policies. */
public static final class Util {
/**
* An attribute policy equivalent to applying all the given policies in
* order, failing early if any of them fails.
*/
@CheckReturnValue
public static final AttributePolicy join(AttributePolicy... policies) {
Set uniq = new LinkedHashSet();
for (AttributePolicy p : policies) {
if (p instanceof JoinedAttributePolicy) {
uniq.addAll(((JoinedAttributePolicy) p).policies);
} else if (p != null) {
uniq.add(p);
}
}
if (uniq.contains(REJECT_ALL_ATTRIBUTE_POLICY)) {
return REJECT_ALL_ATTRIBUTE_POLICY;
}
uniq.remove(IDENTITY_ATTRIBUTE_POLICY);
switch (uniq.size()) {
case 0: return IDENTITY_ATTRIBUTE_POLICY;
case 1: return uniq.iterator().next();
default: return new JoinedAttributePolicy(uniq);
}
}
}
public static final AttributePolicy IDENTITY_ATTRIBUTE_POLICY
= new AttributePolicy() {
public String apply(
String elementName, String attributeName, String value) {
return value;
}
};
public static final AttributePolicy REJECT_ALL_ATTRIBUTE_POLICY
= new AttributePolicy() {
public @Nullable String apply(
String elementName, String attributeName, String value) {
return null;
}
};
}
@Immutable
final class JoinedAttributePolicy implements AttributePolicy {
final ImmutableList policies;
JoinedAttributePolicy(Collection policies) {
this.policies = ImmutableList.copyOf(policies);
}
public @Nullable String apply(
String elementName, String attributeName, @Nullable String value) {
for (AttributePolicy p : policies) {
if (value == null) { break; }
value = p.apply(elementName, attributeName, value);
}
return value;
}
@Override
public boolean equals(Object o) {
return o != null && this.getClass() == o.getClass()
&& policies.equals(((JoinedAttributePolicy) o).policies);
}
@Override
public int hashCode() {
return policies.hashCode();
}
}