Exemplo n.º 1
0
  /**
   * Writes an array of bytes on the stream. This method may not be used until this stream has been
   * passed to one of the methods in HTTPConnection (i.e. until it has been associated with a
   * request).
   *
   * @param buf an array containing the data to write
   * @param off the offset of the data whithin the buffer
   * @param len the number bytes (starting at <var>off</var>) to write
   * @exception IOException if any exception is thrown by the socket, or if writing <var>len</var>
   *     bytes would cause more bytes to be written than this stream is willing to accept.
   * @exception IllegalAccessError if this stream has not been associated with a request yet
   */
  public synchronized void write(byte[] buf, int off, int len)
      throws IOException, IllegalAccessError {
    if (req == null) throw new IllegalAccessError("Stream not associated with a request");

    if (ignore) return;

    if (length != -1 && rcvd + len > length) {
      IOException ioe =
          new IOException("Tried to write too many bytes (" + (rcvd + len) + " > " + length + ")");
      req.getConnection().closeDemux(ioe, false);
      req.getConnection().outputFinished();
      throw ioe;
    }

    try {
      if (bos != null) bos.write(buf, off, len);
      else if (length != -1) os.write(buf, off, len);
      else os.write(Codecs.chunkedEncode(buf, off, len, null, false));
    } catch (IOException ioe) {
      req.getConnection().closeDemux(ioe, true);
      req.getConnection().outputFinished();
      throw ioe;
    }

    rcvd += len;
  }
Exemplo n.º 2
0
  /**
   * Closes the stream and causes the data to be sent if it has not already been done so. This
   * method <strong>must</strong> be invoked when all data has been written.
   *
   * @exception IOException if any exception is thrown by the underlying socket, or if too few bytes
   *     were written.
   * @exception IllegalAccessError if this stream has not been associated with a request yet.
   */
  public synchronized void close() throws IOException, IllegalAccessError {
    if (req == null) throw new IllegalAccessError("Stream not associated with a request");

    if (ignore) return;

    if (bos != null) {
      req.setData(bos.toByteArray());
      req.setStream(null);

      if (trailers.length > 0) {
        NVPair[] hdrs = req.getHeaders();

        // remove any Trailer header field

        int len = hdrs.length;
        for (int idx = 0; idx < len; idx++) {
          if (hdrs[idx].getName().equalsIgnoreCase("Trailer")) {
            System.arraycopy(hdrs, idx + 1, hdrs, idx, len - idx - 1);
            len--;
          }
        }

        // add the trailers to the headers

        hdrs = Util.resizeArray(hdrs, len + trailers.length);
        System.arraycopy(trailers, 0, hdrs, len, trailers.length);

        req.setHeaders(hdrs);
      }

      if (DebugConn) System.err.println("OutS:  Sending request");

      try {
        resp = req.getConnection().sendRequest(req, con_to);
      } catch (ModuleException me) {
        throw new IOException(me.toString());
      }
      notify();
    } else {
      if (rcvd < length) {
        IOException ioe =
            new IOException(
                "Premature close: only "
                    + rcvd
                    + " bytes written instead of the "
                    + "expected "
                    + length);
        req.getConnection().closeDemux(ioe, false);
        req.getConnection().outputFinished();
        throw ioe;
      }

      try {
        if (length == -1) {
          if (DebugConn && trailers.length > 0) {
            System.err.println("OutS:  Sending trailers:");
            for (int idx = 0; idx < trailers.length; idx++)
              System.err.println(
                  "       " + trailers[idx].getName() + ": " + trailers[idx].getValue());
          }

          os.write(Codecs.chunkedEncode(null, 0, 0, trailers, true));
        }

        os.flush();

        if (DebugConn) System.err.println("OutS:  All data sent");
      } catch (IOException ioe) {
        req.getConnection().closeDemux(ioe, true);
        throw ioe;
      } finally {
        req.getConnection().outputFinished();
      }
    }
  }