001 /**
002 * GRANITE DATA SERVICES
003 * Copyright (C) 2006-2013 GRANITE DATA SERVICES S.A.S.
004 *
005 * This file is part of the Granite Data Services Platform.
006 *
007 * ***
008 *
009 * Community License: GPL 3.0
010 *
011 * This file is free software: you can redistribute it and/or modify
012 * it under the terms of the GNU General Public License as published
013 * by the Free Software Foundation, either version 3 of the License,
014 * or (at your option) any later version.
015 *
016 * This file is distributed in the hope that it will be useful, but
017 * WITHOUT ANY WARRANTY; without even the implied warranty of
018 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
019 * GNU General Public License for more details.
020 *
021 * You should have received a copy of the GNU General Public License
022 * along with this program. If not, see <http://www.gnu.org/licenses/>.
023 *
024 * ***
025 *
026 * Available Commercial License: GraniteDS SLA 1.0
027 *
028 * This is the appropriate option if you are creating proprietary
029 * applications and you are not prepared to distribute and share the
030 * source code of your application under the GPL v3 license.
031 *
032 * Please visit http://www.granitedataservices.com/license for more
033 * details.
034 */
035 package org.granite.gravity.udp;
036
037 import java.io.IOException;
038 import java.net.InetSocketAddress;
039 import java.net.SocketAddress;
040 import java.nio.ByteBuffer;
041 import java.nio.channels.DatagramChannel;
042
043 import org.granite.gravity.AbstractChannel;
044
045 /**
046 * @author Franck WOLFF
047 */
048 public class UdpNIOChannel extends AbstractUdpChannel {
049
050 private DatagramChannel channel = null;
051
052 public UdpNIOChannel(UdpChannelFactory channelFactory, AbstractChannel gravityChannel, DatagramChannel channel) {
053 this(channelFactory, gravityChannel, channel, null);
054 }
055
056 public UdpNIOChannel(UdpChannelFactory channelFactory, AbstractChannel gravityChannel, DatagramChannel channel, InetSocketAddress address) {
057 super(channelFactory, gravityChannel, address);
058
059 if ((address == null && !channel.isConnected()) || (address != null && channel.isConnected()))
060 throw new IllegalArgumentException("Inconsistent arguments");
061
062 this.channel = channel;
063 }
064
065 public SocketAddress getClientAddress() {
066 return (channel.isConnected() ? channel.socket().getRemoteSocketAddress() : address);
067 }
068
069 @Override
070 public int getServerPort() {
071 return (channel.isConnected() ? channel.socket().getLocalPort() : address.getPort());
072 }
073
074 public int write(byte[] data, int offset, int length) throws IOException {
075 int size = length - offset;
076
077 if (size > MAX_PACKET_SIZE)
078 throw new UdpPacketTooLargeException("UDP packet size cannot exceed 64K: " + size);
079
080 ByteBuffer buf = ByteBuffer.wrap(data, offset, length);
081 return (channel.isConnected() ? channel.write(buf) : channel.send(buf, address));
082 }
083
084 public void close() {
085 super.close();
086
087 try {
088 if (channel.isConnected()) {
089 try {
090 channel.close();
091 }
092 catch (IOException e) {
093 throw new RuntimeException(e);
094 }
095 }
096 }
097 finally {
098 channel = null;
099 }
100 }
101 }