X-Git-Url: http://ftp.carnet.hr/carnet-debian/scm?a=blobdiff_plain;f=src%2Ffunctions.sh;h=5fce053fcbe1fa8af0b1cc86a31d608bc6ccbadb;hb=6f4f76a38153d21d9c8d3034f997d28d7f2fba83;hp=0af728bf701decf0cbe1aadd0dfe5e9b1084cb6b;hpb=a97320a33fcf92f0c683b0830dbf700b88726a99;p=carnet-upgrade.git

diff --git a/src/functions.sh b/src/functions.sh
index 0af728b..5fce053 100644
--- a/src/functions.sh
+++ b/src/functions.sh
@@ -678,10 +678,12 @@ upgrade_apache2 () {
 upgrade_amavis () {
   local conf
 
-  # remove init script diversion
-  if [ -L /etc/init.d/amavis -a -f /etc/init.d/amavis.amavisd-new ]; then
-    rm -vf /etc/init.d/amavis
-    dpkg-divert --quiet --remove /etc/init.d/amavis
+  # remove init script diversion before upgrade
+  if pkg amavisd-cn lt 3:2.4.2-4; then 
+    if [ -L /etc/init.d/amavis -a -f /etc/init.d/amavis.amavisd-new ]; then
+      rm -vf /etc/init.d/amavis
+      dpkg-divert --remove /etc/init.d/amavis
+    fi
   fi
 
   # move old config aside
@@ -699,6 +701,11 @@ upgrade_amavis () {
                 's/^\(clamd.*\)5.clamav.log$/\14\tsocket/g' \
                 /etc/init.d/amavisd-cn || true
 
+  # check if link got lost
+  if pkg amavisd-cn && [ ! -L /etc/init.d/amavis ]; then
+    apt-get install --reinstall --yes amavisd-cn
+  fi
+
   # start new packages
   /etc/init.d/amavis restart
 }
@@ -824,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"
 
@@ -839,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"
 
@@ -863,7 +880,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}')
+         | 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"
 }
@@ -872,7 +897,14 @@ installed_size() {
   local package size
   package=$1
   size=$(apt-cache show $package \
-         | awk '$1 == "Installed-Size:" {print $2; exit}')
+         | 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"
 }