2009-06-11 02:48:49 +02:00
|
|
|
/*
|
2011-12-09 06:18:34 +01:00
|
|
|
* Copyright 2011 The Netty Project
|
2009-06-11 02:48:49 +02:00
|
|
|
*
|
2011-12-09 06:18:34 +01:00
|
|
|
* The Netty Project licenses this file to you under the Apache License,
|
|
|
|
* version 2.0 (the "License"); you may not use this file except in compliance
|
|
|
|
* with the License. You may obtain a copy of the License at:
|
2009-06-11 02:48:49 +02:00
|
|
|
*
|
2011-12-09 06:18:34 +01:00
|
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
2009-06-11 02:48:49 +02:00
|
|
|
*
|
2009-08-28 09:15:49 +02:00
|
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
|
|
* distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
2011-12-09 06:18:34 +01:00
|
|
|
* WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
2009-08-28 09:15:49 +02:00
|
|
|
* License for the specific language governing permissions and limitations
|
|
|
|
* under the License.
|
2009-06-11 02:48:49 +02:00
|
|
|
*/
|
2011-12-09 04:38:59 +01:00
|
|
|
package io.netty.channel.socket.nio;
|
2009-06-11 02:48:49 +02:00
|
|
|
|
2012-05-24 17:57:10 +02:00
|
|
|
import io.netty.buffer.ChannelBuffers;
|
|
|
|
import io.netty.channel.ChannelBufferHolder;
|
|
|
|
import io.netty.channel.ChannelBufferHolders;
|
2012-02-18 23:02:56 +01:00
|
|
|
import io.netty.channel.ChannelException;
|
2012-03-20 09:43:00 +01:00
|
|
|
import io.netty.channel.ChannelFuture;
|
2012-02-18 23:02:56 +01:00
|
|
|
import io.netty.channel.socket.DatagramChannelConfig;
|
2012-05-24 17:57:10 +02:00
|
|
|
import io.netty.channel.socket.DatagramPacket;
|
|
|
|
import io.netty.logging.InternalLogger;
|
|
|
|
import io.netty.logging.InternalLoggerFactory;
|
2012-04-02 11:07:11 +02:00
|
|
|
import io.netty.util.internal.DetectionUtil;
|
2009-06-11 02:48:49 +02:00
|
|
|
|
|
|
|
import java.io.IOException;
|
2009-06-11 08:10:46 +02:00
|
|
|
import java.net.InetAddress;
|
2009-06-11 02:48:49 +02:00
|
|
|
import java.net.InetSocketAddress;
|
2009-06-11 08:10:46 +02:00
|
|
|
import java.net.NetworkInterface;
|
2012-03-20 09:43:00 +01:00
|
|
|
import java.net.SocketAddress;
|
2012-04-02 11:07:11 +02:00
|
|
|
import java.net.SocketException;
|
2012-05-24 17:57:10 +02:00
|
|
|
import java.nio.ByteBuffer;
|
2009-06-11 02:48:49 +02:00
|
|
|
import java.nio.channels.DatagramChannel;
|
2012-04-02 11:07:11 +02:00
|
|
|
import java.nio.channels.MembershipKey;
|
2012-05-24 17:57:10 +02:00
|
|
|
import java.nio.channels.SelectionKey;
|
2012-04-02 11:07:11 +02:00
|
|
|
import java.util.ArrayList;
|
|
|
|
import java.util.HashMap;
|
|
|
|
import java.util.Iterator;
|
|
|
|
import java.util.List;
|
|
|
|
import java.util.Map;
|
2012-05-24 17:57:10 +02:00
|
|
|
import java.util.Queue;
|
2009-06-11 02:48:49 +02:00
|
|
|
|
|
|
|
/**
|
2011-12-09 04:38:59 +01:00
|
|
|
* Provides an NIO based {@link io.netty.channel.socket.DatagramChannel}.
|
2009-06-11 02:48:49 +02:00
|
|
|
*/
|
2012-03-29 17:07:19 +02:00
|
|
|
public final class NioDatagramChannel extends AbstractNioChannel implements io.netty.channel.socket.DatagramChannel {
|
2009-06-11 02:48:49 +02:00
|
|
|
|
2012-05-24 17:57:10 +02:00
|
|
|
private static final InternalLogger logger = InternalLoggerFactory.getInstance(NioDatagramChannel.class);
|
|
|
|
|
|
|
|
private final DatagramChannelConfig config;
|
|
|
|
private final Map<InetAddress, List<MembershipKey>> memberships =
|
|
|
|
new HashMap<InetAddress, List<MembershipKey>>();
|
|
|
|
private final ChannelBufferHolder<Object> out = ChannelBufferHolders.messageBuffer();
|
|
|
|
|
|
|
|
private static DatagramChannel newSocket() {
|
|
|
|
try {
|
|
|
|
return DatagramChannel.open();
|
|
|
|
} catch (IOException e) {
|
|
|
|
throw new ChannelException("Failed to open a socket.", e);
|
|
|
|
}
|
2012-04-03 12:04:33 +02:00
|
|
|
}
|
2012-05-24 17:57:10 +02:00
|
|
|
|
|
|
|
public NioDatagramChannel() {
|
|
|
|
this(newSocket());
|
2011-11-02 19:17:10 +01:00
|
|
|
}
|
|
|
|
|
2012-05-24 17:57:10 +02:00
|
|
|
public NioDatagramChannel(DatagramChannel socket) {
|
|
|
|
this(null, socket);
|
2009-06-11 02:48:49 +02:00
|
|
|
}
|
|
|
|
|
2012-05-24 17:57:10 +02:00
|
|
|
public NioDatagramChannel(Integer id, DatagramChannel socket) {
|
|
|
|
super(null, id, socket);
|
2009-06-11 02:48:49 +02:00
|
|
|
try {
|
2012-05-24 17:57:10 +02:00
|
|
|
socket.configureBlocking(false);
|
|
|
|
} catch (IOException e) {
|
|
|
|
try {
|
|
|
|
socket.close();
|
|
|
|
} catch (IOException e2) {
|
|
|
|
if (logger.isWarnEnabled()) {
|
|
|
|
logger.warn(
|
|
|
|
"Failed to close a partially initialized socket.", e2);
|
2012-04-03 12:04:33 +02:00
|
|
|
}
|
2012-05-24 17:57:10 +02:00
|
|
|
|
2012-04-03 12:04:33 +02:00
|
|
|
}
|
2012-05-24 17:57:10 +02:00
|
|
|
|
|
|
|
throw new ChannelException("Failed to enter non-blocking mode.", e);
|
2009-06-11 02:48:49 +02:00
|
|
|
}
|
2012-05-24 17:57:10 +02:00
|
|
|
|
|
|
|
config = new DefaultNioDatagramChannelConfig(socket);
|
|
|
|
|
2009-06-11 02:48:49 +02:00
|
|
|
}
|
|
|
|
|
2012-05-24 17:57:10 +02:00
|
|
|
@Override
|
|
|
|
public DatagramChannelConfig config() {
|
|
|
|
return config;
|
|
|
|
}
|
2009-06-11 02:48:49 +02:00
|
|
|
|
2012-03-29 17:07:19 +02:00
|
|
|
@Override
|
2012-05-24 17:57:10 +02:00
|
|
|
public boolean isActive() {
|
|
|
|
DatagramChannel ch = javaChannel();
|
|
|
|
return ch.isOpen() && ch.socket().isBound();
|
2012-03-29 17:07:19 +02:00
|
|
|
}
|
|
|
|
|
2012-03-06 19:26:32 +01:00
|
|
|
@Override
|
2012-05-24 17:57:10 +02:00
|
|
|
protected DatagramChannel javaChannel() {
|
|
|
|
return (DatagramChannel) super.javaChannel();
|
2012-03-06 19:26:32 +01:00
|
|
|
}
|
|
|
|
|
2010-11-12 01:45:39 +01:00
|
|
|
@Override
|
2012-05-24 17:57:10 +02:00
|
|
|
protected ChannelBufferHolder<Object> firstOut() {
|
|
|
|
return out;
|
2009-06-11 02:48:49 +02:00
|
|
|
}
|
|
|
|
|
2010-11-12 01:45:39 +01:00
|
|
|
@Override
|
2012-05-24 17:57:10 +02:00
|
|
|
protected SocketAddress localAddress0() {
|
|
|
|
return javaChannel().socket().getLocalSocketAddress();
|
2009-06-11 02:48:49 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
@Override
|
2012-05-24 17:57:10 +02:00
|
|
|
protected SocketAddress remoteAddress0() {
|
|
|
|
return javaChannel().socket().getRemoteSocketAddress();
|
2009-06-11 02:48:49 +02:00
|
|
|
}
|
|
|
|
|
2010-11-12 01:45:39 +01:00
|
|
|
@Override
|
2012-05-24 17:57:10 +02:00
|
|
|
protected void doBind(SocketAddress localAddress) throws Exception {
|
2012-05-24 18:32:14 +02:00
|
|
|
javaChannel().socket().bind(localAddress);
|
2012-05-24 17:57:10 +02:00
|
|
|
selectionKey().interestOps(SelectionKey.OP_READ);
|
|
|
|
}
|
|
|
|
|
|
|
|
@Override
|
|
|
|
protected boolean doConnect(SocketAddress remoteAddress,
|
|
|
|
SocketAddress localAddress) throws Exception {
|
|
|
|
if (localAddress != null) {
|
|
|
|
javaChannel().socket().bind(localAddress);
|
|
|
|
}
|
|
|
|
|
|
|
|
boolean success = false;
|
|
|
|
try {
|
|
|
|
javaChannel().connect(remoteAddress);
|
|
|
|
selectionKey().interestOps(selectionKey().interestOps() | SelectionKey.OP_READ);
|
|
|
|
success = true;
|
|
|
|
return true;
|
|
|
|
} finally {
|
|
|
|
if (!success) {
|
|
|
|
doClose();
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
@Override
|
|
|
|
protected void doFinishConnect() throws Exception {
|
|
|
|
throw new Error();
|
|
|
|
}
|
|
|
|
|
|
|
|
@Override
|
|
|
|
protected void doDisconnect() throws Exception {
|
|
|
|
javaChannel().disconnect();
|
|
|
|
}
|
|
|
|
|
|
|
|
@Override
|
|
|
|
protected void doClose() throws Exception {
|
|
|
|
javaChannel().close();
|
|
|
|
}
|
|
|
|
|
|
|
|
@Override
|
|
|
|
protected void doDeregister() throws Exception {
|
|
|
|
selectionKey().cancel();
|
|
|
|
((SingleThreadSelectorEventLoop) eventLoop()).cancelledKeys ++;
|
|
|
|
}
|
|
|
|
|
|
|
|
@Override
|
|
|
|
protected int doRead(ChannelBufferHolder<Object> buf) throws Exception {
|
|
|
|
DatagramChannel ch = javaChannel();
|
2012-05-24 19:37:51 +02:00
|
|
|
ByteBuffer data = ByteBuffer.allocate(config().getReceivePacketSize());
|
2012-05-24 17:57:10 +02:00
|
|
|
InetSocketAddress remoteAddress = (InetSocketAddress) ch.receive(data);
|
|
|
|
if (remoteAddress == null) {
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
data.flip();
|
|
|
|
buf.messageBuffer().add(new DatagramPacket(ChannelBuffers.wrappedBuffer(data), remoteAddress));
|
|
|
|
return 1;
|
|
|
|
}
|
|
|
|
|
|
|
|
@Override
|
|
|
|
protected int doFlush(boolean lastSpin) throws Exception {
|
|
|
|
final Queue<Object> buf = unsafe().out().messageBuffer();
|
|
|
|
if (buf.isEmpty()) {
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
DatagramPacket packet = (DatagramPacket) buf.peek();
|
|
|
|
final int writtenBytes = javaChannel().send(packet.data().toByteBuffer(), packet.remoteAddress());
|
|
|
|
|
|
|
|
final SelectionKey key = selectionKey();
|
|
|
|
final int interestOps = key.interestOps();
|
|
|
|
if (writtenBytes <= 0) {
|
|
|
|
// Did not write a packet.
|
|
|
|
// 1) If 'lastSpin' is false, the caller will call this method again real soon.
|
|
|
|
// - Do not update OP_WRITE.
|
|
|
|
// 2) If 'lastSpin' is true, the caller will not retry.
|
|
|
|
// - Set OP_WRITE so that the event loop calls flushForcibly() later.
|
|
|
|
if (lastSpin) {
|
|
|
|
if ((interestOps & SelectionKey.OP_WRITE) == 0) {
|
|
|
|
key.interestOps(interestOps | SelectionKey.OP_WRITE);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
// Wrote a packet.
|
|
|
|
buf.remove();
|
|
|
|
if (buf.isEmpty()) {
|
|
|
|
// Wrote the outbound buffer completely - clear OP_WRITE.
|
|
|
|
if ((interestOps & SelectionKey.OP_WRITE) != 0) {
|
|
|
|
key.interestOps(interestOps & ~SelectionKey.OP_WRITE);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return 1;
|
2009-06-11 02:48:49 +02:00
|
|
|
}
|
|
|
|
|
2010-11-12 01:45:39 +01:00
|
|
|
@Override
|
2012-04-02 11:57:32 +02:00
|
|
|
public ChannelFuture joinGroup(InetAddress multicastAddress) {
|
2012-05-24 17:57:10 +02:00
|
|
|
return joinGroup(multicastAddress, newFuture());
|
|
|
|
}
|
|
|
|
|
|
|
|
@Override
|
|
|
|
public ChannelFuture joinGroup(InetAddress multicastAddress, ChannelFuture future) {
|
|
|
|
try {
|
|
|
|
return joinGroup(
|
|
|
|
multicastAddress,
|
|
|
|
NetworkInterface.getByInetAddress(localAddress().getAddress()),
|
|
|
|
null, future);
|
2012-04-02 11:07:11 +02:00
|
|
|
} catch (SocketException e) {
|
2012-05-24 17:57:10 +02:00
|
|
|
future.setFailure(e);
|
2012-04-02 11:07:11 +02:00
|
|
|
}
|
2012-05-24 17:57:10 +02:00
|
|
|
return future;
|
2009-06-11 08:10:46 +02:00
|
|
|
}
|
|
|
|
|
2010-11-12 01:45:39 +01:00
|
|
|
@Override
|
2012-05-24 17:57:10 +02:00
|
|
|
public ChannelFuture joinGroup(
|
|
|
|
InetSocketAddress multicastAddress, NetworkInterface networkInterface) {
|
|
|
|
return joinGroup(multicastAddress, networkInterface, newFuture());
|
2009-06-11 08:10:46 +02:00
|
|
|
}
|
|
|
|
|
2012-05-24 17:57:10 +02:00
|
|
|
@Override
|
|
|
|
public ChannelFuture joinGroup(
|
|
|
|
InetSocketAddress multicastAddress, NetworkInterface networkInterface,
|
|
|
|
ChannelFuture future) {
|
|
|
|
return joinGroup(multicastAddress.getAddress(), networkInterface, null, future);
|
|
|
|
}
|
|
|
|
|
|
|
|
@Override
|
|
|
|
public ChannelFuture joinGroup(
|
|
|
|
InetAddress multicastAddress, NetworkInterface networkInterface, InetAddress source) {
|
|
|
|
return joinGroup(multicastAddress, networkInterface, source, newFuture());
|
|
|
|
}
|
|
|
|
|
|
|
|
@Override
|
|
|
|
public ChannelFuture joinGroup(
|
|
|
|
InetAddress multicastAddress, NetworkInterface networkInterface,
|
|
|
|
InetAddress source, ChannelFuture future) {
|
2012-04-02 11:07:11 +02:00
|
|
|
if (DetectionUtil.javaVersion() < 7) {
|
|
|
|
throw new UnsupportedOperationException();
|
|
|
|
} else {
|
|
|
|
if (multicastAddress == null) {
|
|
|
|
throw new NullPointerException("multicastAddress");
|
|
|
|
}
|
2012-05-24 17:57:10 +02:00
|
|
|
|
2012-04-02 11:07:11 +02:00
|
|
|
if (networkInterface == null) {
|
|
|
|
throw new NullPointerException("networkInterface");
|
|
|
|
}
|
2012-05-24 17:57:10 +02:00
|
|
|
|
2012-04-02 11:07:11 +02:00
|
|
|
try {
|
2012-04-02 15:25:40 +02:00
|
|
|
MembershipKey key;
|
|
|
|
if (source == null) {
|
2012-05-24 17:57:10 +02:00
|
|
|
key = javaChannel().join(multicastAddress, networkInterface);
|
2012-04-02 15:25:40 +02:00
|
|
|
} else {
|
2012-05-24 17:57:10 +02:00
|
|
|
key = javaChannel().join(multicastAddress, networkInterface, source);
|
2012-04-02 15:25:40 +02:00
|
|
|
}
|
|
|
|
|
2012-04-02 11:07:11 +02:00
|
|
|
synchronized (this) {
|
|
|
|
List<MembershipKey> keys = memberships.get(multicastAddress);
|
|
|
|
if (keys == null) {
|
|
|
|
keys = new ArrayList<MembershipKey>();
|
|
|
|
memberships.put(multicastAddress, keys);
|
|
|
|
}
|
|
|
|
keys.add(key);
|
|
|
|
}
|
2012-05-24 17:57:10 +02:00
|
|
|
|
|
|
|
future.setSuccess();
|
2012-04-02 11:57:32 +02:00
|
|
|
} catch (Throwable e) {
|
2012-05-24 17:57:10 +02:00
|
|
|
future.setFailure(e);
|
2012-04-02 11:07:11 +02:00
|
|
|
}
|
|
|
|
}
|
2012-05-24 17:57:10 +02:00
|
|
|
return future;
|
2012-04-02 11:07:11 +02:00
|
|
|
}
|
2012-05-24 17:57:10 +02:00
|
|
|
|
2010-11-12 01:45:39 +01:00
|
|
|
@Override
|
2012-04-02 11:57:32 +02:00
|
|
|
public ChannelFuture leaveGroup(InetAddress multicastAddress) {
|
2012-05-24 17:57:10 +02:00
|
|
|
return leaveGroup(multicastAddress, newFuture());
|
|
|
|
}
|
|
|
|
|
|
|
|
@Override
|
|
|
|
public ChannelFuture leaveGroup(InetAddress multicastAddress, ChannelFuture future) {
|
2012-04-02 11:07:11 +02:00
|
|
|
try {
|
2012-05-24 17:57:10 +02:00
|
|
|
return leaveGroup(multicastAddress, NetworkInterface.getByInetAddress(localAddress().getAddress()), null, future);
|
2012-04-02 11:07:11 +02:00
|
|
|
} catch (SocketException e) {
|
2012-05-24 17:57:10 +02:00
|
|
|
future.setFailure(e);
|
2012-04-02 11:07:11 +02:00
|
|
|
}
|
2012-05-24 17:57:10 +02:00
|
|
|
return future;
|
2009-06-11 08:10:46 +02:00
|
|
|
}
|
|
|
|
|
2010-11-12 01:45:39 +01:00
|
|
|
@Override
|
2012-05-24 17:57:10 +02:00
|
|
|
public ChannelFuture leaveGroup(
|
|
|
|
InetSocketAddress multicastAddress, NetworkInterface networkInterface) {
|
|
|
|
return leaveGroup(multicastAddress, networkInterface, newFuture());
|
2009-06-11 08:10:46 +02:00
|
|
|
}
|
2012-02-18 23:02:56 +01:00
|
|
|
|
2012-05-24 17:57:10 +02:00
|
|
|
@Override
|
|
|
|
public ChannelFuture leaveGroup(
|
|
|
|
InetSocketAddress multicastAddress,
|
|
|
|
NetworkInterface networkInterface, ChannelFuture future) {
|
|
|
|
return leaveGroup(multicastAddress.getAddress(), networkInterface, null, future);
|
|
|
|
}
|
|
|
|
|
|
|
|
@Override
|
|
|
|
public ChannelFuture leaveGroup(
|
|
|
|
InetAddress multicastAddress, NetworkInterface networkInterface, InetAddress source) {
|
|
|
|
return leaveGroup(multicastAddress, networkInterface, source, newFuture());
|
|
|
|
}
|
|
|
|
|
|
|
|
@Override
|
|
|
|
public ChannelFuture leaveGroup(
|
|
|
|
InetAddress multicastAddress, NetworkInterface networkInterface, InetAddress source,
|
|
|
|
ChannelFuture future) {
|
2012-04-02 11:07:11 +02:00
|
|
|
if (DetectionUtil.javaVersion() < 7) {
|
|
|
|
throw new UnsupportedOperationException();
|
2012-05-24 17:57:10 +02:00
|
|
|
}
|
|
|
|
if (multicastAddress == null) {
|
|
|
|
throw new NullPointerException("multicastAddress");
|
|
|
|
}
|
|
|
|
if (networkInterface == null) {
|
|
|
|
throw new NullPointerException("networkInterface");
|
|
|
|
}
|
|
|
|
|
|
|
|
synchronized (this) {
|
|
|
|
if (memberships != null) {
|
|
|
|
List<MembershipKey> keys = memberships.get(multicastAddress);
|
|
|
|
if (keys != null) {
|
|
|
|
Iterator<MembershipKey> keyIt = keys.iterator();
|
|
|
|
|
|
|
|
while (keyIt.hasNext()) {
|
|
|
|
MembershipKey key = keyIt.next();
|
|
|
|
if (networkInterface.equals(key.networkInterface())) {
|
|
|
|
if (source == null && key.sourceAddress() == null || source != null && source.equals(key.sourceAddress())) {
|
|
|
|
key.drop();
|
|
|
|
keyIt.remove();
|
|
|
|
}
|
2012-04-02 11:07:11 +02:00
|
|
|
}
|
|
|
|
}
|
2012-05-24 17:57:10 +02:00
|
|
|
if (keys.isEmpty()) {
|
|
|
|
memberships.remove(multicastAddress);
|
|
|
|
}
|
2012-04-02 11:07:11 +02:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
2012-05-24 17:57:10 +02:00
|
|
|
|
|
|
|
future.setSuccess();
|
|
|
|
return future;
|
|
|
|
}
|
|
|
|
|
|
|
|
/**
|
|
|
|
* Block the given sourceToBlock address for the given multicastAddress on the given networkInterface
|
|
|
|
*/
|
|
|
|
@Override
|
|
|
|
public ChannelFuture block(
|
|
|
|
InetAddress multicastAddress, NetworkInterface networkInterface,
|
|
|
|
InetAddress sourceToBlock) {
|
|
|
|
return block(multicastAddress, networkInterface, sourceToBlock, newFuture());
|
2012-04-02 11:07:11 +02:00
|
|
|
}
|
2012-05-24 17:57:10 +02:00
|
|
|
|
2012-04-02 11:07:11 +02:00
|
|
|
/**
|
|
|
|
* Block the given sourceToBlock address for the given multicastAddress on the given networkInterface
|
|
|
|
*/
|
2012-05-24 17:57:10 +02:00
|
|
|
@Override
|
|
|
|
public ChannelFuture block(
|
|
|
|
InetAddress multicastAddress, NetworkInterface networkInterface,
|
|
|
|
InetAddress sourceToBlock, ChannelFuture future) {
|
2012-04-02 11:07:11 +02:00
|
|
|
if (DetectionUtil.javaVersion() < 7) {
|
|
|
|
throw new UnsupportedOperationException();
|
|
|
|
} else {
|
|
|
|
if (multicastAddress == null) {
|
|
|
|
throw new NullPointerException("multicastAddress");
|
|
|
|
}
|
|
|
|
if (sourceToBlock == null) {
|
|
|
|
throw new NullPointerException("sourceToBlock");
|
|
|
|
}
|
2012-05-24 17:57:10 +02:00
|
|
|
|
2012-04-02 11:07:11 +02:00
|
|
|
if (networkInterface == null) {
|
|
|
|
throw new NullPointerException("networkInterface");
|
|
|
|
}
|
|
|
|
synchronized (this) {
|
|
|
|
if (memberships != null) {
|
|
|
|
List<MembershipKey> keys = memberships.get(multicastAddress);
|
|
|
|
for (MembershipKey key: keys) {
|
|
|
|
if (networkInterface.equals(key.networkInterface())) {
|
|
|
|
try {
|
|
|
|
key.block(sourceToBlock);
|
|
|
|
} catch (IOException e) {
|
2012-05-24 17:57:10 +02:00
|
|
|
future.setFailure(e);
|
2012-04-02 11:07:11 +02:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
2012-05-24 17:57:10 +02:00
|
|
|
future.setSuccess();
|
|
|
|
return future;
|
2012-04-02 11:07:11 +02:00
|
|
|
}
|
|
|
|
}
|
2012-05-24 17:57:10 +02:00
|
|
|
|
2012-04-02 11:07:11 +02:00
|
|
|
/**
|
|
|
|
* Block the given sourceToBlock address for the given multicastAddress
|
2012-05-24 17:57:10 +02:00
|
|
|
*
|
2012-04-02 11:07:11 +02:00
|
|
|
*/
|
2012-05-24 17:57:10 +02:00
|
|
|
@Override
|
2012-04-02 11:57:32 +02:00
|
|
|
public ChannelFuture block(InetAddress multicastAddress, InetAddress sourceToBlock) {
|
2012-05-24 17:57:10 +02:00
|
|
|
return block(multicastAddress, sourceToBlock, newFuture());
|
2012-04-02 11:07:11 +02:00
|
|
|
}
|
2012-05-24 17:57:10 +02:00
|
|
|
|
|
|
|
/**
|
|
|
|
* Block the given sourceToBlock address for the given multicastAddress
|
|
|
|
*
|
|
|
|
*/
|
2012-03-20 09:43:00 +01:00
|
|
|
@Override
|
2012-05-24 17:57:10 +02:00
|
|
|
public ChannelFuture block(
|
|
|
|
InetAddress multicastAddress, InetAddress sourceToBlock, ChannelFuture future) {
|
|
|
|
try {
|
|
|
|
return block(
|
|
|
|
multicastAddress,
|
|
|
|
NetworkInterface.getByInetAddress(localAddress().getAddress()),
|
|
|
|
sourceToBlock, future);
|
|
|
|
} catch (SocketException e) {
|
|
|
|
future.setFailure(e);
|
2012-03-20 09:43:00 +01:00
|
|
|
}
|
2012-05-24 17:57:10 +02:00
|
|
|
return future;
|
2012-03-20 09:43:00 +01:00
|
|
|
}
|
2009-06-11 02:48:49 +02:00
|
|
|
}
|