libsnapshot: Add script to test snapshot updates
$time ./system/core/fs_mgr/libsnapshot/apply-update.sh --update-static-partitions
Update completed
real 1m19.850s
user 0m10.319s
sys 0m7.448s
Dynamic parition update only:
$time ./system/core/fs_mgr/libsnapshot/apply-update.sh
Update completed
real 0m19.032s
user 0m10.480s
sys 0m7.462s
Bug: 329255363
Test: apply-update.sh
Change-Id: I2af4fe022e35b2b813ff83e2084d22295b3057dc
Signed-off-by: Akilesh Kailash <akailash@google.com>
diff --git a/fs_mgr/libsnapshot/scripts/apply-update.sh b/fs_mgr/libsnapshot/scripts/apply-update.sh
new file mode 100755
index 0000000..90b0119
--- /dev/null
+++ b/fs_mgr/libsnapshot/scripts/apply-update.sh
@@ -0,0 +1,77 @@
+#!/bin/bash
+
+# This is a debug script to quicky test end-to-end flow
+# of snapshot updates without going through update-engine.
+#
+# Usage:
+#
+# To update both dynamic and static partitions:
+#
+# ./system/core/fs_mgr/libsnapshot/apply_update.sh [--update-static-partitions] [--wipe]
+#
+# --update-static-partitions: This will update bootloader and static A/B
+# partitions
+# --wipe: Allows data wipe as part of update flow
+#
+# To update dynamic partitions only (this should be used when static
+# partitions are present in both the slots):
+#
+# ./system/core/fs_mgr/libsnapshot/apply_update.sh
+#
+#
+
+rm -f $OUT/*.patch
+
+# Compare images and create snapshot patches. Currently, this
+# just compares two identical images in $OUT. In general, any source
+# and target images could be passed to create snapshot patches. However,
+# care must be taken to ensure source images are already present on the device.
+#
+# create_snapshot is a host side binary. Build it with `m create_snapshot`
+create_snapshot --source=$OUT/system.img --target=$OUT/system.img &
+create_snapshot --source=$OUT/product.img --target=$OUT/product.img &
+create_snapshot --source=$OUT/vendor.img --target=$OUT/vendor.img &
+create_snapshot --source=$OUT/system_ext.img --target=$OUT/system_ext.img &
+create_snapshot --source=$OUT/vendor_dlkm.img --target=$OUT/vendor_dlkm.img &
+create_snapshot --source=$OUT/system_dlkm.img --target=$OUT/system_dlkm.img &
+
+echo "Waiting for snapshot patch creation"
+wait $(jobs -p)
+echo "Snapshot patch creation completed"
+
+mv *.patch $OUT/
+
+adb root
+adb wait-for-device
+adb shell mkdir -p /data/update/
+adb push $OUT/*.patch /data/update/
+
+if [[ "$2" == "--wipe" ]]; then
+ adb shell snapshotctl apply-update /data/update/ -w
+else
+ adb shell snapshotctl apply-update /data/update/
+fi
+
+# Check if the --update-static-partitions option is provided.
+# For quick developer workflow, there is no need to repeatedly
+# apply static partitions.
+if [[ "$1" == "--update-static-partitions" ]]; then
+ adb reboot bootloader
+ sleep 5
+ if [[ "$2" == "--wipe" ]]; then
+ fastboot -w
+ fi
+ fastboot flash bootloader $OUT/bootloader.img
+ sleep 1
+ fastboot reboot bootloader
+ sleep 1
+ fastboot flash radio $OUT/radio.img
+ sleep 1
+ fastboot reboot bootloader
+ sleep 1
+ fastboot flashall --exclude-dynamic-partitions --disable-super-optimization
+else
+ adb reboot
+fi
+
+echo "Update completed"