sjero.net Git - linphone/blob - p2pproxy/dependencies-src/jxse-src-2.5/api/src/net/jxta/socket/JxtaSocketOutputStream.java

   1 /*
   2  * Copyright (c) 2001-2007 Sun Microsystems, Inc.  All rights reserved.
   3  *
   4  *  The Sun Project JXTA(TM) Software License
   5  *
   6  *  Redistribution and use in source and binary forms, with or without
   7  *  modification, are permitted provided that the following conditions are met:
   8  *
   9  *  1. Redistributions of source code must retain the above copyright notice,
  10  *     this list of conditions and the following disclaimer.
  11  *
  12  *  2. Redistributions in binary form must reproduce the above copyright notice,
  13  *     this list of conditions and the following disclaimer in the documentation
  14  *     and/or other materials provided with the distribution.
  15  *
  16  *  3. The end-user documentation included with the redistribution, if any, must
  17  *     include the following acknowledgment: "This product includes software
  18  *     developed by Sun Microsystems, Inc. for JXTA(TM) technology."
  19  *     Alternately, this acknowledgment may appear in the software itself, if
  20  *     and wherever such third-party acknowledgments normally appear.
  21  *
  22  *  4. The names "Sun", "Sun Microsystems, Inc.", "JXTA" and "Project JXTA" must
  23  *     not be used to endorse or promote products derived from this software
  24  *     without prior written permission. For written permission, please contact
  25  *     Project JXTA at http://www.jxta.org.
  26  *
  27  *  5. Products derived from this software may not be called "JXTA", nor may
  28  *     "JXTA" appear in their name, without prior written permission of Sun.
  29  *
  30  *  THIS SOFTWARE IS PROVIDED ``AS IS'' AND ANY EXPRESSED OR IMPLIED WARRANTIES,
  31  *  INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND
  32  *  FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL SUN
  33  *  MICROSYSTEMS OR ITS CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
  34  *  INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
  35  *  LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA,
  36  *  OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
  37  *  LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
  38  *  NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE,
  39  *  EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  40  *
  41  *  JXTA is a registered trademark of Sun Microsystems, Inc. in the United
  42  *  States and other countries.
  43  *
  44  *  Please see the license information page at :
  45  *  <http://www.jxta.org/project/www/license.html> for instructions on use of
  46  *  the license in source files.
  47  *
  48  *  ====================================================================
  49  *
  50  *  This software consists of voluntary contributions made by many individuals
  51  *  on behalf of Project JXTA. For more information on Project JXTA, please see
  52  *  http://www.jxta.org.
  53  *
  54  *  This license is based on the BSD license adopted by the Apache Foundation.
  55  */
  56 package net.jxta.socket;
  57
  58
  59 import java.io.IOException;
  60 import java.io.OutputStream;
  61 import java.net.SocketException;
  62
  63
  64 /**
  65  * This class implements a buffered output stream. By setting up such an output
  66  * stream, an application can write bytes to the underlying output stream
  67  * without necessarily causing a call to the underlying system for each byte
  68  * written. Data buffer is flushed to the underlying stream, when it is full,
  69  * or an explicit call to flush is made.
  70  */
  71 class JxtaSocketOutputStream extends OutputStream {
  72
  73     /**
  74      * If {@code true} then this socket is closed.
  75      */
  76     protected boolean closed = false;
  77
  78     /**
  79      * Data buffer
  80      */
  81     protected byte buf[];
  82
  83     /**
  84      * byte count in buffer
  85      */
  86     protected int count;
  87
  88     /**
  89      * JxtaSocket associated with this stream
  90      */
  91     protected JxtaSocket socket;
  92
  93     /**
  94      * Constructor for the JxtaSocketOutputStream object
  95      *
  96      * @param socket JxtaSocket associated with this stream
  97      * @param size   buffer size in bytes
  98      */
  99     public JxtaSocketOutputStream(JxtaSocket socket, int size) {
 100         if (size <= 0) {
 101             throw new IllegalArgumentException("Buffer size <= 0");
 102         }
 103         buf = new byte[size];
 104         this.socket = socket;
 105     }
 106
 107     /**
 108      * {@inheritDoc}
 109      */
 110     @Override
 111     public synchronized void close() throws IOException {
 112         flushBuffer();
 113         closed = true;
 114     }
 115
 116     /**
 117      * Similar to close except that any buffered data is discarded.
 118      */
 119     synchronized void hardClose() {
 120         count = 0;
 121         closed = true;
 122     }
 123
 124     /**
 125      * Flush the internal buffer
 126      *
 127      * @throws IOException if an i/o error occurs
 128      */
 129     private void flushBuffer() throws IOException {
 130         if (count > 0) {
 131             // send the message
 132             socket.write(buf, 0, count);
 133             count = 0;
 134         }
 135     }
 136
 137     /**
 138      * {@inheritDoc}
 139      */
 140     @Override
 141     public synchronized void write(int b) throws IOException {
 142
 143         if (closed) {
 144             throw new SocketException("Socket Closed.");
 145         }
 146
 147         if (count >= buf.length) {
 148             flushBuffer();
 149         }
 150         buf[count++] = (byte) b;
 151     }
 152
 153     /**
 154      * {@inheritDoc}
 155      */
 156     @Override
 157     public synchronized void write(byte b[], int off, int len) throws IOException {
 158         int left = buf.length - count;
 159
 160         if (closed) {
 161             throw new SocketException("Socket Closed.");
 162         }
 163
 164         if (len > left) {
 165             System.arraycopy(b, off, buf, count, left);
 166             len -= left;
 167             off += left;
 168             count += left;
 169             flushBuffer();
 170         }
 171
 172         // chunk data if larger than buf.length
 173         while (len >= buf.length) {
 174             socket.write(b, off, buf.length);
 175             len -= buf.length;
 176             off += buf.length;
 177         }
 178         System.arraycopy(b, off, buf, count, len);
 179         count += len;
 180     }
 181
 182     /**
 183      * {@inheritDoc}
 184      */
 185     @Override
 186     public synchronized void flush() throws IOException {
 187         flushBuffer();
 188     }
 189 }
 190