80 lines · 3.0 KB · Java
RawDownload
/* * Copyright (c) 1996, Oracle and/or its affiliates. All rights reserved. * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER. * * This code is free software; you can redistribute it and/or modify it * under the terms of the GNU General Public License version 2 only, as * published by the Free Software Foundation.  Oracle designates this * particular file as subject to the "Classpath" exception as provided * by Oracle in the LICENSE file that accompanied this code. * * This code is distributed in the hope that it will be useful, but WITHOUT * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License * version 2 for more details (a copy is included in the LICENSE file that * accompanied this code). * * You should have received a copy of the GNU General Public License version * 2 along with this work; if not, write to the Free Software Foundation, * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA. * * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA * or visit www.oracle.com if you need additional information or have any * questions. */package sun.rmi.transport.proxy; import java.io.*; /** * The HttpOutputStream class assists the HttpSendSocket and HttpReceiveSocket * classes by providing an output stream that buffers its entire input until * closed, and then it sends the complete transmission prefixed by the end of * an HTTP header that specifies the content length. */class HttpOutputStream extends ByteArrayOutputStream {     /** the output stream to send response to */    protected OutputStream out;     /** true if HTTP response has been sent */    boolean responseSent = false;     /**     * Begin buffering new HTTP response to be sent to a given stream.     * @param out the OutputStream to send response to     */    public HttpOutputStream(OutputStream out) {        super();        this.out = out;    }     /**     * On close, send HTTP-packaged response.     */    public synchronized void close() throws IOException {        if (!responseSent) {            /*             * If response would have zero content length, then make it             * have some arbitrary data so that certain clients will not             * fail because the "document contains no data".             */            if (size() == 0)                write(emptyData);             DataOutputStream dos = new DataOutputStream(out);            dos.writeBytes("Content-type: application/octet-stream\r\n");            dos.writeBytes("Content-length: " + size() + "\r\n");            dos.writeBytes("\r\n");            writeTo(dos);            dos.flush();            // Do not close the underlying stream here, because that would            // close the underlying socket and prevent reading a response.            reset(); // reset byte array            responseSent = true;        }    }     /** data to send if the response would otherwise be empty */    private static byte[] emptyData = { 0 };}