From 10324566ae4ef2e48983e8700dbfb3c6372c01cb Mon Sep 17 00:00:00 2001
From: James Rasell <jrasell@users.noreply.github.com>
Date: Mon, 29 Jan 2024 09:54:52 +0100
Subject: [PATCH] driver/rawexec: populate OOM killed exit result. (#19829)

---
 .changelog/19829.txt      | 3 +++
 drivers/rawexec/driver.go | 5 +++--
 2 files changed, 6 insertions(+), 2 deletions(-)
 create mode 100644 .changelog/19829.txt

diff --git a/.changelog/19829.txt b/.changelog/19829.txt
new file mode 100644
index 000000000..b2e300c77
--- /dev/null
+++ b/.changelog/19829.txt
@@ -0,0 +1,3 @@
+```release-note:bug
+driver/rawexec: Ensure the OOM killed response is populated when the task exits
+```
diff --git a/drivers/rawexec/driver.go b/drivers/rawexec/driver.go
index 18f20b4f7..8bcde8b8a 100644
--- a/drivers/rawexec/driver.go
+++ b/drivers/rawexec/driver.go
@@ -393,8 +393,9 @@ func (d *Driver) handleWait(ctx context.Context, handle *taskHandle, ch chan *dr
 		}
 	} else {
 		result = &drivers.ExitResult{
-			ExitCode: ps.ExitCode,
-			Signal:   ps.Signal,
+			ExitCode:  ps.ExitCode,
+			Signal:    ps.Signal,
+			OOMKilled: ps.OOMKilled,
 		}
 	}