public static byte[] decode(MessageHeader message, byte[] content) throws MessageFormatException {
try {
InputStream is = new ByteArrayInputStream(content);
is = decode(message, is);
ByteArrayOutputStream copy = new ByteArrayOutputStream();
byte[] buff = new byte[4096];
int got;
while ((got = is.read(buff)) > 0) copy.write(buff, 0, got);
return copy.toByteArray();
} catch (IOException ioe) {
throw new MessageFormatException("Malformed encoded content: " + ioe.getMessage(), ioe);
}
}
public static byte[] encode(MessageHeader header, byte[] content) throws MessageFormatException {
InputStream contentStream = new ByteArrayInputStream(content);
contentStream = encode(header, contentStream);
ByteArrayOutputStream copy = new ByteArrayOutputStream();
byte[] buff = new byte[4096];
int got;
try {
while ((got = contentStream.read(buff)) > 0) copy.write(buff, 0, got);
} catch (IOException ioe) {
throw new MessageFormatException("Error encoding content", ioe);
}
return copy.toByteArray();
}
public static MessageHeader readHeader(InputStream in)
throws IOException, MessageFormatException {
int match = 0;
ByteArrayOutputStream out = new ByteArrayOutputStream();
int i;
try {
do {
i = in.read();
if (i == -1) {
if (out.size() > 0)
throw new MessageFormatException("Unexpected end of stream", out.toByteArray());
return null;
}
out.write(i);
if (i == CRLF[match % 2]) match++;
} while (match < 4);
} catch (IOException e) {
if (out.size() > 0)
throw new MessageFormatException("Incomplete message header", e, out.toByteArray());
throw e;
}
MutableMessageHeader header = new MutableMessageHeader.Impl();
header.setHeader(out.toByteArray());
return header;
}
public static boolean flushContent(MutableMessageHeader header, InputStream in, OutputStream out)
throws MessageFormatException, IOException {
boolean read = false;
String te = header.getHeader(TRANSFER_ENCODING);
if (CHUNKED.equalsIgnoreCase(te)) {
in = new ChunkedInputStream(in);
} else if (te != null) {
throw new IOException("Unknown Transfer-Encoding '" + te + "'");
} else {
String cl = header.getHeader(CONTENT_LENGTH);
if (cl != null) {
try {
int l = Integer.parseInt(cl.trim());
if (l == 0) return read;
in = new FixedLengthInputStream(in, l);
} catch (NumberFormatException nfe) {
throw new MessageFormatException("Invalid Content-Length header: " + cl, nfe);
}
}
}
byte[] buff = new byte[2048];
int got;
while ((got = in.read(buff)) > 0) {
read = true;
if (out != null) out.write(buff, 0, got);
}
return read;
}
private static void buffer(StreamingMessage message, MutableBufferedMessage buffered, int max)
throws IOException, SizeLimitExceededException {
buffered.setHeader(message.getHeader());
InputStream in = message.getContent();
if (in != null) {
ByteArrayOutputStream copy;
copy = new SizeLimitedByteArrayOutputStream(max);
byte[] b = new byte[1024];
int got;
try {
while ((got = in.read(b)) > -1) {
copy.write(b, 0, got);
}
buffered.setContent(copy.toByteArray());
} catch (SizeLimitExceededException slee) {
buffered.setContent(copy.toByteArray());
throw slee;
}
}
}
