File: ConstructStubExecutorInputFileListTaskAction.swift

package info (click to toggle)
swiftlang 6.2.3-1
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 2,856,264 kB
  • sloc: cpp: 9,995,718; ansic: 2,234,019; asm: 1,092,167; python: 313,940; objc: 82,726; f90: 80,126; lisp: 38,373; pascal: 25,580; sh: 20,378; ml: 5,058; perl: 4,751; makefile: 4,725; awk: 3,535; javascript: 3,018; xml: 918; fortran: 664; cs: 573; ruby: 396
file content (104 lines) | stat: -rw-r--r-- 4,643 bytes parent folder | download
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
//===----------------------------------------------------------------------===//
//
// This source file is part of the Swift open source project
//
// Copyright (c) 2025 Apple Inc. and the Swift project authors
// Licensed under Apache License v2.0 with Runtime Library Exception
//
// See http://swift.org/LICENSE.txt for license information
// See http://swift.org/CONTRIBUTORS.txt for the list of Swift project authors
//
//===----------------------------------------------------------------------===//

public import SWBCore
import SWBLibc
import SWBUtil
public import SWBLLBuild
import Foundation

/// rdar://125894897 (🚨 fetchOperationServiceEndpoint seems completely broken for app extensions implemented in Swift (SwiftUI: Swift entry point data not found.))
///
/// We need to the final stub executor to contain the `__swift5_entry` trampoline to
/// jump to the debug dylib's `__swift5_entry` IFF it has one. Prior to addressing
/// the above, we always emitted the trampoline and assumed it would be inert unless
/// used. But for older delegate style extensions, the presence of this linker
/// section determines the launch lifecycle used.
///
/// To accommodate this at link time, we are choosing a variant of the stub executor
/// library, one with or without the entry point, based on whether the entry point
/// is present in the user's debug dylib. To do this, we need to inspect the binary
/// so we'll do this in a new task that will write the chosen library path to a file
/// list that is added to the stub executor link arguments.
public final class ConstructStubExecutorInputFileListTaskAction: TaskAction {

    /// A unique identifier for the tool, used for binding in llbuild.
    override public class var toolIdentifier: String {
        "construct-stub-executor-input-file-list"
    }

    override public func performTaskAction(
        _ task: any ExecutableTask,
        dynamicExecutionDelegate: any DynamicTaskExecutionDelegate,
        executionDelegate: any TaskExecutionDelegate,
        clientDelegate: any TaskExecutionClientDelegate,
        outputDelegate: any TaskOutputDelegate
    ) async -> CommandResult {
        guard task.inputPaths.count == 3 else {
            outputDelegate.emitError(
                "Expected 3 inputs to task for constructing stub executor input file list but got \(task.inputPaths.count)"
            )
            return .failed
        }

        let fs = executionDelegate.fs

        let baseLibraryVariantPath = task.inputPaths[0]
        let swiftEntryLibraryVariantPath = task.inputPaths[1]
        let dylibPath = task.inputPaths[2]

        guard task.outputPaths.count == 1 else {
            outputDelegate.emitError(
                "Expected only one output to task for constructing stub executor input list file but got \(task.outputPaths.count)"
            )
            return .failed
        }

        let fileListPath = task.outputPaths[0]

        do {
            let machOData = try MachO(data: fs.read(dylibPath))
            let slices = try machOData.slices()

            let slicesWithEntryPointSection = try machOData.slices()
                .filter { try $0.containsSwift5EntrySection }

            let chosenLibraryPath: Path
            let containsSwift5EntrySection = slicesWithEntryPointSection.count > 0
            if containsSwift5EntrySection {
                if slices.count != slicesWithEntryPointSection.count {
                    // It should not be the case that you'd have some slices with and without the Swift
                    // entry point. If we happen to encounter that, let's print out a warning.
                    outputDelegate.emitWarning(
                        "Only \(slicesWithEntryPointSection.count) out of \(slices.count) slices in the debug dylib MachO contained Swift entry point sections. Using stub executor library with Swift entry point."
                    )
                }
                else {
                    outputDelegate.emitNote("Using stub executor library with Swift entry point.")
                }
                chosenLibraryPath = swiftEntryLibraryVariantPath
            } else {
                outputDelegate.emitNote("Using stub executor library without Swift entry point.")
                chosenLibraryPath = baseLibraryVariantPath
            }

            try fs.write(fileListPath, contents: ByteString(encodingAsUTF8: "\(chosenLibraryPath.str)\n"))

            return .succeeded
        }
        catch {
            outputDelegate.emitError("Unable to process debug dylib: \(error.localizedDescription)")
            return .failed
        }
    }

}