/* ******************************************************************************
* Copyright (c) 2006-2012 XMind Ltd. and others.
*
* This file is a part of XMind 3. XMind releases 3 and
* above are dual-licensed under the Eclipse Public License (EPL),
* which is available at http://www.eclipse.org/legal/epl-v10.html
* and the GNU Lesser General Public License (LGPL),
* which is available at http://www.gnu.org/licenses/lgpl.html
* See http://www.xmind.net/license.html for details.
*
* Contributors:
* XMind Ltd. - initial API and implementation
*******************************************************************************/
package org.xmind.core.internal.security;
import java.io.FilterOutputStream;
import java.io.IOException;
import java.io.OutputStream;
import org.bouncycastle.crypto.Digest;
import org.bouncycastle.crypto.digests.MD5Digest;
import org.xmind.core.CoreException;
import org.xmind.core.IChecksumStream;
/**
* @author Frank Shaka
*
*/
public class ChecksumOutputStream extends FilterOutputStream
implements IChecksumStream {
private Digest digest;
private String checksum = null;
/**
* @throws CoreException
*
*/
public ChecksumOutputStream(OutputStream output) throws CoreException {
super(output);
this.digest = new MD5Digest();
}
/*
* (non-Javadoc)
*
* @see java.io.OutputStream#write(int)
*/
public void write(int b) throws IOException {
digest.update((byte) b);
out.write(b);
}
public void write(byte[] b) throws IOException {
digest.update(b, 0, b.length);
out.write(b);
}
public void write(byte[] b, int off, int len) throws IOException {
digest.update(b, off, len);
out.write(b, off, len);
}
public String getChecksum() {
if (this.checksum == null) {
// Generate checksum:
byte[] checksumBytes = new byte[digest.getDigestSize()];
digest.doFinal(checksumBytes, 0);
this.checksum = Base64.byteArrayToBase64(checksumBytes);
}
return this.checksum;
}
}