001/*
002 * Copyright 2007-2018 The jdeb developers.
003 *
004 * Licensed under the Apache License, Version 2.0 (the "License");
005 * you may not use this file except in compliance with the License.
006 * You may obtain a copy of the License at
007 *
008 *      http://www.apache.org/licenses/LICENSE-2.0
009 *
010 * Unless required by applicable law or agreed to in writing, software
011 * distributed under the License is distributed on an "AS IS" BASIS,
012 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
013 * See the License for the specific language governing permissions and
014 * limitations under the License.
015 */
016package org.vafer.jdeb.utils;
017
018import java.io.IOException;
019import java.io.OutputStream;
020import java.security.DigestOutputStream;
021import java.security.MessageDigest;
022
023/**
024 * Convenience class to provide digest info and length of a stream.
025 *
026 * ATTENTION: don't use outside of jdeb
027 */
028public class InformationOutputStream extends DigestOutputStream {
029
030    private final MessageDigest digest;
031    private long size;
032
033    public InformationOutputStream( OutputStream pStream, MessageDigest pDigest ) {
034        super(pStream, pDigest);
035        digest = pDigest;
036        size = 0;
037    }
038
039    public String getHexDigest() {
040        return Utils.toHex(digest.digest());
041    }
042
043    public void write( byte[] b, int off, int len ) throws IOException {
044        super.write(b, off, len);
045        size += len;
046    }
047
048    public void write( int b ) throws IOException {
049        super.write(b);
050        size++;
051    }
052
053    public long getSize() {
054        return size;
055    }
056}