From 7e3dd619d1defe9a59e3e04767e8b3f397439565 Mon Sep 17 00:00:00 2001
From: Michael Marineau <michael.marineau@coreos.com>
Date: Fri, 17 Jan 2014 21:07:15 -0800
Subject: [PATCH] fix(build_image): Add fstrim commands for -usr images.

---
 build_library/base_image_util.sh | 6 ++++--
 build_library/dev_image_util.sh  | 6 ++++--
 2 files changed, 8 insertions(+), 4 deletions(-)

diff --git a/build_library/base_image_util.sh b/build_library/base_image_util.sh
index 5b26bed20e..be72ffb3b8 100755
--- a/build_library/base_image_util.sh
+++ b/build_library/base_image_util.sh
@@ -60,8 +60,10 @@ create_base_image() {
   # Zero all fs free space to make it more compressible so auto-update
   # payloads become smaller, not fatal since it won't work on linux < 3.2
   sudo fstrim "${root_fs_dir}" || true
-  if [[ -d "${root_fs_dir}/media/state" ]]; then
-      sudo fstrim "${root_fs_dir}/media/state" || true
+  if [[ "${disk_layout}" == *-usr ]]; then
+    sudo fstrim "${root_fs_dir}/usr" || true
+  else
+    sudo fstrim "${root_fs_dir}/media/state" || true
   fi
 
   cleanup_mounts "${root_fs_dir}"
diff --git a/build_library/dev_image_util.sh b/build_library/dev_image_util.sh
index ebf442b63f..4b3d965d0a 100755
--- a/build_library/dev_image_util.sh
+++ b/build_library/dev_image_util.sh
@@ -48,8 +48,10 @@ install_dev_packages() {
 
   # Zero all fs free space, not fatal since it won't work on linux < 3.2
   sudo fstrim "${root_fs_dir}" || true
-  if [[ -d "${root_fs_dir}/media/state" ]]; then
-      sudo fstrim "${root_fs_dir}/media/state" || true
+  if [[ "${disk_layout}" == *-usr ]]; then
+    sudo fstrim "${root_fs_dir}/usr" || true
+  else
+    sudo fstrim "${root_fs_dir}/media/state" || true
   fi
 
   info "Developer image built and stored at ${image_name}"