1 /*
2 * Copyright 2014 The Netty Project
3 *
4 * The Netty Project licenses this file to you under the Apache License,
5 * version 2.0 (the "License"); you may not use this file except in compliance
6 * with the License. You may obtain a copy of the License at:
7 *
8 * https://www.apache.org/licenses/LICENSE-2.0
9 *
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
12 * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
13 * License for the specific language governing permissions and limitations
14 * under the License.
15 */
16 package io.netty5.handler.ipfilter;
17
18 import io.netty5.channel.Channel;
19 import io.netty5.channel.ChannelHandlerContext;
20
21 import java.net.InetAddress;
22 import java.net.InetSocketAddress;
23 import java.util.Set;
24 import java.util.concurrent.ConcurrentHashMap;
25
26 /**
27 * This class allows one to ensure that at all times for every IP address there is at most one
28 * {@link Channel} connected to the server.
29 */
30 public class UniqueIpFilter extends AbstractRemoteAddressFilter<InetSocketAddress> {
31
32 private final Set<InetAddress> connected = ConcurrentHashMap.newKeySet();
33
34 @Override
35 public boolean isSharable() {
36 return true;
37 }
38
39 @Override
40 protected boolean accept(ChannelHandlerContext ctx, InetSocketAddress remoteAddress) {
41 final InetAddress remoteIp = remoteAddress.getAddress();
42 if (!connected.add(remoteIp)) {
43 return false;
44 } else {
45 ctx.channel().closeFuture().addListener(future -> connected.remove(remoteIp));
46 return true;
47 }
48 }
49 }