PEKit/Tests/PEKitTests/ImportObjectFileTests.swift at main · p-x9/PEKit · GitHub

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
import Foundation
import XCTest
@testable import PEKit
import PEArchiveKit

final class ImportObjectFileTests: XCTestCase {
    var importObject: ImportObjectFile!

    override func setUp() {
        let path = "/path/to/archive-file"
        let url = URL(fileURLWithPath: path)
        let archive = try! ArchiveFile(url: url)
        importObject = try! archive.importObjectFiles().first!
    }
}

extension ImportObjectFileTests {
    func testHeaderAndStrings() throws {
        printHeader(importObject)
        printPayload(importObject)
        printStrings(importObject)
        printNames(importObject)
    }

    func testNames() throws {
        printHeader(importObject)
        printNames(importObject)
    }
}

extension ImportObjectFileTests {
    private func printHeader(_ importObject: ImportObjectFile) {
        let header = importObject.header

        print(
            """
            ImportObjectHeader {
              Version: \(header.version)
              Machine: \(header.machine?.description ?? "unknown") (0x\(String(header.layout.Machine, radix: 16)))
              TimeDateStamp: \(Date(timeIntervalSince1970: header.timeDateStamp)) (0x\(String(header.layout.TimeDateStamp, radix: 16)))
              SizeOfData: \(header.sizeOfData)
              Ordinal: \(header.ordinal)
              Hint: \(header.hint)
              Type: \(header.type?.description ?? "unknown") (0x\(String(header.layout.Type, radix: 16)))
              NameType: \(header.nameType?.description ?? "unknown") (0x\(String(header.layout.NameType, radix: 16)))
              Reserved: \(header.reserved)
            }
            """
        )
    }

    private func printPayload(_ importObject: ImportObjectFile) {
        if let data = importObject.data {
            print(
                """
                Payload {
                  Size: \(data.count)
                  Bytes: \(data.map { String(format: "%02X", $0) }.joined(separator: " "))
                }
                """
            )
        } else {
            print("Payload: nil")
        }
    }

    private func printStrings(_ importObject: ImportObjectFile) {
        print("Strings [")
        for (i, string) in importObject.strings.enumerated() {
            print("  \(i): \(string)")
        }
        print("]")
    }

    private func printNames(_ importObject: ImportObjectFile) {
        print(
            """
            ImportNames {
              SymbolName: \(importObject.symbolName ?? "nil")
              ImportName: \(importObject.importName ?? "nil")
              DLLName: \(importObject.dllName ?? "nil")
              ExportAsName: \(importObject.exportAsName ?? "nil")
              ExportName: \(importObject.exportName ?? "nil")
            }
            """
        )
    }
}