diff options
Diffstat (limited to 'jdisc-security-filters')
3 files changed, 113 insertions, 0 deletions
diff --git a/jdisc-security-filters/src/main/java/com/yahoo/jdisc/http/filter/security/misc/LocalhostFilter.java b/jdisc-security-filters/src/main/java/com/yahoo/jdisc/http/filter/security/misc/LocalhostFilter.java new file mode 100644 index 00000000000..1623128fac2 --- /dev/null +++ b/jdisc-security-filters/src/main/java/com/yahoo/jdisc/http/filter/security/misc/LocalhostFilter.java @@ -0,0 +1,33 @@ +// Copyright 2018 Yahoo Holdings. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root. +package com.yahoo.jdisc.http.filter.security.misc; + +import com.google.common.net.InetAddresses; +import com.yahoo.jdisc.Response; +import com.yahoo.jdisc.http.filter.DiscFilterRequest; +import com.yahoo.jdisc.http.filter.security.base.JsonSecurityRequestFilterBase; + +import java.net.InetAddress; +import java.util.Optional; + +/** + * A security filter that only allows self-originating requests. + * + * @author mpolden + * @author bjorncs + */ +@SuppressWarnings("unused") // Injected +public class LocalhostFilter extends JsonSecurityRequestFilterBase { + + @Override + protected Optional<ErrorResponse> filter(DiscFilterRequest request) { + InetAddress remoteAddr = InetAddresses.forString(request.getRemoteAddr()); + if (!remoteAddr.isLoopbackAddress() && !request.getRemoteAddr().equals(request.getLocalAddr())) { + return Optional.of(new ErrorResponse( + Response.Status.UNAUTHORIZED, + String.format("%s %s denied for %s: Unauthorized host", request.getMethod(), + request.getUri().getPath(), request.getRemoteAddr()))); + } + return Optional.empty(); + } + +} diff --git a/jdisc-security-filters/src/main/java/com/yahoo/jdisc/http/filter/security/misc/NoopFilter.java b/jdisc-security-filters/src/main/java/com/yahoo/jdisc/http/filter/security/misc/NoopFilter.java new file mode 100644 index 00000000000..cb1130e8825 --- /dev/null +++ b/jdisc-security-filters/src/main/java/com/yahoo/jdisc/http/filter/security/misc/NoopFilter.java @@ -0,0 +1,20 @@ +// Copyright 2018 Yahoo Holdings. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root. +package com.yahoo.jdisc.http.filter.security.misc; + +import com.yahoo.jdisc.handler.ResponseHandler; +import com.yahoo.jdisc.http.filter.DiscFilterRequest; +import com.yahoo.jdisc.http.filter.SecurityRequestFilter; + +/** + * A no-op filter. Used for bindings that are whitelisted and do not require any authorization. + * + * @author mpolden + */ +@SuppressWarnings("unused") // Injected +public class NoopFilter implements SecurityRequestFilter { + + @Override + public void filter(DiscFilterRequest request, ResponseHandler handler) { + } + +} diff --git a/jdisc-security-filters/src/test/java/com/yahoo/jdisc/http/filter/security/misc/LocalhostFilterTest.java b/jdisc-security-filters/src/test/java/com/yahoo/jdisc/http/filter/security/misc/LocalhostFilterTest.java new file mode 100644 index 00000000000..39c3783caec --- /dev/null +++ b/jdisc-security-filters/src/test/java/com/yahoo/jdisc/http/filter/security/misc/LocalhostFilterTest.java @@ -0,0 +1,60 @@ +// Copyright 2018 Yahoo Holdings. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root. +package com.yahoo.jdisc.http.filter.security.misc; + +import com.yahoo.container.jdisc.RequestHandlerTestDriver; +import com.yahoo.jdisc.Response; +import com.yahoo.jdisc.http.filter.DiscFilterRequest; +import org.junit.Test; +import org.mockito.Mockito; + +import java.net.URI; + +import static org.junit.Assert.assertEquals; +import static org.junit.Assert.assertNull; +import static org.mockito.Mockito.when; + +/** + * @author mpolden + * @author bjorncs + */ +public class LocalhostFilterTest { + + @Test + public void filter() { + // Reject from non-loopback + assertUnauthorized(createRequest("1.2.3.4", null)); + + // Allow requests from loopback addresses + assertSuccess(createRequest("127.0.0.1", null)); + assertSuccess(createRequest("127.127.0.1", null)); + assertSuccess(createRequest("0:0:0:0:0:0:0:1", null)); + + // Allow requests originating from self + assertSuccess(createRequest("1.3.3.7", "1.3.3.7")); + } + + private static DiscFilterRequest createRequest(String remoteAddr, String localAddr) { + DiscFilterRequest request = Mockito.mock(DiscFilterRequest.class); + when(request.getRemoteAddr()).thenReturn(remoteAddr); + when(request.getLocalAddr()).thenReturn(localAddr); + when(request.getMethod()).thenReturn("GET"); + when(request.getUri()).thenReturn(URI.create("http://localhost:8080/")); + return request; + } + + private static void assertUnauthorized(DiscFilterRequest request) { + LocalhostFilter filter = new LocalhostFilter(); + RequestHandlerTestDriver.MockResponseHandler handler = new RequestHandlerTestDriver.MockResponseHandler(); + filter.filter(request, handler); + assertEquals(Response.Status.UNAUTHORIZED, handler.getStatus()); + } + + + private static void assertSuccess(DiscFilterRequest request) { + LocalhostFilter filter = new LocalhostFilter(); + RequestHandlerTestDriver.MockResponseHandler handler = new RequestHandlerTestDriver.MockResponseHandler(); + filter.filter(request, handler); + assertNull(handler.getResponse()); + } + +} |