1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
|
/*
* Copyright 2023 Code Intelligence GmbH
*
* Licensed 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 com.code_intelligence.jazzer.utils;
import java.io.BufferedInputStream;
import java.io.BufferedOutputStream;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.lang.IllegalArgumentException;
import java.nio.file.FileVisitResult;
import java.nio.file.Files;
import java.nio.file.Path;
import java.nio.file.Paths;
import java.nio.file.SimpleFileVisitor;
import java.nio.file.attribute.BasicFileAttributes;
import java.util.ArrayList;
import java.util.Enumeration;
import java.util.HashSet;
import java.util.List;
import java.util.Set;
import java.util.jar.JarEntry;
import java.util.jar.JarFile;
import java.util.stream.Stream;
import java.util.zip.ZipEntry;
import java.util.zip.ZipInputStream;
import java.util.zip.ZipOutputStream;
public final class ZipUtils {
private ZipUtils() {}
public static Set<String> mergeZipToZip(String src, ZipOutputStream zos, Set<String> skipFiles)
throws IOException {
HashSet<String> filesAdded = new HashSet<>();
try (JarFile jarFile = new JarFile(src)) {
// Copy entries from src to dst (jarFile to ZipOutputStream)
Enumeration<JarEntry> allEntries = jarFile.entries();
while (allEntries.hasMoreElements()) {
JarEntry entry = allEntries.nextElement();
if (skipFiles != null && skipFiles.contains(entry.getName())) {
continue;
}
zos.putNextEntry(new ZipEntry(entry.getName()));
try (InputStream is = jarFile.getInputStream(entry)) {
byte[] buf = new byte[1024];
int i = 0;
while ((i = is.read(buf)) != -1) {
zos.write(buf, 0, i);
}
zos.closeEntry();
filesAdded.add(entry.getName());
}
}
}
return filesAdded;
}
public static Set<String> mergeDirectoryToZip(String src, ZipOutputStream zos,
Set<String> skipFiles) throws IllegalArgumentException, IOException {
HashSet<String> filesAdded = new HashSet<>();
File sourceDir = new File(src);
if (!sourceDir.isDirectory()) {
throw new IllegalArgumentException("Argument src must be a directory.");
}
Files.walkFileTree(sourceDir.toPath(), new SimpleFileVisitor<Path>() {
public FileVisitResult visitFile(Path file, BasicFileAttributes attrs) throws IOException {
String zipPath = sourceDir.toPath().relativize(file).toString();
if (skipFiles.stream().anyMatch(zipPath::endsWith)) {
return FileVisitResult.CONTINUE;
}
zos.putNextEntry(new ZipEntry(zipPath));
Files.copy(file, zos);
filesAdded.add(zipPath);
return FileVisitResult.CONTINUE;
}
});
return filesAdded;
}
public static void extractFile(String srcZip, String targetFile, String outputFilePath)
throws IOException {
try (OutputStream out = new FileOutputStream(outputFilePath);
ZipInputStream zis = new ZipInputStream(new FileInputStream(srcZip));) {
ZipEntry ze = zis.getNextEntry();
while (ze != null) {
if (ze.getName().equals(targetFile)) {
byte[] buf = new byte[1024];
int read = 0;
while ((read = zis.read(buf)) > -1) {
out.write(buf, 0, read);
}
out.close();
break;
}
ze = zis.getNextEntry();
}
}
}
}
|