View Javadoc
1   /*
2    * Copyright (c) 1997, 2012, Oracle and/or its affiliates. All rights reserved.
3    * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
4    *
5    * This code is free software; you can redistribute it and/or modify it
6    * under the terms of the GNU General Public License version 2 only, as
7    * published by the Free Software Foundation.  Oracle designates this
8    * particular file as subject to the "Classpath" exception as provided
9    * by Oracle in the LICENSE file that accompanied this code.
10   *
11   * This code is distributed in the hope that it will be useful, but WITHOUT
12   * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
13   * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
14   * version 2 for more details (a copy is included in the LICENSE file that
15   * accompanied this code).
16   *
17   * You should have received a copy of the GNU General Public License version
18   * 2 along with this work; if not, write to the Free Software Foundation,
19   * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
20   *
21   * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
22   * or visit www.oracle.com if you need additional information or have any
23   * questions.
24   */
25  
26  package com.sun.xml.internal.messaging.saaj.util;
27  
28  import java.io.BufferedOutputStream;
29  import java.io.IOException;
30  import java.io.OutputStream;
31  import java.io.InputStream;
32  import java.io.ByteArrayInputStream;
33  
34  /**
35   * Customized {@link BufferedOutputStream}.
36   *
37   * <p>
38   * Compared to {@link BufferedOutputStream},
39   * this class:
40   *
41   * <ol>
42   * <li>doesn't do synchronization
43   * <li>allows access to the raw buffer
44   * <li>almost no parameter check
45   */
46  public final class ByteOutputStream extends OutputStream {
47      /**
48       * The buffer where data is stored.
49       */
50      protected byte[] buf;
51  
52      /**
53       * The number of valid bytes in the buffer.
54       */
55      protected int count = 0;
56  
57      public ByteOutputStream() {
58          this(1024);
59      }
60  
61      public ByteOutputStream(int size) {
62          buf = new byte[size];
63      }
64  
65      /**
66       * Copies all the bytes from this input into this buffer.
67       */
68      public void write(InputStream in) throws IOException {
69          if (in instanceof ByteArrayInputStream) {
70              int size = in.available();
71              ensureCapacity(size);
72              count += in.read(buf,count,size);
73              return;
74          }
75          while(true) {
76              int cap = buf.length-count;
77              int sz = in.read(buf,count,cap);
78              if(sz<0)    return;     // hit EOS
79  
80              count += sz;
81              if(cap==sz)
82                  // the buffer filled up. double the buffer
83                  ensureCapacity(count);
84          }
85      }
86  
87      public void write(int b) {
88          ensureCapacity(1);
89          buf[count] = (byte) b;
90          count++;
91      }
92  
93      /**
94       * Ensure that the buffer has at least this much space.
95       */
96      private void ensureCapacity(int space) {
97          int newcount = space + count;
98          if (newcount > buf.length) {
99              byte[] newbuf = new byte[Math.max(buf.length << 1, newcount)];
100             System.arraycopy(buf, 0, newbuf, 0, count);
101             buf = newbuf;
102         }
103     }
104 
105     public void write(byte[] b, int off, int len) {
106         ensureCapacity(len);
107         System.arraycopy(b, off, buf, count, len);
108         count += len;
109     }
110 
111     public void write(byte[] b) {
112         write(b, 0, b.length);
113     }
114 
115     /**
116      * Writes a string as ASCII string.
117      */
118     public void writeAsAscii(String s) {
119         int len = s.length();
120 
121         ensureCapacity(len);
122 
123         int ptr = count;
124         for( int i=0; i<len; i++ )
125             buf[ptr++] = (byte)s.charAt(i);
126         count = ptr;
127     }
128 
129     public void writeTo(OutputStream out) throws IOException {
130         out.write(buf, 0, count);
131     }
132 
133     public void reset() {
134         count = 0;
135     }
136 
137     /**
138      * Evil buffer reallocation method.
139      * Don't use it unless you absolutely have to.
140      *
141      * @deprecated
142      *      because this is evil!
143      */
144     public byte toByteArray()[] {
145         byte[] newbuf = new byte[count];
146         System.arraycopy(buf, 0, newbuf, 0, count);
147         return newbuf;
148     }
149 
150     public int size() {
151         return count;
152     }
153 
154     public ByteInputStream newInputStream() {
155         return new ByteInputStream(buf,count);
156     }
157 
158     /**
159      * Converts the buffer's contents into a string, translating bytes into
160      * characters according to the platform's default character encoding.
161      *
162      * @return String translated from the buffer's contents.
163      * @since JDK1.1
164      */
165     public String toString() {
166         return new String(buf, 0, count);
167     }
168 
169     public void close() {
170     }
171 
172     public byte[] getBytes() {
173         return buf;
174     }
175 
176 
177     public int getCount() {
178         return count;
179     }
180 }