org.eclipse.jetty.ee8.nested.InetAccessSet Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of jetty-ee8-nested Show documentation
The jetty core handler adapted to legacy ee8 handler.
The newest version!
//
// ========================================================================
// Copyright (c) 1995 Mort Bay Consulting Pty Ltd and others.
//
// This program and the accompanying materials are made available under the
// terms of the Eclipse Public License v. 2.0 which is available at
// https://www.eclipse.org/legal/epl-2.0, or the Apache License, Version 2.0
// which is available at https://www.apache.org/licenses/LICENSE-2.0.
//
// SPDX-License-Identifier: EPL-2.0 OR Apache-2.0
// ========================================================================
//
package org.eclipse.jetty.ee8.nested;

import java.net.InetAddress;
import java.util.AbstractSet;
import java.util.ArrayList;
import java.util.Iterator;
import java.util.Set;
import java.util.function.Predicate;
import org.eclipse.jetty.http.pathmap.PathSpec;
import org.eclipse.jetty.http.pathmap.ServletPathSpec;
import org.eclipse.jetty.util.InetAddressPattern;
import org.eclipse.jetty.util.StringUtil;

public class InetAccessSet extends AbstractSet implements Set, Predicate {

    private ArrayList tuples = new ArrayList<>();

    @Override
    public boolean add(PatternTuple storageTuple) {
        return tuples.add(storageTuple);
    }

    @Override
    public boolean remove(Object o) {
        return tuples.remove(o);
    }

    @Override
    public Iterator iterator() {
        return tuples.iterator();
    }

    @Override
    public int size() {
        return tuples.size();
    }

    @Override
    public boolean test(AccessTuple entry) {
        if (entry == null)
            return false;
        for (PatternTuple tuple : tuples) {
            if (tuple.test(entry))
                return true;
        }
        return false;
    }

    static class PatternTuple implements Predicate {

        private final String connector;

        private final InetAddressPattern address;

        private final PathSpec pathSpec;

        public static PatternTuple from(String pattern) {
            String path = null;
            int pathIndex = pattern.indexOf('|');
            if (pathIndex >= 0)
                path = pattern.substring(pathIndex + 1);
            String connector = null;
            int connectorIndex = pattern.indexOf('@');
            if (connectorIndex >= 0)
                connector = pattern.substring(0, connectorIndex);
            String addr = null;
            int addrStart = (connectorIndex < 0) ? 0 : connectorIndex + 1;
            int addrEnd = (pathIndex < 0) ? pattern.length() : pathIndex;
            if (addrStart != addrEnd)
                addr = pattern.substring(addrStart, addrEnd);
            return new PatternTuple(connector, InetAddressPattern.from(addr), StringUtil.isEmpty(path) ? null : new ServletPathSpec(path));
        }

        public PatternTuple(String connector, InetAddressPattern address, PathSpec pathSpec) {
            this.connector = connector;
            this.address = address;
            this.pathSpec = pathSpec;
        }

        @Override
        public boolean test(AccessTuple entry) {
            // Match for connector.
            if ((connector != null) && !connector.equals(entry.getConnector()))
                return false;
            // If we have a path we must must be at this path to match for an address.
            if ((pathSpec != null) && !pathSpec.matches(entry.getPath()))
                return false;
            // Match for InetAddress.
            if ((address != null) && !address.test(entry.getAddress()))
                return false;
            return true;
        }
    }

    static class AccessTuple {

        private final String connector;

        private final InetAddress address;

        private final String path;

        public AccessTuple(String connector, InetAddress address, String path) {
            this.connector = connector;
            this.address = address;
            this.path = path;
        }

        public String getConnector() {
            return connector;
        }

        public InetAddress getAddress() {
            return address;
        }

        public String getPath() {
            return path;
        }
    }
}