net.sf.mmm.crypto.crypt.CryptorOutputStream Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of mmm-crypto-jce Show documentation

Library for simple and more secure usage of cryptography (JCA and JCE).

The newest version!

package net.sf.mmm.crypto.crypt;

import java.io.IOException;
import java.io.OutputStream;

import net.sf.mmm.crypto.io.OutputStreamWrapper;

/**
 * Implementation of {@link OutputStream} for {@link Encryptor#wrapStream(OutputStream)} based on
 * {@link OutputStreamWrapper}.
 *
 * @author Joerg Hohwiller (hohwille at users.sourceforge.net)
 * @since 1.0.0
 */
public class CryptorOutputStream extends OutputStreamWrapper {

  private Cryptor cryptor;

  /**
   * The constructor.
   *
   * @param cryptor the {@link Cryptor}.
   * @param delegate the {@link OutputStream} to wrap.
   */
  public CryptorOutputStream(Cryptor cryptor, OutputStream delegate) {
    super(delegate);
    this.cryptor = cryptor;
  }

  @Override
  public void write(byte[] b, int off, int len) throws IOException {

    requireNotClosed();
    byte[] data = this.cryptor.crypt(b, off, len, false);
    super.write(data);
  }

  @Override
  public void write(int b) throws IOException {

    byte[] data = this.cryptor.crypt(new byte[] { (byte) b }, false);
    super.write(data);
  }

  @Override
  public void close() throws IOException {

    if (!isClosed()) {
      byte[] data = this.cryptor.doFinal();
      if ((data != null) && (data.length > 0)) {
        super.write(data);
      }
    }
    super.close();
    this.cryptor = null;
  }

}