com.martiansoftware.nailgun.NGInputStream Maven / Gradle / Ivy
/*
Copyright 2004-2012, Martian Software, Inc.
Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at
http://www.apache.org/licenses/LICENSE-2.0
Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
*/
package com.martiansoftware.nailgun;
import java.io.DataInputStream;
import java.io.DataOutputStream;
import java.io.FilterInputStream;
import java.io.IOException;
/**
* A FilterInputStream that is able to read the chunked stdin stream
* from a NailGun client.
*
* @author Marty Lamb
*/
class NGInputStream extends FilterInputStream {
private DataInputStream din;
private boolean eof = false;
private long remaining = 0;
private byte[] oneByteBuffer = null;
private final DataOutputStream out;
private boolean started = false;
/**
* Creates a new NGInputStream wrapping the specified InputStream
* @param in the InputStream to wrap
* @param out the OutputStream to which a STARTINPUT chunk should
* be sent prior to the first read.
*/
public NGInputStream(java.io.InputStream in, DataOutputStream out) {
super(in);
din = (DataInputStream) this.in;
this.out = out;
}
/**
* Reads a NailGun chunk header from the underlying InputStream.
*
* @throws IOException if thrown by the underlying InputStream,
* or if an unexpected NailGun chunk type is encountered.
*/
private void readHeader() throws IOException {
if (eof) return;
int hlen = din.readInt();
byte chunkType = din.readByte();
switch(chunkType) {
case NGConstants.CHUNKTYPE_STDIN:
remaining = hlen;
break;
case NGConstants.CHUNKTYPE_STDIN_EOF:
eof = true;
break;
default: throw(new IOException("Unknown stream type: " + (char) chunkType));
}
}
/**
* @see java.io.InputStream#available()
*/
public int available() throws IOException {
if (eof) return(0);
if (remaining > 0) return (in.available());
return (Math.max(0, in.available() - 5));
}
/**
* @see java.io.InputStream#markSupported()
*/
public boolean markSupported() {
return (false);
}
/**
* @see java.io.InputStream#read()
*/
public int read() throws IOException {
if (oneByteBuffer == null) oneByteBuffer = new byte[1];
return((read(oneByteBuffer, 0, 1) == -1) ? -1 : (int) oneByteBuffer[0]);
}
/**
* @see java.io.InputStream.read(byte[])
*/
public int read(byte[] b) throws IOException {
return (read(b, 0, b.length));
}
/**
* @see java.io.InputStream.read(byte[],offset,length)
*/
public int read(byte[] b, int offset, int length) throws IOException {
if (!started) {
synchronized(out) {
out.writeInt(0);
out.writeByte(NGConstants.CHUNKTYPE_STARTINPUT);
out.flush();
started = true;
}
}
if (remaining == 0) readHeader();
if (eof) return(-1);
int bytesToRead = Math.min((int) remaining, length);
int result = in.read(b, offset, bytesToRead);
remaining -= result;
return (result);
}
}