From 62e5f4865e4eac3c71d80c071d46ed61e8449e71 Mon Sep 17 00:00:00 2001
From: Kubernetes Prow Robot <20407524+k8s-ci-robot@users.noreply.github.com>
Date: Mon, 8 Apr 2024 03:35:48 -0700
Subject: [PATCH] Merge pull request #1325 from cvvz/fix-1236

feat: support setting kubelet path for blobfuse-proxy endpoint
---
 charts/latest/blob-csi-driver/templates/csi-blob-node.yaml | 2 ++
 pkg/blobfuse-proxy/init.sh                                 | 6 ++++++
 pkg/blobfuse-proxy/install-proxy-rhcos.sh                  | 2 +-
 pkg/blobfuse-proxy/install-proxy.sh                        | 2 +-
 4 files changed, 10 insertions(+), 2 deletions(-)

diff --git a/charts/latest/blob-csi-driver/templates/csi-blob-node.yaml b/charts/latest/blob-csi-driver/templates/csi-blob-node.yaml
index b0398cc76..fb74de39a 100644
--- a/charts/latest/blob-csi-driver/templates/csi-blob-node.yaml
+++ b/charts/latest/blob-csi-driver/templates/csi-blob-node.yaml
@@ -97,6 +97,8 @@ spec:
               value: "{{ .Values.node.blobfuseProxy.maxOpenFileNum }}"
             - name: DISABLE_UPDATEDB
               value: "{{ .Values.node.blobfuseProxy.disableUpdateDB }}"
+            - name: KUBELET_PATH
+              value: "{{ .Values.linux.kubelet }}"
           volumeMounts:
             - name: host-usr
               mountPath: /host/usr
diff --git a/pkg/blobfuse-proxy/init.sh b/pkg/blobfuse-proxy/init.sh
index 31fa8d7e5..c0cefe953 100755
--- a/pkg/blobfuse-proxy/init.sh
+++ b/pkg/blobfuse-proxy/init.sh
@@ -21,6 +21,12 @@ DISABLE_UPDATEDB=${DISABLE_UPDATEDB:-true}
 SET_MAX_OPEN_FILE_NUM=${SET_MAX_OPEN_FILE_NUM:-true}
 SET_READ_AHEAD_SIZE=${SET_READ_AHEAD_SIZE:-true}
 READ_AHEAD_KB=${READ_AHEAD_KB:-15380}
+KUBELET_PATH=${KUBELET_PATH:-/var/lib/kubelet}
+if [ "$KUBELET_PATH" != "/var/lib/kubelet" ];then
+  echo "kubelet path is $KUBELET_PATH, update blobfuse-proxy.service...."
+  sed -i "s#--blobfuse-proxy-endpoint[^ ]*#--blobfuse-proxy-endpoint=unix:/${KUBELET_PATH}/plugins/blob.csi.azure.com/blobfuse-proxy.sock#" /blobfuse-proxy/blobfuse-proxy.service
+  echo "blobfuse-proxy endpoint is updated to unix:/$KUBELET_PATH/plugins/blob.csi.azure.com/blobfuse-proxy.sock"
+fi 
 
 HOST_CMD="nsenter --mount=/proc/1/ns/mnt"
 
diff --git a/pkg/blobfuse-proxy/install-proxy-rhcos.sh b/pkg/blobfuse-proxy/install-proxy-rhcos.sh
index 91f51f8bc..2b445de06 100644
--- a/pkg/blobfuse-proxy/install-proxy-rhcos.sh
+++ b/pkg/blobfuse-proxy/install-proxy-rhcos.sh
@@ -35,7 +35,7 @@ if [ -f "/host/usr/local/bin/blobfuse-proxy" ];then
 fi
 if [ "$updateBlobfuseProxy" = "true" ];then
   echo "copy blobfuse-proxy...."
-  rm -rf /host/var/lib/kubelet/plugins/blob.csi.azure.com/blobfuse-proxy.sock
+  rm -rf /host/"$KUBELET_PATH"/plugins/blob.csi.azure.com/blobfuse-proxy.sock
   rm -rf /host/usr/local/bin/blobfuse-proxy
   cp /blobfuse-proxy/blobfuse-proxy /host/usr/local/bin/blobfuse-proxy
   chmod 755 /host/usr/local/bin/blobfuse-proxy
diff --git a/pkg/blobfuse-proxy/install-proxy.sh b/pkg/blobfuse-proxy/install-proxy.sh
index a642bb3aa..422109179 100644
--- a/pkg/blobfuse-proxy/install-proxy.sh
+++ b/pkg/blobfuse-proxy/install-proxy.sh
@@ -89,7 +89,7 @@ if [ -f "/host/usr/bin/blobfuse-proxy" ];then
 fi
 if [ "$updateBlobfuseProxy" = "true" ];then
   echo "copy blobfuse-proxy...."
-  rm -rf /host/var/lib/kubelet/plugins/blob.csi.azure.com/blobfuse-proxy.sock
+  rm -rf /host/"$KUBELET_PATH"/plugins/blob.csi.azure.com/blobfuse-proxy.sock
   rm -rf /host/usr/bin/blobfuse-proxy
   cp /blobfuse-proxy/blobfuse-proxy /host/usr/bin/blobfuse-proxy
   chmod 755 /host/usr/bin/blobfuse-proxy