/* ***** BEGIN LICENSE BLOCK *****
* Version: MPL 1.1/GPL 2.0/LGPL 2.1
*
* The contents of this file are subject to the Mozilla Public License Version
* 1.1 (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.mozilla.org/MPL/
*
* Software distributed under the License is distributed on an "AS IS" basis,
* WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License
* for the specific language governing rights and limitations under the
* License.
*
* The Original Code is part of dcm4che, an implementation of DICOM(TM) in
* Java(TM), hosted at https://github.com/gunterze/dcm4che.
*
* The Initial Developer of the Original Code is
* Agfa Healthcare.
* Portions created by the Initial Developer are Copyright (C) 2011-2014
* the Initial Developer. All Rights Reserved.
*
* Contributor(s):
* See @authors listed below
*
* Alternatively, the contents of this file may be used under the terms of
* either the GNU General Public License Version 2 or later (the "GPL"), or
* the GNU Lesser General Public License Version 2.1 or later (the "LGPL"),
* in which case the provisions of the GPL or the LGPL are applicable instead
* of those above. If you wish to allow use of your version of this file only
* under the terms of either the GPL or the LGPL, and not to allow others to
* use your version of this file under the terms of the MPL, indicate your
* decision by deleting the provisions above and replace them with the notice
* and other provisions required by the GPL or the LGPL. If you do not delete
* the provisions above, a recipient may use your version of this file under
* the terms of any one of the MPL, the GPL or the LGPL.
*
* ***** END LICENSE BLOCK ***** */
package org.dcm4chee.storage.zip;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.nio.file.Files;
import java.nio.file.Path;
import java.security.DigestInputStream;
import java.security.MessageDigest;
import java.security.NoSuchAlgorithmException;
import java.util.Arrays;
import java.util.List;
import java.util.Map;
import java.util.zip.CRC32;
import java.util.zip.ZipEntry;
import java.util.zip.ZipInputStream;
import java.util.zip.ZipOutputStream;
import javax.enterprise.context.Dependent;
import javax.inject.Named;
import org.dcm4chee.storage.ContainerEntry;
import org.dcm4chee.storage.ExtractTask;
import org.dcm4chee.storage.ObjectNotFoundException;
import org.dcm4chee.storage.RetrieveContext;
import org.dcm4chee.storage.StorageContext;
import org.dcm4chee.storage.ChecksumException;
import org.dcm4chee.storage.conf.Container;
import org.dcm4chee.storage.spi.ContainerProvider;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
/**
* @author Gunter Zeilinger <gunterze@gmail.com>
*
*/
@Named("org.dcm4chee.storage.zip")
@Dependent
public class ZipContainerProvider implements ContainerProvider {
private static final Logger LOG = LoggerFactory
.getLogger(ZipContainerProvider.class);
private Container container;
@Override
public void init(Container container) {
this.container = container;
}
@Override
public void writeEntriesTo(StorageContext ctx,
List<ContainerEntry> entries, OutputStream out) throws IOException {
ZipOutputStream zip = new ZipOutputStream(out);
boolean compress = container.isCompress();
String checksumEntry = container.getChecksumEntry();
if (checksumEntry != null) {
ZipEntry zipEntry = new ZipEntry(checksumEntry);
if (!compress) {
CRC32OutputStream crc32 = new CRC32OutputStream();
ContainerEntry.writeChecksumsTo(entries, crc32);
crc32.updateEntry(zipEntry);
}
zip.putNextEntry(zipEntry);
ContainerEntry.writeChecksumsTo(entries, zip);
zip.closeEntry();
}
for (ContainerEntry entry : entries) {
Path path = entry.getSourcePath();
ZipEntry zipEntry = new ZipEntry(entry.getName());
zipEntry.setTime(Files.getLastModifiedTime(path).toMillis());
if (!compress) {
CRC32OutputStream crc32 = new CRC32OutputStream();
Files.copy(path, crc32);
crc32.updateEntry(zipEntry);
}
zip.putNextEntry(zipEntry);
Files.copy(path, zip);
zip.closeEntry();
}
zip.finish();
}
@Override
public InputStream seekEntry(RetrieveContext ctx, String name,
String entryName, InputStream in) throws IOException {
ZipInputStream zip = new ZipInputStream(in);
String checksumEntry = container.getChecksumEntry();
ZipEntry nextEntry;
while ((nextEntry = zip.getNextEntry()) != null) {
String nextEntryName = nextEntry.getName();
if (nextEntry.isDirectory() || nextEntryName.equals(checksumEntry))
continue;
if (nextEntryName.equals(entryName))
return zip;
}
throw new ObjectNotFoundException(ctx.getStorageSystem()
.getStorageSystemPath(), name, entryName);
}
@Override
public void extractEntries(RetrieveContext ctx, String name,
ExtractTask extractTask, InputStream in) throws IOException {
ZipInputStream zip = new ZipInputStream(in);
ZipEntry entry = skipDirectoryEntries(zip);
if (entry == null)
throw new IOException("No entries in " + name);
String entryName = entry.getName();
Map<String, byte[]> checksums = null;
String checksumEntry = container.getChecksumEntry();
MessageDigest digest = null;
if (checksumEntry != null) {
if (checksumEntry.equals(entryName)) {
try {
digest = MessageDigest.getInstance(ctx.getStorageSystem().getStorageSystemGroup()
.getDigestAlgorithm());
} catch (NoSuchAlgorithmException e) {
throw new RuntimeException(e);
}
checksums = ContainerEntry.readChecksumsFrom(zip);
} else
LOG.warn("Misssing checksum entry in {}", name);
entry = skipDirectoryEntries(zip);
}
for (; entry != null; entry = skipDirectoryEntries(zip)) {
entryName = entry.getName();
InputStream tmp = zip;
byte[] checksum = null;
if (checksums != null && digest != null) {
checksum = checksums.remove(entryName);
if (checksum == null)
throw new ChecksumException(
"Missing checksum for container entry: "
+ entryName + " in " + name);
digest.reset();
tmp = new DigestInputStream(zip, digest);
}
extractTask.copyStream(entryName, tmp);
if (checksums != null && digest != null) {
if (!Arrays.equals(digest.digest(), checksum)) {
throw new ChecksumException(
"Checksums do not match for container entry: "
+ entry.getName() + " in " + name);
}
}
extractTask.entryExtracted(entryName);
}
}
private ZipEntry skipDirectoryEntries(ZipInputStream zip)
throws IOException {
for (ZipEntry entry = zip.getNextEntry(); entry != null; entry = zip
.getNextEntry()) {
if (!entry.isDirectory())
return entry;
}
return null;
}
private static class CRC32OutputStream extends java.io.OutputStream {
final CRC32 crc = new CRC32();
long n = 0;
CRC32OutputStream() {}
public void write(int r) throws IOException {
crc.update(r);
n++;
}
public void write(byte[] b, int off, int len) throws IOException {
crc.update(b, off, len);
n += len;
}
public void updateEntry(ZipEntry e) {
e.setMethod(ZipEntry.STORED);
e.setSize(n);
e.setCrc(crc.getValue());
}
}
}