From 6d1ae9e89008a96bd5ac7dae1f7996766a57743f Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E5=BD=A6=E7=BA=BE?= <wangyanshu.wys@alibaba-inc.com>
Date: Wed, 5 Aug 2026 11:42:57 +0800
Subject: [PATCH] fix(install): warn for affected ublk kernels

---
 scripts/docker-setup.sh | 14 ++++++++++++++
 scripts/install.sh      | 14 ++++++++++++++
 2 files changed, 28 insertions(+)

diff --git a/scripts/docker-setup.sh b/scripts/docker-setup.sh
index 8b7cffdd..31b72b33 100755
--- a/scripts/docker-setup.sh
+++ b/scripts/docker-setup.sh
@@ -16,6 +16,20 @@ if [[ $EUID -ne 0 ]]; then
     exit 1
 fi
 
+KERNEL_RELEASE="$(uname -r)"
+case "$KERNEL_RELEASE" in
+    6.18.4|6.18.4[-+]*|6.18.5|6.18.5[-+]*|\
+    6.17.0-1015-azure|6.17.0-1015-azure[-+]*|\
+    6.17.0-1018-azure|6.17.0-1018-azure[-+]*)
+        cat >&2 <<EOF
+warning: Linux kernel ${KERNEL_RELEASE} is known to contain a ublk initialization regression.
+         Creating the first ublk device may hang or panic the host.
+         Upgrade to a fixed vendor kernel (upstream 6.18.6+ for 6.18.y) before running AgentENV.
+         See: https://github.com/kvcache-ai/AgentENV/issues/20
+EOF
+        ;;
+esac
+
 # ---------------------------------------------------------------------------
 # 1. ublk_drv kernel module
 # ---------------------------------------------------------------------------
diff --git a/scripts/install.sh b/scripts/install.sh
index 33de4e25..b61b1453 100644
--- a/scripts/install.sh
+++ b/scripts/install.sh
@@ -65,6 +65,20 @@ if [[ "$OS" != "linux" ]]; then
     exit 1
 fi
 
+KERNEL_RELEASE="$(uname -r)"
+case "$KERNEL_RELEASE" in
+    6.18.4|6.18.4[-+]*|6.18.5|6.18.5[-+]*|\
+    6.17.0-1015-azure|6.17.0-1015-azure[-+]*|\
+    6.17.0-1018-azure|6.17.0-1018-azure[-+]*)
+        cat >&2 <<EOF
+warning: Linux kernel ${KERNEL_RELEASE} is known to contain a ublk initialization regression.
+         Creating the first ublk device may hang or panic the host.
+         Upgrade to a fixed vendor kernel (upstream 6.18.6+ for 6.18.y) before running AgentENV.
+         See: https://github.com/kvcache-ai/AgentENV/issues/20
+EOF
+        ;;
+esac
+
 RELEASE_API="https://api.github.com/repos/${REPO}/releases/latest"
 if [[ "$VIRTUALIZATION_MODE" == "pvm" ]]; then
     TARBALL="aenv-server-${OS}-${ARCH_TAG}-pvm.tar.gz"