From ef730dba907fc78bed3ab9f2a40d0ea2de1ccb2b Mon Sep 17 00:00:00 2001
From: Argyrios Kyrtzidis <kyrtzidis@apple.com>
Date: Mon, 18 Dec 2023 11:38:20 -0800
Subject: [PATCH] Revert "[Driver] For immediate mode prefer using the default
 SDK over the `SDKROOT` variable"

This reverts commit 3615d0645bc2a32f8cf8b48bb4fc1c0226b60f39.

rdar://117027065
---
 Sources/SwiftDriver/Driver/Driver.swift       |  8 ++------
 Tests/SwiftDriverTests/SwiftDriverTests.swift | 19 ++-----------------
 2 files changed, 4 insertions(+), 23 deletions(-)

diff --git a/Sources/SwiftDriver/Driver/Driver.swift b/Sources/SwiftDriver/Driver/Driver.swift
index e37c87954..0beb90af2 100644
--- a/Sources/SwiftDriver/Driver/Driver.swift
+++ b/Sources/SwiftDriver/Driver/Driver.swift
@@ -2752,17 +2752,13 @@ extension Driver {
 
     if let arg = parsedOptions.getLastArgument(.sdk) {
       sdkPath = arg.asSingle
+    } else if let SDKROOT = env["SDKROOT"] {
+      sdkPath = SDKROOT
     } else if compilerMode == .immediate || compilerMode == .repl {
       // In immediate modes, query the toolchain for a default SDK.
-      // We avoid using `SDKROOT` because that may be set to a compilation platform (like iOS)
-      // that is different than the host.
       sdkPath = try? toolchain.defaultSDKPath(targetTriple)?.pathString
     }
 
-    if sdkPath == nil, let SDKROOT = env["SDKROOT"] {
-      sdkPath = SDKROOT
-    }
-
     // An empty string explicitly clears the SDK.
     if sdkPath == "" {
       sdkPath = nil
diff --git a/Tests/SwiftDriverTests/SwiftDriverTests.swift b/Tests/SwiftDriverTests/SwiftDriverTests.swift
index a4743ec08..9cfb124e4 100644
--- a/Tests/SwiftDriverTests/SwiftDriverTests.swift
+++ b/Tests/SwiftDriverTests/SwiftDriverTests.swift
@@ -3689,17 +3689,7 @@ final class SwiftDriverTests: XCTestCase {
 
   func testImmediateMode() throws {
     do {
-      var env: [String: String] = ProcessEnv.vars
-#if os(macOS)
-      let executor = try SwiftDriverExecutor(diagnosticsEngine: DiagnosticsEngine(handlers: [Driver.stderrDiagnosticsHandler]),
-                                             processSet: ProcessSet(),
-                                             fileSystem: localFileSystem,
-                                             env: ProcessEnv.vars)
-      let iosSDKPath = try executor.checkNonZeroExit(
-        args: "xcrun", "-sdk", "iphoneos", "--show-sdk-path").spm_chomp()
-      env["SDKROOT"] = iosSDKPath
-#endif
-      var driver = try Driver(args: ["swift", "foo.swift"], env: env)
+      var driver = try Driver(args: ["swift", "foo.swift"])
       let plannedJobs = try driver.planBuild()
       XCTAssertEqual(plannedJobs.count, 1)
       let job = plannedJobs[0]
@@ -3713,12 +3703,7 @@ final class SwiftDriverTests: XCTestCase {
       XCTAssertTrue(job.commandLine.contains(.flag("foo")))
 
       if driver.targetTriple.isMacOSX {
-        let sdkIdx = try XCTUnwrap(job.commandLine.firstIndex(of: .flag("-sdk")))
-        if case .path(let path) = job.commandLine[sdkIdx + 1] {
-          XCTAssertTrue(path.name.contains("MacOSX.platform"))
-        } else {
-          XCTFail("Missing SDK path")
-        }
+        XCTAssertTrue(job.commandLine.contains(.flag("-sdk")))
       }
 
       XCTAssertFalse(job.commandLine.contains(.flag("--")))