| /* |
| * 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.assertEquals; |
| import static org.junit.Assert.assertFalse; |
| import static org.junit.Assert.assertNotNull; |
| import static org.junit.Assert.assertTrue; |
| |
| import java.io.BufferedInputStream; |
| import java.io.ByteArrayOutputStream; |
| import java.io.File; |
| import java.io.InputStream; |
| import java.io.OutputStream; |
| import java.nio.file.Files; |
| |
| import org.apache.commons.compress.AbstractTestCase; |
| import org.apache.commons.compress.archivers.ar.ArArchiveEntry; |
| import org.apache.commons.compress.archivers.ar.ArArchiveInputStream; |
| import org.apache.commons.compress.archivers.ar.ArArchiveOutputStream; |
| import org.apache.commons.compress.utils.IOUtils; |
| import org.junit.Ignore; |
| import org.junit.Test; |
| |
| public final class ArTestCase extends AbstractTestCase { |
| |
| @Test |
| public void testArArchiveCreation() throws Exception { |
| final File output = new File(dir, "bla.ar"); |
| |
| final File file1 = getFile("test1.xml"); |
| final File file2 = getFile("test2.xml"); |
| |
| final OutputStream out = Files.newOutputStream(output.toPath()); |
| final ArchiveOutputStream os = ArchiveStreamFactory.DEFAULT.createArchiveOutputStream("ar", out); |
| os.putArchiveEntry(new ArArchiveEntry("test1.xml", file1.length())); |
| IOUtils.copy(Files.newInputStream(file1.toPath()), os); |
| os.closeArchiveEntry(); |
| |
| os.putArchiveEntry(new ArArchiveEntry("test2.xml", file2.length())); |
| IOUtils.copy(Files.newInputStream(file2.toPath()), os); |
| os.closeArchiveEntry(); |
| |
| os.close(); |
| } |
| |
| @Test |
| public void testArUnarchive() throws Exception { |
| final File output = new File(dir, "bla.ar"); |
| { |
| final File file1 = getFile("test1.xml"); |
| final File file2 = getFile("test2.xml"); |
| |
| final OutputStream out = Files.newOutputStream(output.toPath()); |
| final ArchiveOutputStream os = ArchiveStreamFactory.DEFAULT.createArchiveOutputStream("ar", out); |
| os.putArchiveEntry(new ArArchiveEntry("test1.xml", file1.length())); |
| IOUtils.copy(Files.newInputStream(file1.toPath()), os); |
| os.closeArchiveEntry(); |
| |
| os.putArchiveEntry(new ArArchiveEntry("test2.xml", file2.length())); |
| IOUtils.copy(Files.newInputStream(file2.toPath()), os); |
| os.closeArchiveEntry(); |
| os.close(); |
| out.close(); |
| } |
| |
| // UnArArchive Operation |
| try (final InputStream is = Files.newInputStream(output.toPath()); |
| final ArchiveInputStream in = new ArchiveStreamFactory() |
| .createArchiveInputStream(new BufferedInputStream(is))) { |
| final ArArchiveEntry entry = (ArArchiveEntry) in.getNextEntry(); |
| |
| final File target = new File(dir, entry.getName()); |
| try (final OutputStream out = Files.newOutputStream(target.toPath())) { |
| IOUtils.copy(in, out); |
| } |
| } |
| } |
| |
| @Test |
| public void testArDelete() throws Exception { |
| final File output = new File(dir, "bla.ar"); |
| |
| final File file1 = getFile("test1.xml"); |
| final File file2 = getFile("test2.xml"); |
| { |
| // create |
| |
| final OutputStream out = Files.newOutputStream(output.toPath()); |
| final ArchiveOutputStream os = ArchiveStreamFactory.DEFAULT.createArchiveOutputStream("ar", out); |
| os.putArchiveEntry(new ArArchiveEntry("test1.xml", file1.length())); |
| IOUtils.copy(Files.newInputStream(file1.toPath()), os); |
| os.closeArchiveEntry(); |
| |
| os.putArchiveEntry(new ArArchiveEntry("test2.xml", file2.length())); |
| IOUtils.copy(Files.newInputStream(file2.toPath()), os); |
| os.closeArchiveEntry(); |
| os.close(); |
| out.close(); |
| } |
| |
| assertEquals(8 |
| + 60 + file1.length() + (file1.length() % 2) |
| + 60 + file2.length() + (file2.length() % 2), |
| output.length()); |
| |
| final File output2 = new File(dir, "bla2.ar"); |
| |
| int copied = 0; |
| int deleted = 0; |
| |
| { |
| // remove all but one file |
| |
| final InputStream is = Files.newInputStream(output.toPath()); |
| final OutputStream os = Files.newOutputStream(output2.toPath()); |
| final ArchiveOutputStream aos = new ArchiveStreamFactory().createArchiveOutputStream("ar", os); |
| final ArchiveInputStream ais = new ArchiveStreamFactory().createArchiveInputStream(new BufferedInputStream(is)); |
| while(true) { |
| final ArArchiveEntry entry = (ArArchiveEntry)ais.getNextEntry(); |
| if (entry == null) { |
| break; |
| } |
| |
| if ("test1.xml".equals(entry.getName())) { |
| aos.putArchiveEntry(entry); |
| IOUtils.copy(ais, aos); |
| aos.closeArchiveEntry(); |
| copied++; |
| } else { |
| IOUtils.copy(ais, new ByteArrayOutputStream()); |
| deleted++; |
| } |
| |
| } |
| ais.close(); |
| aos.close(); |
| is.close(); |
| os.close(); |
| } |
| |
| assertEquals(1, copied); |
| assertEquals(1, deleted); |
| assertEquals(8 |
| + 60 + file1.length() + (file1.length() % 2), |
| output2.length()); |
| |
| long files = 0; |
| long sum = 0; |
| |
| { |
| final InputStream is = Files.newInputStream(output2.toPath()); |
| final ArchiveInputStream ais = new ArchiveStreamFactory().createArchiveInputStream(new BufferedInputStream(is)); |
| while(true) { |
| final ArArchiveEntry entry = (ArArchiveEntry)ais.getNextEntry(); |
| if (entry == null) { |
| break; |
| } |
| |
| IOUtils.copy(ais, new ByteArrayOutputStream()); |
| |
| sum += entry.getLength(); |
| files++; |
| } |
| ais.close(); |
| is.close(); |
| } |
| |
| assertEquals(1, files); |
| assertEquals(file1.length(), sum); |
| |
| } |
| |
| // TODO: revisit - does AR not support storing directories? |
| @Ignore |
| @Test |
| public void XtestDirectoryEntryFromFile() throws Exception { |
| final File[] tmp = createTempDirAndFile(); |
| File archive = null; |
| ArArchiveOutputStream aos = null; |
| ArArchiveInputStream ais = null; |
| try { |
| archive = File.createTempFile("test.", ".ar", tmp[0]); |
| archive.deleteOnExit(); |
| aos = new ArArchiveOutputStream(Files.newOutputStream(archive.toPath())); |
| final long beforeArchiveWrite = tmp[0].lastModified(); |
| final ArArchiveEntry in = new ArArchiveEntry(tmp[0], "foo"); |
| aos.putArchiveEntry(in); |
| aos.closeArchiveEntry(); |
| aos.close(); |
| aos = null; |
| ais = new ArArchiveInputStream(Files.newInputStream(archive.toPath())); |
| final ArArchiveEntry out = ais.getNextArEntry(); |
| ais.close(); |
| ais = null; |
| assertNotNull(out); |
| assertEquals("foo/", out.getName()); |
| assertEquals(0, out.getSize()); |
| // AR stores time with a granularity of 1 second |
| assertEquals(beforeArchiveWrite / 1000, |
| out.getLastModifiedDate().getTime() / 1000); |
| assertTrue(out.isDirectory()); |
| } finally { |
| if (ais != null) { |
| ais.close(); |
| } |
| if (aos != null) { |
| aos.close(); |
| } |
| tryHardToDelete(archive); |
| tryHardToDelete(tmp[1]); |
| rmdir(tmp[0]); |
| } |
| } |
| |
| // TODO: revisit - does AR not support storing directories? |
| @Ignore |
| @Test |
| public void XtestExplicitDirectoryEntry() throws Exception { |
| final File[] tmp = createTempDirAndFile(); |
| File archive = null; |
| ArArchiveOutputStream aos = null; |
| ArArchiveInputStream ais = null; |
| try { |
| archive = File.createTempFile("test.", ".ar", tmp[0]); |
| archive.deleteOnExit(); |
| aos = new ArArchiveOutputStream(Files.newOutputStream(archive.toPath())); |
| final long beforeArchiveWrite = tmp[0].lastModified(); |
| final ArArchiveEntry in = new ArArchiveEntry("foo", 0, 0, 0, 0, |
| tmp[1].lastModified() / 1000); |
| aos.putArchiveEntry(in); |
| aos.closeArchiveEntry(); |
| aos.close(); |
| aos = null; |
| ais = new ArArchiveInputStream(Files.newInputStream(archive.toPath())); |
| final ArArchiveEntry out = ais.getNextArEntry(); |
| ais.close(); |
| ais = null; |
| assertNotNull(out); |
| assertEquals("foo/", out.getName()); |
| assertEquals(0, out.getSize()); |
| assertEquals(beforeArchiveWrite / 1000, |
| out.getLastModifiedDate().getTime() / 1000); |
| assertTrue(out.isDirectory()); |
| } finally { |
| if (ais != null) { |
| ais.close(); |
| } |
| if (aos != null) { |
| aos.close(); |
| } |
| tryHardToDelete(archive); |
| tryHardToDelete(tmp[1]); |
| rmdir(tmp[0]); |
| } |
| } |
| |
| @Test |
| public void testFileEntryFromFile() throws Exception { |
| final File[] tmp = createTempDirAndFile(); |
| File archive = null; |
| ArArchiveOutputStream aos = null; |
| ArArchiveInputStream ais = null; |
| InputStream fis = null; |
| final File directory = tmp[0]; |
| final File file = tmp[1]; |
| try { |
| archive = File.createTempFile("test.", ".ar", directory); |
| archive.deleteOnExit(); |
| aos = new ArArchiveOutputStream(Files.newOutputStream(archive.toPath())); |
| final ArArchiveEntry in = new ArArchiveEntry(file, "foo"); |
| aos.putArchiveEntry(in); |
| final byte[] b = new byte[(int) file.length()]; |
| fis = Files.newInputStream(file.toPath()); |
| while (fis.read(b) > 0) { |
| aos.write(b); |
| } |
| fis.close(); |
| fis = null; |
| aos.closeArchiveEntry(); |
| aos.close(); |
| aos = null; |
| ais = new ArArchiveInputStream(Files.newInputStream(archive.toPath())); |
| final ArArchiveEntry out = ais.getNextArEntry(); |
| ais.close(); |
| ais = null; |
| assertNotNull(out); |
| assertEquals("foo", out.getName()); |
| assertEquals(file.length(), out.getSize()); |
| // AR stores time with a granularity of 1 second |
| assertEquals(file.lastModified() / 1000, out.getLastModifiedDate().getTime() / 1000); |
| assertFalse(out.isDirectory()); |
| } finally { |
| if (ais != null) { |
| ais.close(); |
| } |
| if (aos != null) { |
| aos.close(); |
| } |
| tryHardToDelete(archive); |
| if (fis != null) { |
| fis.close(); |
| } |
| tryHardToDelete(file); |
| rmdir(directory); |
| } |
| } |
| |
| @Test |
| public void testFileEntryFromPath() throws Exception { |
| final File[] tmp = createTempDirAndFile(); |
| File archive = null; |
| ArArchiveOutputStream aos = null; |
| ArArchiveInputStream ais = null; |
| InputStream fis = null; |
| final File directory = tmp[0]; |
| final File file = tmp[1]; |
| try { |
| archive = File.createTempFile("test.", ".ar", directory); |
| archive.deleteOnExit(); |
| aos = new ArArchiveOutputStream(Files.newOutputStream(archive.toPath())); |
| final ArArchiveEntry in = new ArArchiveEntry(file.toPath(), "foo"); |
| aos.putArchiveEntry(in); |
| final byte[] b = new byte[(int) file.length()]; |
| fis = Files.newInputStream(file.toPath()); |
| while (fis.read(b) > 0) { |
| aos.write(b); |
| } |
| fis.close(); |
| fis = null; |
| aos.closeArchiveEntry(); |
| aos.close(); |
| aos = null; |
| ais = new ArArchiveInputStream(Files.newInputStream(archive.toPath())); |
| final ArArchiveEntry out = ais.getNextArEntry(); |
| ais.close(); |
| ais = null; |
| assertNotNull(out); |
| assertEquals("foo", out.getName()); |
| assertEquals(file.length(), out.getSize()); |
| // AR stores time with a granularity of 1 second |
| assertEquals(file.lastModified() / 1000, out.getLastModifiedDate().getTime() / 1000); |
| assertFalse(out.isDirectory()); |
| } finally { |
| if (ais != null) { |
| ais.close(); |
| } |
| if (aos != null) { |
| aos.close(); |
| } |
| tryHardToDelete(archive); |
| if (fis != null) { |
| fis.close(); |
| } |
| tryHardToDelete(file); |
| rmdir(directory); |
| } |
| } |
| |
| @Test |
| public void testExplicitFileEntry() throws Exception { |
| final File[] tmp = createTempDirAndFile(); |
| File archive = null; |
| ArArchiveOutputStream aos = null; |
| ArArchiveInputStream ais = null; |
| InputStream fis = null; |
| try { |
| archive = File.createTempFile("test.", ".ar", tmp[0]); |
| archive.deleteOnExit(); |
| aos = new ArArchiveOutputStream(Files.newOutputStream(archive.toPath())); |
| final ArArchiveEntry in = new ArArchiveEntry("foo", tmp[1].length(), |
| 0, 0, 0, |
| tmp[1].lastModified() / 1000); |
| aos.putArchiveEntry(in); |
| final byte[] b = new byte[(int) tmp[1].length()]; |
| fis = Files.newInputStream(tmp[1].toPath()); |
| while (fis.read(b) > 0) { |
| aos.write(b); |
| } |
| fis.close(); |
| fis = null; |
| aos.closeArchiveEntry(); |
| aos.close(); |
| aos = null; |
| ais = new ArArchiveInputStream(Files.newInputStream(archive.toPath())); |
| final ArArchiveEntry out = ais.getNextArEntry(); |
| ais.close(); |
| ais = 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 (ais != null) { |
| ais.close(); |
| } |
| if (aos != null) { |
| aos.close(); |
| } |
| tryHardToDelete(archive); |
| if (fis != null) { |
| fis.close(); |
| } |
| tryHardToDelete(tmp[1]); |
| rmdir(tmp[0]); |
| } |
| } |
| } |