X-Git-Url: http://ftp.carnet.hr/carnet-debian/scm?a=blobdiff_plain;f=src%2Ffunctions.sh;h=cbb7d6152f5fc1c8c3798a25831d644e45ad9905;hb=f726ca67f881a8d8218bec3bffc2ae9960ad5fbd;hp=0ddd941fabbbde73317fb6e41b081581497c86cc;hpb=fdd153655d85070a231c60f89f343abd4ed79bf0;p=carnet-upgrade.git

diff --git a/src/functions.sh b/src/functions.sh
index 0ddd941..cbb7d61 100644
--- a/src/functions.sh
+++ b/src/functions.sh
@@ -831,6 +831,11 @@ check_archives_space() {
   available_disk_space=$(free_space /var/cache/apt/archives/)
   archives_size=$(download_size dist-upgrade)
 
+  if [ $? -ne 0 ]; then
+    log "Failed to get download size, skipping check."
+    return 0
+  fi
+
   log "Archives space: ${available_disk_space}MB"
   log "Download size: ${archives_size}MB"
 
@@ -846,6 +851,11 @@ check_kernel_space() {
   available_disk_space=$(free_space /)
   kernel_size=$(installed_size kernel-2.6-cn)
 
+  if [ $? -ne 0 ]; then
+    log "Failed to get installed size, skipping check."
+    return 0
+  fi
+
   log "Free space in /: ${available_disk_space}MB"
   log "Kernel size: ${kernel_size}MB"
 
@@ -869,8 +879,15 @@ free_space() {
 
 download_size() {
   local size
-  size=$(LC_ALL=C apt-get --yes --print-uris "$@" \
-         | sed -n '/^Need to get / {s///; s/B.*//; p}')
+  size=$(LC_ALL=C apt-get --yes --print-uris "$@" | grep '^Need to get ')
+
+  if [ $? -ne 0 ]; then
+    log "Failed to get download size for 'apt-get $@'."
+    return 1
+  fi
+
+  size=${size#Need to get }
+  size=${size%%B*}
 
   size_to_mb "${size}B"
 }
@@ -878,8 +895,14 @@ download_size() {
 installed_size() {
   local package size
   package=$1
-  size=$(apt-cache show $package \
-         | awk '$1 == "Installed-Size:" {print $2; exit}')
+  size=$(apt-cache show $package | grep '^Installed-Size:')
+
+  if [ $? -ne 0 ]; then
+    log "Failed to get installed size for '$package'."
+    return 1
+  fi
+
+  size=${size#*: }
 
   size_to_mb "${size}kB"
 }