Handle php.ini upgrade.
[carnet-upgrade.git] / src / functions.sh
1 # Ivan 'ico' Rako <irako@srce.hr>
2 # Zoran 'jelly' Dzelajlija <jelly@srce.hr>
3
4 notice () {
5   LC_MESSAGES=hr_HR dialog --backtitle "$title" --msgbox "$*" 18 75
6 }
7
8 first_warning () {
9   LC_MESSAGES=hr_HR dialog --backtitle "$title" --yesno "$msg_first_warning" 18 75
10   if [ $? -ne 0 ]; then
11     log "Upgrade canceled."
12     exit 0
13   fi
14 }
15
16 pkg() {
17   local version
18   dpkg -l "$1" 2> /dev/null | egrep -q "^.i" || return 1
19   [ -z "$2" ] && return 0
20   version=$(dpkg -s "$1" | awk '/^Version:/ {print $2}')
21   dpkg --compare-versions "$version" "$2" "$3" 2>&1 > /dev/null
22 }
23
24 log() {
25   local old_umask
26   logfile=${logfile:=/var/log/carnet-upgrade.log}
27
28   old_umask=$(umask)
29   umask 0077
30
31   echo "$(date +'%Y-%m-%d %H:%M:%S') $*" >> $logfile
32   echo "CN: $*"
33
34   umask $old_umask
35   chmod og= $logfile
36 }
37
38 # find first free uid/gid in range
39 # find_id passwd 100 999
40 find_id() {
41   local i db first last ids
42   db=$1
43   first=$2
44   last=$3
45   ids=$(getent $db | awk -F: "\$3 >= $first && \$3 <= $last {print \$3}")
46   for i in $(seq $first $last)
47   do
48     if ! echo $ids | grep -q $i; then
49       echo $i
50       return 0
51     fi
52   done
53   return 1
54 }
55
56 # safe in-place s///
57 check_and_sed() {
58   [ "$DEBIAN_SCRIPT_DEBUG" ] && set -vx
59   local s sedcmd ret i
60   s="$1"
61   shift
62   sedcmd="$1"
63   shift
64   ret=2
65   for i in $*
66   do
67     [ -e "$i" ]        || continue
68     egrep -q "$s" "$i" || continue
69     [ -h "$i" ]        && i=$(readlink -f "$i")
70     sed "$sedcmd" "$i" > "$i.dpkg-tmp"
71     chown --reference "$i" "$i.dpkg-tmp"
72     chmod --reference "$i" "$i.dpkg-tmp"
73     mv "$i.dpkg-tmp" "$i"
74     ret=0
75   done
76   if [ "$ret" -eq 0 ]; then
77     log "Done \"$sedcmd\" to $*"
78   fi
79   return $ret
80 }
81
82 #
83 # Update uid for user from reserved system range (0-99) to dynamic system
84 # range (100-999).  Optionally update ownerships of given directories.
85 # $0 user [directory ...]
86 #
87 check_and_update_ugid() {
88   local user newgid newuid
89   user=$1
90   if [ "$(getent passwd $user | awk -F: '$3 >= 100 {print "ok"; exit 0}')" ]; then
91     return 0
92   fi
93   shift
94   newgid=$(find_id group 100 999)
95   newuid=$(find_id passwd 100 999)
96   # other directories/files
97   chown -R $newuid:$newgid $*
98   groupmod -g $newgid $user
99   usermod -u $newuid -g $newgid $user
100   log "Fixed $user user uid/gid."
101 }
102
103 pkgadd () {
104   [ -n "$*" ] || return 0
105   log "Installing $* ..."
106   apt-get -y$s install $* || {
107     dpkg --configure -a
108     apt-get -yf install
109     apt-get -y$s install $*
110   }
111 }
112
113 pkgrm () {
114   [ -n "$*" ] || return 0
115   log "Purging $* ..."
116   apt-get -y$s --purge remove $* || {
117     dpkg --configure -a
118     apt-get -yf install
119     apt-get -y$s --purge remove $*
120   }
121 }
122
123 pkgrm_only () {
124   [ -n "$*" ] || return 0
125   log "Removing $* ..."
126   apt-get -y$s remove $* || {
127     dpkg --configure -a
128     apt-get -yf install
129     apt-get -y$s remove $*
130   }
131 }
132
133
134 apt_update() {
135   log "Running apt-get update ..."
136   apt-get -y$s update || {
137     apt-get -yf install
138     apt-get -y$s update
139   }
140   apt-get autoclean
141 }
142
143 apt_download() {
144   log "Running apt-get download ..."
145   apt-get -dy$s dist-upgrade
146 }
147
148 dist_upgrade () {
149   log "Running dist-upgrade ..."
150   apt-get -y$s dist-upgrade || {
151     dpkg --configure -a
152     apt-get -y$s dist-upgrade
153   }
154 }
155
156 remove_x() {
157   LC_MESSAGES=hr_HR dialog --backtitle "$title" --yesno "$msg_remove_x" 18 75
158   if [ $? -eq 0 ]; then
159     log "Removing X Window System"
160     pkgrm xlibs
161   else
162     log "Skipping X Window System removal"
163   fi
164 }
165
166 remove_bloat() {
167   local to_remove add_this i j n text to_show cmd tmpfile installed
168   local update_selections pkgs_to_deselect pkgs_to_remove oldifs pkglist
169   text="$msg_remove_bloat"
170
171   [ "$DEBUG" ] && set -x
172   n=0
173   pkglist=$(mktemp /var/lib/carnet-upgrade/pkglist.XXXXXX)
174   COLUMNS=200 dpkg -l | awk '$1 ~ /^.i/ { print $2 }' > $pkglist
175   for i in $(echo "$bloats" | sed 's/           */      /g'| awk -F'    ' '{print $2}')
176   do
177     if grep -q "^$i$" $pkglist; then
178       installed="$installed $i"
179       add_this="$(echo "$bloats" | egrep "      .*\b${i}\b" | sed 's/           */      /g'| awk -F'    ' '{print $1}')"
180       if ! echo "$to_show" | grep -q "$add_this"; then
181         to_show="$to_show \"$add_this\" \"\" on"
182         n=$(($n+1))
183       fi
184     fi
185   done
186   [ "$DEBUG" ] || rm -f $pkglist
187   [ -z "$to_show" ] && return 0
188   test $n -gt 8 && n=8
189   tmpfile=$(mktemp /var/lib/carnet-upgrade/dialog-tmp.XXXXXX)
190   cmd="LC_MESSAGES=hr_HR dialog --nocancel --backtitle \""$title"\" --checklist \""$text"\" 20 75 $n $to_show"
191   [ "$DEBUG" ] && set +x
192     eval $cmd 2> $tmpfile
193   [ "$DEBUG" ] && set -x
194   to_remove="$(cat $tmpfile | sed 's/\"\ \"/    /g; s/\"\ *//g; s/\\//g')"
195   [ "$DEBUG" ] || rm -f $tmpfile 
196   [ "$DEBUG" ] && echo "DEBUG: to_remove=$to_remove" 1>&2
197   oldifs="$IFS"
198   IFS=" " # tab
199   for i in $(echo "$to_remove")
200   do
201     IFS="$oldifs"
202     for j in $(echo "$bloats" | grep "^$i" | sed 's/            */      /g'| awk -F'    ' '{print $2}')
203     do
204       # hopefully this will tell apt not to reinstall them at upgrade time
205       update_selections="${update_selections}
206         $j      deinstall"
207       pkgs_to_deselect="$pkgs_to_deselect $j"
208       # only remove those that really are installed
209       if echo "$installed" | grep -q "$j"; then
210         pkgs_to_remove="$pkgs_to_remove $j"
211       fi
212     done
213   done
214   [ "$DEBUG" ] && echo "DEBUG: pkgs_to_remove=$pkgs_to_remove" 1>&2
215   # Finally, remove those
216   if [ -n "$update_selections" ]; then
217     echo "$update_selections" | dpkg --set-selections
218     log "Deselected:${pkgs_to_deselect}"
219   fi
220   eval pkgrm $pkgs_to_remove
221 }
222
223 remove_cn_bloat() {
224   local installed pkg cn_pkgs dialog_list selection cn_del to_del del_pkgs
225
226   # get installed packages
227   installed=$(mktemp /var/lib/carnet-upgrade/pkglist.XXXXXX)
228   COLUMNS=200 dpkg -l | awk '$1 ~ /^.i/ { print $2 }' >$installed
229
230   # detect installed CN packagees
231   for pkg in $(echo "$cn_bloats" | cut -d' ' -f1); do
232     if grep -qx $pkg $installed; then
233       cn_pkgs="$cn_pkgs $pkg"
234       dialog_list="$dialog_list $pkg '' off"
235     fi
236   done
237   rm -f $installed
238
239   # report status or finish if no CN packages found
240   if [ "$cn_pkgs" ]; then
241     log "Found CN packages:$cn_pkgs"
242   else
243     log "No CN packages found."
244     return
245   fi
246
247   # user selects the packages to delete
248   selection=$(mktemp /var/lib/carnet-upgrade/selection.XXXXXX)
249   eval LC_MESSAGES=hr_HR dialog --nocancel --backtitle \""$title"\" \
250     --checklist \""$msg_remove_cn_bloat"\" 20 75 6 $dialog_list 2>$selection
251
252   # expand the package list to include the services
253   for pkg in $(cat $selection); do
254     pkg=${pkg%\"}
255     pkg=${pkg#\"}
256     cn_del="$cn_del $pkg"
257     to_del=$(echo "$cn_bloats" | grep ^$pkg)
258     del_pkgs="$del_pkgs $to_del"
259   done
260   rm -f $selection
261
262   # report selection or finish if no service selected
263   if [ "$cn_del" ]; then
264     log "Removing CN services:$cn_del"
265   else
266     log "No CN services selected for removal."
267     return
268   fi
269
270   # delete the cn packages and relevant services
271   pkgrm_only $del_pkgs
272 }
273
274 post_upgrade() {
275   touch "$post_upgrade_queuetab"
276   if ! grep -q "true    $*" "$post_upgrade_queuetab"; then
277     echo "true  $*" >> "$post_upgrade_queuetab"
278   fi
279 }
280
281 run_post_upgrade() {
282   if [ -f "$post_upgrade_queuetab" ]; then
283     run_actions post_upgrade "$post_upgrade_queuetab"
284     mv "$post_upgrade_queuetab" "${post_upgrade_queuetab}.done.$(date '+%Y-%m-%d.%H:%M:%S')"
285   fi      
286 }
287
288 ###
289
290 run_actions() {
291   local line lineno lineno2 f n name action checkmsg check
292   [ "$DEBUG" ] && set -vx
293   name="$1"
294   f="$2"
295   [ -f "$f" ] ||   return 2 # no such file
296   n=$(wc -l "$f" | awk '{print $1}')
297   [ "$n" -lt 1 ] && return 1 # no lines to process
298   log "run_actions ($name) started, processing $n lines from \"$f\"."
299   log "-----"
300   lineno=0
301   lineno2=0
302   while [ "$lineno" -lt "$n" ]
303   do
304     lineno=$(($lineno+1))
305     lineno2=$lineno
306     line=$(head -$lineno "$f" | tail -1)
307     if echo $line | grep -q '^#'; then  # comment
308       continue
309     fi
310     while echo "$line" | egrep -q '\\$' # continuation
311     do
312       lineno2=$(($lineno2+1))
313       line=$(echo -n "$line"| sed 's/\\$//'; head -$lineno2 "$f" | tail -1)
314     done
315     check=$(echo "$line" | sed 's/              */      /g'| awk -F'    ' '{print $1}')
316     action=$(echo "$line" | sed 's/             */      /g'| awk -F'    ' '{print $2}')
317     checkmsg=""
318     [ "$check" != "true" ] && checkmsg="if '$check'"
319     if eval $check; then
320       log "$(printf 'action %02d' $lineno) starting: '${action}' $checkmsg "
321       if ! eval $action; then
322         log "$(printf 'action %02d' $lineno) failed: '${action}' $checkmsg "
323         display_action_failure "$action"
324         exit 1
325       else  
326         log "$(printf 'action %02d' $lineno) finished: '${action}' $checkmsg"
327       fi
328     else
329       log "$(printf 'action %02d' $lineno) skipping: '${action}' $checkmsg"
330     fi
331     [ "$lineno2" -gt "$lineno" ] && lineno=$lineno2
332   done
333   log "run_actions ($1) finished"
334   log "-----"
335 }
336
337 # report action failure
338 display_action_failure () {
339   local action=$1
340
341   notice "
342 Doslo je do greske prilikom izvrsavanja akcije '$action'. Detalje problema moguce je vidjeti u log datotekama:
343
344   /var/log/carnet-upgrade.log
345   /var/log/carnet-upgrade.typescript
346
347 Da bi dovrsili zapocete operacije nad paketima pokrenite:
348
349   # dpkg --configure -a
350   # apt-get -f install
351
352 Nakon toga ponovno pozovite carnet-upgrade skriptu."
353 }
354
355 # funkcija vraca popis paketa koje treba upgradeati
356 what_upgrade () {
357   local IFSOLD name version package
358
359   for package in `echo "$needed_versions" | awk '{print $1}'| grep -v ^$`; do
360     IFSOLD="$IFS"
361     IFS="       " # tab
362     read name version <<-EOPTS
363         $(echo "$needed_versions" | sed 's/             */      /g' | grep ^$package)
364         EOPTS
365     IFS="$IFSOLD"
366     if pkg "$name" lt "$version"; then
367         echo -n " $name"
368     fi
369   done
370 }
371
372 # funkcija za ispis paketa koje treba upgradeati
373 check_upgrade () {
374   local upgrade
375   upgrade=${1:-"`what_upgrade`"}
376
377   if [ "$upgrade" ]; then
378     notice "
379 Prije prelaska na Debian Lenny potrebno je napraviti nadogradnju ovih paketa:
380   $upgrade
381
382 Pokrenite:
383   # apt-get update
384   # apt-get dist-upgrade
385 ili:
386   # apt-get update
387   # apt-get install${upgrade}
388   
389 Nakon toga ponovno pozovite ovu skriptu."
390     log "carnet-upgrade aborted, upgrade of$upgrade needed."
391     exit 1
392   fi
393 }
394
395 # update sarge urls in sources list
396 fix_sarge_sources () {
397   local apt_sources='/etc/apt/sources.list'
398   local fixed_sources=$(mktemp /var/lib/carnet-upgrade/sources.list.XXXXXX)
399
400   # sarge (oldstable) was archived recently -> sources.list needs update
401   awk -v archive='http://archive.debian.org/debian' \
402       -v security='http://archive.debian.org/debian-security' \
403       '$3 == "sarge" && $2 ~ /[/]debian$/ { $2 = archive }
404        $3 == "sarge/updates" && $2 ~ /security\.debian\.org$/ { $2 = security }
405        { print }' $apt_sources > $fixed_sources
406
407   # did anything get updated
408   if ! cmp $apt_sources $fixed_sources >/dev/null; then
409       log "sources.list fixed to use archive.debian.org"
410
411       # backup current version
412       if [ ! -f "$apt_sources.$backup_ext.sarge" ]; then
413           cp -av $apt_sources "$apt_sources.$backup_ext.sarge"
414       fi
415
416       # install fixed version
417       cp -av $fixed_sources $apt_sources
418
419       # update the database
420       apt_update
421   fi
422
423   # delete temporary
424   rm -f $fixed_sources
425 }
426
427 # try to check if we are using the latest version of carnet-upgrade script
428 check_my_version () {
429   local packages latest_version
430
431   packages=$(mktemp /var/lib/carnet-upgrade/Packages.XXXXXX)
432   if wget -q -O$packages $packages_uri; then
433     latest_version=$(sed -ne \
434         '/Package: carnet-upgrade/,/Description:/{ /Version: /s///p }' \
435         $packages)
436     rm -f $packages
437
438     if [ "$latest_version" ]; then
439       log "Running carnet-upgrade version is $version"
440       log "Latest carnet-upgrade version is $latest_version"
441       if dpkg --compare-versions "$version" lt "$latest_version"; then
442         check_upgrade " carnet-upgrade"
443       else log "Already using the latest carnet-upgrade version."; fi
444     else log "Failed to get latest version, skipping check."; fi
445   else log "Failed to download Packages file, skipping check."; fi
446
447   rm -f $packages
448 }
449
450 # if php is installed make sure it's version 5
451 check_php_version () {
452
453   if pkg php4-cn && ! pkg php5-cn; then
454     # prepare for php upgrade by installing php5 extensions
455     install_php5_extensions
456
457     notice "
458 Prije prelaska na Debian Lenny potrebno je napraviti nadogradnju sa PHP 4 na PHP 5.
459
460 Pokrenite:
461   # apt-get update
462   # apt-get install php5-cn
463   
464 Nakon sto provjerite da sve web aplikacije rade ispravno, ponovno pozovite ovu skriptu."
465     log "carnet-upgrade aborted, upgrade to php5-cn needed."
466     exit 1
467   fi
468 }
469
470 # install php5 extensions based on installed php4 extensions
471 install_php5_extensions () {
472   local php4_list php5_pkg
473
474   # get a list of installed php4 packages
475   php4_list=$(mktemp /var/lib/carnet-upgrade/php4_list.XXXXXX)
476   dpkg -l php4-\* | awk '/^ii/ {
477     pkg=$2; sub("^php4", "php5", pkg); print pkg }' > $php4_list
478
479   # compare it with the list of available php5 packages
480   php5_pkg=$(apt-cache search -n ^php5- | cut -d' ' -f 1 \
481     | grep -F -x -f $php4_list | grep -F -x -v php5-cn)
482
483   # handle special cases
484   if pkg php4-xslt; then
485      php5_pkg="$php5_pkg php5-xsl"
486   fi
487
488   # install the required php5 modules
489   if [ "$php5_pkg" ]; then
490      pkgadd $php5_pkg
491   fi
492
493   # cleanup
494   rm -f $php4_list
495 }
496
497 # if grsec is installed make sure it is fresh
498 check_grsec_kernel () {
499   local ver=`uname -r`
500
501   log "Kernel version: $ver"
502   if [ "$ver" = "${ver%-grsec}" ]; then
503     # not grsec variant
504     return 0
505   fi
506
507   if [ "$ver" = "${ver#2.6.2[4-9]}" ]; then
508     # kernel too old
509     notice "
510 Prije prelaska na Debian Lenny potrebno je napraviti nadogradnju na najnoviji kernel.
511
512 Pokrenite:
513   # apt-get update
514   # apt-get install kernel-2.6-cn
515   # reboot
516   
517 Nakon toga ponovno pozovite ovu skriptu."
518     log "carnet-upgrade aborted, kernel upgrade needed."
519     exit 1
520   fi
521 }
522
523 check_reboot () {
524   reboot_required || return 0
525
526   LC_MESSAGES=hr_HR dialog --backtitle "$title" --yesno "$reboot_warning" 18 +75
527   if [ $? -eq 0 ]; then
528     log "Reboot before continuing."
529     exit 0
530   else
531     log "Proceed without reboot."
532   fi
533 }
534
535 # check if the latest kernel is booted
536 reboot_required () {
537   local default_kernel kernel_package running_release running_version
538
539   # try to get default kernel from lilo
540   if type lilo >/dev/null 2>&1; then
541       default_kernel=$( lilo -t -v 2>/dev/null \
542                         | grep -B3 '[*]$' \
543                         | sed -n -e '/^Boot image: / { s///p; q; }' )
544   fi
545
546   # fallback to grub if lilo not installed
547   if [ -z "$default_kernel" -a -r "/boot/grub/menu.lst" ]; then
548       default_kernel=$( awk '
549            $1 == "default" { default = $2; 
550                              if (default !~ /^[0-9]+$/) { exit } }
551            $1 == "title" { title+=1 }
552            $1 == "kernel" && title==default+1 { print $2; exit }' \
553            /boot/grub/menu.lst 2>/dev/null )
554   fi
555
556   if [ -z "$default_kernel" ]; then
557       log "Could not find default kernel"
558       return 1
559   fi
560
561   log "Default kernel image: $default_kernel"
562
563   if [ ! -e "$default_kernel" ]; then
564       log "Default kernel image doesn't exist!?"
565       return 1
566   fi
567
568   kernel_package=$( dpkg -S $default_kernel | cut -d: -f 1 )
569
570   if [ -z "$kernel_package" ]; then
571       log "Default kernel not installed from a package"
572       return 1
573   fi
574
575   log "Default kernel package: $kernel_package"
576
577   running_release=$( uname -r )
578   running_version=$( uname -v )
579   log "Running kernel: $running_release $running_version"
580
581   if grep -q "$running_release .* $running_version" $default_kernel; then
582       log "Latest kernel version is running." 
583       return 1
584   else
585       log "Reboot to latest version required."
586       return 0
587   fi
588 }
589
590 upgrade () {
591   local i to_install packages
592   case "$1" in
593     first)
594       packages="$packages_first"
595     ;;
596     second)
597       packages="$packages_second"
598     ;;
599     *)
600     return 1
601   esac
602
603   for i in $(echo $packages)
604   do
605     pkg $i && to_install="$to_install $i"
606   done
607
608   if [ -n "$to_install" ];then
609     eval pkgadd $to_install
610   else
611     return 0
612   fi
613 }
614
615 # funkcija utrpa ispravan sources.list
616 create_sources_list () {
617     local sl sl_new
618     sl=/etc/apt/sources.list
619     sl_new=/usr/share/carnet-upgrade/files/etc/apt/sources.list
620
621     # check if already installed
622     if ! cmp $sl $sl_new >/dev/null; then
623
624         # backup old version
625         if [ -f $sl -a ! -f $sl.$backup_ext ]; then
626             cp -av $sl $sl.$backup_ext
627         fi
628
629         # install new version
630         cp -av $sl_new $sl
631         notice "Novi sadrzaj datoteke /etc/apt/sources.list:
632
633 `cat /etc/apt/sources.list`"
634     fi
635 }
636
637
638 # funkcija koja brise grupu proc
639 remove_group_proc () {
640   if getent group proc > /dev/null; then
641     groupdel proc || true
642     log "groupdel proc (temporarily)"
643     post_upgrade add_group_proc
644   fi
645 }
646
647 add_group_proc () {
648   if ! getent group proc > /dev/null; then
649     # grsec uses gid 99 for /proc files
650     groupadd -g 99 proc 
651     log "groupadd -g 99 proc"
652
653     # update oidentd so it uses the proc group
654     if [ -x /etc/init.d/oidentd ]; then
655         if gpasswd -a oident proc; then
656             /etc/init.d/oidentd restart
657         fi
658     fi
659   fi
660 }
661
662 # backup ldap database
663 backup_slapd_db () {
664   local ldap_backup=/var/backups/slapcat.$backup_ext.gz
665   local old_umask result
666
667   if [ ! -f $ldap_backup ]; then
668     [ -x /etc/init.d/freeradius ] && /etc/init.d/freeradius stop || true
669
670     if [ -x /etc/init.d/slapd -a -d /var/lib/ldap ]; then
671       /etc/init.d/slapd stop || true
672       old_umask=`umask`
673       umask 0077 # protect the backup file
674       slapcat | gzip --best > $ldap_backup
675       result=$?
676       umask $old_umask
677       /etc/init.d/slapd start || true
678
679       if [ "$result" -eq 0 ]; then
680         log "$ldap_backup created"
681         notice "Napravljen backup slapd baze u $ldap_backup."
682       else
683         log "$ldap_backup was not created"
684         notice "Backup slapd baze u $ldap_backup nije uspio."
685         exit 1
686       fi
687     fi
688
689     [ -x /etc/init.d/freeradius ] && /etc/init.d/freeradius start || true
690   else
691     log "slapd backup already at $ldap_backup, skipping."
692     #notice "Backup slapd baze u $ldap_backup vec postoji!"
693   fi
694 }
695
696 # backup etc direktorija
697 backup_etc_dir () {
698   local etc_backup=/var/backups/etc.$backup_ext.tar.gz
699   local old_umask
700
701   if [ ! -f $etc_backup ]; then
702     old_umask=`umask`
703     umask 0077 # protect the backup file
704
705     if tar cfz $etc_backup /etc 2> /dev/null; then
706       log "$etc_backup created"
707       notice "Napravljen backup /etc direktorija u $etc_backup."
708     else
709       log "$etc_backup was not created"
710       notice "Backup /etc direktorija u $etc_backup nije uspio."
711       exit 1
712     fi
713     umask $old_umask
714   else
715     log "/etc backup already present in $etc_backup, skipping."
716     #notice "Backup /etc direktorija u $etc_backup vec postoji!"
717   fi
718 }
719
720 # fixaj /etc/default/raid2
721 fix_etc_default_raid2 () {
722   if [ -f /etc/default/raid2 ]; then
723     sed "s/^AUTOSTART.*/AUTOSTART=true/g" /etc/default/raid2 > /etc/default/raid2.$backup_ext.$$
724     chown --reference=/etc/default/raid2 /etc/default/raid2.$backup_ext.$$
725     chmod --reference=/etc/default/raid2 /etc/default/raid2.$backup_ext.$$
726
727     if ! cmp -s /etc/default/raid2.$backup_ext.$$ /etc/default/raid2; then
728       log "Fixed /etc/default/raid2 AUTOSTART option."
729       mv -v /etc/default/raid2.$backup_ext.$$ /etc/default/raid2
730     else
731       rm -v /etc/default/raid2.$backup_ext.$$
732     fi
733   fi
734 }
735
736 # restore a configuration file if it contains only CN modifications
737 restore_file () {
738   local file file_expect file_restore file_backup
739
740   file=$1
741   file_expect=$2
742   file_restore=$3
743   file_backup=$file.$backup_ext
744
745   [ "$file_expect" ] || file_expect=$cnup/files/$file.expect
746   [ "$file_restore" ] || file_restore=$cnup/files/$file.restore
747
748   # check if all required files are there
749   if ! [ -e $file -a -e $file_expect -a -e $file_restore ]; then
750     log "Required files missing, skipping $file restore."
751     return 1
752   fi 
753
754   # is restore needed
755   if ! cmp $file_expect $file >/dev/null; then
756     log "Config doesn't match the template, skipping $file restore."
757     return 1
758   fi
759
760   # backup and restore
761   if [ ! -e $file_backup ]; then
762     log "Restoring config file $file."
763
764     # backup local changes
765     cp -av $file $file_backup
766
767     # restore original
768     cp -v $file_restore $file
769
770     # restore succedded
771     return 0
772   else 
773     log "Backup file exists, skipping $file restore."
774     return 1
775   fi
776 }
777
778 # force reconfiguration at the end if package is not upgraded automatically
779 postupgrade_reconfigure () {
780   local cn_package cn_version
781
782   cn_package=$1
783   cn_version=$( dpkg -s "$cn_package" | awk '/^Version:/ {print $2}' )
784   post_upgrade "pkg $cn_package gt $cn_version || dpkg-reconfigure $cn_package"
785 }
786
787 # copy template into a temporary file
788 copy_template () {
789   local path file template config_new
790
791   path=$1
792   file=$(basename $path)
793   template=$cnup/files/$path.template
794   config_new=$(mktemp /var/lib/carnet-upgrade/$file.XXXXXX)
795   cp $template $config_new
796
797   echo $config_new
798 }
799
800 # restore modified config to their package defaults
801 # so the upgrade doesn't complain so much
802 restore_configs () {
803   local hostname domain config_new memtotal memlimit
804
805   # restore simple configs
806   if pkg kernel-2.6-cn && pkg procps lt 1:3.2.7-11; then
807      if restore_file /etc/sysctl.conf; then
808         postupgrade_reconfigure kernel-2.6-cn
809      fi
810   fi
811
812   if pkg ntp-cn && pkg ntp lt 1:4.2.4p4+dfsg-8; then
813      if restore_file /etc/ntp.conf; then
814         postupgrade_reconfigure ntp-cn
815      fi
816   fi
817
818   if pkg apache2-cn && pkg apache2.2-common lt 2.2.9-10+lenny2; then
819      if restore_file /etc/apache2/ports.conf; then
820         postupgrade_reconfigure apache2-cn
821      fi
822   fi
823
824   if pkg postfix-cn && pkg sasl2-bin lt 2.1.22.dfsg1-23; then
825      if restore_file /etc/default/saslauthd; then
826         postupgrade_reconfigure postfix-cn
827      fi
828   fi
829
830   if pkg bind9-cn && pkg bind9 lt 1:9.5.1.dfsg.P1-1; then
831      if restore_file /etc/bind/named.conf.options; then
832         postupgrade_reconfigure bind9-cn
833      fi
834   fi
835
836   if pkg amavisd-cn && pkg amavisd-new lt 1:2.6.1.dfsg-1; then
837      if restore_file /etc/cron.daily/amavisd-new; then
838         rm -f /etc/cron.daily/amavisd-new.$backup_ext
839         postupgrade_reconfigure amavisd-cn
840      fi
841   fi
842
843   if pkg spamassassin-cn && pkg razor lt 1:2.85-1; then
844      if restore_file /etc/razor/razor-agent.conf; then
845         postupgrade_reconfigure spamassassin-cn
846      fi
847   fi
848
849   if pkg kernel-2.6-cn && pkg libpam-modules lt 1.0.1-5; then
850      if restore_file /etc/security/limits.conf; then
851         postupgrade_reconfigure kernel-2.6-cn
852      fi
853   fi
854
855   if pkg samba-cn && pkg samba-common lt 2:3.2.5-4lenny2; then
856      if restore_file /etc/samba/smb.conf; then
857         postupgrade_reconfigure samba-cn
858      fi
859   fi
860
861   if pkg base-files lt 5lenny2; then
862      restore_file /etc/issue
863      restore_file /etc/issue.net
864   fi
865
866   # restore php.ini
867   memtotal=$(awk '$1 == "MemTotal:" { print int($2/1024) }' /proc/meminfo)
868   if [ $memtotal -gt 1024 ]; then
869     memlimit=64
870   elif [ $memtotal -ge 512 ]; then
871     memlimit=32
872   else
873     memlimit=16
874   fi
875   
876   if pkg php5-cn && pkg php5-cli lt 5.2.6.dfsg.1-1+lenny2; then
877      config_new=$(copy_template /etc/php5/cli/php.ini)
878      sed -i "s/^[[:space:]]*memory_limit[[:space:]]*=.*/memory_limit = ${memlimit}M/" $config_new
879      sed -i "s/^[[:space:]]*post_max_size[[:space:]]*=.*/post_max_size = ${memlimit}M/" $config_new
880      if restore_file /etc/php5/cli/php.ini $config_new; then
881         postupgrade_reconfigure php5-cn
882      fi
883      rm -f $config_new
884   fi
885
886   if pkg php5-cn && pkg libapache2-mod-php5 lt 5.2.6.dfsg.1-1+lenny2; then
887      config_new=$(copy_template /etc/php5/apache2/php.ini)
888      sed -i "s/^[[:space:]]*memory_limit[[:space:]]*=.*/memory_limit = ${memlimit}M/" $config_new
889      sed -i "s/^[[:space:]]*post_max_size[[:space:]]*=.*/post_max_size = ${memlimit}M/" $config_new
890      if restore_file /etc/php5/apache2/php.ini $config_new; then
891         postupgrade_reconfigure php5-cn
892      fi
893      rm -f $config_new
894   fi
895
896   # check if monitrc is template based
897   if [ -f /etc/monit/monitrc ]; then
898      # regenerate config from template
899      hostname=$(hostname)
900      domain=$(hostname --domain)
901      config_new=$(copy_template /etc/monit/monitrc)
902      check_and_sed "@localhost" \
903        "s/@localhost/@$hostname.$domain/g" $config_new || true
904      restore_file /etc/monit/monitrc $config_new
905      rm -f $config_new
906   fi
907
908 #  restore_config login 1:4.0.18.1 /etc/pam.d/login
909 #  restore_config mysql-server 5.0.3 /etc/init.d/mysql \
910 #                                    /etc/logrotate.d/mysql-server \
911 #                                    /etc/mysql/my.cnf
912 #  restore_config ntpdate 1:4.2.2 /etc/default/ntpdate
913 #  restore_config oidentd 2.0.8 /etc/default/oidentd
914 #  restore_config slapd 2.3.30 /etc/default/slapd /etc/init.d/slapd
915 #  restore_config squirrelmail 2:1.4.9a /etc/squirrelmail/apache.conf
916 #  restore_config vsftpd 2.0.5 /etc/vsftpd.conf
917 #  restore_config xinetd 1:2.3.14 /etc/xinetd.conf
918 #
919 #  # orphaned config file - no owner
920 #  if pkg logcheck-database lt 1.2.54; then
921 #      rm -vf /etc/logcheck/ignore.d.server/imap
922 #  fi
923 #
924 #  # aide switched to ucf, move old configs aside
925 #  if pkg aide lt 0.13.1; then
926 #      for file in /etc/aide/aide.conf \
927 #                  /etc/cron.daily/aide \
928 #                  /etc/default/aide;
929 #      do
930 #          if [ ! -e "$file.$backup_ext" ]; then
931 #              mv -v "$file" "$file.$backup_ext"
932 #          fi
933 #
934 #          rm -vf "$file"
935 #      done
936 #  fi
937 #
938 #  # register cn changes in ucf managed files
939 #  for config in /etc/apache/modules.conf \
940 #                /etc/clamav/clamd.conf \
941 #                /etc/clamav/freshclam.conf
942 #  do 
943 #      config_backup=$config.$backup_ext
944 #      if [ -e $config -a ! -e $config_backup ]; then
945 #          log "Saving config file $config"
946 #          cp -av $config $config_backup
947 #          ucf $config_backup $config
948 #      fi
949 #  done
950 }
951
952 # temporary disable listchanges packages to reduce clutter during upgrade
953 apt_listchanges () {
954   local file command=$1
955
956   for file in /etc/apt/apt.conf.d/20listchanges \
957               /etc/apt/apt.conf.d/20listchanges-cn
958   do
959       case $command in
960            disable)
961                if [ -f $file ]; then
962                    dpkg-divert --local --rename --divert $file.$backup_ext \
963                                --add $file || true
964                fi
965                ;;
966
967            enable)
968                if [ -f $file.$backup_ext ]; then
969                    dpkg-divert --remove $file || true
970                fi
971                ;;
972      esac
973   done
974 }
975
976 # make a silent installation of carnet and srce keyrings
977 install_keyrings () {
978   pkgadd carnet-keyring srce-keyring debian-archive-keyring
979   pkgadd dpkg apt debconf python-apt
980
981   dpkg-reconfigure carnet-keyring srce-keyring debian-archive-keyring
982   apt_update
983 }
984
985 # remove skey (not supported anymore)
986 remove_skey () {
987   pkgrm skey-cn libpam-skey
988
989   if getent group skey > /dev/null; then
990     groupdel skey || true
991     log "groupdel skey"
992   fi
993 }
994
995 # remove unsupported php version
996 # but leave the configuration just in case
997 remove_php4 () {
998   if pkg php5-cn && pkg php4-common; then
999      pkgrm_only php4-common
1000   fi
1001 }
1002
1003 # check if package is orphaned (nothing depends on it)
1004 is_orphaned () {
1005   local package deps
1006
1007   package=$1
1008   deps=$(apt-get remove -s $package | grep ^Remv | wc -l)
1009   if [ "$deps" -eq 1 ]; then 
1010     return 0
1011   else 
1012     return 1
1013   fi
1014 }
1015
1016 # remove old and unused libraries
1017 remove_orphaned () {
1018   local package
1019
1020   for package in $orphaned_packages; do
1021     if is_orphaned $package; then
1022       pkgrm $package
1023     fi
1024   done
1025 }
1026
1027 # monit it causing problems for postinst scripts
1028 # restarting daemons so try to disable it
1029 disable_monit () {
1030   if pkg monit-cn; then
1031     pkgrm monit-cn # disables monit in prerm
1032   fi
1033
1034   # if configured manually
1035   if pkg monit; then
1036     # remove the binary but leave the config
1037     apt-get remove --yes monit
1038     # stop the binary
1039     pkill -9 monit || true
1040   fi
1041 }
1042
1043 # make a silent upgrade to new libc6
1044 upgrade_libc () {
1045   DEBIAN_FRONTEND=noninteractive pkgadd libc6 tzdata
1046 }
1047
1048 # upgrade apache2/php5 to lenny
1049 upgrade_apache2 () {
1050   local package packages
1051
1052   for package in \
1053       libapache2-mod-php5 \
1054       php5-cli \
1055       php5-cn \
1056       apache2-cn \
1057       php5-odbc \
1058       php5-suhosin \
1059       php-suhosin-cn
1060   do
1061       if pkg $package; then
1062           packages="$packages $package"
1063       fi
1064   done
1065
1066   eval pkgadd $packages
1067 }
1068
1069 upgrade_amavis () {
1070   local conf
1071
1072   # remove init script diversion before upgrade
1073   if pkg amavisd-cn lt 3:2.6.1-1; then 
1074     if [ -L /etc/init.d/amavis -a -f /etc/init.d/amavis.amavisd-new ]; then
1075       rm -vf /etc/init.d/amavis
1076       dpkg-divert --remove /etc/init.d/amavis
1077     fi
1078   fi
1079
1080   # move old config aside
1081   for conf in /etc/amavisd.conf /etc/amavis/amavisd.conf; do
1082     if [ -e $conf -a ! -e $conf.$backup_ext ]; then
1083       mv -v $conf $conf.$backup_ext
1084     fi
1085   done
1086
1087   # install new packages
1088   pkgadd amavisd-cn amavisd-new
1089
1090   # fix new packages
1091   check_and_sed '^clamd.*5.clamav.log$' \
1092                 's/^\(clamd.*\)5.clamav.log$/\14\tsocket/g' \
1093                 /etc/init.d/amavisd-cn || true
1094
1095   # check if link got lost
1096   if pkg amavisd-cn && [ ! -L /etc/init.d/amavis ]; then
1097     apt-get install --reinstall --yes amavisd-cn
1098   fi
1099
1100   # start new packages
1101   /etc/init.d/amavis restart
1102 }
1103
1104 # handle freerdius config files upgrade
1105 upgrade_freeradius () {
1106   local template config_new password realm hostname basedn
1107
1108   pkg freeradius-aai lt 2.1.3-0lenny0 || return 0
1109
1110   # handle static configs
1111   restore_file /etc/freeradius/clients.conf
1112   restore_file /etc/freeradius/hints
1113   restore_file /etc/freeradius/ldap.attrmap
1114
1115   # handle template based configs
1116   basedn=$(sed -n '/^[[:space:]]*suffix[[:space:]]*/ { 
1117     s///; s/"//g; p; q }' /etc/ldap/slapd.conf)
1118   hostname=$(hostname -f)
1119   config_new=$(copy_template /etc/freeradius/radiusd.conf)
1120   sed -i "s/#HOSTNAME#/$hostname/" $config_new
1121   sed -i "s/#BASEDN#/$basedn/" $config_new
1122   restore_file /etc/freeradius/radiusd.conf $config_new
1123   rm -f $config_new
1124
1125   password=$(grep -s '^[[:space:]]*private_key_password[[:space:]]*=' \
1126     /etc/freeradius/eap.conf)
1127   config_new=$(copy_template /etc/freeradius/eap.conf)
1128   sed -i "s/.*#PASSWORD#.*/$password/" $config_new
1129   restore_file /etc/freeradius/eap.conf $config_new
1130   rm -f $config_new
1131   
1132   realm=$(sed -n '/^[[:space:]]*suffix[[:space:]]*/ { 
1133     s///; s/"//g; s/,dc=/./g; s/dc=//; s/.hr$//; p; q }' /etc/ldap/slapd.conf)
1134   config_new=$(copy_template /etc/freeradius/proxy.conf)
1135   sed -i "s/#REALM#/$realm/" $config_new
1136   restore_file /etc/freeradius/proxy.conf $config_new
1137   rm -f $config_new
1138
1139   # install the new packages
1140   if ! apt-get -y install freeradius; then
1141      # freeradius upgrade fails here, try to fix it by creating certificates
1142      [ -x /etc/freeradius/certs/bootstrap ] && /etc/freeradius/certs/bootstrap
1143      
1144      dpkg --configure -a
1145      apt-get -y -f install
1146   fi
1147
1148   # finally try to cleanup this mess
1149   pkgadd freeradius-aai
1150 }
1151
1152 # prepare for slapd upgrade (caused by freeradius installation)
1153 prepare_openldap () {
1154   pkg openldap-aai lt 2.4 || return 0
1155
1156   # openldap-aai expects org.ldif to exists
1157   local ldif=/var/lib/ldap/org.ldif
1158   local ldif_backup=/var/lib/carnet-upgrade/org.ldif
1159   if [ -f $ldif ]; then
1160     cp -v $ldif $ldif_backup
1161   fi
1162 }
1163
1164 # fix openldap-aai postinst user handling
1165 upgrade_openldap () {
1166   pkg openldap-aai lt 2.4 || return 0
1167
1168   # slapd postinst fails if move_old_database is false
1169   echo 'slapd slapd/move_old_database boolean true' | debconf-set-selections 
1170   pkgadd slapd
1171
1172   # slapd removes org.ldif during switch from ldap -> openldap user
1173   local ldif=/var/lib/ldap/org.ldif
1174   local ldif_backup=/var/lib/carnet-upgrade/org.ldif
1175   if [ -f $ldif_backup -a ! -f $ldif ]; then
1176     cp -v $ldif_backup $ldif
1177   fi
1178
1179   # openldap-aai postinst fails here because of database permissions
1180   apt-get -y install openldap-aai || true
1181   chown -R openldap:openldap /var/lib/ldap
1182   /etc/init.d/slapd restart
1183   dpkg --configure -a
1184
1185   # remove ldap user if openldap works
1186   if getent passwd ldap      > /dev/null && \
1187      getent passwd openldap  > /dev/null && \
1188      pgrep -U openldap slapd > /dev/null
1189   then
1190      log "userdel ldap (replaced by openldap)"
1191      userdel ldap || true
1192   fi
1193 }
1194
1195 # fix postgresql deinstallation during upgrade
1196 upgrade_postgresql () {
1197   if check_services show 2>&1 | grep -qw postgresql; then
1198      apt-get install -y postgresql || apt-get -yf install
1199   fi
1200 }
1201
1202 # upgrade the MTA
1203 upgrade_postfix () {
1204   pkgadd sasl2-bin postfix-cn
1205
1206   if pkg postgrey; then
1207      pkgrm postgrey # not used by postfix-cn anymore
1208   fi
1209 }
1210
1211 get_variable () {
1212   local name=$1 file=$2 val
1213
1214   if [ -f "$file" ]; then
1215      val=$(sed -n "/^[[:space:]]*$name[[:space:]]*=/s/[[:space:]]*//gp" $file)
1216      val=${val#*=}
1217   fi
1218
1219   echo $val
1220 }
1221
1222 # preserve mail location on upgrade
1223 upgrade_dovecot () {
1224   local default_mail_env mail_location
1225
1226   # check if already upgraded
1227   if ! pkg dovecot-common lt 1.0; then
1228       return
1229   fi
1230
1231   # save old mail location
1232   default_mail_env=$(get_variable default_mail_env /etc/dovecot/dovecot.conf)
1233
1234   # silently upgrade package
1235   restore_file /etc/dovecot/dovecot.conf
1236   pkgadd dovecot-cn
1237
1238   # insert old mail location
1239   mail_location=$(get_variable mail_location /etc/dovecot/dovecot.conf)
1240   if [ "$default_mail_env" -a ! "$mail_location" ]; then
1241       check_and_sed '^#mail_location[[:space:]]*=' \
1242                    "s|#mail_location[[:space:]]*=.*|mail_location = $default_mail_env|" /etc/dovecot/dovecot.conf || return 0
1243
1244       if [ -x /etc/init.d/dovecot ]; then
1245           if [ -x /usr/sbin/invoke-rc.d ]; then
1246               /usr/sbin/invoke-rc.d dovecot restart
1247           else
1248               /etc/init.d/dovecot restart
1249           fi
1250       fi
1251   fi
1252 }
1253
1254 # restore munin configs for silent upgrade
1255 upgrade_munin () {
1256   local config backup hostname domain
1257
1258   for pkg_name in munin munin-node; do
1259      pkg $pkg_name lt '1.2.6-10~lenny1' || continue
1260
1261      config=/etc/munin/$pkg_name.conf
1262      backup=$config.$backup_ext
1263      if [ -f $config -a ! -f $backup ]; then
1264         hostname=$(hostname)
1265         domain=$(hostname --domain)
1266
1267         cp -av $config $backup
1268         check_and_sed "$hostname.$domain" \
1269           "s/$hostname.$domain/localhost.localdomain/g" $config
1270      fi
1271      pkgadd $pkg_name
1272   done
1273
1274   # force localhost -> hostname replacement in config files
1275   pkgadd munin-cn
1276   dpkg-reconfigure munin-cn
1277 }
1278
1279 # udev.preinst fails if devfs configs still exist
1280 upgrade_udev () {
1281   local config
1282
1283   for config in /etc/udev/rules.d/devfs.rules \
1284                 /etc/udev/rules.d/compat-full.rules \
1285                 /etc/udev/rules.d/compat.rules
1286   do
1287      if [ -e $config ]; then
1288         rm -v -f $config
1289      fi
1290   done
1291
1292   if [ -f /etc/udev/udev.rules -a ! -f /etc/udev/rules.d/udev.rules ]; then
1293      ln -sfv ../udev.rules /etc/udev/rules.d/
1294   fi
1295
1296   pkgadd udev
1297 }
1298
1299 # fix /etc/dpkg/dpkg.cfg
1300 comment_force_overwrite () {
1301   if [ -f /etc/dpkg/dpkg.cfg ]; then
1302     if check_and_sed '^force-overwrite' \
1303                   's,\(^force-overwrite\),#\1,' \
1304                   /etc/dpkg/dpkg.cfg; then
1305       notice "$msg_comment_force_overwrite"
1306     fi
1307   fi
1308 }
1309
1310 fix_issue () {
1311   cat > /etc/issue <<EOF
1312 Debian GNU/Linux 5.0 (CARNet Debian 5.0) \\n \\l
1313
1314 EOF
1315
1316   cat > /etc/issue.net <<EOF
1317 Debian GNU/Linux 5.0 (CARNet Debian 5.0) %h
1318 EOF
1319 }
1320
1321 check_archives_space() {
1322   local available_disk_space archives_size
1323   available_disk_space=$(free_space /var/cache/apt/archives/)
1324   archives_size=$(download_size dist-upgrade)
1325
1326   if [ $? -ne 0 ]; then
1327     log "Failed to get download size, skipping check."
1328     return 0
1329   fi
1330
1331   log "Archives space: ${available_disk_space}MB"
1332   log "Download size: ${archives_size}MB"
1333
1334   if [ $available_disk_space -lt $archives_size ]; then
1335     log "carnet-upgrade aborted: not enough space in /var/cache/apt/archives/ (have ${available_disk_space}MB, need ${archives_size}MB)"
1336     notice "Premalo slobodnog mjesta u /var/cache/apt/archives/ za skidanje paketa (potrebno je barem ${archives_size}MB). Procitajte poglavlje 'Izvanredne situacije' u dokumentaciji."
1337     exit 1
1338   fi
1339 }
1340
1341 check_kernel_space() {
1342   local available_disk_space kernel_size
1343   available_disk_space=$(free_space /)
1344   kernel_size=$(installed_size kernel-2.6-cn)
1345
1346   if [ $? -ne 0 ]; then
1347     log "Failed to get installed size, skipping check."
1348     return 0
1349   fi
1350
1351   log "Free space in /: ${available_disk_space}MB"
1352   log "Kernel size: ${kernel_size}MB"
1353
1354   if LC_ALL=C apt-get --yes --print-uris install kernel-2.6-cn \
1355      | grep -q '^kernel-2.6-cn is already the newest version.$'
1356   then
1357      log "Not checking free space in /: kernel-2.6-cn already installed"
1358      return
1359   fi
1360
1361   if [ $available_disk_space -lt $kernel_size ]; then
1362     log "carnet-upgrade aborted: not enough space in / (have ${available_disk_space}MB, need ${kernel_size}MB)"
1363     notice "Premalo mjesta u / za instalaciju kernela (potrebno je barem ${kernel_size}MB). Procitajte poglavlje 'Izvanredne situacije' u dokumentaciji."
1364     exit 1
1365   fi
1366 }
1367
1368 free_space() {
1369   df --portability --block-size=1M "$1" | tail -1 | awk '{print $4}'
1370 }
1371
1372 download_size() {
1373   local size
1374   size=$(LC_ALL=C apt-get --yes --print-uris "$@" | grep '^Need to get ')
1375
1376   if [ $? -ne 0 ]; then
1377     log "Failed to get download size for 'apt-get $@'."
1378     return 1
1379   fi
1380
1381   size=${size#Need to get }
1382   size=${size%%B*}
1383
1384   size_to_mb "${size}B"
1385 }
1386
1387 installed_size() {
1388   local package size
1389   package=$1
1390   size=$(apt-cache show $package | grep '^Installed-Size:')
1391
1392   if [ $? -ne 0 ]; then
1393     log "Failed to get installed size for '$package'."
1394     return 1
1395   fi
1396
1397   size=${size#*: }
1398
1399   size_to_mb "${size}kB"
1400 }
1401
1402 size_to_mb() {
1403   local size
1404   size=$1
1405   if [ -z "$size" ]; then
1406     size=0;
1407   fi
1408
1409   if [ "${size%kB}" != "$size" ]; then
1410     size=$(round ${size%kB})
1411     size=$(($size/1024+1))
1412   elif [ "${size%MB}" != "$size" ]; then
1413     size=$(round ${size%MB})
1414   elif [ "${size%GB}" != "$size" ]; then
1415     size=$(round ${size%GB})
1416     size=$(($size*1024))
1417   elif [ "${size%[0-9]B}" != "$size" ]; then
1418     size=$(round ${size%B})
1419     size=$(($size/1024/1024+1))
1420   else
1421     log "unrecognized size unit: $size"
1422     return 1
1423   fi
1424
1425   echo $size
1426 }
1427
1428 round() {
1429   local num
1430   num=$1
1431
1432   if [ "${num%.*}" != "$num" ]; then
1433     num=${num%.*}
1434     num=$(($num+1))
1435   fi
1436
1437   echo $num
1438 }