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.io.OutputStream;
039
040/**
041 * @author Franck WOLFF
042 */
043public class UdpOutputStream extends OutputStream {
044
045        private static final int MAX_SIZE = 64 * 1024; // 64K.
046        
047        private byte[] buffer = new byte[512];
048        private int size = 0;
049        
050        public UdpOutputStream() {
051        }
052
053        @Override
054        public void write(int b) throws IOException {
055                if (size >= buffer.length) {
056                        if (size >= MAX_SIZE)
057                                throw new UdpPacketTooLargeException("Max capacity of 64K reached");
058
059                        byte[] newBuffer = new byte[buffer.length << 1];
060                        System.arraycopy(buffer, 0, newBuffer, 0, buffer.length);
061                        buffer = newBuffer;
062                }
063                buffer[size++] = (byte)b;
064        }
065
066        @Override
067        public void write(byte[] b, int off, int len) throws IOException {
068                if (off < 0 || len < 0 || off + len > b.length)
069                        throw new IndexOutOfBoundsException();
070                
071                if (len == 0)
072                        return;
073                
074                int newSize = len + size;
075                if (newSize > MAX_SIZE)
076                        throw new UdpPacketTooLargeException("Max capacity of 64K reached");
077                
078                if (newSize > buffer.length) {
079                        int newCapacity = buffer.length << 1;
080                        while (newCapacity < newSize)
081                                newCapacity <<= 1;
082                        
083                        byte[] newBuffer = new byte[newCapacity];
084                        System.arraycopy(buffer, 0, newBuffer, 0, buffer.length);
085                        buffer = newBuffer;
086                }
087
088                System.arraycopy(b, off, buffer, size, len);
089                size = newSize;
090        }
091        
092        public byte[] buffer() {
093                return buffer;
094        }
095        
096        public int size() {
097                return size;
098        }
099
100        @Override
101        public void flush() {
102        }
103
104        @Override
105        public void close() {
106        }
107}