Zip archive: Handle files within folders inside the zip

Signed-off-by: Roopesh Chander <roop@roopc.net>

1 files changed, 4 insertions, 4 deletions

diff --git a/WireGuard/WireGuard/ZipArchive/ZipArchive.swift b/WireGuard/WireGuard/ZipArchive/ZipArchive.swift
index 7f654ad..d907803 100644
--- a/WireGuard/WireGuard/ZipArchive/ZipArchive.swift
+++ b/WireGuard/WireGuard/ZipArchive/ZipArchive.swift
@@ -61,10 +61,10 @@ class ZipArchive {
                 throw ZipArchiveError.badArchive
             }
 
-            let fileName = String(cString: fileNameBuffer)
-            let fileExtension = URL(string: fileName)?.pathExtension ?? ""
+            if let fileURL = URL(string: String(cString: fileNameBuffer)),
+                !fileURL.hasDirectoryPath,
+                requiredFileExtensions.contains(fileURL.pathExtension) {
 
-            if (requiredFileExtensions.contains(fileExtension)) {
                 var unzippedData = Data()
                 var bytesRead: Int32 = 0
                 repeat {
@@ -77,7 +77,7 @@ class ZipArchive {
                         unzippedData.append(dataRead)
                     }
                 } while (bytesRead > 0)
-                results.append((fileName: fileName, contents: unzippedData))
+                results.append((fileName: fileURL.lastPathComponent, contents: unzippedData))
             }
 
             guard (unzCloseCurrentFile(zipFile) == UNZ_OK) else {