Java tutorial
/** * Copyright 2012, Board of Regents of the University of * Wisconsin System. See the NOTICE file distributed with * this work for additional information regarding copyright * ownership. Board of Regents of the University of Wisconsin * System licenses this file to you under the Apache License, * Version 2.0 (the "License"); you may not use this file * except in compliance with the License. You may obtain a * copy of the License at: * * http://www.apache.org/licenses/LICENSE-2.0 * * Unless required by applicable law or agreed to in writing, * software distributed under the License is distributed on * an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY * KIND, either express or implied. See the License for the * specific language governing permissions and limitations * under the License. */ package edu.wisc.doit.tcrypt.ant; import java.io.BufferedInputStream; import java.io.BufferedOutputStream; import java.io.File; import java.io.FileOutputStream; import java.io.IOException; import java.io.InputStream; import java.io.InputStreamReader; import java.io.OutputStream; import java.io.Reader; import org.apache.commons.io.FilenameUtils; import org.apache.commons.io.IOUtils; import org.apache.tools.ant.BuildException; import org.apache.tools.ant.types.Resource; import org.bouncycastle.crypto.InvalidCipherTextException; import edu.wisc.doit.tcrypt.BouncyCastleFileEncrypter; import edu.wisc.doit.tcrypt.FileEncrypter; /** * Encrypts a file using the public key, if neither {@link #setDestDir(File)} or {@link #setDestFile(File)} are * specified ${basedir} is used * * @author Eric Dalquist */ public class EncryptFileTask extends AbstractFileTask { private Resource publicKey; /** * @param publicKey The public key used to encrypt */ public void setPublicKey(Resource privateKey) { this.publicKey = privateKey; } /** * @param srcFile The file to encrypt */ public void setSrcFile(Resource srcFile) { this.srcFile = srcFile; } /** * @param destDir Dest directory to encrypt to */ public void setDestDir(File destDir) { this.destDir = destDir; } /** * @param destFile Dest file to encrypt to */ public void setDestFile(File destFile) { this.destFile = destFile; } @Override public void execute() throws BuildException { final FileEncrypter fileEncrypter; Reader publicKeyReader = null; try { publicKeyReader = new InputStreamReader(new BufferedInputStream(this.publicKey.getInputStream()), FileEncrypter.CHARSET); fileEncrypter = new BouncyCastleFileEncrypter(publicKeyReader); } catch (IOException e) { throw new BuildException("Failed to create BouncyCastleFileEncrypter for public key: " + this.publicKey, e); } finally { IOUtils.closeQuietly(publicKeyReader); } final File outputFile = getOutputFile(); InputStream srcFileInputStream = null; OutputStream encryptedFileOutputStream = null; try { srcFileInputStream = new BufferedInputStream(this.srcFile.getInputStream()); encryptedFileOutputStream = new BufferedOutputStream(new FileOutputStream(outputFile)); this.log("Encrypting '" + this.srcFile + "' to '" + outputFile + "'"); final String fileName = FilenameUtils.getName(this.srcFile.getName()); final long size = this.srcFile.getSize(); fileEncrypter.encrypt(fileName, (int) size, srcFileInputStream, encryptedFileOutputStream); } catch (IOException e) { throw new BuildException("Failed to encrypt file from '" + this.srcFile + "' to '" + outputFile + "'", e); } catch (InvalidCipherTextException e) { throw new BuildException("Public key '" + this.publicKey + "' is invalid", e); } finally { IOUtils.closeQuietly(srcFileInputStream); IOUtils.closeQuietly(encryptedFileOutputStream); } } @Override String getFileName(String srcFileName) { return srcFileName + ".tar"; } }