196c5af40d639d629267794f4f0338a267ff94ce5The Android Open Source Project/* 296c5af40d639d629267794f4f0338a267ff94ce5The Android Open Source Project * Copyright (C) 2008 The Android Open Source Project 396c5af40d639d629267794f4f0338a267ff94ce5The Android Open Source Project * 496c5af40d639d629267794f4f0338a267ff94ce5The Android Open Source Project * Licensed under the Apache License, Version 2.0 (the "License"); 596c5af40d639d629267794f4f0338a267ff94ce5The Android Open Source Project * you may not use this file except in compliance with the License. 696c5af40d639d629267794f4f0338a267ff94ce5The Android Open Source Project * You may obtain a copy of the License at 796c5af40d639d629267794f4f0338a267ff94ce5The Android Open Source Project * 896c5af40d639d629267794f4f0338a267ff94ce5The Android Open Source Project * http://www.apache.org/licenses/LICENSE-2.0 996c5af40d639d629267794f4f0338a267ff94ce5The Android Open Source Project * 1096c5af40d639d629267794f4f0338a267ff94ce5The Android Open Source Project * Unless required by applicable law or agreed to in writing, software 1196c5af40d639d629267794f4f0338a267ff94ce5The Android Open Source Project * distributed under the License is distributed on an "AS IS" BASIS, 1296c5af40d639d629267794f4f0338a267ff94ce5The Android Open Source Project * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 1396c5af40d639d629267794f4f0338a267ff94ce5The Android Open Source Project * See the License for the specific language governing permissions and 1496c5af40d639d629267794f4f0338a267ff94ce5The Android Open Source Project * limitations under the License. 1596c5af40d639d629267794f4f0338a267ff94ce5The Android Open Source Project */ 1696c5af40d639d629267794f4f0338a267ff94ce5The Android Open Source Project 1796c5af40d639d629267794f4f0338a267ff94ce5The Android Open Source Projectpackage com.android.email; 1896c5af40d639d629267794f4f0338a267ff94ce5The Android Open Source Project 1996c5af40d639d629267794f4f0338a267ff94ce5The Android Open Source Projectimport java.io.IOException; 2096c5af40d639d629267794f4f0338a267ff94ce5The Android Open Source Projectimport java.io.InputStream; 2196c5af40d639d629267794f4f0338a267ff94ce5The Android Open Source Project 2296c5af40d639d629267794f4f0338a267ff94ce5The Android Open Source Project/** 2396c5af40d639d629267794f4f0338a267ff94ce5The Android Open Source Project * A filtering InputStream that stops allowing reads after the given length has been read. This 2496c5af40d639d629267794f4f0338a267ff94ce5The Android Open Source Project * is used to allow a client to read directly from an underlying protocol stream without reading 2596c5af40d639d629267794f4f0338a267ff94ce5The Android Open Source Project * past where the protocol handler intended the client to read. 2696c5af40d639d629267794f4f0338a267ff94ce5The Android Open Source Project */ 2796c5af40d639d629267794f4f0338a267ff94ce5The Android Open Source Projectpublic class FixedLengthInputStream extends InputStream { 28b3f7dd0169a35221184b9327c8ce337b09dc6d1fMakoto Onuki private final InputStream mIn; 29b3f7dd0169a35221184b9327c8ce337b09dc6d1fMakoto Onuki private final int mLength; 3096c5af40d639d629267794f4f0338a267ff94ce5The Android Open Source Project private int mCount; 3196c5af40d639d629267794f4f0338a267ff94ce5The Android Open Source Project 3296c5af40d639d629267794f4f0338a267ff94ce5The Android Open Source Project public FixedLengthInputStream(InputStream in, int length) { 3396c5af40d639d629267794f4f0338a267ff94ce5The Android Open Source Project this.mIn = in; 3496c5af40d639d629267794f4f0338a267ff94ce5The Android Open Source Project this.mLength = length; 3596c5af40d639d629267794f4f0338a267ff94ce5The Android Open Source Project } 3696c5af40d639d629267794f4f0338a267ff94ce5The Android Open Source Project 3796c5af40d639d629267794f4f0338a267ff94ce5The Android Open Source Project @Override 3896c5af40d639d629267794f4f0338a267ff94ce5The Android Open Source Project public int available() throws IOException { 3996c5af40d639d629267794f4f0338a267ff94ce5The Android Open Source Project return mLength - mCount; 4096c5af40d639d629267794f4f0338a267ff94ce5The Android Open Source Project } 4196c5af40d639d629267794f4f0338a267ff94ce5The Android Open Source Project 4296c5af40d639d629267794f4f0338a267ff94ce5The Android Open Source Project @Override 4396c5af40d639d629267794f4f0338a267ff94ce5The Android Open Source Project public int read() throws IOException { 4496c5af40d639d629267794f4f0338a267ff94ce5The Android Open Source Project if (mCount < mLength) { 4596c5af40d639d629267794f4f0338a267ff94ce5The Android Open Source Project mCount++; 4696c5af40d639d629267794f4f0338a267ff94ce5The Android Open Source Project return mIn.read(); 4796c5af40d639d629267794f4f0338a267ff94ce5The Android Open Source Project } else { 4896c5af40d639d629267794f4f0338a267ff94ce5The Android Open Source Project return -1; 4996c5af40d639d629267794f4f0338a267ff94ce5The Android Open Source Project } 5096c5af40d639d629267794f4f0338a267ff94ce5The Android Open Source Project } 5196c5af40d639d629267794f4f0338a267ff94ce5The Android Open Source Project 5296c5af40d639d629267794f4f0338a267ff94ce5The Android Open Source Project @Override 5396c5af40d639d629267794f4f0338a267ff94ce5The Android Open Source Project public int read(byte[] b, int offset, int length) throws IOException { 5496c5af40d639d629267794f4f0338a267ff94ce5The Android Open Source Project if (mCount < mLength) { 5596c5af40d639d629267794f4f0338a267ff94ce5The Android Open Source Project int d = mIn.read(b, offset, Math.min(mLength - mCount, length)); 5696c5af40d639d629267794f4f0338a267ff94ce5The Android Open Source Project if (d == -1) { 5796c5af40d639d629267794f4f0338a267ff94ce5The Android Open Source Project return -1; 5896c5af40d639d629267794f4f0338a267ff94ce5The Android Open Source Project } else { 5996c5af40d639d629267794f4f0338a267ff94ce5The Android Open Source Project mCount += d; 6096c5af40d639d629267794f4f0338a267ff94ce5The Android Open Source Project return d; 6196c5af40d639d629267794f4f0338a267ff94ce5The Android Open Source Project } 6296c5af40d639d629267794f4f0338a267ff94ce5The Android Open Source Project } else { 6396c5af40d639d629267794f4f0338a267ff94ce5The Android Open Source Project return -1; 6496c5af40d639d629267794f4f0338a267ff94ce5The Android Open Source Project } 6596c5af40d639d629267794f4f0338a267ff94ce5The Android Open Source Project } 6696c5af40d639d629267794f4f0338a267ff94ce5The Android Open Source Project 6796c5af40d639d629267794f4f0338a267ff94ce5The Android Open Source Project @Override 6896c5af40d639d629267794f4f0338a267ff94ce5The Android Open Source Project public int read(byte[] b) throws IOException { 6996c5af40d639d629267794f4f0338a267ff94ce5The Android Open Source Project return read(b, 0, b.length); 7096c5af40d639d629267794f4f0338a267ff94ce5The Android Open Source Project } 7196c5af40d639d629267794f4f0338a267ff94ce5The Android Open Source Project 727e5ba0e1eaee76ab6e6c7ea9362348f660796596Makoto Onuki public int getLength() { 737e5ba0e1eaee76ab6e6c7ea9362348f660796596Makoto Onuki return mLength; 747e5ba0e1eaee76ab6e6c7ea9362348f660796596Makoto Onuki } 757e5ba0e1eaee76ab6e6c7ea9362348f660796596Makoto Onuki 76165e8bfe7a46564ea98bb16cb059102ba356a1b5Makoto Onuki @Override 7796c5af40d639d629267794f4f0338a267ff94ce5The Android Open Source Project public String toString() { 7896c5af40d639d629267794f4f0338a267ff94ce5The Android Open Source Project return String.format("FixedLengthInputStream(in=%s, length=%d)", mIn.toString(), mLength); 7996c5af40d639d629267794f4f0338a267ff94ce5The Android Open Source Project } 8096c5af40d639d629267794f4f0338a267ff94ce5The Android Open Source Project} 81