/*
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF 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 org.apache.commons.compress.archivers;
import static org.junit.Assert.*;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.InputStream;
import java.io.OutputStream;
import java.util.HashMap;
import java.util.Map;
import org.apache.commons.compress.AbstractTestCase;
import org.apache.commons.compress.archivers.cpio.CpioArchiveEntry;
import org.apache.commons.compress.archivers.cpio.CpioArchiveInputStream;
import org.apache.commons.compress.archivers.cpio.CpioArchiveOutputStream;
import org.apache.commons.compress.archivers.cpio.CpioConstants;
import org.apache.commons.compress.utils.IOUtils;
import org.junit.Test;
public final class CpioTestCase extends AbstractTestCase {
@Test
public void testCpioArchiveCreation() throws Exception {
final File output = new File(dir, "bla.cpio");
final File file1 = getFile("test1.xml");
final File file2 = getFile("test2.xml");
final OutputStream out = new FileOutputStream(output);
final ArchiveOutputStream os = new ArchiveStreamFactory().createArchiveOutputStream("cpio", out);
os.putArchiveEntry(new CpioArchiveEntry("test1.xml", file1.length()));
IOUtils.copy(new FileInputStream(file1), os);
os.closeArchiveEntry();
os.putArchiveEntry(new CpioArchiveEntry("test2.xml", file2.length()));
IOUtils.copy(new FileInputStream(file2), os);
os.closeArchiveEntry();
os.close();
out.close();
}
@Test
public void testCpioUnarchive() throws Exception {
final File output = new File(dir, "bla.cpio");
final File file1 = getFile("test1.xml");
final File file2 = getFile("test2.xml");
final long file1Length = file1.length();
final long file2Length = file2.length();
{
final OutputStream out = new FileOutputStream(output);
final ArchiveOutputStream os = new ArchiveStreamFactory().createArchiveOutputStream("cpio", out);
CpioArchiveEntry entry = new CpioArchiveEntry("test1.xml", file1Length);
entry.setMode(CpioConstants.C_ISREG);
os.putArchiveEntry(entry);
IOUtils.copy(new FileInputStream(file1), os);
os.closeArchiveEntry();
entry = new CpioArchiveEntry("test2.xml", file2Length);
entry.setMode(CpioConstants.C_ISREG);
os.putArchiveEntry(entry);
IOUtils.copy(new FileInputStream(file2), os);
os.closeArchiveEntry();
os.finish();
os.close();
out.close();
}
// Unarchive Operation
final File input = output;
final InputStream is = new FileInputStream(input);
final ArchiveInputStream in = new ArchiveStreamFactory().createArchiveInputStream("cpio", is);
final Map<String, File> result = new HashMap<>();
ArchiveEntry entry = null;
while ((entry = in.getNextEntry()) != null) {
final File cpioget = new File(dir, entry.getName());
final OutputStream out = new FileOutputStream(cpioget);
IOUtils.copy(in, out);
out.close();
result.put(entry.getName(), cpioget);
}
in.close();
is.close();
File t = result.get("test1.xml");
assertTrue("Expected " + t.getAbsolutePath() + " to exist", t.exists());
assertEquals("length of " + t.getAbsolutePath(), file1Length, t.length());
t = result.get("test2.xml");
assertTrue("Expected " + t.getAbsolutePath() + " to exist", t.exists());
assertEquals("length of " + t.getAbsolutePath(), file2Length, t.length());
}
@Test
public void testDirectoryEntryFromFile() throws Exception {
final File[] tmp = createTempDirAndFile();
File archive = null;
CpioArchiveOutputStream tos = null;
CpioArchiveInputStream tis = null;
try {
archive = File.createTempFile("test.", ".cpio", tmp[0]);
archive.deleteOnExit();
tos = new CpioArchiveOutputStream(new FileOutputStream(archive));
final long beforeArchiveWrite = tmp[0].lastModified();
final CpioArchiveEntry in = new CpioArchiveEntry(tmp[0], "foo");
tos.putArchiveEntry(in);
tos.closeArchiveEntry();
tos.close();
tos = null;
tis = new CpioArchiveInputStream(new FileInputStream(archive));
final CpioArchiveEntry out = tis.getNextCPIOEntry();
tis.close();
tis = null;
assertNotNull(out);
assertEquals("foo", out.getName());
assertEquals(0, out.getSize());
// CPIO stores time with a granularity of 1 second
assertEquals(beforeArchiveWrite / 1000,
out.getLastModifiedDate().getTime() / 1000);
assertTrue(out.isDirectory());
} finally {
if (tis != null) {
tis.close();
}
if (tos != null) {
tos.close();
}
tryHardToDelete(archive);
tryHardToDelete(tmp[1]);
rmdir(tmp[0]);
}
}
@Test
public void testExplicitDirectoryEntry() throws Exception {
final File[] tmp = createTempDirAndFile();
File archive = null;
CpioArchiveOutputStream tos = null;
CpioArchiveInputStream tis = null;
try {
archive = File.createTempFile("test.", ".cpio", tmp[0]);
archive.deleteOnExit();
tos = new CpioArchiveOutputStream(new FileOutputStream(archive));
final long beforeArchiveWrite = tmp[0].lastModified();
final CpioArchiveEntry in = new CpioArchiveEntry("foo/");
in.setTime(beforeArchiveWrite / 1000);
in.setMode(CpioConstants.C_ISDIR);
tos.putArchiveEntry(in);
tos.closeArchiveEntry();
tos.close();
tos = null;
tis = new CpioArchiveInputStream(new FileInputStream(archive));
final CpioArchiveEntry out = tis.getNextCPIOEntry();
tis.close();
tis = null;
assertNotNull(out);
assertEquals("foo/", out.getName());
assertEquals(0, out.getSize());
assertEquals(beforeArchiveWrite / 1000,
out.getLastModifiedDate().getTime() / 1000);
assertTrue(out.isDirectory());
} finally {
if (tis != null) {
tis.close();
}
if (tos != null) {
tos.close();
}
tryHardToDelete(archive);
tryHardToDelete(tmp[1]);
rmdir(tmp[0]);
}
}
@Test
public void testFileEntryFromFile() throws Exception {
final File[] tmp = createTempDirAndFile();
File archive = null;
CpioArchiveOutputStream tos = null;
CpioArchiveInputStream tis = null;
FileInputStream fis = null;
try {
archive = File.createTempFile("test.", ".cpio", tmp[0]);
archive.deleteOnExit();
tos = new CpioArchiveOutputStream(new FileOutputStream(archive));
final CpioArchiveEntry in = new CpioArchiveEntry(tmp[1], "foo");
tos.putArchiveEntry(in);
final byte[] b = new byte[(int) tmp[1].length()];
fis = new FileInputStream(tmp[1]);
while (fis.read(b) > 0) {
tos.write(b);
}
fis.close();
fis = null;
tos.closeArchiveEntry();
tos.close();
tos = null;
tis = new CpioArchiveInputStream(new FileInputStream(archive));
final CpioArchiveEntry out = tis.getNextCPIOEntry();
tis.close();
tis = null;
assertNotNull(out);
assertEquals("foo", out.getName());
assertEquals(tmp[1].length(), out.getSize());
assertEquals(tmp[1].lastModified() / 1000,
out.getLastModifiedDate().getTime() / 1000);
assertFalse(out.isDirectory());
} finally {
if (tis != null) {
tis.close();
}
if (tos != null) {
tos.close();
}
tryHardToDelete(archive);
if (fis != null) {
fis.close();
}
tryHardToDelete(tmp[1]);
rmdir(tmp[0]);
}
}
@Test
public void testExplicitFileEntry() throws Exception {
final File[] tmp = createTempDirAndFile();
File archive = null;
CpioArchiveOutputStream tos = null;
CpioArchiveInputStream tis = null;
FileInputStream fis = null;
try {
archive = File.createTempFile("test.", ".cpio", tmp[0]);
archive.deleteOnExit();
tos = new CpioArchiveOutputStream(new FileOutputStream(archive));
final CpioArchiveEntry in = new CpioArchiveEntry("foo");
in.setTime(tmp[1].lastModified() / 1000);
in.setSize(tmp[1].length());
in.setMode(CpioConstants.C_ISREG);
tos.putArchiveEntry(in);
final byte[] b = new byte[(int) tmp[1].length()];
fis = new FileInputStream(tmp[1]);
while (fis.read(b) > 0) {
tos.write(b);
}
fis.close();
fis = null;
tos.closeArchiveEntry();
tos.close();
tos = null;
tis = new CpioArchiveInputStream(new FileInputStream(archive));
final CpioArchiveEntry out = tis.getNextCPIOEntry();
tis.close();
tis = null;
assertNotNull(out);
assertEquals("foo", out.getName());
assertEquals(tmp[1].length(), out.getSize());
assertEquals(tmp[1].lastModified() / 1000,
out.getLastModifiedDate().getTime() / 1000);
assertFalse(out.isDirectory());
} finally {
if (tis != null) {
tis.close();
}
if (tos != null) {
tos.close();
}
tryHardToDelete(archive);
if (fis != null) {
fis.close();
}
tryHardToDelete(tmp[1]);
rmdir(tmp[0]);
}
}
}