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 */
035package org.granite.gravity.udp;
036
037import java.io.IOException;
038import java.net.DatagramPacket;
039import java.net.DatagramSocket;
040import java.net.InetSocketAddress;
041import java.net.SocketAddress;
042
043import org.granite.gravity.AbstractChannel;
044
045/**
046 * @author Franck WOLFF
047 */
048public class UdpIOChannel extends AbstractUdpChannel {
049
050        private DatagramSocket socket = null;
051        
052        public UdpIOChannel(UdpChannelFactory channelFactory, AbstractChannel gravityChannel, DatagramSocket socket) {
053                this(channelFactory, gravityChannel, socket, null);
054        }
055        
056        public UdpIOChannel(UdpChannelFactory channelFactory, AbstractChannel gravityChannel, DatagramSocket socket, InetSocketAddress address) {
057                super(channelFactory, gravityChannel, address);
058                
059                if ((address == null && !socket.isConnected()) || (address != null && socket.isConnected()))
060                        throw new IllegalArgumentException("Inconsistent arguments");
061
062                this.socket = socket;
063        }
064
065        public SocketAddress getClientAddress() {
066                return (socket.isConnected() ? socket.getRemoteSocketAddress() : address);
067        }
068
069        @Override
070        public int getServerPort() {
071                return (socket.isConnected() ? 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                DatagramPacket packet = (
081                        socket.isConnected() ?
082                        new DatagramPacket(data, offset, length) :
083                        new DatagramPacket(data, offset, length, address)
084                );
085                
086                socket.send(packet);
087                return size;
088        }
089
090        public void close() {
091                super.close();
092                
093                try {
094                        if (socket.isConnected())
095                                socket.close();
096                }
097                finally {
098                        socket = null;
099                }
100        }
101}