| // |
| // ======================================================================== |
| // Copyright (c) 1995-2017 Mort Bay Consulting Pty. Ltd. |
| // ------------------------------------------------------------------------ |
| // All rights reserved. This program and the accompanying materials |
| // are made available under the terms of the Eclipse Public License v1.0 |
| // and Apache License v2.0 which accompanies this distribution. |
| // |
| // The Eclipse Public License is available at |
| // http://www.eclipse.org/legal/epl-v10.html |
| // |
| // The Apache License v2.0 is available at |
| // http://www.opensource.org/licenses/apache2.0.php |
| // |
| // You may elect to redistribute this code under either of these licenses. |
| // ======================================================================== |
| // |
| |
| package org.eclipse.jetty.io; |
| |
| import java.io.IOException; |
| import java.net.InetSocketAddress; |
| import java.nio.ByteBuffer; |
| import java.nio.channels.ClosedChannelException; |
| import java.nio.charset.Charset; |
| import java.nio.charset.StandardCharsets; |
| |
| import org.eclipse.jetty.util.BufferUtil; |
| import org.eclipse.jetty.util.log.Log; |
| import org.eclipse.jetty.util.log.Logger; |
| import org.eclipse.jetty.util.thread.Scheduler; |
| |
| |
| /* ------------------------------------------------------------ */ |
| /** ByteArrayEndPoint. |
| * |
| */ |
| public class ByteArrayEndPoint extends AbstractEndPoint |
| { |
| static final Logger LOG = Log.getLogger(ByteArrayEndPoint.class); |
| public final static InetSocketAddress NOIP=new InetSocketAddress(0); |
| |
| protected volatile ByteBuffer _in; |
| protected volatile ByteBuffer _out; |
| protected volatile boolean _ishut; |
| protected volatile boolean _oshut; |
| protected volatile boolean _closed; |
| protected volatile boolean _growOutput; |
| |
| |
| /* ------------------------------------------------------------ */ |
| /** |
| * |
| */ |
| public ByteArrayEndPoint() |
| { |
| this(null,0,null,null); |
| } |
| |
| /* ------------------------------------------------------------ */ |
| /** |
| * |
| */ |
| public ByteArrayEndPoint(byte[] input, int outputSize) |
| { |
| this(null,0,input!=null?BufferUtil.toBuffer(input):null,BufferUtil.allocate(outputSize)); |
| } |
| |
| /* ------------------------------------------------------------ */ |
| /** |
| * |
| */ |
| public ByteArrayEndPoint(String input, int outputSize) |
| { |
| this(null,0,input!=null?BufferUtil.toBuffer(input):null,BufferUtil.allocate(outputSize)); |
| } |
| |
| /* ------------------------------------------------------------ */ |
| public ByteArrayEndPoint(Scheduler scheduler, long idleTimeoutMs) |
| { |
| this(scheduler,idleTimeoutMs,null,null); |
| } |
| |
| /* ------------------------------------------------------------ */ |
| public ByteArrayEndPoint(Scheduler timer, long idleTimeoutMs, byte[] input, int outputSize) |
| { |
| this(timer,idleTimeoutMs,input!=null?BufferUtil.toBuffer(input):null,BufferUtil.allocate(outputSize)); |
| } |
| |
| /* ------------------------------------------------------------ */ |
| public ByteArrayEndPoint(Scheduler timer, long idleTimeoutMs, String input, int outputSize) |
| { |
| this(timer,idleTimeoutMs,input!=null?BufferUtil.toBuffer(input):null,BufferUtil.allocate(outputSize)); |
| } |
| |
| /* ------------------------------------------------------------ */ |
| public ByteArrayEndPoint(Scheduler timer, long idleTimeoutMs, ByteBuffer input, ByteBuffer output) |
| { |
| super(timer,NOIP,NOIP); |
| _in=input==null?BufferUtil.EMPTY_BUFFER:input; |
| _out=output==null?BufferUtil.allocate(1024):output; |
| setIdleTimeout(idleTimeoutMs); |
| } |
| |
| |
| |
| |
| |
| /* ------------------------------------------------------------ */ |
| @Override |
| protected void onIncompleteFlush() |
| { |
| // Don't need to do anything here as takeOutput does the signalling. |
| } |
| |
| /* ------------------------------------------------------------ */ |
| @Override |
| protected boolean needsFill() throws IOException |
| { |
| if (_closed) |
| throw new ClosedChannelException(); |
| return _in == null || BufferUtil.hasContent(_in); |
| } |
| |
| /* ------------------------------------------------------------ */ |
| /** |
| * @return Returns the in. |
| */ |
| public ByteBuffer getIn() |
| { |
| return _in; |
| } |
| |
| /* ------------------------------------------------------------ */ |
| /** |
| */ |
| public void setInputEOF() |
| { |
| _in = null; |
| } |
| |
| /* ------------------------------------------------------------ */ |
| /** |
| * @param in The in to set. |
| */ |
| public void setInput(ByteBuffer in) |
| { |
| _in = in; |
| if (in == null || BufferUtil.hasContent(in)) |
| getFillInterest().fillable(); |
| } |
| |
| /* ------------------------------------------------------------ */ |
| public void setInput(String s) |
| { |
| setInput(BufferUtil.toBuffer(s,StandardCharsets.UTF_8)); |
| } |
| |
| /* ------------------------------------------------------------ */ |
| public void setInput(String s,Charset charset) |
| { |
| setInput(BufferUtil.toBuffer(s,charset)); |
| } |
| |
| /* ------------------------------------------------------------ */ |
| /** |
| * @return Returns the out. |
| */ |
| public ByteBuffer getOutput() |
| { |
| return _out; |
| } |
| |
| /* ------------------------------------------------------------ */ |
| /** |
| * @return Returns the out. |
| */ |
| public String getOutputString() |
| { |
| return getOutputString(StandardCharsets.UTF_8); |
| } |
| |
| /* ------------------------------------------------------------ */ |
| /** |
| * @return Returns the out. |
| */ |
| public String getOutputString(Charset charset) |
| { |
| return BufferUtil.toString(_out,charset); |
| } |
| |
| /* ------------------------------------------------------------ */ |
| /** |
| * @return Returns the out. |
| */ |
| public ByteBuffer takeOutput() |
| { |
| ByteBuffer b=_out; |
| _out=BufferUtil.allocate(b.capacity()); |
| getWriteFlusher().completeWrite(); |
| return b; |
| } |
| |
| /* ------------------------------------------------------------ */ |
| /** |
| * @return Returns the out. |
| */ |
| public String takeOutputString() |
| { |
| return takeOutputString(StandardCharsets.UTF_8); |
| } |
| |
| /* ------------------------------------------------------------ */ |
| /** |
| * @return Returns the out. |
| */ |
| public String takeOutputString(Charset charset) |
| { |
| ByteBuffer buffer=takeOutput(); |
| return BufferUtil.toString(buffer,charset); |
| } |
| |
| /* ------------------------------------------------------------ */ |
| /** |
| * @param out The out to set. |
| */ |
| public void setOutput(ByteBuffer out) |
| { |
| _out = out; |
| getWriteFlusher().completeWrite(); |
| } |
| |
| /* ------------------------------------------------------------ */ |
| /* |
| * @see org.eclipse.io.EndPoint#isOpen() |
| */ |
| @Override |
| public boolean isOpen() |
| { |
| return !_closed; |
| } |
| |
| /* ------------------------------------------------------------ */ |
| /* |
| */ |
| @Override |
| public boolean isInputShutdown() |
| { |
| return _ishut||_closed; |
| } |
| |
| /* ------------------------------------------------------------ */ |
| /* |
| */ |
| @Override |
| public boolean isOutputShutdown() |
| { |
| return _oshut||_closed; |
| } |
| |
| /* ------------------------------------------------------------ */ |
| private void shutdownInput() |
| { |
| _ishut=true; |
| if (_oshut) |
| close(); |
| } |
| |
| /* ------------------------------------------------------------ */ |
| /* |
| * @see org.eclipse.io.EndPoint#shutdownOutput() |
| */ |
| @Override |
| public void shutdownOutput() |
| { |
| _oshut=true; |
| if (_ishut) |
| close(); |
| } |
| |
| /* ------------------------------------------------------------ */ |
| /* |
| * @see org.eclipse.io.EndPoint#close() |
| */ |
| @Override |
| public void close() |
| { |
| super.close(); |
| _closed=true; |
| } |
| |
| /* ------------------------------------------------------------ */ |
| /** |
| * @return <code>true</code> if there are bytes remaining to be read from the encoded input |
| */ |
| public boolean hasMore() |
| { |
| return getOutput().position()>0; |
| } |
| |
| /* ------------------------------------------------------------ */ |
| /* |
| * @see org.eclipse.io.EndPoint#fill(org.eclipse.io.Buffer) |
| */ |
| @Override |
| public int fill(ByteBuffer buffer) throws IOException |
| { |
| if (_closed) |
| throw new EofException("CLOSED"); |
| if (_in==null) |
| shutdownInput(); |
| if (_ishut) |
| return -1; |
| int filled=BufferUtil.append(buffer,_in); |
| if (filled>0) |
| notIdle(); |
| return filled; |
| } |
| |
| /* ------------------------------------------------------------ */ |
| /* |
| * @see org.eclipse.io.EndPoint#flush(org.eclipse.io.Buffer, org.eclipse.io.Buffer, org.eclipse.io.Buffer) |
| */ |
| @Override |
| public boolean flush(ByteBuffer... buffers) throws IOException |
| { |
| if (_closed) |
| throw new IOException("CLOSED"); |
| if (_oshut) |
| throw new IOException("OSHUT"); |
| |
| boolean flushed=true; |
| boolean idle=true; |
| |
| for (ByteBuffer b : buffers) |
| { |
| if (BufferUtil.hasContent(b)) |
| { |
| if (_growOutput && b.remaining()>BufferUtil.space(_out)) |
| { |
| BufferUtil.compact(_out); |
| if (b.remaining()>BufferUtil.space(_out)) |
| { |
| ByteBuffer n = BufferUtil.allocate(_out.capacity()+b.remaining()*2); |
| BufferUtil.append(n,_out); |
| _out=n; |
| } |
| } |
| |
| if (BufferUtil.append(_out,b)>0) |
| idle=false; |
| |
| if (BufferUtil.hasContent(b)) |
| { |
| flushed=false; |
| break; |
| } |
| } |
| } |
| if (!idle) |
| notIdle(); |
| return flushed; |
| } |
| |
| /* ------------------------------------------------------------ */ |
| /** |
| * |
| */ |
| public void reset() |
| { |
| getFillInterest().onClose(); |
| getWriteFlusher().onClose(); |
| _ishut=false; |
| _oshut=false; |
| _closed=false; |
| _in=null; |
| BufferUtil.clear(_out); |
| } |
| |
| /* ------------------------------------------------------------ */ |
| /* |
| * @see org.eclipse.io.EndPoint#getConnection() |
| */ |
| @Override |
| public Object getTransport() |
| { |
| return null; |
| } |
| |
| /* ------------------------------------------------------------ */ |
| /** |
| * @return the growOutput |
| */ |
| public boolean isGrowOutput() |
| { |
| return _growOutput; |
| } |
| |
| /* ------------------------------------------------------------ */ |
| /** |
| * @param growOutput the growOutput to set |
| */ |
| public void setGrowOutput(boolean growOutput) |
| { |
| _growOutput=growOutput; |
| } |
| |
| |
| } |