Source for gnu.inet.http.ByteArrayResponseBodyReader

   1: /*
   2:  * ByteArrayResponseBodyReader.java
   3:  * Copyright (C) 2004 The Free Software Foundation
   4:  * 
   5:  * This file is part of GNU inetlib, a library.
   6:  * 
   7:  * GNU inetlib is free software; you can redistribute it and/or modify
   8:  * it under the terms of the GNU General Public License as published by
   9:  * the Free Software Foundation; either version 2 of the License, or
  10:  * (at your option) any later version.
  11:  * 
  12:  * GNU inetlib is distributed in the hope that it will be useful,
  13:  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  14:  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  15:  * GNU General Public License for more details.
  16:  * 
  17:  * You should have received a copy of the GNU General Public License
  18:  * along with this library; if not, write to the Free Software
  19:  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  20:  * 
  21:  * Linking this library statically or dynamically with other modules is
  22:  * making a combined work based on this library.  Thus, the terms and
  23:  * conditions of the GNU General Public License cover the whole
  24:  * combination.
  25:  *
  26:  * As a special exception, the copyright holders of this library give you
  27:  * permission to link this library with independent modules to produce an
  28:  * executable, regardless of the license terms of these independent
  29:  * modules, and to copy and distribute the resulting executable under
  30:  * terms of your choice, provided that you also meet, for each linked
  31:  * independent module, the terms and conditions of the license of that
  32:  * module.  An independent module is a module which is not derived from
  33:  * or based on this library.  If you modify this library, you may extend
  34:  * this exception to your version of the library, but you are not
  35:  * obliged to do so.  If you do not wish to do so, delete this
  36:  * exception statement from your version.
  37:  */
  38: 
  39: package gnu.inet.http;
  40: 
  41: /**
  42:  * Simple response body reader that stores content in a byte array.
  43:  *
  44:  * @author <a href='mailto:dog@gnu.org'>Chris Burdess</a>
  45:  */
  46: public class ByteArrayResponseBodyReader
  47:   implements ResponseBodyReader
  48: {
  49: 
  50:   /**
  51:    * The content.
  52:    */
  53:   protected byte[] content;
  54: 
  55:   /**
  56:    * The position in the content at which the next write will occur.
  57:    */
  58:   protected int pos;
  59: 
  60:   /**
  61:    * The length of the buffer.
  62:    */
  63:   protected int len;
  64: 
  65:   /**
  66:    * Constructs a new byte array response body reader.
  67:    */
  68:   public ByteArrayResponseBodyReader()
  69:   {
  70:     this(4096);
  71:   }
  72:   
  73:   /**
  74:    * Constructs a new byte array response body reader with the specified
  75:    * initial buffer size.
  76:    * @param size the initial buffer size
  77:    */
  78:   public ByteArrayResponseBodyReader(int size)
  79:   {
  80:     content = new byte[size];
  81:     pos = len = 0;
  82:   }
  83: 
  84:   /**
  85:    * This reader accepts all responses.
  86:    */ 
  87:   public boolean accept(Request request, Response response)
  88:   {
  89:     return true;
  90:   }
  91: 
  92:   public void read(byte[] buffer, int offset, int length)
  93:   {
  94:     int l = length - offset;
  95:     if (pos + l > content.length)
  96:       {
  97:         byte[] tmp = new byte[content.length * 2];
  98:         System.arraycopy(content, 0, tmp, 0, pos);
  99:         content = tmp;
 100:       }
 101:     System.arraycopy(buffer, offset, content, pos, l);
 102:     pos += l;
 103:     len = pos;
 104:   }
 105: 
 106:   public void close()
 107:   {
 108:     pos = 0;
 109:   }
 110: 
 111:   /**
 112:    * Retrieves the content of this reader as a byte array.
 113:    * The size of the returned array is the number of bytes read.
 114:    */
 115:   public byte[] toByteArray()
 116:   {
 117:     byte[] ret = new byte[len];
 118:     System.arraycopy(content, 0, ret, 0, len);
 119:     return ret;
 120:   }
 121:   
 122: }