Fix repository checkout ownership on Linux when running as non-root

src/Agent.Sdk/Util/IOUtil.cs

@@ -303,6 +303,9 @@ public static void MoveDirectory(string sourceDir, string targetDir, string stag
 
             // move staging to target
             Directory.Move(stagingDir, targetDir);
+
+            // On Linux, ensure the target directory has the correct ownership
+            EnsureDirectoryOwnership(targetDir);
         }
 
         /// <summary>
@@ -466,6 +469,9 @@ public static void CopyDirectory(string source, string target, CancellationToken
                     target: Path.Combine(target, subDir.Name),
                     cancellationToken: cancellationToken);
             }
+
+            // On Linux, ensure the target directory has the correct ownership
+            EnsureDirectoryOwnership(target);
         }
 
         public static void ValidateExecutePermission(string directory)
@@ -569,6 +575,58 @@ private static void RemoveReadOnly(FileSystemInfo item)
             }
         }
 
+        public static void EnsureDirectoryOwnership(string directoryPath)
+        {
+            // Only apply ownership fix on Linux platforms
+            if (!PlatformUtil.RunningOnLinux)
+            {
+                return;
+            }
+
+            try
+            {
+                // First get the current user and group ID
+                System.Diagnostics.Process idProcess = new System.Diagnostics.Process();
+                idProcess.StartInfo.FileName = "id";
+                idProcess.StartInfo.Arguments = "-u";
+                idProcess.StartInfo.UseShellExecute = false;
+                idProcess.StartInfo.RedirectStandardOutput = true;
+                idProcess.Start();
+                string userId = idProcess.StandardOutput.ReadToEnd().Trim();
+                idProcess.WaitForExit();
+
+                idProcess = new System.Diagnostics.Process();
+                idProcess.StartInfo.FileName = "id";
+                idProcess.StartInfo.Arguments = "-g";
+                idProcess.StartInfo.UseShellExecute = false;
+                idProcess.StartInfo.RedirectStandardOutput = true;
+                idProcess.Start();
+                string groupId = idProcess.StandardOutput.ReadToEnd().Trim();
+                idProcess.WaitForExit();
+
+                // Now use chown with the explicit UIDs
+                System.Diagnostics.Process chownProcess = new System.Diagnostics.Process();
+                chownProcess.StartInfo.FileName = "chown";
+                chownProcess.StartInfo.Arguments = $"-R {userId}:{groupId} \"{directoryPath}\"";
+                chownProcess.StartInfo.UseShellExecute = false;
+                chownProcess.StartInfo.RedirectStandardOutput = true;
+                chownProcess.StartInfo.RedirectStandardError = true;
+                chownProcess.Start();
+                chownProcess.WaitForExit();
+
+                // Log error if the command failed
+                if (chownProcess.ExitCode != 0)
+                {
+                    Trace.Error($"Failed to set directory ownership for '{directoryPath}'. Exit code: {chownProcess.ExitCode}");
+                }
+            }
+            catch (Exception ex)
+            {
+                // Log but don't fail the operation if ownership change fails
+                Trace.Error($"Exception while trying to set directory ownership for '{directoryPath}': {ex.Message}");
+            }
+        }
+
         public static string GetDirectoryName(string path, PlatformUtil.OS platform)
         {
             if (string.IsNullOrWhiteSpace(path))

src/Agent.Worker/Build/BuildDirectoryManager.cs

@@ -256,6 +256,12 @@ public void CreateDirectory(IExecutionContext executionContext, string descripti
                 executionContext.Debug($"Creating {description}: '{path}'");
                 Trace.Info($"Creating {description}.");
                 Directory.CreateDirectory(path);
+
+                // Ensure correct ownership on Linux platforms
+                if (PlatformUtil.RunningOnLinux && Directory.Exists(path))
+                {
+                    IOUtil.EnsureDirectoryOwnership(path);
+                }
             }
         }