1f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project/* 2f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project * Licensed to the Apache Software Foundation (ASF) under one or more 3f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project * contributor license agreements. See the NOTICE file distributed with 4f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project * this work for additional information regarding copyright ownership. 5f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project * The ASF licenses this file to You under the Apache License, Version 2.0 6f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project * (the "License"); you may not use this file except in compliance with 7f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project * the License. You may obtain a copy of the License at 8f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project * 9f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project * http://www.apache.org/licenses/LICENSE-2.0 10f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project * 11f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project * Unless required by applicable law or agreed to in writing, software 12f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project * distributed under the License is distributed on an "AS IS" BASIS, 13f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 14f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project * See the License for the specific language governing permissions and 15f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project * limitations under the License. 16f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project */ 17f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project 18f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Projectpackage java.security; 19f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project 20f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Projectimport java.io.FilterOutputStream; 21f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Projectimport java.io.IOException; 22f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Projectimport java.io.OutputStream; 23f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project 24f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project/** 25f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project * {@code DigestOutputStream} is a {@code FilterOutputStream} which maintains an 26f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project * associated message digest. 27f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project */ 28f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Projectpublic class DigestOutputStream extends FilterOutputStream { 29f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project 30f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project /** 31f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project * The message digest for this stream. 32f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project */ 33f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project protected MessageDigest digest; 34f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project 35f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project // Indicates whether digest functionality is on or off 36f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project private boolean isOn = true; 37f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project 38f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project /** 39f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project * Constructs a new instance of this {@code DigestOutputStream}, using the 40f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project * given {@code stream} and the {@code digest}. 41e7637fe9734c4e3bece51db6773505c04e49fabaElliott Hughes * 42f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project * @param stream 43f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project * the output stream. 44f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project * @param digest 45f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project * the message digest. 46f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project */ 47f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project public DigestOutputStream(OutputStream stream, MessageDigest digest) { 48f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project super(stream); 49f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project this.digest = digest; 50f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project } 51f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project 52f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project /** 53f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project * Returns the message digest for this stream. 54e7637fe9734c4e3bece51db6773505c04e49fabaElliott Hughes * 55f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project * @return the message digest for this stream. 56f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project */ 57f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project public MessageDigest getMessageDigest() { 58f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project return digest; 59f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project } 60f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project 61f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project /** 62f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project * Sets the message digest which this stream will use. 63e7637fe9734c4e3bece51db6773505c04e49fabaElliott Hughes * 64f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project * @param digest 65f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project * the message digest which this stream will use. 66f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project */ 67f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project public void setMessageDigest(MessageDigest digest) { 68f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project this.digest = digest; 69f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project } 70f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project 71f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project /** 72f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project * Writes the specified {@code int} to the stream. Updates the digest if 73f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project * this function is {@link #on(boolean)}. 74e7637fe9734c4e3bece51db6773505c04e49fabaElliott Hughes * 75f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project * @param b 76f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project * the byte to be written. 77f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project * @throws IOException 78f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project * if writing to the stream causes a {@code IOException} 79f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project */ 80e7637fe9734c4e3bece51db6773505c04e49fabaElliott Hughes @Override 81f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project public void write(int b) throws IOException { 82f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project // update digest only if digest functionality is on 83f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project if (isOn) { 84f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project digest.update((byte)b); 85f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project } 86f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project // write the byte 87f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project out.write(b); 88f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project } 89f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project 90f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project /** 91f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project * Writes {@code len} bytes into the stream, starting from the specified 92f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project * offset. Updates the digest if this function is {@link #on(boolean)}. 93e7637fe9734c4e3bece51db6773505c04e49fabaElliott Hughes * 94f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project * @param b 95f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project * the buffer to write to. 96f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project * @param off 97f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project * the index of the first byte in {@code b} to write. 98f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project * @param len 99f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project * the number of bytes in {@code b} to write. 100f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project * @throws IOException 101f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project * if writing to the stream causes an {@code IOException}. 102f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project */ 103e7637fe9734c4e3bece51db6773505c04e49fabaElliott Hughes @Override 104f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project public void write(byte[] b, int off, int len) throws IOException { 105f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project // update digest only if digest functionality is on 106f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project if (isOn) { 107f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project digest.update(b, off, len); 108f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project } 109f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project // write len bytes 110f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project out.write(b, off, len); 111f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project } 112f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project 113f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project /** 114f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project * Enables or disables the digest function (default is on). 115e7637fe9734c4e3bece51db6773505c04e49fabaElliott Hughes * 116f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project * @param on 117f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project * {@code true} if the digest should be computed, {@code false} 118f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project * otherwise. 119e7637fe9734c4e3bece51db6773505c04e49fabaElliott Hughes * @see MessageDigest 120f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project */ 121f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project public void on(boolean on) { 122f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project isOn = on; 123f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project } 124f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project 125f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project /** 126f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project * Returns a string containing a concise, human-readable description of this 127f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project * {@code DigestOutputStream} including the digest. 128e7637fe9734c4e3bece51db6773505c04e49fabaElliott Hughes * 129f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project * @return a printable representation for this {@code DigestOutputStream}. 130f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project */ 131e7637fe9734c4e3bece51db6773505c04e49fabaElliott Hughes @Override 132f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project public String toString() { 133f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project return super.toString() + ", " + digest.toString() + //$NON-NLS-1$ 134f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project (isOn ? ", is on" : ", is off"); //$NON-NLS-1$ //$NON-NLS-2$ 135f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project } 136f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project} 137