blob: 71c1af5d39606c987714bc5b9a2a978ebc779b2c (
plain) (
blame)
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
|
// Copyright Vespa.ai. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package com.yahoo.jrt;
import com.yahoo.security.tls.Capability;
import com.yahoo.security.tls.CapabilitySet;
import com.yahoo.security.tls.MissingCapabilitiesException;
/**
* @author bjorncs
*/
public class RequireCapabilitiesFilter implements RequestAccessFilter {
private static final RequireCapabilitiesFilter UNCLASSIFIED =
new RequireCapabilitiesFilter(Capability.RPC_UNCLASSIFIED);
private final CapabilitySet requiredCapabilities;
public RequireCapabilitiesFilter(CapabilitySet requiredCapabilities) {
this.requiredCapabilities = requiredCapabilities;
}
public RequireCapabilitiesFilter(Capability... requiredCapabilities) {
this(CapabilitySet.of(requiredCapabilities));
}
public static RequireCapabilitiesFilter unclassified() { return UNCLASSIFIED; }
@Override
public boolean allow(Request r) {
try {
r.target().connectionAuthContext()
.verifyCapabilities(requiredCapabilities, "RPC", r.methodName(), r.target().peerSpec().toString());
return true;
} catch (MissingCapabilitiesException e) {
return false;
}
}
}
|