WireGuard/WireGuard/ZipArchive/ZipImporter.swift


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

// SPDX-License-Identifier: MIT
// Copyright © 2018 WireGuard LLC. All Rights Reserved.

import UIKit

enum ZipImporterError: Error {
    case noTunnelsInZipArchive
}

class ZipImporter {
    static func importConfigFiles(from url: URL, completion: @escaping ([TunnelConfiguration?], Error?) -> Void) {
        DispatchQueue.global(qos: .userInitiated).async {
            var unarchivedFiles: [(fileName: String, contents: Data)]
            do {
                unarchivedFiles = try ZipArchive.unarchive(url: url, requiredFileExtensions: ["conf"])

                for (i, unarchivedFile) in unarchivedFiles.enumerated().reversed() {
                    let fileBaseName = URL(string: unarchivedFile.fileName)?.deletingPathExtension().lastPathComponent
                    if let trimmedName = fileBaseName?.trimmingCharacters(in: .whitespacesAndNewlines), !trimmedName.isEmpty {
                        unarchivedFiles[i].fileName = trimmedName
                    } else {
                        unarchivedFiles.remove(at: i)
                    }
                }

                if (unarchivedFiles.isEmpty) {
                    throw ZipImporterError.noTunnelsInZipArchive
                }
            } catch (let error) {
                DispatchQueue.main.async { completion([], error) }
                return
            }

            unarchivedFiles.sort { $0.fileName < $1.fileName }
            var configs = Array<TunnelConfiguration?>(repeating: nil, count: unarchivedFiles.count)
            for (i, file) in unarchivedFiles.enumerated() {
                if (i > 0 && file == unarchivedFiles[i - 1]) {
                    continue
                }
                guard let fileContents = String(data: file.contents, encoding: .utf8) else {
                    continue
                }
                guard let tunnelConfig = try? WgQuickConfigFileParser.parse(fileContents, name: file.fileName) else {
                    continue
                }
                configs[i] = tunnelConfig
            }
            DispatchQueue.main.async { completion(configs, nil) }
        }
    }
}