| /* |
| * 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 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.CpioConstants; |
| import org.apache.commons.compress.archivers.cpio.CpioArchiveInputStream; |
| import org.apache.commons.compress.archivers.cpio.CpioArchiveOutputStream; |
| import org.apache.commons.compress.utils.IOUtils; |
| |
| public final class CpioTestCase extends AbstractTestCase { |
| |
| 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(); |
| } |
| |
| 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); |
| |
| |
| Map<String, File> result = new HashMap<String, File>(); |
| ArchiveEntry entry = null; |
| while ((entry = in.getNextEntry()) != null) { |
| 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()); |
| } |
| |
| public void testDirectoryEntryFromFile() throws Exception { |
| 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)); |
| long beforeArchiveWrite = tmp[0].lastModified(); |
| CpioArchiveEntry in = new CpioArchiveEntry(tmp[0], "foo"); |
| tos.putArchiveEntry(in); |
| tos.closeArchiveEntry(); |
| tos.close(); |
| tos = null; |
| tis = new CpioArchiveInputStream(new FileInputStream(archive)); |
| 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]); |
| } |
| } |
| |
| public void testExplicitDirectoryEntry() throws Exception { |
| 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)); |
| long beforeArchiveWrite = tmp[0].lastModified(); |
| 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)); |
| 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]); |
| } |
| } |
| |
| public void testFileEntryFromFile() throws Exception { |
| 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)); |
| CpioArchiveEntry in = new CpioArchiveEntry(tmp[1], "foo"); |
| tos.putArchiveEntry(in); |
| 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)); |
| 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]); |
| } |
| } |
| |
| public void testExplicitFileEntry() throws Exception { |
| 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)); |
| CpioArchiveEntry in = new CpioArchiveEntry("foo"); |
| in.setTime(tmp[1].lastModified() / 1000); |
| in.setSize(tmp[1].length()); |
| in.setMode(CpioConstants.C_ISREG); |
| tos.putArchiveEntry(in); |
| 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)); |
| 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]); |
| } |
| } |
| } |