From 8329190a5c07bcb4934461805f766a458d6c943d Mon Sep 17 00:00:00 2001
From: Leo Liou <leoliou@google.com>
Date: Tue, 14 Mar 2023 15:37:19 +0800
Subject: [PATCH] gs201: conf: run ufs ffu script after boot complete

The ufs script is to work for ffu preparing. If device's ufs has an
old firmware, the script will copy the valid firmware into disk and
wait for ffu update on bootloader.

Bug: 273305212
Test: full build and test ffu update
Change-Id: Ic5069561b67e2773cd87edb31bfa3d682867b838
Signed-off-by: Leo Liou <leoliou@google.com>
---
 device.mk                      |  3 +++
 storage/Android.bp             | 27 +++++++++++++++++++++++++++
 storage/storage.gs201.rc       |  7 +++++++
 storage/ufs_firmware_update.sh | 19 +++++++++++++++++++
 4 files changed, 56 insertions(+)
 create mode 100644 storage/Android.bp
 create mode 100644 storage/storage.gs201.rc
 create mode 100755 storage/ufs_firmware_update.sh

diff --git a/device.mk b/device.mk
index ceeabd59..63d9ba3d 100644
--- a/device.mk
+++ b/device.mk
@@ -1176,3 +1176,6 @@ PRODUCT_VENDOR_PROPERTIES += ro.crypto.metadata_init_delete_all_keys.enabled?=tr
 
 # Hardware Info
 include hardware/google/pixel/HardwareInfo/HardwareInfo.mk
+
+# UFS: the script is used to select the corresponding firmware to run FFU.
+PRODUCT_PACKAGES += ufs_firmware_update.sh
diff --git a/storage/Android.bp b/storage/Android.bp
new file mode 100644
index 00000000..90a919b8
--- /dev/null
+++ b/storage/Android.bp
@@ -0,0 +1,27 @@
+//
+// Copyright (C) 2017 The Android Open Source Project
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package {
+    default_applicable_licenses: [
+        "//device/google/gs201:device_google_gs201_license",
+    ],
+}
+
+sh_binary {
+    name: "ufs_firmware_update.sh",
+    src: "ufs_firmware_update.sh",
+    init_rc: ["storage.gs201.rc"],
+    vendor: true,
+}
diff --git a/storage/storage.gs201.rc b/storage/storage.gs201.rc
new file mode 100644
index 00000000..9873325d
--- /dev/null
+++ b/storage/storage.gs201.rc
@@ -0,0 +1,7 @@
+on property:sys.boot_completed=1
+    # Copy ufs firmware to disk
+    start ufs_firmware_update
+
+service ufs_firmware_update /vendor/bin/ufs_firmware_update.sh
+    disabled
+    oneshot
diff --git a/storage/ufs_firmware_update.sh b/storage/ufs_firmware_update.sh
new file mode 100755
index 00000000..d118115f
--- /dev/null
+++ b/storage/ufs_firmware_update.sh
@@ -0,0 +1,19 @@
+#!/vendor/bin/sh
+#
+# The script belongs to the feature of UFS FFU via OTA: go/p23-ffu-ota
+# Its purpose is to copy the corresponding firmware into partition for UFS FFU.
+
+ufs_dev="/dev/sys/block/bootdevice"
+fw_dir="/vendor/firmware"
+blk_dev="/dev/block/by-name/fips"
+
+vendor=$(cat ${ufs_dev}/vendor | tr -d "[:space:]")
+model=$(cat ${ufs_dev}/model | tr -d "[:space:]")
+rev=$(cat ${ufs_dev}/rev | tr -d "[:space:]")
+
+file=$(find ${fw_dir} -name "*${vendor}${model}${rev}*" | head -n 1)
+if [ -n "$file" ]; then
+	# The first 4KB block at fips partition has been occupied, and unused space begins from 4 KB
+	# Refer to: go/pixel-mp-ffu-ota-1p
+	dd if="$file" of=$blk_dev bs=4k seek=1
+fi